]> git.proxmox.com Git - mirror_qemu.git/blob - block/raw-posix.c
linux-aio: make it more type safe
[mirror_qemu.git] / block / raw-posix.c
1 /*
2 * Block driver for RAW files (posix)
3 *
4 * Copyright (c) 2006 Fabrice Bellard
5 *
6 * Permission is hereby granted, free of charge, to any person obtaining a copy
7 * of this software and associated documentation files (the "Software"), to deal
8 * in the Software without restriction, including without limitation the rights
9 * to use, copy, modify, merge, publish, distribute, sublicense, and/or sell
10 * copies of the Software, and to permit persons to whom the Software is
11 * furnished to do so, subject to the following conditions:
12 *
13 * The above copyright notice and this permission notice shall be included in
14 * all copies or substantial portions of the Software.
15 *
16 * THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
17 * IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
18 * FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL
19 * THE AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
20 * LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,
21 * OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN
22 * THE SOFTWARE.
23 */
24 #include "qemu/osdep.h"
25 #include "qapi/error.h"
26 #include "qemu/cutils.h"
27 #include "qemu/error-report.h"
28 #include "qemu/timer.h"
29 #include "qemu/log.h"
30 #include "block/block_int.h"
31 #include "qemu/module.h"
32 #include "trace.h"
33 #include "block/thread-pool.h"
34 #include "qemu/iov.h"
35 #include "raw-aio.h"
36 #include "qapi/util.h"
37 #include "qapi/qmp/qstring.h"
38
39 #if defined(__APPLE__) && (__MACH__)
40 #include <paths.h>
41 #include <sys/param.h>
42 #include <IOKit/IOKitLib.h>
43 #include <IOKit/IOBSD.h>
44 #include <IOKit/storage/IOMediaBSDClient.h>
45 #include <IOKit/storage/IOMedia.h>
46 #include <IOKit/storage/IOCDMedia.h>
47 //#include <IOKit/storage/IOCDTypes.h>
48 #include <IOKit/storage/IODVDMedia.h>
49 #include <CoreFoundation/CoreFoundation.h>
50 #endif
51
52 #ifdef __sun__
53 #define _POSIX_PTHREAD_SEMANTICS 1
54 #include <sys/dkio.h>
55 #endif
56 #ifdef __linux__
57 #include <sys/ioctl.h>
58 #include <sys/param.h>
59 #include <linux/cdrom.h>
60 #include <linux/fd.h>
61 #include <linux/fs.h>
62 #include <linux/hdreg.h>
63 #include <scsi/sg.h>
64 #ifdef __s390__
65 #include <asm/dasd.h>
66 #endif
67 #ifndef FS_NOCOW_FL
68 #define FS_NOCOW_FL 0x00800000 /* Do not cow file */
69 #endif
70 #endif
71 #if defined(CONFIG_FALLOCATE_PUNCH_HOLE) || defined(CONFIG_FALLOCATE_ZERO_RANGE)
72 #include <linux/falloc.h>
73 #endif
74 #if defined (__FreeBSD__) || defined(__FreeBSD_kernel__)
75 #include <sys/disk.h>
76 #include <sys/cdio.h>
77 #endif
78
79 #ifdef __OpenBSD__
80 #include <sys/ioctl.h>
81 #include <sys/disklabel.h>
82 #include <sys/dkio.h>
83 #endif
84
85 #ifdef __NetBSD__
86 #include <sys/ioctl.h>
87 #include <sys/disklabel.h>
88 #include <sys/dkio.h>
89 #include <sys/disk.h>
90 #endif
91
92 #ifdef __DragonFly__
93 #include <sys/ioctl.h>
94 #include <sys/diskslice.h>
95 #endif
96
97 #ifdef CONFIG_XFS
98 #include <xfs/xfs.h>
99 #endif
100
101 //#define DEBUG_BLOCK
102
103 #ifdef DEBUG_BLOCK
104 # define DEBUG_BLOCK_PRINT 1
105 #else
106 # define DEBUG_BLOCK_PRINT 0
107 #endif
108 #define DPRINTF(fmt, ...) \
109 do { \
110 if (DEBUG_BLOCK_PRINT) { \
111 printf(fmt, ## __VA_ARGS__); \
112 } \
113 } while (0)
114
115 /* OS X does not have O_DSYNC */
116 #ifndef O_DSYNC
117 #ifdef O_SYNC
118 #define O_DSYNC O_SYNC
119 #elif defined(O_FSYNC)
120 #define O_DSYNC O_FSYNC
121 #endif
122 #endif
123
124 /* Approximate O_DIRECT with O_DSYNC if O_DIRECT isn't available */
125 #ifndef O_DIRECT
126 #define O_DIRECT O_DSYNC
127 #endif
128
129 #define FTYPE_FILE 0
130 #define FTYPE_CD 1
131
132 #define MAX_BLOCKSIZE 4096
133
134 typedef struct BDRVRawState {
135 int fd;
136 int type;
137 int open_flags;
138 size_t buf_align;
139
140 #ifdef CONFIG_LINUX_AIO
141 int use_aio;
142 LinuxAioState *aio_ctx;
143 #endif
144 #ifdef CONFIG_XFS
145 bool is_xfs:1;
146 #endif
147 bool has_discard:1;
148 bool has_write_zeroes:1;
149 bool discard_zeroes:1;
150 bool has_fallocate;
151 bool needs_alignment;
152 } BDRVRawState;
153
154 typedef struct BDRVRawReopenState {
155 int fd;
156 int open_flags;
157 #ifdef CONFIG_LINUX_AIO
158 int use_aio;
159 #endif
160 } BDRVRawReopenState;
161
162 static int fd_open(BlockDriverState *bs);
163 static int64_t raw_getlength(BlockDriverState *bs);
164
165 typedef struct RawPosixAIOData {
166 BlockDriverState *bs;
167 int aio_fildes;
168 union {
169 struct iovec *aio_iov;
170 void *aio_ioctl_buf;
171 };
172 int aio_niov;
173 uint64_t aio_nbytes;
174 #define aio_ioctl_cmd aio_nbytes /* for QEMU_AIO_IOCTL */
175 off_t aio_offset;
176 int aio_type;
177 } RawPosixAIOData;
178
179 #if defined(__FreeBSD__) || defined(__FreeBSD_kernel__)
180 static int cdrom_reopen(BlockDriverState *bs);
181 #endif
182
183 #if defined(__NetBSD__)
184 static int raw_normalize_devicepath(const char **filename)
185 {
186 static char namebuf[PATH_MAX];
187 const char *dp, *fname;
188 struct stat sb;
189
190 fname = *filename;
191 dp = strrchr(fname, '/');
192 if (lstat(fname, &sb) < 0) {
193 fprintf(stderr, "%s: stat failed: %s\n",
194 fname, strerror(errno));
195 return -errno;
196 }
197
198 if (!S_ISBLK(sb.st_mode)) {
199 return 0;
200 }
201
202 if (dp == NULL) {
203 snprintf(namebuf, PATH_MAX, "r%s", fname);
204 } else {
205 snprintf(namebuf, PATH_MAX, "%.*s/r%s",
206 (int)(dp - fname), fname, dp + 1);
207 }
208 fprintf(stderr, "%s is a block device", fname);
209 *filename = namebuf;
210 fprintf(stderr, ", using %s\n", *filename);
211
212 return 0;
213 }
214 #else
215 static int raw_normalize_devicepath(const char **filename)
216 {
217 return 0;
218 }
219 #endif
220
221 /*
222 * Get logical block size via ioctl. On success store it in @sector_size_p.
223 */
224 static int probe_logical_blocksize(int fd, unsigned int *sector_size_p)
225 {
226 unsigned int sector_size;
227 bool success = false;
228
229 errno = ENOTSUP;
230
231 /* Try a few ioctls to get the right size */
232 #ifdef BLKSSZGET
233 if (ioctl(fd, BLKSSZGET, &sector_size) >= 0) {
234 *sector_size_p = sector_size;
235 success = true;
236 }
237 #endif
238 #ifdef DKIOCGETBLOCKSIZE
239 if (ioctl(fd, DKIOCGETBLOCKSIZE, &sector_size) >= 0) {
240 *sector_size_p = sector_size;
241 success = true;
242 }
243 #endif
244 #ifdef DIOCGSECTORSIZE
245 if (ioctl(fd, DIOCGSECTORSIZE, &sector_size) >= 0) {
246 *sector_size_p = sector_size;
247 success = true;
248 }
249 #endif
250
251 return success ? 0 : -errno;
252 }
253
254 /**
255 * Get physical block size of @fd.
256 * On success, store it in @blk_size and return 0.
257 * On failure, return -errno.
258 */
259 static int probe_physical_blocksize(int fd, unsigned int *blk_size)
260 {
261 #ifdef BLKPBSZGET
262 if (ioctl(fd, BLKPBSZGET, blk_size) < 0) {
263 return -errno;
264 }
265 return 0;
266 #else
267 return -ENOTSUP;
268 #endif
269 }
270
271 /* Check if read is allowed with given memory buffer and length.
272 *
273 * This function is used to check O_DIRECT memory buffer and request alignment.
274 */
275 static bool raw_is_io_aligned(int fd, void *buf, size_t len)
276 {
277 ssize_t ret = pread(fd, buf, len, 0);
278
279 if (ret >= 0) {
280 return true;
281 }
282
283 #ifdef __linux__
284 /* The Linux kernel returns EINVAL for misaligned O_DIRECT reads. Ignore
285 * other errors (e.g. real I/O error), which could happen on a failed
286 * drive, since we only care about probing alignment.
287 */
288 if (errno != EINVAL) {
289 return true;
290 }
291 #endif
292
293 return false;
294 }
295
296 static void raw_probe_alignment(BlockDriverState *bs, int fd, Error **errp)
297 {
298 BDRVRawState *s = bs->opaque;
299 char *buf;
300 size_t max_align = MAX(MAX_BLOCKSIZE, getpagesize());
301
302 /* For SCSI generic devices the alignment is not really used.
303 With buffered I/O, we don't have any restrictions. */
304 if (bdrv_is_sg(bs) || !s->needs_alignment) {
305 bs->request_alignment = 1;
306 s->buf_align = 1;
307 return;
308 }
309
310 bs->request_alignment = 0;
311 s->buf_align = 0;
312 /* Let's try to use the logical blocksize for the alignment. */
313 if (probe_logical_blocksize(fd, &bs->request_alignment) < 0) {
314 bs->request_alignment = 0;
315 }
316 #ifdef CONFIG_XFS
317 if (s->is_xfs) {
318 struct dioattr da;
319 if (xfsctl(NULL, fd, XFS_IOC_DIOINFO, &da) >= 0) {
320 bs->request_alignment = da.d_miniosz;
321 /* The kernel returns wrong information for d_mem */
322 /* s->buf_align = da.d_mem; */
323 }
324 }
325 #endif
326
327 /* If we could not get the sizes so far, we can only guess them */
328 if (!s->buf_align) {
329 size_t align;
330 buf = qemu_memalign(max_align, 2 * max_align);
331 for (align = 512; align <= max_align; align <<= 1) {
332 if (raw_is_io_aligned(fd, buf + align, max_align)) {
333 s->buf_align = align;
334 break;
335 }
336 }
337 qemu_vfree(buf);
338 }
339
340 if (!bs->request_alignment) {
341 size_t align;
342 buf = qemu_memalign(s->buf_align, max_align);
343 for (align = 512; align <= max_align; align <<= 1) {
344 if (raw_is_io_aligned(fd, buf, align)) {
345 bs->request_alignment = align;
346 break;
347 }
348 }
349 qemu_vfree(buf);
350 }
351
352 if (!s->buf_align || !bs->request_alignment) {
353 error_setg(errp, "Could not find working O_DIRECT alignment. "
354 "Try cache.direct=off.");
355 }
356 }
357
358 static void raw_parse_flags(int bdrv_flags, int *open_flags)
359 {
360 assert(open_flags != NULL);
361
362 *open_flags |= O_BINARY;
363 *open_flags &= ~O_ACCMODE;
364 if (bdrv_flags & BDRV_O_RDWR) {
365 *open_flags |= O_RDWR;
366 } else {
367 *open_flags |= O_RDONLY;
368 }
369
370 /* Use O_DSYNC for write-through caching, no flags for write-back caching,
371 * and O_DIRECT for no caching. */
372 if ((bdrv_flags & BDRV_O_NOCACHE)) {
373 *open_flags |= O_DIRECT;
374 }
375 }
376
377 static void raw_detach_aio_context(BlockDriverState *bs)
378 {
379 #ifdef CONFIG_LINUX_AIO
380 BDRVRawState *s = bs->opaque;
381
382 if (s->use_aio) {
383 laio_detach_aio_context(s->aio_ctx, bdrv_get_aio_context(bs));
384 }
385 #endif
386 }
387
388 static void raw_attach_aio_context(BlockDriverState *bs,
389 AioContext *new_context)
390 {
391 #ifdef CONFIG_LINUX_AIO
392 BDRVRawState *s = bs->opaque;
393
394 if (s->use_aio) {
395 laio_attach_aio_context(s->aio_ctx, new_context);
396 }
397 #endif
398 }
399
400 #ifdef CONFIG_LINUX_AIO
401 static int raw_set_aio(LinuxAioState **aio_ctx, int *use_aio, int bdrv_flags)
402 {
403 int ret = -1;
404 assert(aio_ctx != NULL);
405 assert(use_aio != NULL);
406 /*
407 * Currently Linux do AIO only for files opened with O_DIRECT
408 * specified so check NOCACHE flag too
409 */
410 if ((bdrv_flags & (BDRV_O_NOCACHE|BDRV_O_NATIVE_AIO)) ==
411 (BDRV_O_NOCACHE|BDRV_O_NATIVE_AIO)) {
412
413 /* if non-NULL, laio_init() has already been run */
414 if (*aio_ctx == NULL) {
415 *aio_ctx = laio_init();
416 if (!*aio_ctx) {
417 goto error;
418 }
419 }
420 *use_aio = 1;
421 } else {
422 *use_aio = 0;
423 }
424
425 ret = 0;
426
427 error:
428 return ret;
429 }
430 #endif
431
432 static void raw_parse_filename(const char *filename, QDict *options,
433 Error **errp)
434 {
435 /* The filename does not have to be prefixed by the protocol name, since
436 * "file" is the default protocol; therefore, the return value of this
437 * function call can be ignored. */
438 strstart(filename, "file:", &filename);
439
440 qdict_put_obj(options, "filename", QOBJECT(qstring_from_str(filename)));
441 }
442
443 static QemuOptsList raw_runtime_opts = {
444 .name = "raw",
445 .head = QTAILQ_HEAD_INITIALIZER(raw_runtime_opts.head),
446 .desc = {
447 {
448 .name = "filename",
449 .type = QEMU_OPT_STRING,
450 .help = "File name of the image",
451 },
452 { /* end of list */ }
453 },
454 };
455
456 static int raw_open_common(BlockDriverState *bs, QDict *options,
457 int bdrv_flags, int open_flags, Error **errp)
458 {
459 BDRVRawState *s = bs->opaque;
460 QemuOpts *opts;
461 Error *local_err = NULL;
462 const char *filename = NULL;
463 int fd, ret;
464 struct stat st;
465
466 opts = qemu_opts_create(&raw_runtime_opts, NULL, 0, &error_abort);
467 qemu_opts_absorb_qdict(opts, options, &local_err);
468 if (local_err) {
469 error_propagate(errp, local_err);
470 ret = -EINVAL;
471 goto fail;
472 }
473
474 filename = qemu_opt_get(opts, "filename");
475
476 ret = raw_normalize_devicepath(&filename);
477 if (ret != 0) {
478 error_setg_errno(errp, -ret, "Could not normalize device path");
479 goto fail;
480 }
481
482 s->open_flags = open_flags;
483 raw_parse_flags(bdrv_flags, &s->open_flags);
484
485 s->fd = -1;
486 fd = qemu_open(filename, s->open_flags, 0644);
487 if (fd < 0) {
488 ret = -errno;
489 if (ret == -EROFS) {
490 ret = -EACCES;
491 }
492 goto fail;
493 }
494 s->fd = fd;
495
496 #ifdef CONFIG_LINUX_AIO
497 if (raw_set_aio(&s->aio_ctx, &s->use_aio, bdrv_flags)) {
498 qemu_close(fd);
499 ret = -errno;
500 error_setg_errno(errp, -ret, "Could not set AIO state");
501 goto fail;
502 }
503 if (!s->use_aio && (bdrv_flags & BDRV_O_NATIVE_AIO)) {
504 error_setg(errp, "aio=native was specified, but it requires "
505 "cache.direct=on, which was not specified.");
506 ret = -EINVAL;
507 goto fail;
508 }
509 #else
510 if (bdrv_flags & BDRV_O_NATIVE_AIO) {
511 error_setg(errp, "aio=native was specified, but is not supported "
512 "in this build.");
513 ret = -EINVAL;
514 goto fail;
515 }
516 #endif /* !defined(CONFIG_LINUX_AIO) */
517
518 s->has_discard = true;
519 s->has_write_zeroes = true;
520 if ((bs->open_flags & BDRV_O_NOCACHE) != 0) {
521 s->needs_alignment = true;
522 }
523
524 if (fstat(s->fd, &st) < 0) {
525 ret = -errno;
526 error_setg_errno(errp, errno, "Could not stat file");
527 goto fail;
528 }
529 if (S_ISREG(st.st_mode)) {
530 s->discard_zeroes = true;
531 s->has_fallocate = true;
532 }
533 if (S_ISBLK(st.st_mode)) {
534 #ifdef BLKDISCARDZEROES
535 unsigned int arg;
536 if (ioctl(s->fd, BLKDISCARDZEROES, &arg) == 0 && arg) {
537 s->discard_zeroes = true;
538 }
539 #endif
540 #ifdef __linux__
541 /* On Linux 3.10, BLKDISCARD leaves stale data in the page cache. Do
542 * not rely on the contents of discarded blocks unless using O_DIRECT.
543 * Same for BLKZEROOUT.
544 */
545 if (!(bs->open_flags & BDRV_O_NOCACHE)) {
546 s->discard_zeroes = false;
547 s->has_write_zeroes = false;
548 }
549 #endif
550 }
551 #ifdef __FreeBSD__
552 if (S_ISCHR(st.st_mode)) {
553 /*
554 * The file is a char device (disk), which on FreeBSD isn't behind
555 * a pager, so force all requests to be aligned. This is needed
556 * so QEMU makes sure all IO operations on the device are aligned
557 * to sector size, or else FreeBSD will reject them with EINVAL.
558 */
559 s->needs_alignment = true;
560 }
561 #endif
562
563 #ifdef CONFIG_XFS
564 if (platform_test_xfs_fd(s->fd)) {
565 s->is_xfs = true;
566 }
567 #endif
568
569 raw_attach_aio_context(bs, bdrv_get_aio_context(bs));
570
571 ret = 0;
572 fail:
573 if (filename && (bdrv_flags & BDRV_O_TEMPORARY)) {
574 unlink(filename);
575 }
576 qemu_opts_del(opts);
577 return ret;
578 }
579
580 static int raw_open(BlockDriverState *bs, QDict *options, int flags,
581 Error **errp)
582 {
583 BDRVRawState *s = bs->opaque;
584 Error *local_err = NULL;
585 int ret;
586
587 s->type = FTYPE_FILE;
588 ret = raw_open_common(bs, options, flags, 0, &local_err);
589 if (local_err) {
590 error_propagate(errp, local_err);
591 }
592 return ret;
593 }
594
595 static int raw_reopen_prepare(BDRVReopenState *state,
596 BlockReopenQueue *queue, Error **errp)
597 {
598 BDRVRawState *s;
599 BDRVRawReopenState *raw_s;
600 int ret = 0;
601 Error *local_err = NULL;
602
603 assert(state != NULL);
604 assert(state->bs != NULL);
605
606 s = state->bs->opaque;
607
608 state->opaque = g_new0(BDRVRawReopenState, 1);
609 raw_s = state->opaque;
610
611 #ifdef CONFIG_LINUX_AIO
612 raw_s->use_aio = s->use_aio;
613
614 /* we can use s->aio_ctx instead of a copy, because the use_aio flag is
615 * valid in the 'false' condition even if aio_ctx is set, and raw_set_aio()
616 * won't override aio_ctx if aio_ctx is non-NULL */
617 if (raw_set_aio(&s->aio_ctx, &raw_s->use_aio, state->flags)) {
618 error_setg(errp, "Could not set AIO state");
619 return -1;
620 }
621 #endif
622
623 if (s->type == FTYPE_CD) {
624 raw_s->open_flags |= O_NONBLOCK;
625 }
626
627 raw_parse_flags(state->flags, &raw_s->open_flags);
628
629 raw_s->fd = -1;
630
631 int fcntl_flags = O_APPEND | O_NONBLOCK;
632 #ifdef O_NOATIME
633 fcntl_flags |= O_NOATIME;
634 #endif
635
636 #ifdef O_ASYNC
637 /* Not all operating systems have O_ASYNC, and those that don't
638 * will not let us track the state into raw_s->open_flags (typically
639 * you achieve the same effect with an ioctl, for example I_SETSIG
640 * on Solaris). But we do not use O_ASYNC, so that's fine.
641 */
642 assert((s->open_flags & O_ASYNC) == 0);
643 #endif
644
645 if ((raw_s->open_flags & ~fcntl_flags) == (s->open_flags & ~fcntl_flags)) {
646 /* dup the original fd */
647 /* TODO: use qemu fcntl wrapper */
648 #ifdef F_DUPFD_CLOEXEC
649 raw_s->fd = fcntl(s->fd, F_DUPFD_CLOEXEC, 0);
650 #else
651 raw_s->fd = dup(s->fd);
652 if (raw_s->fd != -1) {
653 qemu_set_cloexec(raw_s->fd);
654 }
655 #endif
656 if (raw_s->fd >= 0) {
657 ret = fcntl_setfl(raw_s->fd, raw_s->open_flags);
658 if (ret) {
659 qemu_close(raw_s->fd);
660 raw_s->fd = -1;
661 }
662 }
663 }
664
665 /* If we cannot use fcntl, or fcntl failed, fall back to qemu_open() */
666 if (raw_s->fd == -1) {
667 const char *normalized_filename = state->bs->filename;
668 ret = raw_normalize_devicepath(&normalized_filename);
669 if (ret < 0) {
670 error_setg_errno(errp, -ret, "Could not normalize device path");
671 } else {
672 assert(!(raw_s->open_flags & O_CREAT));
673 raw_s->fd = qemu_open(normalized_filename, raw_s->open_flags);
674 if (raw_s->fd == -1) {
675 error_setg_errno(errp, errno, "Could not reopen file");
676 ret = -1;
677 }
678 }
679 }
680
681 /* Fail already reopen_prepare() if we can't get a working O_DIRECT
682 * alignment with the new fd. */
683 if (raw_s->fd != -1) {
684 raw_probe_alignment(state->bs, raw_s->fd, &local_err);
685 if (local_err) {
686 qemu_close(raw_s->fd);
687 raw_s->fd = -1;
688 error_propagate(errp, local_err);
689 ret = -EINVAL;
690 }
691 }
692
693 return ret;
694 }
695
696 static void raw_reopen_commit(BDRVReopenState *state)
697 {
698 BDRVRawReopenState *raw_s = state->opaque;
699 BDRVRawState *s = state->bs->opaque;
700
701 s->open_flags = raw_s->open_flags;
702
703 qemu_close(s->fd);
704 s->fd = raw_s->fd;
705 #ifdef CONFIG_LINUX_AIO
706 s->use_aio = raw_s->use_aio;
707 #endif
708
709 g_free(state->opaque);
710 state->opaque = NULL;
711 }
712
713
714 static void raw_reopen_abort(BDRVReopenState *state)
715 {
716 BDRVRawReopenState *raw_s = state->opaque;
717
718 /* nothing to do if NULL, we didn't get far enough */
719 if (raw_s == NULL) {
720 return;
721 }
722
723 if (raw_s->fd >= 0) {
724 qemu_close(raw_s->fd);
725 raw_s->fd = -1;
726 }
727 g_free(state->opaque);
728 state->opaque = NULL;
729 }
730
731 static void raw_refresh_limits(BlockDriverState *bs, Error **errp)
732 {
733 BDRVRawState *s = bs->opaque;
734
735 raw_probe_alignment(bs, s->fd, errp);
736 bs->bl.min_mem_alignment = s->buf_align;
737 bs->bl.opt_mem_alignment = MAX(s->buf_align, getpagesize());
738 }
739
740 static int check_for_dasd(int fd)
741 {
742 #ifdef BIODASDINFO2
743 struct dasd_information2_t info = {0};
744
745 return ioctl(fd, BIODASDINFO2, &info);
746 #else
747 return -1;
748 #endif
749 }
750
751 /**
752 * Try to get @bs's logical and physical block size.
753 * On success, store them in @bsz and return zero.
754 * On failure, return negative errno.
755 */
756 static int hdev_probe_blocksizes(BlockDriverState *bs, BlockSizes *bsz)
757 {
758 BDRVRawState *s = bs->opaque;
759 int ret;
760
761 /* If DASD, get blocksizes */
762 if (check_for_dasd(s->fd) < 0) {
763 return -ENOTSUP;
764 }
765 ret = probe_logical_blocksize(s->fd, &bsz->log);
766 if (ret < 0) {
767 return ret;
768 }
769 return probe_physical_blocksize(s->fd, &bsz->phys);
770 }
771
772 /**
773 * Try to get @bs's geometry: cyls, heads, sectors.
774 * On success, store them in @geo and return 0.
775 * On failure return -errno.
776 * (Allows block driver to assign default geometry values that guest sees)
777 */
778 #ifdef __linux__
779 static int hdev_probe_geometry(BlockDriverState *bs, HDGeometry *geo)
780 {
781 BDRVRawState *s = bs->opaque;
782 struct hd_geometry ioctl_geo = {0};
783
784 /* If DASD, get its geometry */
785 if (check_for_dasd(s->fd) < 0) {
786 return -ENOTSUP;
787 }
788 if (ioctl(s->fd, HDIO_GETGEO, &ioctl_geo) < 0) {
789 return -errno;
790 }
791 /* HDIO_GETGEO may return success even though geo contains zeros
792 (e.g. certain multipath setups) */
793 if (!ioctl_geo.heads || !ioctl_geo.sectors || !ioctl_geo.cylinders) {
794 return -ENOTSUP;
795 }
796 /* Do not return a geometry for partition */
797 if (ioctl_geo.start != 0) {
798 return -ENOTSUP;
799 }
800 geo->heads = ioctl_geo.heads;
801 geo->sectors = ioctl_geo.sectors;
802 geo->cylinders = ioctl_geo.cylinders;
803
804 return 0;
805 }
806 #else /* __linux__ */
807 static int hdev_probe_geometry(BlockDriverState *bs, HDGeometry *geo)
808 {
809 return -ENOTSUP;
810 }
811 #endif
812
813 static ssize_t handle_aiocb_ioctl(RawPosixAIOData *aiocb)
814 {
815 int ret;
816
817 ret = ioctl(aiocb->aio_fildes, aiocb->aio_ioctl_cmd, aiocb->aio_ioctl_buf);
818 if (ret == -1) {
819 return -errno;
820 }
821
822 return 0;
823 }
824
825 static ssize_t handle_aiocb_flush(RawPosixAIOData *aiocb)
826 {
827 int ret;
828
829 ret = qemu_fdatasync(aiocb->aio_fildes);
830 if (ret == -1) {
831 return -errno;
832 }
833 return 0;
834 }
835
836 #ifdef CONFIG_PREADV
837
838 static bool preadv_present = true;
839
840 static ssize_t
841 qemu_preadv(int fd, const struct iovec *iov, int nr_iov, off_t offset)
842 {
843 return preadv(fd, iov, nr_iov, offset);
844 }
845
846 static ssize_t
847 qemu_pwritev(int fd, const struct iovec *iov, int nr_iov, off_t offset)
848 {
849 return pwritev(fd, iov, nr_iov, offset);
850 }
851
852 #else
853
854 static bool preadv_present = false;
855
856 static ssize_t
857 qemu_preadv(int fd, const struct iovec *iov, int nr_iov, off_t offset)
858 {
859 return -ENOSYS;
860 }
861
862 static ssize_t
863 qemu_pwritev(int fd, const struct iovec *iov, int nr_iov, off_t offset)
864 {
865 return -ENOSYS;
866 }
867
868 #endif
869
870 static ssize_t handle_aiocb_rw_vector(RawPosixAIOData *aiocb)
871 {
872 ssize_t len;
873
874 do {
875 if (aiocb->aio_type & QEMU_AIO_WRITE)
876 len = qemu_pwritev(aiocb->aio_fildes,
877 aiocb->aio_iov,
878 aiocb->aio_niov,
879 aiocb->aio_offset);
880 else
881 len = qemu_preadv(aiocb->aio_fildes,
882 aiocb->aio_iov,
883 aiocb->aio_niov,
884 aiocb->aio_offset);
885 } while (len == -1 && errno == EINTR);
886
887 if (len == -1) {
888 return -errno;
889 }
890 return len;
891 }
892
893 /*
894 * Read/writes the data to/from a given linear buffer.
895 *
896 * Returns the number of bytes handles or -errno in case of an error. Short
897 * reads are only returned if the end of the file is reached.
898 */
899 static ssize_t handle_aiocb_rw_linear(RawPosixAIOData *aiocb, char *buf)
900 {
901 ssize_t offset = 0;
902 ssize_t len;
903
904 while (offset < aiocb->aio_nbytes) {
905 if (aiocb->aio_type & QEMU_AIO_WRITE) {
906 len = pwrite(aiocb->aio_fildes,
907 (const char *)buf + offset,
908 aiocb->aio_nbytes - offset,
909 aiocb->aio_offset + offset);
910 } else {
911 len = pread(aiocb->aio_fildes,
912 buf + offset,
913 aiocb->aio_nbytes - offset,
914 aiocb->aio_offset + offset);
915 }
916 if (len == -1 && errno == EINTR) {
917 continue;
918 } else if (len == -1 && errno == EINVAL &&
919 (aiocb->bs->open_flags & BDRV_O_NOCACHE) &&
920 !(aiocb->aio_type & QEMU_AIO_WRITE) &&
921 offset > 0) {
922 /* O_DIRECT pread() may fail with EINVAL when offset is unaligned
923 * after a short read. Assume that O_DIRECT short reads only occur
924 * at EOF. Therefore this is a short read, not an I/O error.
925 */
926 break;
927 } else if (len == -1) {
928 offset = -errno;
929 break;
930 } else if (len == 0) {
931 break;
932 }
933 offset += len;
934 }
935
936 return offset;
937 }
938
939 static ssize_t handle_aiocb_rw(RawPosixAIOData *aiocb)
940 {
941 ssize_t nbytes;
942 char *buf;
943
944 if (!(aiocb->aio_type & QEMU_AIO_MISALIGNED)) {
945 /*
946 * If there is just a single buffer, and it is properly aligned
947 * we can just use plain pread/pwrite without any problems.
948 */
949 if (aiocb->aio_niov == 1) {
950 return handle_aiocb_rw_linear(aiocb, aiocb->aio_iov->iov_base);
951 }
952 /*
953 * We have more than one iovec, and all are properly aligned.
954 *
955 * Try preadv/pwritev first and fall back to linearizing the
956 * buffer if it's not supported.
957 */
958 if (preadv_present) {
959 nbytes = handle_aiocb_rw_vector(aiocb);
960 if (nbytes == aiocb->aio_nbytes ||
961 (nbytes < 0 && nbytes != -ENOSYS)) {
962 return nbytes;
963 }
964 preadv_present = false;
965 }
966
967 /*
968 * XXX(hch): short read/write. no easy way to handle the reminder
969 * using these interfaces. For now retry using plain
970 * pread/pwrite?
971 */
972 }
973
974 /*
975 * Ok, we have to do it the hard way, copy all segments into
976 * a single aligned buffer.
977 */
978 buf = qemu_try_blockalign(aiocb->bs, aiocb->aio_nbytes);
979 if (buf == NULL) {
980 return -ENOMEM;
981 }
982
983 if (aiocb->aio_type & QEMU_AIO_WRITE) {
984 char *p = buf;
985 int i;
986
987 for (i = 0; i < aiocb->aio_niov; ++i) {
988 memcpy(p, aiocb->aio_iov[i].iov_base, aiocb->aio_iov[i].iov_len);
989 p += aiocb->aio_iov[i].iov_len;
990 }
991 assert(p - buf == aiocb->aio_nbytes);
992 }
993
994 nbytes = handle_aiocb_rw_linear(aiocb, buf);
995 if (!(aiocb->aio_type & QEMU_AIO_WRITE)) {
996 char *p = buf;
997 size_t count = aiocb->aio_nbytes, copy;
998 int i;
999
1000 for (i = 0; i < aiocb->aio_niov && count; ++i) {
1001 copy = count;
1002 if (copy > aiocb->aio_iov[i].iov_len) {
1003 copy = aiocb->aio_iov[i].iov_len;
1004 }
1005 memcpy(aiocb->aio_iov[i].iov_base, p, copy);
1006 assert(count >= copy);
1007 p += copy;
1008 count -= copy;
1009 }
1010 assert(count == 0);
1011 }
1012 qemu_vfree(buf);
1013
1014 return nbytes;
1015 }
1016
1017 #ifdef CONFIG_XFS
1018 static int xfs_write_zeroes(BDRVRawState *s, int64_t offset, uint64_t bytes)
1019 {
1020 struct xfs_flock64 fl;
1021 int err;
1022
1023 memset(&fl, 0, sizeof(fl));
1024 fl.l_whence = SEEK_SET;
1025 fl.l_start = offset;
1026 fl.l_len = bytes;
1027
1028 if (xfsctl(NULL, s->fd, XFS_IOC_ZERO_RANGE, &fl) < 0) {
1029 err = errno;
1030 DPRINTF("cannot write zero range (%s)\n", strerror(errno));
1031 return -err;
1032 }
1033
1034 return 0;
1035 }
1036
1037 static int xfs_discard(BDRVRawState *s, int64_t offset, uint64_t bytes)
1038 {
1039 struct xfs_flock64 fl;
1040 int err;
1041
1042 memset(&fl, 0, sizeof(fl));
1043 fl.l_whence = SEEK_SET;
1044 fl.l_start = offset;
1045 fl.l_len = bytes;
1046
1047 if (xfsctl(NULL, s->fd, XFS_IOC_UNRESVSP64, &fl) < 0) {
1048 err = errno;
1049 DPRINTF("cannot punch hole (%s)\n", strerror(errno));
1050 return -err;
1051 }
1052
1053 return 0;
1054 }
1055 #endif
1056
1057 static int translate_err(int err)
1058 {
1059 if (err == -ENODEV || err == -ENOSYS || err == -EOPNOTSUPP ||
1060 err == -ENOTTY) {
1061 err = -ENOTSUP;
1062 }
1063 return err;
1064 }
1065
1066 #ifdef CONFIG_FALLOCATE
1067 static int do_fallocate(int fd, int mode, off_t offset, off_t len)
1068 {
1069 do {
1070 if (fallocate(fd, mode, offset, len) == 0) {
1071 return 0;
1072 }
1073 } while (errno == EINTR);
1074 return translate_err(-errno);
1075 }
1076 #endif
1077
1078 static ssize_t handle_aiocb_write_zeroes_block(RawPosixAIOData *aiocb)
1079 {
1080 int ret = -ENOTSUP;
1081 BDRVRawState *s = aiocb->bs->opaque;
1082
1083 if (!s->has_write_zeroes) {
1084 return -ENOTSUP;
1085 }
1086
1087 #ifdef BLKZEROOUT
1088 do {
1089 uint64_t range[2] = { aiocb->aio_offset, aiocb->aio_nbytes };
1090 if (ioctl(aiocb->aio_fildes, BLKZEROOUT, range) == 0) {
1091 return 0;
1092 }
1093 } while (errno == EINTR);
1094
1095 ret = translate_err(-errno);
1096 #endif
1097
1098 if (ret == -ENOTSUP) {
1099 s->has_write_zeroes = false;
1100 }
1101 return ret;
1102 }
1103
1104 static ssize_t handle_aiocb_write_zeroes(RawPosixAIOData *aiocb)
1105 {
1106 #if defined(CONFIG_FALLOCATE) || defined(CONFIG_XFS)
1107 BDRVRawState *s = aiocb->bs->opaque;
1108 #endif
1109
1110 if (aiocb->aio_type & QEMU_AIO_BLKDEV) {
1111 return handle_aiocb_write_zeroes_block(aiocb);
1112 }
1113
1114 #ifdef CONFIG_XFS
1115 if (s->is_xfs) {
1116 return xfs_write_zeroes(s, aiocb->aio_offset, aiocb->aio_nbytes);
1117 }
1118 #endif
1119
1120 #ifdef CONFIG_FALLOCATE_ZERO_RANGE
1121 if (s->has_write_zeroes) {
1122 int ret = do_fallocate(s->fd, FALLOC_FL_ZERO_RANGE,
1123 aiocb->aio_offset, aiocb->aio_nbytes);
1124 if (ret == 0 || ret != -ENOTSUP) {
1125 return ret;
1126 }
1127 s->has_write_zeroes = false;
1128 }
1129 #endif
1130
1131 #ifdef CONFIG_FALLOCATE_PUNCH_HOLE
1132 if (s->has_discard && s->has_fallocate) {
1133 int ret = do_fallocate(s->fd,
1134 FALLOC_FL_PUNCH_HOLE | FALLOC_FL_KEEP_SIZE,
1135 aiocb->aio_offset, aiocb->aio_nbytes);
1136 if (ret == 0) {
1137 ret = do_fallocate(s->fd, 0, aiocb->aio_offset, aiocb->aio_nbytes);
1138 if (ret == 0 || ret != -ENOTSUP) {
1139 return ret;
1140 }
1141 s->has_fallocate = false;
1142 } else if (ret != -ENOTSUP) {
1143 return ret;
1144 } else {
1145 s->has_discard = false;
1146 }
1147 }
1148 #endif
1149
1150 #ifdef CONFIG_FALLOCATE
1151 if (s->has_fallocate && aiocb->aio_offset >= bdrv_getlength(aiocb->bs)) {
1152 int ret = do_fallocate(s->fd, 0, aiocb->aio_offset, aiocb->aio_nbytes);
1153 if (ret == 0 || ret != -ENOTSUP) {
1154 return ret;
1155 }
1156 s->has_fallocate = false;
1157 }
1158 #endif
1159
1160 return -ENOTSUP;
1161 }
1162
1163 static ssize_t handle_aiocb_discard(RawPosixAIOData *aiocb)
1164 {
1165 int ret = -EOPNOTSUPP;
1166 BDRVRawState *s = aiocb->bs->opaque;
1167
1168 if (!s->has_discard) {
1169 return -ENOTSUP;
1170 }
1171
1172 if (aiocb->aio_type & QEMU_AIO_BLKDEV) {
1173 #ifdef BLKDISCARD
1174 do {
1175 uint64_t range[2] = { aiocb->aio_offset, aiocb->aio_nbytes };
1176 if (ioctl(aiocb->aio_fildes, BLKDISCARD, range) == 0) {
1177 return 0;
1178 }
1179 } while (errno == EINTR);
1180
1181 ret = -errno;
1182 #endif
1183 } else {
1184 #ifdef CONFIG_XFS
1185 if (s->is_xfs) {
1186 return xfs_discard(s, aiocb->aio_offset, aiocb->aio_nbytes);
1187 }
1188 #endif
1189
1190 #ifdef CONFIG_FALLOCATE_PUNCH_HOLE
1191 ret = do_fallocate(s->fd, FALLOC_FL_PUNCH_HOLE | FALLOC_FL_KEEP_SIZE,
1192 aiocb->aio_offset, aiocb->aio_nbytes);
1193 #endif
1194 }
1195
1196 ret = translate_err(ret);
1197 if (ret == -ENOTSUP) {
1198 s->has_discard = false;
1199 }
1200 return ret;
1201 }
1202
1203 static int aio_worker(void *arg)
1204 {
1205 RawPosixAIOData *aiocb = arg;
1206 ssize_t ret = 0;
1207
1208 switch (aiocb->aio_type & QEMU_AIO_TYPE_MASK) {
1209 case QEMU_AIO_READ:
1210 ret = handle_aiocb_rw(aiocb);
1211 if (ret >= 0 && ret < aiocb->aio_nbytes) {
1212 iov_memset(aiocb->aio_iov, aiocb->aio_niov, ret,
1213 0, aiocb->aio_nbytes - ret);
1214
1215 ret = aiocb->aio_nbytes;
1216 }
1217 if (ret == aiocb->aio_nbytes) {
1218 ret = 0;
1219 } else if (ret >= 0 && ret < aiocb->aio_nbytes) {
1220 ret = -EINVAL;
1221 }
1222 break;
1223 case QEMU_AIO_WRITE:
1224 ret = handle_aiocb_rw(aiocb);
1225 if (ret == aiocb->aio_nbytes) {
1226 ret = 0;
1227 } else if (ret >= 0 && ret < aiocb->aio_nbytes) {
1228 ret = -EINVAL;
1229 }
1230 break;
1231 case QEMU_AIO_FLUSH:
1232 ret = handle_aiocb_flush(aiocb);
1233 break;
1234 case QEMU_AIO_IOCTL:
1235 ret = handle_aiocb_ioctl(aiocb);
1236 break;
1237 case QEMU_AIO_DISCARD:
1238 ret = handle_aiocb_discard(aiocb);
1239 break;
1240 case QEMU_AIO_WRITE_ZEROES:
1241 ret = handle_aiocb_write_zeroes(aiocb);
1242 break;
1243 default:
1244 fprintf(stderr, "invalid aio request (0x%x)\n", aiocb->aio_type);
1245 ret = -EINVAL;
1246 break;
1247 }
1248
1249 g_free(aiocb);
1250 return ret;
1251 }
1252
1253 static int paio_submit_co(BlockDriverState *bs, int fd,
1254 int64_t sector_num, QEMUIOVector *qiov, int nb_sectors,
1255 int type)
1256 {
1257 RawPosixAIOData *acb = g_new(RawPosixAIOData, 1);
1258 ThreadPool *pool;
1259
1260 acb->bs = bs;
1261 acb->aio_type = type;
1262 acb->aio_fildes = fd;
1263
1264 acb->aio_nbytes = nb_sectors * BDRV_SECTOR_SIZE;
1265 acb->aio_offset = sector_num * BDRV_SECTOR_SIZE;
1266
1267 if (qiov) {
1268 acb->aio_iov = qiov->iov;
1269 acb->aio_niov = qiov->niov;
1270 assert(qiov->size == acb->aio_nbytes);
1271 }
1272
1273 trace_paio_submit_co(sector_num, nb_sectors, type);
1274 pool = aio_get_thread_pool(bdrv_get_aio_context(bs));
1275 return thread_pool_submit_co(pool, aio_worker, acb);
1276 }
1277
1278 static BlockAIOCB *paio_submit(BlockDriverState *bs, int fd,
1279 int64_t sector_num, QEMUIOVector *qiov, int nb_sectors,
1280 BlockCompletionFunc *cb, void *opaque, int type)
1281 {
1282 RawPosixAIOData *acb = g_new(RawPosixAIOData, 1);
1283 ThreadPool *pool;
1284
1285 acb->bs = bs;
1286 acb->aio_type = type;
1287 acb->aio_fildes = fd;
1288
1289 acb->aio_nbytes = nb_sectors * BDRV_SECTOR_SIZE;
1290 acb->aio_offset = sector_num * BDRV_SECTOR_SIZE;
1291
1292 if (qiov) {
1293 acb->aio_iov = qiov->iov;
1294 acb->aio_niov = qiov->niov;
1295 assert(qiov->size == acb->aio_nbytes);
1296 }
1297
1298 trace_paio_submit(acb, opaque, sector_num, nb_sectors, type);
1299 pool = aio_get_thread_pool(bdrv_get_aio_context(bs));
1300 return thread_pool_submit_aio(pool, aio_worker, acb, cb, opaque);
1301 }
1302
1303 static BlockAIOCB *raw_aio_submit(BlockDriverState *bs,
1304 int64_t sector_num, QEMUIOVector *qiov, int nb_sectors,
1305 BlockCompletionFunc *cb, void *opaque, int type)
1306 {
1307 BDRVRawState *s = bs->opaque;
1308
1309 if (fd_open(bs) < 0)
1310 return NULL;
1311
1312 /*
1313 * Check if the underlying device requires requests to be aligned,
1314 * and if the request we are trying to submit is aligned or not.
1315 * If this is the case tell the low-level driver that it needs
1316 * to copy the buffer.
1317 */
1318 if (s->needs_alignment) {
1319 if (!bdrv_qiov_is_aligned(bs, qiov)) {
1320 type |= QEMU_AIO_MISALIGNED;
1321 #ifdef CONFIG_LINUX_AIO
1322 } else if (s->use_aio) {
1323 return laio_submit(bs, s->aio_ctx, s->fd, sector_num, qiov,
1324 nb_sectors, cb, opaque, type);
1325 #endif
1326 }
1327 }
1328
1329 return paio_submit(bs, s->fd, sector_num, qiov, nb_sectors,
1330 cb, opaque, type);
1331 }
1332
1333 static void raw_aio_plug(BlockDriverState *bs)
1334 {
1335 #ifdef CONFIG_LINUX_AIO
1336 BDRVRawState *s = bs->opaque;
1337 if (s->use_aio) {
1338 laio_io_plug(bs, s->aio_ctx);
1339 }
1340 #endif
1341 }
1342
1343 static void raw_aio_unplug(BlockDriverState *bs)
1344 {
1345 #ifdef CONFIG_LINUX_AIO
1346 BDRVRawState *s = bs->opaque;
1347 if (s->use_aio) {
1348 laio_io_unplug(bs, s->aio_ctx);
1349 }
1350 #endif
1351 }
1352
1353 static BlockAIOCB *raw_aio_readv(BlockDriverState *bs,
1354 int64_t sector_num, QEMUIOVector *qiov, int nb_sectors,
1355 BlockCompletionFunc *cb, void *opaque)
1356 {
1357 return raw_aio_submit(bs, sector_num, qiov, nb_sectors,
1358 cb, opaque, QEMU_AIO_READ);
1359 }
1360
1361 static BlockAIOCB *raw_aio_writev(BlockDriverState *bs,
1362 int64_t sector_num, QEMUIOVector *qiov, int nb_sectors,
1363 BlockCompletionFunc *cb, void *opaque)
1364 {
1365 return raw_aio_submit(bs, sector_num, qiov, nb_sectors,
1366 cb, opaque, QEMU_AIO_WRITE);
1367 }
1368
1369 static BlockAIOCB *raw_aio_flush(BlockDriverState *bs,
1370 BlockCompletionFunc *cb, void *opaque)
1371 {
1372 BDRVRawState *s = bs->opaque;
1373
1374 if (fd_open(bs) < 0)
1375 return NULL;
1376
1377 return paio_submit(bs, s->fd, 0, NULL, 0, cb, opaque, QEMU_AIO_FLUSH);
1378 }
1379
1380 static void raw_close(BlockDriverState *bs)
1381 {
1382 BDRVRawState *s = bs->opaque;
1383
1384 raw_detach_aio_context(bs);
1385
1386 #ifdef CONFIG_LINUX_AIO
1387 if (s->use_aio) {
1388 laio_cleanup(s->aio_ctx);
1389 }
1390 #endif
1391 if (s->fd >= 0) {
1392 qemu_close(s->fd);
1393 s->fd = -1;
1394 }
1395 }
1396
1397 static int raw_truncate(BlockDriverState *bs, int64_t offset)
1398 {
1399 BDRVRawState *s = bs->opaque;
1400 struct stat st;
1401
1402 if (fstat(s->fd, &st)) {
1403 return -errno;
1404 }
1405
1406 if (S_ISREG(st.st_mode)) {
1407 if (ftruncate(s->fd, offset) < 0) {
1408 return -errno;
1409 }
1410 } else if (S_ISCHR(st.st_mode) || S_ISBLK(st.st_mode)) {
1411 if (offset > raw_getlength(bs)) {
1412 return -EINVAL;
1413 }
1414 } else {
1415 return -ENOTSUP;
1416 }
1417
1418 return 0;
1419 }
1420
1421 #ifdef __OpenBSD__
1422 static int64_t raw_getlength(BlockDriverState *bs)
1423 {
1424 BDRVRawState *s = bs->opaque;
1425 int fd = s->fd;
1426 struct stat st;
1427
1428 if (fstat(fd, &st))
1429 return -errno;
1430 if (S_ISCHR(st.st_mode) || S_ISBLK(st.st_mode)) {
1431 struct disklabel dl;
1432
1433 if (ioctl(fd, DIOCGDINFO, &dl))
1434 return -errno;
1435 return (uint64_t)dl.d_secsize *
1436 dl.d_partitions[DISKPART(st.st_rdev)].p_size;
1437 } else
1438 return st.st_size;
1439 }
1440 #elif defined(__NetBSD__)
1441 static int64_t raw_getlength(BlockDriverState *bs)
1442 {
1443 BDRVRawState *s = bs->opaque;
1444 int fd = s->fd;
1445 struct stat st;
1446
1447 if (fstat(fd, &st))
1448 return -errno;
1449 if (S_ISCHR(st.st_mode) || S_ISBLK(st.st_mode)) {
1450 struct dkwedge_info dkw;
1451
1452 if (ioctl(fd, DIOCGWEDGEINFO, &dkw) != -1) {
1453 return dkw.dkw_size * 512;
1454 } else {
1455 struct disklabel dl;
1456
1457 if (ioctl(fd, DIOCGDINFO, &dl))
1458 return -errno;
1459 return (uint64_t)dl.d_secsize *
1460 dl.d_partitions[DISKPART(st.st_rdev)].p_size;
1461 }
1462 } else
1463 return st.st_size;
1464 }
1465 #elif defined(__sun__)
1466 static int64_t raw_getlength(BlockDriverState *bs)
1467 {
1468 BDRVRawState *s = bs->opaque;
1469 struct dk_minfo minfo;
1470 int ret;
1471 int64_t size;
1472
1473 ret = fd_open(bs);
1474 if (ret < 0) {
1475 return ret;
1476 }
1477
1478 /*
1479 * Use the DKIOCGMEDIAINFO ioctl to read the size.
1480 */
1481 ret = ioctl(s->fd, DKIOCGMEDIAINFO, &minfo);
1482 if (ret != -1) {
1483 return minfo.dki_lbsize * minfo.dki_capacity;
1484 }
1485
1486 /*
1487 * There are reports that lseek on some devices fails, but
1488 * irc discussion said that contingency on contingency was overkill.
1489 */
1490 size = lseek(s->fd, 0, SEEK_END);
1491 if (size < 0) {
1492 return -errno;
1493 }
1494 return size;
1495 }
1496 #elif defined(CONFIG_BSD)
1497 static int64_t raw_getlength(BlockDriverState *bs)
1498 {
1499 BDRVRawState *s = bs->opaque;
1500 int fd = s->fd;
1501 int64_t size;
1502 struct stat sb;
1503 #if defined (__FreeBSD__) || defined(__FreeBSD_kernel__)
1504 int reopened = 0;
1505 #endif
1506 int ret;
1507
1508 ret = fd_open(bs);
1509 if (ret < 0)
1510 return ret;
1511
1512 #if defined (__FreeBSD__) || defined(__FreeBSD_kernel__)
1513 again:
1514 #endif
1515 if (!fstat(fd, &sb) && (S_IFCHR & sb.st_mode)) {
1516 #ifdef DIOCGMEDIASIZE
1517 if (ioctl(fd, DIOCGMEDIASIZE, (off_t *)&size))
1518 #elif defined(DIOCGPART)
1519 {
1520 struct partinfo pi;
1521 if (ioctl(fd, DIOCGPART, &pi) == 0)
1522 size = pi.media_size;
1523 else
1524 size = 0;
1525 }
1526 if (size == 0)
1527 #endif
1528 #if defined(__APPLE__) && defined(__MACH__)
1529 {
1530 uint64_t sectors = 0;
1531 uint32_t sector_size = 0;
1532
1533 if (ioctl(fd, DKIOCGETBLOCKCOUNT, &sectors) == 0
1534 && ioctl(fd, DKIOCGETBLOCKSIZE, &sector_size) == 0) {
1535 size = sectors * sector_size;
1536 } else {
1537 size = lseek(fd, 0LL, SEEK_END);
1538 if (size < 0) {
1539 return -errno;
1540 }
1541 }
1542 }
1543 #else
1544 size = lseek(fd, 0LL, SEEK_END);
1545 if (size < 0) {
1546 return -errno;
1547 }
1548 #endif
1549 #if defined(__FreeBSD__) || defined(__FreeBSD_kernel__)
1550 switch(s->type) {
1551 case FTYPE_CD:
1552 /* XXX FreeBSD acd returns UINT_MAX sectors for an empty drive */
1553 if (size == 2048LL * (unsigned)-1)
1554 size = 0;
1555 /* XXX no disc? maybe we need to reopen... */
1556 if (size <= 0 && !reopened && cdrom_reopen(bs) >= 0) {
1557 reopened = 1;
1558 goto again;
1559 }
1560 }
1561 #endif
1562 } else {
1563 size = lseek(fd, 0, SEEK_END);
1564 if (size < 0) {
1565 return -errno;
1566 }
1567 }
1568 return size;
1569 }
1570 #else
1571 static int64_t raw_getlength(BlockDriverState *bs)
1572 {
1573 BDRVRawState *s = bs->opaque;
1574 int ret;
1575 int64_t size;
1576
1577 ret = fd_open(bs);
1578 if (ret < 0) {
1579 return ret;
1580 }
1581
1582 size = lseek(s->fd, 0, SEEK_END);
1583 if (size < 0) {
1584 return -errno;
1585 }
1586 return size;
1587 }
1588 #endif
1589
1590 static int64_t raw_get_allocated_file_size(BlockDriverState *bs)
1591 {
1592 struct stat st;
1593 BDRVRawState *s = bs->opaque;
1594
1595 if (fstat(s->fd, &st) < 0) {
1596 return -errno;
1597 }
1598 return (int64_t)st.st_blocks * 512;
1599 }
1600
1601 static int raw_create(const char *filename, QemuOpts *opts, Error **errp)
1602 {
1603 int fd;
1604 int result = 0;
1605 int64_t total_size = 0;
1606 bool nocow = false;
1607 PreallocMode prealloc;
1608 char *buf = NULL;
1609 Error *local_err = NULL;
1610
1611 strstart(filename, "file:", &filename);
1612
1613 /* Read out options */
1614 total_size = ROUND_UP(qemu_opt_get_size_del(opts, BLOCK_OPT_SIZE, 0),
1615 BDRV_SECTOR_SIZE);
1616 nocow = qemu_opt_get_bool(opts, BLOCK_OPT_NOCOW, false);
1617 buf = qemu_opt_get_del(opts, BLOCK_OPT_PREALLOC);
1618 prealloc = qapi_enum_parse(PreallocMode_lookup, buf,
1619 PREALLOC_MODE__MAX, PREALLOC_MODE_OFF,
1620 &local_err);
1621 g_free(buf);
1622 if (local_err) {
1623 error_propagate(errp, local_err);
1624 result = -EINVAL;
1625 goto out;
1626 }
1627
1628 fd = qemu_open(filename, O_RDWR | O_CREAT | O_TRUNC | O_BINARY,
1629 0644);
1630 if (fd < 0) {
1631 result = -errno;
1632 error_setg_errno(errp, -result, "Could not create file");
1633 goto out;
1634 }
1635
1636 if (nocow) {
1637 #ifdef __linux__
1638 /* Set NOCOW flag to solve performance issue on fs like btrfs.
1639 * This is an optimisation. The FS_IOC_SETFLAGS ioctl return value
1640 * will be ignored since any failure of this operation should not
1641 * block the left work.
1642 */
1643 int attr;
1644 if (ioctl(fd, FS_IOC_GETFLAGS, &attr) == 0) {
1645 attr |= FS_NOCOW_FL;
1646 ioctl(fd, FS_IOC_SETFLAGS, &attr);
1647 }
1648 #endif
1649 }
1650
1651 if (ftruncate(fd, total_size) != 0) {
1652 result = -errno;
1653 error_setg_errno(errp, -result, "Could not resize file");
1654 goto out_close;
1655 }
1656
1657 switch (prealloc) {
1658 #ifdef CONFIG_POSIX_FALLOCATE
1659 case PREALLOC_MODE_FALLOC:
1660 /* posix_fallocate() doesn't set errno. */
1661 result = -posix_fallocate(fd, 0, total_size);
1662 if (result != 0) {
1663 error_setg_errno(errp, -result,
1664 "Could not preallocate data for the new file");
1665 }
1666 break;
1667 #endif
1668 case PREALLOC_MODE_FULL:
1669 {
1670 int64_t num = 0, left = total_size;
1671 buf = g_malloc0(65536);
1672
1673 while (left > 0) {
1674 num = MIN(left, 65536);
1675 result = write(fd, buf, num);
1676 if (result < 0) {
1677 result = -errno;
1678 error_setg_errno(errp, -result,
1679 "Could not write to the new file");
1680 break;
1681 }
1682 left -= result;
1683 }
1684 if (result >= 0) {
1685 result = fsync(fd);
1686 if (result < 0) {
1687 result = -errno;
1688 error_setg_errno(errp, -result,
1689 "Could not flush new file to disk");
1690 }
1691 }
1692 g_free(buf);
1693 break;
1694 }
1695 case PREALLOC_MODE_OFF:
1696 break;
1697 default:
1698 result = -EINVAL;
1699 error_setg(errp, "Unsupported preallocation mode: %s",
1700 PreallocMode_lookup[prealloc]);
1701 break;
1702 }
1703
1704 out_close:
1705 if (qemu_close(fd) != 0 && result == 0) {
1706 result = -errno;
1707 error_setg_errno(errp, -result, "Could not close the new file");
1708 }
1709 out:
1710 return result;
1711 }
1712
1713 /*
1714 * Find allocation range in @bs around offset @start.
1715 * May change underlying file descriptor's file offset.
1716 * If @start is not in a hole, store @start in @data, and the
1717 * beginning of the next hole in @hole, and return 0.
1718 * If @start is in a non-trailing hole, store @start in @hole and the
1719 * beginning of the next non-hole in @data, and return 0.
1720 * If @start is in a trailing hole or beyond EOF, return -ENXIO.
1721 * If we can't find out, return a negative errno other than -ENXIO.
1722 */
1723 static int find_allocation(BlockDriverState *bs, off_t start,
1724 off_t *data, off_t *hole)
1725 {
1726 #if defined SEEK_HOLE && defined SEEK_DATA
1727 BDRVRawState *s = bs->opaque;
1728 off_t offs;
1729
1730 /*
1731 * SEEK_DATA cases:
1732 * D1. offs == start: start is in data
1733 * D2. offs > start: start is in a hole, next data at offs
1734 * D3. offs < 0, errno = ENXIO: either start is in a trailing hole
1735 * or start is beyond EOF
1736 * If the latter happens, the file has been truncated behind
1737 * our back since we opened it. All bets are off then.
1738 * Treating like a trailing hole is simplest.
1739 * D4. offs < 0, errno != ENXIO: we learned nothing
1740 */
1741 offs = lseek(s->fd, start, SEEK_DATA);
1742 if (offs < 0) {
1743 return -errno; /* D3 or D4 */
1744 }
1745 assert(offs >= start);
1746
1747 if (offs > start) {
1748 /* D2: in hole, next data at offs */
1749 *hole = start;
1750 *data = offs;
1751 return 0;
1752 }
1753
1754 /* D1: in data, end not yet known */
1755
1756 /*
1757 * SEEK_HOLE cases:
1758 * H1. offs == start: start is in a hole
1759 * If this happens here, a hole has been dug behind our back
1760 * since the previous lseek().
1761 * H2. offs > start: either start is in data, next hole at offs,
1762 * or start is in trailing hole, EOF at offs
1763 * Linux treats trailing holes like any other hole: offs ==
1764 * start. Solaris seeks to EOF instead: offs > start (blech).
1765 * If that happens here, a hole has been dug behind our back
1766 * since the previous lseek().
1767 * H3. offs < 0, errno = ENXIO: start is beyond EOF
1768 * If this happens, the file has been truncated behind our
1769 * back since we opened it. Treat it like a trailing hole.
1770 * H4. offs < 0, errno != ENXIO: we learned nothing
1771 * Pretend we know nothing at all, i.e. "forget" about D1.
1772 */
1773 offs = lseek(s->fd, start, SEEK_HOLE);
1774 if (offs < 0) {
1775 return -errno; /* D1 and (H3 or H4) */
1776 }
1777 assert(offs >= start);
1778
1779 if (offs > start) {
1780 /*
1781 * D1 and H2: either in data, next hole at offs, or it was in
1782 * data but is now in a trailing hole. In the latter case,
1783 * all bets are off. Treating it as if it there was data all
1784 * the way to EOF is safe, so simply do that.
1785 */
1786 *data = start;
1787 *hole = offs;
1788 return 0;
1789 }
1790
1791 /* D1 and H1 */
1792 return -EBUSY;
1793 #else
1794 return -ENOTSUP;
1795 #endif
1796 }
1797
1798 /*
1799 * Returns the allocation status of the specified sectors.
1800 *
1801 * If 'sector_num' is beyond the end of the disk image the return value is 0
1802 * and 'pnum' is set to 0.
1803 *
1804 * 'pnum' is set to the number of sectors (including and immediately following
1805 * the specified sector) that are known to be in the same
1806 * allocated/unallocated state.
1807 *
1808 * 'nb_sectors' is the max value 'pnum' should be set to. If nb_sectors goes
1809 * beyond the end of the disk image it will be clamped.
1810 */
1811 static int64_t coroutine_fn raw_co_get_block_status(BlockDriverState *bs,
1812 int64_t sector_num,
1813 int nb_sectors, int *pnum,
1814 BlockDriverState **file)
1815 {
1816 off_t start, data = 0, hole = 0;
1817 int64_t total_size;
1818 int ret;
1819
1820 ret = fd_open(bs);
1821 if (ret < 0) {
1822 return ret;
1823 }
1824
1825 start = sector_num * BDRV_SECTOR_SIZE;
1826 total_size = bdrv_getlength(bs);
1827 if (total_size < 0) {
1828 return total_size;
1829 } else if (start >= total_size) {
1830 *pnum = 0;
1831 return 0;
1832 } else if (start + nb_sectors * BDRV_SECTOR_SIZE > total_size) {
1833 nb_sectors = DIV_ROUND_UP(total_size - start, BDRV_SECTOR_SIZE);
1834 }
1835
1836 ret = find_allocation(bs, start, &data, &hole);
1837 if (ret == -ENXIO) {
1838 /* Trailing hole */
1839 *pnum = nb_sectors;
1840 ret = BDRV_BLOCK_ZERO;
1841 } else if (ret < 0) {
1842 /* No info available, so pretend there are no holes */
1843 *pnum = nb_sectors;
1844 ret = BDRV_BLOCK_DATA;
1845 } else if (data == start) {
1846 /* On a data extent, compute sectors to the end of the extent,
1847 * possibly including a partial sector at EOF. */
1848 *pnum = MIN(nb_sectors, DIV_ROUND_UP(hole - start, BDRV_SECTOR_SIZE));
1849 ret = BDRV_BLOCK_DATA;
1850 } else {
1851 /* On a hole, compute sectors to the beginning of the next extent. */
1852 assert(hole == start);
1853 *pnum = MIN(nb_sectors, (data - start) / BDRV_SECTOR_SIZE);
1854 ret = BDRV_BLOCK_ZERO;
1855 }
1856 *file = bs;
1857 return ret | BDRV_BLOCK_OFFSET_VALID | start;
1858 }
1859
1860 static coroutine_fn BlockAIOCB *raw_aio_discard(BlockDriverState *bs,
1861 int64_t sector_num, int nb_sectors,
1862 BlockCompletionFunc *cb, void *opaque)
1863 {
1864 BDRVRawState *s = bs->opaque;
1865
1866 return paio_submit(bs, s->fd, sector_num, NULL, nb_sectors,
1867 cb, opaque, QEMU_AIO_DISCARD);
1868 }
1869
1870 static int coroutine_fn raw_co_write_zeroes(
1871 BlockDriverState *bs, int64_t sector_num,
1872 int nb_sectors, BdrvRequestFlags flags)
1873 {
1874 BDRVRawState *s = bs->opaque;
1875
1876 if (!(flags & BDRV_REQ_MAY_UNMAP)) {
1877 return paio_submit_co(bs, s->fd, sector_num, NULL, nb_sectors,
1878 QEMU_AIO_WRITE_ZEROES);
1879 } else if (s->discard_zeroes) {
1880 return paio_submit_co(bs, s->fd, sector_num, NULL, nb_sectors,
1881 QEMU_AIO_DISCARD);
1882 }
1883 return -ENOTSUP;
1884 }
1885
1886 static int raw_get_info(BlockDriverState *bs, BlockDriverInfo *bdi)
1887 {
1888 BDRVRawState *s = bs->opaque;
1889
1890 bdi->unallocated_blocks_are_zero = s->discard_zeroes;
1891 bdi->can_write_zeroes_with_unmap = s->discard_zeroes;
1892 return 0;
1893 }
1894
1895 static QemuOptsList raw_create_opts = {
1896 .name = "raw-create-opts",
1897 .head = QTAILQ_HEAD_INITIALIZER(raw_create_opts.head),
1898 .desc = {
1899 {
1900 .name = BLOCK_OPT_SIZE,
1901 .type = QEMU_OPT_SIZE,
1902 .help = "Virtual disk size"
1903 },
1904 {
1905 .name = BLOCK_OPT_NOCOW,
1906 .type = QEMU_OPT_BOOL,
1907 .help = "Turn off copy-on-write (valid only on btrfs)"
1908 },
1909 {
1910 .name = BLOCK_OPT_PREALLOC,
1911 .type = QEMU_OPT_STRING,
1912 .help = "Preallocation mode (allowed values: off, falloc, full)"
1913 },
1914 { /* end of list */ }
1915 }
1916 };
1917
1918 BlockDriver bdrv_file = {
1919 .format_name = "file",
1920 .protocol_name = "file",
1921 .instance_size = sizeof(BDRVRawState),
1922 .bdrv_needs_filename = true,
1923 .bdrv_probe = NULL, /* no probe for protocols */
1924 .bdrv_parse_filename = raw_parse_filename,
1925 .bdrv_file_open = raw_open,
1926 .bdrv_reopen_prepare = raw_reopen_prepare,
1927 .bdrv_reopen_commit = raw_reopen_commit,
1928 .bdrv_reopen_abort = raw_reopen_abort,
1929 .bdrv_close = raw_close,
1930 .bdrv_create = raw_create,
1931 .bdrv_has_zero_init = bdrv_has_zero_init_1,
1932 .bdrv_co_get_block_status = raw_co_get_block_status,
1933 .bdrv_co_write_zeroes = raw_co_write_zeroes,
1934
1935 .bdrv_aio_readv = raw_aio_readv,
1936 .bdrv_aio_writev = raw_aio_writev,
1937 .bdrv_aio_flush = raw_aio_flush,
1938 .bdrv_aio_discard = raw_aio_discard,
1939 .bdrv_refresh_limits = raw_refresh_limits,
1940 .bdrv_io_plug = raw_aio_plug,
1941 .bdrv_io_unplug = raw_aio_unplug,
1942
1943 .bdrv_truncate = raw_truncate,
1944 .bdrv_getlength = raw_getlength,
1945 .bdrv_get_info = raw_get_info,
1946 .bdrv_get_allocated_file_size
1947 = raw_get_allocated_file_size,
1948
1949 .bdrv_detach_aio_context = raw_detach_aio_context,
1950 .bdrv_attach_aio_context = raw_attach_aio_context,
1951
1952 .create_opts = &raw_create_opts,
1953 };
1954
1955 /***********************************************/
1956 /* host device */
1957
1958 #if defined(__APPLE__) && defined(__MACH__)
1959 static kern_return_t GetBSDPath(io_iterator_t mediaIterator, char *bsdPath,
1960 CFIndex maxPathSize, int flags);
1961 static char *FindEjectableOpticalMedia(io_iterator_t *mediaIterator)
1962 {
1963 kern_return_t kernResult = KERN_FAILURE;
1964 mach_port_t masterPort;
1965 CFMutableDictionaryRef classesToMatch;
1966 const char *matching_array[] = {kIODVDMediaClass, kIOCDMediaClass};
1967 char *mediaType = NULL;
1968
1969 kernResult = IOMasterPort( MACH_PORT_NULL, &masterPort );
1970 if ( KERN_SUCCESS != kernResult ) {
1971 printf( "IOMasterPort returned %d\n", kernResult );
1972 }
1973
1974 int index;
1975 for (index = 0; index < ARRAY_SIZE(matching_array); index++) {
1976 classesToMatch = IOServiceMatching(matching_array[index]);
1977 if (classesToMatch == NULL) {
1978 error_report("IOServiceMatching returned NULL for %s",
1979 matching_array[index]);
1980 continue;
1981 }
1982 CFDictionarySetValue(classesToMatch, CFSTR(kIOMediaEjectableKey),
1983 kCFBooleanTrue);
1984 kernResult = IOServiceGetMatchingServices(masterPort, classesToMatch,
1985 mediaIterator);
1986 if (kernResult != KERN_SUCCESS) {
1987 error_report("Note: IOServiceGetMatchingServices returned %d",
1988 kernResult);
1989 continue;
1990 }
1991
1992 /* If a match was found, leave the loop */
1993 if (*mediaIterator != 0) {
1994 DPRINTF("Matching using %s\n", matching_array[index]);
1995 mediaType = g_strdup(matching_array[index]);
1996 break;
1997 }
1998 }
1999 return mediaType;
2000 }
2001
2002 kern_return_t GetBSDPath(io_iterator_t mediaIterator, char *bsdPath,
2003 CFIndex maxPathSize, int flags)
2004 {
2005 io_object_t nextMedia;
2006 kern_return_t kernResult = KERN_FAILURE;
2007 *bsdPath = '\0';
2008 nextMedia = IOIteratorNext( mediaIterator );
2009 if ( nextMedia )
2010 {
2011 CFTypeRef bsdPathAsCFString;
2012 bsdPathAsCFString = IORegistryEntryCreateCFProperty( nextMedia, CFSTR( kIOBSDNameKey ), kCFAllocatorDefault, 0 );
2013 if ( bsdPathAsCFString ) {
2014 size_t devPathLength;
2015 strcpy( bsdPath, _PATH_DEV );
2016 if (flags & BDRV_O_NOCACHE) {
2017 strcat(bsdPath, "r");
2018 }
2019 devPathLength = strlen( bsdPath );
2020 if ( CFStringGetCString( bsdPathAsCFString, bsdPath + devPathLength, maxPathSize - devPathLength, kCFStringEncodingASCII ) ) {
2021 kernResult = KERN_SUCCESS;
2022 }
2023 CFRelease( bsdPathAsCFString );
2024 }
2025 IOObjectRelease( nextMedia );
2026 }
2027
2028 return kernResult;
2029 }
2030
2031 /* Sets up a real cdrom for use in QEMU */
2032 static bool setup_cdrom(char *bsd_path, Error **errp)
2033 {
2034 int index, num_of_test_partitions = 2, fd;
2035 char test_partition[MAXPATHLEN];
2036 bool partition_found = false;
2037
2038 /* look for a working partition */
2039 for (index = 0; index < num_of_test_partitions; index++) {
2040 snprintf(test_partition, sizeof(test_partition), "%ss%d", bsd_path,
2041 index);
2042 fd = qemu_open(test_partition, O_RDONLY | O_BINARY | O_LARGEFILE);
2043 if (fd >= 0) {
2044 partition_found = true;
2045 qemu_close(fd);
2046 break;
2047 }
2048 }
2049
2050 /* if a working partition on the device was not found */
2051 if (partition_found == false) {
2052 error_setg(errp, "Failed to find a working partition on disc");
2053 } else {
2054 DPRINTF("Using %s as optical disc\n", test_partition);
2055 pstrcpy(bsd_path, MAXPATHLEN, test_partition);
2056 }
2057 return partition_found;
2058 }
2059
2060 /* Prints directions on mounting and unmounting a device */
2061 static void print_unmounting_directions(const char *file_name)
2062 {
2063 error_report("If device %s is mounted on the desktop, unmount"
2064 " it first before using it in QEMU", file_name);
2065 error_report("Command to unmount device: diskutil unmountDisk %s",
2066 file_name);
2067 error_report("Command to mount device: diskutil mountDisk %s", file_name);
2068 }
2069
2070 #endif /* defined(__APPLE__) && defined(__MACH__) */
2071
2072 static int hdev_probe_device(const char *filename)
2073 {
2074 struct stat st;
2075
2076 /* allow a dedicated CD-ROM driver to match with a higher priority */
2077 if (strstart(filename, "/dev/cdrom", NULL))
2078 return 50;
2079
2080 if (stat(filename, &st) >= 0 &&
2081 (S_ISCHR(st.st_mode) || S_ISBLK(st.st_mode))) {
2082 return 100;
2083 }
2084
2085 return 0;
2086 }
2087
2088 static int check_hdev_writable(BDRVRawState *s)
2089 {
2090 #if defined(BLKROGET)
2091 /* Linux block devices can be configured "read-only" using blockdev(8).
2092 * This is independent of device node permissions and therefore open(2)
2093 * with O_RDWR succeeds. Actual writes fail with EPERM.
2094 *
2095 * bdrv_open() is supposed to fail if the disk is read-only. Explicitly
2096 * check for read-only block devices so that Linux block devices behave
2097 * properly.
2098 */
2099 struct stat st;
2100 int readonly = 0;
2101
2102 if (fstat(s->fd, &st)) {
2103 return -errno;
2104 }
2105
2106 if (!S_ISBLK(st.st_mode)) {
2107 return 0;
2108 }
2109
2110 if (ioctl(s->fd, BLKROGET, &readonly) < 0) {
2111 return -errno;
2112 }
2113
2114 if (readonly) {
2115 return -EACCES;
2116 }
2117 #endif /* defined(BLKROGET) */
2118 return 0;
2119 }
2120
2121 static void hdev_parse_filename(const char *filename, QDict *options,
2122 Error **errp)
2123 {
2124 /* The prefix is optional, just as for "file". */
2125 strstart(filename, "host_device:", &filename);
2126
2127 qdict_put_obj(options, "filename", QOBJECT(qstring_from_str(filename)));
2128 }
2129
2130 static bool hdev_is_sg(BlockDriverState *bs)
2131 {
2132
2133 #if defined(__linux__)
2134
2135 struct stat st;
2136 struct sg_scsi_id scsiid;
2137 int sg_version;
2138
2139 if (stat(bs->filename, &st) >= 0 && S_ISCHR(st.st_mode) &&
2140 !bdrv_ioctl(bs, SG_GET_VERSION_NUM, &sg_version) &&
2141 !bdrv_ioctl(bs, SG_GET_SCSI_ID, &scsiid)) {
2142 DPRINTF("SG device found: type=%d, version=%d\n",
2143 scsiid.scsi_type, sg_version);
2144 return true;
2145 }
2146
2147 #endif
2148
2149 return false;
2150 }
2151
2152 static int hdev_open(BlockDriverState *bs, QDict *options, int flags,
2153 Error **errp)
2154 {
2155 BDRVRawState *s = bs->opaque;
2156 Error *local_err = NULL;
2157 int ret;
2158
2159 #if defined(__APPLE__) && defined(__MACH__)
2160 const char *filename = qdict_get_str(options, "filename");
2161 char bsd_path[MAXPATHLEN] = "";
2162 bool error_occurred = false;
2163
2164 /* If using a real cdrom */
2165 if (strcmp(filename, "/dev/cdrom") == 0) {
2166 char *mediaType = NULL;
2167 kern_return_t ret_val;
2168 io_iterator_t mediaIterator = 0;
2169
2170 mediaType = FindEjectableOpticalMedia(&mediaIterator);
2171 if (mediaType == NULL) {
2172 error_setg(errp, "Please make sure your CD/DVD is in the optical"
2173 " drive");
2174 error_occurred = true;
2175 goto hdev_open_Mac_error;
2176 }
2177
2178 ret_val = GetBSDPath(mediaIterator, bsd_path, sizeof(bsd_path), flags);
2179 if (ret_val != KERN_SUCCESS) {
2180 error_setg(errp, "Could not get BSD path for optical drive");
2181 error_occurred = true;
2182 goto hdev_open_Mac_error;
2183 }
2184
2185 /* If a real optical drive was not found */
2186 if (bsd_path[0] == '\0') {
2187 error_setg(errp, "Failed to obtain bsd path for optical drive");
2188 error_occurred = true;
2189 goto hdev_open_Mac_error;
2190 }
2191
2192 /* If using a cdrom disc and finding a partition on the disc failed */
2193 if (strncmp(mediaType, kIOCDMediaClass, 9) == 0 &&
2194 setup_cdrom(bsd_path, errp) == false) {
2195 print_unmounting_directions(bsd_path);
2196 error_occurred = true;
2197 goto hdev_open_Mac_error;
2198 }
2199
2200 qdict_put(options, "filename", qstring_from_str(bsd_path));
2201
2202 hdev_open_Mac_error:
2203 g_free(mediaType);
2204 if (mediaIterator) {
2205 IOObjectRelease(mediaIterator);
2206 }
2207 if (error_occurred) {
2208 return -ENOENT;
2209 }
2210 }
2211 #endif /* defined(__APPLE__) && defined(__MACH__) */
2212
2213 s->type = FTYPE_FILE;
2214
2215 ret = raw_open_common(bs, options, flags, 0, &local_err);
2216 if (ret < 0) {
2217 if (local_err) {
2218 error_propagate(errp, local_err);
2219 }
2220 #if defined(__APPLE__) && defined(__MACH__)
2221 if (*bsd_path) {
2222 filename = bsd_path;
2223 }
2224 /* if a physical device experienced an error while being opened */
2225 if (strncmp(filename, "/dev/", 5) == 0) {
2226 print_unmounting_directions(filename);
2227 }
2228 #endif /* defined(__APPLE__) && defined(__MACH__) */
2229 return ret;
2230 }
2231
2232 /* Since this does ioctl the device must be already opened */
2233 bs->sg = hdev_is_sg(bs);
2234
2235 if (flags & BDRV_O_RDWR) {
2236 ret = check_hdev_writable(s);
2237 if (ret < 0) {
2238 raw_close(bs);
2239 error_setg_errno(errp, -ret, "The device is not writable");
2240 return ret;
2241 }
2242 }
2243
2244 return ret;
2245 }
2246
2247 #if defined(__linux__)
2248
2249 static BlockAIOCB *hdev_aio_ioctl(BlockDriverState *bs,
2250 unsigned long int req, void *buf,
2251 BlockCompletionFunc *cb, void *opaque)
2252 {
2253 BDRVRawState *s = bs->opaque;
2254 RawPosixAIOData *acb;
2255 ThreadPool *pool;
2256
2257 if (fd_open(bs) < 0)
2258 return NULL;
2259
2260 acb = g_new(RawPosixAIOData, 1);
2261 acb->bs = bs;
2262 acb->aio_type = QEMU_AIO_IOCTL;
2263 acb->aio_fildes = s->fd;
2264 acb->aio_offset = 0;
2265 acb->aio_ioctl_buf = buf;
2266 acb->aio_ioctl_cmd = req;
2267 pool = aio_get_thread_pool(bdrv_get_aio_context(bs));
2268 return thread_pool_submit_aio(pool, aio_worker, acb, cb, opaque);
2269 }
2270 #endif /* linux */
2271
2272 static int fd_open(BlockDriverState *bs)
2273 {
2274 BDRVRawState *s = bs->opaque;
2275
2276 /* this is just to ensure s->fd is sane (its called by io ops) */
2277 if (s->fd >= 0)
2278 return 0;
2279 return -EIO;
2280 }
2281
2282 static coroutine_fn BlockAIOCB *hdev_aio_discard(BlockDriverState *bs,
2283 int64_t sector_num, int nb_sectors,
2284 BlockCompletionFunc *cb, void *opaque)
2285 {
2286 BDRVRawState *s = bs->opaque;
2287
2288 if (fd_open(bs) < 0) {
2289 return NULL;
2290 }
2291 return paio_submit(bs, s->fd, sector_num, NULL, nb_sectors,
2292 cb, opaque, QEMU_AIO_DISCARD|QEMU_AIO_BLKDEV);
2293 }
2294
2295 static coroutine_fn int hdev_co_write_zeroes(BlockDriverState *bs,
2296 int64_t sector_num, int nb_sectors, BdrvRequestFlags flags)
2297 {
2298 BDRVRawState *s = bs->opaque;
2299 int rc;
2300
2301 rc = fd_open(bs);
2302 if (rc < 0) {
2303 return rc;
2304 }
2305 if (!(flags & BDRV_REQ_MAY_UNMAP)) {
2306 return paio_submit_co(bs, s->fd, sector_num, NULL, nb_sectors,
2307 QEMU_AIO_WRITE_ZEROES|QEMU_AIO_BLKDEV);
2308 } else if (s->discard_zeroes) {
2309 return paio_submit_co(bs, s->fd, sector_num, NULL, nb_sectors,
2310 QEMU_AIO_DISCARD|QEMU_AIO_BLKDEV);
2311 }
2312 return -ENOTSUP;
2313 }
2314
2315 static int hdev_create(const char *filename, QemuOpts *opts,
2316 Error **errp)
2317 {
2318 int fd;
2319 int ret = 0;
2320 struct stat stat_buf;
2321 int64_t total_size = 0;
2322 bool has_prefix;
2323
2324 /* This function is used by both protocol block drivers and therefore either
2325 * of these prefixes may be given.
2326 * The return value has to be stored somewhere, otherwise this is an error
2327 * due to -Werror=unused-value. */
2328 has_prefix =
2329 strstart(filename, "host_device:", &filename) ||
2330 strstart(filename, "host_cdrom:" , &filename);
2331
2332 (void)has_prefix;
2333
2334 ret = raw_normalize_devicepath(&filename);
2335 if (ret < 0) {
2336 error_setg_errno(errp, -ret, "Could not normalize device path");
2337 return ret;
2338 }
2339
2340 /* Read out options */
2341 total_size = ROUND_UP(qemu_opt_get_size_del(opts, BLOCK_OPT_SIZE, 0),
2342 BDRV_SECTOR_SIZE);
2343
2344 fd = qemu_open(filename, O_WRONLY | O_BINARY);
2345 if (fd < 0) {
2346 ret = -errno;
2347 error_setg_errno(errp, -ret, "Could not open device");
2348 return ret;
2349 }
2350
2351 if (fstat(fd, &stat_buf) < 0) {
2352 ret = -errno;
2353 error_setg_errno(errp, -ret, "Could not stat device");
2354 } else if (!S_ISBLK(stat_buf.st_mode) && !S_ISCHR(stat_buf.st_mode)) {
2355 error_setg(errp,
2356 "The given file is neither a block nor a character device");
2357 ret = -ENODEV;
2358 } else if (lseek(fd, 0, SEEK_END) < total_size) {
2359 error_setg(errp, "Device is too small");
2360 ret = -ENOSPC;
2361 }
2362
2363 qemu_close(fd);
2364 return ret;
2365 }
2366
2367 static BlockDriver bdrv_host_device = {
2368 .format_name = "host_device",
2369 .protocol_name = "host_device",
2370 .instance_size = sizeof(BDRVRawState),
2371 .bdrv_needs_filename = true,
2372 .bdrv_probe_device = hdev_probe_device,
2373 .bdrv_parse_filename = hdev_parse_filename,
2374 .bdrv_file_open = hdev_open,
2375 .bdrv_close = raw_close,
2376 .bdrv_reopen_prepare = raw_reopen_prepare,
2377 .bdrv_reopen_commit = raw_reopen_commit,
2378 .bdrv_reopen_abort = raw_reopen_abort,
2379 .bdrv_create = hdev_create,
2380 .create_opts = &raw_create_opts,
2381 .bdrv_co_write_zeroes = hdev_co_write_zeroes,
2382
2383 .bdrv_aio_readv = raw_aio_readv,
2384 .bdrv_aio_writev = raw_aio_writev,
2385 .bdrv_aio_flush = raw_aio_flush,
2386 .bdrv_aio_discard = hdev_aio_discard,
2387 .bdrv_refresh_limits = raw_refresh_limits,
2388 .bdrv_io_plug = raw_aio_plug,
2389 .bdrv_io_unplug = raw_aio_unplug,
2390
2391 .bdrv_truncate = raw_truncate,
2392 .bdrv_getlength = raw_getlength,
2393 .bdrv_get_info = raw_get_info,
2394 .bdrv_get_allocated_file_size
2395 = raw_get_allocated_file_size,
2396 .bdrv_probe_blocksizes = hdev_probe_blocksizes,
2397 .bdrv_probe_geometry = hdev_probe_geometry,
2398
2399 .bdrv_detach_aio_context = raw_detach_aio_context,
2400 .bdrv_attach_aio_context = raw_attach_aio_context,
2401
2402 /* generic scsi device */
2403 #ifdef __linux__
2404 .bdrv_aio_ioctl = hdev_aio_ioctl,
2405 #endif
2406 };
2407
2408 #if defined(__linux__) || defined(__FreeBSD__) || defined(__FreeBSD_kernel__)
2409 static void cdrom_parse_filename(const char *filename, QDict *options,
2410 Error **errp)
2411 {
2412 /* The prefix is optional, just as for "file". */
2413 strstart(filename, "host_cdrom:", &filename);
2414
2415 qdict_put_obj(options, "filename", QOBJECT(qstring_from_str(filename)));
2416 }
2417 #endif
2418
2419 #ifdef __linux__
2420 static int cdrom_open(BlockDriverState *bs, QDict *options, int flags,
2421 Error **errp)
2422 {
2423 BDRVRawState *s = bs->opaque;
2424 Error *local_err = NULL;
2425 int ret;
2426
2427 s->type = FTYPE_CD;
2428
2429 /* open will not fail even if no CD is inserted, so add O_NONBLOCK */
2430 ret = raw_open_common(bs, options, flags, O_NONBLOCK, &local_err);
2431 if (local_err) {
2432 error_propagate(errp, local_err);
2433 }
2434 return ret;
2435 }
2436
2437 static int cdrom_probe_device(const char *filename)
2438 {
2439 int fd, ret;
2440 int prio = 0;
2441 struct stat st;
2442
2443 fd = qemu_open(filename, O_RDONLY | O_NONBLOCK);
2444 if (fd < 0) {
2445 goto out;
2446 }
2447 ret = fstat(fd, &st);
2448 if (ret == -1 || !S_ISBLK(st.st_mode)) {
2449 goto outc;
2450 }
2451
2452 /* Attempt to detect via a CDROM specific ioctl */
2453 ret = ioctl(fd, CDROM_DRIVE_STATUS, CDSL_CURRENT);
2454 if (ret >= 0)
2455 prio = 100;
2456
2457 outc:
2458 qemu_close(fd);
2459 out:
2460 return prio;
2461 }
2462
2463 static bool cdrom_is_inserted(BlockDriverState *bs)
2464 {
2465 BDRVRawState *s = bs->opaque;
2466 int ret;
2467
2468 ret = ioctl(s->fd, CDROM_DRIVE_STATUS, CDSL_CURRENT);
2469 return ret == CDS_DISC_OK;
2470 }
2471
2472 static void cdrom_eject(BlockDriverState *bs, bool eject_flag)
2473 {
2474 BDRVRawState *s = bs->opaque;
2475
2476 if (eject_flag) {
2477 if (ioctl(s->fd, CDROMEJECT, NULL) < 0)
2478 perror("CDROMEJECT");
2479 } else {
2480 if (ioctl(s->fd, CDROMCLOSETRAY, NULL) < 0)
2481 perror("CDROMEJECT");
2482 }
2483 }
2484
2485 static void cdrom_lock_medium(BlockDriverState *bs, bool locked)
2486 {
2487 BDRVRawState *s = bs->opaque;
2488
2489 if (ioctl(s->fd, CDROM_LOCKDOOR, locked) < 0) {
2490 /*
2491 * Note: an error can happen if the distribution automatically
2492 * mounts the CD-ROM
2493 */
2494 /* perror("CDROM_LOCKDOOR"); */
2495 }
2496 }
2497
2498 static BlockDriver bdrv_host_cdrom = {
2499 .format_name = "host_cdrom",
2500 .protocol_name = "host_cdrom",
2501 .instance_size = sizeof(BDRVRawState),
2502 .bdrv_needs_filename = true,
2503 .bdrv_probe_device = cdrom_probe_device,
2504 .bdrv_parse_filename = cdrom_parse_filename,
2505 .bdrv_file_open = cdrom_open,
2506 .bdrv_close = raw_close,
2507 .bdrv_reopen_prepare = raw_reopen_prepare,
2508 .bdrv_reopen_commit = raw_reopen_commit,
2509 .bdrv_reopen_abort = raw_reopen_abort,
2510 .bdrv_create = hdev_create,
2511 .create_opts = &raw_create_opts,
2512
2513 .bdrv_aio_readv = raw_aio_readv,
2514 .bdrv_aio_writev = raw_aio_writev,
2515 .bdrv_aio_flush = raw_aio_flush,
2516 .bdrv_refresh_limits = raw_refresh_limits,
2517 .bdrv_io_plug = raw_aio_plug,
2518 .bdrv_io_unplug = raw_aio_unplug,
2519
2520 .bdrv_truncate = raw_truncate,
2521 .bdrv_getlength = raw_getlength,
2522 .has_variable_length = true,
2523 .bdrv_get_allocated_file_size
2524 = raw_get_allocated_file_size,
2525
2526 .bdrv_detach_aio_context = raw_detach_aio_context,
2527 .bdrv_attach_aio_context = raw_attach_aio_context,
2528
2529 /* removable device support */
2530 .bdrv_is_inserted = cdrom_is_inserted,
2531 .bdrv_eject = cdrom_eject,
2532 .bdrv_lock_medium = cdrom_lock_medium,
2533
2534 /* generic scsi device */
2535 .bdrv_aio_ioctl = hdev_aio_ioctl,
2536 };
2537 #endif /* __linux__ */
2538
2539 #if defined (__FreeBSD__) || defined(__FreeBSD_kernel__)
2540 static int cdrom_open(BlockDriverState *bs, QDict *options, int flags,
2541 Error **errp)
2542 {
2543 BDRVRawState *s = bs->opaque;
2544 Error *local_err = NULL;
2545 int ret;
2546
2547 s->type = FTYPE_CD;
2548
2549 ret = raw_open_common(bs, options, flags, 0, &local_err);
2550 if (ret) {
2551 if (local_err) {
2552 error_propagate(errp, local_err);
2553 }
2554 return ret;
2555 }
2556
2557 /* make sure the door isn't locked at this time */
2558 ioctl(s->fd, CDIOCALLOW);
2559 return 0;
2560 }
2561
2562 static int cdrom_probe_device(const char *filename)
2563 {
2564 if (strstart(filename, "/dev/cd", NULL) ||
2565 strstart(filename, "/dev/acd", NULL))
2566 return 100;
2567 return 0;
2568 }
2569
2570 static int cdrom_reopen(BlockDriverState *bs)
2571 {
2572 BDRVRawState *s = bs->opaque;
2573 int fd;
2574
2575 /*
2576 * Force reread of possibly changed/newly loaded disc,
2577 * FreeBSD seems to not notice sometimes...
2578 */
2579 if (s->fd >= 0)
2580 qemu_close(s->fd);
2581 fd = qemu_open(bs->filename, s->open_flags, 0644);
2582 if (fd < 0) {
2583 s->fd = -1;
2584 return -EIO;
2585 }
2586 s->fd = fd;
2587
2588 /* make sure the door isn't locked at this time */
2589 ioctl(s->fd, CDIOCALLOW);
2590 return 0;
2591 }
2592
2593 static bool cdrom_is_inserted(BlockDriverState *bs)
2594 {
2595 return raw_getlength(bs) > 0;
2596 }
2597
2598 static void cdrom_eject(BlockDriverState *bs, bool eject_flag)
2599 {
2600 BDRVRawState *s = bs->opaque;
2601
2602 if (s->fd < 0)
2603 return;
2604
2605 (void) ioctl(s->fd, CDIOCALLOW);
2606
2607 if (eject_flag) {
2608 if (ioctl(s->fd, CDIOCEJECT) < 0)
2609 perror("CDIOCEJECT");
2610 } else {
2611 if (ioctl(s->fd, CDIOCCLOSE) < 0)
2612 perror("CDIOCCLOSE");
2613 }
2614
2615 cdrom_reopen(bs);
2616 }
2617
2618 static void cdrom_lock_medium(BlockDriverState *bs, bool locked)
2619 {
2620 BDRVRawState *s = bs->opaque;
2621
2622 if (s->fd < 0)
2623 return;
2624 if (ioctl(s->fd, (locked ? CDIOCPREVENT : CDIOCALLOW)) < 0) {
2625 /*
2626 * Note: an error can happen if the distribution automatically
2627 * mounts the CD-ROM
2628 */
2629 /* perror("CDROM_LOCKDOOR"); */
2630 }
2631 }
2632
2633 static BlockDriver bdrv_host_cdrom = {
2634 .format_name = "host_cdrom",
2635 .protocol_name = "host_cdrom",
2636 .instance_size = sizeof(BDRVRawState),
2637 .bdrv_needs_filename = true,
2638 .bdrv_probe_device = cdrom_probe_device,
2639 .bdrv_parse_filename = cdrom_parse_filename,
2640 .bdrv_file_open = cdrom_open,
2641 .bdrv_close = raw_close,
2642 .bdrv_reopen_prepare = raw_reopen_prepare,
2643 .bdrv_reopen_commit = raw_reopen_commit,
2644 .bdrv_reopen_abort = raw_reopen_abort,
2645 .bdrv_create = hdev_create,
2646 .create_opts = &raw_create_opts,
2647
2648 .bdrv_aio_readv = raw_aio_readv,
2649 .bdrv_aio_writev = raw_aio_writev,
2650 .bdrv_aio_flush = raw_aio_flush,
2651 .bdrv_refresh_limits = raw_refresh_limits,
2652 .bdrv_io_plug = raw_aio_plug,
2653 .bdrv_io_unplug = raw_aio_unplug,
2654
2655 .bdrv_truncate = raw_truncate,
2656 .bdrv_getlength = raw_getlength,
2657 .has_variable_length = true,
2658 .bdrv_get_allocated_file_size
2659 = raw_get_allocated_file_size,
2660
2661 .bdrv_detach_aio_context = raw_detach_aio_context,
2662 .bdrv_attach_aio_context = raw_attach_aio_context,
2663
2664 /* removable device support */
2665 .bdrv_is_inserted = cdrom_is_inserted,
2666 .bdrv_eject = cdrom_eject,
2667 .bdrv_lock_medium = cdrom_lock_medium,
2668 };
2669 #endif /* __FreeBSD__ */
2670
2671 static void bdrv_file_init(void)
2672 {
2673 /*
2674 * Register all the drivers. Note that order is important, the driver
2675 * registered last will get probed first.
2676 */
2677 bdrv_register(&bdrv_file);
2678 bdrv_register(&bdrv_host_device);
2679 #ifdef __linux__
2680 bdrv_register(&bdrv_host_cdrom);
2681 #endif
2682 #if defined(__FreeBSD__) || defined(__FreeBSD_kernel__)
2683 bdrv_register(&bdrv_host_cdrom);
2684 #endif
2685 }
2686
2687 block_init(bdrv_file_init);