]> git.proxmox.com Git - mirror_qemu.git/blame - block.c
qapi: Convert query-block
[mirror_qemu.git] / block.c
CommitLineData
fc01f7e7
FB
1/*
2 * QEMU System Emulator block driver
5fafdf24 3 *
fc01f7e7 4 * Copyright (c) 2003 Fabrice Bellard
5fafdf24 5 *
fc01f7e7
FB
6 * Permission is hereby granted, free of charge, to any person obtaining a copy
7 * of this software and associated documentation files (the "Software"), to deal
8 * in the Software without restriction, including without limitation the rights
9 * to use, copy, modify, merge, publish, distribute, sublicense, and/or sell
10 * copies of the Software, and to permit persons to whom the Software is
11 * furnished to do so, subject to the following conditions:
12 *
13 * The above copyright notice and this permission notice shall be included in
14 * all copies or substantial portions of the Software.
15 *
16 * THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
17 * IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
18 * FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL
19 * THE AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
20 * LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,
21 * OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN
22 * THE SOFTWARE.
23 */
3990d09a 24#include "config-host.h"
faf07963 25#include "qemu-common.h"
6d519a5f 26#include "trace.h"
376253ec 27#include "monitor.h"
ea2384d3 28#include "block_int.h"
5efa9d5a 29#include "module.h"
d15e5465 30#include "qemu-objects.h"
68485420 31#include "qemu-coroutine.h"
b2023818 32#include "qmp-commands.h"
fc01f7e7 33
71e72a19 34#ifdef CONFIG_BSD
7674e7bf
FB
35#include <sys/types.h>
36#include <sys/stat.h>
37#include <sys/ioctl.h>
72cf2d4f 38#include <sys/queue.h>
c5e97233 39#ifndef __DragonFly__
7674e7bf
FB
40#include <sys/disk.h>
41#endif
c5e97233 42#endif
7674e7bf 43
49dc768d
AL
44#ifdef _WIN32
45#include <windows.h>
46#endif
47
1c9805a3
SH
48#define NOT_DONE 0x7fffffff /* used while emulated sync operation in progress */
49
7d4b4ba5 50static void bdrv_dev_change_media_cb(BlockDriverState *bs, bool load);
f141eafe
AL
51static BlockDriverAIOCB *bdrv_aio_readv_em(BlockDriverState *bs,
52 int64_t sector_num, QEMUIOVector *qiov, int nb_sectors,
c87c0672 53 BlockDriverCompletionFunc *cb, void *opaque);
f141eafe
AL
54static BlockDriverAIOCB *bdrv_aio_writev_em(BlockDriverState *bs,
55 int64_t sector_num, QEMUIOVector *qiov, int nb_sectors,
ce1a14dc 56 BlockDriverCompletionFunc *cb, void *opaque);
f9f05dc5
KW
57static int coroutine_fn bdrv_co_readv_em(BlockDriverState *bs,
58 int64_t sector_num, int nb_sectors,
59 QEMUIOVector *iov);
60static int coroutine_fn bdrv_co_writev_em(BlockDriverState *bs,
61 int64_t sector_num, int nb_sectors,
62 QEMUIOVector *iov);
c5fbe571
SH
63static int coroutine_fn bdrv_co_do_readv(BlockDriverState *bs,
64 int64_t sector_num, int nb_sectors, QEMUIOVector *qiov);
1c9805a3
SH
65static int coroutine_fn bdrv_co_do_writev(BlockDriverState *bs,
66 int64_t sector_num, int nb_sectors, QEMUIOVector *qiov);
b2a61371
SH
67static BlockDriverAIOCB *bdrv_co_aio_rw_vector(BlockDriverState *bs,
68 int64_t sector_num,
69 QEMUIOVector *qiov,
70 int nb_sectors,
71 BlockDriverCompletionFunc *cb,
72 void *opaque,
8c5873d6 73 bool is_write);
b2a61371 74static void coroutine_fn bdrv_co_do_rw(void *opaque);
ec530c81 75
1b7bdbc1
SH
76static QTAILQ_HEAD(, BlockDriverState) bdrv_states =
77 QTAILQ_HEAD_INITIALIZER(bdrv_states);
7ee930d0 78
8a22f02a
SH
79static QLIST_HEAD(, BlockDriver) bdrv_drivers =
80 QLIST_HEAD_INITIALIZER(bdrv_drivers);
ea2384d3 81
f9092b10
MA
82/* The device to use for VM snapshots */
83static BlockDriverState *bs_snapshots;
84
eb852011
MA
85/* If non-zero, use only whitelisted block drivers */
86static int use_bdrv_whitelist;
87
9e0b22f4
SH
88#ifdef _WIN32
89static int is_windows_drive_prefix(const char *filename)
90{
91 return (((filename[0] >= 'a' && filename[0] <= 'z') ||
92 (filename[0] >= 'A' && filename[0] <= 'Z')) &&
93 filename[1] == ':');
94}
95
96int is_windows_drive(const char *filename)
97{
98 if (is_windows_drive_prefix(filename) &&
99 filename[2] == '\0')
100 return 1;
101 if (strstart(filename, "\\\\.\\", NULL) ||
102 strstart(filename, "//./", NULL))
103 return 1;
104 return 0;
105}
106#endif
107
108/* check if the path starts with "<protocol>:" */
109static int path_has_protocol(const char *path)
110{
111#ifdef _WIN32
112 if (is_windows_drive(path) ||
113 is_windows_drive_prefix(path)) {
114 return 0;
115 }
116#endif
117
118 return strchr(path, ':') != NULL;
119}
120
83f64091 121int path_is_absolute(const char *path)
3b0d4f61 122{
83f64091 123 const char *p;
21664424
FB
124#ifdef _WIN32
125 /* specific case for names like: "\\.\d:" */
126 if (*path == '/' || *path == '\\')
127 return 1;
128#endif
83f64091
FB
129 p = strchr(path, ':');
130 if (p)
131 p++;
132 else
133 p = path;
3b9f94e1
FB
134#ifdef _WIN32
135 return (*p == '/' || *p == '\\');
136#else
137 return (*p == '/');
138#endif
3b0d4f61
FB
139}
140
83f64091
FB
141/* if filename is absolute, just copy it to dest. Otherwise, build a
142 path to it by considering it is relative to base_path. URL are
143 supported. */
144void path_combine(char *dest, int dest_size,
145 const char *base_path,
146 const char *filename)
3b0d4f61 147{
83f64091
FB
148 const char *p, *p1;
149 int len;
150
151 if (dest_size <= 0)
152 return;
153 if (path_is_absolute(filename)) {
154 pstrcpy(dest, dest_size, filename);
155 } else {
156 p = strchr(base_path, ':');
157 if (p)
158 p++;
159 else
160 p = base_path;
3b9f94e1
FB
161 p1 = strrchr(base_path, '/');
162#ifdef _WIN32
163 {
164 const char *p2;
165 p2 = strrchr(base_path, '\\');
166 if (!p1 || p2 > p1)
167 p1 = p2;
168 }
169#endif
83f64091
FB
170 if (p1)
171 p1++;
172 else
173 p1 = base_path;
174 if (p1 > p)
175 p = p1;
176 len = p - base_path;
177 if (len > dest_size - 1)
178 len = dest_size - 1;
179 memcpy(dest, base_path, len);
180 dest[len] = '\0';
181 pstrcat(dest, dest_size, filename);
3b0d4f61 182 }
3b0d4f61
FB
183}
184
5efa9d5a 185void bdrv_register(BlockDriver *bdrv)
ea2384d3 186{
8c5873d6
SH
187 /* Block drivers without coroutine functions need emulation */
188 if (!bdrv->bdrv_co_readv) {
f9f05dc5
KW
189 bdrv->bdrv_co_readv = bdrv_co_readv_em;
190 bdrv->bdrv_co_writev = bdrv_co_writev_em;
191
f8c35c1d
SH
192 /* bdrv_co_readv_em()/brdv_co_writev_em() work in terms of aio, so if
193 * the block driver lacks aio we need to emulate that too.
194 */
f9f05dc5
KW
195 if (!bdrv->bdrv_aio_readv) {
196 /* add AIO emulation layer */
197 bdrv->bdrv_aio_readv = bdrv_aio_readv_em;
198 bdrv->bdrv_aio_writev = bdrv_aio_writev_em;
f9f05dc5 199 }
83f64091 200 }
b2e12bc6 201
8a22f02a 202 QLIST_INSERT_HEAD(&bdrv_drivers, bdrv, list);
ea2384d3 203}
b338082b
FB
204
205/* create a new block device (by default it is empty) */
206BlockDriverState *bdrv_new(const char *device_name)
207{
1b7bdbc1 208 BlockDriverState *bs;
b338082b 209
7267c094 210 bs = g_malloc0(sizeof(BlockDriverState));
b338082b 211 pstrcpy(bs->device_name, sizeof(bs->device_name), device_name);
ea2384d3 212 if (device_name[0] != '\0') {
1b7bdbc1 213 QTAILQ_INSERT_TAIL(&bdrv_states, bs, list);
ea2384d3 214 }
28a7282a 215 bdrv_iostatus_disable(bs);
b338082b
FB
216 return bs;
217}
218
ea2384d3
FB
219BlockDriver *bdrv_find_format(const char *format_name)
220{
221 BlockDriver *drv1;
8a22f02a
SH
222 QLIST_FOREACH(drv1, &bdrv_drivers, list) {
223 if (!strcmp(drv1->format_name, format_name)) {
ea2384d3 224 return drv1;
8a22f02a 225 }
ea2384d3
FB
226 }
227 return NULL;
228}
229
eb852011
MA
230static int bdrv_is_whitelisted(BlockDriver *drv)
231{
232 static const char *whitelist[] = {
233 CONFIG_BDRV_WHITELIST
234 };
235 const char **p;
236
237 if (!whitelist[0])
238 return 1; /* no whitelist, anything goes */
239
240 for (p = whitelist; *p; p++) {
241 if (!strcmp(drv->format_name, *p)) {
242 return 1;
243 }
244 }
245 return 0;
246}
247
248BlockDriver *bdrv_find_whitelisted_format(const char *format_name)
249{
250 BlockDriver *drv = bdrv_find_format(format_name);
251 return drv && bdrv_is_whitelisted(drv) ? drv : NULL;
252}
253
0e7e1989
KW
254int bdrv_create(BlockDriver *drv, const char* filename,
255 QEMUOptionParameter *options)
ea2384d3
FB
256{
257 if (!drv->bdrv_create)
258 return -ENOTSUP;
0e7e1989
KW
259
260 return drv->bdrv_create(filename, options);
ea2384d3
FB
261}
262
84a12e66
CH
263int bdrv_create_file(const char* filename, QEMUOptionParameter *options)
264{
265 BlockDriver *drv;
266
b50cbabc 267 drv = bdrv_find_protocol(filename);
84a12e66 268 if (drv == NULL) {
16905d71 269 return -ENOENT;
84a12e66
CH
270 }
271
272 return bdrv_create(drv, filename, options);
273}
274
d5249393 275#ifdef _WIN32
95389c86 276void get_tmp_filename(char *filename, int size)
d5249393 277{
3b9f94e1 278 char temp_dir[MAX_PATH];
3b46e624 279
3b9f94e1
FB
280 GetTempPath(MAX_PATH, temp_dir);
281 GetTempFileName(temp_dir, "qem", 0, filename);
d5249393
FB
282}
283#else
95389c86 284void get_tmp_filename(char *filename, int size)
fc01f7e7 285{
67b915a5 286 int fd;
7ccfb2eb 287 const char *tmpdir;
d5249393 288 /* XXX: race condition possible */
0badc1ee
AJ
289 tmpdir = getenv("TMPDIR");
290 if (!tmpdir)
291 tmpdir = "/tmp";
292 snprintf(filename, size, "%s/vl.XXXXXX", tmpdir);
ea2384d3
FB
293 fd = mkstemp(filename);
294 close(fd);
295}
d5249393 296#endif
fc01f7e7 297
84a12e66
CH
298/*
299 * Detect host devices. By convention, /dev/cdrom[N] is always
300 * recognized as a host CDROM.
301 */
302static BlockDriver *find_hdev_driver(const char *filename)
303{
304 int score_max = 0, score;
305 BlockDriver *drv = NULL, *d;
306
307 QLIST_FOREACH(d, &bdrv_drivers, list) {
308 if (d->bdrv_probe_device) {
309 score = d->bdrv_probe_device(filename);
310 if (score > score_max) {
311 score_max = score;
312 drv = d;
313 }
314 }
315 }
316
317 return drv;
318}
319
b50cbabc 320BlockDriver *bdrv_find_protocol(const char *filename)
83f64091
FB
321{
322 BlockDriver *drv1;
323 char protocol[128];
1cec71e3 324 int len;
83f64091 325 const char *p;
19cb3738 326
66f82cee
KW
327 /* TODO Drivers without bdrv_file_open must be specified explicitly */
328
39508e7a
CH
329 /*
330 * XXX(hch): we really should not let host device detection
331 * override an explicit protocol specification, but moving this
332 * later breaks access to device names with colons in them.
333 * Thanks to the brain-dead persistent naming schemes on udev-
334 * based Linux systems those actually are quite common.
335 */
336 drv1 = find_hdev_driver(filename);
337 if (drv1) {
338 return drv1;
339 }
340
9e0b22f4 341 if (!path_has_protocol(filename)) {
39508e7a 342 return bdrv_find_format("file");
84a12e66 343 }
9e0b22f4
SH
344 p = strchr(filename, ':');
345 assert(p != NULL);
1cec71e3
AL
346 len = p - filename;
347 if (len > sizeof(protocol) - 1)
348 len = sizeof(protocol) - 1;
349 memcpy(protocol, filename, len);
350 protocol[len] = '\0';
8a22f02a 351 QLIST_FOREACH(drv1, &bdrv_drivers, list) {
5fafdf24 352 if (drv1->protocol_name &&
8a22f02a 353 !strcmp(drv1->protocol_name, protocol)) {
83f64091 354 return drv1;
8a22f02a 355 }
83f64091
FB
356 }
357 return NULL;
358}
359
c98ac35d 360static int find_image_format(const char *filename, BlockDriver **pdrv)
f3a5d3f8
CH
361{
362 int ret, score, score_max;
363 BlockDriver *drv1, *drv;
364 uint8_t buf[2048];
365 BlockDriverState *bs;
366
f5edb014 367 ret = bdrv_file_open(&bs, filename, 0);
c98ac35d
SW
368 if (ret < 0) {
369 *pdrv = NULL;
370 return ret;
371 }
f8ea0b00 372
08a00559
KW
373 /* Return the raw BlockDriver * to scsi-generic devices or empty drives */
374 if (bs->sg || !bdrv_is_inserted(bs)) {
1a396859 375 bdrv_delete(bs);
c98ac35d
SW
376 drv = bdrv_find_format("raw");
377 if (!drv) {
378 ret = -ENOENT;
379 }
380 *pdrv = drv;
381 return ret;
1a396859 382 }
f8ea0b00 383
83f64091
FB
384 ret = bdrv_pread(bs, 0, buf, sizeof(buf));
385 bdrv_delete(bs);
386 if (ret < 0) {
c98ac35d
SW
387 *pdrv = NULL;
388 return ret;
83f64091
FB
389 }
390
ea2384d3 391 score_max = 0;
84a12e66 392 drv = NULL;
8a22f02a 393 QLIST_FOREACH(drv1, &bdrv_drivers, list) {
83f64091
FB
394 if (drv1->bdrv_probe) {
395 score = drv1->bdrv_probe(buf, ret, filename);
396 if (score > score_max) {
397 score_max = score;
398 drv = drv1;
399 }
0849bf08 400 }
fc01f7e7 401 }
c98ac35d
SW
402 if (!drv) {
403 ret = -ENOENT;
404 }
405 *pdrv = drv;
406 return ret;
ea2384d3
FB
407}
408
51762288
SH
409/**
410 * Set the current 'total_sectors' value
411 */
412static int refresh_total_sectors(BlockDriverState *bs, int64_t hint)
413{
414 BlockDriver *drv = bs->drv;
415
396759ad
NB
416 /* Do not attempt drv->bdrv_getlength() on scsi-generic devices */
417 if (bs->sg)
418 return 0;
419
51762288
SH
420 /* query actual device if possible, otherwise just trust the hint */
421 if (drv->bdrv_getlength) {
422 int64_t length = drv->bdrv_getlength(bs);
423 if (length < 0) {
424 return length;
425 }
426 hint = length >> BDRV_SECTOR_BITS;
427 }
428
429 bs->total_sectors = hint;
430 return 0;
431}
432
c3993cdc
SH
433/**
434 * Set open flags for a given cache mode
435 *
436 * Return 0 on success, -1 if the cache mode was invalid.
437 */
438int bdrv_parse_cache_flags(const char *mode, int *flags)
439{
440 *flags &= ~BDRV_O_CACHE_MASK;
441
442 if (!strcmp(mode, "off") || !strcmp(mode, "none")) {
443 *flags |= BDRV_O_NOCACHE | BDRV_O_CACHE_WB;
92196b2f
SH
444 } else if (!strcmp(mode, "directsync")) {
445 *flags |= BDRV_O_NOCACHE;
c3993cdc
SH
446 } else if (!strcmp(mode, "writeback")) {
447 *flags |= BDRV_O_CACHE_WB;
448 } else if (!strcmp(mode, "unsafe")) {
449 *flags |= BDRV_O_CACHE_WB;
450 *flags |= BDRV_O_NO_FLUSH;
451 } else if (!strcmp(mode, "writethrough")) {
452 /* this is the default */
453 } else {
454 return -1;
455 }
456
457 return 0;
458}
459
57915332
KW
460/*
461 * Common part for opening disk images and files
462 */
463static int bdrv_open_common(BlockDriverState *bs, const char *filename,
464 int flags, BlockDriver *drv)
465{
466 int ret, open_flags;
467
468 assert(drv != NULL);
469
28dcee10
SH
470 trace_bdrv_open_common(bs, filename, flags, drv->format_name);
471
66f82cee 472 bs->file = NULL;
51762288 473 bs->total_sectors = 0;
57915332
KW
474 bs->encrypted = 0;
475 bs->valid_key = 0;
476 bs->open_flags = flags;
57915332
KW
477 bs->buffer_alignment = 512;
478
479 pstrcpy(bs->filename, sizeof(bs->filename), filename);
480
481 if (use_bdrv_whitelist && !bdrv_is_whitelisted(drv)) {
482 return -ENOTSUP;
483 }
484
485 bs->drv = drv;
7267c094 486 bs->opaque = g_malloc0(drv->instance_size);
57915332 487
a6599793 488 if (flags & BDRV_O_CACHE_WB)
57915332
KW
489 bs->enable_write_cache = 1;
490
491 /*
492 * Clear flags that are internal to the block layer before opening the
493 * image.
494 */
495 open_flags = flags & ~(BDRV_O_SNAPSHOT | BDRV_O_NO_BACKING);
496
497 /*
ebabb67a 498 * Snapshots should be writable.
57915332
KW
499 */
500 if (bs->is_temporary) {
501 open_flags |= BDRV_O_RDWR;
502 }
503
66f82cee
KW
504 /* Open the image, either directly or using a protocol */
505 if (drv->bdrv_file_open) {
506 ret = drv->bdrv_file_open(bs, filename, open_flags);
507 } else {
508 ret = bdrv_file_open(&bs->file, filename, open_flags);
509 if (ret >= 0) {
510 ret = drv->bdrv_open(bs, open_flags);
511 }
512 }
513
57915332
KW
514 if (ret < 0) {
515 goto free_and_fail;
516 }
517
518 bs->keep_read_only = bs->read_only = !(open_flags & BDRV_O_RDWR);
51762288
SH
519
520 ret = refresh_total_sectors(bs, bs->total_sectors);
521 if (ret < 0) {
522 goto free_and_fail;
57915332 523 }
51762288 524
57915332
KW
525#ifndef _WIN32
526 if (bs->is_temporary) {
527 unlink(filename);
528 }
529#endif
530 return 0;
531
532free_and_fail:
66f82cee
KW
533 if (bs->file) {
534 bdrv_delete(bs->file);
535 bs->file = NULL;
536 }
7267c094 537 g_free(bs->opaque);
57915332
KW
538 bs->opaque = NULL;
539 bs->drv = NULL;
540 return ret;
541}
542
b6ce07aa
KW
543/*
544 * Opens a file using a protocol (file, host_device, nbd, ...)
545 */
83f64091 546int bdrv_file_open(BlockDriverState **pbs, const char *filename, int flags)
ea2384d3 547{
83f64091 548 BlockDriverState *bs;
6db95603 549 BlockDriver *drv;
83f64091
FB
550 int ret;
551
b50cbabc 552 drv = bdrv_find_protocol(filename);
6db95603
CH
553 if (!drv) {
554 return -ENOENT;
555 }
556
83f64091 557 bs = bdrv_new("");
b6ce07aa 558 ret = bdrv_open_common(bs, filename, flags, drv);
83f64091
FB
559 if (ret < 0) {
560 bdrv_delete(bs);
561 return ret;
3b0d4f61 562 }
71d0770c 563 bs->growable = 1;
83f64091
FB
564 *pbs = bs;
565 return 0;
566}
567
b6ce07aa
KW
568/*
569 * Opens a disk image (raw, qcow2, vmdk, ...)
570 */
d6e9098e
KW
571int bdrv_open(BlockDriverState *bs, const char *filename, int flags,
572 BlockDriver *drv)
ea2384d3 573{
b6ce07aa 574 int ret;
712e7874 575
83f64091 576 if (flags & BDRV_O_SNAPSHOT) {
ea2384d3
FB
577 BlockDriverState *bs1;
578 int64_t total_size;
7c96d46e 579 int is_protocol = 0;
91a073a9
KW
580 BlockDriver *bdrv_qcow2;
581 QEMUOptionParameter *options;
b6ce07aa
KW
582 char tmp_filename[PATH_MAX];
583 char backing_filename[PATH_MAX];
3b46e624 584
ea2384d3
FB
585 /* if snapshot, we create a temporary backing file and open it
586 instead of opening 'filename' directly */
33e3963e 587
ea2384d3
FB
588 /* if there is a backing file, use it */
589 bs1 = bdrv_new("");
d6e9098e 590 ret = bdrv_open(bs1, filename, 0, drv);
51d7c00c 591 if (ret < 0) {
ea2384d3 592 bdrv_delete(bs1);
51d7c00c 593 return ret;
ea2384d3 594 }
3e82990b 595 total_size = bdrv_getlength(bs1) & BDRV_SECTOR_MASK;
7c96d46e
AL
596
597 if (bs1->drv && bs1->drv->protocol_name)
598 is_protocol = 1;
599
ea2384d3 600 bdrv_delete(bs1);
3b46e624 601
ea2384d3 602 get_tmp_filename(tmp_filename, sizeof(tmp_filename));
7c96d46e
AL
603
604 /* Real path is meaningless for protocols */
605 if (is_protocol)
606 snprintf(backing_filename, sizeof(backing_filename),
607 "%s", filename);
114cdfa9
KS
608 else if (!realpath(filename, backing_filename))
609 return -errno;
7c96d46e 610
91a073a9
KW
611 bdrv_qcow2 = bdrv_find_format("qcow2");
612 options = parse_option_parameters("", bdrv_qcow2->create_options, NULL);
613
3e82990b 614 set_option_parameter_int(options, BLOCK_OPT_SIZE, total_size);
91a073a9
KW
615 set_option_parameter(options, BLOCK_OPT_BACKING_FILE, backing_filename);
616 if (drv) {
617 set_option_parameter(options, BLOCK_OPT_BACKING_FMT,
618 drv->format_name);
619 }
620
621 ret = bdrv_create(bdrv_qcow2, tmp_filename, options);
d748768c 622 free_option_parameters(options);
51d7c00c
AL
623 if (ret < 0) {
624 return ret;
ea2384d3 625 }
91a073a9 626
ea2384d3 627 filename = tmp_filename;
91a073a9 628 drv = bdrv_qcow2;
ea2384d3
FB
629 bs->is_temporary = 1;
630 }
712e7874 631
b6ce07aa 632 /* Find the right image format driver */
6db95603 633 if (!drv) {
c98ac35d 634 ret = find_image_format(filename, &drv);
51d7c00c 635 }
6987307c 636
51d7c00c 637 if (!drv) {
51d7c00c 638 goto unlink_and_fail;
ea2384d3 639 }
b6ce07aa
KW
640
641 /* Open the image */
642 ret = bdrv_open_common(bs, filename, flags, drv);
643 if (ret < 0) {
6987307c
CH
644 goto unlink_and_fail;
645 }
646
b6ce07aa
KW
647 /* If there is a backing file, use it */
648 if ((flags & BDRV_O_NO_BACKING) == 0 && bs->backing_file[0] != '\0') {
649 char backing_filename[PATH_MAX];
650 int back_flags;
651 BlockDriver *back_drv = NULL;
652
653 bs->backing_hd = bdrv_new("");
df2dbb4a
SH
654
655 if (path_has_protocol(bs->backing_file)) {
656 pstrcpy(backing_filename, sizeof(backing_filename),
657 bs->backing_file);
658 } else {
659 path_combine(backing_filename, sizeof(backing_filename),
660 filename, bs->backing_file);
661 }
662
663 if (bs->backing_format[0] != '\0') {
b6ce07aa 664 back_drv = bdrv_find_format(bs->backing_format);
df2dbb4a 665 }
b6ce07aa
KW
666
667 /* backing files always opened read-only */
668 back_flags =
669 flags & ~(BDRV_O_RDWR | BDRV_O_SNAPSHOT | BDRV_O_NO_BACKING);
670
671 ret = bdrv_open(bs->backing_hd, backing_filename, back_flags, back_drv);
672 if (ret < 0) {
673 bdrv_close(bs);
674 return ret;
675 }
676 if (bs->is_temporary) {
677 bs->backing_hd->keep_read_only = !(flags & BDRV_O_RDWR);
678 } else {
679 /* base image inherits from "parent" */
680 bs->backing_hd->keep_read_only = bs->keep_read_only;
681 }
682 }
683
684 if (!bdrv_key_required(bs)) {
7d4b4ba5 685 bdrv_dev_change_media_cb(bs, true);
b6ce07aa
KW
686 }
687
688 return 0;
689
690unlink_and_fail:
691 if (bs->is_temporary) {
692 unlink(filename);
693 }
694 return ret;
695}
696
fc01f7e7
FB
697void bdrv_close(BlockDriverState *bs)
698{
19cb3738 699 if (bs->drv) {
f9092b10
MA
700 if (bs == bs_snapshots) {
701 bs_snapshots = NULL;
702 }
557df6ac 703 if (bs->backing_hd) {
ea2384d3 704 bdrv_delete(bs->backing_hd);
557df6ac
SH
705 bs->backing_hd = NULL;
706 }
ea2384d3 707 bs->drv->bdrv_close(bs);
7267c094 708 g_free(bs->opaque);
ea2384d3
FB
709#ifdef _WIN32
710 if (bs->is_temporary) {
711 unlink(bs->filename);
712 }
67b915a5 713#endif
ea2384d3
FB
714 bs->opaque = NULL;
715 bs->drv = NULL;
b338082b 716
66f82cee
KW
717 if (bs->file != NULL) {
718 bdrv_close(bs->file);
719 }
720
7d4b4ba5 721 bdrv_dev_change_media_cb(bs, false);
b338082b
FB
722 }
723}
724
2bc93fed
MK
725void bdrv_close_all(void)
726{
727 BlockDriverState *bs;
728
729 QTAILQ_FOREACH(bs, &bdrv_states, list) {
730 bdrv_close(bs);
731 }
732}
733
d22b2f41
RH
734/* make a BlockDriverState anonymous by removing from bdrv_state list.
735 Also, NULL terminate the device_name to prevent double remove */
736void bdrv_make_anon(BlockDriverState *bs)
737{
738 if (bs->device_name[0] != '\0') {
739 QTAILQ_REMOVE(&bdrv_states, bs, list);
740 }
741 bs->device_name[0] = '\0';
742}
743
b338082b
FB
744void bdrv_delete(BlockDriverState *bs)
745{
fa879d62 746 assert(!bs->dev);
18846dee 747
1b7bdbc1 748 /* remove from list, if necessary */
d22b2f41 749 bdrv_make_anon(bs);
34c6f050 750
b338082b 751 bdrv_close(bs);
66f82cee
KW
752 if (bs->file != NULL) {
753 bdrv_delete(bs->file);
754 }
755
f9092b10 756 assert(bs != bs_snapshots);
7267c094 757 g_free(bs);
fc01f7e7
FB
758}
759
fa879d62
MA
760int bdrv_attach_dev(BlockDriverState *bs, void *dev)
761/* TODO change to DeviceState *dev when all users are qdevified */
18846dee 762{
fa879d62 763 if (bs->dev) {
18846dee
MA
764 return -EBUSY;
765 }
fa879d62 766 bs->dev = dev;
28a7282a 767 bdrv_iostatus_reset(bs);
18846dee
MA
768 return 0;
769}
770
fa879d62
MA
771/* TODO qdevified devices don't use this, remove when devices are qdevified */
772void bdrv_attach_dev_nofail(BlockDriverState *bs, void *dev)
18846dee 773{
fa879d62
MA
774 if (bdrv_attach_dev(bs, dev) < 0) {
775 abort();
776 }
777}
778
779void bdrv_detach_dev(BlockDriverState *bs, void *dev)
780/* TODO change to DeviceState *dev when all users are qdevified */
781{
782 assert(bs->dev == dev);
783 bs->dev = NULL;
0e49de52
MA
784 bs->dev_ops = NULL;
785 bs->dev_opaque = NULL;
29e05f20 786 bs->buffer_alignment = 512;
18846dee
MA
787}
788
fa879d62
MA
789/* TODO change to return DeviceState * when all users are qdevified */
790void *bdrv_get_attached_dev(BlockDriverState *bs)
18846dee 791{
fa879d62 792 return bs->dev;
18846dee
MA
793}
794
0e49de52
MA
795void bdrv_set_dev_ops(BlockDriverState *bs, const BlockDevOps *ops,
796 void *opaque)
797{
798 bs->dev_ops = ops;
799 bs->dev_opaque = opaque;
2c6942fa
MA
800 if (bdrv_dev_has_removable_media(bs) && bs == bs_snapshots) {
801 bs_snapshots = NULL;
802 }
0e49de52
MA
803}
804
7d4b4ba5 805static void bdrv_dev_change_media_cb(BlockDriverState *bs, bool load)
0e49de52 806{
145feb17 807 if (bs->dev_ops && bs->dev_ops->change_media_cb) {
7d4b4ba5 808 bs->dev_ops->change_media_cb(bs->dev_opaque, load);
145feb17
MA
809 }
810}
811
2c6942fa
MA
812bool bdrv_dev_has_removable_media(BlockDriverState *bs)
813{
814 return !bs->dev || (bs->dev_ops && bs->dev_ops->change_media_cb);
815}
816
e4def80b
MA
817bool bdrv_dev_is_tray_open(BlockDriverState *bs)
818{
819 if (bs->dev_ops && bs->dev_ops->is_tray_open) {
820 return bs->dev_ops->is_tray_open(bs->dev_opaque);
821 }
822 return false;
823}
824
145feb17
MA
825static void bdrv_dev_resize_cb(BlockDriverState *bs)
826{
827 if (bs->dev_ops && bs->dev_ops->resize_cb) {
828 bs->dev_ops->resize_cb(bs->dev_opaque);
0e49de52
MA
829 }
830}
831
f107639a
MA
832bool bdrv_dev_is_medium_locked(BlockDriverState *bs)
833{
834 if (bs->dev_ops && bs->dev_ops->is_medium_locked) {
835 return bs->dev_ops->is_medium_locked(bs->dev_opaque);
836 }
837 return false;
838}
839
e97fc193
AL
840/*
841 * Run consistency checks on an image
842 *
e076f338 843 * Returns 0 if the check could be completed (it doesn't mean that the image is
a1c7273b 844 * free of errors) or -errno when an internal error occurred. The results of the
e076f338 845 * check are stored in res.
e97fc193 846 */
e076f338 847int bdrv_check(BlockDriverState *bs, BdrvCheckResult *res)
e97fc193
AL
848{
849 if (bs->drv->bdrv_check == NULL) {
850 return -ENOTSUP;
851 }
852
e076f338 853 memset(res, 0, sizeof(*res));
9ac228e0 854 return bs->drv->bdrv_check(bs, res);
e97fc193
AL
855}
856
8a426614
KW
857#define COMMIT_BUF_SECTORS 2048
858
33e3963e
FB
859/* commit COW file into the raw image */
860int bdrv_commit(BlockDriverState *bs)
861{
19cb3738 862 BlockDriver *drv = bs->drv;
ee181196 863 BlockDriver *backing_drv;
8a426614
KW
864 int64_t sector, total_sectors;
865 int n, ro, open_flags;
4dca4b63 866 int ret = 0, rw_ret = 0;
8a426614 867 uint8_t *buf;
4dca4b63
NS
868 char filename[1024];
869 BlockDriverState *bs_rw, *bs_ro;
33e3963e 870
19cb3738
FB
871 if (!drv)
872 return -ENOMEDIUM;
4dca4b63
NS
873
874 if (!bs->backing_hd) {
875 return -ENOTSUP;
33e3963e
FB
876 }
877
4dca4b63
NS
878 if (bs->backing_hd->keep_read_only) {
879 return -EACCES;
880 }
ee181196
KW
881
882 backing_drv = bs->backing_hd->drv;
4dca4b63
NS
883 ro = bs->backing_hd->read_only;
884 strncpy(filename, bs->backing_hd->filename, sizeof(filename));
885 open_flags = bs->backing_hd->open_flags;
886
887 if (ro) {
888 /* re-open as RW */
889 bdrv_delete(bs->backing_hd);
890 bs->backing_hd = NULL;
891 bs_rw = bdrv_new("");
ee181196
KW
892 rw_ret = bdrv_open(bs_rw, filename, open_flags | BDRV_O_RDWR,
893 backing_drv);
4dca4b63
NS
894 if (rw_ret < 0) {
895 bdrv_delete(bs_rw);
896 /* try to re-open read-only */
897 bs_ro = bdrv_new("");
ee181196
KW
898 ret = bdrv_open(bs_ro, filename, open_flags & ~BDRV_O_RDWR,
899 backing_drv);
4dca4b63
NS
900 if (ret < 0) {
901 bdrv_delete(bs_ro);
902 /* drive not functional anymore */
903 bs->drv = NULL;
904 return ret;
905 }
906 bs->backing_hd = bs_ro;
907 return rw_ret;
908 }
909 bs->backing_hd = bs_rw;
ea2384d3 910 }
33e3963e 911
6ea44308 912 total_sectors = bdrv_getlength(bs) >> BDRV_SECTOR_BITS;
7267c094 913 buf = g_malloc(COMMIT_BUF_SECTORS * BDRV_SECTOR_SIZE);
8a426614
KW
914
915 for (sector = 0; sector < total_sectors; sector += n) {
916 if (drv->bdrv_is_allocated(bs, sector, COMMIT_BUF_SECTORS, &n)) {
917
918 if (bdrv_read(bs, sector, buf, n) != 0) {
919 ret = -EIO;
920 goto ro_cleanup;
921 }
922
923 if (bdrv_write(bs->backing_hd, sector, buf, n) != 0) {
924 ret = -EIO;
925 goto ro_cleanup;
926 }
ea2384d3 927 }
33e3963e 928 }
95389c86 929
1d44952f
CH
930 if (drv->bdrv_make_empty) {
931 ret = drv->bdrv_make_empty(bs);
932 bdrv_flush(bs);
933 }
95389c86 934
3f5075ae
CH
935 /*
936 * Make sure all data we wrote to the backing device is actually
937 * stable on disk.
938 */
939 if (bs->backing_hd)
940 bdrv_flush(bs->backing_hd);
4dca4b63
NS
941
942ro_cleanup:
7267c094 943 g_free(buf);
4dca4b63
NS
944
945 if (ro) {
946 /* re-open as RO */
947 bdrv_delete(bs->backing_hd);
948 bs->backing_hd = NULL;
949 bs_ro = bdrv_new("");
ee181196
KW
950 ret = bdrv_open(bs_ro, filename, open_flags & ~BDRV_O_RDWR,
951 backing_drv);
4dca4b63
NS
952 if (ret < 0) {
953 bdrv_delete(bs_ro);
954 /* drive not functional anymore */
955 bs->drv = NULL;
956 return ret;
957 }
958 bs->backing_hd = bs_ro;
959 bs->backing_hd->keep_read_only = 0;
960 }
961
1d44952f 962 return ret;
33e3963e
FB
963}
964
6ab4b5ab
MA
965void bdrv_commit_all(void)
966{
967 BlockDriverState *bs;
968
969 QTAILQ_FOREACH(bs, &bdrv_states, list) {
970 bdrv_commit(bs);
971 }
972}
973
756e6736
KW
974/*
975 * Return values:
976 * 0 - success
977 * -EINVAL - backing format specified, but no file
978 * -ENOSPC - can't update the backing file because no space is left in the
979 * image file header
980 * -ENOTSUP - format driver doesn't support changing the backing file
981 */
982int bdrv_change_backing_file(BlockDriverState *bs,
983 const char *backing_file, const char *backing_fmt)
984{
985 BlockDriver *drv = bs->drv;
986
987 if (drv->bdrv_change_backing_file != NULL) {
988 return drv->bdrv_change_backing_file(bs, backing_file, backing_fmt);
989 } else {
990 return -ENOTSUP;
991 }
992}
993
71d0770c
AL
994static int bdrv_check_byte_request(BlockDriverState *bs, int64_t offset,
995 size_t size)
996{
997 int64_t len;
998
999 if (!bdrv_is_inserted(bs))
1000 return -ENOMEDIUM;
1001
1002 if (bs->growable)
1003 return 0;
1004
1005 len = bdrv_getlength(bs);
1006
fbb7b4e0
KW
1007 if (offset < 0)
1008 return -EIO;
1009
1010 if ((offset > len) || (len - offset < size))
71d0770c
AL
1011 return -EIO;
1012
1013 return 0;
1014}
1015
1016static int bdrv_check_request(BlockDriverState *bs, int64_t sector_num,
1017 int nb_sectors)
1018{
eb5a3165
JS
1019 return bdrv_check_byte_request(bs, sector_num * BDRV_SECTOR_SIZE,
1020 nb_sectors * BDRV_SECTOR_SIZE);
71d0770c
AL
1021}
1022
1c9805a3
SH
1023typedef struct RwCo {
1024 BlockDriverState *bs;
1025 int64_t sector_num;
1026 int nb_sectors;
1027 QEMUIOVector *qiov;
1028 bool is_write;
1029 int ret;
1030} RwCo;
1031
1032static void coroutine_fn bdrv_rw_co_entry(void *opaque)
fc01f7e7 1033{
1c9805a3 1034 RwCo *rwco = opaque;
ea2384d3 1035
1c9805a3
SH
1036 if (!rwco->is_write) {
1037 rwco->ret = bdrv_co_do_readv(rwco->bs, rwco->sector_num,
1038 rwco->nb_sectors, rwco->qiov);
1039 } else {
1040 rwco->ret = bdrv_co_do_writev(rwco->bs, rwco->sector_num,
1041 rwco->nb_sectors, rwco->qiov);
1042 }
1043}
e7a8a783 1044
1c9805a3
SH
1045/*
1046 * Process a synchronous request using coroutines
1047 */
1048static int bdrv_rw_co(BlockDriverState *bs, int64_t sector_num, uint8_t *buf,
1049 int nb_sectors, bool is_write)
1050{
1051 QEMUIOVector qiov;
1052 struct iovec iov = {
1053 .iov_base = (void *)buf,
1054 .iov_len = nb_sectors * BDRV_SECTOR_SIZE,
1055 };
1056 Coroutine *co;
1057 RwCo rwco = {
1058 .bs = bs,
1059 .sector_num = sector_num,
1060 .nb_sectors = nb_sectors,
1061 .qiov = &qiov,
1062 .is_write = is_write,
1063 .ret = NOT_DONE,
1064 };
e7a8a783 1065
1c9805a3 1066 qemu_iovec_init_external(&qiov, &iov, 1);
e7a8a783 1067
1c9805a3
SH
1068 if (qemu_in_coroutine()) {
1069 /* Fast-path if already in coroutine context */
1070 bdrv_rw_co_entry(&rwco);
1071 } else {
1072 co = qemu_coroutine_create(bdrv_rw_co_entry);
1073 qemu_coroutine_enter(co, &rwco);
1074 while (rwco.ret == NOT_DONE) {
1075 qemu_aio_wait();
1076 }
1077 }
1078 return rwco.ret;
1079}
b338082b 1080
1c9805a3
SH
1081/* return < 0 if error. See bdrv_write() for the return codes */
1082int bdrv_read(BlockDriverState *bs, int64_t sector_num,
1083 uint8_t *buf, int nb_sectors)
1084{
1085 return bdrv_rw_co(bs, sector_num, buf, nb_sectors, false);
fc01f7e7
FB
1086}
1087
7cd1e32a 1088static void set_dirty_bitmap(BlockDriverState *bs, int64_t sector_num,
a55eb92c 1089 int nb_sectors, int dirty)
7cd1e32a
LS
1090{
1091 int64_t start, end;
c6d22830 1092 unsigned long val, idx, bit;
a55eb92c 1093
6ea44308 1094 start = sector_num / BDRV_SECTORS_PER_DIRTY_CHUNK;
c6d22830 1095 end = (sector_num + nb_sectors - 1) / BDRV_SECTORS_PER_DIRTY_CHUNK;
a55eb92c
JK
1096
1097 for (; start <= end; start++) {
c6d22830
JK
1098 idx = start / (sizeof(unsigned long) * 8);
1099 bit = start % (sizeof(unsigned long) * 8);
1100 val = bs->dirty_bitmap[idx];
1101 if (dirty) {
6d59fec1 1102 if (!(val & (1UL << bit))) {
aaa0eb75 1103 bs->dirty_count++;
6d59fec1 1104 val |= 1UL << bit;
aaa0eb75 1105 }
c6d22830 1106 } else {
6d59fec1 1107 if (val & (1UL << bit)) {
aaa0eb75 1108 bs->dirty_count--;
6d59fec1 1109 val &= ~(1UL << bit);
aaa0eb75 1110 }
c6d22830
JK
1111 }
1112 bs->dirty_bitmap[idx] = val;
7cd1e32a
LS
1113 }
1114}
1115
5fafdf24 1116/* Return < 0 if error. Important errors are:
19cb3738
FB
1117 -EIO generic I/O error (may happen for all errors)
1118 -ENOMEDIUM No media inserted.
1119 -EINVAL Invalid sector number or nb_sectors
1120 -EACCES Trying to write a read-only device
1121*/
5fafdf24 1122int bdrv_write(BlockDriverState *bs, int64_t sector_num,
fc01f7e7
FB
1123 const uint8_t *buf, int nb_sectors)
1124{
1c9805a3 1125 return bdrv_rw_co(bs, sector_num, (uint8_t *)buf, nb_sectors, true);
83f64091
FB
1126}
1127
eda578e5
AL
1128int bdrv_pread(BlockDriverState *bs, int64_t offset,
1129 void *buf, int count1)
83f64091 1130{
6ea44308 1131 uint8_t tmp_buf[BDRV_SECTOR_SIZE];
83f64091
FB
1132 int len, nb_sectors, count;
1133 int64_t sector_num;
9a8c4cce 1134 int ret;
83f64091
FB
1135
1136 count = count1;
1137 /* first read to align to sector start */
6ea44308 1138 len = (BDRV_SECTOR_SIZE - offset) & (BDRV_SECTOR_SIZE - 1);
83f64091
FB
1139 if (len > count)
1140 len = count;
6ea44308 1141 sector_num = offset >> BDRV_SECTOR_BITS;
83f64091 1142 if (len > 0) {
9a8c4cce
KW
1143 if ((ret = bdrv_read(bs, sector_num, tmp_buf, 1)) < 0)
1144 return ret;
6ea44308 1145 memcpy(buf, tmp_buf + (offset & (BDRV_SECTOR_SIZE - 1)), len);
83f64091
FB
1146 count -= len;
1147 if (count == 0)
1148 return count1;
1149 sector_num++;
1150 buf += len;
1151 }
1152
1153 /* read the sectors "in place" */
6ea44308 1154 nb_sectors = count >> BDRV_SECTOR_BITS;
83f64091 1155 if (nb_sectors > 0) {
9a8c4cce
KW
1156 if ((ret = bdrv_read(bs, sector_num, buf, nb_sectors)) < 0)
1157 return ret;
83f64091 1158 sector_num += nb_sectors;
6ea44308 1159 len = nb_sectors << BDRV_SECTOR_BITS;
83f64091
FB
1160 buf += len;
1161 count -= len;
1162 }
1163
1164 /* add data from the last sector */
1165 if (count > 0) {
9a8c4cce
KW
1166 if ((ret = bdrv_read(bs, sector_num, tmp_buf, 1)) < 0)
1167 return ret;
83f64091
FB
1168 memcpy(buf, tmp_buf, count);
1169 }
1170 return count1;
1171}
1172
eda578e5
AL
1173int bdrv_pwrite(BlockDriverState *bs, int64_t offset,
1174 const void *buf, int count1)
83f64091 1175{
6ea44308 1176 uint8_t tmp_buf[BDRV_SECTOR_SIZE];
83f64091
FB
1177 int len, nb_sectors, count;
1178 int64_t sector_num;
9a8c4cce 1179 int ret;
83f64091
FB
1180
1181 count = count1;
1182 /* first write to align to sector start */
6ea44308 1183 len = (BDRV_SECTOR_SIZE - offset) & (BDRV_SECTOR_SIZE - 1);
83f64091
FB
1184 if (len > count)
1185 len = count;
6ea44308 1186 sector_num = offset >> BDRV_SECTOR_BITS;
83f64091 1187 if (len > 0) {
9a8c4cce
KW
1188 if ((ret = bdrv_read(bs, sector_num, tmp_buf, 1)) < 0)
1189 return ret;
6ea44308 1190 memcpy(tmp_buf + (offset & (BDRV_SECTOR_SIZE - 1)), buf, len);
9a8c4cce
KW
1191 if ((ret = bdrv_write(bs, sector_num, tmp_buf, 1)) < 0)
1192 return ret;
83f64091
FB
1193 count -= len;
1194 if (count == 0)
1195 return count1;
1196 sector_num++;
1197 buf += len;
1198 }
1199
1200 /* write the sectors "in place" */
6ea44308 1201 nb_sectors = count >> BDRV_SECTOR_BITS;
83f64091 1202 if (nb_sectors > 0) {
9a8c4cce
KW
1203 if ((ret = bdrv_write(bs, sector_num, buf, nb_sectors)) < 0)
1204 return ret;
83f64091 1205 sector_num += nb_sectors;
6ea44308 1206 len = nb_sectors << BDRV_SECTOR_BITS;
83f64091
FB
1207 buf += len;
1208 count -= len;
1209 }
1210
1211 /* add data from the last sector */
1212 if (count > 0) {
9a8c4cce
KW
1213 if ((ret = bdrv_read(bs, sector_num, tmp_buf, 1)) < 0)
1214 return ret;
83f64091 1215 memcpy(tmp_buf, buf, count);
9a8c4cce
KW
1216 if ((ret = bdrv_write(bs, sector_num, tmp_buf, 1)) < 0)
1217 return ret;
83f64091
FB
1218 }
1219 return count1;
1220}
83f64091 1221
f08145fe
KW
1222/*
1223 * Writes to the file and ensures that no writes are reordered across this
1224 * request (acts as a barrier)
1225 *
1226 * Returns 0 on success, -errno in error cases.
1227 */
1228int bdrv_pwrite_sync(BlockDriverState *bs, int64_t offset,
1229 const void *buf, int count)
1230{
1231 int ret;
1232
1233 ret = bdrv_pwrite(bs, offset, buf, count);
1234 if (ret < 0) {
1235 return ret;
1236 }
1237
92196b2f
SH
1238 /* No flush needed for cache modes that use O_DSYNC */
1239 if ((bs->open_flags & BDRV_O_CACHE_WB) != 0) {
f08145fe
KW
1240 bdrv_flush(bs);
1241 }
1242
1243 return 0;
1244}
1245
c5fbe571
SH
1246/*
1247 * Handle a read request in coroutine context
1248 */
1249static int coroutine_fn bdrv_co_do_readv(BlockDriverState *bs,
1250 int64_t sector_num, int nb_sectors, QEMUIOVector *qiov)
da1fa91d
KW
1251{
1252 BlockDriver *drv = bs->drv;
1253
da1fa91d
KW
1254 if (!drv) {
1255 return -ENOMEDIUM;
1256 }
1257 if (bdrv_check_request(bs, sector_num, nb_sectors)) {
1258 return -EIO;
1259 }
1260
1261 return drv->bdrv_co_readv(bs, sector_num, nb_sectors, qiov);
1262}
1263
c5fbe571 1264int coroutine_fn bdrv_co_readv(BlockDriverState *bs, int64_t sector_num,
da1fa91d
KW
1265 int nb_sectors, QEMUIOVector *qiov)
1266{
c5fbe571 1267 trace_bdrv_co_readv(bs, sector_num, nb_sectors);
da1fa91d 1268
c5fbe571
SH
1269 return bdrv_co_do_readv(bs, sector_num, nb_sectors, qiov);
1270}
1271
1272/*
1273 * Handle a write request in coroutine context
1274 */
1275static int coroutine_fn bdrv_co_do_writev(BlockDriverState *bs,
1276 int64_t sector_num, int nb_sectors, QEMUIOVector *qiov)
1277{
1278 BlockDriver *drv = bs->drv;
6b7cb247 1279 int ret;
da1fa91d
KW
1280
1281 if (!bs->drv) {
1282 return -ENOMEDIUM;
1283 }
1284 if (bs->read_only) {
1285 return -EACCES;
1286 }
1287 if (bdrv_check_request(bs, sector_num, nb_sectors)) {
1288 return -EIO;
1289 }
1290
6b7cb247
SH
1291 ret = drv->bdrv_co_writev(bs, sector_num, nb_sectors, qiov);
1292
da1fa91d
KW
1293 if (bs->dirty_bitmap) {
1294 set_dirty_bitmap(bs, sector_num, nb_sectors, 1);
1295 }
1296
1297 if (bs->wr_highest_sector < sector_num + nb_sectors - 1) {
1298 bs->wr_highest_sector = sector_num + nb_sectors - 1;
1299 }
1300
6b7cb247 1301 return ret;
da1fa91d
KW
1302}
1303
c5fbe571
SH
1304int coroutine_fn bdrv_co_writev(BlockDriverState *bs, int64_t sector_num,
1305 int nb_sectors, QEMUIOVector *qiov)
1306{
1307 trace_bdrv_co_writev(bs, sector_num, nb_sectors);
1308
1309 return bdrv_co_do_writev(bs, sector_num, nb_sectors, qiov);
1310}
1311
83f64091
FB
1312/**
1313 * Truncate file to 'offset' bytes (needed only for file protocols)
1314 */
1315int bdrv_truncate(BlockDriverState *bs, int64_t offset)
1316{
1317 BlockDriver *drv = bs->drv;
51762288 1318 int ret;
83f64091 1319 if (!drv)
19cb3738 1320 return -ENOMEDIUM;
83f64091
FB
1321 if (!drv->bdrv_truncate)
1322 return -ENOTSUP;
59f2689d
NS
1323 if (bs->read_only)
1324 return -EACCES;
8591675f
MT
1325 if (bdrv_in_use(bs))
1326 return -EBUSY;
51762288
SH
1327 ret = drv->bdrv_truncate(bs, offset);
1328 if (ret == 0) {
1329 ret = refresh_total_sectors(bs, offset >> BDRV_SECTOR_BITS);
145feb17 1330 bdrv_dev_resize_cb(bs);
51762288
SH
1331 }
1332 return ret;
83f64091
FB
1333}
1334
4a1d5e1f
FZ
1335/**
1336 * Length of a allocated file in bytes. Sparse files are counted by actual
1337 * allocated space. Return < 0 if error or unknown.
1338 */
1339int64_t bdrv_get_allocated_file_size(BlockDriverState *bs)
1340{
1341 BlockDriver *drv = bs->drv;
1342 if (!drv) {
1343 return -ENOMEDIUM;
1344 }
1345 if (drv->bdrv_get_allocated_file_size) {
1346 return drv->bdrv_get_allocated_file_size(bs);
1347 }
1348 if (bs->file) {
1349 return bdrv_get_allocated_file_size(bs->file);
1350 }
1351 return -ENOTSUP;
1352}
1353
83f64091
FB
1354/**
1355 * Length of a file in bytes. Return < 0 if error or unknown.
1356 */
1357int64_t bdrv_getlength(BlockDriverState *bs)
1358{
1359 BlockDriver *drv = bs->drv;
1360 if (!drv)
19cb3738 1361 return -ENOMEDIUM;
51762288 1362
2c6942fa 1363 if (bs->growable || bdrv_dev_has_removable_media(bs)) {
46a4e4e6
SH
1364 if (drv->bdrv_getlength) {
1365 return drv->bdrv_getlength(bs);
1366 }
83f64091 1367 }
46a4e4e6 1368 return bs->total_sectors * BDRV_SECTOR_SIZE;
fc01f7e7
FB
1369}
1370
19cb3738 1371/* return 0 as number of sectors if no device present or error */
96b8f136 1372void bdrv_get_geometry(BlockDriverState *bs, uint64_t *nb_sectors_ptr)
fc01f7e7 1373{
19cb3738
FB
1374 int64_t length;
1375 length = bdrv_getlength(bs);
1376 if (length < 0)
1377 length = 0;
1378 else
6ea44308 1379 length = length >> BDRV_SECTOR_BITS;
19cb3738 1380 *nb_sectors_ptr = length;
fc01f7e7 1381}
cf98951b 1382
f3d54fc4
AL
1383struct partition {
1384 uint8_t boot_ind; /* 0x80 - active */
1385 uint8_t head; /* starting head */
1386 uint8_t sector; /* starting sector */
1387 uint8_t cyl; /* starting cylinder */
1388 uint8_t sys_ind; /* What partition type */
1389 uint8_t end_head; /* end head */
1390 uint8_t end_sector; /* end sector */
1391 uint8_t end_cyl; /* end cylinder */
1392 uint32_t start_sect; /* starting sector counting from 0 */
1393 uint32_t nr_sects; /* nr of sectors in partition */
541dc0d4 1394} QEMU_PACKED;
f3d54fc4
AL
1395
1396/* try to guess the disk logical geometry from the MSDOS partition table. Return 0 if OK, -1 if could not guess */
1397static int guess_disk_lchs(BlockDriverState *bs,
1398 int *pcylinders, int *pheads, int *psectors)
1399{
eb5a3165 1400 uint8_t buf[BDRV_SECTOR_SIZE];
f3d54fc4
AL
1401 int ret, i, heads, sectors, cylinders;
1402 struct partition *p;
1403 uint32_t nr_sects;
a38131b6 1404 uint64_t nb_sectors;
f3d54fc4
AL
1405
1406 bdrv_get_geometry(bs, &nb_sectors);
1407
1408 ret = bdrv_read(bs, 0, buf, 1);
1409 if (ret < 0)
1410 return -1;
1411 /* test msdos magic */
1412 if (buf[510] != 0x55 || buf[511] != 0xaa)
1413 return -1;
1414 for(i = 0; i < 4; i++) {
1415 p = ((struct partition *)(buf + 0x1be)) + i;
1416 nr_sects = le32_to_cpu(p->nr_sects);
1417 if (nr_sects && p->end_head) {
1418 /* We make the assumption that the partition terminates on
1419 a cylinder boundary */
1420 heads = p->end_head + 1;
1421 sectors = p->end_sector & 63;
1422 if (sectors == 0)
1423 continue;
1424 cylinders = nb_sectors / (heads * sectors);
1425 if (cylinders < 1 || cylinders > 16383)
1426 continue;
1427 *pheads = heads;
1428 *psectors = sectors;
1429 *pcylinders = cylinders;
1430#if 0
1431 printf("guessed geometry: LCHS=%d %d %d\n",
1432 cylinders, heads, sectors);
1433#endif
1434 return 0;
1435 }
1436 }
1437 return -1;
1438}
1439
1440void bdrv_guess_geometry(BlockDriverState *bs, int *pcyls, int *pheads, int *psecs)
1441{
1442 int translation, lba_detected = 0;
1443 int cylinders, heads, secs;
a38131b6 1444 uint64_t nb_sectors;
f3d54fc4
AL
1445
1446 /* if a geometry hint is available, use it */
1447 bdrv_get_geometry(bs, &nb_sectors);
1448 bdrv_get_geometry_hint(bs, &cylinders, &heads, &secs);
1449 translation = bdrv_get_translation_hint(bs);
1450 if (cylinders != 0) {
1451 *pcyls = cylinders;
1452 *pheads = heads;
1453 *psecs = secs;
1454 } else {
1455 if (guess_disk_lchs(bs, &cylinders, &heads, &secs) == 0) {
1456 if (heads > 16) {
1457 /* if heads > 16, it means that a BIOS LBA
1458 translation was active, so the default
1459 hardware geometry is OK */
1460 lba_detected = 1;
1461 goto default_geometry;
1462 } else {
1463 *pcyls = cylinders;
1464 *pheads = heads;
1465 *psecs = secs;
1466 /* disable any translation to be in sync with
1467 the logical geometry */
1468 if (translation == BIOS_ATA_TRANSLATION_AUTO) {
1469 bdrv_set_translation_hint(bs,
1470 BIOS_ATA_TRANSLATION_NONE);
1471 }
1472 }
1473 } else {
1474 default_geometry:
1475 /* if no geometry, use a standard physical disk geometry */
1476 cylinders = nb_sectors / (16 * 63);
1477
1478 if (cylinders > 16383)
1479 cylinders = 16383;
1480 else if (cylinders < 2)
1481 cylinders = 2;
1482 *pcyls = cylinders;
1483 *pheads = 16;
1484 *psecs = 63;
1485 if ((lba_detected == 1) && (translation == BIOS_ATA_TRANSLATION_AUTO)) {
1486 if ((*pcyls * *pheads) <= 131072) {
1487 bdrv_set_translation_hint(bs,
1488 BIOS_ATA_TRANSLATION_LARGE);
1489 } else {
1490 bdrv_set_translation_hint(bs,
1491 BIOS_ATA_TRANSLATION_LBA);
1492 }
1493 }
1494 }
1495 bdrv_set_geometry_hint(bs, *pcyls, *pheads, *psecs);
1496 }
1497}
1498
5fafdf24 1499void bdrv_set_geometry_hint(BlockDriverState *bs,
b338082b
FB
1500 int cyls, int heads, int secs)
1501{
1502 bs->cyls = cyls;
1503 bs->heads = heads;
1504 bs->secs = secs;
1505}
1506
46d4767d
FB
1507void bdrv_set_translation_hint(BlockDriverState *bs, int translation)
1508{
1509 bs->translation = translation;
1510}
1511
5fafdf24 1512void bdrv_get_geometry_hint(BlockDriverState *bs,
b338082b
FB
1513 int *pcyls, int *pheads, int *psecs)
1514{
1515 *pcyls = bs->cyls;
1516 *pheads = bs->heads;
1517 *psecs = bs->secs;
1518}
1519
5bbdbb46
BS
1520/* Recognize floppy formats */
1521typedef struct FDFormat {
1522 FDriveType drive;
1523 uint8_t last_sect;
1524 uint8_t max_track;
1525 uint8_t max_head;
1526} FDFormat;
1527
1528static const FDFormat fd_formats[] = {
1529 /* First entry is default format */
1530 /* 1.44 MB 3"1/2 floppy disks */
1531 { FDRIVE_DRV_144, 18, 80, 1, },
1532 { FDRIVE_DRV_144, 20, 80, 1, },
1533 { FDRIVE_DRV_144, 21, 80, 1, },
1534 { FDRIVE_DRV_144, 21, 82, 1, },
1535 { FDRIVE_DRV_144, 21, 83, 1, },
1536 { FDRIVE_DRV_144, 22, 80, 1, },
1537 { FDRIVE_DRV_144, 23, 80, 1, },
1538 { FDRIVE_DRV_144, 24, 80, 1, },
1539 /* 2.88 MB 3"1/2 floppy disks */
1540 { FDRIVE_DRV_288, 36, 80, 1, },
1541 { FDRIVE_DRV_288, 39, 80, 1, },
1542 { FDRIVE_DRV_288, 40, 80, 1, },
1543 { FDRIVE_DRV_288, 44, 80, 1, },
1544 { FDRIVE_DRV_288, 48, 80, 1, },
1545 /* 720 kB 3"1/2 floppy disks */
1546 { FDRIVE_DRV_144, 9, 80, 1, },
1547 { FDRIVE_DRV_144, 10, 80, 1, },
1548 { FDRIVE_DRV_144, 10, 82, 1, },
1549 { FDRIVE_DRV_144, 10, 83, 1, },
1550 { FDRIVE_DRV_144, 13, 80, 1, },
1551 { FDRIVE_DRV_144, 14, 80, 1, },
1552 /* 1.2 MB 5"1/4 floppy disks */
1553 { FDRIVE_DRV_120, 15, 80, 1, },
1554 { FDRIVE_DRV_120, 18, 80, 1, },
1555 { FDRIVE_DRV_120, 18, 82, 1, },
1556 { FDRIVE_DRV_120, 18, 83, 1, },
1557 { FDRIVE_DRV_120, 20, 80, 1, },
1558 /* 720 kB 5"1/4 floppy disks */
1559 { FDRIVE_DRV_120, 9, 80, 1, },
1560 { FDRIVE_DRV_120, 11, 80, 1, },
1561 /* 360 kB 5"1/4 floppy disks */
1562 { FDRIVE_DRV_120, 9, 40, 1, },
1563 { FDRIVE_DRV_120, 9, 40, 0, },
1564 { FDRIVE_DRV_120, 10, 41, 1, },
1565 { FDRIVE_DRV_120, 10, 42, 1, },
1566 /* 320 kB 5"1/4 floppy disks */
1567 { FDRIVE_DRV_120, 8, 40, 1, },
1568 { FDRIVE_DRV_120, 8, 40, 0, },
1569 /* 360 kB must match 5"1/4 better than 3"1/2... */
1570 { FDRIVE_DRV_144, 9, 80, 0, },
1571 /* end */
1572 { FDRIVE_DRV_NONE, -1, -1, 0, },
1573};
1574
1575void bdrv_get_floppy_geometry_hint(BlockDriverState *bs, int *nb_heads,
1576 int *max_track, int *last_sect,
1577 FDriveType drive_in, FDriveType *drive)
1578{
1579 const FDFormat *parse;
1580 uint64_t nb_sectors, size;
1581 int i, first_match, match;
1582
1583 bdrv_get_geometry_hint(bs, nb_heads, max_track, last_sect);
1584 if (*nb_heads != 0 && *max_track != 0 && *last_sect != 0) {
1585 /* User defined disk */
1586 } else {
1587 bdrv_get_geometry(bs, &nb_sectors);
1588 match = -1;
1589 first_match = -1;
1590 for (i = 0; ; i++) {
1591 parse = &fd_formats[i];
1592 if (parse->drive == FDRIVE_DRV_NONE) {
1593 break;
1594 }
1595 if (drive_in == parse->drive ||
1596 drive_in == FDRIVE_DRV_NONE) {
1597 size = (parse->max_head + 1) * parse->max_track *
1598 parse->last_sect;
1599 if (nb_sectors == size) {
1600 match = i;
1601 break;
1602 }
1603 if (first_match == -1) {
1604 first_match = i;
1605 }
1606 }
1607 }
1608 if (match == -1) {
1609 if (first_match == -1) {
1610 match = 1;
1611 } else {
1612 match = first_match;
1613 }
1614 parse = &fd_formats[match];
1615 }
1616 *nb_heads = parse->max_head + 1;
1617 *max_track = parse->max_track;
1618 *last_sect = parse->last_sect;
1619 *drive = parse->drive;
1620 }
1621}
1622
46d4767d
FB
1623int bdrv_get_translation_hint(BlockDriverState *bs)
1624{
1625 return bs->translation;
1626}
1627
abd7f68d
MA
1628void bdrv_set_on_error(BlockDriverState *bs, BlockErrorAction on_read_error,
1629 BlockErrorAction on_write_error)
1630{
1631 bs->on_read_error = on_read_error;
1632 bs->on_write_error = on_write_error;
1633}
1634
1635BlockErrorAction bdrv_get_on_error(BlockDriverState *bs, int is_read)
1636{
1637 return is_read ? bs->on_read_error : bs->on_write_error;
1638}
1639
b338082b
FB
1640int bdrv_is_read_only(BlockDriverState *bs)
1641{
1642 return bs->read_only;
1643}
1644
985a03b0
TS
1645int bdrv_is_sg(BlockDriverState *bs)
1646{
1647 return bs->sg;
1648}
1649
e900a7b7
CH
1650int bdrv_enable_write_cache(BlockDriverState *bs)
1651{
1652 return bs->enable_write_cache;
1653}
1654
ea2384d3
FB
1655int bdrv_is_encrypted(BlockDriverState *bs)
1656{
1657 if (bs->backing_hd && bs->backing_hd->encrypted)
1658 return 1;
1659 return bs->encrypted;
1660}
1661
c0f4ce77
AL
1662int bdrv_key_required(BlockDriverState *bs)
1663{
1664 BlockDriverState *backing_hd = bs->backing_hd;
1665
1666 if (backing_hd && backing_hd->encrypted && !backing_hd->valid_key)
1667 return 1;
1668 return (bs->encrypted && !bs->valid_key);
1669}
1670
ea2384d3
FB
1671int bdrv_set_key(BlockDriverState *bs, const char *key)
1672{
1673 int ret;
1674 if (bs->backing_hd && bs->backing_hd->encrypted) {
1675 ret = bdrv_set_key(bs->backing_hd, key);
1676 if (ret < 0)
1677 return ret;
1678 if (!bs->encrypted)
1679 return 0;
1680 }
fd04a2ae
SH
1681 if (!bs->encrypted) {
1682 return -EINVAL;
1683 } else if (!bs->drv || !bs->drv->bdrv_set_key) {
1684 return -ENOMEDIUM;
1685 }
c0f4ce77 1686 ret = bs->drv->bdrv_set_key(bs, key);
bb5fc20f
AL
1687 if (ret < 0) {
1688 bs->valid_key = 0;
1689 } else if (!bs->valid_key) {
1690 bs->valid_key = 1;
1691 /* call the change callback now, we skipped it on open */
7d4b4ba5 1692 bdrv_dev_change_media_cb(bs, true);
bb5fc20f 1693 }
c0f4ce77 1694 return ret;
ea2384d3
FB
1695}
1696
1697void bdrv_get_format(BlockDriverState *bs, char *buf, int buf_size)
1698{
19cb3738 1699 if (!bs->drv) {
ea2384d3
FB
1700 buf[0] = '\0';
1701 } else {
1702 pstrcpy(buf, buf_size, bs->drv->format_name);
1703 }
1704}
1705
5fafdf24 1706void bdrv_iterate_format(void (*it)(void *opaque, const char *name),
ea2384d3
FB
1707 void *opaque)
1708{
1709 BlockDriver *drv;
1710
8a22f02a 1711 QLIST_FOREACH(drv, &bdrv_drivers, list) {
ea2384d3
FB
1712 it(opaque, drv->format_name);
1713 }
1714}
1715
b338082b
FB
1716BlockDriverState *bdrv_find(const char *name)
1717{
1718 BlockDriverState *bs;
1719
1b7bdbc1
SH
1720 QTAILQ_FOREACH(bs, &bdrv_states, list) {
1721 if (!strcmp(name, bs->device_name)) {
b338082b 1722 return bs;
1b7bdbc1 1723 }
b338082b
FB
1724 }
1725 return NULL;
1726}
1727
2f399b0a
MA
1728BlockDriverState *bdrv_next(BlockDriverState *bs)
1729{
1730 if (!bs) {
1731 return QTAILQ_FIRST(&bdrv_states);
1732 }
1733 return QTAILQ_NEXT(bs, list);
1734}
1735
51de9760 1736void bdrv_iterate(void (*it)(void *opaque, BlockDriverState *bs), void *opaque)
81d0912d
FB
1737{
1738 BlockDriverState *bs;
1739
1b7bdbc1 1740 QTAILQ_FOREACH(bs, &bdrv_states, list) {
51de9760 1741 it(opaque, bs);
81d0912d
FB
1742 }
1743}
1744
ea2384d3
FB
1745const char *bdrv_get_device_name(BlockDriverState *bs)
1746{
1747 return bs->device_name;
1748}
1749
c6ca28d6
AL
1750void bdrv_flush_all(void)
1751{
1752 BlockDriverState *bs;
1753
1b7bdbc1 1754 QTAILQ_FOREACH(bs, &bdrv_states, list) {
c602a489 1755 if (!bdrv_is_read_only(bs) && bdrv_is_inserted(bs)) {
c6ca28d6 1756 bdrv_flush(bs);
1b7bdbc1
SH
1757 }
1758 }
c6ca28d6
AL
1759}
1760
f2feebbd
KW
1761int bdrv_has_zero_init(BlockDriverState *bs)
1762{
1763 assert(bs->drv);
1764
336c1c12
KW
1765 if (bs->drv->bdrv_has_zero_init) {
1766 return bs->drv->bdrv_has_zero_init(bs);
f2feebbd
KW
1767 }
1768
1769 return 1;
1770}
1771
f58c7b35
TS
1772/*
1773 * Returns true iff the specified sector is present in the disk image. Drivers
1774 * not implementing the functionality are assumed to not support backing files,
1775 * hence all their sectors are reported as allocated.
1776 *
1777 * 'pnum' is set to the number of sectors (including and immediately following
1778 * the specified sector) that are known to be in the same
1779 * allocated/unallocated state.
1780 *
1781 * 'nb_sectors' is the max value 'pnum' should be set to.
1782 */
1783int bdrv_is_allocated(BlockDriverState *bs, int64_t sector_num, int nb_sectors,
1784 int *pnum)
1785{
1786 int64_t n;
1787 if (!bs->drv->bdrv_is_allocated) {
1788 if (sector_num >= bs->total_sectors) {
1789 *pnum = 0;
1790 return 0;
1791 }
1792 n = bs->total_sectors - sector_num;
1793 *pnum = (n < nb_sectors) ? (n) : (nb_sectors);
1794 return 1;
1795 }
1796 return bs->drv->bdrv_is_allocated(bs, sector_num, nb_sectors, pnum);
1797}
1798
2582bfed
LC
1799void bdrv_mon_event(const BlockDriverState *bdrv,
1800 BlockMonEventAction action, int is_read)
1801{
1802 QObject *data;
1803 const char *action_str;
1804
1805 switch (action) {
1806 case BDRV_ACTION_REPORT:
1807 action_str = "report";
1808 break;
1809 case BDRV_ACTION_IGNORE:
1810 action_str = "ignore";
1811 break;
1812 case BDRV_ACTION_STOP:
1813 action_str = "stop";
1814 break;
1815 default:
1816 abort();
1817 }
1818
1819 data = qobject_from_jsonf("{ 'device': %s, 'action': %s, 'operation': %s }",
1820 bdrv->device_name,
1821 action_str,
1822 is_read ? "read" : "write");
1823 monitor_protocol_event(QEVENT_BLOCK_IO_ERROR, data);
1824
1825 qobject_decref(data);
1826}
1827
b2023818 1828BlockInfoList *qmp_query_block(Error **errp)
b338082b 1829{
b2023818 1830 BlockInfoList *head = NULL, *cur_item = NULL;
b338082b
FB
1831 BlockDriverState *bs;
1832
1b7bdbc1 1833 QTAILQ_FOREACH(bs, &bdrv_states, list) {
b2023818 1834 BlockInfoList *info = g_malloc0(sizeof(*info));
d15e5465 1835
b2023818
LC
1836 info->value = g_malloc0(sizeof(*info->value));
1837 info->value->device = g_strdup(bs->device_name);
1838 info->value->type = g_strdup("unknown");
1839 info->value->locked = bdrv_dev_is_medium_locked(bs);
1840 info->value->removable = bdrv_dev_has_removable_media(bs);
d15e5465 1841
e4def80b 1842 if (bdrv_dev_has_removable_media(bs)) {
b2023818
LC
1843 info->value->has_tray_open = true;
1844 info->value->tray_open = bdrv_dev_is_tray_open(bs);
e4def80b 1845 }
f04ef601
LC
1846
1847 if (bdrv_iostatus_is_enabled(bs)) {
b2023818
LC
1848 info->value->has_io_status = true;
1849 info->value->io_status = bs->iostatus;
f04ef601
LC
1850 }
1851
19cb3738 1852 if (bs->drv) {
b2023818
LC
1853 info->value->has_inserted = true;
1854 info->value->inserted = g_malloc0(sizeof(*info->value->inserted));
1855 info->value->inserted->file = g_strdup(bs->filename);
1856 info->value->inserted->ro = bs->read_only;
1857 info->value->inserted->drv = g_strdup(bs->drv->format_name);
1858 info->value->inserted->encrypted = bs->encrypted;
1859 if (bs->backing_file[0]) {
1860 info->value->inserted->has_backing_file = true;
1861 info->value->inserted->backing_file = g_strdup(bs->backing_file);
376253ec 1862 }
b2023818 1863 }
d15e5465 1864
b2023818
LC
1865 /* XXX: waiting for the qapi to support GSList */
1866 if (!cur_item) {
1867 head = cur_item = info;
1868 } else {
1869 cur_item->next = info;
1870 cur_item = info;
b338082b 1871 }
b338082b 1872 }
d15e5465 1873
b2023818 1874 return head;
b338082b 1875}
a36e69dd 1876
218a536a 1877static void bdrv_stats_iter(QObject *data, void *opaque)
a36e69dd 1878{
218a536a
LC
1879 QDict *qdict;
1880 Monitor *mon = opaque;
1881
1882 qdict = qobject_to_qdict(data);
1883 monitor_printf(mon, "%s:", qdict_get_str(qdict, "device"));
1884
1885 qdict = qobject_to_qdict(qdict_get(qdict, "stats"));
1886 monitor_printf(mon, " rd_bytes=%" PRId64
1887 " wr_bytes=%" PRId64
1888 " rd_operations=%" PRId64
1889 " wr_operations=%" PRId64
e8045d67 1890 " flush_operations=%" PRId64
c488c7f6
CH
1891 " wr_total_time_ns=%" PRId64
1892 " rd_total_time_ns=%" PRId64
1893 " flush_total_time_ns=%" PRId64
218a536a
LC
1894 "\n",
1895 qdict_get_int(qdict, "rd_bytes"),
1896 qdict_get_int(qdict, "wr_bytes"),
1897 qdict_get_int(qdict, "rd_operations"),
e8045d67 1898 qdict_get_int(qdict, "wr_operations"),
c488c7f6
CH
1899 qdict_get_int(qdict, "flush_operations"),
1900 qdict_get_int(qdict, "wr_total_time_ns"),
1901 qdict_get_int(qdict, "rd_total_time_ns"),
1902 qdict_get_int(qdict, "flush_total_time_ns"));
218a536a
LC
1903}
1904
1905void bdrv_stats_print(Monitor *mon, const QObject *data)
1906{
1907 qlist_iter(qobject_to_qlist(data), bdrv_stats_iter, mon);
1908}
1909
294cc35f
KW
1910static QObject* bdrv_info_stats_bs(BlockDriverState *bs)
1911{
1912 QObject *res;
1913 QDict *dict;
1914
1915 res = qobject_from_jsonf("{ 'stats': {"
1916 "'rd_bytes': %" PRId64 ","
1917 "'wr_bytes': %" PRId64 ","
1918 "'rd_operations': %" PRId64 ","
1919 "'wr_operations': %" PRId64 ","
e8045d67 1920 "'wr_highest_offset': %" PRId64 ","
c488c7f6
CH
1921 "'flush_operations': %" PRId64 ","
1922 "'wr_total_time_ns': %" PRId64 ","
1923 "'rd_total_time_ns': %" PRId64 ","
1924 "'flush_total_time_ns': %" PRId64
294cc35f 1925 "} }",
a597e79c
CH
1926 bs->nr_bytes[BDRV_ACCT_READ],
1927 bs->nr_bytes[BDRV_ACCT_WRITE],
1928 bs->nr_ops[BDRV_ACCT_READ],
1929 bs->nr_ops[BDRV_ACCT_WRITE],
5ffbbc67 1930 bs->wr_highest_sector *
e8045d67 1931 (uint64_t)BDRV_SECTOR_SIZE,
c488c7f6
CH
1932 bs->nr_ops[BDRV_ACCT_FLUSH],
1933 bs->total_time_ns[BDRV_ACCT_WRITE],
1934 bs->total_time_ns[BDRV_ACCT_READ],
1935 bs->total_time_ns[BDRV_ACCT_FLUSH]);
294cc35f
KW
1936 dict = qobject_to_qdict(res);
1937
1938 if (*bs->device_name) {
1939 qdict_put(dict, "device", qstring_from_str(bs->device_name));
1940 }
1941
1942 if (bs->file) {
1943 QObject *parent = bdrv_info_stats_bs(bs->file);
1944 qdict_put_obj(dict, "parent", parent);
1945 }
1946
1947 return res;
1948}
1949
218a536a
LC
1950void bdrv_info_stats(Monitor *mon, QObject **ret_data)
1951{
1952 QObject *obj;
1953 QList *devices;
a36e69dd
TS
1954 BlockDriverState *bs;
1955
218a536a
LC
1956 devices = qlist_new();
1957
1b7bdbc1 1958 QTAILQ_FOREACH(bs, &bdrv_states, list) {
294cc35f 1959 obj = bdrv_info_stats_bs(bs);
218a536a 1960 qlist_append_obj(devices, obj);
a36e69dd 1961 }
218a536a
LC
1962
1963 *ret_data = QOBJECT(devices);
a36e69dd 1964}
ea2384d3 1965
045df330
AL
1966const char *bdrv_get_encrypted_filename(BlockDriverState *bs)
1967{
1968 if (bs->backing_hd && bs->backing_hd->encrypted)
1969 return bs->backing_file;
1970 else if (bs->encrypted)
1971 return bs->filename;
1972 else
1973 return NULL;
1974}
1975
5fafdf24 1976void bdrv_get_backing_filename(BlockDriverState *bs,
83f64091
FB
1977 char *filename, int filename_size)
1978{
b783e409 1979 if (!bs->backing_file) {
83f64091
FB
1980 pstrcpy(filename, filename_size, "");
1981 } else {
1982 pstrcpy(filename, filename_size, bs->backing_file);
1983 }
1984}
1985
5fafdf24 1986int bdrv_write_compressed(BlockDriverState *bs, int64_t sector_num,
faea38e7
FB
1987 const uint8_t *buf, int nb_sectors)
1988{
1989 BlockDriver *drv = bs->drv;
1990 if (!drv)
19cb3738 1991 return -ENOMEDIUM;
faea38e7
FB
1992 if (!drv->bdrv_write_compressed)
1993 return -ENOTSUP;
fbb7b4e0
KW
1994 if (bdrv_check_request(bs, sector_num, nb_sectors))
1995 return -EIO;
a55eb92c 1996
c6d22830 1997 if (bs->dirty_bitmap) {
7cd1e32a
LS
1998 set_dirty_bitmap(bs, sector_num, nb_sectors, 1);
1999 }
a55eb92c 2000
faea38e7
FB
2001 return drv->bdrv_write_compressed(bs, sector_num, buf, nb_sectors);
2002}
3b46e624 2003
faea38e7
FB
2004int bdrv_get_info(BlockDriverState *bs, BlockDriverInfo *bdi)
2005{
2006 BlockDriver *drv = bs->drv;
2007 if (!drv)
19cb3738 2008 return -ENOMEDIUM;
faea38e7
FB
2009 if (!drv->bdrv_get_info)
2010 return -ENOTSUP;
2011 memset(bdi, 0, sizeof(*bdi));
2012 return drv->bdrv_get_info(bs, bdi);
2013}
2014
45566e9c
CH
2015int bdrv_save_vmstate(BlockDriverState *bs, const uint8_t *buf,
2016 int64_t pos, int size)
178e08a5
AL
2017{
2018 BlockDriver *drv = bs->drv;
2019 if (!drv)
2020 return -ENOMEDIUM;
7cdb1f6d
MK
2021 if (drv->bdrv_save_vmstate)
2022 return drv->bdrv_save_vmstate(bs, buf, pos, size);
2023 if (bs->file)
2024 return bdrv_save_vmstate(bs->file, buf, pos, size);
2025 return -ENOTSUP;
178e08a5
AL
2026}
2027
45566e9c
CH
2028int bdrv_load_vmstate(BlockDriverState *bs, uint8_t *buf,
2029 int64_t pos, int size)
178e08a5
AL
2030{
2031 BlockDriver *drv = bs->drv;
2032 if (!drv)
2033 return -ENOMEDIUM;
7cdb1f6d
MK
2034 if (drv->bdrv_load_vmstate)
2035 return drv->bdrv_load_vmstate(bs, buf, pos, size);
2036 if (bs->file)
2037 return bdrv_load_vmstate(bs->file, buf, pos, size);
2038 return -ENOTSUP;
178e08a5
AL
2039}
2040
8b9b0cc2
KW
2041void bdrv_debug_event(BlockDriverState *bs, BlkDebugEvent event)
2042{
2043 BlockDriver *drv = bs->drv;
2044
2045 if (!drv || !drv->bdrv_debug_event) {
2046 return;
2047 }
2048
2049 return drv->bdrv_debug_event(bs, event);
2050
2051}
2052
faea38e7
FB
2053/**************************************************************/
2054/* handling of snapshots */
2055
feeee5ac
MDCF
2056int bdrv_can_snapshot(BlockDriverState *bs)
2057{
2058 BlockDriver *drv = bs->drv;
07b70bfb 2059 if (!drv || !bdrv_is_inserted(bs) || bdrv_is_read_only(bs)) {
feeee5ac
MDCF
2060 return 0;
2061 }
2062
2063 if (!drv->bdrv_snapshot_create) {
2064 if (bs->file != NULL) {
2065 return bdrv_can_snapshot(bs->file);
2066 }
2067 return 0;
2068 }
2069
2070 return 1;
2071}
2072
199630b6
BS
2073int bdrv_is_snapshot(BlockDriverState *bs)
2074{
2075 return !!(bs->open_flags & BDRV_O_SNAPSHOT);
2076}
2077
f9092b10
MA
2078BlockDriverState *bdrv_snapshots(void)
2079{
2080 BlockDriverState *bs;
2081
3ac906f7 2082 if (bs_snapshots) {
f9092b10 2083 return bs_snapshots;
3ac906f7 2084 }
f9092b10
MA
2085
2086 bs = NULL;
2087 while ((bs = bdrv_next(bs))) {
2088 if (bdrv_can_snapshot(bs)) {
3ac906f7
MA
2089 bs_snapshots = bs;
2090 return bs;
f9092b10
MA
2091 }
2092 }
2093 return NULL;
f9092b10
MA
2094}
2095
5fafdf24 2096int bdrv_snapshot_create(BlockDriverState *bs,
faea38e7
FB
2097 QEMUSnapshotInfo *sn_info)
2098{
2099 BlockDriver *drv = bs->drv;
2100 if (!drv)
19cb3738 2101 return -ENOMEDIUM;
7cdb1f6d
MK
2102 if (drv->bdrv_snapshot_create)
2103 return drv->bdrv_snapshot_create(bs, sn_info);
2104 if (bs->file)
2105 return bdrv_snapshot_create(bs->file, sn_info);
2106 return -ENOTSUP;
faea38e7
FB
2107}
2108
5fafdf24 2109int bdrv_snapshot_goto(BlockDriverState *bs,
faea38e7
FB
2110 const char *snapshot_id)
2111{
2112 BlockDriver *drv = bs->drv;
7cdb1f6d
MK
2113 int ret, open_ret;
2114
faea38e7 2115 if (!drv)
19cb3738 2116 return -ENOMEDIUM;
7cdb1f6d
MK
2117 if (drv->bdrv_snapshot_goto)
2118 return drv->bdrv_snapshot_goto(bs, snapshot_id);
2119
2120 if (bs->file) {
2121 drv->bdrv_close(bs);
2122 ret = bdrv_snapshot_goto(bs->file, snapshot_id);
2123 open_ret = drv->bdrv_open(bs, bs->open_flags);
2124 if (open_ret < 0) {
2125 bdrv_delete(bs->file);
2126 bs->drv = NULL;
2127 return open_ret;
2128 }
2129 return ret;
2130 }
2131
2132 return -ENOTSUP;
faea38e7
FB
2133}
2134
2135int bdrv_snapshot_delete(BlockDriverState *bs, const char *snapshot_id)
2136{
2137 BlockDriver *drv = bs->drv;
2138 if (!drv)
19cb3738 2139 return -ENOMEDIUM;
7cdb1f6d
MK
2140 if (drv->bdrv_snapshot_delete)
2141 return drv->bdrv_snapshot_delete(bs, snapshot_id);
2142 if (bs->file)
2143 return bdrv_snapshot_delete(bs->file, snapshot_id);
2144 return -ENOTSUP;
faea38e7
FB
2145}
2146
5fafdf24 2147int bdrv_snapshot_list(BlockDriverState *bs,
faea38e7
FB
2148 QEMUSnapshotInfo **psn_info)
2149{
2150 BlockDriver *drv = bs->drv;
2151 if (!drv)
19cb3738 2152 return -ENOMEDIUM;
7cdb1f6d
MK
2153 if (drv->bdrv_snapshot_list)
2154 return drv->bdrv_snapshot_list(bs, psn_info);
2155 if (bs->file)
2156 return bdrv_snapshot_list(bs->file, psn_info);
2157 return -ENOTSUP;
faea38e7
FB
2158}
2159
51ef6727 2160int bdrv_snapshot_load_tmp(BlockDriverState *bs,
2161 const char *snapshot_name)
2162{
2163 BlockDriver *drv = bs->drv;
2164 if (!drv) {
2165 return -ENOMEDIUM;
2166 }
2167 if (!bs->read_only) {
2168 return -EINVAL;
2169 }
2170 if (drv->bdrv_snapshot_load_tmp) {
2171 return drv->bdrv_snapshot_load_tmp(bs, snapshot_name);
2172 }
2173 return -ENOTSUP;
2174}
2175
faea38e7
FB
2176#define NB_SUFFIXES 4
2177
2178char *get_human_readable_size(char *buf, int buf_size, int64_t size)
2179{
2180 static const char suffixes[NB_SUFFIXES] = "KMGT";
2181 int64_t base;
2182 int i;
2183
2184 if (size <= 999) {
2185 snprintf(buf, buf_size, "%" PRId64, size);
2186 } else {
2187 base = 1024;
2188 for(i = 0; i < NB_SUFFIXES; i++) {
2189 if (size < (10 * base)) {
5fafdf24 2190 snprintf(buf, buf_size, "%0.1f%c",
faea38e7
FB
2191 (double)size / base,
2192 suffixes[i]);
2193 break;
2194 } else if (size < (1000 * base) || i == (NB_SUFFIXES - 1)) {
5fafdf24 2195 snprintf(buf, buf_size, "%" PRId64 "%c",
faea38e7
FB
2196 ((size + (base >> 1)) / base),
2197 suffixes[i]);
2198 break;
2199 }
2200 base = base * 1024;
2201 }
2202 }
2203 return buf;
2204}
2205
2206char *bdrv_snapshot_dump(char *buf, int buf_size, QEMUSnapshotInfo *sn)
2207{
2208 char buf1[128], date_buf[128], clock_buf[128];
3b9f94e1
FB
2209#ifdef _WIN32
2210 struct tm *ptm;
2211#else
faea38e7 2212 struct tm tm;
3b9f94e1 2213#endif
faea38e7
FB
2214 time_t ti;
2215 int64_t secs;
2216
2217 if (!sn) {
5fafdf24
TS
2218 snprintf(buf, buf_size,
2219 "%-10s%-20s%7s%20s%15s",
faea38e7
FB
2220 "ID", "TAG", "VM SIZE", "DATE", "VM CLOCK");
2221 } else {
2222 ti = sn->date_sec;
3b9f94e1
FB
2223#ifdef _WIN32
2224 ptm = localtime(&ti);
2225 strftime(date_buf, sizeof(date_buf),
2226 "%Y-%m-%d %H:%M:%S", ptm);
2227#else
faea38e7
FB
2228 localtime_r(&ti, &tm);
2229 strftime(date_buf, sizeof(date_buf),
2230 "%Y-%m-%d %H:%M:%S", &tm);
3b9f94e1 2231#endif
faea38e7
FB
2232 secs = sn->vm_clock_nsec / 1000000000;
2233 snprintf(clock_buf, sizeof(clock_buf),
2234 "%02d:%02d:%02d.%03d",
2235 (int)(secs / 3600),
2236 (int)((secs / 60) % 60),
5fafdf24 2237 (int)(secs % 60),
faea38e7
FB
2238 (int)((sn->vm_clock_nsec / 1000000) % 1000));
2239 snprintf(buf, buf_size,
5fafdf24 2240 "%-10s%-20s%7s%20s%15s",
faea38e7
FB
2241 sn->id_str, sn->name,
2242 get_human_readable_size(buf1, sizeof(buf1), sn->vm_state_size),
2243 date_buf,
2244 clock_buf);
2245 }
2246 return buf;
2247}
2248
ea2384d3 2249/**************************************************************/
83f64091 2250/* async I/Os */
ea2384d3 2251
3b69e4b9 2252BlockDriverAIOCB *bdrv_aio_readv(BlockDriverState *bs, int64_t sector_num,
f141eafe 2253 QEMUIOVector *qiov, int nb_sectors,
3b69e4b9 2254 BlockDriverCompletionFunc *cb, void *opaque)
83f64091 2255{
bbf0a440
SH
2256 trace_bdrv_aio_readv(bs, sector_num, nb_sectors, opaque);
2257
b2a61371 2258 return bdrv_co_aio_rw_vector(bs, sector_num, qiov, nb_sectors,
8c5873d6 2259 cb, opaque, false);
ea2384d3
FB
2260}
2261
f141eafe
AL
2262BlockDriverAIOCB *bdrv_aio_writev(BlockDriverState *bs, int64_t sector_num,
2263 QEMUIOVector *qiov, int nb_sectors,
2264 BlockDriverCompletionFunc *cb, void *opaque)
ea2384d3 2265{
bbf0a440
SH
2266 trace_bdrv_aio_writev(bs, sector_num, nb_sectors, opaque);
2267
1a6e115b 2268 return bdrv_co_aio_rw_vector(bs, sector_num, qiov, nb_sectors,
8c5873d6 2269 cb, opaque, true);
83f64091
FB
2270}
2271
40b4f539
KW
2272
2273typedef struct MultiwriteCB {
2274 int error;
2275 int num_requests;
2276 int num_callbacks;
2277 struct {
2278 BlockDriverCompletionFunc *cb;
2279 void *opaque;
2280 QEMUIOVector *free_qiov;
2281 void *free_buf;
2282 } callbacks[];
2283} MultiwriteCB;
2284
2285static void multiwrite_user_cb(MultiwriteCB *mcb)
2286{
2287 int i;
2288
2289 for (i = 0; i < mcb->num_callbacks; i++) {
2290 mcb->callbacks[i].cb(mcb->callbacks[i].opaque, mcb->error);
1e1ea48d
SH
2291 if (mcb->callbacks[i].free_qiov) {
2292 qemu_iovec_destroy(mcb->callbacks[i].free_qiov);
2293 }
7267c094 2294 g_free(mcb->callbacks[i].free_qiov);
f8a83245 2295 qemu_vfree(mcb->callbacks[i].free_buf);
40b4f539
KW
2296 }
2297}
2298
2299static void multiwrite_cb(void *opaque, int ret)
2300{
2301 MultiwriteCB *mcb = opaque;
2302
6d519a5f
SH
2303 trace_multiwrite_cb(mcb, ret);
2304
cb6d3ca0 2305 if (ret < 0 && !mcb->error) {
40b4f539 2306 mcb->error = ret;
40b4f539
KW
2307 }
2308
2309 mcb->num_requests--;
2310 if (mcb->num_requests == 0) {
de189a1b 2311 multiwrite_user_cb(mcb);
7267c094 2312 g_free(mcb);
40b4f539
KW
2313 }
2314}
2315
2316static int multiwrite_req_compare(const void *a, const void *b)
2317{
77be4366
CH
2318 const BlockRequest *req1 = a, *req2 = b;
2319
2320 /*
2321 * Note that we can't simply subtract req2->sector from req1->sector
2322 * here as that could overflow the return value.
2323 */
2324 if (req1->sector > req2->sector) {
2325 return 1;
2326 } else if (req1->sector < req2->sector) {
2327 return -1;
2328 } else {
2329 return 0;
2330 }
40b4f539
KW
2331}
2332
2333/*
2334 * Takes a bunch of requests and tries to merge them. Returns the number of
2335 * requests that remain after merging.
2336 */
2337static int multiwrite_merge(BlockDriverState *bs, BlockRequest *reqs,
2338 int num_reqs, MultiwriteCB *mcb)
2339{
2340 int i, outidx;
2341
2342 // Sort requests by start sector
2343 qsort(reqs, num_reqs, sizeof(*reqs), &multiwrite_req_compare);
2344
2345 // Check if adjacent requests touch the same clusters. If so, combine them,
2346 // filling up gaps with zero sectors.
2347 outidx = 0;
2348 for (i = 1; i < num_reqs; i++) {
2349 int merge = 0;
2350 int64_t oldreq_last = reqs[outidx].sector + reqs[outidx].nb_sectors;
2351
2352 // This handles the cases that are valid for all block drivers, namely
2353 // exactly sequential writes and overlapping writes.
2354 if (reqs[i].sector <= oldreq_last) {
2355 merge = 1;
2356 }
2357
2358 // The block driver may decide that it makes sense to combine requests
2359 // even if there is a gap of some sectors between them. In this case,
2360 // the gap is filled with zeros (therefore only applicable for yet
2361 // unused space in format like qcow2).
2362 if (!merge && bs->drv->bdrv_merge_requests) {
2363 merge = bs->drv->bdrv_merge_requests(bs, &reqs[outidx], &reqs[i]);
2364 }
2365
e2a305fb
CH
2366 if (reqs[outidx].qiov->niov + reqs[i].qiov->niov + 1 > IOV_MAX) {
2367 merge = 0;
2368 }
2369
40b4f539
KW
2370 if (merge) {
2371 size_t size;
7267c094 2372 QEMUIOVector *qiov = g_malloc0(sizeof(*qiov));
40b4f539
KW
2373 qemu_iovec_init(qiov,
2374 reqs[outidx].qiov->niov + reqs[i].qiov->niov + 1);
2375
2376 // Add the first request to the merged one. If the requests are
2377 // overlapping, drop the last sectors of the first request.
2378 size = (reqs[i].sector - reqs[outidx].sector) << 9;
2379 qemu_iovec_concat(qiov, reqs[outidx].qiov, size);
2380
2381 // We might need to add some zeros between the two requests
2382 if (reqs[i].sector > oldreq_last) {
2383 size_t zero_bytes = (reqs[i].sector - oldreq_last) << 9;
2384 uint8_t *buf = qemu_blockalign(bs, zero_bytes);
2385 memset(buf, 0, zero_bytes);
2386 qemu_iovec_add(qiov, buf, zero_bytes);
2387 mcb->callbacks[i].free_buf = buf;
2388 }
2389
2390 // Add the second request
2391 qemu_iovec_concat(qiov, reqs[i].qiov, reqs[i].qiov->size);
2392
cbf1dff2 2393 reqs[outidx].nb_sectors = qiov->size >> 9;
40b4f539
KW
2394 reqs[outidx].qiov = qiov;
2395
2396 mcb->callbacks[i].free_qiov = reqs[outidx].qiov;
2397 } else {
2398 outidx++;
2399 reqs[outidx].sector = reqs[i].sector;
2400 reqs[outidx].nb_sectors = reqs[i].nb_sectors;
2401 reqs[outidx].qiov = reqs[i].qiov;
2402 }
2403 }
2404
2405 return outidx + 1;
2406}
2407
2408/*
2409 * Submit multiple AIO write requests at once.
2410 *
2411 * On success, the function returns 0 and all requests in the reqs array have
2412 * been submitted. In error case this function returns -1, and any of the
2413 * requests may or may not be submitted yet. In particular, this means that the
2414 * callback will be called for some of the requests, for others it won't. The
2415 * caller must check the error field of the BlockRequest to wait for the right
2416 * callbacks (if error != 0, no callback will be called).
2417 *
2418 * The implementation may modify the contents of the reqs array, e.g. to merge
2419 * requests. However, the fields opaque and error are left unmodified as they
2420 * are used to signal failure for a single request to the caller.
2421 */
2422int bdrv_aio_multiwrite(BlockDriverState *bs, BlockRequest *reqs, int num_reqs)
2423{
2424 BlockDriverAIOCB *acb;
2425 MultiwriteCB *mcb;
2426 int i;
2427
301db7c2
RH
2428 /* don't submit writes if we don't have a medium */
2429 if (bs->drv == NULL) {
2430 for (i = 0; i < num_reqs; i++) {
2431 reqs[i].error = -ENOMEDIUM;
2432 }
2433 return -1;
2434 }
2435
40b4f539
KW
2436 if (num_reqs == 0) {
2437 return 0;
2438 }
2439
2440 // Create MultiwriteCB structure
7267c094 2441 mcb = g_malloc0(sizeof(*mcb) + num_reqs * sizeof(*mcb->callbacks));
40b4f539
KW
2442 mcb->num_requests = 0;
2443 mcb->num_callbacks = num_reqs;
2444
2445 for (i = 0; i < num_reqs; i++) {
2446 mcb->callbacks[i].cb = reqs[i].cb;
2447 mcb->callbacks[i].opaque = reqs[i].opaque;
2448 }
2449
2450 // Check for mergable requests
2451 num_reqs = multiwrite_merge(bs, reqs, num_reqs, mcb);
2452
6d519a5f
SH
2453 trace_bdrv_aio_multiwrite(mcb, mcb->num_callbacks, num_reqs);
2454
453f9a16
KW
2455 /*
2456 * Run the aio requests. As soon as one request can't be submitted
2457 * successfully, fail all requests that are not yet submitted (we must
2458 * return failure for all requests anyway)
2459 *
2460 * num_requests cannot be set to the right value immediately: If
2461 * bdrv_aio_writev fails for some request, num_requests would be too high
2462 * and therefore multiwrite_cb() would never recognize the multiwrite
2463 * request as completed. We also cannot use the loop variable i to set it
2464 * when the first request fails because the callback may already have been
2465 * called for previously submitted requests. Thus, num_requests must be
2466 * incremented for each request that is submitted.
2467 *
2468 * The problem that callbacks may be called early also means that we need
2469 * to take care that num_requests doesn't become 0 before all requests are
2470 * submitted - multiwrite_cb() would consider the multiwrite request
2471 * completed. A dummy request that is "completed" by a manual call to
2472 * multiwrite_cb() takes care of this.
2473 */
2474 mcb->num_requests = 1;
2475
6d519a5f 2476 // Run the aio requests
40b4f539 2477 for (i = 0; i < num_reqs; i++) {
453f9a16 2478 mcb->num_requests++;
40b4f539
KW
2479 acb = bdrv_aio_writev(bs, reqs[i].sector, reqs[i].qiov,
2480 reqs[i].nb_sectors, multiwrite_cb, mcb);
2481
2482 if (acb == NULL) {
2483 // We can only fail the whole thing if no request has been
2484 // submitted yet. Otherwise we'll wait for the submitted AIOs to
2485 // complete and report the error in the callback.
453f9a16 2486 if (i == 0) {
6d519a5f 2487 trace_bdrv_aio_multiwrite_earlyfail(mcb);
40b4f539
KW
2488 goto fail;
2489 } else {
6d519a5f 2490 trace_bdrv_aio_multiwrite_latefail(mcb, i);
7eb58a6c 2491 multiwrite_cb(mcb, -EIO);
40b4f539
KW
2492 break;
2493 }
40b4f539
KW
2494 }
2495 }
2496
453f9a16
KW
2497 /* Complete the dummy request */
2498 multiwrite_cb(mcb, 0);
2499
40b4f539
KW
2500 return 0;
2501
2502fail:
453f9a16
KW
2503 for (i = 0; i < mcb->num_callbacks; i++) {
2504 reqs[i].error = -EIO;
2505 }
7267c094 2506 g_free(mcb);
40b4f539
KW
2507 return -1;
2508}
2509
83f64091 2510void bdrv_aio_cancel(BlockDriverAIOCB *acb)
83f64091 2511{
6bbff9a0 2512 acb->pool->cancel(acb);
83f64091
FB
2513}
2514
ce1a14dc 2515
83f64091
FB
2516/**************************************************************/
2517/* async block device emulation */
2518
c16b5a2c
CH
2519typedef struct BlockDriverAIOCBSync {
2520 BlockDriverAIOCB common;
2521 QEMUBH *bh;
2522 int ret;
2523 /* vector translation state */
2524 QEMUIOVector *qiov;
2525 uint8_t *bounce;
2526 int is_write;
2527} BlockDriverAIOCBSync;
2528
2529static void bdrv_aio_cancel_em(BlockDriverAIOCB *blockacb)
2530{
b666d239
KW
2531 BlockDriverAIOCBSync *acb =
2532 container_of(blockacb, BlockDriverAIOCBSync, common);
6a7ad299 2533 qemu_bh_delete(acb->bh);
36afc451 2534 acb->bh = NULL;
c16b5a2c
CH
2535 qemu_aio_release(acb);
2536}
2537
2538static AIOPool bdrv_em_aio_pool = {
2539 .aiocb_size = sizeof(BlockDriverAIOCBSync),
2540 .cancel = bdrv_aio_cancel_em,
2541};
2542
ce1a14dc 2543static void bdrv_aio_bh_cb(void *opaque)
83f64091 2544{
ce1a14dc 2545 BlockDriverAIOCBSync *acb = opaque;
f141eafe 2546
f141eafe
AL
2547 if (!acb->is_write)
2548 qemu_iovec_from_buffer(acb->qiov, acb->bounce, acb->qiov->size);
ceb42de8 2549 qemu_vfree(acb->bounce);
ce1a14dc 2550 acb->common.cb(acb->common.opaque, acb->ret);
6a7ad299 2551 qemu_bh_delete(acb->bh);
36afc451 2552 acb->bh = NULL;
ce1a14dc 2553 qemu_aio_release(acb);
83f64091 2554}
beac80cd 2555
f141eafe
AL
2556static BlockDriverAIOCB *bdrv_aio_rw_vector(BlockDriverState *bs,
2557 int64_t sector_num,
2558 QEMUIOVector *qiov,
2559 int nb_sectors,
2560 BlockDriverCompletionFunc *cb,
2561 void *opaque,
2562 int is_write)
2563
83f64091 2564{
ce1a14dc 2565 BlockDriverAIOCBSync *acb;
ce1a14dc 2566
c16b5a2c 2567 acb = qemu_aio_get(&bdrv_em_aio_pool, bs, cb, opaque);
f141eafe
AL
2568 acb->is_write = is_write;
2569 acb->qiov = qiov;
e268ca52 2570 acb->bounce = qemu_blockalign(bs, qiov->size);
f141eafe 2571
ce1a14dc
PB
2572 if (!acb->bh)
2573 acb->bh = qemu_bh_new(bdrv_aio_bh_cb, acb);
f141eafe
AL
2574
2575 if (is_write) {
2576 qemu_iovec_to_buffer(acb->qiov, acb->bounce);
1ed20acf 2577 acb->ret = bs->drv->bdrv_write(bs, sector_num, acb->bounce, nb_sectors);
f141eafe 2578 } else {
1ed20acf 2579 acb->ret = bs->drv->bdrv_read(bs, sector_num, acb->bounce, nb_sectors);
f141eafe
AL
2580 }
2581
ce1a14dc 2582 qemu_bh_schedule(acb->bh);
f141eafe 2583
ce1a14dc 2584 return &acb->common;
beac80cd
FB
2585}
2586
f141eafe
AL
2587static BlockDriverAIOCB *bdrv_aio_readv_em(BlockDriverState *bs,
2588 int64_t sector_num, QEMUIOVector *qiov, int nb_sectors,
ce1a14dc 2589 BlockDriverCompletionFunc *cb, void *opaque)
beac80cd 2590{
f141eafe
AL
2591 return bdrv_aio_rw_vector(bs, sector_num, qiov, nb_sectors, cb, opaque, 0);
2592}
83f64091 2593
f141eafe
AL
2594static BlockDriverAIOCB *bdrv_aio_writev_em(BlockDriverState *bs,
2595 int64_t sector_num, QEMUIOVector *qiov, int nb_sectors,
2596 BlockDriverCompletionFunc *cb, void *opaque)
2597{
2598 return bdrv_aio_rw_vector(bs, sector_num, qiov, nb_sectors, cb, opaque, 1);
beac80cd 2599}
beac80cd 2600
68485420
KW
2601
2602typedef struct BlockDriverAIOCBCoroutine {
2603 BlockDriverAIOCB common;
2604 BlockRequest req;
2605 bool is_write;
2606 QEMUBH* bh;
2607} BlockDriverAIOCBCoroutine;
2608
2609static void bdrv_aio_co_cancel_em(BlockDriverAIOCB *blockacb)
2610{
2611 qemu_aio_flush();
2612}
2613
2614static AIOPool bdrv_em_co_aio_pool = {
2615 .aiocb_size = sizeof(BlockDriverAIOCBCoroutine),
2616 .cancel = bdrv_aio_co_cancel_em,
2617};
2618
35246a68 2619static void bdrv_co_em_bh(void *opaque)
68485420
KW
2620{
2621 BlockDriverAIOCBCoroutine *acb = opaque;
2622
2623 acb->common.cb(acb->common.opaque, acb->req.error);
2624 qemu_bh_delete(acb->bh);
2625 qemu_aio_release(acb);
2626}
2627
b2a61371
SH
2628/* Invoke bdrv_co_do_readv/bdrv_co_do_writev */
2629static void coroutine_fn bdrv_co_do_rw(void *opaque)
2630{
2631 BlockDriverAIOCBCoroutine *acb = opaque;
2632 BlockDriverState *bs = acb->common.bs;
2633
2634 if (!acb->is_write) {
2635 acb->req.error = bdrv_co_do_readv(bs, acb->req.sector,
2636 acb->req.nb_sectors, acb->req.qiov);
2637 } else {
2638 acb->req.error = bdrv_co_do_writev(bs, acb->req.sector,
2639 acb->req.nb_sectors, acb->req.qiov);
2640 }
2641
35246a68 2642 acb->bh = qemu_bh_new(bdrv_co_em_bh, acb);
b2a61371
SH
2643 qemu_bh_schedule(acb->bh);
2644}
2645
68485420
KW
2646static BlockDriverAIOCB *bdrv_co_aio_rw_vector(BlockDriverState *bs,
2647 int64_t sector_num,
2648 QEMUIOVector *qiov,
2649 int nb_sectors,
2650 BlockDriverCompletionFunc *cb,
2651 void *opaque,
8c5873d6 2652 bool is_write)
68485420
KW
2653{
2654 Coroutine *co;
2655 BlockDriverAIOCBCoroutine *acb;
2656
2657 acb = qemu_aio_get(&bdrv_em_co_aio_pool, bs, cb, opaque);
2658 acb->req.sector = sector_num;
2659 acb->req.nb_sectors = nb_sectors;
2660 acb->req.qiov = qiov;
2661 acb->is_write = is_write;
2662
8c5873d6 2663 co = qemu_coroutine_create(bdrv_co_do_rw);
68485420
KW
2664 qemu_coroutine_enter(co, acb);
2665
2666 return &acb->common;
2667}
2668
07f07615 2669static void coroutine_fn bdrv_aio_flush_co_entry(void *opaque)
b2e12bc6 2670{
07f07615
PB
2671 BlockDriverAIOCBCoroutine *acb = opaque;
2672 BlockDriverState *bs = acb->common.bs;
b2e12bc6 2673
07f07615
PB
2674 acb->req.error = bdrv_co_flush(bs);
2675 acb->bh = qemu_bh_new(bdrv_co_em_bh, acb);
b2e12bc6 2676 qemu_bh_schedule(acb->bh);
b2e12bc6
CH
2677}
2678
07f07615 2679BlockDriverAIOCB *bdrv_aio_flush(BlockDriverState *bs,
016f5cf6
AG
2680 BlockDriverCompletionFunc *cb, void *opaque)
2681{
07f07615 2682 trace_bdrv_aio_flush(bs, opaque);
016f5cf6 2683
07f07615
PB
2684 Coroutine *co;
2685 BlockDriverAIOCBCoroutine *acb;
016f5cf6 2686
07f07615
PB
2687 acb = qemu_aio_get(&bdrv_em_co_aio_pool, bs, cb, opaque);
2688 co = qemu_coroutine_create(bdrv_aio_flush_co_entry);
2689 qemu_coroutine_enter(co, acb);
016f5cf6 2690
016f5cf6
AG
2691 return &acb->common;
2692}
2693
4265d620
PB
2694static void coroutine_fn bdrv_aio_discard_co_entry(void *opaque)
2695{
2696 BlockDriverAIOCBCoroutine *acb = opaque;
2697 BlockDriverState *bs = acb->common.bs;
2698
2699 acb->req.error = bdrv_co_discard(bs, acb->req.sector, acb->req.nb_sectors);
2700 acb->bh = qemu_bh_new(bdrv_co_em_bh, acb);
2701 qemu_bh_schedule(acb->bh);
2702}
2703
2704BlockDriverAIOCB *bdrv_aio_discard(BlockDriverState *bs,
2705 int64_t sector_num, int nb_sectors,
2706 BlockDriverCompletionFunc *cb, void *opaque)
2707{
2708 Coroutine *co;
2709 BlockDriverAIOCBCoroutine *acb;
2710
2711 trace_bdrv_aio_discard(bs, sector_num, nb_sectors, opaque);
2712
2713 acb = qemu_aio_get(&bdrv_em_co_aio_pool, bs, cb, opaque);
2714 acb->req.sector = sector_num;
2715 acb->req.nb_sectors = nb_sectors;
2716 co = qemu_coroutine_create(bdrv_aio_discard_co_entry);
2717 qemu_coroutine_enter(co, acb);
2718
2719 return &acb->common;
2720}
2721
ea2384d3
FB
2722void bdrv_init(void)
2723{
5efa9d5a 2724 module_call_init(MODULE_INIT_BLOCK);
ea2384d3 2725}
ce1a14dc 2726
eb852011
MA
2727void bdrv_init_with_whitelist(void)
2728{
2729 use_bdrv_whitelist = 1;
2730 bdrv_init();
2731}
2732
c16b5a2c
CH
2733void *qemu_aio_get(AIOPool *pool, BlockDriverState *bs,
2734 BlockDriverCompletionFunc *cb, void *opaque)
ce1a14dc 2735{
ce1a14dc
PB
2736 BlockDriverAIOCB *acb;
2737
6bbff9a0
AL
2738 if (pool->free_aiocb) {
2739 acb = pool->free_aiocb;
2740 pool->free_aiocb = acb->next;
ce1a14dc 2741 } else {
7267c094 2742 acb = g_malloc0(pool->aiocb_size);
6bbff9a0 2743 acb->pool = pool;
ce1a14dc
PB
2744 }
2745 acb->bs = bs;
2746 acb->cb = cb;
2747 acb->opaque = opaque;
2748 return acb;
2749}
2750
2751void qemu_aio_release(void *p)
2752{
6bbff9a0
AL
2753 BlockDriverAIOCB *acb = (BlockDriverAIOCB *)p;
2754 AIOPool *pool = acb->pool;
2755 acb->next = pool->free_aiocb;
2756 pool->free_aiocb = acb;
ce1a14dc 2757}
19cb3738 2758
f9f05dc5
KW
2759/**************************************************************/
2760/* Coroutine block device emulation */
2761
2762typedef struct CoroutineIOCompletion {
2763 Coroutine *coroutine;
2764 int ret;
2765} CoroutineIOCompletion;
2766
2767static void bdrv_co_io_em_complete(void *opaque, int ret)
2768{
2769 CoroutineIOCompletion *co = opaque;
2770
2771 co->ret = ret;
2772 qemu_coroutine_enter(co->coroutine, NULL);
2773}
2774
2775static int coroutine_fn bdrv_co_io_em(BlockDriverState *bs, int64_t sector_num,
2776 int nb_sectors, QEMUIOVector *iov,
2777 bool is_write)
2778{
2779 CoroutineIOCompletion co = {
2780 .coroutine = qemu_coroutine_self(),
2781 };
2782 BlockDriverAIOCB *acb;
2783
2784 if (is_write) {
a652d160
SH
2785 acb = bs->drv->bdrv_aio_writev(bs, sector_num, iov, nb_sectors,
2786 bdrv_co_io_em_complete, &co);
f9f05dc5 2787 } else {
a652d160
SH
2788 acb = bs->drv->bdrv_aio_readv(bs, sector_num, iov, nb_sectors,
2789 bdrv_co_io_em_complete, &co);
f9f05dc5
KW
2790 }
2791
59370aaa 2792 trace_bdrv_co_io_em(bs, sector_num, nb_sectors, is_write, acb);
f9f05dc5
KW
2793 if (!acb) {
2794 return -EIO;
2795 }
2796 qemu_coroutine_yield();
2797
2798 return co.ret;
2799}
2800
2801static int coroutine_fn bdrv_co_readv_em(BlockDriverState *bs,
2802 int64_t sector_num, int nb_sectors,
2803 QEMUIOVector *iov)
2804{
2805 return bdrv_co_io_em(bs, sector_num, nb_sectors, iov, false);
2806}
2807
2808static int coroutine_fn bdrv_co_writev_em(BlockDriverState *bs,
2809 int64_t sector_num, int nb_sectors,
2810 QEMUIOVector *iov)
2811{
2812 return bdrv_co_io_em(bs, sector_num, nb_sectors, iov, true);
2813}
2814
07f07615 2815static void coroutine_fn bdrv_flush_co_entry(void *opaque)
e7a8a783 2816{
07f07615
PB
2817 RwCo *rwco = opaque;
2818
2819 rwco->ret = bdrv_co_flush(rwco->bs);
2820}
2821
2822int coroutine_fn bdrv_co_flush(BlockDriverState *bs)
2823{
2824 if (bs->open_flags & BDRV_O_NO_FLUSH) {
2825 return 0;
2826 } else if (!bs->drv) {
2827 return 0;
2828 } else if (bs->drv->bdrv_co_flush) {
2829 return bs->drv->bdrv_co_flush(bs);
2830 } else if (bs->drv->bdrv_aio_flush) {
2831 BlockDriverAIOCB *acb;
2832 CoroutineIOCompletion co = {
2833 .coroutine = qemu_coroutine_self(),
2834 };
2835
2836 acb = bs->drv->bdrv_aio_flush(bs, bdrv_co_io_em_complete, &co);
2837 if (acb == NULL) {
2838 return -EIO;
2839 } else {
2840 qemu_coroutine_yield();
2841 return co.ret;
2842 }
07f07615
PB
2843 } else {
2844 /*
2845 * Some block drivers always operate in either writethrough or unsafe
2846 * mode and don't support bdrv_flush therefore. Usually qemu doesn't
2847 * know how the server works (because the behaviour is hardcoded or
2848 * depends on server-side configuration), so we can't ensure that
2849 * everything is safe on disk. Returning an error doesn't work because
2850 * that would break guests even if the server operates in writethrough
2851 * mode.
2852 *
2853 * Let's hope the user knows what he's doing.
2854 */
2855 return 0;
2856 }
2857}
2858
2859int bdrv_flush(BlockDriverState *bs)
2860{
2861 Coroutine *co;
2862 RwCo rwco = {
2863 .bs = bs,
2864 .ret = NOT_DONE,
e7a8a783 2865 };
e7a8a783 2866
07f07615
PB
2867 if (qemu_in_coroutine()) {
2868 /* Fast-path if already in coroutine context */
2869 bdrv_flush_co_entry(&rwco);
2870 } else {
2871 co = qemu_coroutine_create(bdrv_flush_co_entry);
2872 qemu_coroutine_enter(co, &rwco);
2873 while (rwco.ret == NOT_DONE) {
2874 qemu_aio_wait();
2875 }
e7a8a783 2876 }
07f07615
PB
2877
2878 return rwco.ret;
e7a8a783
KW
2879}
2880
4265d620
PB
2881static void coroutine_fn bdrv_discard_co_entry(void *opaque)
2882{
2883 RwCo *rwco = opaque;
2884
2885 rwco->ret = bdrv_co_discard(rwco->bs, rwco->sector_num, rwco->nb_sectors);
2886}
2887
2888int coroutine_fn bdrv_co_discard(BlockDriverState *bs, int64_t sector_num,
2889 int nb_sectors)
2890{
2891 if (!bs->drv) {
2892 return -ENOMEDIUM;
2893 } else if (bdrv_check_request(bs, sector_num, nb_sectors)) {
2894 return -EIO;
2895 } else if (bs->read_only) {
2896 return -EROFS;
2897 } else if (bs->drv->bdrv_co_discard) {
2898 return bs->drv->bdrv_co_discard(bs, sector_num, nb_sectors);
2899 } else if (bs->drv->bdrv_aio_discard) {
2900 BlockDriverAIOCB *acb;
2901 CoroutineIOCompletion co = {
2902 .coroutine = qemu_coroutine_self(),
2903 };
2904
2905 acb = bs->drv->bdrv_aio_discard(bs, sector_num, nb_sectors,
2906 bdrv_co_io_em_complete, &co);
2907 if (acb == NULL) {
2908 return -EIO;
2909 } else {
2910 qemu_coroutine_yield();
2911 return co.ret;
2912 }
4265d620
PB
2913 } else {
2914 return 0;
2915 }
2916}
2917
2918int bdrv_discard(BlockDriverState *bs, int64_t sector_num, int nb_sectors)
2919{
2920 Coroutine *co;
2921 RwCo rwco = {
2922 .bs = bs,
2923 .sector_num = sector_num,
2924 .nb_sectors = nb_sectors,
2925 .ret = NOT_DONE,
2926 };
2927
2928 if (qemu_in_coroutine()) {
2929 /* Fast-path if already in coroutine context */
2930 bdrv_discard_co_entry(&rwco);
2931 } else {
2932 co = qemu_coroutine_create(bdrv_discard_co_entry);
2933 qemu_coroutine_enter(co, &rwco);
2934 while (rwco.ret == NOT_DONE) {
2935 qemu_aio_wait();
2936 }
2937 }
2938
2939 return rwco.ret;
2940}
2941
19cb3738
FB
2942/**************************************************************/
2943/* removable device support */
2944
2945/**
2946 * Return TRUE if the media is present
2947 */
2948int bdrv_is_inserted(BlockDriverState *bs)
2949{
2950 BlockDriver *drv = bs->drv;
a1aff5bf 2951
19cb3738
FB
2952 if (!drv)
2953 return 0;
2954 if (!drv->bdrv_is_inserted)
a1aff5bf
MA
2955 return 1;
2956 return drv->bdrv_is_inserted(bs);
19cb3738
FB
2957}
2958
2959/**
8e49ca46
MA
2960 * Return whether the media changed since the last call to this
2961 * function, or -ENOTSUP if we don't know. Most drivers don't know.
19cb3738
FB
2962 */
2963int bdrv_media_changed(BlockDriverState *bs)
2964{
2965 BlockDriver *drv = bs->drv;
19cb3738 2966
8e49ca46
MA
2967 if (drv && drv->bdrv_media_changed) {
2968 return drv->bdrv_media_changed(bs);
2969 }
2970 return -ENOTSUP;
19cb3738
FB
2971}
2972
2973/**
2974 * If eject_flag is TRUE, eject the media. Otherwise, close the tray
2975 */
fdec4404 2976void bdrv_eject(BlockDriverState *bs, int eject_flag)
19cb3738
FB
2977{
2978 BlockDriver *drv = bs->drv;
19cb3738 2979
822e1cd1
MA
2980 if (drv && drv->bdrv_eject) {
2981 drv->bdrv_eject(bs, eject_flag);
19cb3738
FB
2982 }
2983}
2984
19cb3738
FB
2985/**
2986 * Lock or unlock the media (if it is locked, the user won't be able
2987 * to eject it manually).
2988 */
025e849a 2989void bdrv_lock_medium(BlockDriverState *bs, bool locked)
19cb3738
FB
2990{
2991 BlockDriver *drv = bs->drv;
2992
025e849a 2993 trace_bdrv_lock_medium(bs, locked);
b8c6d095 2994
025e849a
MA
2995 if (drv && drv->bdrv_lock_medium) {
2996 drv->bdrv_lock_medium(bs, locked);
19cb3738
FB
2997 }
2998}
985a03b0
TS
2999
3000/* needed for generic scsi interface */
3001
3002int bdrv_ioctl(BlockDriverState *bs, unsigned long int req, void *buf)
3003{
3004 BlockDriver *drv = bs->drv;
3005
3006 if (drv && drv->bdrv_ioctl)
3007 return drv->bdrv_ioctl(bs, req, buf);
3008 return -ENOTSUP;
3009}
7d780669 3010
221f715d
AL
3011BlockDriverAIOCB *bdrv_aio_ioctl(BlockDriverState *bs,
3012 unsigned long int req, void *buf,
3013 BlockDriverCompletionFunc *cb, void *opaque)
7d780669 3014{
221f715d 3015 BlockDriver *drv = bs->drv;
7d780669 3016
221f715d
AL
3017 if (drv && drv->bdrv_aio_ioctl)
3018 return drv->bdrv_aio_ioctl(bs, req, buf, cb, opaque);
3019 return NULL;
7d780669 3020}
e268ca52 3021
7b6f9300
MA
3022void bdrv_set_buffer_alignment(BlockDriverState *bs, int align)
3023{
3024 bs->buffer_alignment = align;
3025}
7cd1e32a 3026
e268ca52
AL
3027void *qemu_blockalign(BlockDriverState *bs, size_t size)
3028{
3029 return qemu_memalign((bs && bs->buffer_alignment) ? bs->buffer_alignment : 512, size);
3030}
7cd1e32a
LS
3031
3032void bdrv_set_dirty_tracking(BlockDriverState *bs, int enable)
3033{
3034 int64_t bitmap_size;
a55eb92c 3035
aaa0eb75 3036 bs->dirty_count = 0;
a55eb92c 3037 if (enable) {
c6d22830
JK
3038 if (!bs->dirty_bitmap) {
3039 bitmap_size = (bdrv_getlength(bs) >> BDRV_SECTOR_BITS) +
3040 BDRV_SECTORS_PER_DIRTY_CHUNK * 8 - 1;
3041 bitmap_size /= BDRV_SECTORS_PER_DIRTY_CHUNK * 8;
a55eb92c 3042
7267c094 3043 bs->dirty_bitmap = g_malloc0(bitmap_size);
a55eb92c 3044 }
7cd1e32a 3045 } else {
c6d22830 3046 if (bs->dirty_bitmap) {
7267c094 3047 g_free(bs->dirty_bitmap);
c6d22830 3048 bs->dirty_bitmap = NULL;
a55eb92c 3049 }
7cd1e32a
LS
3050 }
3051}
3052
3053int bdrv_get_dirty(BlockDriverState *bs, int64_t sector)
3054{
6ea44308 3055 int64_t chunk = sector / (int64_t)BDRV_SECTORS_PER_DIRTY_CHUNK;
a55eb92c 3056
c6d22830
JK
3057 if (bs->dirty_bitmap &&
3058 (sector << BDRV_SECTOR_BITS) < bdrv_getlength(bs)) {
6d59fec1
MT
3059 return !!(bs->dirty_bitmap[chunk / (sizeof(unsigned long) * 8)] &
3060 (1UL << (chunk % (sizeof(unsigned long) * 8))));
7cd1e32a
LS
3061 } else {
3062 return 0;
3063 }
3064}
3065
a55eb92c
JK
3066void bdrv_reset_dirty(BlockDriverState *bs, int64_t cur_sector,
3067 int nr_sectors)
7cd1e32a
LS
3068{
3069 set_dirty_bitmap(bs, cur_sector, nr_sectors, 0);
3070}
aaa0eb75
LS
3071
3072int64_t bdrv_get_dirty_count(BlockDriverState *bs)
3073{
3074 return bs->dirty_count;
3075}
f88e1a42 3076
db593f25
MT
3077void bdrv_set_in_use(BlockDriverState *bs, int in_use)
3078{
3079 assert(bs->in_use != in_use);
3080 bs->in_use = in_use;
3081}
3082
3083int bdrv_in_use(BlockDriverState *bs)
3084{
3085 return bs->in_use;
3086}
3087
28a7282a
LC
3088void bdrv_iostatus_enable(BlockDriverState *bs)
3089{
d6bf279e 3090 bs->iostatus_enabled = true;
58e21ef5 3091 bs->iostatus = BLOCK_DEVICE_IO_STATUS_OK;
28a7282a
LC
3092}
3093
3094/* The I/O status is only enabled if the drive explicitly
3095 * enables it _and_ the VM is configured to stop on errors */
3096bool bdrv_iostatus_is_enabled(const BlockDriverState *bs)
3097{
d6bf279e 3098 return (bs->iostatus_enabled &&
28a7282a
LC
3099 (bs->on_write_error == BLOCK_ERR_STOP_ENOSPC ||
3100 bs->on_write_error == BLOCK_ERR_STOP_ANY ||
3101 bs->on_read_error == BLOCK_ERR_STOP_ANY));
3102}
3103
3104void bdrv_iostatus_disable(BlockDriverState *bs)
3105{
d6bf279e 3106 bs->iostatus_enabled = false;
28a7282a
LC
3107}
3108
3109void bdrv_iostatus_reset(BlockDriverState *bs)
3110{
3111 if (bdrv_iostatus_is_enabled(bs)) {
58e21ef5 3112 bs->iostatus = BLOCK_DEVICE_IO_STATUS_OK;
28a7282a
LC
3113 }
3114}
3115
3116/* XXX: Today this is set by device models because it makes the implementation
3117 quite simple. However, the block layer knows about the error, so it's
3118 possible to implement this without device models being involved */
3119void bdrv_iostatus_set_err(BlockDriverState *bs, int error)
3120{
58e21ef5
LC
3121 if (bdrv_iostatus_is_enabled(bs) &&
3122 bs->iostatus == BLOCK_DEVICE_IO_STATUS_OK) {
28a7282a 3123 assert(error >= 0);
58e21ef5
LC
3124 bs->iostatus = error == ENOSPC ? BLOCK_DEVICE_IO_STATUS_NOSPACE :
3125 BLOCK_DEVICE_IO_STATUS_FAILED;
28a7282a
LC
3126 }
3127}
3128
a597e79c
CH
3129void
3130bdrv_acct_start(BlockDriverState *bs, BlockAcctCookie *cookie, int64_t bytes,
3131 enum BlockAcctType type)
3132{
3133 assert(type < BDRV_MAX_IOTYPE);
3134
3135 cookie->bytes = bytes;
c488c7f6 3136 cookie->start_time_ns = get_clock();
a597e79c
CH
3137 cookie->type = type;
3138}
3139
3140void
3141bdrv_acct_done(BlockDriverState *bs, BlockAcctCookie *cookie)
3142{
3143 assert(cookie->type < BDRV_MAX_IOTYPE);
3144
3145 bs->nr_bytes[cookie->type] += cookie->bytes;
3146 bs->nr_ops[cookie->type]++;
c488c7f6 3147 bs->total_time_ns[cookie->type] += get_clock() - cookie->start_time_ns;
a597e79c
CH
3148}
3149
f88e1a42
JS
3150int bdrv_img_create(const char *filename, const char *fmt,
3151 const char *base_filename, const char *base_fmt,
3152 char *options, uint64_t img_size, int flags)
3153{
3154 QEMUOptionParameter *param = NULL, *create_options = NULL;
d220894e 3155 QEMUOptionParameter *backing_fmt, *backing_file, *size;
f88e1a42
JS
3156 BlockDriverState *bs = NULL;
3157 BlockDriver *drv, *proto_drv;
96df67d1 3158 BlockDriver *backing_drv = NULL;
f88e1a42
JS
3159 int ret = 0;
3160
3161 /* Find driver and parse its options */
3162 drv = bdrv_find_format(fmt);
3163 if (!drv) {
3164 error_report("Unknown file format '%s'", fmt);
4f70f249 3165 ret = -EINVAL;
f88e1a42
JS
3166 goto out;
3167 }
3168
3169 proto_drv = bdrv_find_protocol(filename);
3170 if (!proto_drv) {
3171 error_report("Unknown protocol '%s'", filename);
4f70f249 3172 ret = -EINVAL;
f88e1a42
JS
3173 goto out;
3174 }
3175
3176 create_options = append_option_parameters(create_options,
3177 drv->create_options);
3178 create_options = append_option_parameters(create_options,
3179 proto_drv->create_options);
3180
3181 /* Create parameter list with default values */
3182 param = parse_option_parameters("", create_options, param);
3183
3184 set_option_parameter_int(param, BLOCK_OPT_SIZE, img_size);
3185
3186 /* Parse -o options */
3187 if (options) {
3188 param = parse_option_parameters(options, create_options, param);
3189 if (param == NULL) {
3190 error_report("Invalid options for file format '%s'.", fmt);
4f70f249 3191 ret = -EINVAL;
f88e1a42
JS
3192 goto out;
3193 }
3194 }
3195
3196 if (base_filename) {
3197 if (set_option_parameter(param, BLOCK_OPT_BACKING_FILE,
3198 base_filename)) {
3199 error_report("Backing file not supported for file format '%s'",
3200 fmt);
4f70f249 3201 ret = -EINVAL;
f88e1a42
JS
3202 goto out;
3203 }
3204 }
3205
3206 if (base_fmt) {
3207 if (set_option_parameter(param, BLOCK_OPT_BACKING_FMT, base_fmt)) {
3208 error_report("Backing file format not supported for file "
3209 "format '%s'", fmt);
4f70f249 3210 ret = -EINVAL;
f88e1a42
JS
3211 goto out;
3212 }
3213 }
3214
792da93a
JS
3215 backing_file = get_option_parameter(param, BLOCK_OPT_BACKING_FILE);
3216 if (backing_file && backing_file->value.s) {
3217 if (!strcmp(filename, backing_file->value.s)) {
3218 error_report("Error: Trying to create an image with the "
3219 "same filename as the backing file");
4f70f249 3220 ret = -EINVAL;
792da93a
JS
3221 goto out;
3222 }
3223 }
3224
f88e1a42
JS
3225 backing_fmt = get_option_parameter(param, BLOCK_OPT_BACKING_FMT);
3226 if (backing_fmt && backing_fmt->value.s) {
96df67d1
SH
3227 backing_drv = bdrv_find_format(backing_fmt->value.s);
3228 if (!backing_drv) {
f88e1a42
JS
3229 error_report("Unknown backing file format '%s'",
3230 backing_fmt->value.s);
4f70f249 3231 ret = -EINVAL;
f88e1a42
JS
3232 goto out;
3233 }
3234 }
3235
3236 // The size for the image must always be specified, with one exception:
3237 // If we are using a backing file, we can obtain the size from there
d220894e
KW
3238 size = get_option_parameter(param, BLOCK_OPT_SIZE);
3239 if (size && size->value.n == -1) {
f88e1a42
JS
3240 if (backing_file && backing_file->value.s) {
3241 uint64_t size;
f88e1a42
JS
3242 char buf[32];
3243
f88e1a42
JS
3244 bs = bdrv_new("");
3245
96df67d1 3246 ret = bdrv_open(bs, backing_file->value.s, flags, backing_drv);
f88e1a42 3247 if (ret < 0) {
96df67d1 3248 error_report("Could not open '%s'", backing_file->value.s);
f88e1a42
JS
3249 goto out;
3250 }
3251 bdrv_get_geometry(bs, &size);
3252 size *= 512;
3253
3254 snprintf(buf, sizeof(buf), "%" PRId64, size);
3255 set_option_parameter(param, BLOCK_OPT_SIZE, buf);
3256 } else {
3257 error_report("Image creation needs a size parameter");
4f70f249 3258 ret = -EINVAL;
f88e1a42
JS
3259 goto out;
3260 }
3261 }
3262
3263 printf("Formatting '%s', fmt=%s ", filename, fmt);
3264 print_option_parameters(param);
3265 puts("");
3266
3267 ret = bdrv_create(drv, filename, param);
3268
3269 if (ret < 0) {
3270 if (ret == -ENOTSUP) {
3271 error_report("Formatting or formatting option not supported for "
3272 "file format '%s'", fmt);
3273 } else if (ret == -EFBIG) {
3274 error_report("The image size is too large for file format '%s'",
3275 fmt);
3276 } else {
3277 error_report("%s: error while creating %s: %s", filename, fmt,
3278 strerror(-ret));
3279 }
3280 }
3281
3282out:
3283 free_option_parameters(create_options);
3284 free_option_parameters(param);
3285
3286 if (bs) {
3287 bdrv_delete(bs);
3288 }
4f70f249
JS
3289
3290 return ret;
f88e1a42 3291}