]> git.proxmox.com Git - mirror_qemu.git/blame - block.c
virtio: Allow guest to defer VIRTIO_F_NOTIFY_ON_EMPTY (Alex Williamson)
[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"
179a2c19 25#ifdef HOST_BSD
3990d09a
BS
26/* include native header before sys-queue.h */
27#include <sys/queue.h>
28#endif
29
faf07963 30#include "qemu-common.h"
376253ec 31#include "monitor.h"
ea2384d3 32#include "block_int.h"
fc01f7e7 33
179a2c19 34#ifdef HOST_BSD
7674e7bf
FB
35#include <sys/types.h>
36#include <sys/stat.h>
37#include <sys/ioctl.h>
c5e97233 38#ifndef __DragonFly__
7674e7bf
FB
39#include <sys/disk.h>
40#endif
c5e97233 41#endif
7674e7bf 42
49dc768d
AL
43#ifdef _WIN32
44#include <windows.h>
45#endif
46
83f64091
FB
47#define SECTOR_BITS 9
48#define SECTOR_SIZE (1 << SECTOR_BITS)
49
90765429
FB
50typedef struct BlockDriverAIOCBSync {
51 BlockDriverAIOCB common;
52 QEMUBH *bh;
53 int ret;
54} BlockDriverAIOCBSync;
55
ce1a14dc
PB
56static BlockDriverAIOCB *bdrv_aio_read_em(BlockDriverState *bs,
57 int64_t sector_num, uint8_t *buf, int nb_sectors,
58 BlockDriverCompletionFunc *cb, void *opaque);
59static BlockDriverAIOCB *bdrv_aio_write_em(BlockDriverState *bs,
60 int64_t sector_num, const uint8_t *buf, int nb_sectors,
61 BlockDriverCompletionFunc *cb, void *opaque);
83f64091 62static void bdrv_aio_cancel_em(BlockDriverAIOCB *acb);
5fafdf24 63static int bdrv_read_em(BlockDriverState *bs, int64_t sector_num,
83f64091
FB
64 uint8_t *buf, int nb_sectors);
65static int bdrv_write_em(BlockDriverState *bs, int64_t sector_num,
66 const uint8_t *buf, int nb_sectors);
ec530c81 67
7ee930d0
BS
68BlockDriverState *bdrv_first;
69
ea2384d3
FB
70static BlockDriver *first_drv;
71
83f64091 72int path_is_absolute(const char *path)
3b0d4f61 73{
83f64091 74 const char *p;
21664424
FB
75#ifdef _WIN32
76 /* specific case for names like: "\\.\d:" */
77 if (*path == '/' || *path == '\\')
78 return 1;
79#endif
83f64091
FB
80 p = strchr(path, ':');
81 if (p)
82 p++;
83 else
84 p = path;
3b9f94e1
FB
85#ifdef _WIN32
86 return (*p == '/' || *p == '\\');
87#else
88 return (*p == '/');
89#endif
3b0d4f61
FB
90}
91
83f64091
FB
92/* if filename is absolute, just copy it to dest. Otherwise, build a
93 path to it by considering it is relative to base_path. URL are
94 supported. */
95void path_combine(char *dest, int dest_size,
96 const char *base_path,
97 const char *filename)
3b0d4f61 98{
83f64091
FB
99 const char *p, *p1;
100 int len;
101
102 if (dest_size <= 0)
103 return;
104 if (path_is_absolute(filename)) {
105 pstrcpy(dest, dest_size, filename);
106 } else {
107 p = strchr(base_path, ':');
108 if (p)
109 p++;
110 else
111 p = base_path;
3b9f94e1
FB
112 p1 = strrchr(base_path, '/');
113#ifdef _WIN32
114 {
115 const char *p2;
116 p2 = strrchr(base_path, '\\');
117 if (!p1 || p2 > p1)
118 p1 = p2;
119 }
120#endif
83f64091
FB
121 if (p1)
122 p1++;
123 else
124 p1 = base_path;
125 if (p1 > p)
126 p = p1;
127 len = p - base_path;
128 if (len > dest_size - 1)
129 len = dest_size - 1;
130 memcpy(dest, base_path, len);
131 dest[len] = '\0';
132 pstrcat(dest, dest_size, filename);
3b0d4f61 133 }
3b0d4f61
FB
134}
135
3b0d4f61 136
9596ebb7 137static void bdrv_register(BlockDriver *bdrv)
ea2384d3 138{
ce1a14dc 139 if (!bdrv->bdrv_aio_read) {
83f64091 140 /* add AIO emulation layer */
83f64091
FB
141 bdrv->bdrv_aio_read = bdrv_aio_read_em;
142 bdrv->bdrv_aio_write = bdrv_aio_write_em;
143 bdrv->bdrv_aio_cancel = bdrv_aio_cancel_em;
90765429 144 bdrv->aiocb_size = sizeof(BlockDriverAIOCBSync);
eda578e5 145 } else if (!bdrv->bdrv_read) {
83f64091
FB
146 /* add synchronous IO emulation layer */
147 bdrv->bdrv_read = bdrv_read_em;
148 bdrv->bdrv_write = bdrv_write_em;
149 }
ea2384d3
FB
150 bdrv->next = first_drv;
151 first_drv = bdrv;
152}
b338082b
FB
153
154/* create a new block device (by default it is empty) */
155BlockDriverState *bdrv_new(const char *device_name)
156{
157 BlockDriverState **pbs, *bs;
158
159 bs = qemu_mallocz(sizeof(BlockDriverState));
b338082b 160 pstrcpy(bs->device_name, sizeof(bs->device_name), device_name);
ea2384d3
FB
161 if (device_name[0] != '\0') {
162 /* insert at the end */
163 pbs = &bdrv_first;
164 while (*pbs != NULL)
165 pbs = &(*pbs)->next;
166 *pbs = bs;
167 }
b338082b
FB
168 return bs;
169}
170
ea2384d3
FB
171BlockDriver *bdrv_find_format(const char *format_name)
172{
173 BlockDriver *drv1;
174 for(drv1 = first_drv; drv1 != NULL; drv1 = drv1->next) {
175 if (!strcmp(drv1->format_name, format_name))
176 return drv1;
177 }
178 return NULL;
179}
180
5fafdf24 181int bdrv_create(BlockDriver *drv,
ea2384d3
FB
182 const char *filename, int64_t size_in_sectors,
183 const char *backing_file, int flags)
184{
185 if (!drv->bdrv_create)
186 return -ENOTSUP;
187 return drv->bdrv_create(filename, size_in_sectors, backing_file, flags);
188}
189
d5249393 190#ifdef _WIN32
95389c86 191void get_tmp_filename(char *filename, int size)
d5249393 192{
3b9f94e1 193 char temp_dir[MAX_PATH];
3b46e624 194
3b9f94e1
FB
195 GetTempPath(MAX_PATH, temp_dir);
196 GetTempFileName(temp_dir, "qem", 0, filename);
d5249393
FB
197}
198#else
95389c86 199void get_tmp_filename(char *filename, int size)
fc01f7e7 200{
67b915a5 201 int fd;
7ccfb2eb 202 const char *tmpdir;
d5249393 203 /* XXX: race condition possible */
0badc1ee
AJ
204 tmpdir = getenv("TMPDIR");
205 if (!tmpdir)
206 tmpdir = "/tmp";
207 snprintf(filename, size, "%s/vl.XXXXXX", tmpdir);
ea2384d3
FB
208 fd = mkstemp(filename);
209 close(fd);
210}
d5249393 211#endif
fc01f7e7 212
19cb3738 213#ifdef _WIN32
f45512fe
FB
214static int is_windows_drive_prefix(const char *filename)
215{
216 return (((filename[0] >= 'a' && filename[0] <= 'z') ||
217 (filename[0] >= 'A' && filename[0] <= 'Z')) &&
218 filename[1] == ':');
219}
3b46e624 220
19cb3738
FB
221static int is_windows_drive(const char *filename)
222{
5fafdf24 223 if (is_windows_drive_prefix(filename) &&
f45512fe 224 filename[2] == '\0')
19cb3738
FB
225 return 1;
226 if (strstart(filename, "\\\\.\\", NULL) ||
227 strstart(filename, "//./", NULL))
228 return 1;
229 return 0;
230}
231#endif
232
83f64091
FB
233static BlockDriver *find_protocol(const char *filename)
234{
235 BlockDriver *drv1;
236 char protocol[128];
237 int len;
238 const char *p;
19cb3738
FB
239
240#ifdef _WIN32
f45512fe
FB
241 if (is_windows_drive(filename) ||
242 is_windows_drive_prefix(filename))
19cb3738
FB
243 return &bdrv_raw;
244#endif
83f64091
FB
245 p = strchr(filename, ':');
246 if (!p)
247 return &bdrv_raw;
248 len = p - filename;
249 if (len > sizeof(protocol) - 1)
250 len = sizeof(protocol) - 1;
83f64091
FB
251 memcpy(protocol, filename, len);
252 protocol[len] = '\0';
253 for(drv1 = first_drv; drv1 != NULL; drv1 = drv1->next) {
5fafdf24 254 if (drv1->protocol_name &&
83f64091
FB
255 !strcmp(drv1->protocol_name, protocol))
256 return drv1;
257 }
258 return NULL;
259}
260
7674e7bf
FB
261/* XXX: force raw format if block or character device ? It would
262 simplify the BSD case */
ea2384d3
FB
263static BlockDriver *find_image_format(const char *filename)
264{
83f64091 265 int ret, score, score_max;
ea2384d3 266 BlockDriver *drv1, *drv;
83f64091
FB
267 uint8_t buf[2048];
268 BlockDriverState *bs;
3b46e624 269
19cb3738
FB
270 /* detect host devices. By convention, /dev/cdrom[N] is always
271 recognized as a host CDROM */
272 if (strstart(filename, "/dev/cdrom", NULL))
273 return &bdrv_host_device;
274#ifdef _WIN32
275 if (is_windows_drive(filename))
276 return &bdrv_host_device;
277#else
278 {
279 struct stat st;
5fafdf24 280 if (stat(filename, &st) >= 0 &&
19cb3738
FB
281 (S_ISCHR(st.st_mode) || S_ISBLK(st.st_mode))) {
282 return &bdrv_host_device;
283 }
284 }
285#endif
3b46e624 286
83f64091 287 drv = find_protocol(filename);
19cb3738 288 /* no need to test disk image formats for vvfat */
83f64091
FB
289 if (drv == &bdrv_vvfat)
290 return drv;
19cb3738 291
83f64091
FB
292 ret = bdrv_file_open(&bs, filename, BDRV_O_RDONLY);
293 if (ret < 0)
294 return NULL;
295 ret = bdrv_pread(bs, 0, buf, sizeof(buf));
296 bdrv_delete(bs);
297 if (ret < 0) {
298 return NULL;
299 }
300
ea2384d3
FB
301 score_max = 0;
302 for(drv1 = first_drv; drv1 != NULL; drv1 = drv1->next) {
83f64091
FB
303 if (drv1->bdrv_probe) {
304 score = drv1->bdrv_probe(buf, ret, filename);
305 if (score > score_max) {
306 score_max = score;
307 drv = drv1;
308 }
0849bf08 309 }
fc01f7e7 310 }
ea2384d3
FB
311 return drv;
312}
313
83f64091 314int bdrv_file_open(BlockDriverState **pbs, const char *filename, int flags)
ea2384d3 315{
83f64091
FB
316 BlockDriverState *bs;
317 int ret;
318
319 bs = bdrv_new("");
83f64091
FB
320 ret = bdrv_open2(bs, filename, flags | BDRV_O_FILE, NULL);
321 if (ret < 0) {
322 bdrv_delete(bs);
323 return ret;
3b0d4f61 324 }
71d0770c 325 bs->growable = 1;
83f64091
FB
326 *pbs = bs;
327 return 0;
328}
329
330int bdrv_open(BlockDriverState *bs, const char *filename, int flags)
331{
332 return bdrv_open2(bs, filename, flags, NULL);
ea2384d3
FB
333}
334
83f64091 335int bdrv_open2(BlockDriverState *bs, const char *filename, int flags,
ea2384d3
FB
336 BlockDriver *drv)
337{
83f64091 338 int ret, open_flags;
eb5c851f
TS
339 char tmp_filename[PATH_MAX];
340 char backing_filename[PATH_MAX];
3b46e624 341
ea2384d3
FB
342 bs->read_only = 0;
343 bs->is_temporary = 0;
344 bs->encrypted = 0;
c0f4ce77 345 bs->valid_key = 0;
712e7874 346
83f64091 347 if (flags & BDRV_O_SNAPSHOT) {
ea2384d3
FB
348 BlockDriverState *bs1;
349 int64_t total_size;
7c96d46e 350 int is_protocol = 0;
3b46e624 351
ea2384d3
FB
352 /* if snapshot, we create a temporary backing file and open it
353 instead of opening 'filename' directly */
33e3963e 354
ea2384d3
FB
355 /* if there is a backing file, use it */
356 bs1 = bdrv_new("");
51d7c00c
AL
357 ret = bdrv_open(bs1, filename, 0);
358 if (ret < 0) {
ea2384d3 359 bdrv_delete(bs1);
51d7c00c 360 return ret;
ea2384d3 361 }
83f64091 362 total_size = bdrv_getlength(bs1) >> SECTOR_BITS;
7c96d46e
AL
363
364 if (bs1->drv && bs1->drv->protocol_name)
365 is_protocol = 1;
366
ea2384d3 367 bdrv_delete(bs1);
3b46e624 368
ea2384d3 369 get_tmp_filename(tmp_filename, sizeof(tmp_filename));
7c96d46e
AL
370
371 /* Real path is meaningless for protocols */
372 if (is_protocol)
373 snprintf(backing_filename, sizeof(backing_filename),
374 "%s", filename);
375 else
376 realpath(filename, backing_filename);
377
51d7c00c
AL
378 ret = bdrv_create(&bdrv_qcow2, tmp_filename,
379 total_size, backing_filename, 0);
380 if (ret < 0) {
381 return ret;
ea2384d3
FB
382 }
383 filename = tmp_filename;
384 bs->is_temporary = 1;
385 }
712e7874 386
ea2384d3 387 pstrcpy(bs->filename, sizeof(bs->filename), filename);
83f64091
FB
388 if (flags & BDRV_O_FILE) {
389 drv = find_protocol(filename);
51d7c00c
AL
390 } else if (!drv) {
391 drv = find_image_format(filename);
392 }
393 if (!drv) {
394 ret = -ENOENT;
395 goto unlink_and_fail;
ea2384d3
FB
396 }
397 bs->drv = drv;
398 bs->opaque = qemu_mallocz(drv->instance_size);
83f64091
FB
399 /* Note: for compatibility, we open disk image files as RDWR, and
400 RDONLY as fallback */
401 if (!(flags & BDRV_O_FILE))
9f7965c7 402 open_flags = BDRV_O_RDWR | (flags & BDRV_O_CACHE_MASK);
83f64091
FB
403 else
404 open_flags = flags & ~(BDRV_O_FILE | BDRV_O_SNAPSHOT);
405 ret = drv->bdrv_open(bs, filename, open_flags);
a0a83536 406 if ((ret == -EACCES || ret == -EPERM) && !(flags & BDRV_O_FILE)) {
9f7965c7 407 ret = drv->bdrv_open(bs, filename, open_flags & ~BDRV_O_RDWR);
83f64091
FB
408 bs->read_only = 1;
409 }
ea2384d3
FB
410 if (ret < 0) {
411 qemu_free(bs->opaque);
6b21b973
FB
412 bs->opaque = NULL;
413 bs->drv = NULL;
51d7c00c
AL
414 unlink_and_fail:
415 if (bs->is_temporary)
416 unlink(filename);
83f64091 417 return ret;
33e3963e 418 }
d15a771d
FB
419 if (drv->bdrv_getlength) {
420 bs->total_sectors = bdrv_getlength(bs) >> SECTOR_BITS;
421 }
67b915a5 422#ifndef _WIN32
ea2384d3
FB
423 if (bs->is_temporary) {
424 unlink(filename);
425 }
426#endif
83f64091 427 if (bs->backing_file[0] != '\0') {
ea2384d3
FB
428 /* if there is a backing file, use it */
429 bs->backing_hd = bdrv_new("");
83f64091
FB
430 path_combine(backing_filename, sizeof(backing_filename),
431 filename, bs->backing_file);
51d7c00c
AL
432 ret = bdrv_open(bs->backing_hd, backing_filename, open_flags);
433 if (ret < 0) {
434 bdrv_close(bs);
435 return ret;
436 }
33e3963e
FB
437 }
438
bb5fc20f
AL
439 if (!bdrv_key_required(bs)) {
440 /* call the change callback */
441 bs->media_changed = 1;
442 if (bs->change_cb)
443 bs->change_cb(bs->change_opaque);
444 }
b338082b 445 return 0;
fc01f7e7
FB
446}
447
448void bdrv_close(BlockDriverState *bs)
449{
19cb3738 450 if (bs->drv) {
ea2384d3
FB
451 if (bs->backing_hd)
452 bdrv_delete(bs->backing_hd);
453 bs->drv->bdrv_close(bs);
454 qemu_free(bs->opaque);
455#ifdef _WIN32
456 if (bs->is_temporary) {
457 unlink(bs->filename);
458 }
67b915a5 459#endif
ea2384d3
FB
460 bs->opaque = NULL;
461 bs->drv = NULL;
b338082b
FB
462
463 /* call the change callback */
19cb3738 464 bs->media_changed = 1;
b338082b
FB
465 if (bs->change_cb)
466 bs->change_cb(bs->change_opaque);
467 }
468}
469
470void bdrv_delete(BlockDriverState *bs)
471{
34c6f050
AJ
472 BlockDriverState **pbs;
473
474 pbs = &bdrv_first;
475 while (*pbs != bs && *pbs != NULL)
476 pbs = &(*pbs)->next;
477 if (*pbs == bs)
478 *pbs = bs->next;
479
b338082b
FB
480 bdrv_close(bs);
481 qemu_free(bs);
fc01f7e7
FB
482}
483
33e3963e
FB
484/* commit COW file into the raw image */
485int bdrv_commit(BlockDriverState *bs)
486{
19cb3738 487 BlockDriver *drv = bs->drv;
83f64091 488 int64_t i, total_sectors;
ea2384d3
FB
489 int n, j;
490 unsigned char sector[512];
33e3963e 491
19cb3738
FB
492 if (!drv)
493 return -ENOMEDIUM;
33e3963e
FB
494
495 if (bs->read_only) {
ea2384d3 496 return -EACCES;
33e3963e
FB
497 }
498
ea2384d3
FB
499 if (!bs->backing_hd) {
500 return -ENOTSUP;
501 }
33e3963e 502
83f64091
FB
503 total_sectors = bdrv_getlength(bs) >> SECTOR_BITS;
504 for (i = 0; i < total_sectors;) {
19cb3738 505 if (drv->bdrv_is_allocated(bs, i, 65536, &n)) {
ea2384d3
FB
506 for(j = 0; j < n; j++) {
507 if (bdrv_read(bs, i, sector, 1) != 0) {
508 return -EIO;
509 }
510
511 if (bdrv_write(bs->backing_hd, i, sector, 1) != 0) {
512 return -EIO;
513 }
514 i++;
33e3963e 515 }
ea2384d3
FB
516 } else {
517 i += n;
518 }
33e3963e 519 }
95389c86 520
19cb3738
FB
521 if (drv->bdrv_make_empty)
522 return drv->bdrv_make_empty(bs);
95389c86 523
33e3963e
FB
524 return 0;
525}
526
71d0770c
AL
527static int bdrv_check_byte_request(BlockDriverState *bs, int64_t offset,
528 size_t size)
529{
530 int64_t len;
531
532 if (!bdrv_is_inserted(bs))
533 return -ENOMEDIUM;
534
535 if (bs->growable)
536 return 0;
537
538 len = bdrv_getlength(bs);
539
540 if ((offset + size) > len)
541 return -EIO;
542
543 return 0;
544}
545
546static int bdrv_check_request(BlockDriverState *bs, int64_t sector_num,
547 int nb_sectors)
548{
549 int64_t offset;
550
551 /* Deal with byte accesses */
552 if (sector_num < 0)
553 offset = -sector_num;
554 else
555 offset = sector_num * 512;
556
557 return bdrv_check_byte_request(bs, offset, nb_sectors * 512);
558}
559
19cb3738 560/* return < 0 if error. See bdrv_write() for the return codes */
5fafdf24 561int bdrv_read(BlockDriverState *bs, int64_t sector_num,
fc01f7e7
FB
562 uint8_t *buf, int nb_sectors)
563{
ea2384d3
FB
564 BlockDriver *drv = bs->drv;
565
19cb3738
FB
566 if (!drv)
567 return -ENOMEDIUM;
71d0770c
AL
568 if (bdrv_check_request(bs, sector_num, nb_sectors))
569 return -EIO;
b338082b 570
eda578e5 571 return drv->bdrv_read(bs, sector_num, buf, nb_sectors);
fc01f7e7
FB
572}
573
5fafdf24 574/* Return < 0 if error. Important errors are:
19cb3738
FB
575 -EIO generic I/O error (may happen for all errors)
576 -ENOMEDIUM No media inserted.
577 -EINVAL Invalid sector number or nb_sectors
578 -EACCES Trying to write a read-only device
579*/
5fafdf24 580int bdrv_write(BlockDriverState *bs, int64_t sector_num,
fc01f7e7
FB
581 const uint8_t *buf, int nb_sectors)
582{
83f64091 583 BlockDriver *drv = bs->drv;
19cb3738
FB
584 if (!bs->drv)
585 return -ENOMEDIUM;
0849bf08 586 if (bs->read_only)
19cb3738 587 return -EACCES;
71d0770c
AL
588 if (bdrv_check_request(bs, sector_num, nb_sectors))
589 return -EIO;
590
42fb2807 591 return drv->bdrv_write(bs, sector_num, buf, nb_sectors);
83f64091
FB
592}
593
eda578e5
AL
594int bdrv_pread(BlockDriverState *bs, int64_t offset,
595 void *buf, int count1)
83f64091 596{
83f64091
FB
597 uint8_t tmp_buf[SECTOR_SIZE];
598 int len, nb_sectors, count;
599 int64_t sector_num;
600
601 count = count1;
602 /* first read to align to sector start */
603 len = (SECTOR_SIZE - offset) & (SECTOR_SIZE - 1);
604 if (len > count)
605 len = count;
606 sector_num = offset >> SECTOR_BITS;
607 if (len > 0) {
608 if (bdrv_read(bs, sector_num, tmp_buf, 1) < 0)
609 return -EIO;
610 memcpy(buf, tmp_buf + (offset & (SECTOR_SIZE - 1)), len);
611 count -= len;
612 if (count == 0)
613 return count1;
614 sector_num++;
615 buf += len;
616 }
617
618 /* read the sectors "in place" */
619 nb_sectors = count >> SECTOR_BITS;
620 if (nb_sectors > 0) {
621 if (bdrv_read(bs, sector_num, buf, nb_sectors) < 0)
622 return -EIO;
623 sector_num += nb_sectors;
624 len = nb_sectors << SECTOR_BITS;
625 buf += len;
626 count -= len;
627 }
628
629 /* add data from the last sector */
630 if (count > 0) {
631 if (bdrv_read(bs, sector_num, tmp_buf, 1) < 0)
632 return -EIO;
633 memcpy(buf, tmp_buf, count);
634 }
635 return count1;
636}
637
eda578e5
AL
638int bdrv_pwrite(BlockDriverState *bs, int64_t offset,
639 const void *buf, int count1)
83f64091 640{
83f64091
FB
641 uint8_t tmp_buf[SECTOR_SIZE];
642 int len, nb_sectors, count;
643 int64_t sector_num;
644
645 count = count1;
646 /* first write to align to sector start */
647 len = (SECTOR_SIZE - offset) & (SECTOR_SIZE - 1);
648 if (len > count)
649 len = count;
650 sector_num = offset >> SECTOR_BITS;
651 if (len > 0) {
652 if (bdrv_read(bs, sector_num, tmp_buf, 1) < 0)
653 return -EIO;
654 memcpy(tmp_buf + (offset & (SECTOR_SIZE - 1)), buf, len);
655 if (bdrv_write(bs, sector_num, tmp_buf, 1) < 0)
656 return -EIO;
657 count -= len;
658 if (count == 0)
659 return count1;
660 sector_num++;
661 buf += len;
662 }
663
664 /* write the sectors "in place" */
665 nb_sectors = count >> SECTOR_BITS;
666 if (nb_sectors > 0) {
667 if (bdrv_write(bs, sector_num, buf, nb_sectors) < 0)
668 return -EIO;
669 sector_num += nb_sectors;
670 len = nb_sectors << SECTOR_BITS;
671 buf += len;
672 count -= len;
673 }
674
675 /* add data from the last sector */
676 if (count > 0) {
677 if (bdrv_read(bs, sector_num, tmp_buf, 1) < 0)
678 return -EIO;
679 memcpy(tmp_buf, buf, count);
680 if (bdrv_write(bs, sector_num, tmp_buf, 1) < 0)
681 return -EIO;
682 }
683 return count1;
684}
83f64091 685
83f64091
FB
686/**
687 * Truncate file to 'offset' bytes (needed only for file protocols)
688 */
689int bdrv_truncate(BlockDriverState *bs, int64_t offset)
690{
691 BlockDriver *drv = bs->drv;
692 if (!drv)
19cb3738 693 return -ENOMEDIUM;
83f64091
FB
694 if (!drv->bdrv_truncate)
695 return -ENOTSUP;
696 return drv->bdrv_truncate(bs, offset);
697}
698
699/**
700 * Length of a file in bytes. Return < 0 if error or unknown.
701 */
702int64_t bdrv_getlength(BlockDriverState *bs)
703{
704 BlockDriver *drv = bs->drv;
705 if (!drv)
19cb3738 706 return -ENOMEDIUM;
83f64091
FB
707 if (!drv->bdrv_getlength) {
708 /* legacy mode */
709 return bs->total_sectors * SECTOR_SIZE;
710 }
711 return drv->bdrv_getlength(bs);
fc01f7e7
FB
712}
713
19cb3738 714/* return 0 as number of sectors if no device present or error */
96b8f136 715void bdrv_get_geometry(BlockDriverState *bs, uint64_t *nb_sectors_ptr)
fc01f7e7 716{
19cb3738
FB
717 int64_t length;
718 length = bdrv_getlength(bs);
719 if (length < 0)
720 length = 0;
721 else
722 length = length >> SECTOR_BITS;
723 *nb_sectors_ptr = length;
fc01f7e7 724}
cf98951b 725
f3d54fc4
AL
726struct partition {
727 uint8_t boot_ind; /* 0x80 - active */
728 uint8_t head; /* starting head */
729 uint8_t sector; /* starting sector */
730 uint8_t cyl; /* starting cylinder */
731 uint8_t sys_ind; /* What partition type */
732 uint8_t end_head; /* end head */
733 uint8_t end_sector; /* end sector */
734 uint8_t end_cyl; /* end cylinder */
735 uint32_t start_sect; /* starting sector counting from 0 */
736 uint32_t nr_sects; /* nr of sectors in partition */
737} __attribute__((packed));
738
739/* try to guess the disk logical geometry from the MSDOS partition table. Return 0 if OK, -1 if could not guess */
740static int guess_disk_lchs(BlockDriverState *bs,
741 int *pcylinders, int *pheads, int *psectors)
742{
743 uint8_t buf[512];
744 int ret, i, heads, sectors, cylinders;
745 struct partition *p;
746 uint32_t nr_sects;
a38131b6 747 uint64_t nb_sectors;
f3d54fc4
AL
748
749 bdrv_get_geometry(bs, &nb_sectors);
750
751 ret = bdrv_read(bs, 0, buf, 1);
752 if (ret < 0)
753 return -1;
754 /* test msdos magic */
755 if (buf[510] != 0x55 || buf[511] != 0xaa)
756 return -1;
757 for(i = 0; i < 4; i++) {
758 p = ((struct partition *)(buf + 0x1be)) + i;
759 nr_sects = le32_to_cpu(p->nr_sects);
760 if (nr_sects && p->end_head) {
761 /* We make the assumption that the partition terminates on
762 a cylinder boundary */
763 heads = p->end_head + 1;
764 sectors = p->end_sector & 63;
765 if (sectors == 0)
766 continue;
767 cylinders = nb_sectors / (heads * sectors);
768 if (cylinders < 1 || cylinders > 16383)
769 continue;
770 *pheads = heads;
771 *psectors = sectors;
772 *pcylinders = cylinders;
773#if 0
774 printf("guessed geometry: LCHS=%d %d %d\n",
775 cylinders, heads, sectors);
776#endif
777 return 0;
778 }
779 }
780 return -1;
781}
782
783void bdrv_guess_geometry(BlockDriverState *bs, int *pcyls, int *pheads, int *psecs)
784{
785 int translation, lba_detected = 0;
786 int cylinders, heads, secs;
a38131b6 787 uint64_t nb_sectors;
f3d54fc4
AL
788
789 /* if a geometry hint is available, use it */
790 bdrv_get_geometry(bs, &nb_sectors);
791 bdrv_get_geometry_hint(bs, &cylinders, &heads, &secs);
792 translation = bdrv_get_translation_hint(bs);
793 if (cylinders != 0) {
794 *pcyls = cylinders;
795 *pheads = heads;
796 *psecs = secs;
797 } else {
798 if (guess_disk_lchs(bs, &cylinders, &heads, &secs) == 0) {
799 if (heads > 16) {
800 /* if heads > 16, it means that a BIOS LBA
801 translation was active, so the default
802 hardware geometry is OK */
803 lba_detected = 1;
804 goto default_geometry;
805 } else {
806 *pcyls = cylinders;
807 *pheads = heads;
808 *psecs = secs;
809 /* disable any translation to be in sync with
810 the logical geometry */
811 if (translation == BIOS_ATA_TRANSLATION_AUTO) {
812 bdrv_set_translation_hint(bs,
813 BIOS_ATA_TRANSLATION_NONE);
814 }
815 }
816 } else {
817 default_geometry:
818 /* if no geometry, use a standard physical disk geometry */
819 cylinders = nb_sectors / (16 * 63);
820
821 if (cylinders > 16383)
822 cylinders = 16383;
823 else if (cylinders < 2)
824 cylinders = 2;
825 *pcyls = cylinders;
826 *pheads = 16;
827 *psecs = 63;
828 if ((lba_detected == 1) && (translation == BIOS_ATA_TRANSLATION_AUTO)) {
829 if ((*pcyls * *pheads) <= 131072) {
830 bdrv_set_translation_hint(bs,
831 BIOS_ATA_TRANSLATION_LARGE);
832 } else {
833 bdrv_set_translation_hint(bs,
834 BIOS_ATA_TRANSLATION_LBA);
835 }
836 }
837 }
838 bdrv_set_geometry_hint(bs, *pcyls, *pheads, *psecs);
839 }
840}
841
5fafdf24 842void bdrv_set_geometry_hint(BlockDriverState *bs,
b338082b
FB
843 int cyls, int heads, int secs)
844{
845 bs->cyls = cyls;
846 bs->heads = heads;
847 bs->secs = secs;
848}
849
850void bdrv_set_type_hint(BlockDriverState *bs, int type)
851{
852 bs->type = type;
853 bs->removable = ((type == BDRV_TYPE_CDROM ||
854 type == BDRV_TYPE_FLOPPY));
855}
856
46d4767d
FB
857void bdrv_set_translation_hint(BlockDriverState *bs, int translation)
858{
859 bs->translation = translation;
860}
861
5fafdf24 862void bdrv_get_geometry_hint(BlockDriverState *bs,
b338082b
FB
863 int *pcyls, int *pheads, int *psecs)
864{
865 *pcyls = bs->cyls;
866 *pheads = bs->heads;
867 *psecs = bs->secs;
868}
869
870int bdrv_get_type_hint(BlockDriverState *bs)
871{
872 return bs->type;
873}
874
46d4767d
FB
875int bdrv_get_translation_hint(BlockDriverState *bs)
876{
877 return bs->translation;
878}
879
b338082b
FB
880int bdrv_is_removable(BlockDriverState *bs)
881{
882 return bs->removable;
883}
884
885int bdrv_is_read_only(BlockDriverState *bs)
886{
887 return bs->read_only;
888}
889
985a03b0
TS
890int bdrv_is_sg(BlockDriverState *bs)
891{
892 return bs->sg;
893}
894
19cb3738 895/* XXX: no longer used */
5fafdf24 896void bdrv_set_change_cb(BlockDriverState *bs,
b338082b
FB
897 void (*change_cb)(void *opaque), void *opaque)
898{
899 bs->change_cb = change_cb;
900 bs->change_opaque = opaque;
901}
902
ea2384d3
FB
903int bdrv_is_encrypted(BlockDriverState *bs)
904{
905 if (bs->backing_hd && bs->backing_hd->encrypted)
906 return 1;
907 return bs->encrypted;
908}
909
c0f4ce77
AL
910int bdrv_key_required(BlockDriverState *bs)
911{
912 BlockDriverState *backing_hd = bs->backing_hd;
913
914 if (backing_hd && backing_hd->encrypted && !backing_hd->valid_key)
915 return 1;
916 return (bs->encrypted && !bs->valid_key);
917}
918
ea2384d3
FB
919int bdrv_set_key(BlockDriverState *bs, const char *key)
920{
921 int ret;
922 if (bs->backing_hd && bs->backing_hd->encrypted) {
923 ret = bdrv_set_key(bs->backing_hd, key);
924 if (ret < 0)
925 return ret;
926 if (!bs->encrypted)
927 return 0;
928 }
929 if (!bs->encrypted || !bs->drv || !bs->drv->bdrv_set_key)
930 return -1;
c0f4ce77 931 ret = bs->drv->bdrv_set_key(bs, key);
bb5fc20f
AL
932 if (ret < 0) {
933 bs->valid_key = 0;
934 } else if (!bs->valid_key) {
935 bs->valid_key = 1;
936 /* call the change callback now, we skipped it on open */
937 bs->media_changed = 1;
938 if (bs->change_cb)
939 bs->change_cb(bs->change_opaque);
940 }
c0f4ce77 941 return ret;
ea2384d3
FB
942}
943
944void bdrv_get_format(BlockDriverState *bs, char *buf, int buf_size)
945{
19cb3738 946 if (!bs->drv) {
ea2384d3
FB
947 buf[0] = '\0';
948 } else {
949 pstrcpy(buf, buf_size, bs->drv->format_name);
950 }
951}
952
5fafdf24 953void bdrv_iterate_format(void (*it)(void *opaque, const char *name),
ea2384d3
FB
954 void *opaque)
955{
956 BlockDriver *drv;
957
958 for (drv = first_drv; drv != NULL; drv = drv->next) {
959 it(opaque, drv->format_name);
960 }
961}
962
b338082b
FB
963BlockDriverState *bdrv_find(const char *name)
964{
965 BlockDriverState *bs;
966
967 for (bs = bdrv_first; bs != NULL; bs = bs->next) {
968 if (!strcmp(name, bs->device_name))
969 return bs;
970 }
971 return NULL;
972}
973
51de9760 974void bdrv_iterate(void (*it)(void *opaque, BlockDriverState *bs), void *opaque)
81d0912d
FB
975{
976 BlockDriverState *bs;
977
978 for (bs = bdrv_first; bs != NULL; bs = bs->next) {
51de9760 979 it(opaque, bs);
81d0912d
FB
980 }
981}
982
ea2384d3
FB
983const char *bdrv_get_device_name(BlockDriverState *bs)
984{
985 return bs->device_name;
986}
987
7a6cba61
PB
988void bdrv_flush(BlockDriverState *bs)
989{
990 if (bs->drv->bdrv_flush)
991 bs->drv->bdrv_flush(bs);
992 if (bs->backing_hd)
993 bdrv_flush(bs->backing_hd);
994}
995
c6ca28d6
AL
996void bdrv_flush_all(void)
997{
998 BlockDriverState *bs;
999
1000 for (bs = bdrv_first; bs != NULL; bs = bs->next)
1001 if (bs->drv && !bdrv_is_read_only(bs) &&
1002 (!bdrv_is_removable(bs) || bdrv_is_inserted(bs)))
1003 bdrv_flush(bs);
1004}
1005
f58c7b35
TS
1006/*
1007 * Returns true iff the specified sector is present in the disk image. Drivers
1008 * not implementing the functionality are assumed to not support backing files,
1009 * hence all their sectors are reported as allocated.
1010 *
1011 * 'pnum' is set to the number of sectors (including and immediately following
1012 * the specified sector) that are known to be in the same
1013 * allocated/unallocated state.
1014 *
1015 * 'nb_sectors' is the max value 'pnum' should be set to.
1016 */
1017int bdrv_is_allocated(BlockDriverState *bs, int64_t sector_num, int nb_sectors,
1018 int *pnum)
1019{
1020 int64_t n;
1021 if (!bs->drv->bdrv_is_allocated) {
1022 if (sector_num >= bs->total_sectors) {
1023 *pnum = 0;
1024 return 0;
1025 }
1026 n = bs->total_sectors - sector_num;
1027 *pnum = (n < nb_sectors) ? (n) : (nb_sectors);
1028 return 1;
1029 }
1030 return bs->drv->bdrv_is_allocated(bs, sector_num, nb_sectors, pnum);
1031}
1032
376253ec 1033void bdrv_info(Monitor *mon)
b338082b
FB
1034{
1035 BlockDriverState *bs;
1036
1037 for (bs = bdrv_first; bs != NULL; bs = bs->next) {
376253ec
AL
1038 monitor_printf(mon, "%s:", bs->device_name);
1039 monitor_printf(mon, " type=");
b338082b
FB
1040 switch(bs->type) {
1041 case BDRV_TYPE_HD:
376253ec 1042 monitor_printf(mon, "hd");
b338082b
FB
1043 break;
1044 case BDRV_TYPE_CDROM:
376253ec 1045 monitor_printf(mon, "cdrom");
b338082b
FB
1046 break;
1047 case BDRV_TYPE_FLOPPY:
376253ec 1048 monitor_printf(mon, "floppy");
b338082b
FB
1049 break;
1050 }
376253ec 1051 monitor_printf(mon, " removable=%d", bs->removable);
b338082b 1052 if (bs->removable) {
376253ec 1053 monitor_printf(mon, " locked=%d", bs->locked);
b338082b 1054 }
19cb3738 1055 if (bs->drv) {
376253ec
AL
1056 monitor_printf(mon, " file=");
1057 monitor_print_filename(mon, bs->filename);
fef30743 1058 if (bs->backing_file[0] != '\0') {
376253ec
AL
1059 monitor_printf(mon, " backing_file=");
1060 monitor_print_filename(mon, bs->backing_file);
1061 }
1062 monitor_printf(mon, " ro=%d", bs->read_only);
1063 monitor_printf(mon, " drv=%s", bs->drv->format_name);
1064 monitor_printf(mon, " encrypted=%d", bdrv_is_encrypted(bs));
b338082b 1065 } else {
376253ec 1066 monitor_printf(mon, " [not inserted]");
b338082b 1067 }
376253ec 1068 monitor_printf(mon, "\n");
b338082b
FB
1069 }
1070}
a36e69dd
TS
1071
1072/* The "info blockstats" command. */
376253ec 1073void bdrv_info_stats(Monitor *mon)
a36e69dd
TS
1074{
1075 BlockDriverState *bs;
1076
1077 for (bs = bdrv_first; bs != NULL; bs = bs->next) {
376253ec
AL
1078 monitor_printf(mon, "%s:"
1079 " rd_bytes=%" PRIu64
1080 " wr_bytes=%" PRIu64
1081 " rd_operations=%" PRIu64
1082 " wr_operations=%" PRIu64
ebf53fcd 1083 "\n",
376253ec
AL
1084 bs->device_name,
1085 bs->rd_bytes, bs->wr_bytes,
1086 bs->rd_ops, bs->wr_ops);
a36e69dd
TS
1087 }
1088}
ea2384d3 1089
045df330
AL
1090const char *bdrv_get_encrypted_filename(BlockDriverState *bs)
1091{
1092 if (bs->backing_hd && bs->backing_hd->encrypted)
1093 return bs->backing_file;
1094 else if (bs->encrypted)
1095 return bs->filename;
1096 else
1097 return NULL;
1098}
1099
5fafdf24 1100void bdrv_get_backing_filename(BlockDriverState *bs,
83f64091
FB
1101 char *filename, int filename_size)
1102{
1103 if (!bs->backing_hd) {
1104 pstrcpy(filename, filename_size, "");
1105 } else {
1106 pstrcpy(filename, filename_size, bs->backing_file);
1107 }
1108}
1109
5fafdf24 1110int bdrv_write_compressed(BlockDriverState *bs, int64_t sector_num,
faea38e7
FB
1111 const uint8_t *buf, int nb_sectors)
1112{
1113 BlockDriver *drv = bs->drv;
1114 if (!drv)
19cb3738 1115 return -ENOMEDIUM;
faea38e7
FB
1116 if (!drv->bdrv_write_compressed)
1117 return -ENOTSUP;
1118 return drv->bdrv_write_compressed(bs, sector_num, buf, nb_sectors);
1119}
3b46e624 1120
faea38e7
FB
1121int bdrv_get_info(BlockDriverState *bs, BlockDriverInfo *bdi)
1122{
1123 BlockDriver *drv = bs->drv;
1124 if (!drv)
19cb3738 1125 return -ENOMEDIUM;
faea38e7
FB
1126 if (!drv->bdrv_get_info)
1127 return -ENOTSUP;
1128 memset(bdi, 0, sizeof(*bdi));
1129 return drv->bdrv_get_info(bs, bdi);
1130}
1131
1132/**************************************************************/
1133/* handling of snapshots */
1134
5fafdf24 1135int bdrv_snapshot_create(BlockDriverState *bs,
faea38e7
FB
1136 QEMUSnapshotInfo *sn_info)
1137{
1138 BlockDriver *drv = bs->drv;
1139 if (!drv)
19cb3738 1140 return -ENOMEDIUM;
faea38e7
FB
1141 if (!drv->bdrv_snapshot_create)
1142 return -ENOTSUP;
1143 return drv->bdrv_snapshot_create(bs, sn_info);
1144}
1145
5fafdf24 1146int bdrv_snapshot_goto(BlockDriverState *bs,
faea38e7
FB
1147 const char *snapshot_id)
1148{
1149 BlockDriver *drv = bs->drv;
1150 if (!drv)
19cb3738 1151 return -ENOMEDIUM;
faea38e7
FB
1152 if (!drv->bdrv_snapshot_goto)
1153 return -ENOTSUP;
1154 return drv->bdrv_snapshot_goto(bs, snapshot_id);
1155}
1156
1157int bdrv_snapshot_delete(BlockDriverState *bs, const char *snapshot_id)
1158{
1159 BlockDriver *drv = bs->drv;
1160 if (!drv)
19cb3738 1161 return -ENOMEDIUM;
faea38e7
FB
1162 if (!drv->bdrv_snapshot_delete)
1163 return -ENOTSUP;
1164 return drv->bdrv_snapshot_delete(bs, snapshot_id);
1165}
1166
5fafdf24 1167int bdrv_snapshot_list(BlockDriverState *bs,
faea38e7
FB
1168 QEMUSnapshotInfo **psn_info)
1169{
1170 BlockDriver *drv = bs->drv;
1171 if (!drv)
19cb3738 1172 return -ENOMEDIUM;
faea38e7
FB
1173 if (!drv->bdrv_snapshot_list)
1174 return -ENOTSUP;
1175 return drv->bdrv_snapshot_list(bs, psn_info);
1176}
1177
1178#define NB_SUFFIXES 4
1179
1180char *get_human_readable_size(char *buf, int buf_size, int64_t size)
1181{
1182 static const char suffixes[NB_SUFFIXES] = "KMGT";
1183 int64_t base;
1184 int i;
1185
1186 if (size <= 999) {
1187 snprintf(buf, buf_size, "%" PRId64, size);
1188 } else {
1189 base = 1024;
1190 for(i = 0; i < NB_SUFFIXES; i++) {
1191 if (size < (10 * base)) {
5fafdf24 1192 snprintf(buf, buf_size, "%0.1f%c",
faea38e7
FB
1193 (double)size / base,
1194 suffixes[i]);
1195 break;
1196 } else if (size < (1000 * base) || i == (NB_SUFFIXES - 1)) {
5fafdf24 1197 snprintf(buf, buf_size, "%" PRId64 "%c",
faea38e7
FB
1198 ((size + (base >> 1)) / base),
1199 suffixes[i]);
1200 break;
1201 }
1202 base = base * 1024;
1203 }
1204 }
1205 return buf;
1206}
1207
1208char *bdrv_snapshot_dump(char *buf, int buf_size, QEMUSnapshotInfo *sn)
1209{
1210 char buf1[128], date_buf[128], clock_buf[128];
3b9f94e1
FB
1211#ifdef _WIN32
1212 struct tm *ptm;
1213#else
faea38e7 1214 struct tm tm;
3b9f94e1 1215#endif
faea38e7
FB
1216 time_t ti;
1217 int64_t secs;
1218
1219 if (!sn) {
5fafdf24
TS
1220 snprintf(buf, buf_size,
1221 "%-10s%-20s%7s%20s%15s",
faea38e7
FB
1222 "ID", "TAG", "VM SIZE", "DATE", "VM CLOCK");
1223 } else {
1224 ti = sn->date_sec;
3b9f94e1
FB
1225#ifdef _WIN32
1226 ptm = localtime(&ti);
1227 strftime(date_buf, sizeof(date_buf),
1228 "%Y-%m-%d %H:%M:%S", ptm);
1229#else
faea38e7
FB
1230 localtime_r(&ti, &tm);
1231 strftime(date_buf, sizeof(date_buf),
1232 "%Y-%m-%d %H:%M:%S", &tm);
3b9f94e1 1233#endif
faea38e7
FB
1234 secs = sn->vm_clock_nsec / 1000000000;
1235 snprintf(clock_buf, sizeof(clock_buf),
1236 "%02d:%02d:%02d.%03d",
1237 (int)(secs / 3600),
1238 (int)((secs / 60) % 60),
5fafdf24 1239 (int)(secs % 60),
faea38e7
FB
1240 (int)((sn->vm_clock_nsec / 1000000) % 1000));
1241 snprintf(buf, buf_size,
5fafdf24 1242 "%-10s%-20s%7s%20s%15s",
faea38e7
FB
1243 sn->id_str, sn->name,
1244 get_human_readable_size(buf1, sizeof(buf1), sn->vm_state_size),
1245 date_buf,
1246 clock_buf);
1247 }
1248 return buf;
1249}
1250
83f64091 1251
ea2384d3 1252/**************************************************************/
83f64091 1253/* async I/Os */
ea2384d3 1254
3b69e4b9
AL
1255typedef struct VectorTranslationState {
1256 QEMUIOVector *iov;
1257 uint8_t *bounce;
1258 int is_write;
1259 BlockDriverAIOCB *aiocb;
1260 BlockDriverAIOCB *this_aiocb;
1261} VectorTranslationState;
1262
1263static void bdrv_aio_rw_vector_cb(void *opaque, int ret)
1264{
1265 VectorTranslationState *s = opaque;
1266
1267 if (!s->is_write) {
249aa745 1268 qemu_iovec_from_buffer(s->iov, s->bounce, s->iov->size);
3b69e4b9 1269 }
d905dba4 1270 qemu_vfree(s->bounce);
3b69e4b9
AL
1271 s->this_aiocb->cb(s->this_aiocb->opaque, ret);
1272 qemu_aio_release(s->this_aiocb);
1273}
1274
1275static BlockDriverAIOCB *bdrv_aio_rw_vector(BlockDriverState *bs,
1276 int64_t sector_num,
1277 QEMUIOVector *iov,
1278 int nb_sectors,
1279 BlockDriverCompletionFunc *cb,
1280 void *opaque,
1281 int is_write)
1282
1283{
1284 VectorTranslationState *s = qemu_mallocz(sizeof(*s));
1285 BlockDriverAIOCB *aiocb = qemu_aio_get(bs, cb, opaque);
1286
1287 s->this_aiocb = aiocb;
1288 s->iov = iov;
1289 s->bounce = qemu_memalign(512, nb_sectors * 512);
1290 s->is_write = is_write;
1291 if (is_write) {
1292 qemu_iovec_to_buffer(s->iov, s->bounce);
1293 s->aiocb = bdrv_aio_write(bs, sector_num, s->bounce, nb_sectors,
1294 bdrv_aio_rw_vector_cb, s);
1295 } else {
1296 s->aiocb = bdrv_aio_read(bs, sector_num, s->bounce, nb_sectors,
1297 bdrv_aio_rw_vector_cb, s);
1298 }
1299 return aiocb;
1300}
1301
1302BlockDriverAIOCB *bdrv_aio_readv(BlockDriverState *bs, int64_t sector_num,
1303 QEMUIOVector *iov, int nb_sectors,
1304 BlockDriverCompletionFunc *cb, void *opaque)
1305{
71d0770c
AL
1306 if (bdrv_check_request(bs, sector_num, nb_sectors))
1307 return NULL;
1308
3b69e4b9
AL
1309 return bdrv_aio_rw_vector(bs, sector_num, iov, nb_sectors,
1310 cb, opaque, 0);
1311}
1312
1313BlockDriverAIOCB *bdrv_aio_writev(BlockDriverState *bs, int64_t sector_num,
1314 QEMUIOVector *iov, int nb_sectors,
1315 BlockDriverCompletionFunc *cb, void *opaque)
1316{
71d0770c
AL
1317 if (bdrv_check_request(bs, sector_num, nb_sectors))
1318 return NULL;
1319
3b69e4b9
AL
1320 return bdrv_aio_rw_vector(bs, sector_num, iov, nb_sectors,
1321 cb, opaque, 1);
1322}
1323
ce1a14dc
PB
1324BlockDriverAIOCB *bdrv_aio_read(BlockDriverState *bs, int64_t sector_num,
1325 uint8_t *buf, int nb_sectors,
1326 BlockDriverCompletionFunc *cb, void *opaque)
83f64091
FB
1327{
1328 BlockDriver *drv = bs->drv;
a36e69dd 1329 BlockDriverAIOCB *ret;
83f64091 1330
19cb3738 1331 if (!drv)
ce1a14dc 1332 return NULL;
71d0770c
AL
1333 if (bdrv_check_request(bs, sector_num, nb_sectors))
1334 return NULL;
3b46e624 1335
a36e69dd
TS
1336 ret = drv->bdrv_aio_read(bs, sector_num, buf, nb_sectors, cb, opaque);
1337
1338 if (ret) {
1339 /* Update stats even though technically transfer has not happened. */
1340 bs->rd_bytes += (unsigned) nb_sectors * SECTOR_SIZE;
1341 bs->rd_ops ++;
1342 }
1343
1344 return ret;
ea2384d3
FB
1345}
1346
ce1a14dc
PB
1347BlockDriverAIOCB *bdrv_aio_write(BlockDriverState *bs, int64_t sector_num,
1348 const uint8_t *buf, int nb_sectors,
1349 BlockDriverCompletionFunc *cb, void *opaque)
ea2384d3 1350{
83f64091 1351 BlockDriver *drv = bs->drv;
a36e69dd 1352 BlockDriverAIOCB *ret;
ea2384d3 1353
19cb3738 1354 if (!drv)
ce1a14dc 1355 return NULL;
83f64091 1356 if (bs->read_only)
ce1a14dc 1357 return NULL;
71d0770c
AL
1358 if (bdrv_check_request(bs, sector_num, nb_sectors))
1359 return NULL;
83f64091 1360
a36e69dd
TS
1361 ret = drv->bdrv_aio_write(bs, sector_num, buf, nb_sectors, cb, opaque);
1362
1363 if (ret) {
1364 /* Update stats even though technically transfer has not happened. */
1365 bs->wr_bytes += (unsigned) nb_sectors * SECTOR_SIZE;
1366 bs->wr_ops ++;
1367 }
1368
1369 return ret;
83f64091
FB
1370}
1371
1372void bdrv_aio_cancel(BlockDriverAIOCB *acb)
83f64091 1373{
ce1a14dc 1374 BlockDriver *drv = acb->bs->drv;
83f64091 1375
3b69e4b9
AL
1376 if (acb->cb == bdrv_aio_rw_vector_cb) {
1377 VectorTranslationState *s = acb->opaque;
1378 acb = s->aiocb;
1379 }
1380
ce1a14dc 1381 drv->bdrv_aio_cancel(acb);
83f64091
FB
1382}
1383
ce1a14dc 1384
83f64091
FB
1385/**************************************************************/
1386/* async block device emulation */
1387
ce1a14dc 1388static void bdrv_aio_bh_cb(void *opaque)
83f64091 1389{
ce1a14dc
PB
1390 BlockDriverAIOCBSync *acb = opaque;
1391 acb->common.cb(acb->common.opaque, acb->ret);
1392 qemu_aio_release(acb);
83f64091 1393}
beac80cd 1394
ce1a14dc
PB
1395static BlockDriverAIOCB *bdrv_aio_read_em(BlockDriverState *bs,
1396 int64_t sector_num, uint8_t *buf, int nb_sectors,
1397 BlockDriverCompletionFunc *cb, void *opaque)
83f64091 1398{
ce1a14dc 1399 BlockDriverAIOCBSync *acb;
83f64091 1400 int ret;
ce1a14dc
PB
1401
1402 acb = qemu_aio_get(bs, cb, opaque);
1403 if (!acb->bh)
1404 acb->bh = qemu_bh_new(bdrv_aio_bh_cb, acb);
1405 ret = bdrv_read(bs, sector_num, buf, nb_sectors);
1406 acb->ret = ret;
1407 qemu_bh_schedule(acb->bh);
1408 return &acb->common;
beac80cd
FB
1409}
1410
ce1a14dc
PB
1411static BlockDriverAIOCB *bdrv_aio_write_em(BlockDriverState *bs,
1412 int64_t sector_num, const uint8_t *buf, int nb_sectors,
1413 BlockDriverCompletionFunc *cb, void *opaque)
beac80cd 1414{
ce1a14dc 1415 BlockDriverAIOCBSync *acb;
83f64091 1416 int ret;
83f64091 1417
ce1a14dc
PB
1418 acb = qemu_aio_get(bs, cb, opaque);
1419 if (!acb->bh)
1420 acb->bh = qemu_bh_new(bdrv_aio_bh_cb, acb);
1421 ret = bdrv_write(bs, sector_num, buf, nb_sectors);
1422 acb->ret = ret;
1423 qemu_bh_schedule(acb->bh);
1424 return &acb->common;
beac80cd 1425}
beac80cd 1426
ce1a14dc 1427static void bdrv_aio_cancel_em(BlockDriverAIOCB *blockacb)
ea2384d3 1428{
ce1a14dc
PB
1429 BlockDriverAIOCBSync *acb = (BlockDriverAIOCBSync *)blockacb;
1430 qemu_bh_cancel(acb->bh);
1431 qemu_aio_release(acb);
83f64091 1432}
ea2384d3 1433
83f64091
FB
1434/**************************************************************/
1435/* sync block device emulation */
ea2384d3 1436
83f64091
FB
1437static void bdrv_rw_em_cb(void *opaque, int ret)
1438{
1439 *(int *)opaque = ret;
ea2384d3
FB
1440}
1441
83f64091
FB
1442#define NOT_DONE 0x7fffffff
1443
5fafdf24 1444static int bdrv_read_em(BlockDriverState *bs, int64_t sector_num,
83f64091 1445 uint8_t *buf, int nb_sectors)
7a6cba61 1446{
ce1a14dc
PB
1447 int async_ret;
1448 BlockDriverAIOCB *acb;
83f64091 1449
83f64091 1450 async_ret = NOT_DONE;
5fafdf24 1451 acb = bdrv_aio_read(bs, sector_num, buf, nb_sectors,
83f64091 1452 bdrv_rw_em_cb, &async_ret);
baf35cb9 1453 if (acb == NULL)
ce1a14dc 1454 return -1;
baf35cb9 1455
83f64091
FB
1456 while (async_ret == NOT_DONE) {
1457 qemu_aio_wait();
1458 }
baf35cb9 1459
83f64091 1460 return async_ret;
7a6cba61
PB
1461}
1462
83f64091
FB
1463static int bdrv_write_em(BlockDriverState *bs, int64_t sector_num,
1464 const uint8_t *buf, int nb_sectors)
1465{
ce1a14dc
PB
1466 int async_ret;
1467 BlockDriverAIOCB *acb;
83f64091 1468
83f64091 1469 async_ret = NOT_DONE;
5fafdf24 1470 acb = bdrv_aio_write(bs, sector_num, buf, nb_sectors,
83f64091 1471 bdrv_rw_em_cb, &async_ret);
baf35cb9 1472 if (acb == NULL)
ce1a14dc 1473 return -1;
83f64091
FB
1474 while (async_ret == NOT_DONE) {
1475 qemu_aio_wait();
1476 }
83f64091
FB
1477 return async_ret;
1478}
ea2384d3
FB
1479
1480void bdrv_init(void)
1481{
1482 bdrv_register(&bdrv_raw);
19cb3738 1483 bdrv_register(&bdrv_host_device);
ea2384d3
FB
1484#ifndef _WIN32
1485 bdrv_register(&bdrv_cow);
1486#endif
1487 bdrv_register(&bdrv_qcow);
1488 bdrv_register(&bdrv_vmdk);
3c56521b 1489 bdrv_register(&bdrv_cloop);
585d0ed9 1490 bdrv_register(&bdrv_dmg);
a8753c34 1491 bdrv_register(&bdrv_bochs);
6a0f9e82 1492 bdrv_register(&bdrv_vpc);
712e7874 1493 bdrv_register(&bdrv_vvfat);
faea38e7 1494 bdrv_register(&bdrv_qcow2);
6ada7453 1495 bdrv_register(&bdrv_parallels);
75818250 1496 bdrv_register(&bdrv_nbd);
ea2384d3 1497}
ce1a14dc
PB
1498
1499void *qemu_aio_get(BlockDriverState *bs, BlockDriverCompletionFunc *cb,
1500 void *opaque)
1501{
1502 BlockDriver *drv;
1503 BlockDriverAIOCB *acb;
1504
1505 drv = bs->drv;
1506 if (drv->free_aiocb) {
1507 acb = drv->free_aiocb;
1508 drv->free_aiocb = acb->next;
1509 } else {
1510 acb = qemu_mallocz(drv->aiocb_size);
ce1a14dc
PB
1511 }
1512 acb->bs = bs;
1513 acb->cb = cb;
1514 acb->opaque = opaque;
1515 return acb;
1516}
1517
1518void qemu_aio_release(void *p)
1519{
1520 BlockDriverAIOCB *acb = p;
1521 BlockDriver *drv = acb->bs->drv;
1522 acb->next = drv->free_aiocb;
1523 drv->free_aiocb = acb;
1524}
19cb3738
FB
1525
1526/**************************************************************/
1527/* removable device support */
1528
1529/**
1530 * Return TRUE if the media is present
1531 */
1532int bdrv_is_inserted(BlockDriverState *bs)
1533{
1534 BlockDriver *drv = bs->drv;
1535 int ret;
1536 if (!drv)
1537 return 0;
1538 if (!drv->bdrv_is_inserted)
1539 return 1;
1540 ret = drv->bdrv_is_inserted(bs);
1541 return ret;
1542}
1543
1544/**
1545 * Return TRUE if the media changed since the last call to this
5fafdf24 1546 * function. It is currently only used for floppy disks
19cb3738
FB
1547 */
1548int bdrv_media_changed(BlockDriverState *bs)
1549{
1550 BlockDriver *drv = bs->drv;
1551 int ret;
1552
1553 if (!drv || !drv->bdrv_media_changed)
1554 ret = -ENOTSUP;
1555 else
1556 ret = drv->bdrv_media_changed(bs);
1557 if (ret == -ENOTSUP)
1558 ret = bs->media_changed;
1559 bs->media_changed = 0;
1560 return ret;
1561}
1562
1563/**
1564 * If eject_flag is TRUE, eject the media. Otherwise, close the tray
1565 */
1566void bdrv_eject(BlockDriverState *bs, int eject_flag)
1567{
1568 BlockDriver *drv = bs->drv;
1569 int ret;
1570
1571 if (!drv || !drv->bdrv_eject) {
1572 ret = -ENOTSUP;
1573 } else {
1574 ret = drv->bdrv_eject(bs, eject_flag);
1575 }
1576 if (ret == -ENOTSUP) {
1577 if (eject_flag)
1578 bdrv_close(bs);
1579 }
1580}
1581
1582int bdrv_is_locked(BlockDriverState *bs)
1583{
1584 return bs->locked;
1585}
1586
1587/**
1588 * Lock or unlock the media (if it is locked, the user won't be able
1589 * to eject it manually).
1590 */
1591void bdrv_set_locked(BlockDriverState *bs, int locked)
1592{
1593 BlockDriver *drv = bs->drv;
1594
1595 bs->locked = locked;
1596 if (drv && drv->bdrv_set_locked) {
1597 drv->bdrv_set_locked(bs, locked);
1598 }
1599}
985a03b0
TS
1600
1601/* needed for generic scsi interface */
1602
1603int bdrv_ioctl(BlockDriverState *bs, unsigned long int req, void *buf)
1604{
1605 BlockDriver *drv = bs->drv;
1606
1607 if (drv && drv->bdrv_ioctl)
1608 return drv->bdrv_ioctl(bs, req, buf);
1609 return -ENOTSUP;
1610}
7d780669
AL
1611
1612int bdrv_sg_send_command(BlockDriverState *bs, void *buf, int count)
1613{
04eeb8b6 1614 return bs->drv->bdrv_sg_send_command(bs, buf, count);
7d780669
AL
1615}
1616
1617int bdrv_sg_recv_response(BlockDriverState *bs, void *buf, int count)
1618{
04eeb8b6 1619 return bs->drv->bdrv_sg_recv_response(bs, buf, count);
7d780669
AL
1620}
1621
1622BlockDriverAIOCB *bdrv_sg_aio_read(BlockDriverState *bs, void *buf, int count,
1623 BlockDriverCompletionFunc *cb, void *opaque)
1624{
04eeb8b6 1625 return bs->drv->bdrv_sg_aio_read(bs, buf, count, cb, opaque);
7d780669
AL
1626}
1627
1628BlockDriverAIOCB *bdrv_sg_aio_write(BlockDriverState *bs, void *buf, int count,
1629 BlockDriverCompletionFunc *cb, void *opaque)
1630{
04eeb8b6 1631 return bs->drv->bdrv_sg_aio_write(bs, buf, count, cb, opaque);
7d780669 1632}