]> Git Repo - qemu.git/blob - block/file-posix.c
f3934c4e1024d2df560667d01e0b738a0cfd7649
[qemu.git] / block / file-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
25 #include "qemu/osdep.h"
26 #include "qemu-common.h"
27 #include "qapi/error.h"
28 #include "qemu/cutils.h"
29 #include "qemu/error-report.h"
30 #include "block/block_int.h"
31 #include "qemu/module.h"
32 #include "qemu/option.h"
33 #include "trace.h"
34 #include "block/thread-pool.h"
35 #include "qemu/iov.h"
36 #include "block/raw-aio.h"
37 #include "qapi/qmp/qdict.h"
38 #include "qapi/qmp/qstring.h"
39
40 #include "scsi/pr-manager.h"
41 #include "scsi/constants.h"
42
43 #if defined(__APPLE__) && (__MACH__)
44 #include <paths.h>
45 #include <sys/param.h>
46 #include <IOKit/IOKitLib.h>
47 #include <IOKit/IOBSD.h>
48 #include <IOKit/storage/IOMediaBSDClient.h>
49 #include <IOKit/storage/IOMedia.h>
50 #include <IOKit/storage/IOCDMedia.h>
51 //#include <IOKit/storage/IOCDTypes.h>
52 #include <IOKit/storage/IODVDMedia.h>
53 #include <CoreFoundation/CoreFoundation.h>
54 #endif
55
56 #ifdef __sun__
57 #define _POSIX_PTHREAD_SEMANTICS 1
58 #include <sys/dkio.h>
59 #endif
60 #ifdef __linux__
61 #include <sys/ioctl.h>
62 #include <sys/param.h>
63 #include <sys/syscall.h>
64 #include <linux/cdrom.h>
65 #include <linux/fd.h>
66 #include <linux/fs.h>
67 #include <linux/hdreg.h>
68 #include <scsi/sg.h>
69 #ifdef __s390__
70 #include <asm/dasd.h>
71 #endif
72 #ifndef FS_NOCOW_FL
73 #define FS_NOCOW_FL                     0x00800000 /* Do not cow file */
74 #endif
75 #endif
76 #if defined(CONFIG_FALLOCATE_PUNCH_HOLE) || defined(CONFIG_FALLOCATE_ZERO_RANGE)
77 #include <linux/falloc.h>
78 #endif
79 #if defined (__FreeBSD__) || defined(__FreeBSD_kernel__)
80 #include <sys/disk.h>
81 #include <sys/cdio.h>
82 #endif
83
84 #ifdef __OpenBSD__
85 #include <sys/ioctl.h>
86 #include <sys/disklabel.h>
87 #include <sys/dkio.h>
88 #endif
89
90 #ifdef __NetBSD__
91 #include <sys/ioctl.h>
92 #include <sys/disklabel.h>
93 #include <sys/dkio.h>
94 #include <sys/disk.h>
95 #endif
96
97 #ifdef __DragonFly__
98 #include <sys/ioctl.h>
99 #include <sys/diskslice.h>
100 #endif
101
102 #ifdef CONFIG_XFS
103 #include <xfs/xfs.h>
104 #endif
105
106 #include "trace.h"
107
108 /* OS X does not have O_DSYNC */
109 #ifndef O_DSYNC
110 #ifdef O_SYNC
111 #define O_DSYNC O_SYNC
112 #elif defined(O_FSYNC)
113 #define O_DSYNC O_FSYNC
114 #endif
115 #endif
116
117 /* Approximate O_DIRECT with O_DSYNC if O_DIRECT isn't available */
118 #ifndef O_DIRECT
119 #define O_DIRECT O_DSYNC
120 #endif
121
122 #define FTYPE_FILE   0
123 #define FTYPE_CD     1
124
125 #define MAX_BLOCKSIZE   4096
126
127 /* Posix file locking bytes. Libvirt takes byte 0, we start from higher bytes,
128  * leaving a few more bytes for its future use. */
129 #define RAW_LOCK_PERM_BASE             100
130 #define RAW_LOCK_SHARED_BASE           200
131
132 typedef struct BDRVRawState {
133     int fd;
134     bool use_lock;
135     int type;
136     int open_flags;
137     size_t buf_align;
138
139     /* The current permissions. */
140     uint64_t perm;
141     uint64_t shared_perm;
142
143     /* The perms bits whose corresponding bytes are already locked in
144      * s->fd. */
145     uint64_t locked_perm;
146     uint64_t locked_shared_perm;
147
148     int perm_change_fd;
149     int perm_change_flags;
150     BDRVReopenState *reopen_state;
151
152 #ifdef CONFIG_XFS
153     bool is_xfs:1;
154 #endif
155     bool has_discard:1;
156     bool has_write_zeroes:1;
157     bool discard_zeroes:1;
158     bool use_linux_aio:1;
159     bool page_cache_inconsistent:1;
160     bool has_fallocate;
161     bool needs_alignment;
162     bool drop_cache;
163     bool check_cache_dropped;
164     struct {
165         uint64_t discard_nb_ok;
166         uint64_t discard_nb_failed;
167         uint64_t discard_bytes_ok;
168     } stats;
169
170     PRManager *pr_mgr;
171 } BDRVRawState;
172
173 typedef struct BDRVRawReopenState {
174     int fd;
175     int open_flags;
176     bool drop_cache;
177     bool check_cache_dropped;
178 } BDRVRawReopenState;
179
180 static int fd_open(BlockDriverState *bs);
181 static int64_t raw_getlength(BlockDriverState *bs);
182
183 typedef struct RawPosixAIOData {
184     BlockDriverState *bs;
185     int aio_type;
186     int aio_fildes;
187
188     off_t aio_offset;
189     uint64_t aio_nbytes;
190
191     union {
192         struct {
193             struct iovec *iov;
194             int niov;
195         } io;
196         struct {
197             uint64_t cmd;
198             void *buf;
199         } ioctl;
200         struct {
201             int aio_fd2;
202             off_t aio_offset2;
203         } copy_range;
204         struct {
205             PreallocMode prealloc;
206             Error **errp;
207         } truncate;
208     };
209 } RawPosixAIOData;
210
211 #if defined(__FreeBSD__) || defined(__FreeBSD_kernel__)
212 static int cdrom_reopen(BlockDriverState *bs);
213 #endif
214
215 #if defined(__NetBSD__)
216 static int raw_normalize_devicepath(const char **filename, Error **errp)
217 {
218     static char namebuf[PATH_MAX];
219     const char *dp, *fname;
220     struct stat sb;
221
222     fname = *filename;
223     dp = strrchr(fname, '/');
224     if (lstat(fname, &sb) < 0) {
225         error_setg_file_open(errp, errno, fname);
226         return -errno;
227     }
228
229     if (!S_ISBLK(sb.st_mode)) {
230         return 0;
231     }
232
233     if (dp == NULL) {
234         snprintf(namebuf, PATH_MAX, "r%s", fname);
235     } else {
236         snprintf(namebuf, PATH_MAX, "%.*s/r%s",
237             (int)(dp - fname), fname, dp + 1);
238     }
239     *filename = namebuf;
240     warn_report("%s is a block device, using %s", fname, *filename);
241
242     return 0;
243 }
244 #else
245 static int raw_normalize_devicepath(const char **filename, Error **errp)
246 {
247     return 0;
248 }
249 #endif
250
251 /*
252  * Get logical block size via ioctl. On success store it in @sector_size_p.
253  */
254 static int probe_logical_blocksize(int fd, unsigned int *sector_size_p)
255 {
256     unsigned int sector_size;
257     bool success = false;
258     int i;
259
260     errno = ENOTSUP;
261     static const unsigned long ioctl_list[] = {
262 #ifdef BLKSSZGET
263         BLKSSZGET,
264 #endif
265 #ifdef DKIOCGETBLOCKSIZE
266         DKIOCGETBLOCKSIZE,
267 #endif
268 #ifdef DIOCGSECTORSIZE
269         DIOCGSECTORSIZE,
270 #endif
271     };
272
273     /* Try a few ioctls to get the right size */
274     for (i = 0; i < (int)ARRAY_SIZE(ioctl_list); i++) {
275         if (ioctl(fd, ioctl_list[i], &sector_size) >= 0) {
276             *sector_size_p = sector_size;
277             success = true;
278         }
279     }
280
281     return success ? 0 : -errno;
282 }
283
284 /**
285  * Get physical block size of @fd.
286  * On success, store it in @blk_size and return 0.
287  * On failure, return -errno.
288  */
289 static int probe_physical_blocksize(int fd, unsigned int *blk_size)
290 {
291 #ifdef BLKPBSZGET
292     if (ioctl(fd, BLKPBSZGET, blk_size) < 0) {
293         return -errno;
294     }
295     return 0;
296 #else
297     return -ENOTSUP;
298 #endif
299 }
300
301 /* Check if read is allowed with given memory buffer and length.
302  *
303  * This function is used to check O_DIRECT memory buffer and request alignment.
304  */
305 static bool raw_is_io_aligned(int fd, void *buf, size_t len)
306 {
307     ssize_t ret = pread(fd, buf, len, 0);
308
309     if (ret >= 0) {
310         return true;
311     }
312
313 #ifdef __linux__
314     /* The Linux kernel returns EINVAL for misaligned O_DIRECT reads.  Ignore
315      * other errors (e.g. real I/O error), which could happen on a failed
316      * drive, since we only care about probing alignment.
317      */
318     if (errno != EINVAL) {
319         return true;
320     }
321 #endif
322
323     return false;
324 }
325
326 static void raw_probe_alignment(BlockDriverState *bs, int fd, Error **errp)
327 {
328     BDRVRawState *s = bs->opaque;
329     char *buf;
330     size_t max_align = MAX(MAX_BLOCKSIZE, getpagesize());
331     size_t alignments[] = {1, 512, 1024, 2048, 4096};
332
333     /* For SCSI generic devices the alignment is not really used.
334        With buffered I/O, we don't have any restrictions. */
335     if (bdrv_is_sg(bs) || !s->needs_alignment) {
336         bs->bl.request_alignment = 1;
337         s->buf_align = 1;
338         return;
339     }
340
341     bs->bl.request_alignment = 0;
342     s->buf_align = 0;
343     /* Let's try to use the logical blocksize for the alignment. */
344     if (probe_logical_blocksize(fd, &bs->bl.request_alignment) < 0) {
345         bs->bl.request_alignment = 0;
346     }
347 #ifdef CONFIG_XFS
348     if (s->is_xfs) {
349         struct dioattr da;
350         if (xfsctl(NULL, fd, XFS_IOC_DIOINFO, &da) >= 0) {
351             bs->bl.request_alignment = da.d_miniosz;
352             /* The kernel returns wrong information for d_mem */
353             /* s->buf_align = da.d_mem; */
354         }
355     }
356 #endif
357
358     /*
359      * If we could not get the sizes so far, we can only guess them. First try
360      * to detect request alignment, since it is more likely to succeed. Then
361      * try to detect buf_align, which cannot be detected in some cases (e.g.
362      * Gluster). If buf_align cannot be detected, we fallback to the value of
363      * request_alignment.
364      */
365
366     if (!bs->bl.request_alignment) {
367         int i;
368         size_t align;
369         buf = qemu_memalign(max_align, max_align);
370         for (i = 0; i < ARRAY_SIZE(alignments); i++) {
371             align = alignments[i];
372             if (raw_is_io_aligned(fd, buf, align)) {
373                 /* Fallback to safe value. */
374                 bs->bl.request_alignment = (align != 1) ? align : max_align;
375                 break;
376             }
377         }
378         qemu_vfree(buf);
379     }
380
381     if (!s->buf_align) {
382         int i;
383         size_t align;
384         buf = qemu_memalign(max_align, 2 * max_align);
385         for (i = 0; i < ARRAY_SIZE(alignments); i++) {
386             align = alignments[i];
387             if (raw_is_io_aligned(fd, buf + align, max_align)) {
388                 /* Fallback to request_alignment. */
389                 s->buf_align = (align != 1) ? align : bs->bl.request_alignment;
390                 break;
391             }
392         }
393         qemu_vfree(buf);
394     }
395
396     if (!s->buf_align || !bs->bl.request_alignment) {
397         error_setg(errp, "Could not find working O_DIRECT alignment");
398         error_append_hint(errp, "Try cache.direct=off\n");
399     }
400 }
401
402 static void raw_parse_flags(int bdrv_flags, int *open_flags, bool has_writers)
403 {
404     bool read_write = false;
405     assert(open_flags != NULL);
406
407     *open_flags |= O_BINARY;
408     *open_flags &= ~O_ACCMODE;
409
410     if (bdrv_flags & BDRV_O_AUTO_RDONLY) {
411         read_write = has_writers;
412     } else if (bdrv_flags & BDRV_O_RDWR) {
413         read_write = true;
414     }
415
416     if (read_write) {
417         *open_flags |= O_RDWR;
418     } else {
419         *open_flags |= O_RDONLY;
420     }
421
422     /* Use O_DSYNC for write-through caching, no flags for write-back caching,
423      * and O_DIRECT for no caching. */
424     if ((bdrv_flags & BDRV_O_NOCACHE)) {
425         *open_flags |= O_DIRECT;
426     }
427 }
428
429 static void raw_parse_filename(const char *filename, QDict *options,
430                                Error **errp)
431 {
432     bdrv_parse_filename_strip_prefix(filename, "file:", options);
433 }
434
435 static QemuOptsList raw_runtime_opts = {
436     .name = "raw",
437     .head = QTAILQ_HEAD_INITIALIZER(raw_runtime_opts.head),
438     .desc = {
439         {
440             .name = "filename",
441             .type = QEMU_OPT_STRING,
442             .help = "File name of the image",
443         },
444         {
445             .name = "aio",
446             .type = QEMU_OPT_STRING,
447             .help = "host AIO implementation (threads, native)",
448         },
449         {
450             .name = "locking",
451             .type = QEMU_OPT_STRING,
452             .help = "file locking mode (on/off/auto, default: auto)",
453         },
454         {
455             .name = "pr-manager",
456             .type = QEMU_OPT_STRING,
457             .help = "id of persistent reservation manager object (default: none)",
458         },
459 #if defined(__linux__)
460         {
461             .name = "drop-cache",
462             .type = QEMU_OPT_BOOL,
463             .help = "invalidate page cache during live migration (default: on)",
464         },
465 #endif
466         {
467             .name = "x-check-cache-dropped",
468             .type = QEMU_OPT_BOOL,
469             .help = "check that page cache was dropped on live migration (default: off)"
470         },
471         { /* end of list */ }
472     },
473 };
474
475 static const char *const mutable_opts[] = { "x-check-cache-dropped", NULL };
476
477 static int raw_open_common(BlockDriverState *bs, QDict *options,
478                            int bdrv_flags, int open_flags,
479                            bool device, Error **errp)
480 {
481     BDRVRawState *s = bs->opaque;
482     QemuOpts *opts;
483     Error *local_err = NULL;
484     const char *filename = NULL;
485     const char *str;
486     BlockdevAioOptions aio, aio_default;
487     int fd, ret;
488     struct stat st;
489     OnOffAuto locking;
490
491     opts = qemu_opts_create(&raw_runtime_opts, NULL, 0, &error_abort);
492     qemu_opts_absorb_qdict(opts, options, &local_err);
493     if (local_err) {
494         error_propagate(errp, local_err);
495         ret = -EINVAL;
496         goto fail;
497     }
498
499     filename = qemu_opt_get(opts, "filename");
500
501     ret = raw_normalize_devicepath(&filename, errp);
502     if (ret != 0) {
503         goto fail;
504     }
505
506     aio_default = (bdrv_flags & BDRV_O_NATIVE_AIO)
507                   ? BLOCKDEV_AIO_OPTIONS_NATIVE
508                   : BLOCKDEV_AIO_OPTIONS_THREADS;
509     aio = qapi_enum_parse(&BlockdevAioOptions_lookup,
510                           qemu_opt_get(opts, "aio"),
511                           aio_default, &local_err);
512     if (local_err) {
513         error_propagate(errp, local_err);
514         ret = -EINVAL;
515         goto fail;
516     }
517     s->use_linux_aio = (aio == BLOCKDEV_AIO_OPTIONS_NATIVE);
518
519     locking = qapi_enum_parse(&OnOffAuto_lookup,
520                               qemu_opt_get(opts, "locking"),
521                               ON_OFF_AUTO_AUTO, &local_err);
522     if (local_err) {
523         error_propagate(errp, local_err);
524         ret = -EINVAL;
525         goto fail;
526     }
527     switch (locking) {
528     case ON_OFF_AUTO_ON:
529         s->use_lock = true;
530         if (!qemu_has_ofd_lock()) {
531             warn_report("File lock requested but OFD locking syscall is "
532                         "unavailable, falling back to POSIX file locks");
533             error_printf("Due to the implementation, locks can be lost "
534                          "unexpectedly.\n");
535         }
536         break;
537     case ON_OFF_AUTO_OFF:
538         s->use_lock = false;
539         break;
540     case ON_OFF_AUTO_AUTO:
541         s->use_lock = qemu_has_ofd_lock();
542         break;
543     default:
544         abort();
545     }
546
547     str = qemu_opt_get(opts, "pr-manager");
548     if (str) {
549         s->pr_mgr = pr_manager_lookup(str, &local_err);
550         if (local_err) {
551             error_propagate(errp, local_err);
552             ret = -EINVAL;
553             goto fail;
554         }
555     }
556
557     s->drop_cache = qemu_opt_get_bool(opts, "drop-cache", true);
558     s->check_cache_dropped = qemu_opt_get_bool(opts, "x-check-cache-dropped",
559                                                false);
560
561     s->open_flags = open_flags;
562     raw_parse_flags(bdrv_flags, &s->open_flags, false);
563
564     s->fd = -1;
565     fd = qemu_open(filename, s->open_flags, 0644);
566     ret = fd < 0 ? -errno : 0;
567
568     if (ret < 0) {
569         error_setg_file_open(errp, -ret, filename);
570         if (ret == -EROFS) {
571             ret = -EACCES;
572         }
573         goto fail;
574     }
575     s->fd = fd;
576
577     s->perm = 0;
578     s->shared_perm = BLK_PERM_ALL;
579
580 #ifdef CONFIG_LINUX_AIO
581      /* Currently Linux does AIO only for files opened with O_DIRECT */
582     if (s->use_linux_aio) {
583         if (!(s->open_flags & O_DIRECT)) {
584             error_setg(errp, "aio=native was specified, but it requires "
585                              "cache.direct=on, which was not specified.");
586             ret = -EINVAL;
587             goto fail;
588         }
589         if (!aio_setup_linux_aio(bdrv_get_aio_context(bs), errp)) {
590             error_prepend(errp, "Unable to use native AIO: ");
591             goto fail;
592         }
593     }
594 #else
595     if (s->use_linux_aio) {
596         error_setg(errp, "aio=native was specified, but is not supported "
597                          "in this build.");
598         ret = -EINVAL;
599         goto fail;
600     }
601 #endif /* !defined(CONFIG_LINUX_AIO) */
602
603     s->has_discard = true;
604     s->has_write_zeroes = true;
605     if ((bs->open_flags & BDRV_O_NOCACHE) != 0) {
606         s->needs_alignment = true;
607     }
608
609     if (fstat(s->fd, &st) < 0) {
610         ret = -errno;
611         error_setg_errno(errp, errno, "Could not stat file");
612         goto fail;
613     }
614
615     if (!device) {
616         if (S_ISBLK(st.st_mode)) {
617             warn_report("Opening a block device as a file using the '%s' "
618                         "driver is deprecated", bs->drv->format_name);
619         } else if (S_ISCHR(st.st_mode)) {
620             warn_report("Opening a character device as a file using the '%s' "
621                         "driver is deprecated", bs->drv->format_name);
622         } else if (!S_ISREG(st.st_mode)) {
623             error_setg(errp, "A regular file was expected by the '%s' driver, "
624                        "but something else was given", bs->drv->format_name);
625             ret = -EINVAL;
626             goto fail;
627         } else {
628             s->discard_zeroes = true;
629             s->has_fallocate = true;
630         }
631     } else {
632         if (!(S_ISCHR(st.st_mode) || S_ISBLK(st.st_mode))) {
633             error_setg(errp, "'%s' driver expects either "
634                        "a character or block device", bs->drv->format_name);
635             ret = -EINVAL;
636             goto fail;
637         }
638     }
639
640     if (S_ISBLK(st.st_mode)) {
641 #ifdef BLKDISCARDZEROES
642         unsigned int arg;
643         if (ioctl(s->fd, BLKDISCARDZEROES, &arg) == 0 && arg) {
644             s->discard_zeroes = true;
645         }
646 #endif
647 #ifdef __linux__
648         /* On Linux 3.10, BLKDISCARD leaves stale data in the page cache.  Do
649          * not rely on the contents of discarded blocks unless using O_DIRECT.
650          * Same for BLKZEROOUT.
651          */
652         if (!(bs->open_flags & BDRV_O_NOCACHE)) {
653             s->discard_zeroes = false;
654             s->has_write_zeroes = false;
655         }
656 #endif
657     }
658 #ifdef __FreeBSD__
659     if (S_ISCHR(st.st_mode)) {
660         /*
661          * The file is a char device (disk), which on FreeBSD isn't behind
662          * a pager, so force all requests to be aligned. This is needed
663          * so QEMU makes sure all IO operations on the device are aligned
664          * to sector size, or else FreeBSD will reject them with EINVAL.
665          */
666         s->needs_alignment = true;
667     }
668 #endif
669
670 #ifdef CONFIG_XFS
671     if (platform_test_xfs_fd(s->fd)) {
672         s->is_xfs = true;
673     }
674 #endif
675
676     bs->supported_zero_flags = BDRV_REQ_MAY_UNMAP | BDRV_REQ_NO_FALLBACK;
677     ret = 0;
678 fail:
679     if (filename && (bdrv_flags & BDRV_O_TEMPORARY)) {
680         unlink(filename);
681     }
682     qemu_opts_del(opts);
683     return ret;
684 }
685
686 static int raw_open(BlockDriverState *bs, QDict *options, int flags,
687                     Error **errp)
688 {
689     BDRVRawState *s = bs->opaque;
690
691     s->type = FTYPE_FILE;
692     return raw_open_common(bs, options, flags, 0, false, errp);
693 }
694
695 typedef enum {
696     RAW_PL_PREPARE,
697     RAW_PL_COMMIT,
698     RAW_PL_ABORT,
699 } RawPermLockOp;
700
701 #define PERM_FOREACH(i) \
702     for ((i) = 0; (1ULL << (i)) <= BLK_PERM_ALL; i++)
703
704 /* Lock bytes indicated by @perm_lock_bits and @shared_perm_lock_bits in the
705  * file; if @unlock == true, also unlock the unneeded bytes.
706  * @shared_perm_lock_bits is the mask of all permissions that are NOT shared.
707  */
708 static int raw_apply_lock_bytes(BDRVRawState *s, int fd,
709                                 uint64_t perm_lock_bits,
710                                 uint64_t shared_perm_lock_bits,
711                                 bool unlock, Error **errp)
712 {
713     int ret;
714     int i;
715     uint64_t locked_perm, locked_shared_perm;
716
717     if (s) {
718         locked_perm = s->locked_perm;
719         locked_shared_perm = s->locked_shared_perm;
720     } else {
721         /*
722          * We don't have the previous bits, just lock/unlock for each of the
723          * requested bits.
724          */
725         if (unlock) {
726             locked_perm = BLK_PERM_ALL;
727             locked_shared_perm = BLK_PERM_ALL;
728         } else {
729             locked_perm = 0;
730             locked_shared_perm = 0;
731         }
732     }
733
734     PERM_FOREACH(i) {
735         int off = RAW_LOCK_PERM_BASE + i;
736         uint64_t bit = (1ULL << i);
737         if ((perm_lock_bits & bit) && !(locked_perm & bit)) {
738             ret = qemu_lock_fd(fd, off, 1, false);
739             if (ret) {
740                 error_setg(errp, "Failed to lock byte %d", off);
741                 return ret;
742             } else if (s) {
743                 s->locked_perm |= bit;
744             }
745         } else if (unlock && (locked_perm & bit) && !(perm_lock_bits & bit)) {
746             ret = qemu_unlock_fd(fd, off, 1);
747             if (ret) {
748                 error_setg(errp, "Failed to unlock byte %d", off);
749                 return ret;
750             } else if (s) {
751                 s->locked_perm &= ~bit;
752             }
753         }
754     }
755     PERM_FOREACH(i) {
756         int off = RAW_LOCK_SHARED_BASE + i;
757         uint64_t bit = (1ULL << i);
758         if ((shared_perm_lock_bits & bit) && !(locked_shared_perm & bit)) {
759             ret = qemu_lock_fd(fd, off, 1, false);
760             if (ret) {
761                 error_setg(errp, "Failed to lock byte %d", off);
762                 return ret;
763             } else if (s) {
764                 s->locked_shared_perm |= bit;
765             }
766         } else if (unlock && (locked_shared_perm & bit) &&
767                    !(shared_perm_lock_bits & bit)) {
768             ret = qemu_unlock_fd(fd, off, 1);
769             if (ret) {
770                 error_setg(errp, "Failed to unlock byte %d", off);
771                 return ret;
772             } else if (s) {
773                 s->locked_shared_perm &= ~bit;
774             }
775         }
776     }
777     return 0;
778 }
779
780 /* Check "unshared" bytes implied by @perm and ~@shared_perm in the file. */
781 static int raw_check_lock_bytes(int fd, uint64_t perm, uint64_t shared_perm,
782                                 Error **errp)
783 {
784     int ret;
785     int i;
786
787     PERM_FOREACH(i) {
788         int off = RAW_LOCK_SHARED_BASE + i;
789         uint64_t p = 1ULL << i;
790         if (perm & p) {
791             ret = qemu_lock_fd_test(fd, off, 1, true);
792             if (ret) {
793                 char *perm_name = bdrv_perm_names(p);
794                 error_setg(errp,
795                            "Failed to get \"%s\" lock",
796                            perm_name);
797                 g_free(perm_name);
798                 return ret;
799             }
800         }
801     }
802     PERM_FOREACH(i) {
803         int off = RAW_LOCK_PERM_BASE + i;
804         uint64_t p = 1ULL << i;
805         if (!(shared_perm & p)) {
806             ret = qemu_lock_fd_test(fd, off, 1, true);
807             if (ret) {
808                 char *perm_name = bdrv_perm_names(p);
809                 error_setg(errp,
810                            "Failed to get shared \"%s\" lock",
811                            perm_name);
812                 g_free(perm_name);
813                 return ret;
814             }
815         }
816     }
817     return 0;
818 }
819
820 static int raw_handle_perm_lock(BlockDriverState *bs,
821                                 RawPermLockOp op,
822                                 uint64_t new_perm, uint64_t new_shared,
823                                 Error **errp)
824 {
825     BDRVRawState *s = bs->opaque;
826     int ret = 0;
827     Error *local_err = NULL;
828
829     if (!s->use_lock) {
830         return 0;
831     }
832
833     if (bdrv_get_flags(bs) & BDRV_O_INACTIVE) {
834         return 0;
835     }
836
837     switch (op) {
838     case RAW_PL_PREPARE:
839         if ((s->perm | new_perm) == s->perm &&
840             (s->shared_perm & new_shared) == s->shared_perm)
841         {
842             /*
843              * We are going to unlock bytes, it should not fail. If it fail due
844              * to some fs-dependent permission-unrelated reasons (which occurs
845              * sometimes on NFS and leads to abort in bdrv_replace_child) we
846              * can't prevent such errors by any check here. And we ignore them
847              * anyway in ABORT and COMMIT.
848              */
849             return 0;
850         }
851         ret = raw_apply_lock_bytes(s, s->fd, s->perm | new_perm,
852                                    ~s->shared_perm | ~new_shared,
853                                    false, errp);
854         if (!ret) {
855             ret = raw_check_lock_bytes(s->fd, new_perm, new_shared, errp);
856             if (!ret) {
857                 return 0;
858             }
859             error_append_hint(errp,
860                               "Is another process using the image [%s]?\n",
861                               bs->filename);
862         }
863         op = RAW_PL_ABORT;
864         /* fall through to unlock bytes. */
865     case RAW_PL_ABORT:
866         raw_apply_lock_bytes(s, s->fd, s->perm, ~s->shared_perm,
867                              true, &local_err);
868         if (local_err) {
869             /* Theoretically the above call only unlocks bytes and it cannot
870              * fail. Something weird happened, report it.
871              */
872             warn_report_err(local_err);
873         }
874         break;
875     case RAW_PL_COMMIT:
876         raw_apply_lock_bytes(s, s->fd, new_perm, ~new_shared,
877                              true, &local_err);
878         if (local_err) {
879             /* Theoretically the above call only unlocks bytes and it cannot
880              * fail. Something weird happened, report it.
881              */
882             warn_report_err(local_err);
883         }
884         break;
885     }
886     return ret;
887 }
888
889 static int raw_reconfigure_getfd(BlockDriverState *bs, int flags,
890                                  int *open_flags, uint64_t perm, bool force_dup,
891                                  Error **errp)
892 {
893     BDRVRawState *s = bs->opaque;
894     int fd = -1;
895     int ret;
896     bool has_writers = perm &
897         (BLK_PERM_WRITE | BLK_PERM_WRITE_UNCHANGED | BLK_PERM_RESIZE);
898     int fcntl_flags = O_APPEND | O_NONBLOCK;
899 #ifdef O_NOATIME
900     fcntl_flags |= O_NOATIME;
901 #endif
902
903     *open_flags = 0;
904     if (s->type == FTYPE_CD) {
905         *open_flags |= O_NONBLOCK;
906     }
907
908     raw_parse_flags(flags, open_flags, has_writers);
909
910 #ifdef O_ASYNC
911     /* Not all operating systems have O_ASYNC, and those that don't
912      * will not let us track the state into rs->open_flags (typically
913      * you achieve the same effect with an ioctl, for example I_SETSIG
914      * on Solaris). But we do not use O_ASYNC, so that's fine.
915      */
916     assert((s->open_flags & O_ASYNC) == 0);
917 #endif
918
919     if (!force_dup && *open_flags == s->open_flags) {
920         /* We're lucky, the existing fd is fine */
921         return s->fd;
922     }
923
924     if ((*open_flags & ~fcntl_flags) == (s->open_flags & ~fcntl_flags)) {
925         /* dup the original fd */
926         fd = qemu_dup(s->fd);
927         if (fd >= 0) {
928             ret = fcntl_setfl(fd, *open_flags);
929             if (ret) {
930                 qemu_close(fd);
931                 fd = -1;
932             }
933         }
934     }
935
936     /* If we cannot use fcntl, or fcntl failed, fall back to qemu_open() */
937     if (fd == -1) {
938         const char *normalized_filename = bs->filename;
939         ret = raw_normalize_devicepath(&normalized_filename, errp);
940         if (ret >= 0) {
941             assert(!(*open_flags & O_CREAT));
942             fd = qemu_open(normalized_filename, *open_flags);
943             if (fd == -1) {
944                 error_setg_errno(errp, errno, "Could not reopen file");
945                 return -1;
946             }
947         }
948     }
949
950     return fd;
951 }
952
953 static int raw_reopen_prepare(BDRVReopenState *state,
954                               BlockReopenQueue *queue, Error **errp)
955 {
956     BDRVRawState *s;
957     BDRVRawReopenState *rs;
958     QemuOpts *opts;
959     int ret;
960     Error *local_err = NULL;
961
962     assert(state != NULL);
963     assert(state->bs != NULL);
964
965     s = state->bs->opaque;
966
967     state->opaque = g_new0(BDRVRawReopenState, 1);
968     rs = state->opaque;
969
970     /* Handle options changes */
971     opts = qemu_opts_create(&raw_runtime_opts, NULL, 0, &error_abort);
972     qemu_opts_absorb_qdict(opts, state->options, &local_err);
973     if (local_err) {
974         error_propagate(errp, local_err);
975         ret = -EINVAL;
976         goto out;
977     }
978
979     rs->drop_cache = qemu_opt_get_bool_del(opts, "drop-cache", true);
980     rs->check_cache_dropped =
981         qemu_opt_get_bool_del(opts, "x-check-cache-dropped", false);
982
983     /* This driver's reopen function doesn't currently allow changing
984      * other options, so let's put them back in the original QDict and
985      * bdrv_reopen_prepare() will detect changes and complain. */
986     qemu_opts_to_qdict(opts, state->options);
987
988     rs->fd = raw_reconfigure_getfd(state->bs, state->flags, &rs->open_flags,
989                                    state->perm, true, &local_err);
990     if (local_err) {
991         error_propagate(errp, local_err);
992         ret = -1;
993         goto out;
994     }
995
996     /* Fail already reopen_prepare() if we can't get a working O_DIRECT
997      * alignment with the new fd. */
998     if (rs->fd != -1) {
999         raw_probe_alignment(state->bs, rs->fd, &local_err);
1000         if (local_err) {
1001             error_propagate(errp, local_err);
1002             ret = -EINVAL;
1003             goto out_fd;
1004         }
1005     }
1006
1007     s->reopen_state = state;
1008     ret = 0;
1009 out_fd:
1010     if (ret < 0) {
1011         qemu_close(rs->fd);
1012         rs->fd = -1;
1013     }
1014 out:
1015     qemu_opts_del(opts);
1016     return ret;
1017 }
1018
1019 static void raw_reopen_commit(BDRVReopenState *state)
1020 {
1021     BDRVRawReopenState *rs = state->opaque;
1022     BDRVRawState *s = state->bs->opaque;
1023
1024     s->drop_cache = rs->drop_cache;
1025     s->check_cache_dropped = rs->check_cache_dropped;
1026     s->open_flags = rs->open_flags;
1027
1028     qemu_close(s->fd);
1029     s->fd = rs->fd;
1030
1031     g_free(state->opaque);
1032     state->opaque = NULL;
1033
1034     assert(s->reopen_state == state);
1035     s->reopen_state = NULL;
1036 }
1037
1038
1039 static void raw_reopen_abort(BDRVReopenState *state)
1040 {
1041     BDRVRawReopenState *rs = state->opaque;
1042     BDRVRawState *s = state->bs->opaque;
1043
1044      /* nothing to do if NULL, we didn't get far enough */
1045     if (rs == NULL) {
1046         return;
1047     }
1048
1049     if (rs->fd >= 0) {
1050         qemu_close(rs->fd);
1051         rs->fd = -1;
1052     }
1053     g_free(state->opaque);
1054     state->opaque = NULL;
1055
1056     assert(s->reopen_state == state);
1057     s->reopen_state = NULL;
1058 }
1059
1060 static int sg_get_max_transfer_length(int fd)
1061 {
1062 #ifdef BLKSECTGET
1063     int max_bytes = 0;
1064
1065     if (ioctl(fd, BLKSECTGET, &max_bytes) == 0) {
1066         return max_bytes;
1067     } else {
1068         return -errno;
1069     }
1070 #else
1071     return -ENOSYS;
1072 #endif
1073 }
1074
1075 static int sg_get_max_segments(int fd)
1076 {
1077 #ifdef CONFIG_LINUX
1078     char buf[32];
1079     const char *end;
1080     char *sysfspath = NULL;
1081     int ret;
1082     int sysfd = -1;
1083     long max_segments;
1084     struct stat st;
1085
1086     if (fstat(fd, &st)) {
1087         ret = -errno;
1088         goto out;
1089     }
1090
1091     sysfspath = g_strdup_printf("/sys/dev/block/%u:%u/queue/max_segments",
1092                                 major(st.st_rdev), minor(st.st_rdev));
1093     sysfd = open(sysfspath, O_RDONLY);
1094     if (sysfd == -1) {
1095         ret = -errno;
1096         goto out;
1097     }
1098     do {
1099         ret = read(sysfd, buf, sizeof(buf) - 1);
1100     } while (ret == -1 && errno == EINTR);
1101     if (ret < 0) {
1102         ret = -errno;
1103         goto out;
1104     } else if (ret == 0) {
1105         ret = -EIO;
1106         goto out;
1107     }
1108     buf[ret] = 0;
1109     /* The file is ended with '\n', pass 'end' to accept that. */
1110     ret = qemu_strtol(buf, &end, 10, &max_segments);
1111     if (ret == 0 && end && *end == '\n') {
1112         ret = max_segments;
1113     }
1114
1115 out:
1116     if (sysfd != -1) {
1117         close(sysfd);
1118     }
1119     g_free(sysfspath);
1120     return ret;
1121 #else
1122     return -ENOTSUP;
1123 #endif
1124 }
1125
1126 static void raw_refresh_limits(BlockDriverState *bs, Error **errp)
1127 {
1128     BDRVRawState *s = bs->opaque;
1129
1130     if (bs->sg) {
1131         int ret = sg_get_max_transfer_length(s->fd);
1132
1133         if (ret > 0 && ret <= BDRV_REQUEST_MAX_BYTES) {
1134             bs->bl.max_transfer = pow2floor(ret);
1135         }
1136
1137         ret = sg_get_max_segments(s->fd);
1138         if (ret > 0) {
1139             bs->bl.max_transfer = MIN(bs->bl.max_transfer, ret * getpagesize());
1140         }
1141     }
1142
1143     raw_probe_alignment(bs, s->fd, errp);
1144     bs->bl.min_mem_alignment = s->buf_align;
1145     bs->bl.opt_mem_alignment = MAX(s->buf_align, getpagesize());
1146 }
1147
1148 static int check_for_dasd(int fd)
1149 {
1150 #ifdef BIODASDINFO2
1151     struct dasd_information2_t info = {0};
1152
1153     return ioctl(fd, BIODASDINFO2, &info);
1154 #else
1155     return -1;
1156 #endif
1157 }
1158
1159 /**
1160  * Try to get @bs's logical and physical block size.
1161  * On success, store them in @bsz and return zero.
1162  * On failure, return negative errno.
1163  */
1164 static int hdev_probe_blocksizes(BlockDriverState *bs, BlockSizes *bsz)
1165 {
1166     BDRVRawState *s = bs->opaque;
1167     int ret;
1168
1169     /* If DASD, get blocksizes */
1170     if (check_for_dasd(s->fd) < 0) {
1171         return -ENOTSUP;
1172     }
1173     ret = probe_logical_blocksize(s->fd, &bsz->log);
1174     if (ret < 0) {
1175         return ret;
1176     }
1177     return probe_physical_blocksize(s->fd, &bsz->phys);
1178 }
1179
1180 /**
1181  * Try to get @bs's geometry: cyls, heads, sectors.
1182  * On success, store them in @geo and return 0.
1183  * On failure return -errno.
1184  * (Allows block driver to assign default geometry values that guest sees)
1185  */
1186 #ifdef __linux__
1187 static int hdev_probe_geometry(BlockDriverState *bs, HDGeometry *geo)
1188 {
1189     BDRVRawState *s = bs->opaque;
1190     struct hd_geometry ioctl_geo = {0};
1191
1192     /* If DASD, get its geometry */
1193     if (check_for_dasd(s->fd) < 0) {
1194         return -ENOTSUP;
1195     }
1196     if (ioctl(s->fd, HDIO_GETGEO, &ioctl_geo) < 0) {
1197         return -errno;
1198     }
1199     /* HDIO_GETGEO may return success even though geo contains zeros
1200        (e.g. certain multipath setups) */
1201     if (!ioctl_geo.heads || !ioctl_geo.sectors || !ioctl_geo.cylinders) {
1202         return -ENOTSUP;
1203     }
1204     /* Do not return a geometry for partition */
1205     if (ioctl_geo.start != 0) {
1206         return -ENOTSUP;
1207     }
1208     geo->heads = ioctl_geo.heads;
1209     geo->sectors = ioctl_geo.sectors;
1210     geo->cylinders = ioctl_geo.cylinders;
1211
1212     return 0;
1213 }
1214 #else /* __linux__ */
1215 static int hdev_probe_geometry(BlockDriverState *bs, HDGeometry *geo)
1216 {
1217     return -ENOTSUP;
1218 }
1219 #endif
1220
1221 #if defined(__linux__)
1222 static int handle_aiocb_ioctl(void *opaque)
1223 {
1224     RawPosixAIOData *aiocb = opaque;
1225     int ret;
1226
1227     ret = ioctl(aiocb->aio_fildes, aiocb->ioctl.cmd, aiocb->ioctl.buf);
1228     if (ret == -1) {
1229         return -errno;
1230     }
1231
1232     return 0;
1233 }
1234 #endif /* linux */
1235
1236 static int handle_aiocb_flush(void *opaque)
1237 {
1238     RawPosixAIOData *aiocb = opaque;
1239     BDRVRawState *s = aiocb->bs->opaque;
1240     int ret;
1241
1242     if (s->page_cache_inconsistent) {
1243         return -EIO;
1244     }
1245
1246     ret = qemu_fdatasync(aiocb->aio_fildes);
1247     if (ret == -1) {
1248         /* There is no clear definition of the semantics of a failing fsync(),
1249          * so we may have to assume the worst. The sad truth is that this
1250          * assumption is correct for Linux. Some pages are now probably marked
1251          * clean in the page cache even though they are inconsistent with the
1252          * on-disk contents. The next fdatasync() call would succeed, but no
1253          * further writeback attempt will be made. We can't get back to a state
1254          * in which we know what is on disk (we would have to rewrite
1255          * everything that was touched since the last fdatasync() at least), so
1256          * make bdrv_flush() fail permanently. Given that the behaviour isn't
1257          * really defined, I have little hope that other OSes are doing better.
1258          *
1259          * Obviously, this doesn't affect O_DIRECT, which bypasses the page
1260          * cache. */
1261         if ((s->open_flags & O_DIRECT) == 0) {
1262             s->page_cache_inconsistent = true;
1263         }
1264         return -errno;
1265     }
1266     return 0;
1267 }
1268
1269 #ifdef CONFIG_PREADV
1270
1271 static bool preadv_present = true;
1272
1273 static ssize_t
1274 qemu_preadv(int fd, const struct iovec *iov, int nr_iov, off_t offset)
1275 {
1276     return preadv(fd, iov, nr_iov, offset);
1277 }
1278
1279 static ssize_t
1280 qemu_pwritev(int fd, const struct iovec *iov, int nr_iov, off_t offset)
1281 {
1282     return pwritev(fd, iov, nr_iov, offset);
1283 }
1284
1285 #else
1286
1287 static bool preadv_present = false;
1288
1289 static ssize_t
1290 qemu_preadv(int fd, const struct iovec *iov, int nr_iov, off_t offset)
1291 {
1292     return -ENOSYS;
1293 }
1294
1295 static ssize_t
1296 qemu_pwritev(int fd, const struct iovec *iov, int nr_iov, off_t offset)
1297 {
1298     return -ENOSYS;
1299 }
1300
1301 #endif
1302
1303 static ssize_t handle_aiocb_rw_vector(RawPosixAIOData *aiocb)
1304 {
1305     ssize_t len;
1306
1307     do {
1308         if (aiocb->aio_type & QEMU_AIO_WRITE)
1309             len = qemu_pwritev(aiocb->aio_fildes,
1310                                aiocb->io.iov,
1311                                aiocb->io.niov,
1312                                aiocb->aio_offset);
1313          else
1314             len = qemu_preadv(aiocb->aio_fildes,
1315                               aiocb->io.iov,
1316                               aiocb->io.niov,
1317                               aiocb->aio_offset);
1318     } while (len == -1 && errno == EINTR);
1319
1320     if (len == -1) {
1321         return -errno;
1322     }
1323     return len;
1324 }
1325
1326 /*
1327  * Read/writes the data to/from a given linear buffer.
1328  *
1329  * Returns the number of bytes handles or -errno in case of an error. Short
1330  * reads are only returned if the end of the file is reached.
1331  */
1332 static ssize_t handle_aiocb_rw_linear(RawPosixAIOData *aiocb, char *buf)
1333 {
1334     ssize_t offset = 0;
1335     ssize_t len;
1336
1337     while (offset < aiocb->aio_nbytes) {
1338         if (aiocb->aio_type & QEMU_AIO_WRITE) {
1339             len = pwrite(aiocb->aio_fildes,
1340                          (const char *)buf + offset,
1341                          aiocb->aio_nbytes - offset,
1342                          aiocb->aio_offset + offset);
1343         } else {
1344             len = pread(aiocb->aio_fildes,
1345                         buf + offset,
1346                         aiocb->aio_nbytes - offset,
1347                         aiocb->aio_offset + offset);
1348         }
1349         if (len == -1 && errno == EINTR) {
1350             continue;
1351         } else if (len == -1 && errno == EINVAL &&
1352                    (aiocb->bs->open_flags & BDRV_O_NOCACHE) &&
1353                    !(aiocb->aio_type & QEMU_AIO_WRITE) &&
1354                    offset > 0) {
1355             /* O_DIRECT pread() may fail with EINVAL when offset is unaligned
1356              * after a short read.  Assume that O_DIRECT short reads only occur
1357              * at EOF.  Therefore this is a short read, not an I/O error.
1358              */
1359             break;
1360         } else if (len == -1) {
1361             offset = -errno;
1362             break;
1363         } else if (len == 0) {
1364             break;
1365         }
1366         offset += len;
1367     }
1368
1369     return offset;
1370 }
1371
1372 static int handle_aiocb_rw(void *opaque)
1373 {
1374     RawPosixAIOData *aiocb = opaque;
1375     ssize_t nbytes;
1376     char *buf;
1377
1378     if (!(aiocb->aio_type & QEMU_AIO_MISALIGNED)) {
1379         /*
1380          * If there is just a single buffer, and it is properly aligned
1381          * we can just use plain pread/pwrite without any problems.
1382          */
1383         if (aiocb->io.niov == 1) {
1384             nbytes = handle_aiocb_rw_linear(aiocb, aiocb->io.iov->iov_base);
1385             goto out;
1386         }
1387         /*
1388          * We have more than one iovec, and all are properly aligned.
1389          *
1390          * Try preadv/pwritev first and fall back to linearizing the
1391          * buffer if it's not supported.
1392          */
1393         if (preadv_present) {
1394             nbytes = handle_aiocb_rw_vector(aiocb);
1395             if (nbytes == aiocb->aio_nbytes ||
1396                 (nbytes < 0 && nbytes != -ENOSYS)) {
1397                 goto out;
1398             }
1399             preadv_present = false;
1400         }
1401
1402         /*
1403          * XXX(hch): short read/write.  no easy way to handle the reminder
1404          * using these interfaces.  For now retry using plain
1405          * pread/pwrite?
1406          */
1407     }
1408
1409     /*
1410      * Ok, we have to do it the hard way, copy all segments into
1411      * a single aligned buffer.
1412      */
1413     buf = qemu_try_blockalign(aiocb->bs, aiocb->aio_nbytes);
1414     if (buf == NULL) {
1415         nbytes = -ENOMEM;
1416         goto out;
1417     }
1418
1419     if (aiocb->aio_type & QEMU_AIO_WRITE) {
1420         char *p = buf;
1421         int i;
1422
1423         for (i = 0; i < aiocb->io.niov; ++i) {
1424             memcpy(p, aiocb->io.iov[i].iov_base, aiocb->io.iov[i].iov_len);
1425             p += aiocb->io.iov[i].iov_len;
1426         }
1427         assert(p - buf == aiocb->aio_nbytes);
1428     }
1429
1430     nbytes = handle_aiocb_rw_linear(aiocb, buf);
1431     if (!(aiocb->aio_type & QEMU_AIO_WRITE)) {
1432         char *p = buf;
1433         size_t count = aiocb->aio_nbytes, copy;
1434         int i;
1435
1436         for (i = 0; i < aiocb->io.niov && count; ++i) {
1437             copy = count;
1438             if (copy > aiocb->io.iov[i].iov_len) {
1439                 copy = aiocb->io.iov[i].iov_len;
1440             }
1441             memcpy(aiocb->io.iov[i].iov_base, p, copy);
1442             assert(count >= copy);
1443             p     += copy;
1444             count -= copy;
1445         }
1446         assert(count == 0);
1447     }
1448     qemu_vfree(buf);
1449
1450 out:
1451     if (nbytes == aiocb->aio_nbytes) {
1452         return 0;
1453     } else if (nbytes >= 0 && nbytes < aiocb->aio_nbytes) {
1454         if (aiocb->aio_type & QEMU_AIO_WRITE) {
1455             return -EINVAL;
1456         } else {
1457             iov_memset(aiocb->io.iov, aiocb->io.niov, nbytes,
1458                       0, aiocb->aio_nbytes - nbytes);
1459             return 0;
1460         }
1461     } else {
1462         assert(nbytes < 0);
1463         return nbytes;
1464     }
1465 }
1466
1467 static int translate_err(int err)
1468 {
1469     if (err == -ENODEV || err == -ENOSYS || err == -EOPNOTSUPP ||
1470         err == -ENOTTY) {
1471         err = -ENOTSUP;
1472     }
1473     return err;
1474 }
1475
1476 #ifdef CONFIG_FALLOCATE
1477 static int do_fallocate(int fd, int mode, off_t offset, off_t len)
1478 {
1479     do {
1480         if (fallocate(fd, mode, offset, len) == 0) {
1481             return 0;
1482         }
1483     } while (errno == EINTR);
1484     return translate_err(-errno);
1485 }
1486 #endif
1487
1488 static ssize_t handle_aiocb_write_zeroes_block(RawPosixAIOData *aiocb)
1489 {
1490     int ret = -ENOTSUP;
1491     BDRVRawState *s = aiocb->bs->opaque;
1492
1493     if (!s->has_write_zeroes) {
1494         return -ENOTSUP;
1495     }
1496
1497 #ifdef BLKZEROOUT
1498     /* The BLKZEROOUT implementation in the kernel doesn't set
1499      * BLKDEV_ZERO_NOFALLBACK, so we can't call this if we have to avoid slow
1500      * fallbacks. */
1501     if (!(aiocb->aio_type & QEMU_AIO_NO_FALLBACK)) {
1502         do {
1503             uint64_t range[2] = { aiocb->aio_offset, aiocb->aio_nbytes };
1504             if (ioctl(aiocb->aio_fildes, BLKZEROOUT, range) == 0) {
1505                 return 0;
1506             }
1507         } while (errno == EINTR);
1508
1509         ret = translate_err(-errno);
1510         if (ret == -ENOTSUP) {
1511             s->has_write_zeroes = false;
1512         }
1513     }
1514 #endif
1515
1516     return ret;
1517 }
1518
1519 static int handle_aiocb_write_zeroes(void *opaque)
1520 {
1521     RawPosixAIOData *aiocb = opaque;
1522 #ifdef CONFIG_FALLOCATE
1523     BDRVRawState *s = aiocb->bs->opaque;
1524     int64_t len;
1525 #endif
1526
1527     if (aiocb->aio_type & QEMU_AIO_BLKDEV) {
1528         return handle_aiocb_write_zeroes_block(aiocb);
1529     }
1530
1531 #ifdef CONFIG_FALLOCATE_ZERO_RANGE
1532     if (s->has_write_zeroes) {
1533         int ret = do_fallocate(s->fd, FALLOC_FL_ZERO_RANGE,
1534                                aiocb->aio_offset, aiocb->aio_nbytes);
1535         if (ret == -EINVAL) {
1536             /*
1537              * Allow falling back to pwrite for file systems that
1538              * do not support fallocate() for an unaligned byte range.
1539              */
1540             return -ENOTSUP;
1541         }
1542         if (ret == 0 || ret != -ENOTSUP) {
1543             return ret;
1544         }
1545         s->has_write_zeroes = false;
1546     }
1547 #endif
1548
1549 #ifdef CONFIG_FALLOCATE_PUNCH_HOLE
1550     if (s->has_discard && s->has_fallocate) {
1551         int ret = do_fallocate(s->fd,
1552                                FALLOC_FL_PUNCH_HOLE | FALLOC_FL_KEEP_SIZE,
1553                                aiocb->aio_offset, aiocb->aio_nbytes);
1554         if (ret == 0) {
1555             ret = do_fallocate(s->fd, 0, aiocb->aio_offset, aiocb->aio_nbytes);
1556             if (ret == 0 || ret != -ENOTSUP) {
1557                 return ret;
1558             }
1559             s->has_fallocate = false;
1560         } else if (ret != -ENOTSUP) {
1561             return ret;
1562         } else {
1563             s->has_discard = false;
1564         }
1565     }
1566 #endif
1567
1568 #ifdef CONFIG_FALLOCATE
1569     /* Last resort: we are trying to extend the file with zeroed data. This
1570      * can be done via fallocate(fd, 0) */
1571     len = bdrv_getlength(aiocb->bs);
1572     if (s->has_fallocate && len >= 0 && aiocb->aio_offset >= len) {
1573         int ret = do_fallocate(s->fd, 0, aiocb->aio_offset, aiocb->aio_nbytes);
1574         if (ret == 0 || ret != -ENOTSUP) {
1575             return ret;
1576         }
1577         s->has_fallocate = false;
1578     }
1579 #endif
1580
1581     return -ENOTSUP;
1582 }
1583
1584 static int handle_aiocb_write_zeroes_unmap(void *opaque)
1585 {
1586     RawPosixAIOData *aiocb = opaque;
1587     BDRVRawState *s G_GNUC_UNUSED = aiocb->bs->opaque;
1588     int ret;
1589
1590     /* First try to write zeros and unmap at the same time */
1591
1592 #ifdef CONFIG_FALLOCATE_PUNCH_HOLE
1593     ret = do_fallocate(s->fd, FALLOC_FL_PUNCH_HOLE | FALLOC_FL_KEEP_SIZE,
1594                        aiocb->aio_offset, aiocb->aio_nbytes);
1595     if (ret != -ENOTSUP) {
1596         return ret;
1597     }
1598 #endif
1599
1600     /* If we couldn't manage to unmap while guaranteed that the area reads as
1601      * all-zero afterwards, just write zeroes without unmapping */
1602     ret = handle_aiocb_write_zeroes(aiocb);
1603     return ret;
1604 }
1605
1606 #ifndef HAVE_COPY_FILE_RANGE
1607 static off_t copy_file_range(int in_fd, off_t *in_off, int out_fd,
1608                              off_t *out_off, size_t len, unsigned int flags)
1609 {
1610 #ifdef __NR_copy_file_range
1611     return syscall(__NR_copy_file_range, in_fd, in_off, out_fd,
1612                    out_off, len, flags);
1613 #else
1614     errno = ENOSYS;
1615     return -1;
1616 #endif
1617 }
1618 #endif
1619
1620 static int handle_aiocb_copy_range(void *opaque)
1621 {
1622     RawPosixAIOData *aiocb = opaque;
1623     uint64_t bytes = aiocb->aio_nbytes;
1624     off_t in_off = aiocb->aio_offset;
1625     off_t out_off = aiocb->copy_range.aio_offset2;
1626
1627     while (bytes) {
1628         ssize_t ret = copy_file_range(aiocb->aio_fildes, &in_off,
1629                                       aiocb->copy_range.aio_fd2, &out_off,
1630                                       bytes, 0);
1631         trace_file_copy_file_range(aiocb->bs, aiocb->aio_fildes, in_off,
1632                                    aiocb->copy_range.aio_fd2, out_off, bytes,
1633                                    0, ret);
1634         if (ret == 0) {
1635             /* No progress (e.g. when beyond EOF), let the caller fall back to
1636              * buffer I/O. */
1637             return -ENOSPC;
1638         }
1639         if (ret < 0) {
1640             switch (errno) {
1641             case ENOSYS:
1642                 return -ENOTSUP;
1643             case EINTR:
1644                 continue;
1645             default:
1646                 return -errno;
1647             }
1648         }
1649         bytes -= ret;
1650     }
1651     return 0;
1652 }
1653
1654 static int handle_aiocb_discard(void *opaque)
1655 {
1656     RawPosixAIOData *aiocb = opaque;
1657     int ret = -EOPNOTSUPP;
1658     BDRVRawState *s = aiocb->bs->opaque;
1659
1660     if (!s->has_discard) {
1661         return -ENOTSUP;
1662     }
1663
1664     if (aiocb->aio_type & QEMU_AIO_BLKDEV) {
1665 #ifdef BLKDISCARD
1666         do {
1667             uint64_t range[2] = { aiocb->aio_offset, aiocb->aio_nbytes };
1668             if (ioctl(aiocb->aio_fildes, BLKDISCARD, range) == 0) {
1669                 return 0;
1670             }
1671         } while (errno == EINTR);
1672
1673         ret = -errno;
1674 #endif
1675     } else {
1676 #ifdef CONFIG_FALLOCATE_PUNCH_HOLE
1677         ret = do_fallocate(s->fd, FALLOC_FL_PUNCH_HOLE | FALLOC_FL_KEEP_SIZE,
1678                            aiocb->aio_offset, aiocb->aio_nbytes);
1679 #endif
1680     }
1681
1682     ret = translate_err(ret);
1683     if (ret == -ENOTSUP) {
1684         s->has_discard = false;
1685     }
1686     return ret;
1687 }
1688
1689 /*
1690  * Help alignment probing by allocating the first block.
1691  *
1692  * When reading with direct I/O from unallocated area on Gluster backed by XFS,
1693  * reading succeeds regardless of request length. In this case we fallback to
1694  * safe alignment which is not optimal. Allocating the first block avoids this
1695  * fallback.
1696  *
1697  * fd may be opened with O_DIRECT, but we don't know the buffer alignment or
1698  * request alignment, so we use safe values.
1699  *
1700  * Returns: 0 on success, -errno on failure. Since this is an optimization,
1701  * caller may ignore failures.
1702  */
1703 static int allocate_first_block(int fd, size_t max_size)
1704 {
1705     size_t write_size = (max_size < MAX_BLOCKSIZE)
1706         ? BDRV_SECTOR_SIZE
1707         : MAX_BLOCKSIZE;
1708     size_t max_align = MAX(MAX_BLOCKSIZE, getpagesize());
1709     void *buf;
1710     ssize_t n;
1711     int ret;
1712
1713     buf = qemu_memalign(max_align, write_size);
1714     memset(buf, 0, write_size);
1715
1716     do {
1717         n = pwrite(fd, buf, write_size, 0);
1718     } while (n == -1 && errno == EINTR);
1719
1720     ret = (n == -1) ? -errno : 0;
1721
1722     qemu_vfree(buf);
1723     return ret;
1724 }
1725
1726 static int handle_aiocb_truncate(void *opaque)
1727 {
1728     RawPosixAIOData *aiocb = opaque;
1729     int result = 0;
1730     int64_t current_length = 0;
1731     char *buf = NULL;
1732     struct stat st;
1733     int fd = aiocb->aio_fildes;
1734     int64_t offset = aiocb->aio_offset;
1735     PreallocMode prealloc = aiocb->truncate.prealloc;
1736     Error **errp = aiocb->truncate.errp;
1737
1738     if (fstat(fd, &st) < 0) {
1739         result = -errno;
1740         error_setg_errno(errp, -result, "Could not stat file");
1741         return result;
1742     }
1743
1744     current_length = st.st_size;
1745     if (current_length > offset && prealloc != PREALLOC_MODE_OFF) {
1746         error_setg(errp, "Cannot use preallocation for shrinking files");
1747         return -ENOTSUP;
1748     }
1749
1750     switch (prealloc) {
1751 #ifdef CONFIG_POSIX_FALLOCATE
1752     case PREALLOC_MODE_FALLOC:
1753         /*
1754          * Truncating before posix_fallocate() makes it about twice slower on
1755          * file systems that do not support fallocate(), trying to check if a
1756          * block is allocated before allocating it, so don't do that here.
1757          */
1758         if (offset != current_length) {
1759             result = -posix_fallocate(fd, current_length,
1760                                       offset - current_length);
1761             if (result != 0) {
1762                 /* posix_fallocate() doesn't set errno. */
1763                 error_setg_errno(errp, -result,
1764                                  "Could not preallocate new data");
1765             } else if (current_length == 0) {
1766                 /*
1767                  * posix_fallocate() uses fallocate() if the filesystem
1768                  * supports it, or fallback to manually writing zeroes. If
1769                  * fallocate() was used, unaligned reads from the fallocated
1770                  * area in raw_probe_alignment() will succeed, hence we need to
1771                  * allocate the first block.
1772                  *
1773                  * Optimize future alignment probing; ignore failures.
1774                  */
1775                 allocate_first_block(fd, offset);
1776             }
1777         } else {
1778             result = 0;
1779         }
1780         goto out;
1781 #endif
1782     case PREALLOC_MODE_FULL:
1783     {
1784         int64_t num = 0, left = offset - current_length;
1785         off_t seek_result;
1786
1787         /*
1788          * Knowing the final size from the beginning could allow the file
1789          * system driver to do less allocations and possibly avoid
1790          * fragmentation of the file.
1791          */
1792         if (ftruncate(fd, offset) != 0) {
1793             result = -errno;
1794             error_setg_errno(errp, -result, "Could not resize file");
1795             goto out;
1796         }
1797
1798         buf = g_malloc0(65536);
1799
1800         seek_result = lseek(fd, current_length, SEEK_SET);
1801         if (seek_result < 0) {
1802             result = -errno;
1803             error_setg_errno(errp, -result,
1804                              "Failed to seek to the old end of file");
1805             goto out;
1806         }
1807
1808         while (left > 0) {
1809             num = MIN(left, 65536);
1810             result = write(fd, buf, num);
1811             if (result < 0) {
1812                 if (errno == EINTR) {
1813                     continue;
1814                 }
1815                 result = -errno;
1816                 error_setg_errno(errp, -result,
1817                                  "Could not write zeros for preallocation");
1818                 goto out;
1819             }
1820             left -= result;
1821         }
1822         if (result >= 0) {
1823             result = fsync(fd);
1824             if (result < 0) {
1825                 result = -errno;
1826                 error_setg_errno(errp, -result,
1827                                  "Could not flush file to disk");
1828                 goto out;
1829             }
1830         }
1831         goto out;
1832     }
1833     case PREALLOC_MODE_OFF:
1834         if (ftruncate(fd, offset) != 0) {
1835             result = -errno;
1836             error_setg_errno(errp, -result, "Could not resize file");
1837         } else if (current_length == 0 && offset > current_length) {
1838             /* Optimize future alignment probing; ignore failures. */
1839             allocate_first_block(fd, offset);
1840         }
1841         return result;
1842     default:
1843         result = -ENOTSUP;
1844         error_setg(errp, "Unsupported preallocation mode: %s",
1845                    PreallocMode_str(prealloc));
1846         return result;
1847     }
1848
1849 out:
1850     if (result < 0) {
1851         if (ftruncate(fd, current_length) < 0) {
1852             error_report("Failed to restore old file length: %s",
1853                          strerror(errno));
1854         }
1855     }
1856
1857     g_free(buf);
1858     return result;
1859 }
1860
1861 static int coroutine_fn raw_thread_pool_submit(BlockDriverState *bs,
1862                                                ThreadPoolFunc func, void *arg)
1863 {
1864     /* @bs can be NULL, bdrv_get_aio_context() returns the main context then */
1865     ThreadPool *pool = aio_get_thread_pool(bdrv_get_aio_context(bs));
1866     return thread_pool_submit_co(pool, func, arg);
1867 }
1868
1869 static int coroutine_fn raw_co_prw(BlockDriverState *bs, uint64_t offset,
1870                                    uint64_t bytes, QEMUIOVector *qiov, int type)
1871 {
1872     BDRVRawState *s = bs->opaque;
1873     RawPosixAIOData acb;
1874
1875     if (fd_open(bs) < 0)
1876         return -EIO;
1877
1878     /*
1879      * Check if the underlying device requires requests to be aligned,
1880      * and if the request we are trying to submit is aligned or not.
1881      * If this is the case tell the low-level driver that it needs
1882      * to copy the buffer.
1883      */
1884     if (s->needs_alignment) {
1885         if (!bdrv_qiov_is_aligned(bs, qiov)) {
1886             type |= QEMU_AIO_MISALIGNED;
1887 #ifdef CONFIG_LINUX_AIO
1888         } else if (s->use_linux_aio) {
1889             LinuxAioState *aio = aio_get_linux_aio(bdrv_get_aio_context(bs));
1890             assert(qiov->size == bytes);
1891             return laio_co_submit(bs, aio, s->fd, offset, qiov, type);
1892 #endif
1893         }
1894     }
1895
1896     acb = (RawPosixAIOData) {
1897         .bs             = bs,
1898         .aio_fildes     = s->fd,
1899         .aio_type       = type,
1900         .aio_offset     = offset,
1901         .aio_nbytes     = bytes,
1902         .io             = {
1903             .iov            = qiov->iov,
1904             .niov           = qiov->niov,
1905         },
1906     };
1907
1908     assert(qiov->size == bytes);
1909     return raw_thread_pool_submit(bs, handle_aiocb_rw, &acb);
1910 }
1911
1912 static int coroutine_fn raw_co_preadv(BlockDriverState *bs, uint64_t offset,
1913                                       uint64_t bytes, QEMUIOVector *qiov,
1914                                       int flags)
1915 {
1916     return raw_co_prw(bs, offset, bytes, qiov, QEMU_AIO_READ);
1917 }
1918
1919 static int coroutine_fn raw_co_pwritev(BlockDriverState *bs, uint64_t offset,
1920                                        uint64_t bytes, QEMUIOVector *qiov,
1921                                        int flags)
1922 {
1923     assert(flags == 0);
1924     return raw_co_prw(bs, offset, bytes, qiov, QEMU_AIO_WRITE);
1925 }
1926
1927 static void raw_aio_plug(BlockDriverState *bs)
1928 {
1929 #ifdef CONFIG_LINUX_AIO
1930     BDRVRawState *s = bs->opaque;
1931     if (s->use_linux_aio) {
1932         LinuxAioState *aio = aio_get_linux_aio(bdrv_get_aio_context(bs));
1933         laio_io_plug(bs, aio);
1934     }
1935 #endif
1936 }
1937
1938 static void raw_aio_unplug(BlockDriverState *bs)
1939 {
1940 #ifdef CONFIG_LINUX_AIO
1941     BDRVRawState *s = bs->opaque;
1942     if (s->use_linux_aio) {
1943         LinuxAioState *aio = aio_get_linux_aio(bdrv_get_aio_context(bs));
1944         laio_io_unplug(bs, aio);
1945     }
1946 #endif
1947 }
1948
1949 static int raw_co_flush_to_disk(BlockDriverState *bs)
1950 {
1951     BDRVRawState *s = bs->opaque;
1952     RawPosixAIOData acb;
1953     int ret;
1954
1955     ret = fd_open(bs);
1956     if (ret < 0) {
1957         return ret;
1958     }
1959
1960     acb = (RawPosixAIOData) {
1961         .bs             = bs,
1962         .aio_fildes     = s->fd,
1963         .aio_type       = QEMU_AIO_FLUSH,
1964     };
1965
1966     return raw_thread_pool_submit(bs, handle_aiocb_flush, &acb);
1967 }
1968
1969 static void raw_aio_attach_aio_context(BlockDriverState *bs,
1970                                        AioContext *new_context)
1971 {
1972 #ifdef CONFIG_LINUX_AIO
1973     BDRVRawState *s = bs->opaque;
1974     if (s->use_linux_aio) {
1975         Error *local_err;
1976         if (!aio_setup_linux_aio(new_context, &local_err)) {
1977             error_reportf_err(local_err, "Unable to use native AIO, "
1978                                          "falling back to thread pool: ");
1979             s->use_linux_aio = false;
1980         }
1981     }
1982 #endif
1983 }
1984
1985 static void raw_close(BlockDriverState *bs)
1986 {
1987     BDRVRawState *s = bs->opaque;
1988
1989     if (s->fd >= 0) {
1990         qemu_close(s->fd);
1991         s->fd = -1;
1992     }
1993 }
1994
1995 /**
1996  * Truncates the given regular file @fd to @offset and, when growing, fills the
1997  * new space according to @prealloc.
1998  *
1999  * Returns: 0 on success, -errno on failure.
2000  */
2001 static int coroutine_fn
2002 raw_regular_truncate(BlockDriverState *bs, int fd, int64_t offset,
2003                      PreallocMode prealloc, Error **errp)
2004 {
2005     RawPosixAIOData acb;
2006
2007     acb = (RawPosixAIOData) {
2008         .bs             = bs,
2009         .aio_fildes     = fd,
2010         .aio_type       = QEMU_AIO_TRUNCATE,
2011         .aio_offset     = offset,
2012         .truncate       = {
2013             .prealloc       = prealloc,
2014             .errp           = errp,
2015         },
2016     };
2017
2018     return raw_thread_pool_submit(bs, handle_aiocb_truncate, &acb);
2019 }
2020
2021 static int coroutine_fn raw_co_truncate(BlockDriverState *bs, int64_t offset,
2022                                         PreallocMode prealloc, Error **errp)
2023 {
2024     BDRVRawState *s = bs->opaque;
2025     struct stat st;
2026     int ret;
2027
2028     if (fstat(s->fd, &st)) {
2029         ret = -errno;
2030         error_setg_errno(errp, -ret, "Failed to fstat() the file");
2031         return ret;
2032     }
2033
2034     if (S_ISREG(st.st_mode)) {
2035         return raw_regular_truncate(bs, s->fd, offset, prealloc, errp);
2036     }
2037
2038     if (prealloc != PREALLOC_MODE_OFF) {
2039         error_setg(errp, "Preallocation mode '%s' unsupported for this "
2040                    "non-regular file", PreallocMode_str(prealloc));
2041         return -ENOTSUP;
2042     }
2043
2044     if (S_ISCHR(st.st_mode) || S_ISBLK(st.st_mode)) {
2045         if (offset > raw_getlength(bs)) {
2046             error_setg(errp, "Cannot grow device files");
2047             return -EINVAL;
2048         }
2049     } else {
2050         error_setg(errp, "Resizing this file is not supported");
2051         return -ENOTSUP;
2052     }
2053
2054     return 0;
2055 }
2056
2057 #ifdef __OpenBSD__
2058 static int64_t raw_getlength(BlockDriverState *bs)
2059 {
2060     BDRVRawState *s = bs->opaque;
2061     int fd = s->fd;
2062     struct stat st;
2063
2064     if (fstat(fd, &st))
2065         return -errno;
2066     if (S_ISCHR(st.st_mode) || S_ISBLK(st.st_mode)) {
2067         struct disklabel dl;
2068
2069         if (ioctl(fd, DIOCGDINFO, &dl))
2070             return -errno;
2071         return (uint64_t)dl.d_secsize *
2072             dl.d_partitions[DISKPART(st.st_rdev)].p_size;
2073     } else
2074         return st.st_size;
2075 }
2076 #elif defined(__NetBSD__)
2077 static int64_t raw_getlength(BlockDriverState *bs)
2078 {
2079     BDRVRawState *s = bs->opaque;
2080     int fd = s->fd;
2081     struct stat st;
2082
2083     if (fstat(fd, &st))
2084         return -errno;
2085     if (S_ISCHR(st.st_mode) || S_ISBLK(st.st_mode)) {
2086         struct dkwedge_info dkw;
2087
2088         if (ioctl(fd, DIOCGWEDGEINFO, &dkw) != -1) {
2089             return dkw.dkw_size * 512;
2090         } else {
2091             struct disklabel dl;
2092
2093             if (ioctl(fd, DIOCGDINFO, &dl))
2094                 return -errno;
2095             return (uint64_t)dl.d_secsize *
2096                 dl.d_partitions[DISKPART(st.st_rdev)].p_size;
2097         }
2098     } else
2099         return st.st_size;
2100 }
2101 #elif defined(__sun__)
2102 static int64_t raw_getlength(BlockDriverState *bs)
2103 {
2104     BDRVRawState *s = bs->opaque;
2105     struct dk_minfo minfo;
2106     int ret;
2107     int64_t size;
2108
2109     ret = fd_open(bs);
2110     if (ret < 0) {
2111         return ret;
2112     }
2113
2114     /*
2115      * Use the DKIOCGMEDIAINFO ioctl to read the size.
2116      */
2117     ret = ioctl(s->fd, DKIOCGMEDIAINFO, &minfo);
2118     if (ret != -1) {
2119         return minfo.dki_lbsize * minfo.dki_capacity;
2120     }
2121
2122     /*
2123      * There are reports that lseek on some devices fails, but
2124      * irc discussion said that contingency on contingency was overkill.
2125      */
2126     size = lseek(s->fd, 0, SEEK_END);
2127     if (size < 0) {
2128         return -errno;
2129     }
2130     return size;
2131 }
2132 #elif defined(CONFIG_BSD)
2133 static int64_t raw_getlength(BlockDriverState *bs)
2134 {
2135     BDRVRawState *s = bs->opaque;
2136     int fd = s->fd;
2137     int64_t size;
2138     struct stat sb;
2139 #if defined (__FreeBSD__) || defined(__FreeBSD_kernel__)
2140     int reopened = 0;
2141 #endif
2142     int ret;
2143
2144     ret = fd_open(bs);
2145     if (ret < 0)
2146         return ret;
2147
2148 #if defined (__FreeBSD__) || defined(__FreeBSD_kernel__)
2149 again:
2150 #endif
2151     if (!fstat(fd, &sb) && (S_IFCHR & sb.st_mode)) {
2152 #ifdef DIOCGMEDIASIZE
2153         if (ioctl(fd, DIOCGMEDIASIZE, (off_t *)&size))
2154 #elif defined(DIOCGPART)
2155         {
2156                 struct partinfo pi;
2157                 if (ioctl(fd, DIOCGPART, &pi) == 0)
2158                         size = pi.media_size;
2159                 else
2160                         size = 0;
2161         }
2162         if (size == 0)
2163 #endif
2164 #if defined(__APPLE__) && defined(__MACH__)
2165         {
2166             uint64_t sectors = 0;
2167             uint32_t sector_size = 0;
2168
2169             if (ioctl(fd, DKIOCGETBLOCKCOUNT, &sectors) == 0
2170                && ioctl(fd, DKIOCGETBLOCKSIZE, &sector_size) == 0) {
2171                 size = sectors * sector_size;
2172             } else {
2173                 size = lseek(fd, 0LL, SEEK_END);
2174                 if (size < 0) {
2175                     return -errno;
2176                 }
2177             }
2178         }
2179 #else
2180         size = lseek(fd, 0LL, SEEK_END);
2181         if (size < 0) {
2182             return -errno;
2183         }
2184 #endif
2185 #if defined(__FreeBSD__) || defined(__FreeBSD_kernel__)
2186         switch(s->type) {
2187         case FTYPE_CD:
2188             /* XXX FreeBSD acd returns UINT_MAX sectors for an empty drive */
2189             if (size == 2048LL * (unsigned)-1)
2190                 size = 0;
2191             /* XXX no disc?  maybe we need to reopen... */
2192             if (size <= 0 && !reopened && cdrom_reopen(bs) >= 0) {
2193                 reopened = 1;
2194                 goto again;
2195             }
2196         }
2197 #endif
2198     } else {
2199         size = lseek(fd, 0, SEEK_END);
2200         if (size < 0) {
2201             return -errno;
2202         }
2203     }
2204     return size;
2205 }
2206 #else
2207 static int64_t raw_getlength(BlockDriverState *bs)
2208 {
2209     BDRVRawState *s = bs->opaque;
2210     int ret;
2211     int64_t size;
2212
2213     ret = fd_open(bs);
2214     if (ret < 0) {
2215         return ret;
2216     }
2217
2218     size = lseek(s->fd, 0, SEEK_END);
2219     if (size < 0) {
2220         return -errno;
2221     }
2222     return size;
2223 }
2224 #endif
2225
2226 static int64_t raw_get_allocated_file_size(BlockDriverState *bs)
2227 {
2228     struct stat st;
2229     BDRVRawState *s = bs->opaque;
2230
2231     if (fstat(s->fd, &st) < 0) {
2232         return -errno;
2233     }
2234     return (int64_t)st.st_blocks * 512;
2235 }
2236
2237 static int coroutine_fn
2238 raw_co_create(BlockdevCreateOptions *options, Error **errp)
2239 {
2240     BlockdevCreateOptionsFile *file_opts;
2241     Error *local_err = NULL;
2242     int fd;
2243     uint64_t perm, shared;
2244     int result = 0;
2245
2246     /* Validate options and set default values */
2247     assert(options->driver == BLOCKDEV_DRIVER_FILE);
2248     file_opts = &options->u.file;
2249
2250     if (!file_opts->has_nocow) {
2251         file_opts->nocow = false;
2252     }
2253     if (!file_opts->has_preallocation) {
2254         file_opts->preallocation = PREALLOC_MODE_OFF;
2255     }
2256
2257     /* Create file */
2258     fd = qemu_open(file_opts->filename, O_RDWR | O_CREAT | O_BINARY, 0644);
2259     if (fd < 0) {
2260         result = -errno;
2261         error_setg_errno(errp, -result, "Could not create file");
2262         goto out;
2263     }
2264
2265     /* Take permissions: We want to discard everything, so we need
2266      * BLK_PERM_WRITE; and truncation to the desired size requires
2267      * BLK_PERM_RESIZE.
2268      * On the other hand, we cannot share the RESIZE permission
2269      * because we promise that after this function, the file has the
2270      * size given in the options.  If someone else were to resize it
2271      * concurrently, we could not guarantee that.
2272      * Note that after this function, we can no longer guarantee that
2273      * the file is not touched by a third party, so it may be resized
2274      * then. */
2275     perm = BLK_PERM_WRITE | BLK_PERM_RESIZE;
2276     shared = BLK_PERM_ALL & ~BLK_PERM_RESIZE;
2277
2278     /* Step one: Take locks */
2279     result = raw_apply_lock_bytes(NULL, fd, perm, ~shared, false, errp);
2280     if (result < 0) {
2281         goto out_close;
2282     }
2283
2284     /* Step two: Check that nobody else has taken conflicting locks */
2285     result = raw_check_lock_bytes(fd, perm, shared, errp);
2286     if (result < 0) {
2287         error_append_hint(errp,
2288                           "Is another process using the image [%s]?\n",
2289                           file_opts->filename);
2290         goto out_unlock;
2291     }
2292
2293     /* Clear the file by truncating it to 0 */
2294     result = raw_regular_truncate(NULL, fd, 0, PREALLOC_MODE_OFF, errp);
2295     if (result < 0) {
2296         goto out_unlock;
2297     }
2298
2299     if (file_opts->nocow) {
2300 #ifdef __linux__
2301         /* Set NOCOW flag to solve performance issue on fs like btrfs.
2302          * This is an optimisation. The FS_IOC_SETFLAGS ioctl return value
2303          * will be ignored since any failure of this operation should not
2304          * block the left work.
2305          */
2306         int attr;
2307         if (ioctl(fd, FS_IOC_GETFLAGS, &attr) == 0) {
2308             attr |= FS_NOCOW_FL;
2309             ioctl(fd, FS_IOC_SETFLAGS, &attr);
2310         }
2311 #endif
2312     }
2313
2314     /* Resize and potentially preallocate the file to the desired
2315      * final size */
2316     result = raw_regular_truncate(NULL, fd, file_opts->size,
2317                                   file_opts->preallocation, errp);
2318     if (result < 0) {
2319         goto out_unlock;
2320     }
2321
2322 out_unlock:
2323     raw_apply_lock_bytes(NULL, fd, 0, 0, true, &local_err);
2324     if (local_err) {
2325         /* The above call should not fail, and if it does, that does
2326          * not mean the whole creation operation has failed.  So
2327          * report it the user for their convenience, but do not report
2328          * it to the caller. */
2329         warn_report_err(local_err);
2330     }
2331
2332 out_close:
2333     if (qemu_close(fd) != 0 && result == 0) {
2334         result = -errno;
2335         error_setg_errno(errp, -result, "Could not close the new file");
2336     }
2337 out:
2338     return result;
2339 }
2340
2341 static int coroutine_fn raw_co_create_opts(const char *filename, QemuOpts *opts,
2342                                            Error **errp)
2343 {
2344     BlockdevCreateOptions options;
2345     int64_t total_size = 0;
2346     bool nocow = false;
2347     PreallocMode prealloc;
2348     char *buf = NULL;
2349     Error *local_err = NULL;
2350
2351     /* Skip file: protocol prefix */
2352     strstart(filename, "file:", &filename);
2353
2354     /* Read out options */
2355     total_size = ROUND_UP(qemu_opt_get_size_del(opts, BLOCK_OPT_SIZE, 0),
2356                           BDRV_SECTOR_SIZE);
2357     nocow = qemu_opt_get_bool(opts, BLOCK_OPT_NOCOW, false);
2358     buf = qemu_opt_get_del(opts, BLOCK_OPT_PREALLOC);
2359     prealloc = qapi_enum_parse(&PreallocMode_lookup, buf,
2360                                PREALLOC_MODE_OFF, &local_err);
2361     g_free(buf);
2362     if (local_err) {
2363         error_propagate(errp, local_err);
2364         return -EINVAL;
2365     }
2366
2367     options = (BlockdevCreateOptions) {
2368         .driver     = BLOCKDEV_DRIVER_FILE,
2369         .u.file     = {
2370             .filename           = (char *) filename,
2371             .size               = total_size,
2372             .has_preallocation  = true,
2373             .preallocation      = prealloc,
2374             .has_nocow          = true,
2375             .nocow              = nocow,
2376         },
2377     };
2378     return raw_co_create(&options, errp);
2379 }
2380
2381 /*
2382  * Find allocation range in @bs around offset @start.
2383  * May change underlying file descriptor's file offset.
2384  * If @start is not in a hole, store @start in @data, and the
2385  * beginning of the next hole in @hole, and return 0.
2386  * If @start is in a non-trailing hole, store @start in @hole and the
2387  * beginning of the next non-hole in @data, and return 0.
2388  * If @start is in a trailing hole or beyond EOF, return -ENXIO.
2389  * If we can't find out, return a negative errno other than -ENXIO.
2390  */
2391 static int find_allocation(BlockDriverState *bs, off_t start,
2392                            off_t *data, off_t *hole)
2393 {
2394 #if defined SEEK_HOLE && defined SEEK_DATA
2395     BDRVRawState *s = bs->opaque;
2396     off_t offs;
2397
2398     /*
2399      * SEEK_DATA cases:
2400      * D1. offs == start: start is in data
2401      * D2. offs > start: start is in a hole, next data at offs
2402      * D3. offs < 0, errno = ENXIO: either start is in a trailing hole
2403      *                              or start is beyond EOF
2404      *     If the latter happens, the file has been truncated behind
2405      *     our back since we opened it.  All bets are off then.
2406      *     Treating like a trailing hole is simplest.
2407      * D4. offs < 0, errno != ENXIO: we learned nothing
2408      */
2409     offs = lseek(s->fd, start, SEEK_DATA);
2410     if (offs < 0) {
2411         return -errno;          /* D3 or D4 */
2412     }
2413
2414     if (offs < start) {
2415         /* This is not a valid return by lseek().  We are safe to just return
2416          * -EIO in this case, and we'll treat it like D4. */
2417         return -EIO;
2418     }
2419
2420     if (offs > start) {
2421         /* D2: in hole, next data at offs */
2422         *hole = start;
2423         *data = offs;
2424         return 0;
2425     }
2426
2427     /* D1: in data, end not yet known */
2428
2429     /*
2430      * SEEK_HOLE cases:
2431      * H1. offs == start: start is in a hole
2432      *     If this happens here, a hole has been dug behind our back
2433      *     since the previous lseek().
2434      * H2. offs > start: either start is in data, next hole at offs,
2435      *                   or start is in trailing hole, EOF at offs
2436      *     Linux treats trailing holes like any other hole: offs ==
2437      *     start.  Solaris seeks to EOF instead: offs > start (blech).
2438      *     If that happens here, a hole has been dug behind our back
2439      *     since the previous lseek().
2440      * H3. offs < 0, errno = ENXIO: start is beyond EOF
2441      *     If this happens, the file has been truncated behind our
2442      *     back since we opened it.  Treat it like a trailing hole.
2443      * H4. offs < 0, errno != ENXIO: we learned nothing
2444      *     Pretend we know nothing at all, i.e. "forget" about D1.
2445      */
2446     offs = lseek(s->fd, start, SEEK_HOLE);
2447     if (offs < 0) {
2448         return -errno;          /* D1 and (H3 or H4) */
2449     }
2450
2451     if (offs < start) {
2452         /* This is not a valid return by lseek().  We are safe to just return
2453          * -EIO in this case, and we'll treat it like H4. */
2454         return -EIO;
2455     }
2456
2457     if (offs > start) {
2458         /*
2459          * D1 and H2: either in data, next hole at offs, or it was in
2460          * data but is now in a trailing hole.  In the latter case,
2461          * all bets are off.  Treating it as if it there was data all
2462          * the way to EOF is safe, so simply do that.
2463          */
2464         *data = start;
2465         *hole = offs;
2466         return 0;
2467     }
2468
2469     /* D1 and H1 */
2470     return -EBUSY;
2471 #else
2472     return -ENOTSUP;
2473 #endif
2474 }
2475
2476 /*
2477  * Returns the allocation status of the specified offset.
2478  *
2479  * The block layer guarantees 'offset' and 'bytes' are within bounds.
2480  *
2481  * 'pnum' is set to the number of bytes (including and immediately following
2482  * the specified offset) that are known to be in the same
2483  * allocated/unallocated state.
2484  *
2485  * 'bytes' is the max value 'pnum' should be set to.
2486  */
2487 static int coroutine_fn raw_co_block_status(BlockDriverState *bs,
2488                                             bool want_zero,
2489                                             int64_t offset,
2490                                             int64_t bytes, int64_t *pnum,
2491                                             int64_t *map,
2492                                             BlockDriverState **file)
2493 {
2494     off_t data = 0, hole = 0;
2495     int ret;
2496
2497     assert(QEMU_IS_ALIGNED(offset | bytes, bs->bl.request_alignment));
2498
2499     ret = fd_open(bs);
2500     if (ret < 0) {
2501         return ret;
2502     }
2503
2504     if (!want_zero) {
2505         *pnum = bytes;
2506         *map = offset;
2507         *file = bs;
2508         return BDRV_BLOCK_DATA | BDRV_BLOCK_OFFSET_VALID;
2509     }
2510
2511     ret = find_allocation(bs, offset, &data, &hole);
2512     if (ret == -ENXIO) {
2513         /* Trailing hole */
2514         *pnum = bytes;
2515         ret = BDRV_BLOCK_ZERO;
2516     } else if (ret < 0) {
2517         /* No info available, so pretend there are no holes */
2518         *pnum = bytes;
2519         ret = BDRV_BLOCK_DATA;
2520     } else if (data == offset) {
2521         /* On a data extent, compute bytes to the end of the extent,
2522          * possibly including a partial sector at EOF. */
2523         *pnum = MIN(bytes, hole - offset);
2524
2525         /*
2526          * We are not allowed to return partial sectors, though, so
2527          * round up if necessary.
2528          */
2529         if (!QEMU_IS_ALIGNED(*pnum, bs->bl.request_alignment)) {
2530             int64_t file_length = raw_getlength(bs);
2531             if (file_length > 0) {
2532                 /* Ignore errors, this is just a safeguard */
2533                 assert(hole == file_length);
2534             }
2535             *pnum = ROUND_UP(*pnum, bs->bl.request_alignment);
2536         }
2537
2538         ret = BDRV_BLOCK_DATA;
2539     } else {
2540         /* On a hole, compute bytes to the beginning of the next extent.  */
2541         assert(hole == offset);
2542         *pnum = MIN(bytes, data - offset);
2543         ret = BDRV_BLOCK_ZERO;
2544     }
2545     *map = offset;
2546     *file = bs;
2547     return ret | BDRV_BLOCK_OFFSET_VALID;
2548 }
2549
2550 #if defined(__linux__)
2551 /* Verify that the file is not in the page cache */
2552 static void check_cache_dropped(BlockDriverState *bs, Error **errp)
2553 {
2554     const size_t window_size = 128 * 1024 * 1024;
2555     BDRVRawState *s = bs->opaque;
2556     void *window = NULL;
2557     size_t length = 0;
2558     unsigned char *vec;
2559     size_t page_size;
2560     off_t offset;
2561     off_t end;
2562
2563     /* mincore(2) page status information requires 1 byte per page */
2564     page_size = sysconf(_SC_PAGESIZE);
2565     vec = g_malloc(DIV_ROUND_UP(window_size, page_size));
2566
2567     end = raw_getlength(bs);
2568
2569     for (offset = 0; offset < end; offset += window_size) {
2570         void *new_window;
2571         size_t new_length;
2572         size_t vec_end;
2573         size_t i;
2574         int ret;
2575
2576         /* Unmap previous window if size has changed */
2577         new_length = MIN(end - offset, window_size);
2578         if (new_length != length) {
2579             munmap(window, length);
2580             window = NULL;
2581             length = 0;
2582         }
2583
2584         new_window = mmap(window, new_length, PROT_NONE, MAP_PRIVATE,
2585                           s->fd, offset);
2586         if (new_window == MAP_FAILED) {
2587             error_setg_errno(errp, errno, "mmap failed");
2588             break;
2589         }
2590
2591         window = new_window;
2592         length = new_length;
2593
2594         ret = mincore(window, length, vec);
2595         if (ret < 0) {
2596             error_setg_errno(errp, errno, "mincore failed");
2597             break;
2598         }
2599
2600         vec_end = DIV_ROUND_UP(length, page_size);
2601         for (i = 0; i < vec_end; i++) {
2602             if (vec[i] & 0x1) {
2603                 error_setg(errp, "page cache still in use!");
2604                 break;
2605             }
2606         }
2607     }
2608
2609     if (window) {
2610         munmap(window, length);
2611     }
2612
2613     g_free(vec);
2614 }
2615 #endif /* __linux__ */
2616
2617 static void coroutine_fn raw_co_invalidate_cache(BlockDriverState *bs,
2618                                                  Error **errp)
2619 {
2620     BDRVRawState *s = bs->opaque;
2621     int ret;
2622
2623     ret = fd_open(bs);
2624     if (ret < 0) {
2625         error_setg_errno(errp, -ret, "The file descriptor is not open");
2626         return;
2627     }
2628
2629     if (!s->drop_cache) {
2630         return;
2631     }
2632
2633     if (s->open_flags & O_DIRECT) {
2634         return; /* No host kernel page cache */
2635     }
2636
2637 #if defined(__linux__)
2638     /* This sets the scene for the next syscall... */
2639     ret = bdrv_co_flush(bs);
2640     if (ret < 0) {
2641         error_setg_errno(errp, -ret, "flush failed");
2642         return;
2643     }
2644
2645     /* Linux does not invalidate pages that are dirty, locked, or mmapped by a
2646      * process.  These limitations are okay because we just fsynced the file,
2647      * we don't use mmap, and the file should not be in use by other processes.
2648      */
2649     ret = posix_fadvise(s->fd, 0, 0, POSIX_FADV_DONTNEED);
2650     if (ret != 0) { /* the return value is a positive errno */
2651         error_setg_errno(errp, ret, "fadvise failed");
2652         return;
2653     }
2654
2655     if (s->check_cache_dropped) {
2656         check_cache_dropped(bs, errp);
2657     }
2658 #else /* __linux__ */
2659     /* Do nothing.  Live migration to a remote host with cache.direct=off is
2660      * unsupported on other host operating systems.  Cache consistency issues
2661      * may occur but no error is reported here, partly because that's the
2662      * historical behavior and partly because it's hard to differentiate valid
2663      * configurations that should not cause errors.
2664      */
2665 #endif /* !__linux__ */
2666 }
2667
2668 static void raw_account_discard(BDRVRawState *s, uint64_t nbytes, int ret)
2669 {
2670     if (ret) {
2671         s->stats.discard_nb_failed++;
2672     } else {
2673         s->stats.discard_nb_ok++;
2674         s->stats.discard_bytes_ok += nbytes;
2675     }
2676 }
2677
2678 static coroutine_fn int
2679 raw_do_pdiscard(BlockDriverState *bs, int64_t offset, int bytes, bool blkdev)
2680 {
2681     BDRVRawState *s = bs->opaque;
2682     RawPosixAIOData acb;
2683     int ret;
2684
2685     acb = (RawPosixAIOData) {
2686         .bs             = bs,
2687         .aio_fildes     = s->fd,
2688         .aio_type       = QEMU_AIO_DISCARD,
2689         .aio_offset     = offset,
2690         .aio_nbytes     = bytes,
2691     };
2692
2693     if (blkdev) {
2694         acb.aio_type |= QEMU_AIO_BLKDEV;
2695     }
2696
2697     ret = raw_thread_pool_submit(bs, handle_aiocb_discard, &acb);
2698     raw_account_discard(s, bytes, ret);
2699     return ret;
2700 }
2701
2702 static coroutine_fn int
2703 raw_co_pdiscard(BlockDriverState *bs, int64_t offset, int bytes)
2704 {
2705     return raw_do_pdiscard(bs, offset, bytes, false);
2706 }
2707
2708 static int coroutine_fn
2709 raw_do_pwrite_zeroes(BlockDriverState *bs, int64_t offset, int bytes,
2710                      BdrvRequestFlags flags, bool blkdev)
2711 {
2712     BDRVRawState *s = bs->opaque;
2713     RawPosixAIOData acb;
2714     ThreadPoolFunc *handler;
2715
2716     acb = (RawPosixAIOData) {
2717         .bs             = bs,
2718         .aio_fildes     = s->fd,
2719         .aio_type       = QEMU_AIO_WRITE_ZEROES,
2720         .aio_offset     = offset,
2721         .aio_nbytes     = bytes,
2722     };
2723
2724     if (blkdev) {
2725         acb.aio_type |= QEMU_AIO_BLKDEV;
2726     }
2727     if (flags & BDRV_REQ_NO_FALLBACK) {
2728         acb.aio_type |= QEMU_AIO_NO_FALLBACK;
2729     }
2730
2731     if (flags & BDRV_REQ_MAY_UNMAP) {
2732         acb.aio_type |= QEMU_AIO_DISCARD;
2733         handler = handle_aiocb_write_zeroes_unmap;
2734     } else {
2735         handler = handle_aiocb_write_zeroes;
2736     }
2737
2738     return raw_thread_pool_submit(bs, handler, &acb);
2739 }
2740
2741 static int coroutine_fn raw_co_pwrite_zeroes(
2742     BlockDriverState *bs, int64_t offset,
2743     int bytes, BdrvRequestFlags flags)
2744 {
2745     return raw_do_pwrite_zeroes(bs, offset, bytes, flags, false);
2746 }
2747
2748 static int raw_get_info(BlockDriverState *bs, BlockDriverInfo *bdi)
2749 {
2750     BDRVRawState *s = bs->opaque;
2751
2752     bdi->unallocated_blocks_are_zero = s->discard_zeroes;
2753     return 0;
2754 }
2755
2756 static QemuOptsList raw_create_opts = {
2757     .name = "raw-create-opts",
2758     .head = QTAILQ_HEAD_INITIALIZER(raw_create_opts.head),
2759     .desc = {
2760         {
2761             .name = BLOCK_OPT_SIZE,
2762             .type = QEMU_OPT_SIZE,
2763             .help = "Virtual disk size"
2764         },
2765         {
2766             .name = BLOCK_OPT_NOCOW,
2767             .type = QEMU_OPT_BOOL,
2768             .help = "Turn off copy-on-write (valid only on btrfs)"
2769         },
2770         {
2771             .name = BLOCK_OPT_PREALLOC,
2772             .type = QEMU_OPT_STRING,
2773             .help = "Preallocation mode (allowed values: off"
2774 #ifdef CONFIG_POSIX_FALLOCATE
2775                     ", falloc"
2776 #endif
2777                     ", full)"
2778         },
2779         { /* end of list */ }
2780     }
2781 };
2782
2783 static int raw_check_perm(BlockDriverState *bs, uint64_t perm, uint64_t shared,
2784                           Error **errp)
2785 {
2786     BDRVRawState *s = bs->opaque;
2787     BDRVRawReopenState *rs = NULL;
2788     int open_flags;
2789     int ret;
2790
2791     if (s->perm_change_fd) {
2792         /*
2793          * In the context of reopen, this function may be called several times
2794          * (directly and recursively while change permissions of the parent).
2795          * This is even true for children that don't inherit from the original
2796          * reopen node, so s->reopen_state is not set.
2797          *
2798          * Ignore all but the first call.
2799          */
2800         return 0;
2801     }
2802
2803     if (s->reopen_state) {
2804         /* We already have a new file descriptor to set permissions for */
2805         assert(s->reopen_state->perm == perm);
2806         assert(s->reopen_state->shared_perm == shared);
2807         rs = s->reopen_state->opaque;
2808         s->perm_change_fd = rs->fd;
2809         s->perm_change_flags = rs->open_flags;
2810     } else {
2811         /* We may need a new fd if auto-read-only switches the mode */
2812         ret = raw_reconfigure_getfd(bs, bs->open_flags, &open_flags, perm,
2813                                     false, errp);
2814         if (ret < 0) {
2815             return ret;
2816         } else if (ret != s->fd) {
2817             s->perm_change_fd = ret;
2818             s->perm_change_flags = open_flags;
2819         }
2820     }
2821
2822     /* Prepare permissions on old fd to avoid conflicts between old and new,
2823      * but keep everything locked that new will need. */
2824     ret = raw_handle_perm_lock(bs, RAW_PL_PREPARE, perm, shared, errp);
2825     if (ret < 0) {
2826         goto fail;
2827     }
2828
2829     /* Copy locks to the new fd */
2830     if (s->perm_change_fd) {
2831         ret = raw_apply_lock_bytes(NULL, s->perm_change_fd, perm, ~shared,
2832                                    false, errp);
2833         if (ret < 0) {
2834             raw_handle_perm_lock(bs, RAW_PL_ABORT, 0, 0, NULL);
2835             goto fail;
2836         }
2837     }
2838     return 0;
2839
2840 fail:
2841     if (s->perm_change_fd && !s->reopen_state) {
2842         qemu_close(s->perm_change_fd);
2843     }
2844     s->perm_change_fd = 0;
2845     return ret;
2846 }
2847
2848 static void raw_set_perm(BlockDriverState *bs, uint64_t perm, uint64_t shared)
2849 {
2850     BDRVRawState *s = bs->opaque;
2851
2852     /* For reopen, we have already switched to the new fd (.bdrv_set_perm is
2853      * called after .bdrv_reopen_commit) */
2854     if (s->perm_change_fd && s->fd != s->perm_change_fd) {
2855         qemu_close(s->fd);
2856         s->fd = s->perm_change_fd;
2857         s->open_flags = s->perm_change_flags;
2858     }
2859     s->perm_change_fd = 0;
2860
2861     raw_handle_perm_lock(bs, RAW_PL_COMMIT, perm, shared, NULL);
2862     s->perm = perm;
2863     s->shared_perm = shared;
2864 }
2865
2866 static void raw_abort_perm_update(BlockDriverState *bs)
2867 {
2868     BDRVRawState *s = bs->opaque;
2869
2870     /* For reopen, .bdrv_reopen_abort is called afterwards and will close
2871      * the file descriptor. */
2872     if (s->perm_change_fd && !s->reopen_state) {
2873         qemu_close(s->perm_change_fd);
2874     }
2875     s->perm_change_fd = 0;
2876
2877     raw_handle_perm_lock(bs, RAW_PL_ABORT, 0, 0, NULL);
2878 }
2879
2880 static int coroutine_fn raw_co_copy_range_from(
2881         BlockDriverState *bs, BdrvChild *src, uint64_t src_offset,
2882         BdrvChild *dst, uint64_t dst_offset, uint64_t bytes,
2883         BdrvRequestFlags read_flags, BdrvRequestFlags write_flags)
2884 {
2885     return bdrv_co_copy_range_to(src, src_offset, dst, dst_offset, bytes,
2886                                  read_flags, write_flags);
2887 }
2888
2889 static int coroutine_fn raw_co_copy_range_to(BlockDriverState *bs,
2890                                              BdrvChild *src,
2891                                              uint64_t src_offset,
2892                                              BdrvChild *dst,
2893                                              uint64_t dst_offset,
2894                                              uint64_t bytes,
2895                                              BdrvRequestFlags read_flags,
2896                                              BdrvRequestFlags write_flags)
2897 {
2898     RawPosixAIOData acb;
2899     BDRVRawState *s = bs->opaque;
2900     BDRVRawState *src_s;
2901
2902     assert(dst->bs == bs);
2903     if (src->bs->drv->bdrv_co_copy_range_to != raw_co_copy_range_to) {
2904         return -ENOTSUP;
2905     }
2906
2907     src_s = src->bs->opaque;
2908     if (fd_open(src->bs) < 0 || fd_open(dst->bs) < 0) {
2909         return -EIO;
2910     }
2911
2912     acb = (RawPosixAIOData) {
2913         .bs             = bs,
2914         .aio_type       = QEMU_AIO_COPY_RANGE,
2915         .aio_fildes     = src_s->fd,
2916         .aio_offset     = src_offset,
2917         .aio_nbytes     = bytes,
2918         .copy_range     = {
2919             .aio_fd2        = s->fd,
2920             .aio_offset2    = dst_offset,
2921         },
2922     };
2923
2924     return raw_thread_pool_submit(bs, handle_aiocb_copy_range, &acb);
2925 }
2926
2927 BlockDriver bdrv_file = {
2928     .format_name = "file",
2929     .protocol_name = "file",
2930     .instance_size = sizeof(BDRVRawState),
2931     .bdrv_needs_filename = true,
2932     .bdrv_probe = NULL, /* no probe for protocols */
2933     .bdrv_parse_filename = raw_parse_filename,
2934     .bdrv_file_open = raw_open,
2935     .bdrv_reopen_prepare = raw_reopen_prepare,
2936     .bdrv_reopen_commit = raw_reopen_commit,
2937     .bdrv_reopen_abort = raw_reopen_abort,
2938     .bdrv_close = raw_close,
2939     .bdrv_co_create = raw_co_create,
2940     .bdrv_co_create_opts = raw_co_create_opts,
2941     .bdrv_has_zero_init = bdrv_has_zero_init_1,
2942     .bdrv_has_zero_init_truncate = bdrv_has_zero_init_1,
2943     .bdrv_co_block_status = raw_co_block_status,
2944     .bdrv_co_invalidate_cache = raw_co_invalidate_cache,
2945     .bdrv_co_pwrite_zeroes = raw_co_pwrite_zeroes,
2946
2947     .bdrv_co_preadv         = raw_co_preadv,
2948     .bdrv_co_pwritev        = raw_co_pwritev,
2949     .bdrv_co_flush_to_disk  = raw_co_flush_to_disk,
2950     .bdrv_co_pdiscard       = raw_co_pdiscard,
2951     .bdrv_co_copy_range_from = raw_co_copy_range_from,
2952     .bdrv_co_copy_range_to  = raw_co_copy_range_to,
2953     .bdrv_refresh_limits = raw_refresh_limits,
2954     .bdrv_io_plug = raw_aio_plug,
2955     .bdrv_io_unplug = raw_aio_unplug,
2956     .bdrv_attach_aio_context = raw_aio_attach_aio_context,
2957
2958     .bdrv_co_truncate = raw_co_truncate,
2959     .bdrv_getlength = raw_getlength,
2960     .bdrv_get_info = raw_get_info,
2961     .bdrv_get_allocated_file_size
2962                         = raw_get_allocated_file_size,
2963     .bdrv_check_perm = raw_check_perm,
2964     .bdrv_set_perm   = raw_set_perm,
2965     .bdrv_abort_perm_update = raw_abort_perm_update,
2966     .create_opts = &raw_create_opts,
2967     .mutable_opts = mutable_opts,
2968 };
2969
2970 /***********************************************/
2971 /* host device */
2972
2973 #if defined(__APPLE__) && defined(__MACH__)
2974 static kern_return_t GetBSDPath(io_iterator_t mediaIterator, char *bsdPath,
2975                                 CFIndex maxPathSize, int flags);
2976 static char *FindEjectableOpticalMedia(io_iterator_t *mediaIterator)
2977 {
2978     kern_return_t kernResult = KERN_FAILURE;
2979     mach_port_t     masterPort;
2980     CFMutableDictionaryRef  classesToMatch;
2981     const char *matching_array[] = {kIODVDMediaClass, kIOCDMediaClass};
2982     char *mediaType = NULL;
2983
2984     kernResult = IOMasterPort( MACH_PORT_NULL, &masterPort );
2985     if ( KERN_SUCCESS != kernResult ) {
2986         printf( "IOMasterPort returned %d\n", kernResult );
2987     }
2988
2989     int index;
2990     for (index = 0; index < ARRAY_SIZE(matching_array); index++) {
2991         classesToMatch = IOServiceMatching(matching_array[index]);
2992         if (classesToMatch == NULL) {
2993             error_report("IOServiceMatching returned NULL for %s",
2994                          matching_array[index]);
2995             continue;
2996         }
2997         CFDictionarySetValue(classesToMatch, CFSTR(kIOMediaEjectableKey),
2998                              kCFBooleanTrue);
2999         kernResult = IOServiceGetMatchingServices(masterPort, classesToMatch,
3000                                                   mediaIterator);
3001         if (kernResult != KERN_SUCCESS) {
3002             error_report("Note: IOServiceGetMatchingServices returned %d",
3003                          kernResult);
3004             continue;
3005         }
3006
3007         /* If a match was found, leave the loop */
3008         if (*mediaIterator != 0) {
3009             trace_file_FindEjectableOpticalMedia(matching_array[index]);
3010             mediaType = g_strdup(matching_array[index]);
3011             break;
3012         }
3013     }
3014     return mediaType;
3015 }
3016
3017 kern_return_t GetBSDPath(io_iterator_t mediaIterator, char *bsdPath,
3018                          CFIndex maxPathSize, int flags)
3019 {
3020     io_object_t     nextMedia;
3021     kern_return_t   kernResult = KERN_FAILURE;
3022     *bsdPath = '\0';
3023     nextMedia = IOIteratorNext( mediaIterator );
3024     if ( nextMedia )
3025     {
3026         CFTypeRef   bsdPathAsCFString;
3027     bsdPathAsCFString = IORegistryEntryCreateCFProperty( nextMedia, CFSTR( kIOBSDNameKey ), kCFAllocatorDefault, 0 );
3028         if ( bsdPathAsCFString ) {
3029             size_t devPathLength;
3030             strcpy( bsdPath, _PATH_DEV );
3031             if (flags & BDRV_O_NOCACHE) {
3032                 strcat(bsdPath, "r");
3033             }
3034             devPathLength = strlen( bsdPath );
3035             if ( CFStringGetCString( bsdPathAsCFString, bsdPath + devPathLength, maxPathSize - devPathLength, kCFStringEncodingASCII ) ) {
3036                 kernResult = KERN_SUCCESS;
3037             }
3038             CFRelease( bsdPathAsCFString );
3039         }
3040         IOObjectRelease( nextMedia );
3041     }
3042
3043     return kernResult;
3044 }
3045
3046 /* Sets up a real cdrom for use in QEMU */
3047 static bool setup_cdrom(char *bsd_path, Error **errp)
3048 {
3049     int index, num_of_test_partitions = 2, fd;
3050     char test_partition[MAXPATHLEN];
3051     bool partition_found = false;
3052
3053     /* look for a working partition */
3054     for (index = 0; index < num_of_test_partitions; index++) {
3055         snprintf(test_partition, sizeof(test_partition), "%ss%d", bsd_path,
3056                  index);
3057         fd = qemu_open(test_partition, O_RDONLY | O_BINARY | O_LARGEFILE);
3058         if (fd >= 0) {
3059             partition_found = true;
3060             qemu_close(fd);
3061             break;
3062         }
3063     }
3064
3065     /* if a working partition on the device was not found */
3066     if (partition_found == false) {
3067         error_setg(errp, "Failed to find a working partition on disc");
3068     } else {
3069         trace_file_setup_cdrom(test_partition);
3070         pstrcpy(bsd_path, MAXPATHLEN, test_partition);
3071     }
3072     return partition_found;
3073 }
3074
3075 /* Prints directions on mounting and unmounting a device */
3076 static void print_unmounting_directions(const char *file_name)
3077 {
3078     error_report("If device %s is mounted on the desktop, unmount"
3079                  " it first before using it in QEMU", file_name);
3080     error_report("Command to unmount device: diskutil unmountDisk %s",
3081                  file_name);
3082     error_report("Command to mount device: diskutil mountDisk %s", file_name);
3083 }
3084
3085 #endif /* defined(__APPLE__) && defined(__MACH__) */
3086
3087 static int hdev_probe_device(const char *filename)
3088 {
3089     struct stat st;
3090
3091     /* allow a dedicated CD-ROM driver to match with a higher priority */
3092     if (strstart(filename, "/dev/cdrom", NULL))
3093         return 50;
3094
3095     if (stat(filename, &st) >= 0 &&
3096             (S_ISCHR(st.st_mode) || S_ISBLK(st.st_mode))) {
3097         return 100;
3098     }
3099
3100     return 0;
3101 }
3102
3103 static int check_hdev_writable(BDRVRawState *s)
3104 {
3105 #if defined(BLKROGET)
3106     /* Linux block devices can be configured "read-only" using blockdev(8).
3107      * This is independent of device node permissions and therefore open(2)
3108      * with O_RDWR succeeds.  Actual writes fail with EPERM.
3109      *
3110      * bdrv_open() is supposed to fail if the disk is read-only.  Explicitly
3111      * check for read-only block devices so that Linux block devices behave
3112      * properly.
3113      */
3114     struct stat st;
3115     int readonly = 0;
3116
3117     if (fstat(s->fd, &st)) {
3118         return -errno;
3119     }
3120
3121     if (!S_ISBLK(st.st_mode)) {
3122         return 0;
3123     }
3124
3125     if (ioctl(s->fd, BLKROGET, &readonly) < 0) {
3126         return -errno;
3127     }
3128
3129     if (readonly) {
3130         return -EACCES;
3131     }
3132 #endif /* defined(BLKROGET) */
3133     return 0;
3134 }
3135
3136 static void hdev_parse_filename(const char *filename, QDict *options,
3137                                 Error **errp)
3138 {
3139     bdrv_parse_filename_strip_prefix(filename, "host_device:", options);
3140 }
3141
3142 static bool hdev_is_sg(BlockDriverState *bs)
3143 {
3144
3145 #if defined(__linux__)
3146
3147     BDRVRawState *s = bs->opaque;
3148     struct stat st;
3149     struct sg_scsi_id scsiid;
3150     int sg_version;
3151     int ret;
3152
3153     if (stat(bs->filename, &st) < 0 || !S_ISCHR(st.st_mode)) {
3154         return false;
3155     }
3156
3157     ret = ioctl(s->fd, SG_GET_VERSION_NUM, &sg_version);
3158     if (ret < 0) {
3159         return false;
3160     }
3161
3162     ret = ioctl(s->fd, SG_GET_SCSI_ID, &scsiid);
3163     if (ret >= 0) {
3164         trace_file_hdev_is_sg(scsiid.scsi_type, sg_version);
3165         return true;
3166     }
3167
3168 #endif
3169
3170     return false;
3171 }
3172
3173 static int hdev_open(BlockDriverState *bs, QDict *options, int flags,
3174                      Error **errp)
3175 {
3176     BDRVRawState *s = bs->opaque;
3177     Error *local_err = NULL;
3178     int ret;
3179
3180 #if defined(__APPLE__) && defined(__MACH__)
3181     /*
3182      * Caution: while qdict_get_str() is fine, getting non-string types
3183      * would require more care.  When @options come from -blockdev or
3184      * blockdev_add, its members are typed according to the QAPI
3185      * schema, but when they come from -drive, they're all QString.
3186      */
3187     const char *filename = qdict_get_str(options, "filename");
3188     char bsd_path[MAXPATHLEN] = "";
3189     bool error_occurred = false;
3190
3191     /* If using a real cdrom */
3192     if (strcmp(filename, "/dev/cdrom") == 0) {
3193         char *mediaType = NULL;
3194         kern_return_t ret_val;
3195         io_iterator_t mediaIterator = 0;
3196
3197         mediaType = FindEjectableOpticalMedia(&mediaIterator);
3198         if (mediaType == NULL) {
3199             error_setg(errp, "Please make sure your CD/DVD is in the optical"
3200                        " drive");
3201             error_occurred = true;
3202             goto hdev_open_Mac_error;
3203         }
3204
3205         ret_val = GetBSDPath(mediaIterator, bsd_path, sizeof(bsd_path), flags);
3206         if (ret_val != KERN_SUCCESS) {
3207             error_setg(errp, "Could not get BSD path for optical drive");
3208             error_occurred = true;
3209             goto hdev_open_Mac_error;
3210         }
3211
3212         /* If a real optical drive was not found */
3213         if (bsd_path[0] == '\0') {
3214             error_setg(errp, "Failed to obtain bsd path for optical drive");
3215             error_occurred = true;
3216             goto hdev_open_Mac_error;
3217         }
3218
3219         /* If using a cdrom disc and finding a partition on the disc failed */
3220         if (strncmp(mediaType, kIOCDMediaClass, 9) == 0 &&
3221             setup_cdrom(bsd_path, errp) == false) {
3222             print_unmounting_directions(bsd_path);
3223             error_occurred = true;
3224             goto hdev_open_Mac_error;
3225         }
3226
3227         qdict_put_str(options, "filename", bsd_path);
3228
3229 hdev_open_Mac_error:
3230         g_free(mediaType);
3231         if (mediaIterator) {
3232             IOObjectRelease(mediaIterator);
3233         }
3234         if (error_occurred) {
3235             return -ENOENT;
3236         }
3237     }
3238 #endif /* defined(__APPLE__) && defined(__MACH__) */
3239
3240     s->type = FTYPE_FILE;
3241
3242     ret = raw_open_common(bs, options, flags, 0, true, &local_err);
3243     if (ret < 0) {
3244         error_propagate(errp, local_err);
3245 #if defined(__APPLE__) && defined(__MACH__)
3246         if (*bsd_path) {
3247             filename = bsd_path;
3248         }
3249         /* if a physical device experienced an error while being opened */
3250         if (strncmp(filename, "/dev/", 5) == 0) {
3251             print_unmounting_directions(filename);
3252         }
3253 #endif /* defined(__APPLE__) && defined(__MACH__) */
3254         return ret;
3255     }
3256
3257     /* Since this does ioctl the device must be already opened */
3258     bs->sg = hdev_is_sg(bs);
3259
3260     if (flags & BDRV_O_RDWR) {
3261         ret = check_hdev_writable(s);
3262         if (ret < 0) {
3263             raw_close(bs);
3264             error_setg_errno(errp, -ret, "The device is not writable");
3265             return ret;
3266         }
3267     }
3268
3269     return ret;
3270 }
3271
3272 #if defined(__linux__)
3273 static int coroutine_fn
3274 hdev_co_ioctl(BlockDriverState *bs, unsigned long int req, void *buf)
3275 {
3276     BDRVRawState *s = bs->opaque;
3277     RawPosixAIOData acb;
3278     int ret;
3279
3280     ret = fd_open(bs);
3281     if (ret < 0) {
3282         return ret;
3283     }
3284
3285     if (req == SG_IO && s->pr_mgr) {
3286         struct sg_io_hdr *io_hdr = buf;
3287         if (io_hdr->cmdp[0] == PERSISTENT_RESERVE_OUT ||
3288             io_hdr->cmdp[0] == PERSISTENT_RESERVE_IN) {
3289             return pr_manager_execute(s->pr_mgr, bdrv_get_aio_context(bs),
3290                                       s->fd, io_hdr);
3291         }
3292     }
3293
3294     acb = (RawPosixAIOData) {
3295         .bs         = bs,
3296         .aio_type   = QEMU_AIO_IOCTL,
3297         .aio_fildes = s->fd,
3298         .aio_offset = 0,
3299         .ioctl      = {
3300             .buf        = buf,
3301             .cmd        = req,
3302         },
3303     };
3304
3305     return raw_thread_pool_submit(bs, handle_aiocb_ioctl, &acb);
3306 }
3307 #endif /* linux */
3308
3309 static int fd_open(BlockDriverState *bs)
3310 {
3311     BDRVRawState *s = bs->opaque;
3312
3313     /* this is just to ensure s->fd is sane (its called by io ops) */
3314     if (s->fd >= 0)
3315         return 0;
3316     return -EIO;
3317 }
3318
3319 static coroutine_fn int
3320 hdev_co_pdiscard(BlockDriverState *bs, int64_t offset, int bytes)
3321 {
3322     BDRVRawState *s = bs->opaque;
3323     int ret;
3324
3325     ret = fd_open(bs);
3326     if (ret < 0) {
3327         raw_account_discard(s, bytes, ret);
3328         return ret;
3329     }
3330     return raw_do_pdiscard(bs, offset, bytes, true);
3331 }
3332
3333 static coroutine_fn int hdev_co_pwrite_zeroes(BlockDriverState *bs,
3334     int64_t offset, int bytes, BdrvRequestFlags flags)
3335 {
3336     int rc;
3337
3338     rc = fd_open(bs);
3339     if (rc < 0) {
3340         return rc;
3341     }
3342
3343     return raw_do_pwrite_zeroes(bs, offset, bytes, flags, true);
3344 }
3345
3346 static int coroutine_fn hdev_co_create_opts(const char *filename, QemuOpts *opts,
3347                                             Error **errp)
3348 {
3349     int fd;
3350     int ret = 0;
3351     struct stat stat_buf;
3352     int64_t total_size = 0;
3353     bool has_prefix;
3354
3355     /* This function is used by both protocol block drivers and therefore either
3356      * of these prefixes may be given.
3357      * The return value has to be stored somewhere, otherwise this is an error
3358      * due to -Werror=unused-value. */
3359     has_prefix =
3360         strstart(filename, "host_device:", &filename) ||
3361         strstart(filename, "host_cdrom:" , &filename);
3362
3363     (void)has_prefix;
3364
3365     ret = raw_normalize_devicepath(&filename, errp);
3366     if (ret < 0) {
3367         return ret;
3368     }
3369
3370     /* Read out options */
3371     total_size = ROUND_UP(qemu_opt_get_size_del(opts, BLOCK_OPT_SIZE, 0),
3372                           BDRV_SECTOR_SIZE);
3373
3374     fd = qemu_open(filename, O_WRONLY | O_BINARY);
3375     if (fd < 0) {
3376         ret = -errno;
3377         error_setg_errno(errp, -ret, "Could not open device");
3378         return ret;
3379     }
3380
3381     if (fstat(fd, &stat_buf) < 0) {
3382         ret = -errno;
3383         error_setg_errno(errp, -ret, "Could not stat device");
3384     } else if (!S_ISBLK(stat_buf.st_mode) && !S_ISCHR(stat_buf.st_mode)) {
3385         error_setg(errp,
3386                    "The given file is neither a block nor a character device");
3387         ret = -ENODEV;
3388     } else if (lseek(fd, 0, SEEK_END) < total_size) {
3389         error_setg(errp, "Device is too small");
3390         ret = -ENOSPC;
3391     }
3392
3393     if (!ret && total_size) {
3394         uint8_t buf[BDRV_SECTOR_SIZE] = { 0 };
3395         int64_t zero_size = MIN(BDRV_SECTOR_SIZE, total_size);
3396         if (lseek(fd, 0, SEEK_SET) == -1) {
3397             ret = -errno;
3398         } else {
3399             ret = qemu_write_full(fd, buf, zero_size);
3400             ret = ret == zero_size ? 0 : -errno;
3401         }
3402     }
3403     qemu_close(fd);
3404     return ret;
3405 }
3406
3407 static BlockDriver bdrv_host_device = {
3408     .format_name        = "host_device",
3409     .protocol_name        = "host_device",
3410     .instance_size      = sizeof(BDRVRawState),
3411     .bdrv_needs_filename = true,
3412     .bdrv_probe_device  = hdev_probe_device,
3413     .bdrv_parse_filename = hdev_parse_filename,
3414     .bdrv_file_open     = hdev_open,
3415     .bdrv_close         = raw_close,
3416     .bdrv_reopen_prepare = raw_reopen_prepare,
3417     .bdrv_reopen_commit  = raw_reopen_commit,
3418     .bdrv_reopen_abort   = raw_reopen_abort,
3419     .bdrv_co_create_opts = hdev_co_create_opts,
3420     .create_opts         = &raw_create_opts,
3421     .mutable_opts        = mutable_opts,
3422     .bdrv_co_invalidate_cache = raw_co_invalidate_cache,
3423     .bdrv_co_pwrite_zeroes = hdev_co_pwrite_zeroes,
3424
3425     .bdrv_co_preadv         = raw_co_preadv,
3426     .bdrv_co_pwritev        = raw_co_pwritev,
3427     .bdrv_co_flush_to_disk  = raw_co_flush_to_disk,
3428     .bdrv_co_pdiscard       = hdev_co_pdiscard,
3429     .bdrv_co_copy_range_from = raw_co_copy_range_from,
3430     .bdrv_co_copy_range_to  = raw_co_copy_range_to,
3431     .bdrv_refresh_limits = raw_refresh_limits,
3432     .bdrv_io_plug = raw_aio_plug,
3433     .bdrv_io_unplug = raw_aio_unplug,
3434     .bdrv_attach_aio_context = raw_aio_attach_aio_context,
3435
3436     .bdrv_co_truncate       = raw_co_truncate,
3437     .bdrv_getlength     = raw_getlength,
3438     .bdrv_get_info = raw_get_info,
3439     .bdrv_get_allocated_file_size
3440                         = raw_get_allocated_file_size,
3441     .bdrv_check_perm = raw_check_perm,
3442     .bdrv_set_perm   = raw_set_perm,
3443     .bdrv_abort_perm_update = raw_abort_perm_update,
3444     .bdrv_probe_blocksizes = hdev_probe_blocksizes,
3445     .bdrv_probe_geometry = hdev_probe_geometry,
3446
3447     /* generic scsi device */
3448 #ifdef __linux__
3449     .bdrv_co_ioctl          = hdev_co_ioctl,
3450 #endif
3451 };
3452
3453 #if defined(__linux__) || defined(__FreeBSD__) || defined(__FreeBSD_kernel__)
3454 static void cdrom_parse_filename(const char *filename, QDict *options,
3455                                  Error **errp)
3456 {
3457     bdrv_parse_filename_strip_prefix(filename, "host_cdrom:", options);
3458 }
3459 #endif
3460
3461 #ifdef __linux__
3462 static int cdrom_open(BlockDriverState *bs, QDict *options, int flags,
3463                       Error **errp)
3464 {
3465     BDRVRawState *s = bs->opaque;
3466
3467     s->type = FTYPE_CD;
3468
3469     /* open will not fail even if no CD is inserted, so add O_NONBLOCK */
3470     return raw_open_common(bs, options, flags, O_NONBLOCK, true, errp);
3471 }
3472
3473 static int cdrom_probe_device(const char *filename)
3474 {
3475     int fd, ret;
3476     int prio = 0;
3477     struct stat st;
3478
3479     fd = qemu_open(filename, O_RDONLY | O_NONBLOCK);
3480     if (fd < 0) {
3481         goto out;
3482     }
3483     ret = fstat(fd, &st);
3484     if (ret == -1 || !S_ISBLK(st.st_mode)) {
3485         goto outc;
3486     }
3487
3488     /* Attempt to detect via a CDROM specific ioctl */
3489     ret = ioctl(fd, CDROM_DRIVE_STATUS, CDSL_CURRENT);
3490     if (ret >= 0)
3491         prio = 100;
3492
3493 outc:
3494     qemu_close(fd);
3495 out:
3496     return prio;
3497 }
3498
3499 static bool cdrom_is_inserted(BlockDriverState *bs)
3500 {
3501     BDRVRawState *s = bs->opaque;
3502     int ret;
3503
3504     ret = ioctl(s->fd, CDROM_DRIVE_STATUS, CDSL_CURRENT);
3505     return ret == CDS_DISC_OK;
3506 }
3507
3508 static void cdrom_eject(BlockDriverState *bs, bool eject_flag)
3509 {
3510     BDRVRawState *s = bs->opaque;
3511
3512     if (eject_flag) {
3513         if (ioctl(s->fd, CDROMEJECT, NULL) < 0)
3514             perror("CDROMEJECT");
3515     } else {
3516         if (ioctl(s->fd, CDROMCLOSETRAY, NULL) < 0)
3517             perror("CDROMEJECT");
3518     }
3519 }
3520
3521 static void cdrom_lock_medium(BlockDriverState *bs, bool locked)
3522 {
3523     BDRVRawState *s = bs->opaque;
3524
3525     if (ioctl(s->fd, CDROM_LOCKDOOR, locked) < 0) {
3526         /*
3527          * Note: an error can happen if the distribution automatically
3528          * mounts the CD-ROM
3529          */
3530         /* perror("CDROM_LOCKDOOR"); */
3531     }
3532 }
3533
3534 static BlockDriver bdrv_host_cdrom = {
3535     .format_name        = "host_cdrom",
3536     .protocol_name      = "host_cdrom",
3537     .instance_size      = sizeof(BDRVRawState),
3538     .bdrv_needs_filename = true,
3539     .bdrv_probe_device  = cdrom_probe_device,
3540     .bdrv_parse_filename = cdrom_parse_filename,
3541     .bdrv_file_open     = cdrom_open,
3542     .bdrv_close         = raw_close,
3543     .bdrv_reopen_prepare = raw_reopen_prepare,
3544     .bdrv_reopen_commit  = raw_reopen_commit,
3545     .bdrv_reopen_abort   = raw_reopen_abort,
3546     .bdrv_co_create_opts = hdev_co_create_opts,
3547     .create_opts         = &raw_create_opts,
3548     .mutable_opts        = mutable_opts,
3549     .bdrv_co_invalidate_cache = raw_co_invalidate_cache,
3550
3551
3552     .bdrv_co_preadv         = raw_co_preadv,
3553     .bdrv_co_pwritev        = raw_co_pwritev,
3554     .bdrv_co_flush_to_disk  = raw_co_flush_to_disk,
3555     .bdrv_refresh_limits = raw_refresh_limits,
3556     .bdrv_io_plug = raw_aio_plug,
3557     .bdrv_io_unplug = raw_aio_unplug,
3558     .bdrv_attach_aio_context = raw_aio_attach_aio_context,
3559
3560     .bdrv_co_truncate    = raw_co_truncate,
3561     .bdrv_getlength      = raw_getlength,
3562     .has_variable_length = true,
3563     .bdrv_get_allocated_file_size
3564                         = raw_get_allocated_file_size,
3565
3566     /* removable device support */
3567     .bdrv_is_inserted   = cdrom_is_inserted,
3568     .bdrv_eject         = cdrom_eject,
3569     .bdrv_lock_medium   = cdrom_lock_medium,
3570
3571     /* generic scsi device */
3572     .bdrv_co_ioctl      = hdev_co_ioctl,
3573 };
3574 #endif /* __linux__ */
3575
3576 #if defined (__FreeBSD__) || defined(__FreeBSD_kernel__)
3577 static int cdrom_open(BlockDriverState *bs, QDict *options, int flags,
3578                       Error **errp)
3579 {
3580     BDRVRawState *s = bs->opaque;
3581     Error *local_err = NULL;
3582     int ret;
3583
3584     s->type = FTYPE_CD;
3585
3586     ret = raw_open_common(bs, options, flags, 0, true, &local_err);
3587     if (ret) {
3588         error_propagate(errp, local_err);
3589         return ret;
3590     }
3591
3592     /* make sure the door isn't locked at this time */
3593     ioctl(s->fd, CDIOCALLOW);
3594     return 0;
3595 }
3596
3597 static int cdrom_probe_device(const char *filename)
3598 {
3599     if (strstart(filename, "/dev/cd", NULL) ||
3600             strstart(filename, "/dev/acd", NULL))
3601         return 100;
3602     return 0;
3603 }
3604
3605 static int cdrom_reopen(BlockDriverState *bs)
3606 {
3607     BDRVRawState *s = bs->opaque;
3608     int fd;
3609
3610     /*
3611      * Force reread of possibly changed/newly loaded disc,
3612      * FreeBSD seems to not notice sometimes...
3613      */
3614     if (s->fd >= 0)
3615         qemu_close(s->fd);
3616     fd = qemu_open(bs->filename, s->open_flags, 0644);
3617     if (fd < 0) {
3618         s->fd = -1;
3619         return -EIO;
3620     }
3621     s->fd = fd;
3622
3623     /* make sure the door isn't locked at this time */
3624     ioctl(s->fd, CDIOCALLOW);
3625     return 0;
3626 }
3627
3628 static bool cdrom_is_inserted(BlockDriverState *bs)
3629 {
3630     return raw_getlength(bs) > 0;
3631 }
3632
3633 static void cdrom_eject(BlockDriverState *bs, bool eject_flag)
3634 {
3635     BDRVRawState *s = bs->opaque;
3636
3637     if (s->fd < 0)
3638         return;
3639
3640     (void) ioctl(s->fd, CDIOCALLOW);
3641
3642     if (eject_flag) {
3643         if (ioctl(s->fd, CDIOCEJECT) < 0)
3644             perror("CDIOCEJECT");
3645     } else {
3646         if (ioctl(s->fd, CDIOCCLOSE) < 0)
3647             perror("CDIOCCLOSE");
3648     }
3649
3650     cdrom_reopen(bs);
3651 }
3652
3653 static void cdrom_lock_medium(BlockDriverState *bs, bool locked)
3654 {
3655     BDRVRawState *s = bs->opaque;
3656
3657     if (s->fd < 0)
3658         return;
3659     if (ioctl(s->fd, (locked ? CDIOCPREVENT : CDIOCALLOW)) < 0) {
3660         /*
3661          * Note: an error can happen if the distribution automatically
3662          * mounts the CD-ROM
3663          */
3664         /* perror("CDROM_LOCKDOOR"); */
3665     }
3666 }
3667
3668 static BlockDriver bdrv_host_cdrom = {
3669     .format_name        = "host_cdrom",
3670     .protocol_name      = "host_cdrom",
3671     .instance_size      = sizeof(BDRVRawState),
3672     .bdrv_needs_filename = true,
3673     .bdrv_probe_device  = cdrom_probe_device,
3674     .bdrv_parse_filename = cdrom_parse_filename,
3675     .bdrv_file_open     = cdrom_open,
3676     .bdrv_close         = raw_close,
3677     .bdrv_reopen_prepare = raw_reopen_prepare,
3678     .bdrv_reopen_commit  = raw_reopen_commit,
3679     .bdrv_reopen_abort   = raw_reopen_abort,
3680     .bdrv_co_create_opts = hdev_co_create_opts,
3681     .create_opts        = &raw_create_opts,
3682     .mutable_opts       = mutable_opts,
3683
3684     .bdrv_co_preadv         = raw_co_preadv,
3685     .bdrv_co_pwritev        = raw_co_pwritev,
3686     .bdrv_co_flush_to_disk  = raw_co_flush_to_disk,
3687     .bdrv_refresh_limits = raw_refresh_limits,
3688     .bdrv_io_plug = raw_aio_plug,
3689     .bdrv_io_unplug = raw_aio_unplug,
3690     .bdrv_attach_aio_context = raw_aio_attach_aio_context,
3691
3692     .bdrv_co_truncate    = raw_co_truncate,
3693     .bdrv_getlength      = raw_getlength,
3694     .has_variable_length = true,
3695     .bdrv_get_allocated_file_size
3696                         = raw_get_allocated_file_size,
3697
3698     /* removable device support */
3699     .bdrv_is_inserted   = cdrom_is_inserted,
3700     .bdrv_eject         = cdrom_eject,
3701     .bdrv_lock_medium   = cdrom_lock_medium,
3702 };
3703 #endif /* __FreeBSD__ */
3704
3705 static void bdrv_file_init(void)
3706 {
3707     /*
3708      * Register all the drivers.  Note that order is important, the driver
3709      * registered last will get probed first.
3710      */
3711     bdrv_register(&bdrv_file);
3712     bdrv_register(&bdrv_host_device);
3713 #ifdef __linux__
3714     bdrv_register(&bdrv_host_cdrom);
3715 #endif
3716 #if defined(__FreeBSD__) || defined(__FreeBSD_kernel__)
3717     bdrv_register(&bdrv_host_cdrom);
3718 #endif
3719 }
3720
3721 block_init(bdrv_file_init);
This page took 0.212658 seconds and 2 git commands to generate.