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