]> git.proxmox.com Git - qemu.git/blame - block.c
slavio_misc: QOM cast cleanup for APCState
[qemu.git] / block.c
CommitLineData
fc01f7e7
FB
1/*
2 * QEMU System Emulator block driver
5fafdf24 3 *
fc01f7e7 4 * Copyright (c) 2003 Fabrice Bellard
5fafdf24 5 *
fc01f7e7
FB
6 * Permission is hereby granted, free of charge, to any person obtaining a copy
7 * of this software and associated documentation files (the "Software"), to deal
8 * in the Software without restriction, including without limitation the rights
9 * to use, copy, modify, merge, publish, distribute, sublicense, and/or sell
10 * copies of the Software, and to permit persons to whom the Software is
11 * furnished to do so, subject to the following conditions:
12 *
13 * The above copyright notice and this permission notice shall be included in
14 * all copies or substantial portions of the Software.
15 *
16 * THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
17 * IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
18 * FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL
19 * THE AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
20 * LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,
21 * OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN
22 * THE SOFTWARE.
23 */
3990d09a 24#include "config-host.h"
faf07963 25#include "qemu-common.h"
6d519a5f 26#include "trace.h"
83c9089e 27#include "monitor/monitor.h"
737e150e
PB
28#include "block/block_int.h"
29#include "block/blockjob.h"
1de7afc9 30#include "qemu/module.h"
7b1b5d19 31#include "qapi/qmp/qjson.h"
9c17d615 32#include "sysemu/sysemu.h"
1de7afc9 33#include "qemu/notify.h"
737e150e 34#include "block/coroutine.h"
b2023818 35#include "qmp-commands.h"
1de7afc9 36#include "qemu/timer.h"
fc01f7e7 37
71e72a19 38#ifdef CONFIG_BSD
7674e7bf
FB
39#include <sys/types.h>
40#include <sys/stat.h>
41#include <sys/ioctl.h>
72cf2d4f 42#include <sys/queue.h>
c5e97233 43#ifndef __DragonFly__
7674e7bf
FB
44#include <sys/disk.h>
45#endif
c5e97233 46#endif
7674e7bf 47
49dc768d
AL
48#ifdef _WIN32
49#include <windows.h>
50#endif
51
1c9805a3
SH
52#define NOT_DONE 0x7fffffff /* used while emulated sync operation in progress */
53
470c0504
SH
54typedef enum {
55 BDRV_REQ_COPY_ON_READ = 0x1,
f08f2dda 56 BDRV_REQ_ZERO_WRITE = 0x2,
470c0504
SH
57} BdrvRequestFlags;
58
7d4b4ba5 59static void bdrv_dev_change_media_cb(BlockDriverState *bs, bool load);
f141eafe
AL
60static BlockDriverAIOCB *bdrv_aio_readv_em(BlockDriverState *bs,
61 int64_t sector_num, QEMUIOVector *qiov, int nb_sectors,
c87c0672 62 BlockDriverCompletionFunc *cb, void *opaque);
f141eafe
AL
63static BlockDriverAIOCB *bdrv_aio_writev_em(BlockDriverState *bs,
64 int64_t sector_num, QEMUIOVector *qiov, int nb_sectors,
ce1a14dc 65 BlockDriverCompletionFunc *cb, void *opaque);
f9f05dc5
KW
66static int coroutine_fn bdrv_co_readv_em(BlockDriverState *bs,
67 int64_t sector_num, int nb_sectors,
68 QEMUIOVector *iov);
69static int coroutine_fn bdrv_co_writev_em(BlockDriverState *bs,
70 int64_t sector_num, int nb_sectors,
71 QEMUIOVector *iov);
c5fbe571 72static int coroutine_fn bdrv_co_do_readv(BlockDriverState *bs,
470c0504
SH
73 int64_t sector_num, int nb_sectors, QEMUIOVector *qiov,
74 BdrvRequestFlags flags);
1c9805a3 75static int coroutine_fn bdrv_co_do_writev(BlockDriverState *bs,
f08f2dda
SH
76 int64_t sector_num, int nb_sectors, QEMUIOVector *qiov,
77 BdrvRequestFlags flags);
b2a61371
SH
78static BlockDriverAIOCB *bdrv_co_aio_rw_vector(BlockDriverState *bs,
79 int64_t sector_num,
80 QEMUIOVector *qiov,
81 int nb_sectors,
82 BlockDriverCompletionFunc *cb,
83 void *opaque,
8c5873d6 84 bool is_write);
b2a61371 85static void coroutine_fn bdrv_co_do_rw(void *opaque);
621f0589
KW
86static int coroutine_fn bdrv_co_do_write_zeroes(BlockDriverState *bs,
87 int64_t sector_num, int nb_sectors);
ec530c81 88
98f90dba
ZYW
89static bool bdrv_exceed_bps_limits(BlockDriverState *bs, int nb_sectors,
90 bool is_write, double elapsed_time, uint64_t *wait);
91static bool bdrv_exceed_iops_limits(BlockDriverState *bs, bool is_write,
92 double elapsed_time, uint64_t *wait);
93static bool bdrv_exceed_io_limits(BlockDriverState *bs, int nb_sectors,
94 bool is_write, int64_t *wait);
95
1b7bdbc1
SH
96static QTAILQ_HEAD(, BlockDriverState) bdrv_states =
97 QTAILQ_HEAD_INITIALIZER(bdrv_states);
7ee930d0 98
8a22f02a
SH
99static QLIST_HEAD(, BlockDriver) bdrv_drivers =
100 QLIST_HEAD_INITIALIZER(bdrv_drivers);
ea2384d3 101
eb852011
MA
102/* If non-zero, use only whitelisted block drivers */
103static int use_bdrv_whitelist;
104
9e0b22f4
SH
105#ifdef _WIN32
106static int is_windows_drive_prefix(const char *filename)
107{
108 return (((filename[0] >= 'a' && filename[0] <= 'z') ||
109 (filename[0] >= 'A' && filename[0] <= 'Z')) &&
110 filename[1] == ':');
111}
112
113int is_windows_drive(const char *filename)
114{
115 if (is_windows_drive_prefix(filename) &&
116 filename[2] == '\0')
117 return 1;
118 if (strstart(filename, "\\\\.\\", NULL) ||
119 strstart(filename, "//./", NULL))
120 return 1;
121 return 0;
122}
123#endif
124
0563e191 125/* throttling disk I/O limits */
98f90dba
ZYW
126void bdrv_io_limits_disable(BlockDriverState *bs)
127{
128 bs->io_limits_enabled = false;
129
b681a1c7 130 do {} while (qemu_co_enter_next(&bs->throttled_reqs));
98f90dba
ZYW
131
132 if (bs->block_timer) {
133 qemu_del_timer(bs->block_timer);
134 qemu_free_timer(bs->block_timer);
135 bs->block_timer = NULL;
136 }
137
138 bs->slice_start = 0;
139 bs->slice_end = 0;
98f90dba
ZYW
140}
141
0563e191
ZYW
142static void bdrv_block_timer(void *opaque)
143{
144 BlockDriverState *bs = opaque;
145
b681a1c7 146 qemu_co_enter_next(&bs->throttled_reqs);
0563e191
ZYW
147}
148
149void bdrv_io_limits_enable(BlockDriverState *bs)
150{
151 qemu_co_queue_init(&bs->throttled_reqs);
152 bs->block_timer = qemu_new_timer_ns(vm_clock, bdrv_block_timer, bs);
0563e191
ZYW
153 bs->io_limits_enabled = true;
154}
155
156bool bdrv_io_limits_enabled(BlockDriverState *bs)
157{
158 BlockIOLimit *io_limits = &bs->io_limits;
159 return io_limits->bps[BLOCK_IO_LIMIT_READ]
160 || io_limits->bps[BLOCK_IO_LIMIT_WRITE]
161 || io_limits->bps[BLOCK_IO_LIMIT_TOTAL]
162 || io_limits->iops[BLOCK_IO_LIMIT_READ]
163 || io_limits->iops[BLOCK_IO_LIMIT_WRITE]
164 || io_limits->iops[BLOCK_IO_LIMIT_TOTAL];
165}
166
98f90dba
ZYW
167static void bdrv_io_limits_intercept(BlockDriverState *bs,
168 bool is_write, int nb_sectors)
169{
170 int64_t wait_time = -1;
171
172 if (!qemu_co_queue_empty(&bs->throttled_reqs)) {
173 qemu_co_queue_wait(&bs->throttled_reqs);
174 }
175
176 /* In fact, we hope to keep each request's timing, in FIFO mode. The next
177 * throttled requests will not be dequeued until the current request is
178 * allowed to be serviced. So if the current request still exceeds the
179 * limits, it will be inserted to the head. All requests followed it will
180 * be still in throttled_reqs queue.
181 */
182
183 while (bdrv_exceed_io_limits(bs, nb_sectors, is_write, &wait_time)) {
184 qemu_mod_timer(bs->block_timer,
185 wait_time + qemu_get_clock_ns(vm_clock));
186 qemu_co_queue_wait_insert_head(&bs->throttled_reqs);
187 }
188
189 qemu_co_queue_next(&bs->throttled_reqs);
190}
191
9e0b22f4
SH
192/* check if the path starts with "<protocol>:" */
193static int path_has_protocol(const char *path)
194{
947995c0
PB
195 const char *p;
196
9e0b22f4
SH
197#ifdef _WIN32
198 if (is_windows_drive(path) ||
199 is_windows_drive_prefix(path)) {
200 return 0;
201 }
947995c0
PB
202 p = path + strcspn(path, ":/\\");
203#else
204 p = path + strcspn(path, ":/");
9e0b22f4
SH
205#endif
206
947995c0 207 return *p == ':';
9e0b22f4
SH
208}
209
83f64091 210int path_is_absolute(const char *path)
3b0d4f61 211{
21664424
FB
212#ifdef _WIN32
213 /* specific case for names like: "\\.\d:" */
f53f4da9 214 if (is_windows_drive(path) || is_windows_drive_prefix(path)) {
21664424 215 return 1;
f53f4da9
PB
216 }
217 return (*path == '/' || *path == '\\');
3b9f94e1 218#else
f53f4da9 219 return (*path == '/');
3b9f94e1 220#endif
3b0d4f61
FB
221}
222
83f64091
FB
223/* if filename is absolute, just copy it to dest. Otherwise, build a
224 path to it by considering it is relative to base_path. URL are
225 supported. */
226void path_combine(char *dest, int dest_size,
227 const char *base_path,
228 const char *filename)
3b0d4f61 229{
83f64091
FB
230 const char *p, *p1;
231 int len;
232
233 if (dest_size <= 0)
234 return;
235 if (path_is_absolute(filename)) {
236 pstrcpy(dest, dest_size, filename);
237 } else {
238 p = strchr(base_path, ':');
239 if (p)
240 p++;
241 else
242 p = base_path;
3b9f94e1
FB
243 p1 = strrchr(base_path, '/');
244#ifdef _WIN32
245 {
246 const char *p2;
247 p2 = strrchr(base_path, '\\');
248 if (!p1 || p2 > p1)
249 p1 = p2;
250 }
251#endif
83f64091
FB
252 if (p1)
253 p1++;
254 else
255 p1 = base_path;
256 if (p1 > p)
257 p = p1;
258 len = p - base_path;
259 if (len > dest_size - 1)
260 len = dest_size - 1;
261 memcpy(dest, base_path, len);
262 dest[len] = '\0';
263 pstrcat(dest, dest_size, filename);
3b0d4f61 264 }
3b0d4f61
FB
265}
266
dc5a1371
PB
267void bdrv_get_full_backing_filename(BlockDriverState *bs, char *dest, size_t sz)
268{
269 if (bs->backing_file[0] == '\0' || path_has_protocol(bs->backing_file)) {
270 pstrcpy(dest, sz, bs->backing_file);
271 } else {
272 path_combine(dest, sz, bs->filename, bs->backing_file);
273 }
274}
275
5efa9d5a 276void bdrv_register(BlockDriver *bdrv)
ea2384d3 277{
8c5873d6
SH
278 /* Block drivers without coroutine functions need emulation */
279 if (!bdrv->bdrv_co_readv) {
f9f05dc5
KW
280 bdrv->bdrv_co_readv = bdrv_co_readv_em;
281 bdrv->bdrv_co_writev = bdrv_co_writev_em;
282
f8c35c1d
SH
283 /* bdrv_co_readv_em()/brdv_co_writev_em() work in terms of aio, so if
284 * the block driver lacks aio we need to emulate that too.
285 */
f9f05dc5
KW
286 if (!bdrv->bdrv_aio_readv) {
287 /* add AIO emulation layer */
288 bdrv->bdrv_aio_readv = bdrv_aio_readv_em;
289 bdrv->bdrv_aio_writev = bdrv_aio_writev_em;
f9f05dc5 290 }
83f64091 291 }
b2e12bc6 292
8a22f02a 293 QLIST_INSERT_HEAD(&bdrv_drivers, bdrv, list);
ea2384d3 294}
b338082b
FB
295
296/* create a new block device (by default it is empty) */
297BlockDriverState *bdrv_new(const char *device_name)
298{
1b7bdbc1 299 BlockDriverState *bs;
b338082b 300
7267c094 301 bs = g_malloc0(sizeof(BlockDriverState));
b338082b 302 pstrcpy(bs->device_name, sizeof(bs->device_name), device_name);
ea2384d3 303 if (device_name[0] != '\0') {
1b7bdbc1 304 QTAILQ_INSERT_TAIL(&bdrv_states, bs, list);
ea2384d3 305 }
28a7282a 306 bdrv_iostatus_disable(bs);
d7d512f6 307 notifier_list_init(&bs->close_notifiers);
d616b224 308 notifier_with_return_list_init(&bs->before_write_notifiers);
d7d512f6 309
b338082b
FB
310 return bs;
311}
312
d7d512f6
PB
313void bdrv_add_close_notifier(BlockDriverState *bs, Notifier *notify)
314{
315 notifier_list_add(&bs->close_notifiers, notify);
316}
317
ea2384d3
FB
318BlockDriver *bdrv_find_format(const char *format_name)
319{
320 BlockDriver *drv1;
8a22f02a
SH
321 QLIST_FOREACH(drv1, &bdrv_drivers, list) {
322 if (!strcmp(drv1->format_name, format_name)) {
ea2384d3 323 return drv1;
8a22f02a 324 }
ea2384d3
FB
325 }
326 return NULL;
327}
328
b64ec4e4 329static int bdrv_is_whitelisted(BlockDriver *drv, bool read_only)
eb852011 330{
b64ec4e4
FZ
331 static const char *whitelist_rw[] = {
332 CONFIG_BDRV_RW_WHITELIST
333 };
334 static const char *whitelist_ro[] = {
335 CONFIG_BDRV_RO_WHITELIST
eb852011
MA
336 };
337 const char **p;
338
b64ec4e4 339 if (!whitelist_rw[0] && !whitelist_ro[0]) {
eb852011 340 return 1; /* no whitelist, anything goes */
b64ec4e4 341 }
eb852011 342
b64ec4e4 343 for (p = whitelist_rw; *p; p++) {
eb852011
MA
344 if (!strcmp(drv->format_name, *p)) {
345 return 1;
346 }
347 }
b64ec4e4
FZ
348 if (read_only) {
349 for (p = whitelist_ro; *p; p++) {
350 if (!strcmp(drv->format_name, *p)) {
351 return 1;
352 }
353 }
354 }
eb852011
MA
355 return 0;
356}
357
b64ec4e4
FZ
358BlockDriver *bdrv_find_whitelisted_format(const char *format_name,
359 bool read_only)
eb852011
MA
360{
361 BlockDriver *drv = bdrv_find_format(format_name);
b64ec4e4 362 return drv && bdrv_is_whitelisted(drv, read_only) ? drv : NULL;
eb852011
MA
363}
364
5b7e1542
ZYW
365typedef struct CreateCo {
366 BlockDriver *drv;
367 char *filename;
368 QEMUOptionParameter *options;
369 int ret;
370} CreateCo;
371
372static void coroutine_fn bdrv_create_co_entry(void *opaque)
373{
374 CreateCo *cco = opaque;
375 assert(cco->drv);
376
377 cco->ret = cco->drv->bdrv_create(cco->filename, cco->options);
378}
379
0e7e1989
KW
380int bdrv_create(BlockDriver *drv, const char* filename,
381 QEMUOptionParameter *options)
ea2384d3 382{
5b7e1542
ZYW
383 int ret;
384
385 Coroutine *co;
386 CreateCo cco = {
387 .drv = drv,
388 .filename = g_strdup(filename),
389 .options = options,
390 .ret = NOT_DONE,
391 };
392
393 if (!drv->bdrv_create) {
80168bff
LC
394 ret = -ENOTSUP;
395 goto out;
5b7e1542
ZYW
396 }
397
398 if (qemu_in_coroutine()) {
399 /* Fast-path if already in coroutine context */
400 bdrv_create_co_entry(&cco);
401 } else {
402 co = qemu_coroutine_create(bdrv_create_co_entry);
403 qemu_coroutine_enter(co, &cco);
404 while (cco.ret == NOT_DONE) {
405 qemu_aio_wait();
406 }
407 }
408
409 ret = cco.ret;
0e7e1989 410
80168bff
LC
411out:
412 g_free(cco.filename);
5b7e1542 413 return ret;
ea2384d3
FB
414}
415
84a12e66
CH
416int bdrv_create_file(const char* filename, QEMUOptionParameter *options)
417{
418 BlockDriver *drv;
419
98289620 420 drv = bdrv_find_protocol(filename, true);
84a12e66 421 if (drv == NULL) {
16905d71 422 return -ENOENT;
84a12e66
CH
423 }
424
425 return bdrv_create(drv, filename, options);
426}
427
eba25057
JM
428/*
429 * Create a uniquely-named empty temporary file.
430 * Return 0 upon success, otherwise a negative errno value.
431 */
432int get_tmp_filename(char *filename, int size)
d5249393 433{
eba25057 434#ifdef _WIN32
3b9f94e1 435 char temp_dir[MAX_PATH];
eba25057
JM
436 /* GetTempFileName requires that its output buffer (4th param)
437 have length MAX_PATH or greater. */
438 assert(size >= MAX_PATH);
439 return (GetTempPath(MAX_PATH, temp_dir)
440 && GetTempFileName(temp_dir, "qem", 0, filename)
441 ? 0 : -GetLastError());
d5249393 442#else
67b915a5 443 int fd;
7ccfb2eb 444 const char *tmpdir;
0badc1ee
AJ
445 tmpdir = getenv("TMPDIR");
446 if (!tmpdir)
447 tmpdir = "/tmp";
eba25057
JM
448 if (snprintf(filename, size, "%s/vl.XXXXXX", tmpdir) >= size) {
449 return -EOVERFLOW;
450 }
ea2384d3 451 fd = mkstemp(filename);
fe235a06
DH
452 if (fd < 0) {
453 return -errno;
454 }
455 if (close(fd) != 0) {
456 unlink(filename);
eba25057
JM
457 return -errno;
458 }
459 return 0;
d5249393 460#endif
eba25057 461}
fc01f7e7 462
84a12e66
CH
463/*
464 * Detect host devices. By convention, /dev/cdrom[N] is always
465 * recognized as a host CDROM.
466 */
467static BlockDriver *find_hdev_driver(const char *filename)
468{
469 int score_max = 0, score;
470 BlockDriver *drv = NULL, *d;
471
472 QLIST_FOREACH(d, &bdrv_drivers, list) {
473 if (d->bdrv_probe_device) {
474 score = d->bdrv_probe_device(filename);
475 if (score > score_max) {
476 score_max = score;
477 drv = d;
478 }
479 }
480 }
481
482 return drv;
483}
484
98289620
KW
485BlockDriver *bdrv_find_protocol(const char *filename,
486 bool allow_protocol_prefix)
83f64091
FB
487{
488 BlockDriver *drv1;
489 char protocol[128];
1cec71e3 490 int len;
83f64091 491 const char *p;
19cb3738 492
66f82cee
KW
493 /* TODO Drivers without bdrv_file_open must be specified explicitly */
494
39508e7a
CH
495 /*
496 * XXX(hch): we really should not let host device detection
497 * override an explicit protocol specification, but moving this
498 * later breaks access to device names with colons in them.
499 * Thanks to the brain-dead persistent naming schemes on udev-
500 * based Linux systems those actually are quite common.
501 */
502 drv1 = find_hdev_driver(filename);
503 if (drv1) {
504 return drv1;
505 }
506
98289620 507 if (!path_has_protocol(filename) || !allow_protocol_prefix) {
39508e7a 508 return bdrv_find_format("file");
84a12e66 509 }
98289620 510
9e0b22f4
SH
511 p = strchr(filename, ':');
512 assert(p != NULL);
1cec71e3
AL
513 len = p - filename;
514 if (len > sizeof(protocol) - 1)
515 len = sizeof(protocol) - 1;
516 memcpy(protocol, filename, len);
517 protocol[len] = '\0';
8a22f02a 518 QLIST_FOREACH(drv1, &bdrv_drivers, list) {
5fafdf24 519 if (drv1->protocol_name &&
8a22f02a 520 !strcmp(drv1->protocol_name, protocol)) {
83f64091 521 return drv1;
8a22f02a 522 }
83f64091
FB
523 }
524 return NULL;
525}
526
f500a6d3
KW
527static int find_image_format(BlockDriverState *bs, const char *filename,
528 BlockDriver **pdrv)
f3a5d3f8 529{
f500a6d3 530 int score, score_max;
f3a5d3f8
CH
531 BlockDriver *drv1, *drv;
532 uint8_t buf[2048];
f500a6d3 533 int ret = 0;
f8ea0b00 534
08a00559 535 /* Return the raw BlockDriver * to scsi-generic devices or empty drives */
8e895599 536 if (bs->sg || !bdrv_is_inserted(bs) || bdrv_getlength(bs) == 0) {
c98ac35d
SW
537 drv = bdrv_find_format("raw");
538 if (!drv) {
539 ret = -ENOENT;
540 }
541 *pdrv = drv;
542 return ret;
1a396859 543 }
f8ea0b00 544
83f64091 545 ret = bdrv_pread(bs, 0, buf, sizeof(buf));
83f64091 546 if (ret < 0) {
c98ac35d
SW
547 *pdrv = NULL;
548 return ret;
83f64091
FB
549 }
550
ea2384d3 551 score_max = 0;
84a12e66 552 drv = NULL;
8a22f02a 553 QLIST_FOREACH(drv1, &bdrv_drivers, list) {
83f64091
FB
554 if (drv1->bdrv_probe) {
555 score = drv1->bdrv_probe(buf, ret, filename);
556 if (score > score_max) {
557 score_max = score;
558 drv = drv1;
559 }
0849bf08 560 }
fc01f7e7 561 }
c98ac35d
SW
562 if (!drv) {
563 ret = -ENOENT;
564 }
565 *pdrv = drv;
566 return ret;
ea2384d3
FB
567}
568
51762288
SH
569/**
570 * Set the current 'total_sectors' value
571 */
572static int refresh_total_sectors(BlockDriverState *bs, int64_t hint)
573{
574 BlockDriver *drv = bs->drv;
575
396759ad
NB
576 /* Do not attempt drv->bdrv_getlength() on scsi-generic devices */
577 if (bs->sg)
578 return 0;
579
51762288
SH
580 /* query actual device if possible, otherwise just trust the hint */
581 if (drv->bdrv_getlength) {
582 int64_t length = drv->bdrv_getlength(bs);
583 if (length < 0) {
584 return length;
585 }
586 hint = length >> BDRV_SECTOR_BITS;
587 }
588
589 bs->total_sectors = hint;
590 return 0;
591}
592
9e8f1835
PB
593/**
594 * Set open flags for a given discard mode
595 *
596 * Return 0 on success, -1 if the discard mode was invalid.
597 */
598int bdrv_parse_discard_flags(const char *mode, int *flags)
599{
600 *flags &= ~BDRV_O_UNMAP;
601
602 if (!strcmp(mode, "off") || !strcmp(mode, "ignore")) {
603 /* do nothing */
604 } else if (!strcmp(mode, "on") || !strcmp(mode, "unmap")) {
605 *flags |= BDRV_O_UNMAP;
606 } else {
607 return -1;
608 }
609
610 return 0;
611}
612
c3993cdc
SH
613/**
614 * Set open flags for a given cache mode
615 *
616 * Return 0 on success, -1 if the cache mode was invalid.
617 */
618int bdrv_parse_cache_flags(const char *mode, int *flags)
619{
620 *flags &= ~BDRV_O_CACHE_MASK;
621
622 if (!strcmp(mode, "off") || !strcmp(mode, "none")) {
623 *flags |= BDRV_O_NOCACHE | BDRV_O_CACHE_WB;
92196b2f
SH
624 } else if (!strcmp(mode, "directsync")) {
625 *flags |= BDRV_O_NOCACHE;
c3993cdc
SH
626 } else if (!strcmp(mode, "writeback")) {
627 *flags |= BDRV_O_CACHE_WB;
628 } else if (!strcmp(mode, "unsafe")) {
629 *flags |= BDRV_O_CACHE_WB;
630 *flags |= BDRV_O_NO_FLUSH;
631 } else if (!strcmp(mode, "writethrough")) {
632 /* this is the default */
633 } else {
634 return -1;
635 }
636
637 return 0;
638}
639
53fec9d3
SH
640/**
641 * The copy-on-read flag is actually a reference count so multiple users may
642 * use the feature without worrying about clobbering its previous state.
643 * Copy-on-read stays enabled until all users have called to disable it.
644 */
645void bdrv_enable_copy_on_read(BlockDriverState *bs)
646{
647 bs->copy_on_read++;
648}
649
650void bdrv_disable_copy_on_read(BlockDriverState *bs)
651{
652 assert(bs->copy_on_read > 0);
653 bs->copy_on_read--;
654}
655
7b272452
KW
656static int bdrv_open_flags(BlockDriverState *bs, int flags)
657{
658 int open_flags = flags | BDRV_O_CACHE_WB;
659
660 /*
661 * Clear flags that are internal to the block layer before opening the
662 * image.
663 */
664 open_flags &= ~(BDRV_O_SNAPSHOT | BDRV_O_NO_BACKING);
665
666 /*
667 * Snapshots should be writable.
668 */
669 if (bs->is_temporary) {
670 open_flags |= BDRV_O_RDWR;
671 }
672
673 return open_flags;
674}
675
57915332
KW
676/*
677 * Common part for opening disk images and files
b6ad491a
KW
678 *
679 * Removes all processed options from *options.
57915332 680 */
f500a6d3 681static int bdrv_open_common(BlockDriverState *bs, BlockDriverState *file,
035fccdf 682 QDict *options, int flags, BlockDriver *drv)
57915332
KW
683{
684 int ret, open_flags;
035fccdf 685 const char *filename;
57915332
KW
686
687 assert(drv != NULL);
6405875c 688 assert(bs->file == NULL);
707ff828 689 assert(options != NULL && bs->options != options);
57915332 690
45673671
KW
691 if (file != NULL) {
692 filename = file->filename;
693 } else {
694 filename = qdict_get_try_str(options, "filename");
695 }
696
697 trace_bdrv_open_common(bs, filename ?: "", flags, drv->format_name);
28dcee10 698
5d186eb0
KW
699 /* bdrv_open() with directly using a protocol as drv. This layer is already
700 * opened, so assign it to bs (while file becomes a closed BlockDriverState)
701 * and return immediately. */
702 if (file != NULL && drv->bdrv_file_open) {
703 bdrv_swap(file, bs);
704 return 0;
705 }
706
57915332 707 bs->open_flags = flags;
57915332 708 bs->buffer_alignment = 512;
b64ec4e4
FZ
709 open_flags = bdrv_open_flags(bs, flags);
710 bs->read_only = !(open_flags & BDRV_O_RDWR);
711
712 if (use_bdrv_whitelist && !bdrv_is_whitelisted(drv, bs->read_only)) {
713 return -ENOTSUP;
714 }
57915332 715
53fec9d3 716 assert(bs->copy_on_read == 0); /* bdrv_new() and bdrv_close() make it so */
b64ec4e4 717 if (!bs->read_only && (flags & BDRV_O_COPY_ON_READ)) {
53fec9d3
SH
718 bdrv_enable_copy_on_read(bs);
719 }
720
c2ad1b0c
KW
721 if (filename != NULL) {
722 pstrcpy(bs->filename, sizeof(bs->filename), filename);
723 } else {
724 bs->filename[0] = '\0';
725 }
57915332 726
57915332 727 bs->drv = drv;
7267c094 728 bs->opaque = g_malloc0(drv->instance_size);
57915332 729
03f541bd 730 bs->enable_write_cache = !!(flags & BDRV_O_CACHE_WB);
e7c63796 731
66f82cee
KW
732 /* Open the image, either directly or using a protocol */
733 if (drv->bdrv_file_open) {
5d186eb0
KW
734 assert(file == NULL);
735 assert(drv->bdrv_parse_filename || filename != NULL);
56d1b4d2 736 ret = drv->bdrv_file_open(bs, options, open_flags);
f500a6d3 737 } else {
2af5ef70
KW
738 if (file == NULL) {
739 qerror_report(ERROR_CLASS_GENERIC_ERROR, "Can't use '%s' as a "
740 "block driver for the protocol level",
741 drv->format_name);
742 ret = -EINVAL;
743 goto free_and_fail;
744 }
f500a6d3
KW
745 assert(file != NULL);
746 bs->file = file;
b6ad491a 747 ret = drv->bdrv_open(bs, options, open_flags);
66f82cee
KW
748 }
749
57915332
KW
750 if (ret < 0) {
751 goto free_and_fail;
752 }
753
51762288
SH
754 ret = refresh_total_sectors(bs, bs->total_sectors);
755 if (ret < 0) {
756 goto free_and_fail;
57915332 757 }
51762288 758
57915332
KW
759#ifndef _WIN32
760 if (bs->is_temporary) {
c2ad1b0c 761 assert(filename != NULL);
57915332
KW
762 unlink(filename);
763 }
764#endif
765 return 0;
766
767free_and_fail:
f500a6d3 768 bs->file = NULL;
7267c094 769 g_free(bs->opaque);
57915332
KW
770 bs->opaque = NULL;
771 bs->drv = NULL;
772 return ret;
773}
774
b6ce07aa
KW
775/*
776 * Opens a file using a protocol (file, host_device, nbd, ...)
787e4a85
KW
777 *
778 * options is a QDict of options to pass to the block drivers, or NULL for an
779 * empty set of options. The reference to the QDict belongs to the block layer
780 * after the call (even on failure), so if the caller intends to reuse the
781 * dictionary, it needs to use QINCREF() before calling bdrv_file_open.
b6ce07aa 782 */
787e4a85
KW
783int bdrv_file_open(BlockDriverState **pbs, const char *filename,
784 QDict *options, int flags)
ea2384d3 785{
83f64091 786 BlockDriverState *bs;
6db95603 787 BlockDriver *drv;
c2ad1b0c 788 const char *drvname;
98289620 789 bool allow_protocol_prefix = false;
83f64091
FB
790 int ret;
791
707ff828
KW
792 /* NULL means an empty set of options */
793 if (options == NULL) {
794 options = qdict_new();
795 }
796
83f64091 797 bs = bdrv_new("");
707ff828
KW
798 bs->options = options;
799 options = qdict_clone_shallow(options);
800
035fccdf
KW
801 /* Fetch the file name from the options QDict if necessary */
802 if (!filename) {
803 filename = qdict_get_try_str(options, "filename");
804 } else if (filename && !qdict_haskey(options, "filename")) {
805 qdict_put(options, "filename", qstring_from_str(filename));
98289620 806 allow_protocol_prefix = true;
035fccdf
KW
807 } else {
808 qerror_report(ERROR_CLASS_GENERIC_ERROR, "Can't specify 'file' and "
809 "'filename' options at the same time");
810 ret = -EINVAL;
811 goto fail;
812 }
813
c2ad1b0c
KW
814 /* Find the right block driver */
815 drvname = qdict_get_try_str(options, "driver");
816 if (drvname) {
b64ec4e4 817 drv = bdrv_find_whitelisted_format(drvname, !(flags & BDRV_O_RDWR));
c2ad1b0c
KW
818 qdict_del(options, "driver");
819 } else if (filename) {
98289620
KW
820 drv = bdrv_find_protocol(filename, allow_protocol_prefix);
821 if (!drv) {
822 qerror_report(ERROR_CLASS_GENERIC_ERROR, "Unknown protocol");
823 }
c2ad1b0c
KW
824 } else {
825 qerror_report(ERROR_CLASS_GENERIC_ERROR,
826 "Must specify either driver or file");
827 drv = NULL;
828 }
829
830 if (!drv) {
831 ret = -ENOENT;
832 goto fail;
833 }
834
835 /* Parse the filename and open it */
836 if (drv->bdrv_parse_filename && filename) {
6963a30d
KW
837 Error *local_err = NULL;
838 drv->bdrv_parse_filename(filename, options, &local_err);
839 if (error_is_set(&local_err)) {
840 qerror_report_err(local_err);
841 error_free(local_err);
842 ret = -EINVAL;
843 goto fail;
844 }
56d1b4d2 845 qdict_del(options, "filename");
c2ad1b0c
KW
846 } else if (!drv->bdrv_parse_filename && !filename) {
847 qerror_report(ERROR_CLASS_GENERIC_ERROR,
848 "The '%s' block driver requires a file name",
849 drv->format_name);
850 ret = -EINVAL;
851 goto fail;
6963a30d
KW
852 }
853
035fccdf 854 ret = bdrv_open_common(bs, NULL, options, flags, drv);
83f64091 855 if (ret < 0) {
707ff828
KW
856 goto fail;
857 }
858
859 /* Check if any unknown options were used */
860 if (qdict_size(options) != 0) {
861 const QDictEntry *entry = qdict_first(options);
862 qerror_report(ERROR_CLASS_GENERIC_ERROR, "Block protocol '%s' doesn't "
863 "support the option '%s'",
864 drv->format_name, entry->key);
865 ret = -EINVAL;
866 goto fail;
3b0d4f61 867 }
707ff828
KW
868 QDECREF(options);
869
71d0770c 870 bs->growable = 1;
83f64091
FB
871 *pbs = bs;
872 return 0;
707ff828
KW
873
874fail:
875 QDECREF(options);
876 if (!bs->drv) {
877 QDECREF(bs->options);
878 }
879 bdrv_delete(bs);
880 return ret;
83f64091
FB
881}
882
31ca6d07
KW
883/*
884 * Opens the backing file for a BlockDriverState if not yet open
885 *
886 * options is a QDict of options to pass to the block drivers, or NULL for an
887 * empty set of options. The reference to the QDict is transferred to this
888 * function (even on failure), so if the caller intends to reuse the dictionary,
889 * it needs to use QINCREF() before calling bdrv_file_open.
890 */
891int bdrv_open_backing_file(BlockDriverState *bs, QDict *options)
9156df12
PB
892{
893 char backing_filename[PATH_MAX];
894 int back_flags, ret;
895 BlockDriver *back_drv = NULL;
896
897 if (bs->backing_hd != NULL) {
31ca6d07 898 QDECREF(options);
9156df12
PB
899 return 0;
900 }
901
31ca6d07
KW
902 /* NULL means an empty set of options */
903 if (options == NULL) {
904 options = qdict_new();
905 }
906
9156df12 907 bs->open_flags &= ~BDRV_O_NO_BACKING;
1cb6f506
KW
908 if (qdict_haskey(options, "file.filename")) {
909 backing_filename[0] = '\0';
910 } else if (bs->backing_file[0] == '\0' && qdict_size(options) == 0) {
31ca6d07 911 QDECREF(options);
9156df12
PB
912 return 0;
913 }
914
915 bs->backing_hd = bdrv_new("");
916 bdrv_get_full_backing_filename(bs, backing_filename,
917 sizeof(backing_filename));
918
919 if (bs->backing_format[0] != '\0') {
920 back_drv = bdrv_find_format(bs->backing_format);
921 }
922
923 /* backing files always opened read-only */
924 back_flags = bs->open_flags & ~(BDRV_O_RDWR | BDRV_O_SNAPSHOT);
925
31ca6d07
KW
926 ret = bdrv_open(bs->backing_hd,
927 *backing_filename ? backing_filename : NULL, options,
de9c0cec 928 back_flags, back_drv);
9156df12
PB
929 if (ret < 0) {
930 bdrv_delete(bs->backing_hd);
931 bs->backing_hd = NULL;
932 bs->open_flags |= BDRV_O_NO_BACKING;
933 return ret;
934 }
935 return 0;
936}
937
707ff828
KW
938static void extract_subqdict(QDict *src, QDict **dst, const char *start)
939{
940 const QDictEntry *entry, *next;
941 const char *p;
942
943 *dst = qdict_new();
944 entry = qdict_first(src);
945
946 while (entry != NULL) {
947 next = qdict_next(src, entry);
948 if (strstart(entry->key, start, &p)) {
949 qobject_incref(entry->value);
950 qdict_put_obj(*dst, p, entry->value);
951 qdict_del(src, entry->key);
952 }
953 entry = next;
954 }
955}
956
b6ce07aa
KW
957/*
958 * Opens a disk image (raw, qcow2, vmdk, ...)
de9c0cec
KW
959 *
960 * options is a QDict of options to pass to the block drivers, or NULL for an
961 * empty set of options. The reference to the QDict belongs to the block layer
962 * after the call (even on failure), so if the caller intends to reuse the
963 * dictionary, it needs to use QINCREF() before calling bdrv_open.
b6ce07aa 964 */
de9c0cec
KW
965int bdrv_open(BlockDriverState *bs, const char *filename, QDict *options,
966 int flags, BlockDriver *drv)
ea2384d3 967{
b6ce07aa 968 int ret;
89c9bc3d
SW
969 /* TODO: extra byte is a hack to ensure MAX_PATH space on Windows. */
970 char tmp_filename[PATH_MAX + 1];
f500a6d3 971 BlockDriverState *file = NULL;
707ff828 972 QDict *file_options = NULL;
74fe54f2 973 const char *drvname;
712e7874 974
de9c0cec
KW
975 /* NULL means an empty set of options */
976 if (options == NULL) {
977 options = qdict_new();
978 }
979
980 bs->options = options;
b6ad491a 981 options = qdict_clone_shallow(options);
de9c0cec
KW
982
983 /* For snapshot=on, create a temporary qcow2 overlay */
83f64091 984 if (flags & BDRV_O_SNAPSHOT) {
ea2384d3
FB
985 BlockDriverState *bs1;
986 int64_t total_size;
91a073a9 987 BlockDriver *bdrv_qcow2;
08b392e1 988 QEMUOptionParameter *create_options;
b6ce07aa 989 char backing_filename[PATH_MAX];
3b46e624 990
c2ad1b0c
KW
991 if (qdict_size(options) != 0) {
992 error_report("Can't use snapshot=on with driver-specific options");
993 ret = -EINVAL;
994 goto fail;
995 }
996 assert(filename != NULL);
997
ea2384d3
FB
998 /* if snapshot, we create a temporary backing file and open it
999 instead of opening 'filename' directly */
33e3963e 1000
ea2384d3
FB
1001 /* if there is a backing file, use it */
1002 bs1 = bdrv_new("");
de9c0cec 1003 ret = bdrv_open(bs1, filename, NULL, 0, drv);
51d7c00c 1004 if (ret < 0) {
ea2384d3 1005 bdrv_delete(bs1);
de9c0cec 1006 goto fail;
ea2384d3 1007 }
3e82990b 1008 total_size = bdrv_getlength(bs1) & BDRV_SECTOR_MASK;
7c96d46e 1009
ea2384d3 1010 bdrv_delete(bs1);
3b46e624 1011
eba25057
JM
1012 ret = get_tmp_filename(tmp_filename, sizeof(tmp_filename));
1013 if (ret < 0) {
de9c0cec 1014 goto fail;
eba25057 1015 }
7c96d46e
AL
1016
1017 /* Real path is meaningless for protocols */
4d70655b 1018 if (path_has_protocol(filename)) {
7c96d46e
AL
1019 snprintf(backing_filename, sizeof(backing_filename),
1020 "%s", filename);
de9c0cec
KW
1021 } else if (!realpath(filename, backing_filename)) {
1022 ret = -errno;
1023 goto fail;
1024 }
7c96d46e 1025
91a073a9 1026 bdrv_qcow2 = bdrv_find_format("qcow2");
08b392e1
KW
1027 create_options = parse_option_parameters("", bdrv_qcow2->create_options,
1028 NULL);
91a073a9 1029
08b392e1
KW
1030 set_option_parameter_int(create_options, BLOCK_OPT_SIZE, total_size);
1031 set_option_parameter(create_options, BLOCK_OPT_BACKING_FILE,
1032 backing_filename);
91a073a9 1033 if (drv) {
08b392e1 1034 set_option_parameter(create_options, BLOCK_OPT_BACKING_FMT,
91a073a9
KW
1035 drv->format_name);
1036 }
1037
08b392e1
KW
1038 ret = bdrv_create(bdrv_qcow2, tmp_filename, create_options);
1039 free_option_parameters(create_options);
51d7c00c 1040 if (ret < 0) {
de9c0cec 1041 goto fail;
ea2384d3 1042 }
91a073a9 1043
ea2384d3 1044 filename = tmp_filename;
91a073a9 1045 drv = bdrv_qcow2;
ea2384d3
FB
1046 bs->is_temporary = 1;
1047 }
712e7874 1048
f500a6d3
KW
1049 /* Open image file without format layer */
1050 if (flags & BDRV_O_RDWR) {
1051 flags |= BDRV_O_ALLOW_RDWR;
1052 }
1053
707ff828
KW
1054 extract_subqdict(options, &file_options, "file.");
1055
1056 ret = bdrv_file_open(&file, filename, file_options,
50b05b6f 1057 bdrv_open_flags(bs, flags | BDRV_O_UNMAP));
f500a6d3 1058 if (ret < 0) {
de9c0cec 1059 goto fail;
f500a6d3
KW
1060 }
1061
b6ce07aa 1062 /* Find the right image format driver */
74fe54f2
KW
1063 drvname = qdict_get_try_str(options, "driver");
1064 if (drvname) {
1065 drv = bdrv_find_whitelisted_format(drvname, !(flags & BDRV_O_RDWR));
1066 qdict_del(options, "driver");
1067 }
1068
6db95603 1069 if (!drv) {
f500a6d3 1070 ret = find_image_format(file, filename, &drv);
51d7c00c 1071 }
6987307c 1072
51d7c00c 1073 if (!drv) {
51d7c00c 1074 goto unlink_and_fail;
ea2384d3 1075 }
b6ce07aa
KW
1076
1077 /* Open the image */
035fccdf 1078 ret = bdrv_open_common(bs, file, options, flags, drv);
b6ce07aa 1079 if (ret < 0) {
6987307c
CH
1080 goto unlink_and_fail;
1081 }
1082
f500a6d3
KW
1083 if (bs->file != file) {
1084 bdrv_delete(file);
1085 file = NULL;
1086 }
1087
b6ce07aa 1088 /* If there is a backing file, use it */
9156df12 1089 if ((flags & BDRV_O_NO_BACKING) == 0) {
31ca6d07
KW
1090 QDict *backing_options;
1091
1092 extract_subqdict(options, &backing_options, "backing.");
1093 ret = bdrv_open_backing_file(bs, backing_options);
b6ce07aa 1094 if (ret < 0) {
b6ad491a 1095 goto close_and_fail;
b6ce07aa 1096 }
b6ce07aa
KW
1097 }
1098
b6ad491a
KW
1099 /* Check if any unknown options were used */
1100 if (qdict_size(options) != 0) {
1101 const QDictEntry *entry = qdict_first(options);
1102 qerror_report(ERROR_CLASS_GENERIC_ERROR, "Block format '%s' used by "
1103 "device '%s' doesn't support the option '%s'",
1104 drv->format_name, bs->device_name, entry->key);
1105
1106 ret = -EINVAL;
1107 goto close_and_fail;
1108 }
1109 QDECREF(options);
1110
b6ce07aa 1111 if (!bdrv_key_required(bs)) {
7d4b4ba5 1112 bdrv_dev_change_media_cb(bs, true);
b6ce07aa
KW
1113 }
1114
98f90dba
ZYW
1115 /* throttling disk I/O limits */
1116 if (bs->io_limits_enabled) {
1117 bdrv_io_limits_enable(bs);
1118 }
1119
b6ce07aa
KW
1120 return 0;
1121
1122unlink_and_fail:
f500a6d3
KW
1123 if (file != NULL) {
1124 bdrv_delete(file);
1125 }
b6ce07aa
KW
1126 if (bs->is_temporary) {
1127 unlink(filename);
1128 }
de9c0cec
KW
1129fail:
1130 QDECREF(bs->options);
b6ad491a 1131 QDECREF(options);
de9c0cec 1132 bs->options = NULL;
b6ad491a 1133 return ret;
de9c0cec 1134
b6ad491a
KW
1135close_and_fail:
1136 bdrv_close(bs);
1137 QDECREF(options);
b6ce07aa
KW
1138 return ret;
1139}
1140
e971aa12
JC
1141typedef struct BlockReopenQueueEntry {
1142 bool prepared;
1143 BDRVReopenState state;
1144 QSIMPLEQ_ENTRY(BlockReopenQueueEntry) entry;
1145} BlockReopenQueueEntry;
1146
1147/*
1148 * Adds a BlockDriverState to a simple queue for an atomic, transactional
1149 * reopen of multiple devices.
1150 *
1151 * bs_queue can either be an existing BlockReopenQueue that has had QSIMPLE_INIT
1152 * already performed, or alternatively may be NULL a new BlockReopenQueue will
1153 * be created and initialized. This newly created BlockReopenQueue should be
1154 * passed back in for subsequent calls that are intended to be of the same
1155 * atomic 'set'.
1156 *
1157 * bs is the BlockDriverState to add to the reopen queue.
1158 *
1159 * flags contains the open flags for the associated bs
1160 *
1161 * returns a pointer to bs_queue, which is either the newly allocated
1162 * bs_queue, or the existing bs_queue being used.
1163 *
1164 */
1165BlockReopenQueue *bdrv_reopen_queue(BlockReopenQueue *bs_queue,
1166 BlockDriverState *bs, int flags)
1167{
1168 assert(bs != NULL);
1169
1170 BlockReopenQueueEntry *bs_entry;
1171 if (bs_queue == NULL) {
1172 bs_queue = g_new0(BlockReopenQueue, 1);
1173 QSIMPLEQ_INIT(bs_queue);
1174 }
1175
1176 if (bs->file) {
1177 bdrv_reopen_queue(bs_queue, bs->file, flags);
1178 }
1179
1180 bs_entry = g_new0(BlockReopenQueueEntry, 1);
1181 QSIMPLEQ_INSERT_TAIL(bs_queue, bs_entry, entry);
1182
1183 bs_entry->state.bs = bs;
1184 bs_entry->state.flags = flags;
1185
1186 return bs_queue;
1187}
1188
1189/*
1190 * Reopen multiple BlockDriverStates atomically & transactionally.
1191 *
1192 * The queue passed in (bs_queue) must have been built up previous
1193 * via bdrv_reopen_queue().
1194 *
1195 * Reopens all BDS specified in the queue, with the appropriate
1196 * flags. All devices are prepared for reopen, and failure of any
1197 * device will cause all device changes to be abandonded, and intermediate
1198 * data cleaned up.
1199 *
1200 * If all devices prepare successfully, then the changes are committed
1201 * to all devices.
1202 *
1203 */
1204int bdrv_reopen_multiple(BlockReopenQueue *bs_queue, Error **errp)
1205{
1206 int ret = -1;
1207 BlockReopenQueueEntry *bs_entry, *next;
1208 Error *local_err = NULL;
1209
1210 assert(bs_queue != NULL);
1211
1212 bdrv_drain_all();
1213
1214 QSIMPLEQ_FOREACH(bs_entry, bs_queue, entry) {
1215 if (bdrv_reopen_prepare(&bs_entry->state, bs_queue, &local_err)) {
1216 error_propagate(errp, local_err);
1217 goto cleanup;
1218 }
1219 bs_entry->prepared = true;
1220 }
1221
1222 /* If we reach this point, we have success and just need to apply the
1223 * changes
1224 */
1225 QSIMPLEQ_FOREACH(bs_entry, bs_queue, entry) {
1226 bdrv_reopen_commit(&bs_entry->state);
1227 }
1228
1229 ret = 0;
1230
1231cleanup:
1232 QSIMPLEQ_FOREACH_SAFE(bs_entry, bs_queue, entry, next) {
1233 if (ret && bs_entry->prepared) {
1234 bdrv_reopen_abort(&bs_entry->state);
1235 }
1236 g_free(bs_entry);
1237 }
1238 g_free(bs_queue);
1239 return ret;
1240}
1241
1242
1243/* Reopen a single BlockDriverState with the specified flags. */
1244int bdrv_reopen(BlockDriverState *bs, int bdrv_flags, Error **errp)
1245{
1246 int ret = -1;
1247 Error *local_err = NULL;
1248 BlockReopenQueue *queue = bdrv_reopen_queue(NULL, bs, bdrv_flags);
1249
1250 ret = bdrv_reopen_multiple(queue, &local_err);
1251 if (local_err != NULL) {
1252 error_propagate(errp, local_err);
1253 }
1254 return ret;
1255}
1256
1257
1258/*
1259 * Prepares a BlockDriverState for reopen. All changes are staged in the
1260 * 'opaque' field of the BDRVReopenState, which is used and allocated by
1261 * the block driver layer .bdrv_reopen_prepare()
1262 *
1263 * bs is the BlockDriverState to reopen
1264 * flags are the new open flags
1265 * queue is the reopen queue
1266 *
1267 * Returns 0 on success, non-zero on error. On error errp will be set
1268 * as well.
1269 *
1270 * On failure, bdrv_reopen_abort() will be called to clean up any data.
1271 * It is the responsibility of the caller to then call the abort() or
1272 * commit() for any other BDS that have been left in a prepare() state
1273 *
1274 */
1275int bdrv_reopen_prepare(BDRVReopenState *reopen_state, BlockReopenQueue *queue,
1276 Error **errp)
1277{
1278 int ret = -1;
1279 Error *local_err = NULL;
1280 BlockDriver *drv;
1281
1282 assert(reopen_state != NULL);
1283 assert(reopen_state->bs->drv != NULL);
1284 drv = reopen_state->bs->drv;
1285
1286 /* if we are to stay read-only, do not allow permission change
1287 * to r/w */
1288 if (!(reopen_state->bs->open_flags & BDRV_O_ALLOW_RDWR) &&
1289 reopen_state->flags & BDRV_O_RDWR) {
1290 error_set(errp, QERR_DEVICE_IS_READ_ONLY,
1291 reopen_state->bs->device_name);
1292 goto error;
1293 }
1294
1295
1296 ret = bdrv_flush(reopen_state->bs);
1297 if (ret) {
1298 error_set(errp, ERROR_CLASS_GENERIC_ERROR, "Error (%s) flushing drive",
1299 strerror(-ret));
1300 goto error;
1301 }
1302
1303 if (drv->bdrv_reopen_prepare) {
1304 ret = drv->bdrv_reopen_prepare(reopen_state, queue, &local_err);
1305 if (ret) {
1306 if (local_err != NULL) {
1307 error_propagate(errp, local_err);
1308 } else {
d8b6895f
LC
1309 error_setg(errp, "failed while preparing to reopen image '%s'",
1310 reopen_state->bs->filename);
e971aa12
JC
1311 }
1312 goto error;
1313 }
1314 } else {
1315 /* It is currently mandatory to have a bdrv_reopen_prepare()
1316 * handler for each supported drv. */
1317 error_set(errp, QERR_BLOCK_FORMAT_FEATURE_NOT_SUPPORTED,
1318 drv->format_name, reopen_state->bs->device_name,
1319 "reopening of file");
1320 ret = -1;
1321 goto error;
1322 }
1323
1324 ret = 0;
1325
1326error:
1327 return ret;
1328}
1329
1330/*
1331 * Takes the staged changes for the reopen from bdrv_reopen_prepare(), and
1332 * makes them final by swapping the staging BlockDriverState contents into
1333 * the active BlockDriverState contents.
1334 */
1335void bdrv_reopen_commit(BDRVReopenState *reopen_state)
1336{
1337 BlockDriver *drv;
1338
1339 assert(reopen_state != NULL);
1340 drv = reopen_state->bs->drv;
1341 assert(drv != NULL);
1342
1343 /* If there are any driver level actions to take */
1344 if (drv->bdrv_reopen_commit) {
1345 drv->bdrv_reopen_commit(reopen_state);
1346 }
1347
1348 /* set BDS specific flags now */
1349 reopen_state->bs->open_flags = reopen_state->flags;
1350 reopen_state->bs->enable_write_cache = !!(reopen_state->flags &
1351 BDRV_O_CACHE_WB);
1352 reopen_state->bs->read_only = !(reopen_state->flags & BDRV_O_RDWR);
1353}
1354
1355/*
1356 * Abort the reopen, and delete and free the staged changes in
1357 * reopen_state
1358 */
1359void bdrv_reopen_abort(BDRVReopenState *reopen_state)
1360{
1361 BlockDriver *drv;
1362
1363 assert(reopen_state != NULL);
1364 drv = reopen_state->bs->drv;
1365 assert(drv != NULL);
1366
1367 if (drv->bdrv_reopen_abort) {
1368 drv->bdrv_reopen_abort(reopen_state);
1369 }
1370}
1371
1372
fc01f7e7
FB
1373void bdrv_close(BlockDriverState *bs)
1374{
3cbc002c
PB
1375 if (bs->job) {
1376 block_job_cancel_sync(bs->job);
1377 }
58fda173
SH
1378 bdrv_drain_all(); /* complete I/O */
1379 bdrv_flush(bs);
1380 bdrv_drain_all(); /* in case flush left pending I/O */
d7d512f6 1381 notifier_list_notify(&bs->close_notifiers, bs);
7094f12f 1382
3cbc002c 1383 if (bs->drv) {
557df6ac 1384 if (bs->backing_hd) {
ea2384d3 1385 bdrv_delete(bs->backing_hd);
557df6ac
SH
1386 bs->backing_hd = NULL;
1387 }
ea2384d3 1388 bs->drv->bdrv_close(bs);
7267c094 1389 g_free(bs->opaque);
ea2384d3
FB
1390#ifdef _WIN32
1391 if (bs->is_temporary) {
1392 unlink(bs->filename);
1393 }
67b915a5 1394#endif
ea2384d3
FB
1395 bs->opaque = NULL;
1396 bs->drv = NULL;
53fec9d3 1397 bs->copy_on_read = 0;
a275fa42
PB
1398 bs->backing_file[0] = '\0';
1399 bs->backing_format[0] = '\0';
6405875c
PB
1400 bs->total_sectors = 0;
1401 bs->encrypted = 0;
1402 bs->valid_key = 0;
1403 bs->sg = 0;
1404 bs->growable = 0;
de9c0cec
KW
1405 QDECREF(bs->options);
1406 bs->options = NULL;
b338082b 1407
66f82cee 1408 if (bs->file != NULL) {
0ac9377d
PB
1409 bdrv_delete(bs->file);
1410 bs->file = NULL;
66f82cee 1411 }
b338082b 1412 }
98f90dba 1413
9ca11154
PH
1414 bdrv_dev_change_media_cb(bs, false);
1415
98f90dba
ZYW
1416 /*throttling disk I/O limits*/
1417 if (bs->io_limits_enabled) {
1418 bdrv_io_limits_disable(bs);
1419 }
b338082b
FB
1420}
1421
2bc93fed
MK
1422void bdrv_close_all(void)
1423{
1424 BlockDriverState *bs;
1425
1426 QTAILQ_FOREACH(bs, &bdrv_states, list) {
1427 bdrv_close(bs);
1428 }
1429}
1430
922453bc
SH
1431/*
1432 * Wait for pending requests to complete across all BlockDriverStates
1433 *
1434 * This function does not flush data to disk, use bdrv_flush_all() for that
1435 * after calling this function.
4c355d53
ZYW
1436 *
1437 * Note that completion of an asynchronous I/O operation can trigger any
1438 * number of other I/O operations on other devices---for example a coroutine
1439 * can be arbitrarily complex and a constant flow of I/O can come until the
1440 * coroutine is complete. Because of this, it is not possible to have a
1441 * function to drain a single device's I/O queue.
922453bc
SH
1442 */
1443void bdrv_drain_all(void)
1444{
1445 BlockDriverState *bs;
4c355d53
ZYW
1446 bool busy;
1447
1448 do {
1449 busy = qemu_aio_wait();
922453bc 1450
4c355d53
ZYW
1451 /* FIXME: We do not have timer support here, so this is effectively
1452 * a busy wait.
1453 */
1454 QTAILQ_FOREACH(bs, &bdrv_states, list) {
b681a1c7 1455 while (qemu_co_enter_next(&bs->throttled_reqs)) {
4c355d53
ZYW
1456 busy = true;
1457 }
1458 }
1459 } while (busy);
922453bc
SH
1460
1461 /* If requests are still pending there is a bug somewhere */
1462 QTAILQ_FOREACH(bs, &bdrv_states, list) {
1463 assert(QLIST_EMPTY(&bs->tracked_requests));
1464 assert(qemu_co_queue_empty(&bs->throttled_reqs));
1465 }
1466}
1467
d22b2f41
RH
1468/* make a BlockDriverState anonymous by removing from bdrv_state list.
1469 Also, NULL terminate the device_name to prevent double remove */
1470void bdrv_make_anon(BlockDriverState *bs)
1471{
1472 if (bs->device_name[0] != '\0') {
1473 QTAILQ_REMOVE(&bdrv_states, bs, list);
1474 }
1475 bs->device_name[0] = '\0';
1476}
1477
e023b2e2
PB
1478static void bdrv_rebind(BlockDriverState *bs)
1479{
1480 if (bs->drv && bs->drv->bdrv_rebind) {
1481 bs->drv->bdrv_rebind(bs);
1482 }
1483}
1484
4ddc07ca
PB
1485static void bdrv_move_feature_fields(BlockDriverState *bs_dest,
1486 BlockDriverState *bs_src)
8802d1fd 1487{
4ddc07ca
PB
1488 /* move some fields that need to stay attached to the device */
1489 bs_dest->open_flags = bs_src->open_flags;
8802d1fd
JC
1490
1491 /* dev info */
4ddc07ca
PB
1492 bs_dest->dev_ops = bs_src->dev_ops;
1493 bs_dest->dev_opaque = bs_src->dev_opaque;
1494 bs_dest->dev = bs_src->dev;
1495 bs_dest->buffer_alignment = bs_src->buffer_alignment;
1496 bs_dest->copy_on_read = bs_src->copy_on_read;
8802d1fd 1497
4ddc07ca 1498 bs_dest->enable_write_cache = bs_src->enable_write_cache;
c4a248a1 1499
8802d1fd 1500 /* i/o timing parameters */
4ddc07ca
PB
1501 bs_dest->slice_start = bs_src->slice_start;
1502 bs_dest->slice_end = bs_src->slice_end;
5905fbc9 1503 bs_dest->slice_submitted = bs_src->slice_submitted;
4ddc07ca 1504 bs_dest->io_limits = bs_src->io_limits;
4ddc07ca
PB
1505 bs_dest->throttled_reqs = bs_src->throttled_reqs;
1506 bs_dest->block_timer = bs_src->block_timer;
1507 bs_dest->io_limits_enabled = bs_src->io_limits_enabled;
8802d1fd 1508
8802d1fd 1509 /* r/w error */
4ddc07ca
PB
1510 bs_dest->on_read_error = bs_src->on_read_error;
1511 bs_dest->on_write_error = bs_src->on_write_error;
8802d1fd
JC
1512
1513 /* i/o status */
4ddc07ca
PB
1514 bs_dest->iostatus_enabled = bs_src->iostatus_enabled;
1515 bs_dest->iostatus = bs_src->iostatus;
8802d1fd 1516
a9fc4408 1517 /* dirty bitmap */
4ddc07ca 1518 bs_dest->dirty_bitmap = bs_src->dirty_bitmap;
a9fc4408
PB
1519
1520 /* job */
4ddc07ca
PB
1521 bs_dest->in_use = bs_src->in_use;
1522 bs_dest->job = bs_src->job;
a9fc4408 1523
8802d1fd 1524 /* keep the same entry in bdrv_states */
4ddc07ca
PB
1525 pstrcpy(bs_dest->device_name, sizeof(bs_dest->device_name),
1526 bs_src->device_name);
1527 bs_dest->list = bs_src->list;
1528}
8802d1fd 1529
4ddc07ca
PB
1530/*
1531 * Swap bs contents for two image chains while they are live,
1532 * while keeping required fields on the BlockDriverState that is
1533 * actually attached to a device.
1534 *
1535 * This will modify the BlockDriverState fields, and swap contents
1536 * between bs_new and bs_old. Both bs_new and bs_old are modified.
1537 *
1538 * bs_new is required to be anonymous.
1539 *
1540 * This function does not create any image files.
1541 */
1542void bdrv_swap(BlockDriverState *bs_new, BlockDriverState *bs_old)
1543{
1544 BlockDriverState tmp;
f6801b83 1545
4ddc07ca
PB
1546 /* bs_new must be anonymous and shouldn't have anything fancy enabled */
1547 assert(bs_new->device_name[0] == '\0');
1548 assert(bs_new->dirty_bitmap == NULL);
1549 assert(bs_new->job == NULL);
1550 assert(bs_new->dev == NULL);
1551 assert(bs_new->in_use == 0);
1552 assert(bs_new->io_limits_enabled == false);
1553 assert(bs_new->block_timer == NULL);
8802d1fd 1554
4ddc07ca
PB
1555 tmp = *bs_new;
1556 *bs_new = *bs_old;
1557 *bs_old = tmp;
a9fc4408 1558
4ddc07ca
PB
1559 /* there are some fields that should not be swapped, move them back */
1560 bdrv_move_feature_fields(&tmp, bs_old);
1561 bdrv_move_feature_fields(bs_old, bs_new);
1562 bdrv_move_feature_fields(bs_new, &tmp);
8802d1fd 1563
4ddc07ca
PB
1564 /* bs_new shouldn't be in bdrv_states even after the swap! */
1565 assert(bs_new->device_name[0] == '\0');
1566
1567 /* Check a few fields that should remain attached to the device */
1568 assert(bs_new->dev == NULL);
1569 assert(bs_new->job == NULL);
1570 assert(bs_new->in_use == 0);
1571 assert(bs_new->io_limits_enabled == false);
1572 assert(bs_new->block_timer == NULL);
e023b2e2
PB
1573
1574 bdrv_rebind(bs_new);
4ddc07ca
PB
1575 bdrv_rebind(bs_old);
1576}
1577
1578/*
1579 * Add new bs contents at the top of an image chain while the chain is
1580 * live, while keeping required fields on the top layer.
1581 *
1582 * This will modify the BlockDriverState fields, and swap contents
1583 * between bs_new and bs_top. Both bs_new and bs_top are modified.
1584 *
1585 * bs_new is required to be anonymous.
1586 *
1587 * This function does not create any image files.
1588 */
1589void bdrv_append(BlockDriverState *bs_new, BlockDriverState *bs_top)
1590{
1591 bdrv_swap(bs_new, bs_top);
1592
1593 /* The contents of 'tmp' will become bs_top, as we are
1594 * swapping bs_new and bs_top contents. */
1595 bs_top->backing_hd = bs_new;
1596 bs_top->open_flags &= ~BDRV_O_NO_BACKING;
1597 pstrcpy(bs_top->backing_file, sizeof(bs_top->backing_file),
1598 bs_new->filename);
1599 pstrcpy(bs_top->backing_format, sizeof(bs_top->backing_format),
1600 bs_new->drv ? bs_new->drv->format_name : "");
8802d1fd
JC
1601}
1602
b338082b
FB
1603void bdrv_delete(BlockDriverState *bs)
1604{
fa879d62 1605 assert(!bs->dev);
3e914655
PB
1606 assert(!bs->job);
1607 assert(!bs->in_use);
18846dee 1608
1b7bdbc1 1609 /* remove from list, if necessary */
d22b2f41 1610 bdrv_make_anon(bs);
34c6f050 1611
b338082b 1612 bdrv_close(bs);
66f82cee 1613
7267c094 1614 g_free(bs);
fc01f7e7
FB
1615}
1616
fa879d62
MA
1617int bdrv_attach_dev(BlockDriverState *bs, void *dev)
1618/* TODO change to DeviceState *dev when all users are qdevified */
18846dee 1619{
fa879d62 1620 if (bs->dev) {
18846dee
MA
1621 return -EBUSY;
1622 }
fa879d62 1623 bs->dev = dev;
28a7282a 1624 bdrv_iostatus_reset(bs);
18846dee
MA
1625 return 0;
1626}
1627
fa879d62
MA
1628/* TODO qdevified devices don't use this, remove when devices are qdevified */
1629void bdrv_attach_dev_nofail(BlockDriverState *bs, void *dev)
18846dee 1630{
fa879d62
MA
1631 if (bdrv_attach_dev(bs, dev) < 0) {
1632 abort();
1633 }
1634}
1635
1636void bdrv_detach_dev(BlockDriverState *bs, void *dev)
1637/* TODO change to DeviceState *dev when all users are qdevified */
1638{
1639 assert(bs->dev == dev);
1640 bs->dev = NULL;
0e49de52
MA
1641 bs->dev_ops = NULL;
1642 bs->dev_opaque = NULL;
29e05f20 1643 bs->buffer_alignment = 512;
18846dee
MA
1644}
1645
fa879d62
MA
1646/* TODO change to return DeviceState * when all users are qdevified */
1647void *bdrv_get_attached_dev(BlockDriverState *bs)
18846dee 1648{
fa879d62 1649 return bs->dev;
18846dee
MA
1650}
1651
0e49de52
MA
1652void bdrv_set_dev_ops(BlockDriverState *bs, const BlockDevOps *ops,
1653 void *opaque)
1654{
1655 bs->dev_ops = ops;
1656 bs->dev_opaque = opaque;
1657}
1658
32c81a4a
PB
1659void bdrv_emit_qmp_error_event(const BlockDriverState *bdrv,
1660 enum MonitorEvent ev,
1661 BlockErrorAction action, bool is_read)
329c0a48
LC
1662{
1663 QObject *data;
1664 const char *action_str;
1665
1666 switch (action) {
1667 case BDRV_ACTION_REPORT:
1668 action_str = "report";
1669 break;
1670 case BDRV_ACTION_IGNORE:
1671 action_str = "ignore";
1672 break;
1673 case BDRV_ACTION_STOP:
1674 action_str = "stop";
1675 break;
1676 default:
1677 abort();
1678 }
1679
1680 data = qobject_from_jsonf("{ 'device': %s, 'action': %s, 'operation': %s }",
1681 bdrv->device_name,
1682 action_str,
1683 is_read ? "read" : "write");
32c81a4a 1684 monitor_protocol_event(ev, data);
329c0a48
LC
1685
1686 qobject_decref(data);
1687}
1688
6f382ed2
LC
1689static void bdrv_emit_qmp_eject_event(BlockDriverState *bs, bool ejected)
1690{
1691 QObject *data;
1692
1693 data = qobject_from_jsonf("{ 'device': %s, 'tray-open': %i }",
1694 bdrv_get_device_name(bs), ejected);
1695 monitor_protocol_event(QEVENT_DEVICE_TRAY_MOVED, data);
1696
1697 qobject_decref(data);
1698}
1699
7d4b4ba5 1700static void bdrv_dev_change_media_cb(BlockDriverState *bs, bool load)
0e49de52 1701{
145feb17 1702 if (bs->dev_ops && bs->dev_ops->change_media_cb) {
6f382ed2 1703 bool tray_was_closed = !bdrv_dev_is_tray_open(bs);
7d4b4ba5 1704 bs->dev_ops->change_media_cb(bs->dev_opaque, load);
6f382ed2
LC
1705 if (tray_was_closed) {
1706 /* tray open */
1707 bdrv_emit_qmp_eject_event(bs, true);
1708 }
1709 if (load) {
1710 /* tray close */
1711 bdrv_emit_qmp_eject_event(bs, false);
1712 }
145feb17
MA
1713 }
1714}
1715
2c6942fa
MA
1716bool bdrv_dev_has_removable_media(BlockDriverState *bs)
1717{
1718 return !bs->dev || (bs->dev_ops && bs->dev_ops->change_media_cb);
1719}
1720
025ccaa7
PB
1721void bdrv_dev_eject_request(BlockDriverState *bs, bool force)
1722{
1723 if (bs->dev_ops && bs->dev_ops->eject_request_cb) {
1724 bs->dev_ops->eject_request_cb(bs->dev_opaque, force);
1725 }
1726}
1727
e4def80b
MA
1728bool bdrv_dev_is_tray_open(BlockDriverState *bs)
1729{
1730 if (bs->dev_ops && bs->dev_ops->is_tray_open) {
1731 return bs->dev_ops->is_tray_open(bs->dev_opaque);
1732 }
1733 return false;
1734}
1735
145feb17
MA
1736static void bdrv_dev_resize_cb(BlockDriverState *bs)
1737{
1738 if (bs->dev_ops && bs->dev_ops->resize_cb) {
1739 bs->dev_ops->resize_cb(bs->dev_opaque);
0e49de52
MA
1740 }
1741}
1742
f107639a
MA
1743bool bdrv_dev_is_medium_locked(BlockDriverState *bs)
1744{
1745 if (bs->dev_ops && bs->dev_ops->is_medium_locked) {
1746 return bs->dev_ops->is_medium_locked(bs->dev_opaque);
1747 }
1748 return false;
1749}
1750
e97fc193
AL
1751/*
1752 * Run consistency checks on an image
1753 *
e076f338 1754 * Returns 0 if the check could be completed (it doesn't mean that the image is
a1c7273b 1755 * free of errors) or -errno when an internal error occurred. The results of the
e076f338 1756 * check are stored in res.
e97fc193 1757 */
4534ff54 1758int bdrv_check(BlockDriverState *bs, BdrvCheckResult *res, BdrvCheckMode fix)
e97fc193
AL
1759{
1760 if (bs->drv->bdrv_check == NULL) {
1761 return -ENOTSUP;
1762 }
1763
e076f338 1764 memset(res, 0, sizeof(*res));
4534ff54 1765 return bs->drv->bdrv_check(bs, res, fix);
e97fc193
AL
1766}
1767
8a426614
KW
1768#define COMMIT_BUF_SECTORS 2048
1769
33e3963e
FB
1770/* commit COW file into the raw image */
1771int bdrv_commit(BlockDriverState *bs)
1772{
19cb3738 1773 BlockDriver *drv = bs->drv;
8a426614
KW
1774 int64_t sector, total_sectors;
1775 int n, ro, open_flags;
0bce597d 1776 int ret = 0;
8a426614 1777 uint8_t *buf;
c2cba3d9 1778 char filename[PATH_MAX];
33e3963e 1779
19cb3738
FB
1780 if (!drv)
1781 return -ENOMEDIUM;
4dca4b63
NS
1782
1783 if (!bs->backing_hd) {
1784 return -ENOTSUP;
33e3963e
FB
1785 }
1786
2d3735d3
SH
1787 if (bdrv_in_use(bs) || bdrv_in_use(bs->backing_hd)) {
1788 return -EBUSY;
1789 }
1790
4dca4b63 1791 ro = bs->backing_hd->read_only;
c2cba3d9
JM
1792 /* Use pstrcpy (not strncpy): filename must be NUL-terminated. */
1793 pstrcpy(filename, sizeof(filename), bs->backing_hd->filename);
4dca4b63
NS
1794 open_flags = bs->backing_hd->open_flags;
1795
1796 if (ro) {
0bce597d
JC
1797 if (bdrv_reopen(bs->backing_hd, open_flags | BDRV_O_RDWR, NULL)) {
1798 return -EACCES;
4dca4b63 1799 }
ea2384d3 1800 }
33e3963e 1801
6ea44308 1802 total_sectors = bdrv_getlength(bs) >> BDRV_SECTOR_BITS;
7267c094 1803 buf = g_malloc(COMMIT_BUF_SECTORS * BDRV_SECTOR_SIZE);
8a426614
KW
1804
1805 for (sector = 0; sector < total_sectors; sector += n) {
05c4af54 1806 if (bdrv_is_allocated(bs, sector, COMMIT_BUF_SECTORS, &n)) {
8a426614
KW
1807
1808 if (bdrv_read(bs, sector, buf, n) != 0) {
1809 ret = -EIO;
1810 goto ro_cleanup;
1811 }
1812
1813 if (bdrv_write(bs->backing_hd, sector, buf, n) != 0) {
1814 ret = -EIO;
1815 goto ro_cleanup;
1816 }
ea2384d3 1817 }
33e3963e 1818 }
95389c86 1819
1d44952f
CH
1820 if (drv->bdrv_make_empty) {
1821 ret = drv->bdrv_make_empty(bs);
1822 bdrv_flush(bs);
1823 }
95389c86 1824
3f5075ae
CH
1825 /*
1826 * Make sure all data we wrote to the backing device is actually
1827 * stable on disk.
1828 */
1829 if (bs->backing_hd)
1830 bdrv_flush(bs->backing_hd);
4dca4b63
NS
1831
1832ro_cleanup:
7267c094 1833 g_free(buf);
4dca4b63
NS
1834
1835 if (ro) {
0bce597d
JC
1836 /* ignoring error return here */
1837 bdrv_reopen(bs->backing_hd, open_flags & ~BDRV_O_RDWR, NULL);
4dca4b63
NS
1838 }
1839
1d44952f 1840 return ret;
33e3963e
FB
1841}
1842
e8877497 1843int bdrv_commit_all(void)
6ab4b5ab
MA
1844{
1845 BlockDriverState *bs;
1846
1847 QTAILQ_FOREACH(bs, &bdrv_states, list) {
272d2d8e
JC
1848 if (bs->drv && bs->backing_hd) {
1849 int ret = bdrv_commit(bs);
1850 if (ret < 0) {
1851 return ret;
1852 }
e8877497 1853 }
6ab4b5ab 1854 }
e8877497 1855 return 0;
6ab4b5ab
MA
1856}
1857
dbffbdcf
SH
1858/**
1859 * Remove an active request from the tracked requests list
1860 *
1861 * This function should be called when a tracked request is completing.
1862 */
1863static void tracked_request_end(BdrvTrackedRequest *req)
1864{
1865 QLIST_REMOVE(req, list);
f4658285 1866 qemu_co_queue_restart_all(&req->wait_queue);
dbffbdcf
SH
1867}
1868
1869/**
1870 * Add an active request to the tracked requests list
1871 */
1872static void tracked_request_begin(BdrvTrackedRequest *req,
1873 BlockDriverState *bs,
1874 int64_t sector_num,
1875 int nb_sectors, bool is_write)
1876{
1877 *req = (BdrvTrackedRequest){
1878 .bs = bs,
1879 .sector_num = sector_num,
1880 .nb_sectors = nb_sectors,
1881 .is_write = is_write,
5f8b6491 1882 .co = qemu_coroutine_self(),
dbffbdcf
SH
1883 };
1884
f4658285
SH
1885 qemu_co_queue_init(&req->wait_queue);
1886
dbffbdcf
SH
1887 QLIST_INSERT_HEAD(&bs->tracked_requests, req, list);
1888}
1889
d83947ac
SH
1890/**
1891 * Round a region to cluster boundaries
1892 */
343bded4
PB
1893void bdrv_round_to_clusters(BlockDriverState *bs,
1894 int64_t sector_num, int nb_sectors,
1895 int64_t *cluster_sector_num,
1896 int *cluster_nb_sectors)
d83947ac
SH
1897{
1898 BlockDriverInfo bdi;
1899
1900 if (bdrv_get_info(bs, &bdi) < 0 || bdi.cluster_size == 0) {
1901 *cluster_sector_num = sector_num;
1902 *cluster_nb_sectors = nb_sectors;
1903 } else {
1904 int64_t c = bdi.cluster_size / BDRV_SECTOR_SIZE;
1905 *cluster_sector_num = QEMU_ALIGN_DOWN(sector_num, c);
1906 *cluster_nb_sectors = QEMU_ALIGN_UP(sector_num - *cluster_sector_num +
1907 nb_sectors, c);
1908 }
1909}
1910
f4658285
SH
1911static bool tracked_request_overlaps(BdrvTrackedRequest *req,
1912 int64_t sector_num, int nb_sectors) {
d83947ac
SH
1913 /* aaaa bbbb */
1914 if (sector_num >= req->sector_num + req->nb_sectors) {
1915 return false;
1916 }
1917 /* bbbb aaaa */
1918 if (req->sector_num >= sector_num + nb_sectors) {
1919 return false;
1920 }
1921 return true;
f4658285
SH
1922}
1923
1924static void coroutine_fn wait_for_overlapping_requests(BlockDriverState *bs,
1925 int64_t sector_num, int nb_sectors)
1926{
1927 BdrvTrackedRequest *req;
d83947ac
SH
1928 int64_t cluster_sector_num;
1929 int cluster_nb_sectors;
f4658285
SH
1930 bool retry;
1931
d83947ac
SH
1932 /* If we touch the same cluster it counts as an overlap. This guarantees
1933 * that allocating writes will be serialized and not race with each other
1934 * for the same cluster. For example, in copy-on-read it ensures that the
1935 * CoR read and write operations are atomic and guest writes cannot
1936 * interleave between them.
1937 */
343bded4
PB
1938 bdrv_round_to_clusters(bs, sector_num, nb_sectors,
1939 &cluster_sector_num, &cluster_nb_sectors);
d83947ac 1940
f4658285
SH
1941 do {
1942 retry = false;
1943 QLIST_FOREACH(req, &bs->tracked_requests, list) {
d83947ac
SH
1944 if (tracked_request_overlaps(req, cluster_sector_num,
1945 cluster_nb_sectors)) {
5f8b6491
SH
1946 /* Hitting this means there was a reentrant request, for
1947 * example, a block driver issuing nested requests. This must
1948 * never happen since it means deadlock.
1949 */
1950 assert(qemu_coroutine_self() != req->co);
1951
f4658285
SH
1952 qemu_co_queue_wait(&req->wait_queue);
1953 retry = true;
1954 break;
1955 }
1956 }
1957 } while (retry);
1958}
1959
756e6736
KW
1960/*
1961 * Return values:
1962 * 0 - success
1963 * -EINVAL - backing format specified, but no file
1964 * -ENOSPC - can't update the backing file because no space is left in the
1965 * image file header
1966 * -ENOTSUP - format driver doesn't support changing the backing file
1967 */
1968int bdrv_change_backing_file(BlockDriverState *bs,
1969 const char *backing_file, const char *backing_fmt)
1970{
1971 BlockDriver *drv = bs->drv;
469ef350 1972 int ret;
756e6736 1973
5f377794
PB
1974 /* Backing file format doesn't make sense without a backing file */
1975 if (backing_fmt && !backing_file) {
1976 return -EINVAL;
1977 }
1978
756e6736 1979 if (drv->bdrv_change_backing_file != NULL) {
469ef350 1980 ret = drv->bdrv_change_backing_file(bs, backing_file, backing_fmt);
756e6736 1981 } else {
469ef350 1982 ret = -ENOTSUP;
756e6736 1983 }
469ef350
PB
1984
1985 if (ret == 0) {
1986 pstrcpy(bs->backing_file, sizeof(bs->backing_file), backing_file ?: "");
1987 pstrcpy(bs->backing_format, sizeof(bs->backing_format), backing_fmt ?: "");
1988 }
1989 return ret;
756e6736
KW
1990}
1991
6ebdcee2
JC
1992/*
1993 * Finds the image layer in the chain that has 'bs' as its backing file.
1994 *
1995 * active is the current topmost image.
1996 *
1997 * Returns NULL if bs is not found in active's image chain,
1998 * or if active == bs.
1999 */
2000BlockDriverState *bdrv_find_overlay(BlockDriverState *active,
2001 BlockDriverState *bs)
2002{
2003 BlockDriverState *overlay = NULL;
2004 BlockDriverState *intermediate;
2005
2006 assert(active != NULL);
2007 assert(bs != NULL);
2008
2009 /* if bs is the same as active, then by definition it has no overlay
2010 */
2011 if (active == bs) {
2012 return NULL;
2013 }
2014
2015 intermediate = active;
2016 while (intermediate->backing_hd) {
2017 if (intermediate->backing_hd == bs) {
2018 overlay = intermediate;
2019 break;
2020 }
2021 intermediate = intermediate->backing_hd;
2022 }
2023
2024 return overlay;
2025}
2026
2027typedef struct BlkIntermediateStates {
2028 BlockDriverState *bs;
2029 QSIMPLEQ_ENTRY(BlkIntermediateStates) entry;
2030} BlkIntermediateStates;
2031
2032
2033/*
2034 * Drops images above 'base' up to and including 'top', and sets the image
2035 * above 'top' to have base as its backing file.
2036 *
2037 * Requires that the overlay to 'top' is opened r/w, so that the backing file
2038 * information in 'bs' can be properly updated.
2039 *
2040 * E.g., this will convert the following chain:
2041 * bottom <- base <- intermediate <- top <- active
2042 *
2043 * to
2044 *
2045 * bottom <- base <- active
2046 *
2047 * It is allowed for bottom==base, in which case it converts:
2048 *
2049 * base <- intermediate <- top <- active
2050 *
2051 * to
2052 *
2053 * base <- active
2054 *
2055 * Error conditions:
2056 * if active == top, that is considered an error
2057 *
2058 */
2059int bdrv_drop_intermediate(BlockDriverState *active, BlockDriverState *top,
2060 BlockDriverState *base)
2061{
2062 BlockDriverState *intermediate;
2063 BlockDriverState *base_bs = NULL;
2064 BlockDriverState *new_top_bs = NULL;
2065 BlkIntermediateStates *intermediate_state, *next;
2066 int ret = -EIO;
2067
2068 QSIMPLEQ_HEAD(states_to_delete, BlkIntermediateStates) states_to_delete;
2069 QSIMPLEQ_INIT(&states_to_delete);
2070
2071 if (!top->drv || !base->drv) {
2072 goto exit;
2073 }
2074
2075 new_top_bs = bdrv_find_overlay(active, top);
2076
2077 if (new_top_bs == NULL) {
2078 /* we could not find the image above 'top', this is an error */
2079 goto exit;
2080 }
2081
2082 /* special case of new_top_bs->backing_hd already pointing to base - nothing
2083 * to do, no intermediate images */
2084 if (new_top_bs->backing_hd == base) {
2085 ret = 0;
2086 goto exit;
2087 }
2088
2089 intermediate = top;
2090
2091 /* now we will go down through the list, and add each BDS we find
2092 * into our deletion queue, until we hit the 'base'
2093 */
2094 while (intermediate) {
2095 intermediate_state = g_malloc0(sizeof(BlkIntermediateStates));
2096 intermediate_state->bs = intermediate;
2097 QSIMPLEQ_INSERT_TAIL(&states_to_delete, intermediate_state, entry);
2098
2099 if (intermediate->backing_hd == base) {
2100 base_bs = intermediate->backing_hd;
2101 break;
2102 }
2103 intermediate = intermediate->backing_hd;
2104 }
2105 if (base_bs == NULL) {
2106 /* something went wrong, we did not end at the base. safely
2107 * unravel everything, and exit with error */
2108 goto exit;
2109 }
2110
2111 /* success - we can delete the intermediate states, and link top->base */
2112 ret = bdrv_change_backing_file(new_top_bs, base_bs->filename,
2113 base_bs->drv ? base_bs->drv->format_name : "");
2114 if (ret) {
2115 goto exit;
2116 }
2117 new_top_bs->backing_hd = base_bs;
2118
2119
2120 QSIMPLEQ_FOREACH_SAFE(intermediate_state, &states_to_delete, entry, next) {
2121 /* so that bdrv_close() does not recursively close the chain */
2122 intermediate_state->bs->backing_hd = NULL;
2123 bdrv_delete(intermediate_state->bs);
2124 }
2125 ret = 0;
2126
2127exit:
2128 QSIMPLEQ_FOREACH_SAFE(intermediate_state, &states_to_delete, entry, next) {
2129 g_free(intermediate_state);
2130 }
2131 return ret;
2132}
2133
2134
71d0770c
AL
2135static int bdrv_check_byte_request(BlockDriverState *bs, int64_t offset,
2136 size_t size)
2137{
2138 int64_t len;
2139
2140 if (!bdrv_is_inserted(bs))
2141 return -ENOMEDIUM;
2142
2143 if (bs->growable)
2144 return 0;
2145
2146 len = bdrv_getlength(bs);
2147
fbb7b4e0
KW
2148 if (offset < 0)
2149 return -EIO;
2150
2151 if ((offset > len) || (len - offset < size))
71d0770c
AL
2152 return -EIO;
2153
2154 return 0;
2155}
2156
2157static int bdrv_check_request(BlockDriverState *bs, int64_t sector_num,
2158 int nb_sectors)
2159{
eb5a3165
JS
2160 return bdrv_check_byte_request(bs, sector_num * BDRV_SECTOR_SIZE,
2161 nb_sectors * BDRV_SECTOR_SIZE);
71d0770c
AL
2162}
2163
1c9805a3
SH
2164typedef struct RwCo {
2165 BlockDriverState *bs;
2166 int64_t sector_num;
2167 int nb_sectors;
2168 QEMUIOVector *qiov;
2169 bool is_write;
2170 int ret;
4105eaaa 2171 BdrvRequestFlags flags;
1c9805a3
SH
2172} RwCo;
2173
2174static void coroutine_fn bdrv_rw_co_entry(void *opaque)
fc01f7e7 2175{
1c9805a3 2176 RwCo *rwco = opaque;
ea2384d3 2177
1c9805a3
SH
2178 if (!rwco->is_write) {
2179 rwco->ret = bdrv_co_do_readv(rwco->bs, rwco->sector_num,
4105eaaa
PL
2180 rwco->nb_sectors, rwco->qiov,
2181 rwco->flags);
1c9805a3
SH
2182 } else {
2183 rwco->ret = bdrv_co_do_writev(rwco->bs, rwco->sector_num,
4105eaaa
PL
2184 rwco->nb_sectors, rwco->qiov,
2185 rwco->flags);
1c9805a3
SH
2186 }
2187}
e7a8a783 2188
1c9805a3 2189/*
8d3b1a2d 2190 * Process a vectored synchronous request using coroutines
1c9805a3 2191 */
8d3b1a2d 2192static int bdrv_rwv_co(BlockDriverState *bs, int64_t sector_num,
4105eaaa
PL
2193 QEMUIOVector *qiov, bool is_write,
2194 BdrvRequestFlags flags)
1c9805a3 2195{
1c9805a3
SH
2196 Coroutine *co;
2197 RwCo rwco = {
2198 .bs = bs,
2199 .sector_num = sector_num,
8d3b1a2d
KW
2200 .nb_sectors = qiov->size >> BDRV_SECTOR_BITS,
2201 .qiov = qiov,
1c9805a3
SH
2202 .is_write = is_write,
2203 .ret = NOT_DONE,
4105eaaa 2204 .flags = flags,
1c9805a3 2205 };
8d3b1a2d 2206 assert((qiov->size & (BDRV_SECTOR_SIZE - 1)) == 0);
e7a8a783 2207
498e386c
ZYW
2208 /**
2209 * In sync call context, when the vcpu is blocked, this throttling timer
2210 * will not fire; so the I/O throttling function has to be disabled here
2211 * if it has been enabled.
2212 */
2213 if (bs->io_limits_enabled) {
2214 fprintf(stderr, "Disabling I/O throttling on '%s' due "
2215 "to synchronous I/O.\n", bdrv_get_device_name(bs));
2216 bdrv_io_limits_disable(bs);
2217 }
2218
1c9805a3
SH
2219 if (qemu_in_coroutine()) {
2220 /* Fast-path if already in coroutine context */
2221 bdrv_rw_co_entry(&rwco);
2222 } else {
2223 co = qemu_coroutine_create(bdrv_rw_co_entry);
2224 qemu_coroutine_enter(co, &rwco);
2225 while (rwco.ret == NOT_DONE) {
2226 qemu_aio_wait();
2227 }
2228 }
2229 return rwco.ret;
2230}
b338082b 2231
8d3b1a2d
KW
2232/*
2233 * Process a synchronous request using coroutines
2234 */
2235static int bdrv_rw_co(BlockDriverState *bs, int64_t sector_num, uint8_t *buf,
4105eaaa 2236 int nb_sectors, bool is_write, BdrvRequestFlags flags)
8d3b1a2d
KW
2237{
2238 QEMUIOVector qiov;
2239 struct iovec iov = {
2240 .iov_base = (void *)buf,
2241 .iov_len = nb_sectors * BDRV_SECTOR_SIZE,
2242 };
2243
2244 qemu_iovec_init_external(&qiov, &iov, 1);
4105eaaa 2245 return bdrv_rwv_co(bs, sector_num, &qiov, is_write, flags);
8d3b1a2d
KW
2246}
2247
1c9805a3
SH
2248/* return < 0 if error. See bdrv_write() for the return codes */
2249int bdrv_read(BlockDriverState *bs, int64_t sector_num,
2250 uint8_t *buf, int nb_sectors)
2251{
4105eaaa 2252 return bdrv_rw_co(bs, sector_num, buf, nb_sectors, false, 0);
fc01f7e7
FB
2253}
2254
07d27a44
MA
2255/* Just like bdrv_read(), but with I/O throttling temporarily disabled */
2256int bdrv_read_unthrottled(BlockDriverState *bs, int64_t sector_num,
2257 uint8_t *buf, int nb_sectors)
2258{
2259 bool enabled;
2260 int ret;
2261
2262 enabled = bs->io_limits_enabled;
2263 bs->io_limits_enabled = false;
4e7395e8 2264 ret = bdrv_read(bs, sector_num, buf, nb_sectors);
07d27a44
MA
2265 bs->io_limits_enabled = enabled;
2266 return ret;
2267}
2268
5fafdf24 2269/* Return < 0 if error. Important errors are:
19cb3738
FB
2270 -EIO generic I/O error (may happen for all errors)
2271 -ENOMEDIUM No media inserted.
2272 -EINVAL Invalid sector number or nb_sectors
2273 -EACCES Trying to write a read-only device
2274*/
5fafdf24 2275int bdrv_write(BlockDriverState *bs, int64_t sector_num,
fc01f7e7
FB
2276 const uint8_t *buf, int nb_sectors)
2277{
4105eaaa 2278 return bdrv_rw_co(bs, sector_num, (uint8_t *)buf, nb_sectors, true, 0);
83f64091
FB
2279}
2280
8d3b1a2d
KW
2281int bdrv_writev(BlockDriverState *bs, int64_t sector_num, QEMUIOVector *qiov)
2282{
4105eaaa
PL
2283 return bdrv_rwv_co(bs, sector_num, qiov, true, 0);
2284}
2285
2286int bdrv_write_zeroes(BlockDriverState *bs, int64_t sector_num, int nb_sectors)
2287{
2288 return bdrv_rw_co(bs, sector_num, NULL, nb_sectors, true,
2289 BDRV_REQ_ZERO_WRITE);
8d3b1a2d
KW
2290}
2291
eda578e5
AL
2292int bdrv_pread(BlockDriverState *bs, int64_t offset,
2293 void *buf, int count1)
83f64091 2294{
6ea44308 2295 uint8_t tmp_buf[BDRV_SECTOR_SIZE];
83f64091
FB
2296 int len, nb_sectors, count;
2297 int64_t sector_num;
9a8c4cce 2298 int ret;
83f64091
FB
2299
2300 count = count1;
2301 /* first read to align to sector start */
6ea44308 2302 len = (BDRV_SECTOR_SIZE - offset) & (BDRV_SECTOR_SIZE - 1);
83f64091
FB
2303 if (len > count)
2304 len = count;
6ea44308 2305 sector_num = offset >> BDRV_SECTOR_BITS;
83f64091 2306 if (len > 0) {
9a8c4cce
KW
2307 if ((ret = bdrv_read(bs, sector_num, tmp_buf, 1)) < 0)
2308 return ret;
6ea44308 2309 memcpy(buf, tmp_buf + (offset & (BDRV_SECTOR_SIZE - 1)), len);
83f64091
FB
2310 count -= len;
2311 if (count == 0)
2312 return count1;
2313 sector_num++;
2314 buf += len;
2315 }
2316
2317 /* read the sectors "in place" */
6ea44308 2318 nb_sectors = count >> BDRV_SECTOR_BITS;
83f64091 2319 if (nb_sectors > 0) {
9a8c4cce
KW
2320 if ((ret = bdrv_read(bs, sector_num, buf, nb_sectors)) < 0)
2321 return ret;
83f64091 2322 sector_num += nb_sectors;
6ea44308 2323 len = nb_sectors << BDRV_SECTOR_BITS;
83f64091
FB
2324 buf += len;
2325 count -= len;
2326 }
2327
2328 /* add data from the last sector */
2329 if (count > 0) {
9a8c4cce
KW
2330 if ((ret = bdrv_read(bs, sector_num, tmp_buf, 1)) < 0)
2331 return ret;
83f64091
FB
2332 memcpy(buf, tmp_buf, count);
2333 }
2334 return count1;
2335}
2336
8d3b1a2d 2337int bdrv_pwritev(BlockDriverState *bs, int64_t offset, QEMUIOVector *qiov)
83f64091 2338{
6ea44308 2339 uint8_t tmp_buf[BDRV_SECTOR_SIZE];
83f64091
FB
2340 int len, nb_sectors, count;
2341 int64_t sector_num;
9a8c4cce 2342 int ret;
83f64091 2343
8d3b1a2d
KW
2344 count = qiov->size;
2345
83f64091 2346 /* first write to align to sector start */
6ea44308 2347 len = (BDRV_SECTOR_SIZE - offset) & (BDRV_SECTOR_SIZE - 1);
83f64091
FB
2348 if (len > count)
2349 len = count;
6ea44308 2350 sector_num = offset >> BDRV_SECTOR_BITS;
83f64091 2351 if (len > 0) {
9a8c4cce
KW
2352 if ((ret = bdrv_read(bs, sector_num, tmp_buf, 1)) < 0)
2353 return ret;
8d3b1a2d
KW
2354 qemu_iovec_to_buf(qiov, 0, tmp_buf + (offset & (BDRV_SECTOR_SIZE - 1)),
2355 len);
9a8c4cce
KW
2356 if ((ret = bdrv_write(bs, sector_num, tmp_buf, 1)) < 0)
2357 return ret;
83f64091
FB
2358 count -= len;
2359 if (count == 0)
8d3b1a2d 2360 return qiov->size;
83f64091 2361 sector_num++;
83f64091
FB
2362 }
2363
2364 /* write the sectors "in place" */
6ea44308 2365 nb_sectors = count >> BDRV_SECTOR_BITS;
83f64091 2366 if (nb_sectors > 0) {
8d3b1a2d
KW
2367 QEMUIOVector qiov_inplace;
2368
2369 qemu_iovec_init(&qiov_inplace, qiov->niov);
2370 qemu_iovec_concat(&qiov_inplace, qiov, len,
2371 nb_sectors << BDRV_SECTOR_BITS);
2372 ret = bdrv_writev(bs, sector_num, &qiov_inplace);
2373 qemu_iovec_destroy(&qiov_inplace);
2374 if (ret < 0) {
9a8c4cce 2375 return ret;
8d3b1a2d
KW
2376 }
2377
83f64091 2378 sector_num += nb_sectors;
6ea44308 2379 len = nb_sectors << BDRV_SECTOR_BITS;
83f64091
FB
2380 count -= len;
2381 }
2382
2383 /* add data from the last sector */
2384 if (count > 0) {
9a8c4cce
KW
2385 if ((ret = bdrv_read(bs, sector_num, tmp_buf, 1)) < 0)
2386 return ret;
8d3b1a2d 2387 qemu_iovec_to_buf(qiov, qiov->size - count, tmp_buf, count);
9a8c4cce
KW
2388 if ((ret = bdrv_write(bs, sector_num, tmp_buf, 1)) < 0)
2389 return ret;
83f64091 2390 }
8d3b1a2d
KW
2391 return qiov->size;
2392}
2393
2394int bdrv_pwrite(BlockDriverState *bs, int64_t offset,
2395 const void *buf, int count1)
2396{
2397 QEMUIOVector qiov;
2398 struct iovec iov = {
2399 .iov_base = (void *) buf,
2400 .iov_len = count1,
2401 };
2402
2403 qemu_iovec_init_external(&qiov, &iov, 1);
2404 return bdrv_pwritev(bs, offset, &qiov);
83f64091 2405}
83f64091 2406
f08145fe
KW
2407/*
2408 * Writes to the file and ensures that no writes are reordered across this
2409 * request (acts as a barrier)
2410 *
2411 * Returns 0 on success, -errno in error cases.
2412 */
2413int bdrv_pwrite_sync(BlockDriverState *bs, int64_t offset,
2414 const void *buf, int count)
2415{
2416 int ret;
2417
2418 ret = bdrv_pwrite(bs, offset, buf, count);
2419 if (ret < 0) {
2420 return ret;
2421 }
2422
f05fa4ad
PB
2423 /* No flush needed for cache modes that already do it */
2424 if (bs->enable_write_cache) {
f08145fe
KW
2425 bdrv_flush(bs);
2426 }
2427
2428 return 0;
2429}
2430
470c0504 2431static int coroutine_fn bdrv_co_do_copy_on_readv(BlockDriverState *bs,
ab185921
SH
2432 int64_t sector_num, int nb_sectors, QEMUIOVector *qiov)
2433{
2434 /* Perform I/O through a temporary buffer so that users who scribble over
2435 * their read buffer while the operation is in progress do not end up
2436 * modifying the image file. This is critical for zero-copy guest I/O
2437 * where anything might happen inside guest memory.
2438 */
2439 void *bounce_buffer;
2440
79c053bd 2441 BlockDriver *drv = bs->drv;
ab185921
SH
2442 struct iovec iov;
2443 QEMUIOVector bounce_qiov;
2444 int64_t cluster_sector_num;
2445 int cluster_nb_sectors;
2446 size_t skip_bytes;
2447 int ret;
2448
2449 /* Cover entire cluster so no additional backing file I/O is required when
2450 * allocating cluster in the image file.
2451 */
343bded4
PB
2452 bdrv_round_to_clusters(bs, sector_num, nb_sectors,
2453 &cluster_sector_num, &cluster_nb_sectors);
ab185921 2454
470c0504
SH
2455 trace_bdrv_co_do_copy_on_readv(bs, sector_num, nb_sectors,
2456 cluster_sector_num, cluster_nb_sectors);
ab185921
SH
2457
2458 iov.iov_len = cluster_nb_sectors * BDRV_SECTOR_SIZE;
2459 iov.iov_base = bounce_buffer = qemu_blockalign(bs, iov.iov_len);
2460 qemu_iovec_init_external(&bounce_qiov, &iov, 1);
2461
79c053bd
SH
2462 ret = drv->bdrv_co_readv(bs, cluster_sector_num, cluster_nb_sectors,
2463 &bounce_qiov);
ab185921
SH
2464 if (ret < 0) {
2465 goto err;
2466 }
2467
79c053bd
SH
2468 if (drv->bdrv_co_write_zeroes &&
2469 buffer_is_zero(bounce_buffer, iov.iov_len)) {
621f0589
KW
2470 ret = bdrv_co_do_write_zeroes(bs, cluster_sector_num,
2471 cluster_nb_sectors);
79c053bd 2472 } else {
f05fa4ad
PB
2473 /* This does not change the data on the disk, it is not necessary
2474 * to flush even in cache=writethrough mode.
2475 */
79c053bd 2476 ret = drv->bdrv_co_writev(bs, cluster_sector_num, cluster_nb_sectors,
ab185921 2477 &bounce_qiov);
79c053bd
SH
2478 }
2479
ab185921
SH
2480 if (ret < 0) {
2481 /* It might be okay to ignore write errors for guest requests. If this
2482 * is a deliberate copy-on-read then we don't want to ignore the error.
2483 * Simply report it in all cases.
2484 */
2485 goto err;
2486 }
2487
2488 skip_bytes = (sector_num - cluster_sector_num) * BDRV_SECTOR_SIZE;
03396148
MT
2489 qemu_iovec_from_buf(qiov, 0, bounce_buffer + skip_bytes,
2490 nb_sectors * BDRV_SECTOR_SIZE);
ab185921
SH
2491
2492err:
2493 qemu_vfree(bounce_buffer);
2494 return ret;
2495}
2496
c5fbe571
SH
2497/*
2498 * Handle a read request in coroutine context
2499 */
2500static int coroutine_fn bdrv_co_do_readv(BlockDriverState *bs,
470c0504
SH
2501 int64_t sector_num, int nb_sectors, QEMUIOVector *qiov,
2502 BdrvRequestFlags flags)
da1fa91d
KW
2503{
2504 BlockDriver *drv = bs->drv;
dbffbdcf
SH
2505 BdrvTrackedRequest req;
2506 int ret;
da1fa91d 2507
da1fa91d
KW
2508 if (!drv) {
2509 return -ENOMEDIUM;
2510 }
2511 if (bdrv_check_request(bs, sector_num, nb_sectors)) {
2512 return -EIO;
2513 }
2514
98f90dba
ZYW
2515 /* throttling disk read I/O */
2516 if (bs->io_limits_enabled) {
2517 bdrv_io_limits_intercept(bs, false, nb_sectors);
2518 }
2519
f4658285 2520 if (bs->copy_on_read) {
470c0504
SH
2521 flags |= BDRV_REQ_COPY_ON_READ;
2522 }
2523 if (flags & BDRV_REQ_COPY_ON_READ) {
2524 bs->copy_on_read_in_flight++;
2525 }
2526
2527 if (bs->copy_on_read_in_flight) {
f4658285
SH
2528 wait_for_overlapping_requests(bs, sector_num, nb_sectors);
2529 }
2530
dbffbdcf 2531 tracked_request_begin(&req, bs, sector_num, nb_sectors, false);
ab185921 2532
470c0504 2533 if (flags & BDRV_REQ_COPY_ON_READ) {
ab185921
SH
2534 int pnum;
2535
2536 ret = bdrv_co_is_allocated(bs, sector_num, nb_sectors, &pnum);
2537 if (ret < 0) {
2538 goto out;
2539 }
2540
2541 if (!ret || pnum != nb_sectors) {
470c0504 2542 ret = bdrv_co_do_copy_on_readv(bs, sector_num, nb_sectors, qiov);
ab185921
SH
2543 goto out;
2544 }
2545 }
2546
dbffbdcf 2547 ret = drv->bdrv_co_readv(bs, sector_num, nb_sectors, qiov);
ab185921
SH
2548
2549out:
dbffbdcf 2550 tracked_request_end(&req);
470c0504
SH
2551
2552 if (flags & BDRV_REQ_COPY_ON_READ) {
2553 bs->copy_on_read_in_flight--;
2554 }
2555
dbffbdcf 2556 return ret;
da1fa91d
KW
2557}
2558
c5fbe571 2559int coroutine_fn bdrv_co_readv(BlockDriverState *bs, int64_t sector_num,
da1fa91d
KW
2560 int nb_sectors, QEMUIOVector *qiov)
2561{
c5fbe571 2562 trace_bdrv_co_readv(bs, sector_num, nb_sectors);
da1fa91d 2563
470c0504
SH
2564 return bdrv_co_do_readv(bs, sector_num, nb_sectors, qiov, 0);
2565}
2566
2567int coroutine_fn bdrv_co_copy_on_readv(BlockDriverState *bs,
2568 int64_t sector_num, int nb_sectors, QEMUIOVector *qiov)
2569{
2570 trace_bdrv_co_copy_on_readv(bs, sector_num, nb_sectors);
2571
2572 return bdrv_co_do_readv(bs, sector_num, nb_sectors, qiov,
2573 BDRV_REQ_COPY_ON_READ);
c5fbe571
SH
2574}
2575
f08f2dda
SH
2576static int coroutine_fn bdrv_co_do_write_zeroes(BlockDriverState *bs,
2577 int64_t sector_num, int nb_sectors)
2578{
2579 BlockDriver *drv = bs->drv;
2580 QEMUIOVector qiov;
2581 struct iovec iov;
2582 int ret;
2583
621f0589
KW
2584 /* TODO Emulate only part of misaligned requests instead of letting block
2585 * drivers return -ENOTSUP and emulate everything */
2586
f08f2dda
SH
2587 /* First try the efficient write zeroes operation */
2588 if (drv->bdrv_co_write_zeroes) {
621f0589
KW
2589 ret = drv->bdrv_co_write_zeroes(bs, sector_num, nb_sectors);
2590 if (ret != -ENOTSUP) {
2591 return ret;
2592 }
f08f2dda
SH
2593 }
2594
2595 /* Fall back to bounce buffer if write zeroes is unsupported */
2596 iov.iov_len = nb_sectors * BDRV_SECTOR_SIZE;
2597 iov.iov_base = qemu_blockalign(bs, iov.iov_len);
2598 memset(iov.iov_base, 0, iov.iov_len);
2599 qemu_iovec_init_external(&qiov, &iov, 1);
2600
2601 ret = drv->bdrv_co_writev(bs, sector_num, nb_sectors, &qiov);
2602
2603 qemu_vfree(iov.iov_base);
2604 return ret;
2605}
2606
c5fbe571
SH
2607/*
2608 * Handle a write request in coroutine context
2609 */
2610static int coroutine_fn bdrv_co_do_writev(BlockDriverState *bs,
f08f2dda
SH
2611 int64_t sector_num, int nb_sectors, QEMUIOVector *qiov,
2612 BdrvRequestFlags flags)
c5fbe571
SH
2613{
2614 BlockDriver *drv = bs->drv;
dbffbdcf 2615 BdrvTrackedRequest req;
6b7cb247 2616 int ret;
da1fa91d
KW
2617
2618 if (!bs->drv) {
2619 return -ENOMEDIUM;
2620 }
2621 if (bs->read_only) {
2622 return -EACCES;
2623 }
2624 if (bdrv_check_request(bs, sector_num, nb_sectors)) {
2625 return -EIO;
2626 }
2627
98f90dba
ZYW
2628 /* throttling disk write I/O */
2629 if (bs->io_limits_enabled) {
2630 bdrv_io_limits_intercept(bs, true, nb_sectors);
2631 }
2632
470c0504 2633 if (bs->copy_on_read_in_flight) {
f4658285
SH
2634 wait_for_overlapping_requests(bs, sector_num, nb_sectors);
2635 }
2636
dbffbdcf
SH
2637 tracked_request_begin(&req, bs, sector_num, nb_sectors, true);
2638
d616b224
SH
2639 ret = notifier_with_return_list_notify(&bs->before_write_notifiers, &req);
2640
2641 if (ret < 0) {
2642 /* Do nothing, write notifier decided to fail this request */
2643 } else if (flags & BDRV_REQ_ZERO_WRITE) {
f08f2dda
SH
2644 ret = bdrv_co_do_write_zeroes(bs, sector_num, nb_sectors);
2645 } else {
2646 ret = drv->bdrv_co_writev(bs, sector_num, nb_sectors, qiov);
2647 }
6b7cb247 2648
f05fa4ad
PB
2649 if (ret == 0 && !bs->enable_write_cache) {
2650 ret = bdrv_co_flush(bs);
2651 }
2652
da1fa91d 2653 if (bs->dirty_bitmap) {
1755da16 2654 bdrv_set_dirty(bs, sector_num, nb_sectors);
da1fa91d
KW
2655 }
2656
2657 if (bs->wr_highest_sector < sector_num + nb_sectors - 1) {
2658 bs->wr_highest_sector = sector_num + nb_sectors - 1;
2659 }
2660
dbffbdcf
SH
2661 tracked_request_end(&req);
2662
6b7cb247 2663 return ret;
da1fa91d
KW
2664}
2665
c5fbe571
SH
2666int coroutine_fn bdrv_co_writev(BlockDriverState *bs, int64_t sector_num,
2667 int nb_sectors, QEMUIOVector *qiov)
2668{
2669 trace_bdrv_co_writev(bs, sector_num, nb_sectors);
2670
f08f2dda
SH
2671 return bdrv_co_do_writev(bs, sector_num, nb_sectors, qiov, 0);
2672}
2673
2674int coroutine_fn bdrv_co_write_zeroes(BlockDriverState *bs,
2675 int64_t sector_num, int nb_sectors)
2676{
2677 trace_bdrv_co_write_zeroes(bs, sector_num, nb_sectors);
2678
2679 return bdrv_co_do_writev(bs, sector_num, nb_sectors, NULL,
2680 BDRV_REQ_ZERO_WRITE);
c5fbe571
SH
2681}
2682
83f64091
FB
2683/**
2684 * Truncate file to 'offset' bytes (needed only for file protocols)
2685 */
2686int bdrv_truncate(BlockDriverState *bs, int64_t offset)
2687{
2688 BlockDriver *drv = bs->drv;
51762288 2689 int ret;
83f64091 2690 if (!drv)
19cb3738 2691 return -ENOMEDIUM;
83f64091
FB
2692 if (!drv->bdrv_truncate)
2693 return -ENOTSUP;
59f2689d
NS
2694 if (bs->read_only)
2695 return -EACCES;
8591675f
MT
2696 if (bdrv_in_use(bs))
2697 return -EBUSY;
51762288
SH
2698 ret = drv->bdrv_truncate(bs, offset);
2699 if (ret == 0) {
2700 ret = refresh_total_sectors(bs, offset >> BDRV_SECTOR_BITS);
145feb17 2701 bdrv_dev_resize_cb(bs);
51762288
SH
2702 }
2703 return ret;
83f64091
FB
2704}
2705
4a1d5e1f
FZ
2706/**
2707 * Length of a allocated file in bytes. Sparse files are counted by actual
2708 * allocated space. Return < 0 if error or unknown.
2709 */
2710int64_t bdrv_get_allocated_file_size(BlockDriverState *bs)
2711{
2712 BlockDriver *drv = bs->drv;
2713 if (!drv) {
2714 return -ENOMEDIUM;
2715 }
2716 if (drv->bdrv_get_allocated_file_size) {
2717 return drv->bdrv_get_allocated_file_size(bs);
2718 }
2719 if (bs->file) {
2720 return bdrv_get_allocated_file_size(bs->file);
2721 }
2722 return -ENOTSUP;
2723}
2724
83f64091
FB
2725/**
2726 * Length of a file in bytes. Return < 0 if error or unknown.
2727 */
2728int64_t bdrv_getlength(BlockDriverState *bs)
2729{
2730 BlockDriver *drv = bs->drv;
2731 if (!drv)
19cb3738 2732 return -ENOMEDIUM;
51762288 2733
2c6942fa 2734 if (bs->growable || bdrv_dev_has_removable_media(bs)) {
46a4e4e6
SH
2735 if (drv->bdrv_getlength) {
2736 return drv->bdrv_getlength(bs);
2737 }
83f64091 2738 }
46a4e4e6 2739 return bs->total_sectors * BDRV_SECTOR_SIZE;
fc01f7e7
FB
2740}
2741
19cb3738 2742/* return 0 as number of sectors if no device present or error */
96b8f136 2743void bdrv_get_geometry(BlockDriverState *bs, uint64_t *nb_sectors_ptr)
fc01f7e7 2744{
19cb3738
FB
2745 int64_t length;
2746 length = bdrv_getlength(bs);
2747 if (length < 0)
2748 length = 0;
2749 else
6ea44308 2750 length = length >> BDRV_SECTOR_BITS;
19cb3738 2751 *nb_sectors_ptr = length;
fc01f7e7 2752}
cf98951b 2753
0563e191
ZYW
2754/* throttling disk io limits */
2755void bdrv_set_io_limits(BlockDriverState *bs,
2756 BlockIOLimit *io_limits)
2757{
2758 bs->io_limits = *io_limits;
2759 bs->io_limits_enabled = bdrv_io_limits_enabled(bs);
2760}
2761
ff06f5f3
PB
2762void bdrv_set_on_error(BlockDriverState *bs, BlockdevOnError on_read_error,
2763 BlockdevOnError on_write_error)
abd7f68d
MA
2764{
2765 bs->on_read_error = on_read_error;
2766 bs->on_write_error = on_write_error;
2767}
2768
1ceee0d5 2769BlockdevOnError bdrv_get_on_error(BlockDriverState *bs, bool is_read)
abd7f68d
MA
2770{
2771 return is_read ? bs->on_read_error : bs->on_write_error;
2772}
2773
3e1caa5f
PB
2774BlockErrorAction bdrv_get_error_action(BlockDriverState *bs, bool is_read, int error)
2775{
2776 BlockdevOnError on_err = is_read ? bs->on_read_error : bs->on_write_error;
2777
2778 switch (on_err) {
2779 case BLOCKDEV_ON_ERROR_ENOSPC:
2780 return (error == ENOSPC) ? BDRV_ACTION_STOP : BDRV_ACTION_REPORT;
2781 case BLOCKDEV_ON_ERROR_STOP:
2782 return BDRV_ACTION_STOP;
2783 case BLOCKDEV_ON_ERROR_REPORT:
2784 return BDRV_ACTION_REPORT;
2785 case BLOCKDEV_ON_ERROR_IGNORE:
2786 return BDRV_ACTION_IGNORE;
2787 default:
2788 abort();
2789 }
2790}
2791
2792/* This is done by device models because, while the block layer knows
2793 * about the error, it does not know whether an operation comes from
2794 * the device or the block layer (from a job, for example).
2795 */
2796void bdrv_error_action(BlockDriverState *bs, BlockErrorAction action,
2797 bool is_read, int error)
2798{
2799 assert(error >= 0);
32c81a4a 2800 bdrv_emit_qmp_error_event(bs, QEVENT_BLOCK_IO_ERROR, action, is_read);
3e1caa5f
PB
2801 if (action == BDRV_ACTION_STOP) {
2802 vm_stop(RUN_STATE_IO_ERROR);
2803 bdrv_iostatus_set_err(bs, error);
2804 }
2805}
2806
b338082b
FB
2807int bdrv_is_read_only(BlockDriverState *bs)
2808{
2809 return bs->read_only;
2810}
2811
985a03b0
TS
2812int bdrv_is_sg(BlockDriverState *bs)
2813{
2814 return bs->sg;
2815}
2816
e900a7b7
CH
2817int bdrv_enable_write_cache(BlockDriverState *bs)
2818{
2819 return bs->enable_write_cache;
2820}
2821
425b0148
PB
2822void bdrv_set_enable_write_cache(BlockDriverState *bs, bool wce)
2823{
2824 bs->enable_write_cache = wce;
55b110f2
JC
2825
2826 /* so a reopen() will preserve wce */
2827 if (wce) {
2828 bs->open_flags |= BDRV_O_CACHE_WB;
2829 } else {
2830 bs->open_flags &= ~BDRV_O_CACHE_WB;
2831 }
425b0148
PB
2832}
2833
ea2384d3
FB
2834int bdrv_is_encrypted(BlockDriverState *bs)
2835{
2836 if (bs->backing_hd && bs->backing_hd->encrypted)
2837 return 1;
2838 return bs->encrypted;
2839}
2840
c0f4ce77
AL
2841int bdrv_key_required(BlockDriverState *bs)
2842{
2843 BlockDriverState *backing_hd = bs->backing_hd;
2844
2845 if (backing_hd && backing_hd->encrypted && !backing_hd->valid_key)
2846 return 1;
2847 return (bs->encrypted && !bs->valid_key);
2848}
2849
ea2384d3
FB
2850int bdrv_set_key(BlockDriverState *bs, const char *key)
2851{
2852 int ret;
2853 if (bs->backing_hd && bs->backing_hd->encrypted) {
2854 ret = bdrv_set_key(bs->backing_hd, key);
2855 if (ret < 0)
2856 return ret;
2857 if (!bs->encrypted)
2858 return 0;
2859 }
fd04a2ae
SH
2860 if (!bs->encrypted) {
2861 return -EINVAL;
2862 } else if (!bs->drv || !bs->drv->bdrv_set_key) {
2863 return -ENOMEDIUM;
2864 }
c0f4ce77 2865 ret = bs->drv->bdrv_set_key(bs, key);
bb5fc20f
AL
2866 if (ret < 0) {
2867 bs->valid_key = 0;
2868 } else if (!bs->valid_key) {
2869 bs->valid_key = 1;
2870 /* call the change callback now, we skipped it on open */
7d4b4ba5 2871 bdrv_dev_change_media_cb(bs, true);
bb5fc20f 2872 }
c0f4ce77 2873 return ret;
ea2384d3
FB
2874}
2875
f8d6bba1 2876const char *bdrv_get_format_name(BlockDriverState *bs)
ea2384d3 2877{
f8d6bba1 2878 return bs->drv ? bs->drv->format_name : NULL;
ea2384d3
FB
2879}
2880
5fafdf24 2881void bdrv_iterate_format(void (*it)(void *opaque, const char *name),
ea2384d3
FB
2882 void *opaque)
2883{
2884 BlockDriver *drv;
2885
8a22f02a 2886 QLIST_FOREACH(drv, &bdrv_drivers, list) {
ea2384d3
FB
2887 it(opaque, drv->format_name);
2888 }
2889}
2890
b338082b
FB
2891BlockDriverState *bdrv_find(const char *name)
2892{
2893 BlockDriverState *bs;
2894
1b7bdbc1
SH
2895 QTAILQ_FOREACH(bs, &bdrv_states, list) {
2896 if (!strcmp(name, bs->device_name)) {
b338082b 2897 return bs;
1b7bdbc1 2898 }
b338082b
FB
2899 }
2900 return NULL;
2901}
2902
2f399b0a
MA
2903BlockDriverState *bdrv_next(BlockDriverState *bs)
2904{
2905 if (!bs) {
2906 return QTAILQ_FIRST(&bdrv_states);
2907 }
2908 return QTAILQ_NEXT(bs, list);
2909}
2910
51de9760 2911void bdrv_iterate(void (*it)(void *opaque, BlockDriverState *bs), void *opaque)
81d0912d
FB
2912{
2913 BlockDriverState *bs;
2914
1b7bdbc1 2915 QTAILQ_FOREACH(bs, &bdrv_states, list) {
51de9760 2916 it(opaque, bs);
81d0912d
FB
2917 }
2918}
2919
ea2384d3
FB
2920const char *bdrv_get_device_name(BlockDriverState *bs)
2921{
2922 return bs->device_name;
2923}
2924
c8433287
MA
2925int bdrv_get_flags(BlockDriverState *bs)
2926{
2927 return bs->open_flags;
2928}
2929
f0f0fdfe 2930int bdrv_flush_all(void)
c6ca28d6
AL
2931{
2932 BlockDriverState *bs;
f0f0fdfe 2933 int result = 0;
c6ca28d6 2934
1b7bdbc1 2935 QTAILQ_FOREACH(bs, &bdrv_states, list) {
f0f0fdfe
KW
2936 int ret = bdrv_flush(bs);
2937 if (ret < 0 && !result) {
2938 result = ret;
2939 }
1b7bdbc1 2940 }
f0f0fdfe
KW
2941
2942 return result;
c6ca28d6
AL
2943}
2944
3ac21627
PL
2945int bdrv_has_zero_init_1(BlockDriverState *bs)
2946{
2947 return 1;
2948}
2949
f2feebbd
KW
2950int bdrv_has_zero_init(BlockDriverState *bs)
2951{
2952 assert(bs->drv);
2953
336c1c12
KW
2954 if (bs->drv->bdrv_has_zero_init) {
2955 return bs->drv->bdrv_has_zero_init(bs);
f2feebbd
KW
2956 }
2957
3ac21627
PL
2958 /* safe default */
2959 return 0;
f2feebbd
KW
2960}
2961
376ae3f1
SH
2962typedef struct BdrvCoIsAllocatedData {
2963 BlockDriverState *bs;
b35b2bba 2964 BlockDriverState *base;
376ae3f1
SH
2965 int64_t sector_num;
2966 int nb_sectors;
2967 int *pnum;
2968 int ret;
2969 bool done;
2970} BdrvCoIsAllocatedData;
2971
f58c7b35
TS
2972/*
2973 * Returns true iff the specified sector is present in the disk image. Drivers
2974 * not implementing the functionality are assumed to not support backing files,
2975 * hence all their sectors are reported as allocated.
2976 *
bd9533e3
SH
2977 * If 'sector_num' is beyond the end of the disk image the return value is 0
2978 * and 'pnum' is set to 0.
2979 *
f58c7b35
TS
2980 * 'pnum' is set to the number of sectors (including and immediately following
2981 * the specified sector) that are known to be in the same
2982 * allocated/unallocated state.
2983 *
bd9533e3
SH
2984 * 'nb_sectors' is the max value 'pnum' should be set to. If nb_sectors goes
2985 * beyond the end of the disk image it will be clamped.
f58c7b35 2986 */
060f51c9
SH
2987int coroutine_fn bdrv_co_is_allocated(BlockDriverState *bs, int64_t sector_num,
2988 int nb_sectors, int *pnum)
f58c7b35 2989{
bd9533e3
SH
2990 int64_t n;
2991
2992 if (sector_num >= bs->total_sectors) {
2993 *pnum = 0;
2994 return 0;
2995 }
2996
2997 n = bs->total_sectors - sector_num;
2998 if (n < nb_sectors) {
2999 nb_sectors = n;
3000 }
3001
6aebab14 3002 if (!bs->drv->bdrv_co_is_allocated) {
bd9533e3 3003 *pnum = nb_sectors;
f58c7b35
TS
3004 return 1;
3005 }
6aebab14 3006
060f51c9
SH
3007 return bs->drv->bdrv_co_is_allocated(bs, sector_num, nb_sectors, pnum);
3008}
3009
3010/* Coroutine wrapper for bdrv_is_allocated() */
3011static void coroutine_fn bdrv_is_allocated_co_entry(void *opaque)
3012{
3013 BdrvCoIsAllocatedData *data = opaque;
3014 BlockDriverState *bs = data->bs;
3015
3016 data->ret = bdrv_co_is_allocated(bs, data->sector_num, data->nb_sectors,
3017 data->pnum);
3018 data->done = true;
3019}
3020
3021/*
3022 * Synchronous wrapper around bdrv_co_is_allocated().
3023 *
3024 * See bdrv_co_is_allocated() for details.
3025 */
3026int bdrv_is_allocated(BlockDriverState *bs, int64_t sector_num, int nb_sectors,
3027 int *pnum)
3028{
6aebab14
SH
3029 Coroutine *co;
3030 BdrvCoIsAllocatedData data = {
3031 .bs = bs,
3032 .sector_num = sector_num,
3033 .nb_sectors = nb_sectors,
3034 .pnum = pnum,
3035 .done = false,
3036 };
3037
3038 co = qemu_coroutine_create(bdrv_is_allocated_co_entry);
3039 qemu_coroutine_enter(co, &data);
3040 while (!data.done) {
3041 qemu_aio_wait();
3042 }
3043 return data.ret;
f58c7b35
TS
3044}
3045
188a7bbf
PB
3046/*
3047 * Given an image chain: ... -> [BASE] -> [INTER1] -> [INTER2] -> [TOP]
3048 *
3049 * Return true if the given sector is allocated in any image between
3050 * BASE and TOP (inclusive). BASE can be NULL to check if the given
3051 * sector is allocated in any image of the chain. Return false otherwise.
3052 *
3053 * 'pnum' is set to the number of sectors (including and immediately following
3054 * the specified sector) that are known to be in the same
3055 * allocated/unallocated state.
3056 *
3057 */
3058int coroutine_fn bdrv_co_is_allocated_above(BlockDriverState *top,
3059 BlockDriverState *base,
3060 int64_t sector_num,
3061 int nb_sectors, int *pnum)
3062{
3063 BlockDriverState *intermediate;
3064 int ret, n = nb_sectors;
3065
3066 intermediate = top;
3067 while (intermediate && intermediate != base) {
3068 int pnum_inter;
3069 ret = bdrv_co_is_allocated(intermediate, sector_num, nb_sectors,
3070 &pnum_inter);
3071 if (ret < 0) {
3072 return ret;
3073 } else if (ret) {
3074 *pnum = pnum_inter;
3075 return 1;
3076 }
3077
3078 /*
3079 * [sector_num, nb_sectors] is unallocated on top but intermediate
3080 * might have
3081 *
3082 * [sector_num+x, nr_sectors] allocated.
3083 */
63ba17d3
VI
3084 if (n > pnum_inter &&
3085 (intermediate == top ||
3086 sector_num + pnum_inter < intermediate->total_sectors)) {
188a7bbf
PB
3087 n = pnum_inter;
3088 }
3089
3090 intermediate = intermediate->backing_hd;
3091 }
3092
3093 *pnum = n;
3094 return 0;
3095}
3096
b35b2bba
MR
3097/* Coroutine wrapper for bdrv_is_allocated_above() */
3098static void coroutine_fn bdrv_is_allocated_above_co_entry(void *opaque)
3099{
3100 BdrvCoIsAllocatedData *data = opaque;
3101 BlockDriverState *top = data->bs;
3102 BlockDriverState *base = data->base;
3103
3104 data->ret = bdrv_co_is_allocated_above(top, base, data->sector_num,
3105 data->nb_sectors, data->pnum);
3106 data->done = true;
3107}
3108
3109/*
3110 * Synchronous wrapper around bdrv_co_is_allocated_above().
3111 *
3112 * See bdrv_co_is_allocated_above() for details.
3113 */
3114int bdrv_is_allocated_above(BlockDriverState *top, BlockDriverState *base,
3115 int64_t sector_num, int nb_sectors, int *pnum)
3116{
3117 Coroutine *co;
3118 BdrvCoIsAllocatedData data = {
3119 .bs = top,
3120 .base = base,
3121 .sector_num = sector_num,
3122 .nb_sectors = nb_sectors,
3123 .pnum = pnum,
3124 .done = false,
3125 };
3126
3127 co = qemu_coroutine_create(bdrv_is_allocated_above_co_entry);
3128 qemu_coroutine_enter(co, &data);
3129 while (!data.done) {
3130 qemu_aio_wait();
3131 }
3132 return data.ret;
3133}
3134
045df330
AL
3135const char *bdrv_get_encrypted_filename(BlockDriverState *bs)
3136{
3137 if (bs->backing_hd && bs->backing_hd->encrypted)
3138 return bs->backing_file;
3139 else if (bs->encrypted)
3140 return bs->filename;
3141 else
3142 return NULL;
3143}
3144
5fafdf24 3145void bdrv_get_backing_filename(BlockDriverState *bs,
83f64091
FB
3146 char *filename, int filename_size)
3147{
3574c608 3148 pstrcpy(filename, filename_size, bs->backing_file);
83f64091
FB
3149}
3150
5fafdf24 3151int bdrv_write_compressed(BlockDriverState *bs, int64_t sector_num,
faea38e7
FB
3152 const uint8_t *buf, int nb_sectors)
3153{
3154 BlockDriver *drv = bs->drv;
3155 if (!drv)
19cb3738 3156 return -ENOMEDIUM;
faea38e7
FB
3157 if (!drv->bdrv_write_compressed)
3158 return -ENOTSUP;
fbb7b4e0
KW
3159 if (bdrv_check_request(bs, sector_num, nb_sectors))
3160 return -EIO;
a55eb92c 3161
1755da16 3162 assert(!bs->dirty_bitmap);
a55eb92c 3163
faea38e7
FB
3164 return drv->bdrv_write_compressed(bs, sector_num, buf, nb_sectors);
3165}
3b46e624 3166
faea38e7
FB
3167int bdrv_get_info(BlockDriverState *bs, BlockDriverInfo *bdi)
3168{
3169 BlockDriver *drv = bs->drv;
3170 if (!drv)
19cb3738 3171 return -ENOMEDIUM;
faea38e7
FB
3172 if (!drv->bdrv_get_info)
3173 return -ENOTSUP;
3174 memset(bdi, 0, sizeof(*bdi));
3175 return drv->bdrv_get_info(bs, bdi);
3176}
3177
45566e9c
CH
3178int bdrv_save_vmstate(BlockDriverState *bs, const uint8_t *buf,
3179 int64_t pos, int size)
cf8074b3
KW
3180{
3181 QEMUIOVector qiov;
3182 struct iovec iov = {
3183 .iov_base = (void *) buf,
3184 .iov_len = size,
3185 };
3186
3187 qemu_iovec_init_external(&qiov, &iov, 1);
3188 return bdrv_writev_vmstate(bs, &qiov, pos);
3189}
3190
3191int bdrv_writev_vmstate(BlockDriverState *bs, QEMUIOVector *qiov, int64_t pos)
178e08a5
AL
3192{
3193 BlockDriver *drv = bs->drv;
cf8074b3
KW
3194
3195 if (!drv) {
178e08a5 3196 return -ENOMEDIUM;
cf8074b3
KW
3197 } else if (drv->bdrv_save_vmstate) {
3198 return drv->bdrv_save_vmstate(bs, qiov, pos);
3199 } else if (bs->file) {
3200 return bdrv_writev_vmstate(bs->file, qiov, pos);
3201 }
3202
7cdb1f6d 3203 return -ENOTSUP;
178e08a5
AL
3204}
3205
45566e9c
CH
3206int bdrv_load_vmstate(BlockDriverState *bs, uint8_t *buf,
3207 int64_t pos, int size)
178e08a5
AL
3208{
3209 BlockDriver *drv = bs->drv;
3210 if (!drv)
3211 return -ENOMEDIUM;
7cdb1f6d
MK
3212 if (drv->bdrv_load_vmstate)
3213 return drv->bdrv_load_vmstate(bs, buf, pos, size);
3214 if (bs->file)
3215 return bdrv_load_vmstate(bs->file, buf, pos, size);
3216 return -ENOTSUP;
178e08a5
AL
3217}
3218
8b9b0cc2
KW
3219void bdrv_debug_event(BlockDriverState *bs, BlkDebugEvent event)
3220{
bf736fe3 3221 if (!bs || !bs->drv || !bs->drv->bdrv_debug_event) {
8b9b0cc2
KW
3222 return;
3223 }
3224
bf736fe3 3225 bs->drv->bdrv_debug_event(bs, event);
41c695c7
KW
3226}
3227
3228int bdrv_debug_breakpoint(BlockDriverState *bs, const char *event,
3229 const char *tag)
3230{
3231 while (bs && bs->drv && !bs->drv->bdrv_debug_breakpoint) {
3232 bs = bs->file;
3233 }
3234
3235 if (bs && bs->drv && bs->drv->bdrv_debug_breakpoint) {
3236 return bs->drv->bdrv_debug_breakpoint(bs, event, tag);
3237 }
3238
3239 return -ENOTSUP;
3240}
3241
3242int bdrv_debug_resume(BlockDriverState *bs, const char *tag)
3243{
3244 while (bs && bs->drv && !bs->drv->bdrv_debug_resume) {
3245 bs = bs->file;
3246 }
8b9b0cc2 3247
41c695c7
KW
3248 if (bs && bs->drv && bs->drv->bdrv_debug_resume) {
3249 return bs->drv->bdrv_debug_resume(bs, tag);
3250 }
3251
3252 return -ENOTSUP;
3253}
3254
3255bool bdrv_debug_is_suspended(BlockDriverState *bs, const char *tag)
3256{
3257 while (bs && bs->drv && !bs->drv->bdrv_debug_is_suspended) {
3258 bs = bs->file;
3259 }
3260
3261 if (bs && bs->drv && bs->drv->bdrv_debug_is_suspended) {
3262 return bs->drv->bdrv_debug_is_suspended(bs, tag);
3263 }
3264
3265 return false;
8b9b0cc2
KW
3266}
3267
199630b6
BS
3268int bdrv_is_snapshot(BlockDriverState *bs)
3269{
3270 return !!(bs->open_flags & BDRV_O_SNAPSHOT);
3271}
3272
b1b1d783
JC
3273/* backing_file can either be relative, or absolute, or a protocol. If it is
3274 * relative, it must be relative to the chain. So, passing in bs->filename
3275 * from a BDS as backing_file should not be done, as that may be relative to
3276 * the CWD rather than the chain. */
e8a6bb9c
MT
3277BlockDriverState *bdrv_find_backing_image(BlockDriverState *bs,
3278 const char *backing_file)
3279{
b1b1d783
JC
3280 char *filename_full = NULL;
3281 char *backing_file_full = NULL;
3282 char *filename_tmp = NULL;
3283 int is_protocol = 0;
3284 BlockDriverState *curr_bs = NULL;
3285 BlockDriverState *retval = NULL;
3286
3287 if (!bs || !bs->drv || !backing_file) {
e8a6bb9c
MT
3288 return NULL;
3289 }
3290
b1b1d783
JC
3291 filename_full = g_malloc(PATH_MAX);
3292 backing_file_full = g_malloc(PATH_MAX);
3293 filename_tmp = g_malloc(PATH_MAX);
3294
3295 is_protocol = path_has_protocol(backing_file);
3296
3297 for (curr_bs = bs; curr_bs->backing_hd; curr_bs = curr_bs->backing_hd) {
3298
3299 /* If either of the filename paths is actually a protocol, then
3300 * compare unmodified paths; otherwise make paths relative */
3301 if (is_protocol || path_has_protocol(curr_bs->backing_file)) {
3302 if (strcmp(backing_file, curr_bs->backing_file) == 0) {
3303 retval = curr_bs->backing_hd;
3304 break;
3305 }
e8a6bb9c 3306 } else {
b1b1d783
JC
3307 /* If not an absolute filename path, make it relative to the current
3308 * image's filename path */
3309 path_combine(filename_tmp, PATH_MAX, curr_bs->filename,
3310 backing_file);
3311
3312 /* We are going to compare absolute pathnames */
3313 if (!realpath(filename_tmp, filename_full)) {
3314 continue;
3315 }
3316
3317 /* We need to make sure the backing filename we are comparing against
3318 * is relative to the current image filename (or absolute) */
3319 path_combine(filename_tmp, PATH_MAX, curr_bs->filename,
3320 curr_bs->backing_file);
3321
3322 if (!realpath(filename_tmp, backing_file_full)) {
3323 continue;
3324 }
3325
3326 if (strcmp(backing_file_full, filename_full) == 0) {
3327 retval = curr_bs->backing_hd;
3328 break;
3329 }
e8a6bb9c
MT
3330 }
3331 }
3332
b1b1d783
JC
3333 g_free(filename_full);
3334 g_free(backing_file_full);
3335 g_free(filename_tmp);
3336 return retval;
e8a6bb9c
MT
3337}
3338
f198fd1c
BC
3339int bdrv_get_backing_file_depth(BlockDriverState *bs)
3340{
3341 if (!bs->drv) {
3342 return 0;
3343 }
3344
3345 if (!bs->backing_hd) {
3346 return 0;
3347 }
3348
3349 return 1 + bdrv_get_backing_file_depth(bs->backing_hd);
3350}
3351
79fac568
JC
3352BlockDriverState *bdrv_find_base(BlockDriverState *bs)
3353{
3354 BlockDriverState *curr_bs = NULL;
3355
3356 if (!bs) {
3357 return NULL;
3358 }
3359
3360 curr_bs = bs;
3361
3362 while (curr_bs->backing_hd) {
3363 curr_bs = curr_bs->backing_hd;
3364 }
3365 return curr_bs;
3366}
3367
ea2384d3 3368/**************************************************************/
83f64091 3369/* async I/Os */
ea2384d3 3370
3b69e4b9 3371BlockDriverAIOCB *bdrv_aio_readv(BlockDriverState *bs, int64_t sector_num,
f141eafe 3372 QEMUIOVector *qiov, int nb_sectors,
3b69e4b9 3373 BlockDriverCompletionFunc *cb, void *opaque)
83f64091 3374{
bbf0a440
SH
3375 trace_bdrv_aio_readv(bs, sector_num, nb_sectors, opaque);
3376
b2a61371 3377 return bdrv_co_aio_rw_vector(bs, sector_num, qiov, nb_sectors,
8c5873d6 3378 cb, opaque, false);
ea2384d3
FB
3379}
3380
f141eafe
AL
3381BlockDriverAIOCB *bdrv_aio_writev(BlockDriverState *bs, int64_t sector_num,
3382 QEMUIOVector *qiov, int nb_sectors,
3383 BlockDriverCompletionFunc *cb, void *opaque)
ea2384d3 3384{
bbf0a440
SH
3385 trace_bdrv_aio_writev(bs, sector_num, nb_sectors, opaque);
3386
1a6e115b 3387 return bdrv_co_aio_rw_vector(bs, sector_num, qiov, nb_sectors,
8c5873d6 3388 cb, opaque, true);
83f64091
FB
3389}
3390
40b4f539
KW
3391
3392typedef struct MultiwriteCB {
3393 int error;
3394 int num_requests;
3395 int num_callbacks;
3396 struct {
3397 BlockDriverCompletionFunc *cb;
3398 void *opaque;
3399 QEMUIOVector *free_qiov;
40b4f539
KW
3400 } callbacks[];
3401} MultiwriteCB;
3402
3403static void multiwrite_user_cb(MultiwriteCB *mcb)
3404{
3405 int i;
3406
3407 for (i = 0; i < mcb->num_callbacks; i++) {
3408 mcb->callbacks[i].cb(mcb->callbacks[i].opaque, mcb->error);
1e1ea48d
SH
3409 if (mcb->callbacks[i].free_qiov) {
3410 qemu_iovec_destroy(mcb->callbacks[i].free_qiov);
3411 }
7267c094 3412 g_free(mcb->callbacks[i].free_qiov);
40b4f539
KW
3413 }
3414}
3415
3416static void multiwrite_cb(void *opaque, int ret)
3417{
3418 MultiwriteCB *mcb = opaque;
3419
6d519a5f
SH
3420 trace_multiwrite_cb(mcb, ret);
3421
cb6d3ca0 3422 if (ret < 0 && !mcb->error) {
40b4f539 3423 mcb->error = ret;
40b4f539
KW
3424 }
3425
3426 mcb->num_requests--;
3427 if (mcb->num_requests == 0) {
de189a1b 3428 multiwrite_user_cb(mcb);
7267c094 3429 g_free(mcb);
40b4f539
KW
3430 }
3431}
3432
3433static int multiwrite_req_compare(const void *a, const void *b)
3434{
77be4366
CH
3435 const BlockRequest *req1 = a, *req2 = b;
3436
3437 /*
3438 * Note that we can't simply subtract req2->sector from req1->sector
3439 * here as that could overflow the return value.
3440 */
3441 if (req1->sector > req2->sector) {
3442 return 1;
3443 } else if (req1->sector < req2->sector) {
3444 return -1;
3445 } else {
3446 return 0;
3447 }
40b4f539
KW
3448}
3449
3450/*
3451 * Takes a bunch of requests and tries to merge them. Returns the number of
3452 * requests that remain after merging.
3453 */
3454static int multiwrite_merge(BlockDriverState *bs, BlockRequest *reqs,
3455 int num_reqs, MultiwriteCB *mcb)
3456{
3457 int i, outidx;
3458
3459 // Sort requests by start sector
3460 qsort(reqs, num_reqs, sizeof(*reqs), &multiwrite_req_compare);
3461
3462 // Check if adjacent requests touch the same clusters. If so, combine them,
3463 // filling up gaps with zero sectors.
3464 outidx = 0;
3465 for (i = 1; i < num_reqs; i++) {
3466 int merge = 0;
3467 int64_t oldreq_last = reqs[outidx].sector + reqs[outidx].nb_sectors;
3468
b6a127a1 3469 // Handle exactly sequential writes and overlapping writes.
40b4f539
KW
3470 if (reqs[i].sector <= oldreq_last) {
3471 merge = 1;
3472 }
3473
e2a305fb
CH
3474 if (reqs[outidx].qiov->niov + reqs[i].qiov->niov + 1 > IOV_MAX) {
3475 merge = 0;
3476 }
3477
40b4f539
KW
3478 if (merge) {
3479 size_t size;
7267c094 3480 QEMUIOVector *qiov = g_malloc0(sizeof(*qiov));
40b4f539
KW
3481 qemu_iovec_init(qiov,
3482 reqs[outidx].qiov->niov + reqs[i].qiov->niov + 1);
3483
3484 // Add the first request to the merged one. If the requests are
3485 // overlapping, drop the last sectors of the first request.
3486 size = (reqs[i].sector - reqs[outidx].sector) << 9;
1b093c48 3487 qemu_iovec_concat(qiov, reqs[outidx].qiov, 0, size);
40b4f539 3488
b6a127a1
PB
3489 // We should need to add any zeros between the two requests
3490 assert (reqs[i].sector <= oldreq_last);
40b4f539
KW
3491
3492 // Add the second request
1b093c48 3493 qemu_iovec_concat(qiov, reqs[i].qiov, 0, reqs[i].qiov->size);
40b4f539 3494
cbf1dff2 3495 reqs[outidx].nb_sectors = qiov->size >> 9;
40b4f539
KW
3496 reqs[outidx].qiov = qiov;
3497
3498 mcb->callbacks[i].free_qiov = reqs[outidx].qiov;
3499 } else {
3500 outidx++;
3501 reqs[outidx].sector = reqs[i].sector;
3502 reqs[outidx].nb_sectors = reqs[i].nb_sectors;
3503 reqs[outidx].qiov = reqs[i].qiov;
3504 }
3505 }
3506
3507 return outidx + 1;
3508}
3509
3510/*
3511 * Submit multiple AIO write requests at once.
3512 *
3513 * On success, the function returns 0 and all requests in the reqs array have
3514 * been submitted. In error case this function returns -1, and any of the
3515 * requests may or may not be submitted yet. In particular, this means that the
3516 * callback will be called for some of the requests, for others it won't. The
3517 * caller must check the error field of the BlockRequest to wait for the right
3518 * callbacks (if error != 0, no callback will be called).
3519 *
3520 * The implementation may modify the contents of the reqs array, e.g. to merge
3521 * requests. However, the fields opaque and error are left unmodified as they
3522 * are used to signal failure for a single request to the caller.
3523 */
3524int bdrv_aio_multiwrite(BlockDriverState *bs, BlockRequest *reqs, int num_reqs)
3525{
40b4f539
KW
3526 MultiwriteCB *mcb;
3527 int i;
3528
301db7c2
RH
3529 /* don't submit writes if we don't have a medium */
3530 if (bs->drv == NULL) {
3531 for (i = 0; i < num_reqs; i++) {
3532 reqs[i].error = -ENOMEDIUM;
3533 }
3534 return -1;
3535 }
3536
40b4f539
KW
3537 if (num_reqs == 0) {
3538 return 0;
3539 }
3540
3541 // Create MultiwriteCB structure
7267c094 3542 mcb = g_malloc0(sizeof(*mcb) + num_reqs * sizeof(*mcb->callbacks));
40b4f539
KW
3543 mcb->num_requests = 0;
3544 mcb->num_callbacks = num_reqs;
3545
3546 for (i = 0; i < num_reqs; i++) {
3547 mcb->callbacks[i].cb = reqs[i].cb;
3548 mcb->callbacks[i].opaque = reqs[i].opaque;
3549 }
3550
3551 // Check for mergable requests
3552 num_reqs = multiwrite_merge(bs, reqs, num_reqs, mcb);
3553
6d519a5f
SH
3554 trace_bdrv_aio_multiwrite(mcb, mcb->num_callbacks, num_reqs);
3555
df9309fb
PB
3556 /* Run the aio requests. */
3557 mcb->num_requests = num_reqs;
40b4f539 3558 for (i = 0; i < num_reqs; i++) {
ad54ae80 3559 bdrv_aio_writev(bs, reqs[i].sector, reqs[i].qiov,
40b4f539 3560 reqs[i].nb_sectors, multiwrite_cb, mcb);
40b4f539
KW
3561 }
3562
3563 return 0;
40b4f539
KW
3564}
3565
83f64091 3566void bdrv_aio_cancel(BlockDriverAIOCB *acb)
83f64091 3567{
d7331bed 3568 acb->aiocb_info->cancel(acb);
83f64091
FB
3569}
3570
98f90dba
ZYW
3571/* block I/O throttling */
3572static bool bdrv_exceed_bps_limits(BlockDriverState *bs, int nb_sectors,
3573 bool is_write, double elapsed_time, uint64_t *wait)
3574{
3575 uint64_t bps_limit = 0;
ae29d6c6 3576 uint64_t extension;
98f90dba
ZYW
3577 double bytes_limit, bytes_base, bytes_res;
3578 double slice_time, wait_time;
3579
3580 if (bs->io_limits.bps[BLOCK_IO_LIMIT_TOTAL]) {
3581 bps_limit = bs->io_limits.bps[BLOCK_IO_LIMIT_TOTAL];
3582 } else if (bs->io_limits.bps[is_write]) {
3583 bps_limit = bs->io_limits.bps[is_write];
3584 } else {
3585 if (wait) {
3586 *wait = 0;
3587 }
3588
3589 return false;
3590 }
3591
3592 slice_time = bs->slice_end - bs->slice_start;
3593 slice_time /= (NANOSECONDS_PER_SECOND);
3594 bytes_limit = bps_limit * slice_time;
5905fbc9 3595 bytes_base = bs->slice_submitted.bytes[is_write];
98f90dba 3596 if (bs->io_limits.bps[BLOCK_IO_LIMIT_TOTAL]) {
5905fbc9 3597 bytes_base += bs->slice_submitted.bytes[!is_write];
98f90dba
ZYW
3598 }
3599
3600 /* bytes_base: the bytes of data which have been read/written; and
3601 * it is obtained from the history statistic info.
3602 * bytes_res: the remaining bytes of data which need to be read/written.
3603 * (bytes_base + bytes_res) / bps_limit: used to calcuate
3604 * the total time for completing reading/writting all data.
3605 */
3606 bytes_res = (unsigned) nb_sectors * BDRV_SECTOR_SIZE;
3607
3608 if (bytes_base + bytes_res <= bytes_limit) {
3609 if (wait) {
3610 *wait = 0;
3611 }
3612
3613 return false;
3614 }
3615
3616 /* Calc approx time to dispatch */
3617 wait_time = (bytes_base + bytes_res) / bps_limit - elapsed_time;
3618
3619 /* When the I/O rate at runtime exceeds the limits,
3620 * bs->slice_end need to be extended in order that the current statistic
3621 * info can be kept until the timer fire, so it is increased and tuned
3622 * based on the result of experiment.
3623 */
ae29d6c6
SH
3624 extension = wait_time * NANOSECONDS_PER_SECOND;
3625 extension = DIV_ROUND_UP(extension, BLOCK_IO_SLICE_TIME) *
3626 BLOCK_IO_SLICE_TIME;
3627 bs->slice_end += extension;
98f90dba 3628 if (wait) {
0775437f 3629 *wait = wait_time * NANOSECONDS_PER_SECOND;
98f90dba
ZYW
3630 }
3631
3632 return true;
3633}
3634
3635static bool bdrv_exceed_iops_limits(BlockDriverState *bs, bool is_write,
3636 double elapsed_time, uint64_t *wait)
3637{
3638 uint64_t iops_limit = 0;
3639 double ios_limit, ios_base;
3640 double slice_time, wait_time;
3641
3642 if (bs->io_limits.iops[BLOCK_IO_LIMIT_TOTAL]) {
3643 iops_limit = bs->io_limits.iops[BLOCK_IO_LIMIT_TOTAL];
3644 } else if (bs->io_limits.iops[is_write]) {
3645 iops_limit = bs->io_limits.iops[is_write];
3646 } else {
3647 if (wait) {
3648 *wait = 0;
3649 }
3650
3651 return false;
3652 }
3653
3654 slice_time = bs->slice_end - bs->slice_start;
3655 slice_time /= (NANOSECONDS_PER_SECOND);
3656 ios_limit = iops_limit * slice_time;
5905fbc9 3657 ios_base = bs->slice_submitted.ios[is_write];
98f90dba 3658 if (bs->io_limits.iops[BLOCK_IO_LIMIT_TOTAL]) {
5905fbc9 3659 ios_base += bs->slice_submitted.ios[!is_write];
98f90dba
ZYW
3660 }
3661
3662 if (ios_base + 1 <= ios_limit) {
3663 if (wait) {
3664 *wait = 0;
3665 }
3666
3667 return false;
3668 }
3669
0775437f 3670 /* Calc approx time to dispatch, in seconds */
98f90dba
ZYW
3671 wait_time = (ios_base + 1) / iops_limit;
3672 if (wait_time > elapsed_time) {
3673 wait_time = wait_time - elapsed_time;
3674 } else {
3675 wait_time = 0;
3676 }
3677
ae29d6c6
SH
3678 /* Exceeded current slice, extend it by another slice time */
3679 bs->slice_end += BLOCK_IO_SLICE_TIME;
98f90dba 3680 if (wait) {
0775437f 3681 *wait = wait_time * NANOSECONDS_PER_SECOND;
98f90dba
ZYW
3682 }
3683
3684 return true;
3685}
3686
3687static bool bdrv_exceed_io_limits(BlockDriverState *bs, int nb_sectors,
3688 bool is_write, int64_t *wait)
3689{
3690 int64_t now, max_wait;
3691 uint64_t bps_wait = 0, iops_wait = 0;
3692 double elapsed_time;
3693 int bps_ret, iops_ret;
3694
3695 now = qemu_get_clock_ns(vm_clock);
e660fb8b 3696 if (now > bs->slice_end) {
98f90dba 3697 bs->slice_start = now;
ae29d6c6 3698 bs->slice_end = now + BLOCK_IO_SLICE_TIME;
5905fbc9 3699 memset(&bs->slice_submitted, 0, sizeof(bs->slice_submitted));
98f90dba
ZYW
3700 }
3701
3702 elapsed_time = now - bs->slice_start;
3703 elapsed_time /= (NANOSECONDS_PER_SECOND);
3704
3705 bps_ret = bdrv_exceed_bps_limits(bs, nb_sectors,
3706 is_write, elapsed_time, &bps_wait);
3707 iops_ret = bdrv_exceed_iops_limits(bs, is_write,
3708 elapsed_time, &iops_wait);
3709 if (bps_ret || iops_ret) {
3710 max_wait = bps_wait > iops_wait ? bps_wait : iops_wait;
3711 if (wait) {
3712 *wait = max_wait;
3713 }
3714
3715 now = qemu_get_clock_ns(vm_clock);
3716 if (bs->slice_end < now + max_wait) {
3717 bs->slice_end = now + max_wait;
3718 }
3719
3720 return true;
3721 }
3722
3723 if (wait) {
3724 *wait = 0;
3725 }
3726
5905fbc9
SH
3727 bs->slice_submitted.bytes[is_write] += (int64_t)nb_sectors *
3728 BDRV_SECTOR_SIZE;
3729 bs->slice_submitted.ios[is_write]++;
3730
98f90dba
ZYW
3731 return false;
3732}
ce1a14dc 3733
83f64091
FB
3734/**************************************************************/
3735/* async block device emulation */
3736
c16b5a2c
CH
3737typedef struct BlockDriverAIOCBSync {
3738 BlockDriverAIOCB common;
3739 QEMUBH *bh;
3740 int ret;
3741 /* vector translation state */
3742 QEMUIOVector *qiov;
3743 uint8_t *bounce;
3744 int is_write;
3745} BlockDriverAIOCBSync;
3746
3747static void bdrv_aio_cancel_em(BlockDriverAIOCB *blockacb)
3748{
b666d239
KW
3749 BlockDriverAIOCBSync *acb =
3750 container_of(blockacb, BlockDriverAIOCBSync, common);
6a7ad299 3751 qemu_bh_delete(acb->bh);
36afc451 3752 acb->bh = NULL;
c16b5a2c
CH
3753 qemu_aio_release(acb);
3754}
3755
d7331bed 3756static const AIOCBInfo bdrv_em_aiocb_info = {
c16b5a2c
CH
3757 .aiocb_size = sizeof(BlockDriverAIOCBSync),
3758 .cancel = bdrv_aio_cancel_em,
3759};
3760
ce1a14dc 3761static void bdrv_aio_bh_cb(void *opaque)
83f64091 3762{
ce1a14dc 3763 BlockDriverAIOCBSync *acb = opaque;
f141eafe 3764
f141eafe 3765 if (!acb->is_write)
03396148 3766 qemu_iovec_from_buf(acb->qiov, 0, acb->bounce, acb->qiov->size);
ceb42de8 3767 qemu_vfree(acb->bounce);
ce1a14dc 3768 acb->common.cb(acb->common.opaque, acb->ret);
6a7ad299 3769 qemu_bh_delete(acb->bh);
36afc451 3770 acb->bh = NULL;
ce1a14dc 3771 qemu_aio_release(acb);
83f64091 3772}
beac80cd 3773
f141eafe
AL
3774static BlockDriverAIOCB *bdrv_aio_rw_vector(BlockDriverState *bs,
3775 int64_t sector_num,
3776 QEMUIOVector *qiov,
3777 int nb_sectors,
3778 BlockDriverCompletionFunc *cb,
3779 void *opaque,
3780 int is_write)
3781
83f64091 3782{
ce1a14dc 3783 BlockDriverAIOCBSync *acb;
ce1a14dc 3784
d7331bed 3785 acb = qemu_aio_get(&bdrv_em_aiocb_info, bs, cb, opaque);
f141eafe
AL
3786 acb->is_write = is_write;
3787 acb->qiov = qiov;
e268ca52 3788 acb->bounce = qemu_blockalign(bs, qiov->size);
3f3aace8 3789 acb->bh = qemu_bh_new(bdrv_aio_bh_cb, acb);
f141eafe
AL
3790
3791 if (is_write) {
d5e6b161 3792 qemu_iovec_to_buf(acb->qiov, 0, acb->bounce, qiov->size);
1ed20acf 3793 acb->ret = bs->drv->bdrv_write(bs, sector_num, acb->bounce, nb_sectors);
f141eafe 3794 } else {
1ed20acf 3795 acb->ret = bs->drv->bdrv_read(bs, sector_num, acb->bounce, nb_sectors);
f141eafe
AL
3796 }
3797
ce1a14dc 3798 qemu_bh_schedule(acb->bh);
f141eafe 3799
ce1a14dc 3800 return &acb->common;
beac80cd
FB
3801}
3802
f141eafe
AL
3803static BlockDriverAIOCB *bdrv_aio_readv_em(BlockDriverState *bs,
3804 int64_t sector_num, QEMUIOVector *qiov, int nb_sectors,
ce1a14dc 3805 BlockDriverCompletionFunc *cb, void *opaque)
beac80cd 3806{
f141eafe
AL
3807 return bdrv_aio_rw_vector(bs, sector_num, qiov, nb_sectors, cb, opaque, 0);
3808}
83f64091 3809
f141eafe
AL
3810static BlockDriverAIOCB *bdrv_aio_writev_em(BlockDriverState *bs,
3811 int64_t sector_num, QEMUIOVector *qiov, int nb_sectors,
3812 BlockDriverCompletionFunc *cb, void *opaque)
3813{
3814 return bdrv_aio_rw_vector(bs, sector_num, qiov, nb_sectors, cb, opaque, 1);
beac80cd 3815}
beac80cd 3816
68485420
KW
3817
3818typedef struct BlockDriverAIOCBCoroutine {
3819 BlockDriverAIOCB common;
3820 BlockRequest req;
3821 bool is_write;
d318aea9 3822 bool *done;
68485420
KW
3823 QEMUBH* bh;
3824} BlockDriverAIOCBCoroutine;
3825
3826static void bdrv_aio_co_cancel_em(BlockDriverAIOCB *blockacb)
3827{
d318aea9
KW
3828 BlockDriverAIOCBCoroutine *acb =
3829 container_of(blockacb, BlockDriverAIOCBCoroutine, common);
3830 bool done = false;
3831
3832 acb->done = &done;
3833 while (!done) {
3834 qemu_aio_wait();
3835 }
68485420
KW
3836}
3837
d7331bed 3838static const AIOCBInfo bdrv_em_co_aiocb_info = {
68485420
KW
3839 .aiocb_size = sizeof(BlockDriverAIOCBCoroutine),
3840 .cancel = bdrv_aio_co_cancel_em,
3841};
3842
35246a68 3843static void bdrv_co_em_bh(void *opaque)
68485420
KW
3844{
3845 BlockDriverAIOCBCoroutine *acb = opaque;
3846
3847 acb->common.cb(acb->common.opaque, acb->req.error);
d318aea9
KW
3848
3849 if (acb->done) {
3850 *acb->done = true;
3851 }
3852
68485420
KW
3853 qemu_bh_delete(acb->bh);
3854 qemu_aio_release(acb);
3855}
3856
b2a61371
SH
3857/* Invoke bdrv_co_do_readv/bdrv_co_do_writev */
3858static void coroutine_fn bdrv_co_do_rw(void *opaque)
3859{
3860 BlockDriverAIOCBCoroutine *acb = opaque;
3861 BlockDriverState *bs = acb->common.bs;
3862
3863 if (!acb->is_write) {
3864 acb->req.error = bdrv_co_do_readv(bs, acb->req.sector,
470c0504 3865 acb->req.nb_sectors, acb->req.qiov, 0);
b2a61371
SH
3866 } else {
3867 acb->req.error = bdrv_co_do_writev(bs, acb->req.sector,
f08f2dda 3868 acb->req.nb_sectors, acb->req.qiov, 0);
b2a61371
SH
3869 }
3870
35246a68 3871 acb->bh = qemu_bh_new(bdrv_co_em_bh, acb);
b2a61371
SH
3872 qemu_bh_schedule(acb->bh);
3873}
3874
68485420
KW
3875static BlockDriverAIOCB *bdrv_co_aio_rw_vector(BlockDriverState *bs,
3876 int64_t sector_num,
3877 QEMUIOVector *qiov,
3878 int nb_sectors,
3879 BlockDriverCompletionFunc *cb,
3880 void *opaque,
8c5873d6 3881 bool is_write)
68485420
KW
3882{
3883 Coroutine *co;
3884 BlockDriverAIOCBCoroutine *acb;
3885
d7331bed 3886 acb = qemu_aio_get(&bdrv_em_co_aiocb_info, bs, cb, opaque);
68485420
KW
3887 acb->req.sector = sector_num;
3888 acb->req.nb_sectors = nb_sectors;
3889 acb->req.qiov = qiov;
3890 acb->is_write = is_write;
d318aea9 3891 acb->done = NULL;
68485420 3892
8c5873d6 3893 co = qemu_coroutine_create(bdrv_co_do_rw);
68485420
KW
3894 qemu_coroutine_enter(co, acb);
3895
3896 return &acb->common;
3897}
3898
07f07615 3899static void coroutine_fn bdrv_aio_flush_co_entry(void *opaque)
b2e12bc6 3900{
07f07615
PB
3901 BlockDriverAIOCBCoroutine *acb = opaque;
3902 BlockDriverState *bs = acb->common.bs;
b2e12bc6 3903
07f07615
PB
3904 acb->req.error = bdrv_co_flush(bs);
3905 acb->bh = qemu_bh_new(bdrv_co_em_bh, acb);
b2e12bc6 3906 qemu_bh_schedule(acb->bh);
b2e12bc6
CH
3907}
3908
07f07615 3909BlockDriverAIOCB *bdrv_aio_flush(BlockDriverState *bs,
016f5cf6
AG
3910 BlockDriverCompletionFunc *cb, void *opaque)
3911{
07f07615 3912 trace_bdrv_aio_flush(bs, opaque);
016f5cf6 3913
07f07615
PB
3914 Coroutine *co;
3915 BlockDriverAIOCBCoroutine *acb;
016f5cf6 3916
d7331bed 3917 acb = qemu_aio_get(&bdrv_em_co_aiocb_info, bs, cb, opaque);
d318aea9
KW
3918 acb->done = NULL;
3919
07f07615
PB
3920 co = qemu_coroutine_create(bdrv_aio_flush_co_entry);
3921 qemu_coroutine_enter(co, acb);
016f5cf6 3922
016f5cf6
AG
3923 return &acb->common;
3924}
3925
4265d620
PB
3926static void coroutine_fn bdrv_aio_discard_co_entry(void *opaque)
3927{
3928 BlockDriverAIOCBCoroutine *acb = opaque;
3929 BlockDriverState *bs = acb->common.bs;
3930
3931 acb->req.error = bdrv_co_discard(bs, acb->req.sector, acb->req.nb_sectors);
3932 acb->bh = qemu_bh_new(bdrv_co_em_bh, acb);
3933 qemu_bh_schedule(acb->bh);
3934}
3935
3936BlockDriverAIOCB *bdrv_aio_discard(BlockDriverState *bs,
3937 int64_t sector_num, int nb_sectors,
3938 BlockDriverCompletionFunc *cb, void *opaque)
3939{
3940 Coroutine *co;
3941 BlockDriverAIOCBCoroutine *acb;
3942
3943 trace_bdrv_aio_discard(bs, sector_num, nb_sectors, opaque);
3944
d7331bed 3945 acb = qemu_aio_get(&bdrv_em_co_aiocb_info, bs, cb, opaque);
4265d620
PB
3946 acb->req.sector = sector_num;
3947 acb->req.nb_sectors = nb_sectors;
d318aea9 3948 acb->done = NULL;
4265d620
PB
3949 co = qemu_coroutine_create(bdrv_aio_discard_co_entry);
3950 qemu_coroutine_enter(co, acb);
3951
3952 return &acb->common;
3953}
3954
ea2384d3
FB
3955void bdrv_init(void)
3956{
5efa9d5a 3957 module_call_init(MODULE_INIT_BLOCK);
ea2384d3 3958}
ce1a14dc 3959
eb852011
MA
3960void bdrv_init_with_whitelist(void)
3961{
3962 use_bdrv_whitelist = 1;
3963 bdrv_init();
3964}
3965
d7331bed 3966void *qemu_aio_get(const AIOCBInfo *aiocb_info, BlockDriverState *bs,
c16b5a2c 3967 BlockDriverCompletionFunc *cb, void *opaque)
ce1a14dc 3968{
ce1a14dc
PB
3969 BlockDriverAIOCB *acb;
3970
d7331bed
SH
3971 acb = g_slice_alloc(aiocb_info->aiocb_size);
3972 acb->aiocb_info = aiocb_info;
ce1a14dc
PB
3973 acb->bs = bs;
3974 acb->cb = cb;
3975 acb->opaque = opaque;
3976 return acb;
3977}
3978
3979void qemu_aio_release(void *p)
3980{
d37c975f 3981 BlockDriverAIOCB *acb = p;
d7331bed 3982 g_slice_free1(acb->aiocb_info->aiocb_size, acb);
ce1a14dc 3983}
19cb3738 3984
f9f05dc5
KW
3985/**************************************************************/
3986/* Coroutine block device emulation */
3987
3988typedef struct CoroutineIOCompletion {
3989 Coroutine *coroutine;
3990 int ret;
3991} CoroutineIOCompletion;
3992
3993static void bdrv_co_io_em_complete(void *opaque, int ret)
3994{
3995 CoroutineIOCompletion *co = opaque;
3996
3997 co->ret = ret;
3998 qemu_coroutine_enter(co->coroutine, NULL);
3999}
4000
4001static int coroutine_fn bdrv_co_io_em(BlockDriverState *bs, int64_t sector_num,
4002 int nb_sectors, QEMUIOVector *iov,
4003 bool is_write)
4004{
4005 CoroutineIOCompletion co = {
4006 .coroutine = qemu_coroutine_self(),
4007 };
4008 BlockDriverAIOCB *acb;
4009
4010 if (is_write) {
a652d160
SH
4011 acb = bs->drv->bdrv_aio_writev(bs, sector_num, iov, nb_sectors,
4012 bdrv_co_io_em_complete, &co);
f9f05dc5 4013 } else {
a652d160
SH
4014 acb = bs->drv->bdrv_aio_readv(bs, sector_num, iov, nb_sectors,
4015 bdrv_co_io_em_complete, &co);
f9f05dc5
KW
4016 }
4017
59370aaa 4018 trace_bdrv_co_io_em(bs, sector_num, nb_sectors, is_write, acb);
f9f05dc5
KW
4019 if (!acb) {
4020 return -EIO;
4021 }
4022 qemu_coroutine_yield();
4023
4024 return co.ret;
4025}
4026
4027static int coroutine_fn bdrv_co_readv_em(BlockDriverState *bs,
4028 int64_t sector_num, int nb_sectors,
4029 QEMUIOVector *iov)
4030{
4031 return bdrv_co_io_em(bs, sector_num, nb_sectors, iov, false);
4032}
4033
4034static int coroutine_fn bdrv_co_writev_em(BlockDriverState *bs,
4035 int64_t sector_num, int nb_sectors,
4036 QEMUIOVector *iov)
4037{
4038 return bdrv_co_io_em(bs, sector_num, nb_sectors, iov, true);
4039}
4040
07f07615 4041static void coroutine_fn bdrv_flush_co_entry(void *opaque)
e7a8a783 4042{
07f07615
PB
4043 RwCo *rwco = opaque;
4044
4045 rwco->ret = bdrv_co_flush(rwco->bs);
4046}
4047
4048int coroutine_fn bdrv_co_flush(BlockDriverState *bs)
4049{
eb489bb1
KW
4050 int ret;
4051
29cdb251 4052 if (!bs || !bdrv_is_inserted(bs) || bdrv_is_read_only(bs)) {
07f07615 4053 return 0;
eb489bb1
KW
4054 }
4055
ca716364 4056 /* Write back cached data to the OS even with cache=unsafe */
bf736fe3 4057 BLKDBG_EVENT(bs->file, BLKDBG_FLUSH_TO_OS);
eb489bb1
KW
4058 if (bs->drv->bdrv_co_flush_to_os) {
4059 ret = bs->drv->bdrv_co_flush_to_os(bs);
4060 if (ret < 0) {
4061 return ret;
4062 }
4063 }
4064
ca716364
KW
4065 /* But don't actually force it to the disk with cache=unsafe */
4066 if (bs->open_flags & BDRV_O_NO_FLUSH) {
d4c82329 4067 goto flush_parent;
ca716364
KW
4068 }
4069
bf736fe3 4070 BLKDBG_EVENT(bs->file, BLKDBG_FLUSH_TO_DISK);
eb489bb1 4071 if (bs->drv->bdrv_co_flush_to_disk) {
29cdb251 4072 ret = bs->drv->bdrv_co_flush_to_disk(bs);
07f07615
PB
4073 } else if (bs->drv->bdrv_aio_flush) {
4074 BlockDriverAIOCB *acb;
4075 CoroutineIOCompletion co = {
4076 .coroutine = qemu_coroutine_self(),
4077 };
4078
4079 acb = bs->drv->bdrv_aio_flush(bs, bdrv_co_io_em_complete, &co);
4080 if (acb == NULL) {
29cdb251 4081 ret = -EIO;
07f07615
PB
4082 } else {
4083 qemu_coroutine_yield();
29cdb251 4084 ret = co.ret;
07f07615 4085 }
07f07615
PB
4086 } else {
4087 /*
4088 * Some block drivers always operate in either writethrough or unsafe
4089 * mode and don't support bdrv_flush therefore. Usually qemu doesn't
4090 * know how the server works (because the behaviour is hardcoded or
4091 * depends on server-side configuration), so we can't ensure that
4092 * everything is safe on disk. Returning an error doesn't work because
4093 * that would break guests even if the server operates in writethrough
4094 * mode.
4095 *
4096 * Let's hope the user knows what he's doing.
4097 */
29cdb251 4098 ret = 0;
07f07615 4099 }
29cdb251
PB
4100 if (ret < 0) {
4101 return ret;
4102 }
4103
4104 /* Now flush the underlying protocol. It will also have BDRV_O_NO_FLUSH
4105 * in the case of cache=unsafe, so there are no useless flushes.
4106 */
d4c82329 4107flush_parent:
29cdb251 4108 return bdrv_co_flush(bs->file);
07f07615
PB
4109}
4110
0f15423c
AL
4111void bdrv_invalidate_cache(BlockDriverState *bs)
4112{
4113 if (bs->drv && bs->drv->bdrv_invalidate_cache) {
4114 bs->drv->bdrv_invalidate_cache(bs);
4115 }
4116}
4117
4118void bdrv_invalidate_cache_all(void)
4119{
4120 BlockDriverState *bs;
4121
4122 QTAILQ_FOREACH(bs, &bdrv_states, list) {
4123 bdrv_invalidate_cache(bs);
4124 }
4125}
4126
07789269
BC
4127void bdrv_clear_incoming_migration_all(void)
4128{
4129 BlockDriverState *bs;
4130
4131 QTAILQ_FOREACH(bs, &bdrv_states, list) {
4132 bs->open_flags = bs->open_flags & ~(BDRV_O_INCOMING);
4133 }
4134}
4135
07f07615
PB
4136int bdrv_flush(BlockDriverState *bs)
4137{
4138 Coroutine *co;
4139 RwCo rwco = {
4140 .bs = bs,
4141 .ret = NOT_DONE,
e7a8a783 4142 };
e7a8a783 4143
07f07615
PB
4144 if (qemu_in_coroutine()) {
4145 /* Fast-path if already in coroutine context */
4146 bdrv_flush_co_entry(&rwco);
4147 } else {
4148 co = qemu_coroutine_create(bdrv_flush_co_entry);
4149 qemu_coroutine_enter(co, &rwco);
4150 while (rwco.ret == NOT_DONE) {
4151 qemu_aio_wait();
4152 }
e7a8a783 4153 }
07f07615
PB
4154
4155 return rwco.ret;
e7a8a783
KW
4156}
4157
4265d620
PB
4158static void coroutine_fn bdrv_discard_co_entry(void *opaque)
4159{
4160 RwCo *rwco = opaque;
4161
4162 rwco->ret = bdrv_co_discard(rwco->bs, rwco->sector_num, rwco->nb_sectors);
4163}
4164
4165int coroutine_fn bdrv_co_discard(BlockDriverState *bs, int64_t sector_num,
4166 int nb_sectors)
4167{
4168 if (!bs->drv) {
4169 return -ENOMEDIUM;
4170 } else if (bdrv_check_request(bs, sector_num, nb_sectors)) {
4171 return -EIO;
4172 } else if (bs->read_only) {
4173 return -EROFS;
df702c9b
PB
4174 }
4175
4176 if (bs->dirty_bitmap) {
8f0720ec 4177 bdrv_reset_dirty(bs, sector_num, nb_sectors);
df702c9b
PB
4178 }
4179
9e8f1835
PB
4180 /* Do nothing if disabled. */
4181 if (!(bs->open_flags & BDRV_O_UNMAP)) {
4182 return 0;
4183 }
4184
df702c9b 4185 if (bs->drv->bdrv_co_discard) {
4265d620
PB
4186 return bs->drv->bdrv_co_discard(bs, sector_num, nb_sectors);
4187 } else if (bs->drv->bdrv_aio_discard) {
4188 BlockDriverAIOCB *acb;
4189 CoroutineIOCompletion co = {
4190 .coroutine = qemu_coroutine_self(),
4191 };
4192
4193 acb = bs->drv->bdrv_aio_discard(bs, sector_num, nb_sectors,
4194 bdrv_co_io_em_complete, &co);
4195 if (acb == NULL) {
4196 return -EIO;
4197 } else {
4198 qemu_coroutine_yield();
4199 return co.ret;
4200 }
4265d620
PB
4201 } else {
4202 return 0;
4203 }
4204}
4205
4206int bdrv_discard(BlockDriverState *bs, int64_t sector_num, int nb_sectors)
4207{
4208 Coroutine *co;
4209 RwCo rwco = {
4210 .bs = bs,
4211 .sector_num = sector_num,
4212 .nb_sectors = nb_sectors,
4213 .ret = NOT_DONE,
4214 };
4215
4216 if (qemu_in_coroutine()) {
4217 /* Fast-path if already in coroutine context */
4218 bdrv_discard_co_entry(&rwco);
4219 } else {
4220 co = qemu_coroutine_create(bdrv_discard_co_entry);
4221 qemu_coroutine_enter(co, &rwco);
4222 while (rwco.ret == NOT_DONE) {
4223 qemu_aio_wait();
4224 }
4225 }
4226
4227 return rwco.ret;
4228}
4229
19cb3738
FB
4230/**************************************************************/
4231/* removable device support */
4232
4233/**
4234 * Return TRUE if the media is present
4235 */
4236int bdrv_is_inserted(BlockDriverState *bs)
4237{
4238 BlockDriver *drv = bs->drv;
a1aff5bf 4239
19cb3738
FB
4240 if (!drv)
4241 return 0;
4242 if (!drv->bdrv_is_inserted)
a1aff5bf
MA
4243 return 1;
4244 return drv->bdrv_is_inserted(bs);
19cb3738
FB
4245}
4246
4247/**
8e49ca46
MA
4248 * Return whether the media changed since the last call to this
4249 * function, or -ENOTSUP if we don't know. Most drivers don't know.
19cb3738
FB
4250 */
4251int bdrv_media_changed(BlockDriverState *bs)
4252{
4253 BlockDriver *drv = bs->drv;
19cb3738 4254
8e49ca46
MA
4255 if (drv && drv->bdrv_media_changed) {
4256 return drv->bdrv_media_changed(bs);
4257 }
4258 return -ENOTSUP;
19cb3738
FB
4259}
4260
4261/**
4262 * If eject_flag is TRUE, eject the media. Otherwise, close the tray
4263 */
f36f3949 4264void bdrv_eject(BlockDriverState *bs, bool eject_flag)
19cb3738
FB
4265{
4266 BlockDriver *drv = bs->drv;
19cb3738 4267
822e1cd1
MA
4268 if (drv && drv->bdrv_eject) {
4269 drv->bdrv_eject(bs, eject_flag);
19cb3738 4270 }
6f382ed2
LC
4271
4272 if (bs->device_name[0] != '\0') {
4273 bdrv_emit_qmp_eject_event(bs, eject_flag);
4274 }
19cb3738
FB
4275}
4276
19cb3738
FB
4277/**
4278 * Lock or unlock the media (if it is locked, the user won't be able
4279 * to eject it manually).
4280 */
025e849a 4281void bdrv_lock_medium(BlockDriverState *bs, bool locked)
19cb3738
FB
4282{
4283 BlockDriver *drv = bs->drv;
4284
025e849a 4285 trace_bdrv_lock_medium(bs, locked);
b8c6d095 4286
025e849a
MA
4287 if (drv && drv->bdrv_lock_medium) {
4288 drv->bdrv_lock_medium(bs, locked);
19cb3738
FB
4289 }
4290}
985a03b0
TS
4291
4292/* needed for generic scsi interface */
4293
4294int bdrv_ioctl(BlockDriverState *bs, unsigned long int req, void *buf)
4295{
4296 BlockDriver *drv = bs->drv;
4297
4298 if (drv && drv->bdrv_ioctl)
4299 return drv->bdrv_ioctl(bs, req, buf);
4300 return -ENOTSUP;
4301}
7d780669 4302
221f715d
AL
4303BlockDriverAIOCB *bdrv_aio_ioctl(BlockDriverState *bs,
4304 unsigned long int req, void *buf,
4305 BlockDriverCompletionFunc *cb, void *opaque)
7d780669 4306{
221f715d 4307 BlockDriver *drv = bs->drv;
7d780669 4308
221f715d
AL
4309 if (drv && drv->bdrv_aio_ioctl)
4310 return drv->bdrv_aio_ioctl(bs, req, buf, cb, opaque);
4311 return NULL;
7d780669 4312}
e268ca52 4313
7b6f9300
MA
4314void bdrv_set_buffer_alignment(BlockDriverState *bs, int align)
4315{
4316 bs->buffer_alignment = align;
4317}
7cd1e32a 4318
e268ca52
AL
4319void *qemu_blockalign(BlockDriverState *bs, size_t size)
4320{
4321 return qemu_memalign((bs && bs->buffer_alignment) ? bs->buffer_alignment : 512, size);
4322}
7cd1e32a 4323
c53b1c51
SH
4324/*
4325 * Check if all memory in this vector is sector aligned.
4326 */
4327bool bdrv_qiov_is_aligned(BlockDriverState *bs, QEMUIOVector *qiov)
4328{
4329 int i;
4330
4331 for (i = 0; i < qiov->niov; i++) {
4332 if ((uintptr_t) qiov->iov[i].iov_base % bs->buffer_alignment) {
4333 return false;
4334 }
4335 }
4336
4337 return true;
4338}
4339
50717e94 4340void bdrv_set_dirty_tracking(BlockDriverState *bs, int granularity)
7cd1e32a 4341{
4342 int64_t bitmap_size;
a55eb92c 4343
50717e94
PB
4344 assert((granularity & (granularity - 1)) == 0);
4345
4346 if (granularity) {
4347 granularity >>= BDRV_SECTOR_BITS;
4348 assert(!bs->dirty_bitmap);
4349 bitmap_size = (bdrv_getlength(bs) >> BDRV_SECTOR_BITS);
4350 bs->dirty_bitmap = hbitmap_alloc(bitmap_size, ffs(granularity) - 1);
7cd1e32a 4351 } else {
c6d22830 4352 if (bs->dirty_bitmap) {
8f0720ec 4353 hbitmap_free(bs->dirty_bitmap);
c6d22830 4354 bs->dirty_bitmap = NULL;
a55eb92c 4355 }
7cd1e32a 4356 }
4357}
4358
4359int bdrv_get_dirty(BlockDriverState *bs, int64_t sector)
4360{
8f0720ec
PB
4361 if (bs->dirty_bitmap) {
4362 return hbitmap_get(bs->dirty_bitmap, sector);
7cd1e32a 4363 } else {
4364 return 0;
4365 }
4366}
4367
8f0720ec 4368void bdrv_dirty_iter_init(BlockDriverState *bs, HBitmapIter *hbi)
1755da16 4369{
8f0720ec 4370 hbitmap_iter_init(hbi, bs->dirty_bitmap, 0);
1755da16
PB
4371}
4372
4373void bdrv_set_dirty(BlockDriverState *bs, int64_t cur_sector,
4374 int nr_sectors)
4375{
8f0720ec 4376 hbitmap_set(bs->dirty_bitmap, cur_sector, nr_sectors);
1755da16
PB
4377}
4378
a55eb92c
JK
4379void bdrv_reset_dirty(BlockDriverState *bs, int64_t cur_sector,
4380 int nr_sectors)
7cd1e32a 4381{
8f0720ec 4382 hbitmap_reset(bs->dirty_bitmap, cur_sector, nr_sectors);
7cd1e32a 4383}
aaa0eb75
LS
4384
4385int64_t bdrv_get_dirty_count(BlockDriverState *bs)
4386{
8f0720ec 4387 if (bs->dirty_bitmap) {
acc906c6 4388 return hbitmap_count(bs->dirty_bitmap);
8f0720ec
PB
4389 } else {
4390 return 0;
4391 }
aaa0eb75 4392}
f88e1a42 4393
db593f25
MT
4394void bdrv_set_in_use(BlockDriverState *bs, int in_use)
4395{
4396 assert(bs->in_use != in_use);
4397 bs->in_use = in_use;
4398}
4399
4400int bdrv_in_use(BlockDriverState *bs)
4401{
4402 return bs->in_use;
4403}
4404
28a7282a
LC
4405void bdrv_iostatus_enable(BlockDriverState *bs)
4406{
d6bf279e 4407 bs->iostatus_enabled = true;
58e21ef5 4408 bs->iostatus = BLOCK_DEVICE_IO_STATUS_OK;
28a7282a
LC
4409}
4410
4411/* The I/O status is only enabled if the drive explicitly
4412 * enables it _and_ the VM is configured to stop on errors */
4413bool bdrv_iostatus_is_enabled(const BlockDriverState *bs)
4414{
d6bf279e 4415 return (bs->iostatus_enabled &&
92aa5c6d
PB
4416 (bs->on_write_error == BLOCKDEV_ON_ERROR_ENOSPC ||
4417 bs->on_write_error == BLOCKDEV_ON_ERROR_STOP ||
4418 bs->on_read_error == BLOCKDEV_ON_ERROR_STOP));
28a7282a
LC
4419}
4420
4421void bdrv_iostatus_disable(BlockDriverState *bs)
4422{
d6bf279e 4423 bs->iostatus_enabled = false;
28a7282a
LC
4424}
4425
4426void bdrv_iostatus_reset(BlockDriverState *bs)
4427{
4428 if (bdrv_iostatus_is_enabled(bs)) {
58e21ef5 4429 bs->iostatus = BLOCK_DEVICE_IO_STATUS_OK;
3bd293c3
PB
4430 if (bs->job) {
4431 block_job_iostatus_reset(bs->job);
4432 }
28a7282a
LC
4433 }
4434}
4435
28a7282a
LC
4436void bdrv_iostatus_set_err(BlockDriverState *bs, int error)
4437{
3e1caa5f
PB
4438 assert(bdrv_iostatus_is_enabled(bs));
4439 if (bs->iostatus == BLOCK_DEVICE_IO_STATUS_OK) {
58e21ef5
LC
4440 bs->iostatus = error == ENOSPC ? BLOCK_DEVICE_IO_STATUS_NOSPACE :
4441 BLOCK_DEVICE_IO_STATUS_FAILED;
28a7282a
LC
4442 }
4443}
4444
a597e79c
CH
4445void
4446bdrv_acct_start(BlockDriverState *bs, BlockAcctCookie *cookie, int64_t bytes,
4447 enum BlockAcctType type)
4448{
4449 assert(type < BDRV_MAX_IOTYPE);
4450
4451 cookie->bytes = bytes;
c488c7f6 4452 cookie->start_time_ns = get_clock();
a597e79c
CH
4453 cookie->type = type;
4454}
4455
4456void
4457bdrv_acct_done(BlockDriverState *bs, BlockAcctCookie *cookie)
4458{
4459 assert(cookie->type < BDRV_MAX_IOTYPE);
4460
4461 bs->nr_bytes[cookie->type] += cookie->bytes;
4462 bs->nr_ops[cookie->type]++;
c488c7f6 4463 bs->total_time_ns[cookie->type] += get_clock() - cookie->start_time_ns;
a597e79c
CH
4464}
4465
d92ada22
LC
4466void bdrv_img_create(const char *filename, const char *fmt,
4467 const char *base_filename, const char *base_fmt,
f382d43a
MR
4468 char *options, uint64_t img_size, int flags,
4469 Error **errp, bool quiet)
f88e1a42
JS
4470{
4471 QEMUOptionParameter *param = NULL, *create_options = NULL;
d220894e 4472 QEMUOptionParameter *backing_fmt, *backing_file, *size;
f88e1a42
JS
4473 BlockDriverState *bs = NULL;
4474 BlockDriver *drv, *proto_drv;
96df67d1 4475 BlockDriver *backing_drv = NULL;
f88e1a42
JS
4476 int ret = 0;
4477
4478 /* Find driver and parse its options */
4479 drv = bdrv_find_format(fmt);
4480 if (!drv) {
71c79813 4481 error_setg(errp, "Unknown file format '%s'", fmt);
d92ada22 4482 return;
f88e1a42
JS
4483 }
4484
98289620 4485 proto_drv = bdrv_find_protocol(filename, true);
f88e1a42 4486 if (!proto_drv) {
71c79813 4487 error_setg(errp, "Unknown protocol '%s'", filename);
d92ada22 4488 return;
f88e1a42
JS
4489 }
4490
4491 create_options = append_option_parameters(create_options,
4492 drv->create_options);
4493 create_options = append_option_parameters(create_options,
4494 proto_drv->create_options);
4495
4496 /* Create parameter list with default values */
4497 param = parse_option_parameters("", create_options, param);
4498
4499 set_option_parameter_int(param, BLOCK_OPT_SIZE, img_size);
4500
4501 /* Parse -o options */
4502 if (options) {
4503 param = parse_option_parameters(options, create_options, param);
4504 if (param == NULL) {
71c79813 4505 error_setg(errp, "Invalid options for file format '%s'.", fmt);
f88e1a42
JS
4506 goto out;
4507 }
4508 }
4509
4510 if (base_filename) {
4511 if (set_option_parameter(param, BLOCK_OPT_BACKING_FILE,
4512 base_filename)) {
71c79813
LC
4513 error_setg(errp, "Backing file not supported for file format '%s'",
4514 fmt);
f88e1a42
JS
4515 goto out;
4516 }
4517 }
4518
4519 if (base_fmt) {
4520 if (set_option_parameter(param, BLOCK_OPT_BACKING_FMT, base_fmt)) {
71c79813
LC
4521 error_setg(errp, "Backing file format not supported for file "
4522 "format '%s'", fmt);
f88e1a42
JS
4523 goto out;
4524 }
4525 }
4526
792da93a
JS
4527 backing_file = get_option_parameter(param, BLOCK_OPT_BACKING_FILE);
4528 if (backing_file && backing_file->value.s) {
4529 if (!strcmp(filename, backing_file->value.s)) {
71c79813
LC
4530 error_setg(errp, "Error: Trying to create an image with the "
4531 "same filename as the backing file");
792da93a
JS
4532 goto out;
4533 }
4534 }
4535
f88e1a42
JS
4536 backing_fmt = get_option_parameter(param, BLOCK_OPT_BACKING_FMT);
4537 if (backing_fmt && backing_fmt->value.s) {
96df67d1
SH
4538 backing_drv = bdrv_find_format(backing_fmt->value.s);
4539 if (!backing_drv) {
71c79813
LC
4540 error_setg(errp, "Unknown backing file format '%s'",
4541 backing_fmt->value.s);
f88e1a42
JS
4542 goto out;
4543 }
4544 }
4545
4546 // The size for the image must always be specified, with one exception:
4547 // If we are using a backing file, we can obtain the size from there
d220894e
KW
4548 size = get_option_parameter(param, BLOCK_OPT_SIZE);
4549 if (size && size->value.n == -1) {
f88e1a42
JS
4550 if (backing_file && backing_file->value.s) {
4551 uint64_t size;
f88e1a42 4552 char buf[32];
63090dac
PB
4553 int back_flags;
4554
4555 /* backing files always opened read-only */
4556 back_flags =
4557 flags & ~(BDRV_O_RDWR | BDRV_O_SNAPSHOT | BDRV_O_NO_BACKING);
f88e1a42 4558
f88e1a42
JS
4559 bs = bdrv_new("");
4560
de9c0cec
KW
4561 ret = bdrv_open(bs, backing_file->value.s, NULL, back_flags,
4562 backing_drv);
f88e1a42 4563 if (ret < 0) {
71c79813
LC
4564 error_setg_errno(errp, -ret, "Could not open '%s'",
4565 backing_file->value.s);
f88e1a42
JS
4566 goto out;
4567 }
4568 bdrv_get_geometry(bs, &size);
4569 size *= 512;
4570
4571 snprintf(buf, sizeof(buf), "%" PRId64, size);
4572 set_option_parameter(param, BLOCK_OPT_SIZE, buf);
4573 } else {
71c79813 4574 error_setg(errp, "Image creation needs a size parameter");
f88e1a42
JS
4575 goto out;
4576 }
4577 }
4578
f382d43a
MR
4579 if (!quiet) {
4580 printf("Formatting '%s', fmt=%s ", filename, fmt);
4581 print_option_parameters(param);
4582 puts("");
4583 }
f88e1a42 4584 ret = bdrv_create(drv, filename, param);
f88e1a42
JS
4585 if (ret < 0) {
4586 if (ret == -ENOTSUP) {
71c79813
LC
4587 error_setg(errp,"Formatting or formatting option not supported for "
4588 "file format '%s'", fmt);
f88e1a42 4589 } else if (ret == -EFBIG) {
f3f4d2c0
KW
4590 const char *cluster_size_hint = "";
4591 if (get_option_parameter(create_options, BLOCK_OPT_CLUSTER_SIZE)) {
4592 cluster_size_hint = " (try using a larger cluster size)";
4593 }
4594 error_setg(errp, "The image size is too large for file format '%s'%s",
4595 fmt, cluster_size_hint);
f88e1a42 4596 } else {
71c79813
LC
4597 error_setg(errp, "%s: error while creating %s: %s", filename, fmt,
4598 strerror(-ret));
f88e1a42
JS
4599 }
4600 }
4601
4602out:
4603 free_option_parameters(create_options);
4604 free_option_parameters(param);
4605
4606 if (bs) {
4607 bdrv_delete(bs);
4608 }
f88e1a42 4609}
85d126f3
SH
4610
4611AioContext *bdrv_get_aio_context(BlockDriverState *bs)
4612{
4613 /* Currently BlockDriverState always uses the main loop AioContext */
4614 return qemu_get_aio_context();
4615}
d616b224
SH
4616
4617void bdrv_add_before_write_notifier(BlockDriverState *bs,
4618 NotifierWithReturn *notifier)
4619{
4620 notifier_with_return_list_add(&bs->before_write_notifiers, notifier);
4621}