]> git.proxmox.com Git - qemu.git/blob - block.c
virtio-9p-device: Avoid freeing uninitialized memory
[qemu.git] / block.c
1 /*
2 * QEMU System Emulator block driver
3 *
4 * Copyright (c) 2003 Fabrice Bellard
5 *
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 */
24 #include "config-host.h"
25 #include "qemu-common.h"
26 #include "trace.h"
27 #include "monitor/monitor.h"
28 #include "block/block_int.h"
29 #include "block/blockjob.h"
30 #include "qemu/module.h"
31 #include "qapi/qmp/qjson.h"
32 #include "sysemu/sysemu.h"
33 #include "qemu/notify.h"
34 #include "block/coroutine.h"
35 #include "qmp-commands.h"
36 #include "qemu/timer.h"
37
38 #ifdef CONFIG_BSD
39 #include <sys/types.h>
40 #include <sys/stat.h>
41 #include <sys/ioctl.h>
42 #include <sys/queue.h>
43 #ifndef __DragonFly__
44 #include <sys/disk.h>
45 #endif
46 #endif
47
48 #ifdef _WIN32
49 #include <windows.h>
50 #endif
51
52 #define NOT_DONE 0x7fffffff /* used while emulated sync operation in progress */
53
54 typedef enum {
55 BDRV_REQ_COPY_ON_READ = 0x1,
56 BDRV_REQ_ZERO_WRITE = 0x2,
57 } BdrvRequestFlags;
58
59 static void bdrv_dev_change_media_cb(BlockDriverState *bs, bool load);
60 static BlockDriverAIOCB *bdrv_aio_readv_em(BlockDriverState *bs,
61 int64_t sector_num, QEMUIOVector *qiov, int nb_sectors,
62 BlockDriverCompletionFunc *cb, void *opaque);
63 static BlockDriverAIOCB *bdrv_aio_writev_em(BlockDriverState *bs,
64 int64_t sector_num, QEMUIOVector *qiov, int nb_sectors,
65 BlockDriverCompletionFunc *cb, void *opaque);
66 static int coroutine_fn bdrv_co_readv_em(BlockDriverState *bs,
67 int64_t sector_num, int nb_sectors,
68 QEMUIOVector *iov);
69 static int coroutine_fn bdrv_co_writev_em(BlockDriverState *bs,
70 int64_t sector_num, int nb_sectors,
71 QEMUIOVector *iov);
72 static int coroutine_fn bdrv_co_do_readv(BlockDriverState *bs,
73 int64_t sector_num, int nb_sectors, QEMUIOVector *qiov,
74 BdrvRequestFlags flags);
75 static int coroutine_fn bdrv_co_do_writev(BlockDriverState *bs,
76 int64_t sector_num, int nb_sectors, QEMUIOVector *qiov,
77 BdrvRequestFlags flags);
78 static 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,
84 bool is_write);
85 static void coroutine_fn bdrv_co_do_rw(void *opaque);
86 static int coroutine_fn bdrv_co_do_write_zeroes(BlockDriverState *bs,
87 int64_t sector_num, int nb_sectors);
88
89 static bool bdrv_exceed_bps_limits(BlockDriverState *bs, int nb_sectors,
90 bool is_write, double elapsed_time, uint64_t *wait);
91 static bool bdrv_exceed_iops_limits(BlockDriverState *bs, bool is_write,
92 double elapsed_time, uint64_t *wait);
93 static bool bdrv_exceed_io_limits(BlockDriverState *bs, int nb_sectors,
94 bool is_write, int64_t *wait);
95
96 static QTAILQ_HEAD(, BlockDriverState) bdrv_states =
97 QTAILQ_HEAD_INITIALIZER(bdrv_states);
98
99 static QLIST_HEAD(, BlockDriver) bdrv_drivers =
100 QLIST_HEAD_INITIALIZER(bdrv_drivers);
101
102 /* If non-zero, use only whitelisted block drivers */
103 static int use_bdrv_whitelist;
104
105 #ifdef _WIN32
106 static 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
113 int 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
125 /* throttling disk I/O limits */
126 void bdrv_io_limits_disable(BlockDriverState *bs)
127 {
128 bs->io_limits_enabled = false;
129
130 do {} while (qemu_co_enter_next(&bs->throttled_reqs));
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;
140 }
141
142 static void bdrv_block_timer(void *opaque)
143 {
144 BlockDriverState *bs = opaque;
145
146 qemu_co_enter_next(&bs->throttled_reqs);
147 }
148
149 void 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);
153 bs->io_limits_enabled = true;
154 }
155
156 bool 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
167 static 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
192 /* check if the path starts with "<protocol>:" */
193 static int path_has_protocol(const char *path)
194 {
195 const char *p;
196
197 #ifdef _WIN32
198 if (is_windows_drive(path) ||
199 is_windows_drive_prefix(path)) {
200 return 0;
201 }
202 p = path + strcspn(path, ":/\\");
203 #else
204 p = path + strcspn(path, ":/");
205 #endif
206
207 return *p == ':';
208 }
209
210 int path_is_absolute(const char *path)
211 {
212 #ifdef _WIN32
213 /* specific case for names like: "\\.\d:" */
214 if (is_windows_drive(path) || is_windows_drive_prefix(path)) {
215 return 1;
216 }
217 return (*path == '/' || *path == '\\');
218 #else
219 return (*path == '/');
220 #endif
221 }
222
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. */
226 void path_combine(char *dest, int dest_size,
227 const char *base_path,
228 const char *filename)
229 {
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;
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
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);
264 }
265 }
266
267 void 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
276 void bdrv_register(BlockDriver *bdrv)
277 {
278 /* Block drivers without coroutine functions need emulation */
279 if (!bdrv->bdrv_co_readv) {
280 bdrv->bdrv_co_readv = bdrv_co_readv_em;
281 bdrv->bdrv_co_writev = bdrv_co_writev_em;
282
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 */
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;
290 }
291 }
292
293 QLIST_INSERT_HEAD(&bdrv_drivers, bdrv, list);
294 }
295
296 /* create a new block device (by default it is empty) */
297 BlockDriverState *bdrv_new(const char *device_name)
298 {
299 BlockDriverState *bs;
300
301 bs = g_malloc0(sizeof(BlockDriverState));
302 pstrcpy(bs->device_name, sizeof(bs->device_name), device_name);
303 if (device_name[0] != '\0') {
304 QTAILQ_INSERT_TAIL(&bdrv_states, bs, list);
305 }
306 bdrv_iostatus_disable(bs);
307 notifier_list_init(&bs->close_notifiers);
308 notifier_with_return_list_init(&bs->before_write_notifiers);
309
310 return bs;
311 }
312
313 void bdrv_add_close_notifier(BlockDriverState *bs, Notifier *notify)
314 {
315 notifier_list_add(&bs->close_notifiers, notify);
316 }
317
318 BlockDriver *bdrv_find_format(const char *format_name)
319 {
320 BlockDriver *drv1;
321 QLIST_FOREACH(drv1, &bdrv_drivers, list) {
322 if (!strcmp(drv1->format_name, format_name)) {
323 return drv1;
324 }
325 }
326 return NULL;
327 }
328
329 static int bdrv_is_whitelisted(BlockDriver *drv, bool read_only)
330 {
331 static const char *whitelist_rw[] = {
332 CONFIG_BDRV_RW_WHITELIST
333 };
334 static const char *whitelist_ro[] = {
335 CONFIG_BDRV_RO_WHITELIST
336 };
337 const char **p;
338
339 if (!whitelist_rw[0] && !whitelist_ro[0]) {
340 return 1; /* no whitelist, anything goes */
341 }
342
343 for (p = whitelist_rw; *p; p++) {
344 if (!strcmp(drv->format_name, *p)) {
345 return 1;
346 }
347 }
348 if (read_only) {
349 for (p = whitelist_ro; *p; p++) {
350 if (!strcmp(drv->format_name, *p)) {
351 return 1;
352 }
353 }
354 }
355 return 0;
356 }
357
358 BlockDriver *bdrv_find_whitelisted_format(const char *format_name,
359 bool read_only)
360 {
361 BlockDriver *drv = bdrv_find_format(format_name);
362 return drv && bdrv_is_whitelisted(drv, read_only) ? drv : NULL;
363 }
364
365 typedef struct CreateCo {
366 BlockDriver *drv;
367 char *filename;
368 QEMUOptionParameter *options;
369 int ret;
370 } CreateCo;
371
372 static 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
380 int bdrv_create(BlockDriver *drv, const char* filename,
381 QEMUOptionParameter *options)
382 {
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) {
394 ret = -ENOTSUP;
395 goto out;
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;
410
411 out:
412 g_free(cco.filename);
413 return ret;
414 }
415
416 int bdrv_create_file(const char* filename, QEMUOptionParameter *options)
417 {
418 BlockDriver *drv;
419
420 drv = bdrv_find_protocol(filename, true);
421 if (drv == NULL) {
422 return -ENOENT;
423 }
424
425 return bdrv_create(drv, filename, options);
426 }
427
428 /*
429 * Create a uniquely-named empty temporary file.
430 * Return 0 upon success, otherwise a negative errno value.
431 */
432 int get_tmp_filename(char *filename, int size)
433 {
434 #ifdef _WIN32
435 char temp_dir[MAX_PATH];
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());
442 #else
443 int fd;
444 const char *tmpdir;
445 tmpdir = getenv("TMPDIR");
446 if (!tmpdir)
447 tmpdir = "/tmp";
448 if (snprintf(filename, size, "%s/vl.XXXXXX", tmpdir) >= size) {
449 return -EOVERFLOW;
450 }
451 fd = mkstemp(filename);
452 if (fd < 0) {
453 return -errno;
454 }
455 if (close(fd) != 0) {
456 unlink(filename);
457 return -errno;
458 }
459 return 0;
460 #endif
461 }
462
463 /*
464 * Detect host devices. By convention, /dev/cdrom[N] is always
465 * recognized as a host CDROM.
466 */
467 static 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
485 BlockDriver *bdrv_find_protocol(const char *filename,
486 bool allow_protocol_prefix)
487 {
488 BlockDriver *drv1;
489 char protocol[128];
490 int len;
491 const char *p;
492
493 /* TODO Drivers without bdrv_file_open must be specified explicitly */
494
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
507 if (!path_has_protocol(filename) || !allow_protocol_prefix) {
508 return bdrv_find_format("file");
509 }
510
511 p = strchr(filename, ':');
512 assert(p != NULL);
513 len = p - filename;
514 if (len > sizeof(protocol) - 1)
515 len = sizeof(protocol) - 1;
516 memcpy(protocol, filename, len);
517 protocol[len] = '\0';
518 QLIST_FOREACH(drv1, &bdrv_drivers, list) {
519 if (drv1->protocol_name &&
520 !strcmp(drv1->protocol_name, protocol)) {
521 return drv1;
522 }
523 }
524 return NULL;
525 }
526
527 static int find_image_format(BlockDriverState *bs, const char *filename,
528 BlockDriver **pdrv)
529 {
530 int score, score_max;
531 BlockDriver *drv1, *drv;
532 uint8_t buf[2048];
533 int ret = 0;
534
535 /* Return the raw BlockDriver * to scsi-generic devices or empty drives */
536 if (bs->sg || !bdrv_is_inserted(bs) || bdrv_getlength(bs) == 0) {
537 drv = bdrv_find_format("raw");
538 if (!drv) {
539 ret = -ENOENT;
540 }
541 *pdrv = drv;
542 return ret;
543 }
544
545 ret = bdrv_pread(bs, 0, buf, sizeof(buf));
546 if (ret < 0) {
547 *pdrv = NULL;
548 return ret;
549 }
550
551 score_max = 0;
552 drv = NULL;
553 QLIST_FOREACH(drv1, &bdrv_drivers, list) {
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 }
560 }
561 }
562 if (!drv) {
563 ret = -ENOENT;
564 }
565 *pdrv = drv;
566 return ret;
567 }
568
569 /**
570 * Set the current 'total_sectors' value
571 */
572 static int refresh_total_sectors(BlockDriverState *bs, int64_t hint)
573 {
574 BlockDriver *drv = bs->drv;
575
576 /* Do not attempt drv->bdrv_getlength() on scsi-generic devices */
577 if (bs->sg)
578 return 0;
579
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
593 /**
594 * Set open flags for a given discard mode
595 *
596 * Return 0 on success, -1 if the discard mode was invalid.
597 */
598 int 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
613 /**
614 * Set open flags for a given cache mode
615 *
616 * Return 0 on success, -1 if the cache mode was invalid.
617 */
618 int 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;
624 } else if (!strcmp(mode, "directsync")) {
625 *flags |= BDRV_O_NOCACHE;
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
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 */
645 void bdrv_enable_copy_on_read(BlockDriverState *bs)
646 {
647 bs->copy_on_read++;
648 }
649
650 void bdrv_disable_copy_on_read(BlockDriverState *bs)
651 {
652 assert(bs->copy_on_read > 0);
653 bs->copy_on_read--;
654 }
655
656 static 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
676 /*
677 * Common part for opening disk images and files
678 *
679 * Removes all processed options from *options.
680 */
681 static int bdrv_open_common(BlockDriverState *bs, BlockDriverState *file,
682 QDict *options, int flags, BlockDriver *drv)
683 {
684 int ret, open_flags;
685 const char *filename;
686
687 assert(drv != NULL);
688 assert(bs->file == NULL);
689 assert(options != NULL && bs->options != options);
690
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);
698
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
707 bs->open_flags = flags;
708 bs->buffer_alignment = 512;
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 }
715
716 assert(bs->copy_on_read == 0); /* bdrv_new() and bdrv_close() make it so */
717 if (!bs->read_only && (flags & BDRV_O_COPY_ON_READ)) {
718 bdrv_enable_copy_on_read(bs);
719 }
720
721 if (filename != NULL) {
722 pstrcpy(bs->filename, sizeof(bs->filename), filename);
723 } else {
724 bs->filename[0] = '\0';
725 }
726
727 bs->drv = drv;
728 bs->opaque = g_malloc0(drv->instance_size);
729
730 bs->enable_write_cache = !!(flags & BDRV_O_CACHE_WB);
731
732 /* Open the image, either directly or using a protocol */
733 if (drv->bdrv_file_open) {
734 assert(file == NULL);
735 assert(drv->bdrv_parse_filename || filename != NULL);
736 ret = drv->bdrv_file_open(bs, options, open_flags);
737 } else {
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 }
745 assert(file != NULL);
746 bs->file = file;
747 ret = drv->bdrv_open(bs, options, open_flags);
748 }
749
750 if (ret < 0) {
751 goto free_and_fail;
752 }
753
754 ret = refresh_total_sectors(bs, bs->total_sectors);
755 if (ret < 0) {
756 goto free_and_fail;
757 }
758
759 #ifndef _WIN32
760 if (bs->is_temporary) {
761 assert(filename != NULL);
762 unlink(filename);
763 }
764 #endif
765 return 0;
766
767 free_and_fail:
768 bs->file = NULL;
769 g_free(bs->opaque);
770 bs->opaque = NULL;
771 bs->drv = NULL;
772 return ret;
773 }
774
775 /*
776 * Opens a file using a protocol (file, host_device, nbd, ...)
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.
782 */
783 int bdrv_file_open(BlockDriverState **pbs, const char *filename,
784 QDict *options, int flags)
785 {
786 BlockDriverState *bs;
787 BlockDriver *drv;
788 const char *drvname;
789 bool allow_protocol_prefix = false;
790 int ret;
791
792 /* NULL means an empty set of options */
793 if (options == NULL) {
794 options = qdict_new();
795 }
796
797 bs = bdrv_new("");
798 bs->options = options;
799 options = qdict_clone_shallow(options);
800
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));
806 allow_protocol_prefix = true;
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
814 /* Find the right block driver */
815 drvname = qdict_get_try_str(options, "driver");
816 if (drvname) {
817 drv = bdrv_find_whitelisted_format(drvname, !(flags & BDRV_O_RDWR));
818 qdict_del(options, "driver");
819 } else if (filename) {
820 drv = bdrv_find_protocol(filename, allow_protocol_prefix);
821 if (!drv) {
822 qerror_report(ERROR_CLASS_GENERIC_ERROR, "Unknown protocol");
823 }
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) {
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 }
845 qdict_del(options, "filename");
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;
852 }
853
854 ret = bdrv_open_common(bs, NULL, options, flags, drv);
855 if (ret < 0) {
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;
867 }
868 QDECREF(options);
869
870 bs->growable = 1;
871 *pbs = bs;
872 return 0;
873
874 fail:
875 QDECREF(options);
876 if (!bs->drv) {
877 QDECREF(bs->options);
878 }
879 bdrv_delete(bs);
880 return ret;
881 }
882
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 */
891 int bdrv_open_backing_file(BlockDriverState *bs, QDict *options)
892 {
893 char backing_filename[PATH_MAX];
894 int back_flags, ret;
895 BlockDriver *back_drv = NULL;
896
897 if (bs->backing_hd != NULL) {
898 QDECREF(options);
899 return 0;
900 }
901
902 /* NULL means an empty set of options */
903 if (options == NULL) {
904 options = qdict_new();
905 }
906
907 bs->open_flags &= ~BDRV_O_NO_BACKING;
908 if (qdict_haskey(options, "file.filename")) {
909 backing_filename[0] = '\0';
910 } else if (bs->backing_file[0] == '\0' && qdict_size(options) == 0) {
911 QDECREF(options);
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
926 ret = bdrv_open(bs->backing_hd,
927 *backing_filename ? backing_filename : NULL, options,
928 back_flags, back_drv);
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
938 static 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
957 /*
958 * Opens a disk image (raw, qcow2, vmdk, ...)
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.
964 */
965 int bdrv_open(BlockDriverState *bs, const char *filename, QDict *options,
966 int flags, BlockDriver *drv)
967 {
968 int ret;
969 /* TODO: extra byte is a hack to ensure MAX_PATH space on Windows. */
970 char tmp_filename[PATH_MAX + 1];
971 BlockDriverState *file = NULL;
972 QDict *file_options = NULL;
973 const char *drvname;
974
975 /* NULL means an empty set of options */
976 if (options == NULL) {
977 options = qdict_new();
978 }
979
980 bs->options = options;
981 options = qdict_clone_shallow(options);
982
983 /* For snapshot=on, create a temporary qcow2 overlay */
984 if (flags & BDRV_O_SNAPSHOT) {
985 BlockDriverState *bs1;
986 int64_t total_size;
987 BlockDriver *bdrv_qcow2;
988 QEMUOptionParameter *create_options;
989 char backing_filename[PATH_MAX];
990
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
998 /* if snapshot, we create a temporary backing file and open it
999 instead of opening 'filename' directly */
1000
1001 /* if there is a backing file, use it */
1002 bs1 = bdrv_new("");
1003 ret = bdrv_open(bs1, filename, NULL, 0, drv);
1004 if (ret < 0) {
1005 bdrv_delete(bs1);
1006 goto fail;
1007 }
1008 total_size = bdrv_getlength(bs1) & BDRV_SECTOR_MASK;
1009
1010 bdrv_delete(bs1);
1011
1012 ret = get_tmp_filename(tmp_filename, sizeof(tmp_filename));
1013 if (ret < 0) {
1014 goto fail;
1015 }
1016
1017 /* Real path is meaningless for protocols */
1018 if (path_has_protocol(filename)) {
1019 snprintf(backing_filename, sizeof(backing_filename),
1020 "%s", filename);
1021 } else if (!realpath(filename, backing_filename)) {
1022 ret = -errno;
1023 goto fail;
1024 }
1025
1026 bdrv_qcow2 = bdrv_find_format("qcow2");
1027 create_options = parse_option_parameters("", bdrv_qcow2->create_options,
1028 NULL);
1029
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);
1033 if (drv) {
1034 set_option_parameter(create_options, BLOCK_OPT_BACKING_FMT,
1035 drv->format_name);
1036 }
1037
1038 ret = bdrv_create(bdrv_qcow2, tmp_filename, create_options);
1039 free_option_parameters(create_options);
1040 if (ret < 0) {
1041 goto fail;
1042 }
1043
1044 filename = tmp_filename;
1045 drv = bdrv_qcow2;
1046 bs->is_temporary = 1;
1047 }
1048
1049 /* Open image file without format layer */
1050 if (flags & BDRV_O_RDWR) {
1051 flags |= BDRV_O_ALLOW_RDWR;
1052 }
1053
1054 extract_subqdict(options, &file_options, "file.");
1055
1056 ret = bdrv_file_open(&file, filename, file_options,
1057 bdrv_open_flags(bs, flags | BDRV_O_UNMAP));
1058 if (ret < 0) {
1059 goto fail;
1060 }
1061
1062 /* Find the right image format driver */
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
1069 if (!drv) {
1070 ret = find_image_format(file, filename, &drv);
1071 }
1072
1073 if (!drv) {
1074 goto unlink_and_fail;
1075 }
1076
1077 /* Open the image */
1078 ret = bdrv_open_common(bs, file, options, flags, drv);
1079 if (ret < 0) {
1080 goto unlink_and_fail;
1081 }
1082
1083 if (bs->file != file) {
1084 bdrv_delete(file);
1085 file = NULL;
1086 }
1087
1088 /* If there is a backing file, use it */
1089 if ((flags & BDRV_O_NO_BACKING) == 0) {
1090 QDict *backing_options;
1091
1092 extract_subqdict(options, &backing_options, "backing.");
1093 ret = bdrv_open_backing_file(bs, backing_options);
1094 if (ret < 0) {
1095 goto close_and_fail;
1096 }
1097 }
1098
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
1111 if (!bdrv_key_required(bs)) {
1112 bdrv_dev_change_media_cb(bs, true);
1113 }
1114
1115 /* throttling disk I/O limits */
1116 if (bs->io_limits_enabled) {
1117 bdrv_io_limits_enable(bs);
1118 }
1119
1120 return 0;
1121
1122 unlink_and_fail:
1123 if (file != NULL) {
1124 bdrv_delete(file);
1125 }
1126 if (bs->is_temporary) {
1127 unlink(filename);
1128 }
1129 fail:
1130 QDECREF(bs->options);
1131 QDECREF(options);
1132 bs->options = NULL;
1133 return ret;
1134
1135 close_and_fail:
1136 bdrv_close(bs);
1137 QDECREF(options);
1138 return ret;
1139 }
1140
1141 typedef 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 */
1165 BlockReopenQueue *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 */
1204 int 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
1231 cleanup:
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. */
1244 int 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 */
1275 int 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 {
1309 error_setg(errp, "failed while preparing to reopen image '%s'",
1310 reopen_state->bs->filename);
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
1326 error:
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 */
1335 void 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 */
1359 void 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
1373 void bdrv_close(BlockDriverState *bs)
1374 {
1375 if (bs->job) {
1376 block_job_cancel_sync(bs->job);
1377 }
1378 bdrv_drain_all(); /* complete I/O */
1379 bdrv_flush(bs);
1380 bdrv_drain_all(); /* in case flush left pending I/O */
1381 notifier_list_notify(&bs->close_notifiers, bs);
1382
1383 if (bs->drv) {
1384 if (bs->backing_hd) {
1385 bdrv_delete(bs->backing_hd);
1386 bs->backing_hd = NULL;
1387 }
1388 bs->drv->bdrv_close(bs);
1389 g_free(bs->opaque);
1390 #ifdef _WIN32
1391 if (bs->is_temporary) {
1392 unlink(bs->filename);
1393 }
1394 #endif
1395 bs->opaque = NULL;
1396 bs->drv = NULL;
1397 bs->copy_on_read = 0;
1398 bs->backing_file[0] = '\0';
1399 bs->backing_format[0] = '\0';
1400 bs->total_sectors = 0;
1401 bs->encrypted = 0;
1402 bs->valid_key = 0;
1403 bs->sg = 0;
1404 bs->growable = 0;
1405 QDECREF(bs->options);
1406 bs->options = NULL;
1407
1408 if (bs->file != NULL) {
1409 bdrv_delete(bs->file);
1410 bs->file = NULL;
1411 }
1412 }
1413
1414 bdrv_dev_change_media_cb(bs, false);
1415
1416 /*throttling disk I/O limits*/
1417 if (bs->io_limits_enabled) {
1418 bdrv_io_limits_disable(bs);
1419 }
1420 }
1421
1422 void bdrv_close_all(void)
1423 {
1424 BlockDriverState *bs;
1425
1426 QTAILQ_FOREACH(bs, &bdrv_states, list) {
1427 bdrv_close(bs);
1428 }
1429 }
1430
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.
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.
1442 */
1443 void bdrv_drain_all(void)
1444 {
1445 BlockDriverState *bs;
1446 bool busy;
1447
1448 do {
1449 busy = qemu_aio_wait();
1450
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) {
1455 while (qemu_co_enter_next(&bs->throttled_reqs)) {
1456 busy = true;
1457 }
1458 }
1459 } while (busy);
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
1468 /* make a BlockDriverState anonymous by removing from bdrv_state list.
1469 Also, NULL terminate the device_name to prevent double remove */
1470 void 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
1478 static void bdrv_rebind(BlockDriverState *bs)
1479 {
1480 if (bs->drv && bs->drv->bdrv_rebind) {
1481 bs->drv->bdrv_rebind(bs);
1482 }
1483 }
1484
1485 static void bdrv_move_feature_fields(BlockDriverState *bs_dest,
1486 BlockDriverState *bs_src)
1487 {
1488 /* move some fields that need to stay attached to the device */
1489 bs_dest->open_flags = bs_src->open_flags;
1490
1491 /* dev info */
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;
1497
1498 bs_dest->enable_write_cache = bs_src->enable_write_cache;
1499
1500 /* i/o timing parameters */
1501 bs_dest->slice_start = bs_src->slice_start;
1502 bs_dest->slice_end = bs_src->slice_end;
1503 bs_dest->slice_submitted = bs_src->slice_submitted;
1504 bs_dest->io_limits = bs_src->io_limits;
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;
1508
1509 /* r/w error */
1510 bs_dest->on_read_error = bs_src->on_read_error;
1511 bs_dest->on_write_error = bs_src->on_write_error;
1512
1513 /* i/o status */
1514 bs_dest->iostatus_enabled = bs_src->iostatus_enabled;
1515 bs_dest->iostatus = bs_src->iostatus;
1516
1517 /* dirty bitmap */
1518 bs_dest->dirty_bitmap = bs_src->dirty_bitmap;
1519
1520 /* job */
1521 bs_dest->in_use = bs_src->in_use;
1522 bs_dest->job = bs_src->job;
1523
1524 /* keep the same entry in bdrv_states */
1525 pstrcpy(bs_dest->device_name, sizeof(bs_dest->device_name),
1526 bs_src->device_name);
1527 bs_dest->list = bs_src->list;
1528 }
1529
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 */
1542 void bdrv_swap(BlockDriverState *bs_new, BlockDriverState *bs_old)
1543 {
1544 BlockDriverState tmp;
1545
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);
1554
1555 tmp = *bs_new;
1556 *bs_new = *bs_old;
1557 *bs_old = tmp;
1558
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);
1563
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);
1573
1574 bdrv_rebind(bs_new);
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 */
1589 void 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 : "");
1601 }
1602
1603 void bdrv_delete(BlockDriverState *bs)
1604 {
1605 assert(!bs->dev);
1606 assert(!bs->job);
1607 assert(!bs->in_use);
1608
1609 /* remove from list, if necessary */
1610 bdrv_make_anon(bs);
1611
1612 bdrv_close(bs);
1613
1614 g_free(bs);
1615 }
1616
1617 int bdrv_attach_dev(BlockDriverState *bs, void *dev)
1618 /* TODO change to DeviceState *dev when all users are qdevified */
1619 {
1620 if (bs->dev) {
1621 return -EBUSY;
1622 }
1623 bs->dev = dev;
1624 bdrv_iostatus_reset(bs);
1625 return 0;
1626 }
1627
1628 /* TODO qdevified devices don't use this, remove when devices are qdevified */
1629 void bdrv_attach_dev_nofail(BlockDriverState *bs, void *dev)
1630 {
1631 if (bdrv_attach_dev(bs, dev) < 0) {
1632 abort();
1633 }
1634 }
1635
1636 void 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;
1641 bs->dev_ops = NULL;
1642 bs->dev_opaque = NULL;
1643 bs->buffer_alignment = 512;
1644 }
1645
1646 /* TODO change to return DeviceState * when all users are qdevified */
1647 void *bdrv_get_attached_dev(BlockDriverState *bs)
1648 {
1649 return bs->dev;
1650 }
1651
1652 void 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
1659 void bdrv_emit_qmp_error_event(const BlockDriverState *bdrv,
1660 enum MonitorEvent ev,
1661 BlockErrorAction action, bool is_read)
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");
1684 monitor_protocol_event(ev, data);
1685
1686 qobject_decref(data);
1687 }
1688
1689 static 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
1700 static void bdrv_dev_change_media_cb(BlockDriverState *bs, bool load)
1701 {
1702 if (bs->dev_ops && bs->dev_ops->change_media_cb) {
1703 bool tray_was_closed = !bdrv_dev_is_tray_open(bs);
1704 bs->dev_ops->change_media_cb(bs->dev_opaque, load);
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 }
1713 }
1714 }
1715
1716 bool bdrv_dev_has_removable_media(BlockDriverState *bs)
1717 {
1718 return !bs->dev || (bs->dev_ops && bs->dev_ops->change_media_cb);
1719 }
1720
1721 void 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
1728 bool 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
1736 static 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);
1740 }
1741 }
1742
1743 bool 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
1751 /*
1752 * Run consistency checks on an image
1753 *
1754 * Returns 0 if the check could be completed (it doesn't mean that the image is
1755 * free of errors) or -errno when an internal error occurred. The results of the
1756 * check are stored in res.
1757 */
1758 int bdrv_check(BlockDriverState *bs, BdrvCheckResult *res, BdrvCheckMode fix)
1759 {
1760 if (bs->drv->bdrv_check == NULL) {
1761 return -ENOTSUP;
1762 }
1763
1764 memset(res, 0, sizeof(*res));
1765 return bs->drv->bdrv_check(bs, res, fix);
1766 }
1767
1768 #define COMMIT_BUF_SECTORS 2048
1769
1770 /* commit COW file into the raw image */
1771 int bdrv_commit(BlockDriverState *bs)
1772 {
1773 BlockDriver *drv = bs->drv;
1774 int64_t sector, total_sectors;
1775 int n, ro, open_flags;
1776 int ret = 0;
1777 uint8_t *buf;
1778 char filename[PATH_MAX];
1779
1780 if (!drv)
1781 return -ENOMEDIUM;
1782
1783 if (!bs->backing_hd) {
1784 return -ENOTSUP;
1785 }
1786
1787 if (bdrv_in_use(bs) || bdrv_in_use(bs->backing_hd)) {
1788 return -EBUSY;
1789 }
1790
1791 ro = bs->backing_hd->read_only;
1792 /* Use pstrcpy (not strncpy): filename must be NUL-terminated. */
1793 pstrcpy(filename, sizeof(filename), bs->backing_hd->filename);
1794 open_flags = bs->backing_hd->open_flags;
1795
1796 if (ro) {
1797 if (bdrv_reopen(bs->backing_hd, open_flags | BDRV_O_RDWR, NULL)) {
1798 return -EACCES;
1799 }
1800 }
1801
1802 total_sectors = bdrv_getlength(bs) >> BDRV_SECTOR_BITS;
1803 buf = g_malloc(COMMIT_BUF_SECTORS * BDRV_SECTOR_SIZE);
1804
1805 for (sector = 0; sector < total_sectors; sector += n) {
1806 if (bdrv_is_allocated(bs, sector, COMMIT_BUF_SECTORS, &n)) {
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 }
1817 }
1818 }
1819
1820 if (drv->bdrv_make_empty) {
1821 ret = drv->bdrv_make_empty(bs);
1822 bdrv_flush(bs);
1823 }
1824
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);
1831
1832 ro_cleanup:
1833 g_free(buf);
1834
1835 if (ro) {
1836 /* ignoring error return here */
1837 bdrv_reopen(bs->backing_hd, open_flags & ~BDRV_O_RDWR, NULL);
1838 }
1839
1840 return ret;
1841 }
1842
1843 int bdrv_commit_all(void)
1844 {
1845 BlockDriverState *bs;
1846
1847 QTAILQ_FOREACH(bs, &bdrv_states, list) {
1848 if (bs->drv && bs->backing_hd) {
1849 int ret = bdrv_commit(bs);
1850 if (ret < 0) {
1851 return ret;
1852 }
1853 }
1854 }
1855 return 0;
1856 }
1857
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 */
1863 static void tracked_request_end(BdrvTrackedRequest *req)
1864 {
1865 QLIST_REMOVE(req, list);
1866 qemu_co_queue_restart_all(&req->wait_queue);
1867 }
1868
1869 /**
1870 * Add an active request to the tracked requests list
1871 */
1872 static 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,
1882 .co = qemu_coroutine_self(),
1883 };
1884
1885 qemu_co_queue_init(&req->wait_queue);
1886
1887 QLIST_INSERT_HEAD(&bs->tracked_requests, req, list);
1888 }
1889
1890 /**
1891 * Round a region to cluster boundaries
1892 */
1893 void 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)
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
1911 static bool tracked_request_overlaps(BdrvTrackedRequest *req,
1912 int64_t sector_num, int nb_sectors) {
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;
1922 }
1923
1924 static void coroutine_fn wait_for_overlapping_requests(BlockDriverState *bs,
1925 int64_t sector_num, int nb_sectors)
1926 {
1927 BdrvTrackedRequest *req;
1928 int64_t cluster_sector_num;
1929 int cluster_nb_sectors;
1930 bool retry;
1931
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 */
1938 bdrv_round_to_clusters(bs, sector_num, nb_sectors,
1939 &cluster_sector_num, &cluster_nb_sectors);
1940
1941 do {
1942 retry = false;
1943 QLIST_FOREACH(req, &bs->tracked_requests, list) {
1944 if (tracked_request_overlaps(req, cluster_sector_num,
1945 cluster_nb_sectors)) {
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
1952 qemu_co_queue_wait(&req->wait_queue);
1953 retry = true;
1954 break;
1955 }
1956 }
1957 } while (retry);
1958 }
1959
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 */
1968 int bdrv_change_backing_file(BlockDriverState *bs,
1969 const char *backing_file, const char *backing_fmt)
1970 {
1971 BlockDriver *drv = bs->drv;
1972 int ret;
1973
1974 /* Backing file format doesn't make sense without a backing file */
1975 if (backing_fmt && !backing_file) {
1976 return -EINVAL;
1977 }
1978
1979 if (drv->bdrv_change_backing_file != NULL) {
1980 ret = drv->bdrv_change_backing_file(bs, backing_file, backing_fmt);
1981 } else {
1982 ret = -ENOTSUP;
1983 }
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;
1990 }
1991
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 */
2000 BlockDriverState *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
2027 typedef 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 */
2059 int 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
2127 exit:
2128 QSIMPLEQ_FOREACH_SAFE(intermediate_state, &states_to_delete, entry, next) {
2129 g_free(intermediate_state);
2130 }
2131 return ret;
2132 }
2133
2134
2135 static 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
2148 if (offset < 0)
2149 return -EIO;
2150
2151 if ((offset > len) || (len - offset < size))
2152 return -EIO;
2153
2154 return 0;
2155 }
2156
2157 static int bdrv_check_request(BlockDriverState *bs, int64_t sector_num,
2158 int nb_sectors)
2159 {
2160 return bdrv_check_byte_request(bs, sector_num * BDRV_SECTOR_SIZE,
2161 nb_sectors * BDRV_SECTOR_SIZE);
2162 }
2163
2164 typedef struct RwCo {
2165 BlockDriverState *bs;
2166 int64_t sector_num;
2167 int nb_sectors;
2168 QEMUIOVector *qiov;
2169 bool is_write;
2170 int ret;
2171 BdrvRequestFlags flags;
2172 } RwCo;
2173
2174 static void coroutine_fn bdrv_rw_co_entry(void *opaque)
2175 {
2176 RwCo *rwco = opaque;
2177
2178 if (!rwco->is_write) {
2179 rwco->ret = bdrv_co_do_readv(rwco->bs, rwco->sector_num,
2180 rwco->nb_sectors, rwco->qiov,
2181 rwco->flags);
2182 } else {
2183 rwco->ret = bdrv_co_do_writev(rwco->bs, rwco->sector_num,
2184 rwco->nb_sectors, rwco->qiov,
2185 rwco->flags);
2186 }
2187 }
2188
2189 /*
2190 * Process a vectored synchronous request using coroutines
2191 */
2192 static int bdrv_rwv_co(BlockDriverState *bs, int64_t sector_num,
2193 QEMUIOVector *qiov, bool is_write,
2194 BdrvRequestFlags flags)
2195 {
2196 Coroutine *co;
2197 RwCo rwco = {
2198 .bs = bs,
2199 .sector_num = sector_num,
2200 .nb_sectors = qiov->size >> BDRV_SECTOR_BITS,
2201 .qiov = qiov,
2202 .is_write = is_write,
2203 .ret = NOT_DONE,
2204 .flags = flags,
2205 };
2206 assert((qiov->size & (BDRV_SECTOR_SIZE - 1)) == 0);
2207
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
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 }
2231
2232 /*
2233 * Process a synchronous request using coroutines
2234 */
2235 static int bdrv_rw_co(BlockDriverState *bs, int64_t sector_num, uint8_t *buf,
2236 int nb_sectors, bool is_write, BdrvRequestFlags flags)
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);
2245 return bdrv_rwv_co(bs, sector_num, &qiov, is_write, flags);
2246 }
2247
2248 /* return < 0 if error. See bdrv_write() for the return codes */
2249 int bdrv_read(BlockDriverState *bs, int64_t sector_num,
2250 uint8_t *buf, int nb_sectors)
2251 {
2252 return bdrv_rw_co(bs, sector_num, buf, nb_sectors, false, 0);
2253 }
2254
2255 /* Just like bdrv_read(), but with I/O throttling temporarily disabled */
2256 int 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;
2264 ret = bdrv_read(bs, sector_num, buf, nb_sectors);
2265 bs->io_limits_enabled = enabled;
2266 return ret;
2267 }
2268
2269 /* Return < 0 if error. Important errors are:
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 */
2275 int bdrv_write(BlockDriverState *bs, int64_t sector_num,
2276 const uint8_t *buf, int nb_sectors)
2277 {
2278 return bdrv_rw_co(bs, sector_num, (uint8_t *)buf, nb_sectors, true, 0);
2279 }
2280
2281 int bdrv_writev(BlockDriverState *bs, int64_t sector_num, QEMUIOVector *qiov)
2282 {
2283 return bdrv_rwv_co(bs, sector_num, qiov, true, 0);
2284 }
2285
2286 int 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);
2290 }
2291
2292 int bdrv_pread(BlockDriverState *bs, int64_t offset,
2293 void *buf, int count1)
2294 {
2295 uint8_t tmp_buf[BDRV_SECTOR_SIZE];
2296 int len, nb_sectors, count;
2297 int64_t sector_num;
2298 int ret;
2299
2300 count = count1;
2301 /* first read to align to sector start */
2302 len = (BDRV_SECTOR_SIZE - offset) & (BDRV_SECTOR_SIZE - 1);
2303 if (len > count)
2304 len = count;
2305 sector_num = offset >> BDRV_SECTOR_BITS;
2306 if (len > 0) {
2307 if ((ret = bdrv_read(bs, sector_num, tmp_buf, 1)) < 0)
2308 return ret;
2309 memcpy(buf, tmp_buf + (offset & (BDRV_SECTOR_SIZE - 1)), len);
2310 count -= len;
2311 if (count == 0)
2312 return count1;
2313 sector_num++;
2314 buf += len;
2315 }
2316
2317 /* read the sectors "in place" */
2318 nb_sectors = count >> BDRV_SECTOR_BITS;
2319 if (nb_sectors > 0) {
2320 if ((ret = bdrv_read(bs, sector_num, buf, nb_sectors)) < 0)
2321 return ret;
2322 sector_num += nb_sectors;
2323 len = nb_sectors << BDRV_SECTOR_BITS;
2324 buf += len;
2325 count -= len;
2326 }
2327
2328 /* add data from the last sector */
2329 if (count > 0) {
2330 if ((ret = bdrv_read(bs, sector_num, tmp_buf, 1)) < 0)
2331 return ret;
2332 memcpy(buf, tmp_buf, count);
2333 }
2334 return count1;
2335 }
2336
2337 int bdrv_pwritev(BlockDriverState *bs, int64_t offset, QEMUIOVector *qiov)
2338 {
2339 uint8_t tmp_buf[BDRV_SECTOR_SIZE];
2340 int len, nb_sectors, count;
2341 int64_t sector_num;
2342 int ret;
2343
2344 count = qiov->size;
2345
2346 /* first write to align to sector start */
2347 len = (BDRV_SECTOR_SIZE - offset) & (BDRV_SECTOR_SIZE - 1);
2348 if (len > count)
2349 len = count;
2350 sector_num = offset >> BDRV_SECTOR_BITS;
2351 if (len > 0) {
2352 if ((ret = bdrv_read(bs, sector_num, tmp_buf, 1)) < 0)
2353 return ret;
2354 qemu_iovec_to_buf(qiov, 0, tmp_buf + (offset & (BDRV_SECTOR_SIZE - 1)),
2355 len);
2356 if ((ret = bdrv_write(bs, sector_num, tmp_buf, 1)) < 0)
2357 return ret;
2358 count -= len;
2359 if (count == 0)
2360 return qiov->size;
2361 sector_num++;
2362 }
2363
2364 /* write the sectors "in place" */
2365 nb_sectors = count >> BDRV_SECTOR_BITS;
2366 if (nb_sectors > 0) {
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) {
2375 return ret;
2376 }
2377
2378 sector_num += nb_sectors;
2379 len = nb_sectors << BDRV_SECTOR_BITS;
2380 count -= len;
2381 }
2382
2383 /* add data from the last sector */
2384 if (count > 0) {
2385 if ((ret = bdrv_read(bs, sector_num, tmp_buf, 1)) < 0)
2386 return ret;
2387 qemu_iovec_to_buf(qiov, qiov->size - count, tmp_buf, count);
2388 if ((ret = bdrv_write(bs, sector_num, tmp_buf, 1)) < 0)
2389 return ret;
2390 }
2391 return qiov->size;
2392 }
2393
2394 int 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);
2405 }
2406
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 */
2413 int 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
2423 /* No flush needed for cache modes that already do it */
2424 if (bs->enable_write_cache) {
2425 bdrv_flush(bs);
2426 }
2427
2428 return 0;
2429 }
2430
2431 static int coroutine_fn bdrv_co_do_copy_on_readv(BlockDriverState *bs,
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
2441 BlockDriver *drv = bs->drv;
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 */
2452 bdrv_round_to_clusters(bs, sector_num, nb_sectors,
2453 &cluster_sector_num, &cluster_nb_sectors);
2454
2455 trace_bdrv_co_do_copy_on_readv(bs, sector_num, nb_sectors,
2456 cluster_sector_num, cluster_nb_sectors);
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
2462 ret = drv->bdrv_co_readv(bs, cluster_sector_num, cluster_nb_sectors,
2463 &bounce_qiov);
2464 if (ret < 0) {
2465 goto err;
2466 }
2467
2468 if (drv->bdrv_co_write_zeroes &&
2469 buffer_is_zero(bounce_buffer, iov.iov_len)) {
2470 ret = bdrv_co_do_write_zeroes(bs, cluster_sector_num,
2471 cluster_nb_sectors);
2472 } else {
2473 /* This does not change the data on the disk, it is not necessary
2474 * to flush even in cache=writethrough mode.
2475 */
2476 ret = drv->bdrv_co_writev(bs, cluster_sector_num, cluster_nb_sectors,
2477 &bounce_qiov);
2478 }
2479
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;
2489 qemu_iovec_from_buf(qiov, 0, bounce_buffer + skip_bytes,
2490 nb_sectors * BDRV_SECTOR_SIZE);
2491
2492 err:
2493 qemu_vfree(bounce_buffer);
2494 return ret;
2495 }
2496
2497 /*
2498 * Handle a read request in coroutine context
2499 */
2500 static int coroutine_fn bdrv_co_do_readv(BlockDriverState *bs,
2501 int64_t sector_num, int nb_sectors, QEMUIOVector *qiov,
2502 BdrvRequestFlags flags)
2503 {
2504 BlockDriver *drv = bs->drv;
2505 BdrvTrackedRequest req;
2506 int ret;
2507
2508 if (!drv) {
2509 return -ENOMEDIUM;
2510 }
2511 if (bdrv_check_request(bs, sector_num, nb_sectors)) {
2512 return -EIO;
2513 }
2514
2515 /* throttling disk read I/O */
2516 if (bs->io_limits_enabled) {
2517 bdrv_io_limits_intercept(bs, false, nb_sectors);
2518 }
2519
2520 if (bs->copy_on_read) {
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) {
2528 wait_for_overlapping_requests(bs, sector_num, nb_sectors);
2529 }
2530
2531 tracked_request_begin(&req, bs, sector_num, nb_sectors, false);
2532
2533 if (flags & BDRV_REQ_COPY_ON_READ) {
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) {
2542 ret = bdrv_co_do_copy_on_readv(bs, sector_num, nb_sectors, qiov);
2543 goto out;
2544 }
2545 }
2546
2547 ret = drv->bdrv_co_readv(bs, sector_num, nb_sectors, qiov);
2548
2549 out:
2550 tracked_request_end(&req);
2551
2552 if (flags & BDRV_REQ_COPY_ON_READ) {
2553 bs->copy_on_read_in_flight--;
2554 }
2555
2556 return ret;
2557 }
2558
2559 int coroutine_fn bdrv_co_readv(BlockDriverState *bs, int64_t sector_num,
2560 int nb_sectors, QEMUIOVector *qiov)
2561 {
2562 trace_bdrv_co_readv(bs, sector_num, nb_sectors);
2563
2564 return bdrv_co_do_readv(bs, sector_num, nb_sectors, qiov, 0);
2565 }
2566
2567 int 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);
2574 }
2575
2576 static 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
2584 /* TODO Emulate only part of misaligned requests instead of letting block
2585 * drivers return -ENOTSUP and emulate everything */
2586
2587 /* First try the efficient write zeroes operation */
2588 if (drv->bdrv_co_write_zeroes) {
2589 ret = drv->bdrv_co_write_zeroes(bs, sector_num, nb_sectors);
2590 if (ret != -ENOTSUP) {
2591 return ret;
2592 }
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
2607 /*
2608 * Handle a write request in coroutine context
2609 */
2610 static int coroutine_fn bdrv_co_do_writev(BlockDriverState *bs,
2611 int64_t sector_num, int nb_sectors, QEMUIOVector *qiov,
2612 BdrvRequestFlags flags)
2613 {
2614 BlockDriver *drv = bs->drv;
2615 BdrvTrackedRequest req;
2616 int ret;
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
2628 /* throttling disk write I/O */
2629 if (bs->io_limits_enabled) {
2630 bdrv_io_limits_intercept(bs, true, nb_sectors);
2631 }
2632
2633 if (bs->copy_on_read_in_flight) {
2634 wait_for_overlapping_requests(bs, sector_num, nb_sectors);
2635 }
2636
2637 tracked_request_begin(&req, bs, sector_num, nb_sectors, true);
2638
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) {
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 }
2648
2649 if (ret == 0 && !bs->enable_write_cache) {
2650 ret = bdrv_co_flush(bs);
2651 }
2652
2653 if (bs->dirty_bitmap) {
2654 bdrv_set_dirty(bs, sector_num, nb_sectors);
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
2661 tracked_request_end(&req);
2662
2663 return ret;
2664 }
2665
2666 int 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
2671 return bdrv_co_do_writev(bs, sector_num, nb_sectors, qiov, 0);
2672 }
2673
2674 int 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);
2681 }
2682
2683 /**
2684 * Truncate file to 'offset' bytes (needed only for file protocols)
2685 */
2686 int bdrv_truncate(BlockDriverState *bs, int64_t offset)
2687 {
2688 BlockDriver *drv = bs->drv;
2689 int ret;
2690 if (!drv)
2691 return -ENOMEDIUM;
2692 if (!drv->bdrv_truncate)
2693 return -ENOTSUP;
2694 if (bs->read_only)
2695 return -EACCES;
2696 if (bdrv_in_use(bs))
2697 return -EBUSY;
2698 ret = drv->bdrv_truncate(bs, offset);
2699 if (ret == 0) {
2700 ret = refresh_total_sectors(bs, offset >> BDRV_SECTOR_BITS);
2701 bdrv_dev_resize_cb(bs);
2702 }
2703 return ret;
2704 }
2705
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 */
2710 int64_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
2725 /**
2726 * Length of a file in bytes. Return < 0 if error or unknown.
2727 */
2728 int64_t bdrv_getlength(BlockDriverState *bs)
2729 {
2730 BlockDriver *drv = bs->drv;
2731 if (!drv)
2732 return -ENOMEDIUM;
2733
2734 if (bs->growable || bdrv_dev_has_removable_media(bs)) {
2735 if (drv->bdrv_getlength) {
2736 return drv->bdrv_getlength(bs);
2737 }
2738 }
2739 return bs->total_sectors * BDRV_SECTOR_SIZE;
2740 }
2741
2742 /* return 0 as number of sectors if no device present or error */
2743 void bdrv_get_geometry(BlockDriverState *bs, uint64_t *nb_sectors_ptr)
2744 {
2745 int64_t length;
2746 length = bdrv_getlength(bs);
2747 if (length < 0)
2748 length = 0;
2749 else
2750 length = length >> BDRV_SECTOR_BITS;
2751 *nb_sectors_ptr = length;
2752 }
2753
2754 /* throttling disk io limits */
2755 void 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
2762 void bdrv_set_on_error(BlockDriverState *bs, BlockdevOnError on_read_error,
2763 BlockdevOnError on_write_error)
2764 {
2765 bs->on_read_error = on_read_error;
2766 bs->on_write_error = on_write_error;
2767 }
2768
2769 BlockdevOnError bdrv_get_on_error(BlockDriverState *bs, bool is_read)
2770 {
2771 return is_read ? bs->on_read_error : bs->on_write_error;
2772 }
2773
2774 BlockErrorAction 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 */
2796 void bdrv_error_action(BlockDriverState *bs, BlockErrorAction action,
2797 bool is_read, int error)
2798 {
2799 assert(error >= 0);
2800 bdrv_emit_qmp_error_event(bs, QEVENT_BLOCK_IO_ERROR, action, is_read);
2801 if (action == BDRV_ACTION_STOP) {
2802 vm_stop(RUN_STATE_IO_ERROR);
2803 bdrv_iostatus_set_err(bs, error);
2804 }
2805 }
2806
2807 int bdrv_is_read_only(BlockDriverState *bs)
2808 {
2809 return bs->read_only;
2810 }
2811
2812 int bdrv_is_sg(BlockDriverState *bs)
2813 {
2814 return bs->sg;
2815 }
2816
2817 int bdrv_enable_write_cache(BlockDriverState *bs)
2818 {
2819 return bs->enable_write_cache;
2820 }
2821
2822 void bdrv_set_enable_write_cache(BlockDriverState *bs, bool wce)
2823 {
2824 bs->enable_write_cache = wce;
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 }
2832 }
2833
2834 int bdrv_is_encrypted(BlockDriverState *bs)
2835 {
2836 if (bs->backing_hd && bs->backing_hd->encrypted)
2837 return 1;
2838 return bs->encrypted;
2839 }
2840
2841 int 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
2850 int 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 }
2860 if (!bs->encrypted) {
2861 return -EINVAL;
2862 } else if (!bs->drv || !bs->drv->bdrv_set_key) {
2863 return -ENOMEDIUM;
2864 }
2865 ret = bs->drv->bdrv_set_key(bs, key);
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 */
2871 bdrv_dev_change_media_cb(bs, true);
2872 }
2873 return ret;
2874 }
2875
2876 const char *bdrv_get_format_name(BlockDriverState *bs)
2877 {
2878 return bs->drv ? bs->drv->format_name : NULL;
2879 }
2880
2881 void bdrv_iterate_format(void (*it)(void *opaque, const char *name),
2882 void *opaque)
2883 {
2884 BlockDriver *drv;
2885
2886 QLIST_FOREACH(drv, &bdrv_drivers, list) {
2887 it(opaque, drv->format_name);
2888 }
2889 }
2890
2891 BlockDriverState *bdrv_find(const char *name)
2892 {
2893 BlockDriverState *bs;
2894
2895 QTAILQ_FOREACH(bs, &bdrv_states, list) {
2896 if (!strcmp(name, bs->device_name)) {
2897 return bs;
2898 }
2899 }
2900 return NULL;
2901 }
2902
2903 BlockDriverState *bdrv_next(BlockDriverState *bs)
2904 {
2905 if (!bs) {
2906 return QTAILQ_FIRST(&bdrv_states);
2907 }
2908 return QTAILQ_NEXT(bs, list);
2909 }
2910
2911 void bdrv_iterate(void (*it)(void *opaque, BlockDriverState *bs), void *opaque)
2912 {
2913 BlockDriverState *bs;
2914
2915 QTAILQ_FOREACH(bs, &bdrv_states, list) {
2916 it(opaque, bs);
2917 }
2918 }
2919
2920 const char *bdrv_get_device_name(BlockDriverState *bs)
2921 {
2922 return bs->device_name;
2923 }
2924
2925 int bdrv_get_flags(BlockDriverState *bs)
2926 {
2927 return bs->open_flags;
2928 }
2929
2930 int bdrv_flush_all(void)
2931 {
2932 BlockDriverState *bs;
2933 int result = 0;
2934
2935 QTAILQ_FOREACH(bs, &bdrv_states, list) {
2936 int ret = bdrv_flush(bs);
2937 if (ret < 0 && !result) {
2938 result = ret;
2939 }
2940 }
2941
2942 return result;
2943 }
2944
2945 int bdrv_has_zero_init_1(BlockDriverState *bs)
2946 {
2947 return 1;
2948 }
2949
2950 int bdrv_has_zero_init(BlockDriverState *bs)
2951 {
2952 assert(bs->drv);
2953
2954 if (bs->drv->bdrv_has_zero_init) {
2955 return bs->drv->bdrv_has_zero_init(bs);
2956 }
2957
2958 /* safe default */
2959 return 0;
2960 }
2961
2962 typedef struct BdrvCoIsAllocatedData {
2963 BlockDriverState *bs;
2964 BlockDriverState *base;
2965 int64_t sector_num;
2966 int nb_sectors;
2967 int *pnum;
2968 int ret;
2969 bool done;
2970 } BdrvCoIsAllocatedData;
2971
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 *
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 *
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 *
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.
2986 */
2987 int coroutine_fn bdrv_co_is_allocated(BlockDriverState *bs, int64_t sector_num,
2988 int nb_sectors, int *pnum)
2989 {
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
3002 if (!bs->drv->bdrv_co_is_allocated) {
3003 *pnum = nb_sectors;
3004 return 1;
3005 }
3006
3007 return bs->drv->bdrv_co_is_allocated(bs, sector_num, nb_sectors, pnum);
3008 }
3009
3010 /* Coroutine wrapper for bdrv_is_allocated() */
3011 static 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 */
3026 int bdrv_is_allocated(BlockDriverState *bs, int64_t sector_num, int nb_sectors,
3027 int *pnum)
3028 {
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;
3044 }
3045
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 */
3058 int 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 */
3084 if (n > pnum_inter &&
3085 (intermediate == top ||
3086 sector_num + pnum_inter < intermediate->total_sectors)) {
3087 n = pnum_inter;
3088 }
3089
3090 intermediate = intermediate->backing_hd;
3091 }
3092
3093 *pnum = n;
3094 return 0;
3095 }
3096
3097 /* Coroutine wrapper for bdrv_is_allocated_above() */
3098 static 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 */
3114 int 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
3135 const 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
3145 void bdrv_get_backing_filename(BlockDriverState *bs,
3146 char *filename, int filename_size)
3147 {
3148 pstrcpy(filename, filename_size, bs->backing_file);
3149 }
3150
3151 int bdrv_write_compressed(BlockDriverState *bs, int64_t sector_num,
3152 const uint8_t *buf, int nb_sectors)
3153 {
3154 BlockDriver *drv = bs->drv;
3155 if (!drv)
3156 return -ENOMEDIUM;
3157 if (!drv->bdrv_write_compressed)
3158 return -ENOTSUP;
3159 if (bdrv_check_request(bs, sector_num, nb_sectors))
3160 return -EIO;
3161
3162 assert(!bs->dirty_bitmap);
3163
3164 return drv->bdrv_write_compressed(bs, sector_num, buf, nb_sectors);
3165 }
3166
3167 int bdrv_get_info(BlockDriverState *bs, BlockDriverInfo *bdi)
3168 {
3169 BlockDriver *drv = bs->drv;
3170 if (!drv)
3171 return -ENOMEDIUM;
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
3178 int bdrv_save_vmstate(BlockDriverState *bs, const uint8_t *buf,
3179 int64_t pos, int size)
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
3191 int bdrv_writev_vmstate(BlockDriverState *bs, QEMUIOVector *qiov, int64_t pos)
3192 {
3193 BlockDriver *drv = bs->drv;
3194
3195 if (!drv) {
3196 return -ENOMEDIUM;
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
3203 return -ENOTSUP;
3204 }
3205
3206 int bdrv_load_vmstate(BlockDriverState *bs, uint8_t *buf,
3207 int64_t pos, int size)
3208 {
3209 BlockDriver *drv = bs->drv;
3210 if (!drv)
3211 return -ENOMEDIUM;
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;
3217 }
3218
3219 void bdrv_debug_event(BlockDriverState *bs, BlkDebugEvent event)
3220 {
3221 if (!bs || !bs->drv || !bs->drv->bdrv_debug_event) {
3222 return;
3223 }
3224
3225 bs->drv->bdrv_debug_event(bs, event);
3226 }
3227
3228 int 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
3242 int bdrv_debug_resume(BlockDriverState *bs, const char *tag)
3243 {
3244 while (bs && bs->drv && !bs->drv->bdrv_debug_resume) {
3245 bs = bs->file;
3246 }
3247
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
3255 bool 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;
3266 }
3267
3268 int bdrv_is_snapshot(BlockDriverState *bs)
3269 {
3270 return !!(bs->open_flags & BDRV_O_SNAPSHOT);
3271 }
3272
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. */
3277 BlockDriverState *bdrv_find_backing_image(BlockDriverState *bs,
3278 const char *backing_file)
3279 {
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) {
3288 return NULL;
3289 }
3290
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 }
3306 } else {
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 }
3330 }
3331 }
3332
3333 g_free(filename_full);
3334 g_free(backing_file_full);
3335 g_free(filename_tmp);
3336 return retval;
3337 }
3338
3339 int 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
3352 BlockDriverState *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
3368 /**************************************************************/
3369 /* async I/Os */
3370
3371 BlockDriverAIOCB *bdrv_aio_readv(BlockDriverState *bs, int64_t sector_num,
3372 QEMUIOVector *qiov, int nb_sectors,
3373 BlockDriverCompletionFunc *cb, void *opaque)
3374 {
3375 trace_bdrv_aio_readv(bs, sector_num, nb_sectors, opaque);
3376
3377 return bdrv_co_aio_rw_vector(bs, sector_num, qiov, nb_sectors,
3378 cb, opaque, false);
3379 }
3380
3381 BlockDriverAIOCB *bdrv_aio_writev(BlockDriverState *bs, int64_t sector_num,
3382 QEMUIOVector *qiov, int nb_sectors,
3383 BlockDriverCompletionFunc *cb, void *opaque)
3384 {
3385 trace_bdrv_aio_writev(bs, sector_num, nb_sectors, opaque);
3386
3387 return bdrv_co_aio_rw_vector(bs, sector_num, qiov, nb_sectors,
3388 cb, opaque, true);
3389 }
3390
3391
3392 typedef 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;
3400 } callbacks[];
3401 } MultiwriteCB;
3402
3403 static 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);
3409 if (mcb->callbacks[i].free_qiov) {
3410 qemu_iovec_destroy(mcb->callbacks[i].free_qiov);
3411 }
3412 g_free(mcb->callbacks[i].free_qiov);
3413 }
3414 }
3415
3416 static void multiwrite_cb(void *opaque, int ret)
3417 {
3418 MultiwriteCB *mcb = opaque;
3419
3420 trace_multiwrite_cb(mcb, ret);
3421
3422 if (ret < 0 && !mcb->error) {
3423 mcb->error = ret;
3424 }
3425
3426 mcb->num_requests--;
3427 if (mcb->num_requests == 0) {
3428 multiwrite_user_cb(mcb);
3429 g_free(mcb);
3430 }
3431 }
3432
3433 static int multiwrite_req_compare(const void *a, const void *b)
3434 {
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 }
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 */
3454 static 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
3469 // Handle exactly sequential writes and overlapping writes.
3470 if (reqs[i].sector <= oldreq_last) {
3471 merge = 1;
3472 }
3473
3474 if (reqs[outidx].qiov->niov + reqs[i].qiov->niov + 1 > IOV_MAX) {
3475 merge = 0;
3476 }
3477
3478 if (merge) {
3479 size_t size;
3480 QEMUIOVector *qiov = g_malloc0(sizeof(*qiov));
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;
3487 qemu_iovec_concat(qiov, reqs[outidx].qiov, 0, size);
3488
3489 // We should need to add any zeros between the two requests
3490 assert (reqs[i].sector <= oldreq_last);
3491
3492 // Add the second request
3493 qemu_iovec_concat(qiov, reqs[i].qiov, 0, reqs[i].qiov->size);
3494
3495 reqs[outidx].nb_sectors = qiov->size >> 9;
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 */
3524 int bdrv_aio_multiwrite(BlockDriverState *bs, BlockRequest *reqs, int num_reqs)
3525 {
3526 MultiwriteCB *mcb;
3527 int i;
3528
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
3537 if (num_reqs == 0) {
3538 return 0;
3539 }
3540
3541 // Create MultiwriteCB structure
3542 mcb = g_malloc0(sizeof(*mcb) + num_reqs * sizeof(*mcb->callbacks));
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
3554 trace_bdrv_aio_multiwrite(mcb, mcb->num_callbacks, num_reqs);
3555
3556 /* Run the aio requests. */
3557 mcb->num_requests = num_reqs;
3558 for (i = 0; i < num_reqs; i++) {
3559 bdrv_aio_writev(bs, reqs[i].sector, reqs[i].qiov,
3560 reqs[i].nb_sectors, multiwrite_cb, mcb);
3561 }
3562
3563 return 0;
3564 }
3565
3566 void bdrv_aio_cancel(BlockDriverAIOCB *acb)
3567 {
3568 acb->aiocb_info->cancel(acb);
3569 }
3570
3571 /* block I/O throttling */
3572 static 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;
3576 uint64_t extension;
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;
3595 bytes_base = bs->slice_submitted.bytes[is_write];
3596 if (bs->io_limits.bps[BLOCK_IO_LIMIT_TOTAL]) {
3597 bytes_base += bs->slice_submitted.bytes[!is_write];
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 */
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;
3628 if (wait) {
3629 *wait = wait_time * NANOSECONDS_PER_SECOND;
3630 }
3631
3632 return true;
3633 }
3634
3635 static 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;
3657 ios_base = bs->slice_submitted.ios[is_write];
3658 if (bs->io_limits.iops[BLOCK_IO_LIMIT_TOTAL]) {
3659 ios_base += bs->slice_submitted.ios[!is_write];
3660 }
3661
3662 if (ios_base + 1 <= ios_limit) {
3663 if (wait) {
3664 *wait = 0;
3665 }
3666
3667 return false;
3668 }
3669
3670 /* Calc approx time to dispatch, in seconds */
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
3678 /* Exceeded current slice, extend it by another slice time */
3679 bs->slice_end += BLOCK_IO_SLICE_TIME;
3680 if (wait) {
3681 *wait = wait_time * NANOSECONDS_PER_SECOND;
3682 }
3683
3684 return true;
3685 }
3686
3687 static 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);
3696 if (now > bs->slice_end) {
3697 bs->slice_start = now;
3698 bs->slice_end = now + BLOCK_IO_SLICE_TIME;
3699 memset(&bs->slice_submitted, 0, sizeof(bs->slice_submitted));
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
3727 bs->slice_submitted.bytes[is_write] += (int64_t)nb_sectors *
3728 BDRV_SECTOR_SIZE;
3729 bs->slice_submitted.ios[is_write]++;
3730
3731 return false;
3732 }
3733
3734 /**************************************************************/
3735 /* async block device emulation */
3736
3737 typedef 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
3747 static void bdrv_aio_cancel_em(BlockDriverAIOCB *blockacb)
3748 {
3749 BlockDriverAIOCBSync *acb =
3750 container_of(blockacb, BlockDriverAIOCBSync, common);
3751 qemu_bh_delete(acb->bh);
3752 acb->bh = NULL;
3753 qemu_aio_release(acb);
3754 }
3755
3756 static const AIOCBInfo bdrv_em_aiocb_info = {
3757 .aiocb_size = sizeof(BlockDriverAIOCBSync),
3758 .cancel = bdrv_aio_cancel_em,
3759 };
3760
3761 static void bdrv_aio_bh_cb(void *opaque)
3762 {
3763 BlockDriverAIOCBSync *acb = opaque;
3764
3765 if (!acb->is_write)
3766 qemu_iovec_from_buf(acb->qiov, 0, acb->bounce, acb->qiov->size);
3767 qemu_vfree(acb->bounce);
3768 acb->common.cb(acb->common.opaque, acb->ret);
3769 qemu_bh_delete(acb->bh);
3770 acb->bh = NULL;
3771 qemu_aio_release(acb);
3772 }
3773
3774 static 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
3782 {
3783 BlockDriverAIOCBSync *acb;
3784
3785 acb = qemu_aio_get(&bdrv_em_aiocb_info, bs, cb, opaque);
3786 acb->is_write = is_write;
3787 acb->qiov = qiov;
3788 acb->bounce = qemu_blockalign(bs, qiov->size);
3789 acb->bh = qemu_bh_new(bdrv_aio_bh_cb, acb);
3790
3791 if (is_write) {
3792 qemu_iovec_to_buf(acb->qiov, 0, acb->bounce, qiov->size);
3793 acb->ret = bs->drv->bdrv_write(bs, sector_num, acb->bounce, nb_sectors);
3794 } else {
3795 acb->ret = bs->drv->bdrv_read(bs, sector_num, acb->bounce, nb_sectors);
3796 }
3797
3798 qemu_bh_schedule(acb->bh);
3799
3800 return &acb->common;
3801 }
3802
3803 static BlockDriverAIOCB *bdrv_aio_readv_em(BlockDriverState *bs,
3804 int64_t sector_num, QEMUIOVector *qiov, int nb_sectors,
3805 BlockDriverCompletionFunc *cb, void *opaque)
3806 {
3807 return bdrv_aio_rw_vector(bs, sector_num, qiov, nb_sectors, cb, opaque, 0);
3808 }
3809
3810 static 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);
3815 }
3816
3817
3818 typedef struct BlockDriverAIOCBCoroutine {
3819 BlockDriverAIOCB common;
3820 BlockRequest req;
3821 bool is_write;
3822 bool *done;
3823 QEMUBH* bh;
3824 } BlockDriverAIOCBCoroutine;
3825
3826 static void bdrv_aio_co_cancel_em(BlockDriverAIOCB *blockacb)
3827 {
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 }
3836 }
3837
3838 static const AIOCBInfo bdrv_em_co_aiocb_info = {
3839 .aiocb_size = sizeof(BlockDriverAIOCBCoroutine),
3840 .cancel = bdrv_aio_co_cancel_em,
3841 };
3842
3843 static void bdrv_co_em_bh(void *opaque)
3844 {
3845 BlockDriverAIOCBCoroutine *acb = opaque;
3846
3847 acb->common.cb(acb->common.opaque, acb->req.error);
3848
3849 if (acb->done) {
3850 *acb->done = true;
3851 }
3852
3853 qemu_bh_delete(acb->bh);
3854 qemu_aio_release(acb);
3855 }
3856
3857 /* Invoke bdrv_co_do_readv/bdrv_co_do_writev */
3858 static 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,
3865 acb->req.nb_sectors, acb->req.qiov, 0);
3866 } else {
3867 acb->req.error = bdrv_co_do_writev(bs, acb->req.sector,
3868 acb->req.nb_sectors, acb->req.qiov, 0);
3869 }
3870
3871 acb->bh = qemu_bh_new(bdrv_co_em_bh, acb);
3872 qemu_bh_schedule(acb->bh);
3873 }
3874
3875 static 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,
3881 bool is_write)
3882 {
3883 Coroutine *co;
3884 BlockDriverAIOCBCoroutine *acb;
3885
3886 acb = qemu_aio_get(&bdrv_em_co_aiocb_info, bs, cb, opaque);
3887 acb->req.sector = sector_num;
3888 acb->req.nb_sectors = nb_sectors;
3889 acb->req.qiov = qiov;
3890 acb->is_write = is_write;
3891 acb->done = NULL;
3892
3893 co = qemu_coroutine_create(bdrv_co_do_rw);
3894 qemu_coroutine_enter(co, acb);
3895
3896 return &acb->common;
3897 }
3898
3899 static void coroutine_fn bdrv_aio_flush_co_entry(void *opaque)
3900 {
3901 BlockDriverAIOCBCoroutine *acb = opaque;
3902 BlockDriverState *bs = acb->common.bs;
3903
3904 acb->req.error = bdrv_co_flush(bs);
3905 acb->bh = qemu_bh_new(bdrv_co_em_bh, acb);
3906 qemu_bh_schedule(acb->bh);
3907 }
3908
3909 BlockDriverAIOCB *bdrv_aio_flush(BlockDriverState *bs,
3910 BlockDriverCompletionFunc *cb, void *opaque)
3911 {
3912 trace_bdrv_aio_flush(bs, opaque);
3913
3914 Coroutine *co;
3915 BlockDriverAIOCBCoroutine *acb;
3916
3917 acb = qemu_aio_get(&bdrv_em_co_aiocb_info, bs, cb, opaque);
3918 acb->done = NULL;
3919
3920 co = qemu_coroutine_create(bdrv_aio_flush_co_entry);
3921 qemu_coroutine_enter(co, acb);
3922
3923 return &acb->common;
3924 }
3925
3926 static 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
3936 BlockDriverAIOCB *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
3945 acb = qemu_aio_get(&bdrv_em_co_aiocb_info, bs, cb, opaque);
3946 acb->req.sector = sector_num;
3947 acb->req.nb_sectors = nb_sectors;
3948 acb->done = NULL;
3949 co = qemu_coroutine_create(bdrv_aio_discard_co_entry);
3950 qemu_coroutine_enter(co, acb);
3951
3952 return &acb->common;
3953 }
3954
3955 void bdrv_init(void)
3956 {
3957 module_call_init(MODULE_INIT_BLOCK);
3958 }
3959
3960 void bdrv_init_with_whitelist(void)
3961 {
3962 use_bdrv_whitelist = 1;
3963 bdrv_init();
3964 }
3965
3966 void *qemu_aio_get(const AIOCBInfo *aiocb_info, BlockDriverState *bs,
3967 BlockDriverCompletionFunc *cb, void *opaque)
3968 {
3969 BlockDriverAIOCB *acb;
3970
3971 acb = g_slice_alloc(aiocb_info->aiocb_size);
3972 acb->aiocb_info = aiocb_info;
3973 acb->bs = bs;
3974 acb->cb = cb;
3975 acb->opaque = opaque;
3976 return acb;
3977 }
3978
3979 void qemu_aio_release(void *p)
3980 {
3981 BlockDriverAIOCB *acb = p;
3982 g_slice_free1(acb->aiocb_info->aiocb_size, acb);
3983 }
3984
3985 /**************************************************************/
3986 /* Coroutine block device emulation */
3987
3988 typedef struct CoroutineIOCompletion {
3989 Coroutine *coroutine;
3990 int ret;
3991 } CoroutineIOCompletion;
3992
3993 static 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
4001 static 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) {
4011 acb = bs->drv->bdrv_aio_writev(bs, sector_num, iov, nb_sectors,
4012 bdrv_co_io_em_complete, &co);
4013 } else {
4014 acb = bs->drv->bdrv_aio_readv(bs, sector_num, iov, nb_sectors,
4015 bdrv_co_io_em_complete, &co);
4016 }
4017
4018 trace_bdrv_co_io_em(bs, sector_num, nb_sectors, is_write, acb);
4019 if (!acb) {
4020 return -EIO;
4021 }
4022 qemu_coroutine_yield();
4023
4024 return co.ret;
4025 }
4026
4027 static 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
4034 static 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
4041 static void coroutine_fn bdrv_flush_co_entry(void *opaque)
4042 {
4043 RwCo *rwco = opaque;
4044
4045 rwco->ret = bdrv_co_flush(rwco->bs);
4046 }
4047
4048 int coroutine_fn bdrv_co_flush(BlockDriverState *bs)
4049 {
4050 int ret;
4051
4052 if (!bs || !bdrv_is_inserted(bs) || bdrv_is_read_only(bs)) {
4053 return 0;
4054 }
4055
4056 /* Write back cached data to the OS even with cache=unsafe */
4057 BLKDBG_EVENT(bs->file, BLKDBG_FLUSH_TO_OS);
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
4065 /* But don't actually force it to the disk with cache=unsafe */
4066 if (bs->open_flags & BDRV_O_NO_FLUSH) {
4067 goto flush_parent;
4068 }
4069
4070 BLKDBG_EVENT(bs->file, BLKDBG_FLUSH_TO_DISK);
4071 if (bs->drv->bdrv_co_flush_to_disk) {
4072 ret = bs->drv->bdrv_co_flush_to_disk(bs);
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) {
4081 ret = -EIO;
4082 } else {
4083 qemu_coroutine_yield();
4084 ret = co.ret;
4085 }
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 */
4098 ret = 0;
4099 }
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 */
4107 flush_parent:
4108 return bdrv_co_flush(bs->file);
4109 }
4110
4111 void 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
4118 void 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
4127 void 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
4136 int bdrv_flush(BlockDriverState *bs)
4137 {
4138 Coroutine *co;
4139 RwCo rwco = {
4140 .bs = bs,
4141 .ret = NOT_DONE,
4142 };
4143
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 }
4153 }
4154
4155 return rwco.ret;
4156 }
4157
4158 static 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
4165 int 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;
4174 }
4175
4176 if (bs->dirty_bitmap) {
4177 bdrv_reset_dirty(bs, sector_num, nb_sectors);
4178 }
4179
4180 /* Do nothing if disabled. */
4181 if (!(bs->open_flags & BDRV_O_UNMAP)) {
4182 return 0;
4183 }
4184
4185 if (bs->drv->bdrv_co_discard) {
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 }
4201 } else {
4202 return 0;
4203 }
4204 }
4205
4206 int 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
4230 /**************************************************************/
4231 /* removable device support */
4232
4233 /**
4234 * Return TRUE if the media is present
4235 */
4236 int bdrv_is_inserted(BlockDriverState *bs)
4237 {
4238 BlockDriver *drv = bs->drv;
4239
4240 if (!drv)
4241 return 0;
4242 if (!drv->bdrv_is_inserted)
4243 return 1;
4244 return drv->bdrv_is_inserted(bs);
4245 }
4246
4247 /**
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.
4250 */
4251 int bdrv_media_changed(BlockDriverState *bs)
4252 {
4253 BlockDriver *drv = bs->drv;
4254
4255 if (drv && drv->bdrv_media_changed) {
4256 return drv->bdrv_media_changed(bs);
4257 }
4258 return -ENOTSUP;
4259 }
4260
4261 /**
4262 * If eject_flag is TRUE, eject the media. Otherwise, close the tray
4263 */
4264 void bdrv_eject(BlockDriverState *bs, bool eject_flag)
4265 {
4266 BlockDriver *drv = bs->drv;
4267
4268 if (drv && drv->bdrv_eject) {
4269 drv->bdrv_eject(bs, eject_flag);
4270 }
4271
4272 if (bs->device_name[0] != '\0') {
4273 bdrv_emit_qmp_eject_event(bs, eject_flag);
4274 }
4275 }
4276
4277 /**
4278 * Lock or unlock the media (if it is locked, the user won't be able
4279 * to eject it manually).
4280 */
4281 void bdrv_lock_medium(BlockDriverState *bs, bool locked)
4282 {
4283 BlockDriver *drv = bs->drv;
4284
4285 trace_bdrv_lock_medium(bs, locked);
4286
4287 if (drv && drv->bdrv_lock_medium) {
4288 drv->bdrv_lock_medium(bs, locked);
4289 }
4290 }
4291
4292 /* needed for generic scsi interface */
4293
4294 int 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 }
4302
4303 BlockDriverAIOCB *bdrv_aio_ioctl(BlockDriverState *bs,
4304 unsigned long int req, void *buf,
4305 BlockDriverCompletionFunc *cb, void *opaque)
4306 {
4307 BlockDriver *drv = bs->drv;
4308
4309 if (drv && drv->bdrv_aio_ioctl)
4310 return drv->bdrv_aio_ioctl(bs, req, buf, cb, opaque);
4311 return NULL;
4312 }
4313
4314 void bdrv_set_buffer_alignment(BlockDriverState *bs, int align)
4315 {
4316 bs->buffer_alignment = align;
4317 }
4318
4319 void *qemu_blockalign(BlockDriverState *bs, size_t size)
4320 {
4321 return qemu_memalign((bs && bs->buffer_alignment) ? bs->buffer_alignment : 512, size);
4322 }
4323
4324 /*
4325 * Check if all memory in this vector is sector aligned.
4326 */
4327 bool 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
4340 void bdrv_set_dirty_tracking(BlockDriverState *bs, int granularity)
4341 {
4342 int64_t bitmap_size;
4343
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);
4351 } else {
4352 if (bs->dirty_bitmap) {
4353 hbitmap_free(bs->dirty_bitmap);
4354 bs->dirty_bitmap = NULL;
4355 }
4356 }
4357 }
4358
4359 int bdrv_get_dirty(BlockDriverState *bs, int64_t sector)
4360 {
4361 if (bs->dirty_bitmap) {
4362 return hbitmap_get(bs->dirty_bitmap, sector);
4363 } else {
4364 return 0;
4365 }
4366 }
4367
4368 void bdrv_dirty_iter_init(BlockDriverState *bs, HBitmapIter *hbi)
4369 {
4370 hbitmap_iter_init(hbi, bs->dirty_bitmap, 0);
4371 }
4372
4373 void bdrv_set_dirty(BlockDriverState *bs, int64_t cur_sector,
4374 int nr_sectors)
4375 {
4376 hbitmap_set(bs->dirty_bitmap, cur_sector, nr_sectors);
4377 }
4378
4379 void bdrv_reset_dirty(BlockDriverState *bs, int64_t cur_sector,
4380 int nr_sectors)
4381 {
4382 hbitmap_reset(bs->dirty_bitmap, cur_sector, nr_sectors);
4383 }
4384
4385 int64_t bdrv_get_dirty_count(BlockDriverState *bs)
4386 {
4387 if (bs->dirty_bitmap) {
4388 return hbitmap_count(bs->dirty_bitmap);
4389 } else {
4390 return 0;
4391 }
4392 }
4393
4394 void 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
4400 int bdrv_in_use(BlockDriverState *bs)
4401 {
4402 return bs->in_use;
4403 }
4404
4405 void bdrv_iostatus_enable(BlockDriverState *bs)
4406 {
4407 bs->iostatus_enabled = true;
4408 bs->iostatus = BLOCK_DEVICE_IO_STATUS_OK;
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 */
4413 bool bdrv_iostatus_is_enabled(const BlockDriverState *bs)
4414 {
4415 return (bs->iostatus_enabled &&
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));
4419 }
4420
4421 void bdrv_iostatus_disable(BlockDriverState *bs)
4422 {
4423 bs->iostatus_enabled = false;
4424 }
4425
4426 void bdrv_iostatus_reset(BlockDriverState *bs)
4427 {
4428 if (bdrv_iostatus_is_enabled(bs)) {
4429 bs->iostatus = BLOCK_DEVICE_IO_STATUS_OK;
4430 if (bs->job) {
4431 block_job_iostatus_reset(bs->job);
4432 }
4433 }
4434 }
4435
4436 void bdrv_iostatus_set_err(BlockDriverState *bs, int error)
4437 {
4438 assert(bdrv_iostatus_is_enabled(bs));
4439 if (bs->iostatus == BLOCK_DEVICE_IO_STATUS_OK) {
4440 bs->iostatus = error == ENOSPC ? BLOCK_DEVICE_IO_STATUS_NOSPACE :
4441 BLOCK_DEVICE_IO_STATUS_FAILED;
4442 }
4443 }
4444
4445 void
4446 bdrv_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;
4452 cookie->start_time_ns = get_clock();
4453 cookie->type = type;
4454 }
4455
4456 void
4457 bdrv_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]++;
4463 bs->total_time_ns[cookie->type] += get_clock() - cookie->start_time_ns;
4464 }
4465
4466 void bdrv_img_create(const char *filename, const char *fmt,
4467 const char *base_filename, const char *base_fmt,
4468 char *options, uint64_t img_size, int flags,
4469 Error **errp, bool quiet)
4470 {
4471 QEMUOptionParameter *param = NULL, *create_options = NULL;
4472 QEMUOptionParameter *backing_fmt, *backing_file, *size;
4473 BlockDriverState *bs = NULL;
4474 BlockDriver *drv, *proto_drv;
4475 BlockDriver *backing_drv = NULL;
4476 int ret = 0;
4477
4478 /* Find driver and parse its options */
4479 drv = bdrv_find_format(fmt);
4480 if (!drv) {
4481 error_setg(errp, "Unknown file format '%s'", fmt);
4482 return;
4483 }
4484
4485 proto_drv = bdrv_find_protocol(filename, true);
4486 if (!proto_drv) {
4487 error_setg(errp, "Unknown protocol '%s'", filename);
4488 return;
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) {
4505 error_setg(errp, "Invalid options for file format '%s'.", fmt);
4506 goto out;
4507 }
4508 }
4509
4510 if (base_filename) {
4511 if (set_option_parameter(param, BLOCK_OPT_BACKING_FILE,
4512 base_filename)) {
4513 error_setg(errp, "Backing file not supported for file format '%s'",
4514 fmt);
4515 goto out;
4516 }
4517 }
4518
4519 if (base_fmt) {
4520 if (set_option_parameter(param, BLOCK_OPT_BACKING_FMT, base_fmt)) {
4521 error_setg(errp, "Backing file format not supported for file "
4522 "format '%s'", fmt);
4523 goto out;
4524 }
4525 }
4526
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)) {
4530 error_setg(errp, "Error: Trying to create an image with the "
4531 "same filename as the backing file");
4532 goto out;
4533 }
4534 }
4535
4536 backing_fmt = get_option_parameter(param, BLOCK_OPT_BACKING_FMT);
4537 if (backing_fmt && backing_fmt->value.s) {
4538 backing_drv = bdrv_find_format(backing_fmt->value.s);
4539 if (!backing_drv) {
4540 error_setg(errp, "Unknown backing file format '%s'",
4541 backing_fmt->value.s);
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
4548 size = get_option_parameter(param, BLOCK_OPT_SIZE);
4549 if (size && size->value.n == -1) {
4550 if (backing_file && backing_file->value.s) {
4551 uint64_t size;
4552 char buf[32];
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);
4558
4559 bs = bdrv_new("");
4560
4561 ret = bdrv_open(bs, backing_file->value.s, NULL, back_flags,
4562 backing_drv);
4563 if (ret < 0) {
4564 error_setg_errno(errp, -ret, "Could not open '%s'",
4565 backing_file->value.s);
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 {
4574 error_setg(errp, "Image creation needs a size parameter");
4575 goto out;
4576 }
4577 }
4578
4579 if (!quiet) {
4580 printf("Formatting '%s', fmt=%s ", filename, fmt);
4581 print_option_parameters(param);
4582 puts("");
4583 }
4584 ret = bdrv_create(drv, filename, param);
4585 if (ret < 0) {
4586 if (ret == -ENOTSUP) {
4587 error_setg(errp,"Formatting or formatting option not supported for "
4588 "file format '%s'", fmt);
4589 } else if (ret == -EFBIG) {
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);
4596 } else {
4597 error_setg(errp, "%s: error while creating %s: %s", filename, fmt,
4598 strerror(-ret));
4599 }
4600 }
4601
4602 out:
4603 free_option_parameters(create_options);
4604 free_option_parameters(param);
4605
4606 if (bs) {
4607 bdrv_delete(bs);
4608 }
4609 }
4610
4611 AioContext *bdrv_get_aio_context(BlockDriverState *bs)
4612 {
4613 /* Currently BlockDriverState always uses the main loop AioContext */
4614 return qemu_get_aio_context();
4615 }
4616
4617 void bdrv_add_before_write_notifier(BlockDriverState *bs,
4618 NotifierWithReturn *notifier)
4619 {
4620 notifier_with_return_list_add(&bs->before_write_notifiers, notifier);
4621 }