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