xref: /qemu/block/file-posix.c (revision 294682cc)
1c1bb86cdSEric Blake /*
2c1bb86cdSEric Blake  * Block driver for RAW files (posix)
3c1bb86cdSEric Blake  *
4c1bb86cdSEric Blake  * Copyright (c) 2006 Fabrice Bellard
5c1bb86cdSEric Blake  *
6c1bb86cdSEric Blake  * Permission is hereby granted, free of charge, to any person obtaining a copy
7c1bb86cdSEric Blake  * of this software and associated documentation files (the "Software"), to deal
8c1bb86cdSEric Blake  * in the Software without restriction, including without limitation the rights
9c1bb86cdSEric Blake  * to use, copy, modify, merge, publish, distribute, sublicense, and/or sell
10c1bb86cdSEric Blake  * copies of the Software, and to permit persons to whom the Software is
11c1bb86cdSEric Blake  * furnished to do so, subject to the following conditions:
12c1bb86cdSEric Blake  *
13c1bb86cdSEric Blake  * The above copyright notice and this permission notice shall be included in
14c1bb86cdSEric Blake  * all copies or substantial portions of the Software.
15c1bb86cdSEric Blake  *
16c1bb86cdSEric Blake  * THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
17c1bb86cdSEric Blake  * IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
18c1bb86cdSEric Blake  * FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL
19c1bb86cdSEric Blake  * THE AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
20c1bb86cdSEric Blake  * LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,
21c1bb86cdSEric Blake  * OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN
22c1bb86cdSEric Blake  * THE SOFTWARE.
23c1bb86cdSEric Blake  */
24922a01a0SMarkus Armbruster 
25c1bb86cdSEric Blake #include "qemu/osdep.h"
26a8d25326SMarkus Armbruster #include "qemu-common.h"
27c1bb86cdSEric Blake #include "qapi/error.h"
28c1bb86cdSEric Blake #include "qemu/cutils.h"
29c1bb86cdSEric Blake #include "qemu/error-report.h"
30c1bb86cdSEric Blake #include "block/block_int.h"
31c1bb86cdSEric Blake #include "qemu/module.h"
32922a01a0SMarkus Armbruster #include "qemu/option.h"
33c1bb86cdSEric Blake #include "trace.h"
34c1bb86cdSEric Blake #include "block/thread-pool.h"
35c1bb86cdSEric Blake #include "qemu/iov.h"
36c1bb86cdSEric Blake #include "block/raw-aio.h"
37452fcdbcSMarkus Armbruster #include "qapi/qmp/qdict.h"
38c1bb86cdSEric Blake #include "qapi/qmp/qstring.h"
39c1bb86cdSEric Blake 
407c9e5276SPaolo Bonzini #include "scsi/pr-manager.h"
417c9e5276SPaolo Bonzini #include "scsi/constants.h"
427c9e5276SPaolo Bonzini 
43c1bb86cdSEric Blake #if defined(__APPLE__) && (__MACH__)
44c1bb86cdSEric Blake #include <paths.h>
45c1bb86cdSEric Blake #include <sys/param.h>
46c1bb86cdSEric Blake #include <IOKit/IOKitLib.h>
47c1bb86cdSEric Blake #include <IOKit/IOBSD.h>
48c1bb86cdSEric Blake #include <IOKit/storage/IOMediaBSDClient.h>
49c1bb86cdSEric Blake #include <IOKit/storage/IOMedia.h>
50c1bb86cdSEric Blake #include <IOKit/storage/IOCDMedia.h>
51c1bb86cdSEric Blake //#include <IOKit/storage/IOCDTypes.h>
52c1bb86cdSEric Blake #include <IOKit/storage/IODVDMedia.h>
53c1bb86cdSEric Blake #include <CoreFoundation/CoreFoundation.h>
54c1bb86cdSEric Blake #endif
55c1bb86cdSEric Blake 
56c1bb86cdSEric Blake #ifdef __sun__
57c1bb86cdSEric Blake #define _POSIX_PTHREAD_SEMANTICS 1
58c1bb86cdSEric Blake #include <sys/dkio.h>
59c1bb86cdSEric Blake #endif
60c1bb86cdSEric Blake #ifdef __linux__
61c1bb86cdSEric Blake #include <sys/ioctl.h>
62c1bb86cdSEric Blake #include <sys/param.h>
631efad060SFam Zheng #include <sys/syscall.h>
64c1bb86cdSEric Blake #include <linux/cdrom.h>
65c1bb86cdSEric Blake #include <linux/fd.h>
66c1bb86cdSEric Blake #include <linux/fs.h>
67c1bb86cdSEric Blake #include <linux/hdreg.h>
68c1bb86cdSEric Blake #include <scsi/sg.h>
69c1bb86cdSEric Blake #ifdef __s390__
70c1bb86cdSEric Blake #include <asm/dasd.h>
71c1bb86cdSEric Blake #endif
72c1bb86cdSEric Blake #ifndef FS_NOCOW_FL
73c1bb86cdSEric Blake #define FS_NOCOW_FL                     0x00800000 /* Do not cow file */
74c1bb86cdSEric Blake #endif
75c1bb86cdSEric Blake #endif
76c1bb86cdSEric Blake #if defined(CONFIG_FALLOCATE_PUNCH_HOLE) || defined(CONFIG_FALLOCATE_ZERO_RANGE)
77c1bb86cdSEric Blake #include <linux/falloc.h>
78c1bb86cdSEric Blake #endif
79c1bb86cdSEric Blake #if defined (__FreeBSD__) || defined(__FreeBSD_kernel__)
80c1bb86cdSEric Blake #include <sys/disk.h>
81c1bb86cdSEric Blake #include <sys/cdio.h>
82c1bb86cdSEric Blake #endif
83c1bb86cdSEric Blake 
84c1bb86cdSEric Blake #ifdef __OpenBSD__
85c1bb86cdSEric Blake #include <sys/ioctl.h>
86c1bb86cdSEric Blake #include <sys/disklabel.h>
87c1bb86cdSEric Blake #include <sys/dkio.h>
88c1bb86cdSEric Blake #endif
89c1bb86cdSEric Blake 
90c1bb86cdSEric Blake #ifdef __NetBSD__
91c1bb86cdSEric Blake #include <sys/ioctl.h>
92c1bb86cdSEric Blake #include <sys/disklabel.h>
93c1bb86cdSEric Blake #include <sys/dkio.h>
94c1bb86cdSEric Blake #include <sys/disk.h>
95c1bb86cdSEric Blake #endif
96c1bb86cdSEric Blake 
97c1bb86cdSEric Blake #ifdef __DragonFly__
98c1bb86cdSEric Blake #include <sys/ioctl.h>
99c1bb86cdSEric Blake #include <sys/diskslice.h>
100c1bb86cdSEric Blake #endif
101c1bb86cdSEric Blake 
102c1bb86cdSEric Blake #ifdef CONFIG_XFS
103c1bb86cdSEric Blake #include <xfs/xfs.h>
104c1bb86cdSEric Blake #endif
105c1bb86cdSEric Blake 
1064f7d28d7SLaurent Vivier #include "trace.h"
107c1bb86cdSEric Blake 
108c1bb86cdSEric Blake /* OS X does not have O_DSYNC */
109c1bb86cdSEric Blake #ifndef O_DSYNC
110c1bb86cdSEric Blake #ifdef O_SYNC
111c1bb86cdSEric Blake #define O_DSYNC O_SYNC
112c1bb86cdSEric Blake #elif defined(O_FSYNC)
113c1bb86cdSEric Blake #define O_DSYNC O_FSYNC
114c1bb86cdSEric Blake #endif
115c1bb86cdSEric Blake #endif
116c1bb86cdSEric Blake 
117c1bb86cdSEric Blake /* Approximate O_DIRECT with O_DSYNC if O_DIRECT isn't available */
118c1bb86cdSEric Blake #ifndef O_DIRECT
119c1bb86cdSEric Blake #define O_DIRECT O_DSYNC
120c1bb86cdSEric Blake #endif
121c1bb86cdSEric Blake 
122c1bb86cdSEric Blake #define FTYPE_FILE   0
123c1bb86cdSEric Blake #define FTYPE_CD     1
124c1bb86cdSEric Blake 
125c1bb86cdSEric Blake #define MAX_BLOCKSIZE	4096
126c1bb86cdSEric Blake 
127244a5668SFam Zheng /* Posix file locking bytes. Libvirt takes byte 0, we start from higher bytes,
128244a5668SFam Zheng  * leaving a few more bytes for its future use. */
129244a5668SFam Zheng #define RAW_LOCK_PERM_BASE             100
130244a5668SFam Zheng #define RAW_LOCK_SHARED_BASE           200
131244a5668SFam Zheng 
132c1bb86cdSEric Blake typedef struct BDRVRawState {
133c1bb86cdSEric Blake     int fd;
134244a5668SFam Zheng     bool use_lock;
135c1bb86cdSEric Blake     int type;
136c1bb86cdSEric Blake     int open_flags;
137c1bb86cdSEric Blake     size_t buf_align;
138c1bb86cdSEric Blake 
139244a5668SFam Zheng     /* The current permissions. */
140244a5668SFam Zheng     uint64_t perm;
141244a5668SFam Zheng     uint64_t shared_perm;
142244a5668SFam Zheng 
1432996ffadSFam Zheng     /* The perms bits whose corresponding bytes are already locked in
144f2e3af29SFam Zheng      * s->fd. */
1452996ffadSFam Zheng     uint64_t locked_perm;
1462996ffadSFam Zheng     uint64_t locked_shared_perm;
1472996ffadSFam Zheng 
1486ceabe6fSKevin Wolf     int perm_change_fd;
149094e3639SMax Reitz     int perm_change_flags;
150e0c9cf3aSKevin Wolf     BDRVReopenState *reopen_state;
151e0c9cf3aSKevin Wolf 
152c1bb86cdSEric Blake #ifdef CONFIG_XFS
153c1bb86cdSEric Blake     bool is_xfs:1;
154c1bb86cdSEric Blake #endif
155c1bb86cdSEric Blake     bool has_discard:1;
156c1bb86cdSEric Blake     bool has_write_zeroes:1;
157c1bb86cdSEric Blake     bool discard_zeroes:1;
158c1bb86cdSEric Blake     bool use_linux_aio:1;
159e5bcf967SKevin Wolf     bool page_cache_inconsistent:1;
160c1bb86cdSEric Blake     bool has_fallocate;
161c1bb86cdSEric Blake     bool needs_alignment;
162f357fcd8SStefan Hajnoczi     bool drop_cache;
16331be8a2aSStefan Hajnoczi     bool check_cache_dropped;
1647c9e5276SPaolo Bonzini 
1657c9e5276SPaolo Bonzini     PRManager *pr_mgr;
166c1bb86cdSEric Blake } BDRVRawState;
167c1bb86cdSEric Blake 
168c1bb86cdSEric Blake typedef struct BDRVRawReopenState {
169c1bb86cdSEric Blake     int fd;
170c1bb86cdSEric Blake     int open_flags;
171f357fcd8SStefan Hajnoczi     bool drop_cache;
17231be8a2aSStefan Hajnoczi     bool check_cache_dropped;
173c1bb86cdSEric Blake } BDRVRawReopenState;
174c1bb86cdSEric Blake 
175c1bb86cdSEric Blake static int fd_open(BlockDriverState *bs);
176c1bb86cdSEric Blake static int64_t raw_getlength(BlockDriverState *bs);
177c1bb86cdSEric Blake 
178c1bb86cdSEric Blake typedef struct RawPosixAIOData {
179c1bb86cdSEric Blake     BlockDriverState *bs;
180c1bb86cdSEric Blake     int aio_type;
181d57c44d0SKevin Wolf     int aio_fildes;
182d57c44d0SKevin Wolf 
183d57c44d0SKevin Wolf     off_t aio_offset;
184d57c44d0SKevin Wolf     uint64_t aio_nbytes;
185d57c44d0SKevin Wolf 
18693f4e2ffSKevin Wolf     union {
18793f4e2ffSKevin Wolf         struct {
188d57c44d0SKevin Wolf             struct iovec *iov;
189d57c44d0SKevin Wolf             int niov;
190d57c44d0SKevin Wolf         } io;
191d57c44d0SKevin Wolf         struct {
192d57c44d0SKevin Wolf             uint64_t cmd;
193d57c44d0SKevin Wolf             void *buf;
194d57c44d0SKevin Wolf         } ioctl;
195d57c44d0SKevin Wolf         struct {
1961efad060SFam Zheng             int aio_fd2;
1971efad060SFam Zheng             off_t aio_offset2;
198d57c44d0SKevin Wolf         } copy_range;
19993f4e2ffSKevin Wolf         struct {
20093f4e2ffSKevin Wolf             PreallocMode prealloc;
20193f4e2ffSKevin Wolf             Error **errp;
202d57c44d0SKevin Wolf         } truncate;
20393f4e2ffSKevin Wolf     };
204c1bb86cdSEric Blake } RawPosixAIOData;
205c1bb86cdSEric Blake 
206c1bb86cdSEric Blake #if defined(__FreeBSD__) || defined(__FreeBSD_kernel__)
207c1bb86cdSEric Blake static int cdrom_reopen(BlockDriverState *bs);
208c1bb86cdSEric Blake #endif
209c1bb86cdSEric Blake 
210c1bb86cdSEric Blake #if defined(__NetBSD__)
211db0754dfSFam Zheng static int raw_normalize_devicepath(const char **filename, Error **errp)
212c1bb86cdSEric Blake {
213c1bb86cdSEric Blake     static char namebuf[PATH_MAX];
214c1bb86cdSEric Blake     const char *dp, *fname;
215c1bb86cdSEric Blake     struct stat sb;
216c1bb86cdSEric Blake 
217c1bb86cdSEric Blake     fname = *filename;
218c1bb86cdSEric Blake     dp = strrchr(fname, '/');
219c1bb86cdSEric Blake     if (lstat(fname, &sb) < 0) {
220f6fc1e30SPaolo Bonzini         error_setg_file_open(errp, errno, fname);
221c1bb86cdSEric Blake         return -errno;
222c1bb86cdSEric Blake     }
223c1bb86cdSEric Blake 
224c1bb86cdSEric Blake     if (!S_ISBLK(sb.st_mode)) {
225c1bb86cdSEric Blake         return 0;
226c1bb86cdSEric Blake     }
227c1bb86cdSEric Blake 
228c1bb86cdSEric Blake     if (dp == NULL) {
229c1bb86cdSEric Blake         snprintf(namebuf, PATH_MAX, "r%s", fname);
230c1bb86cdSEric Blake     } else {
231c1bb86cdSEric Blake         snprintf(namebuf, PATH_MAX, "%.*s/r%s",
232c1bb86cdSEric Blake             (int)(dp - fname), fname, dp + 1);
233c1bb86cdSEric Blake     }
234c1bb86cdSEric Blake     *filename = namebuf;
235db0754dfSFam Zheng     warn_report("%s is a block device, using %s", fname, *filename);
236c1bb86cdSEric Blake 
237c1bb86cdSEric Blake     return 0;
238c1bb86cdSEric Blake }
239c1bb86cdSEric Blake #else
240db0754dfSFam Zheng static int raw_normalize_devicepath(const char **filename, Error **errp)
241c1bb86cdSEric Blake {
242c1bb86cdSEric Blake     return 0;
243c1bb86cdSEric Blake }
244c1bb86cdSEric Blake #endif
245c1bb86cdSEric Blake 
246c1bb86cdSEric Blake /*
247c1bb86cdSEric Blake  * Get logical block size via ioctl. On success store it in @sector_size_p.
248c1bb86cdSEric Blake  */
249c1bb86cdSEric Blake static int probe_logical_blocksize(int fd, unsigned int *sector_size_p)
250c1bb86cdSEric Blake {
251c1bb86cdSEric Blake     unsigned int sector_size;
252c1bb86cdSEric Blake     bool success = false;
253700f9ce0SPeter Maydell     int i;
254c1bb86cdSEric Blake 
255c1bb86cdSEric Blake     errno = ENOTSUP;
256700f9ce0SPeter Maydell     static const unsigned long ioctl_list[] = {
257c1bb86cdSEric Blake #ifdef BLKSSZGET
258700f9ce0SPeter Maydell         BLKSSZGET,
259c1bb86cdSEric Blake #endif
260c1bb86cdSEric Blake #ifdef DKIOCGETBLOCKSIZE
261700f9ce0SPeter Maydell         DKIOCGETBLOCKSIZE,
262c1bb86cdSEric Blake #endif
263c1bb86cdSEric Blake #ifdef DIOCGSECTORSIZE
264700f9ce0SPeter Maydell         DIOCGSECTORSIZE,
265700f9ce0SPeter Maydell #endif
266700f9ce0SPeter Maydell     };
267700f9ce0SPeter Maydell 
268700f9ce0SPeter Maydell     /* Try a few ioctls to get the right size */
269700f9ce0SPeter Maydell     for (i = 0; i < (int)ARRAY_SIZE(ioctl_list); i++) {
270700f9ce0SPeter Maydell         if (ioctl(fd, ioctl_list[i], &sector_size) >= 0) {
271c1bb86cdSEric Blake             *sector_size_p = sector_size;
272c1bb86cdSEric Blake             success = true;
273c1bb86cdSEric Blake         }
274700f9ce0SPeter Maydell     }
275c1bb86cdSEric Blake 
276c1bb86cdSEric Blake     return success ? 0 : -errno;
277c1bb86cdSEric Blake }
278c1bb86cdSEric Blake 
279c1bb86cdSEric Blake /**
280c1bb86cdSEric Blake  * Get physical block size of @fd.
281c1bb86cdSEric Blake  * On success, store it in @blk_size and return 0.
282c1bb86cdSEric Blake  * On failure, return -errno.
283c1bb86cdSEric Blake  */
284c1bb86cdSEric Blake static int probe_physical_blocksize(int fd, unsigned int *blk_size)
285c1bb86cdSEric Blake {
286c1bb86cdSEric Blake #ifdef BLKPBSZGET
287c1bb86cdSEric Blake     if (ioctl(fd, BLKPBSZGET, blk_size) < 0) {
288c1bb86cdSEric Blake         return -errno;
289c1bb86cdSEric Blake     }
290c1bb86cdSEric Blake     return 0;
291c1bb86cdSEric Blake #else
292c1bb86cdSEric Blake     return -ENOTSUP;
293c1bb86cdSEric Blake #endif
294c1bb86cdSEric Blake }
295c1bb86cdSEric Blake 
296c1bb86cdSEric Blake /* Check if read is allowed with given memory buffer and length.
297c1bb86cdSEric Blake  *
298c1bb86cdSEric Blake  * This function is used to check O_DIRECT memory buffer and request alignment.
299c1bb86cdSEric Blake  */
300c1bb86cdSEric Blake static bool raw_is_io_aligned(int fd, void *buf, size_t len)
301c1bb86cdSEric Blake {
302c1bb86cdSEric Blake     ssize_t ret = pread(fd, buf, len, 0);
303c1bb86cdSEric Blake 
304c1bb86cdSEric Blake     if (ret >= 0) {
305c1bb86cdSEric Blake         return true;
306c1bb86cdSEric Blake     }
307c1bb86cdSEric Blake 
308c1bb86cdSEric Blake #ifdef __linux__
309c1bb86cdSEric Blake     /* The Linux kernel returns EINVAL for misaligned O_DIRECT reads.  Ignore
310c1bb86cdSEric Blake      * other errors (e.g. real I/O error), which could happen on a failed
311c1bb86cdSEric Blake      * drive, since we only care about probing alignment.
312c1bb86cdSEric Blake      */
313c1bb86cdSEric Blake     if (errno != EINVAL) {
314c1bb86cdSEric Blake         return true;
315c1bb86cdSEric Blake     }
316c1bb86cdSEric Blake #endif
317c1bb86cdSEric Blake 
318c1bb86cdSEric Blake     return false;
319c1bb86cdSEric Blake }
320c1bb86cdSEric Blake 
321c1bb86cdSEric Blake static void raw_probe_alignment(BlockDriverState *bs, int fd, Error **errp)
322c1bb86cdSEric Blake {
323c1bb86cdSEric Blake     BDRVRawState *s = bs->opaque;
324c1bb86cdSEric Blake     char *buf;
325c1bb86cdSEric Blake     size_t max_align = MAX(MAX_BLOCKSIZE, getpagesize());
326a6b257a0SNir Soffer     size_t alignments[] = {1, 512, 1024, 2048, 4096};
327c1bb86cdSEric Blake 
328c1bb86cdSEric Blake     /* For SCSI generic devices the alignment is not really used.
329c1bb86cdSEric Blake        With buffered I/O, we don't have any restrictions. */
330c1bb86cdSEric Blake     if (bdrv_is_sg(bs) || !s->needs_alignment) {
331c1bb86cdSEric Blake         bs->bl.request_alignment = 1;
332c1bb86cdSEric Blake         s->buf_align = 1;
333c1bb86cdSEric Blake         return;
334c1bb86cdSEric Blake     }
335c1bb86cdSEric Blake 
336c1bb86cdSEric Blake     bs->bl.request_alignment = 0;
337c1bb86cdSEric Blake     s->buf_align = 0;
338c1bb86cdSEric Blake     /* Let's try to use the logical blocksize for the alignment. */
339c1bb86cdSEric Blake     if (probe_logical_blocksize(fd, &bs->bl.request_alignment) < 0) {
340c1bb86cdSEric Blake         bs->bl.request_alignment = 0;
341c1bb86cdSEric Blake     }
342c1bb86cdSEric Blake #ifdef CONFIG_XFS
343c1bb86cdSEric Blake     if (s->is_xfs) {
344c1bb86cdSEric Blake         struct dioattr da;
345c1bb86cdSEric Blake         if (xfsctl(NULL, fd, XFS_IOC_DIOINFO, &da) >= 0) {
346c1bb86cdSEric Blake             bs->bl.request_alignment = da.d_miniosz;
347c1bb86cdSEric Blake             /* The kernel returns wrong information for d_mem */
348c1bb86cdSEric Blake             /* s->buf_align = da.d_mem; */
349c1bb86cdSEric Blake         }
350c1bb86cdSEric Blake     }
351c1bb86cdSEric Blake #endif
352c1bb86cdSEric Blake 
353a6b257a0SNir Soffer     /*
354a6b257a0SNir Soffer      * If we could not get the sizes so far, we can only guess them. First try
355a6b257a0SNir Soffer      * to detect request alignment, since it is more likely to succeed. Then
356a6b257a0SNir Soffer      * try to detect buf_align, which cannot be detected in some cases (e.g.
357a6b257a0SNir Soffer      * Gluster). If buf_align cannot be detected, we fallback to the value of
358a6b257a0SNir Soffer      * request_alignment.
359a6b257a0SNir Soffer      */
360a6b257a0SNir Soffer 
361a6b257a0SNir Soffer     if (!bs->bl.request_alignment) {
362a6b257a0SNir Soffer         int i;
363c1bb86cdSEric Blake         size_t align;
364a6b257a0SNir Soffer         buf = qemu_memalign(max_align, max_align);
365a6b257a0SNir Soffer         for (i = 0; i < ARRAY_SIZE(alignments); i++) {
366a6b257a0SNir Soffer             align = alignments[i];
367a6b257a0SNir Soffer             if (raw_is_io_aligned(fd, buf, align)) {
368a6b257a0SNir Soffer                 /* Fallback to safe value. */
369a6b257a0SNir Soffer                 bs->bl.request_alignment = (align != 1) ? align : max_align;
370c1bb86cdSEric Blake                 break;
371c1bb86cdSEric Blake             }
372c1bb86cdSEric Blake         }
373c1bb86cdSEric Blake         qemu_vfree(buf);
374c1bb86cdSEric Blake     }
375c1bb86cdSEric Blake 
376a6b257a0SNir Soffer     if (!s->buf_align) {
377a6b257a0SNir Soffer         int i;
378c1bb86cdSEric Blake         size_t align;
379a6b257a0SNir Soffer         buf = qemu_memalign(max_align, 2 * max_align);
380a6b257a0SNir Soffer         for (i = 0; i < ARRAY_SIZE(alignments); i++) {
381a6b257a0SNir Soffer             align = alignments[i];
382a6b257a0SNir Soffer             if (raw_is_io_aligned(fd, buf + align, max_align)) {
383236094c7SStefan Hajnoczi                 /* Fallback to request_alignment. */
384a6b257a0SNir Soffer                 s->buf_align = (align != 1) ? align : bs->bl.request_alignment;
385c1bb86cdSEric Blake                 break;
386c1bb86cdSEric Blake             }
387c1bb86cdSEric Blake         }
388c1bb86cdSEric Blake         qemu_vfree(buf);
389c1bb86cdSEric Blake     }
390c1bb86cdSEric Blake 
391c1bb86cdSEric Blake     if (!s->buf_align || !bs->bl.request_alignment) {
392c1bb86cdSEric Blake         error_setg(errp, "Could not find working O_DIRECT alignment");
393c1bb86cdSEric Blake         error_append_hint(errp, "Try cache.direct=off\n");
394c1bb86cdSEric Blake     }
395c1bb86cdSEric Blake }
396c1bb86cdSEric Blake 
39723dece19SKevin Wolf static void raw_parse_flags(int bdrv_flags, int *open_flags, bool has_writers)
398c1bb86cdSEric Blake {
39923dece19SKevin Wolf     bool read_write = false;
400c1bb86cdSEric Blake     assert(open_flags != NULL);
401c1bb86cdSEric Blake 
402c1bb86cdSEric Blake     *open_flags |= O_BINARY;
403c1bb86cdSEric Blake     *open_flags &= ~O_ACCMODE;
40423dece19SKevin Wolf 
40523dece19SKevin Wolf     if (bdrv_flags & BDRV_O_AUTO_RDONLY) {
40623dece19SKevin Wolf         read_write = has_writers;
40723dece19SKevin Wolf     } else if (bdrv_flags & BDRV_O_RDWR) {
40823dece19SKevin Wolf         read_write = true;
40923dece19SKevin Wolf     }
41023dece19SKevin Wolf 
41123dece19SKevin Wolf     if (read_write) {
412c1bb86cdSEric Blake         *open_flags |= O_RDWR;
413c1bb86cdSEric Blake     } else {
414c1bb86cdSEric Blake         *open_flags |= O_RDONLY;
415c1bb86cdSEric Blake     }
416c1bb86cdSEric Blake 
417c1bb86cdSEric Blake     /* Use O_DSYNC for write-through caching, no flags for write-back caching,
418c1bb86cdSEric Blake      * and O_DIRECT for no caching. */
419c1bb86cdSEric Blake     if ((bdrv_flags & BDRV_O_NOCACHE)) {
420c1bb86cdSEric Blake         *open_flags |= O_DIRECT;
421c1bb86cdSEric Blake     }
422c1bb86cdSEric Blake }
423c1bb86cdSEric Blake 
424c1bb86cdSEric Blake static void raw_parse_filename(const char *filename, QDict *options,
425c1bb86cdSEric Blake                                Error **errp)
426c1bb86cdSEric Blake {
42703c320d8SMax Reitz     bdrv_parse_filename_strip_prefix(filename, "file:", options);
428c1bb86cdSEric Blake }
429c1bb86cdSEric Blake 
430c1bb86cdSEric Blake static QemuOptsList raw_runtime_opts = {
431c1bb86cdSEric Blake     .name = "raw",
432c1bb86cdSEric Blake     .head = QTAILQ_HEAD_INITIALIZER(raw_runtime_opts.head),
433c1bb86cdSEric Blake     .desc = {
434c1bb86cdSEric Blake         {
435c1bb86cdSEric Blake             .name = "filename",
436c1bb86cdSEric Blake             .type = QEMU_OPT_STRING,
437c1bb86cdSEric Blake             .help = "File name of the image",
438c1bb86cdSEric Blake         },
439c1bb86cdSEric Blake         {
440c1bb86cdSEric Blake             .name = "aio",
441c1bb86cdSEric Blake             .type = QEMU_OPT_STRING,
442c1bb86cdSEric Blake             .help = "host AIO implementation (threads, native)",
443c1bb86cdSEric Blake         },
44416b48d5dSFam Zheng         {
44516b48d5dSFam Zheng             .name = "locking",
44616b48d5dSFam Zheng             .type = QEMU_OPT_STRING,
44716b48d5dSFam Zheng             .help = "file locking mode (on/off/auto, default: auto)",
44816b48d5dSFam Zheng         },
4497c9e5276SPaolo Bonzini         {
4507c9e5276SPaolo Bonzini             .name = "pr-manager",
4517c9e5276SPaolo Bonzini             .type = QEMU_OPT_STRING,
4527c9e5276SPaolo Bonzini             .help = "id of persistent reservation manager object (default: none)",
4537c9e5276SPaolo Bonzini         },
454f357fcd8SStefan Hajnoczi #if defined(__linux__)
455f357fcd8SStefan Hajnoczi         {
456f357fcd8SStefan Hajnoczi             .name = "drop-cache",
457f357fcd8SStefan Hajnoczi             .type = QEMU_OPT_BOOL,
458f357fcd8SStefan Hajnoczi             .help = "invalidate page cache during live migration (default: on)",
459f357fcd8SStefan Hajnoczi         },
460f357fcd8SStefan Hajnoczi #endif
46131be8a2aSStefan Hajnoczi         {
46231be8a2aSStefan Hajnoczi             .name = "x-check-cache-dropped",
46331be8a2aSStefan Hajnoczi             .type = QEMU_OPT_BOOL,
46431be8a2aSStefan Hajnoczi             .help = "check that page cache was dropped on live migration (default: off)"
46531be8a2aSStefan Hajnoczi         },
466c1bb86cdSEric Blake         { /* end of list */ }
467c1bb86cdSEric Blake     },
468c1bb86cdSEric Blake };
469c1bb86cdSEric Blake 
4708a2ce0bcSAlberto Garcia static const char *const mutable_opts[] = { "x-check-cache-dropped", NULL };
4718a2ce0bcSAlberto Garcia 
472c1bb86cdSEric Blake static int raw_open_common(BlockDriverState *bs, QDict *options,
473230ff739SJohn Snow                            int bdrv_flags, int open_flags,
474230ff739SJohn Snow                            bool device, Error **errp)
475c1bb86cdSEric Blake {
476c1bb86cdSEric Blake     BDRVRawState *s = bs->opaque;
477c1bb86cdSEric Blake     QemuOpts *opts;
478c1bb86cdSEric Blake     Error *local_err = NULL;
479c1bb86cdSEric Blake     const char *filename = NULL;
4807c9e5276SPaolo Bonzini     const char *str;
481c1bb86cdSEric Blake     BlockdevAioOptions aio, aio_default;
482c1bb86cdSEric Blake     int fd, ret;
483c1bb86cdSEric Blake     struct stat st;
484244a5668SFam Zheng     OnOffAuto locking;
485c1bb86cdSEric Blake 
486c1bb86cdSEric Blake     opts = qemu_opts_create(&raw_runtime_opts, NULL, 0, &error_abort);
487c1bb86cdSEric Blake     qemu_opts_absorb_qdict(opts, options, &local_err);
488c1bb86cdSEric Blake     if (local_err) {
489c1bb86cdSEric Blake         error_propagate(errp, local_err);
490c1bb86cdSEric Blake         ret = -EINVAL;
491c1bb86cdSEric Blake         goto fail;
492c1bb86cdSEric Blake     }
493c1bb86cdSEric Blake 
494c1bb86cdSEric Blake     filename = qemu_opt_get(opts, "filename");
495c1bb86cdSEric Blake 
496db0754dfSFam Zheng     ret = raw_normalize_devicepath(&filename, errp);
497c1bb86cdSEric Blake     if (ret != 0) {
498c1bb86cdSEric Blake         goto fail;
499c1bb86cdSEric Blake     }
500c1bb86cdSEric Blake 
501c1bb86cdSEric Blake     aio_default = (bdrv_flags & BDRV_O_NATIVE_AIO)
502c1bb86cdSEric Blake                   ? BLOCKDEV_AIO_OPTIONS_NATIVE
503c1bb86cdSEric Blake                   : BLOCKDEV_AIO_OPTIONS_THREADS;
504f7abe0ecSMarc-André Lureau     aio = qapi_enum_parse(&BlockdevAioOptions_lookup,
505f7abe0ecSMarc-André Lureau                           qemu_opt_get(opts, "aio"),
50606c60b6cSMarkus Armbruster                           aio_default, &local_err);
507c1bb86cdSEric Blake     if (local_err) {
508c1bb86cdSEric Blake         error_propagate(errp, local_err);
509c1bb86cdSEric Blake         ret = -EINVAL;
510c1bb86cdSEric Blake         goto fail;
511c1bb86cdSEric Blake     }
512c1bb86cdSEric Blake     s->use_linux_aio = (aio == BLOCKDEV_AIO_OPTIONS_NATIVE);
513c1bb86cdSEric Blake 
514f7abe0ecSMarc-André Lureau     locking = qapi_enum_parse(&OnOffAuto_lookup,
515f7abe0ecSMarc-André Lureau                               qemu_opt_get(opts, "locking"),
51606c60b6cSMarkus Armbruster                               ON_OFF_AUTO_AUTO, &local_err);
517244a5668SFam Zheng     if (local_err) {
518244a5668SFam Zheng         error_propagate(errp, local_err);
519244a5668SFam Zheng         ret = -EINVAL;
520244a5668SFam Zheng         goto fail;
521244a5668SFam Zheng     }
522244a5668SFam Zheng     switch (locking) {
523244a5668SFam Zheng     case ON_OFF_AUTO_ON:
524244a5668SFam Zheng         s->use_lock = true;
5252b218f5dSFam Zheng         if (!qemu_has_ofd_lock()) {
526db0754dfSFam Zheng             warn_report("File lock requested but OFD locking syscall is "
527db0754dfSFam Zheng                         "unavailable, falling back to POSIX file locks");
528db0754dfSFam Zheng             error_printf("Due to the implementation, locks can be lost "
5292b218f5dSFam Zheng                          "unexpectedly.\n");
5302b218f5dSFam Zheng         }
531244a5668SFam Zheng         break;
532244a5668SFam Zheng     case ON_OFF_AUTO_OFF:
533244a5668SFam Zheng         s->use_lock = false;
534244a5668SFam Zheng         break;
535244a5668SFam Zheng     case ON_OFF_AUTO_AUTO:
5362b218f5dSFam Zheng         s->use_lock = qemu_has_ofd_lock();
537244a5668SFam Zheng         break;
538244a5668SFam Zheng     default:
539244a5668SFam Zheng         abort();
540244a5668SFam Zheng     }
541244a5668SFam Zheng 
5427c9e5276SPaolo Bonzini     str = qemu_opt_get(opts, "pr-manager");
5437c9e5276SPaolo Bonzini     if (str) {
5447c9e5276SPaolo Bonzini         s->pr_mgr = pr_manager_lookup(str, &local_err);
5457c9e5276SPaolo Bonzini         if (local_err) {
5467c9e5276SPaolo Bonzini             error_propagate(errp, local_err);
5477c9e5276SPaolo Bonzini             ret = -EINVAL;
5487c9e5276SPaolo Bonzini             goto fail;
5497c9e5276SPaolo Bonzini         }
5507c9e5276SPaolo Bonzini     }
5517c9e5276SPaolo Bonzini 
552f357fcd8SStefan Hajnoczi     s->drop_cache = qemu_opt_get_bool(opts, "drop-cache", true);
55331be8a2aSStefan Hajnoczi     s->check_cache_dropped = qemu_opt_get_bool(opts, "x-check-cache-dropped",
55431be8a2aSStefan Hajnoczi                                                false);
55531be8a2aSStefan Hajnoczi 
556c1bb86cdSEric Blake     s->open_flags = open_flags;
55723dece19SKevin Wolf     raw_parse_flags(bdrv_flags, &s->open_flags, false);
558c1bb86cdSEric Blake 
559c1bb86cdSEric Blake     s->fd = -1;
560c1bb86cdSEric Blake     fd = qemu_open(filename, s->open_flags, 0644);
56164107dc0SKevin Wolf     ret = fd < 0 ? -errno : 0;
56264107dc0SKevin Wolf 
56364107dc0SKevin Wolf     if (ret < 0) {
564f6fc1e30SPaolo Bonzini         error_setg_file_open(errp, -ret, filename);
565c1bb86cdSEric Blake         if (ret == -EROFS) {
566c1bb86cdSEric Blake             ret = -EACCES;
567c1bb86cdSEric Blake         }
568c1bb86cdSEric Blake         goto fail;
569c1bb86cdSEric Blake     }
570c1bb86cdSEric Blake     s->fd = fd;
571c1bb86cdSEric Blake 
572244a5668SFam Zheng     s->perm = 0;
573244a5668SFam Zheng     s->shared_perm = BLK_PERM_ALL;
574244a5668SFam Zheng 
575c1bb86cdSEric Blake #ifdef CONFIG_LINUX_AIO
576c1bb86cdSEric Blake      /* Currently Linux does AIO only for files opened with O_DIRECT */
577ed6e2161SNishanth Aravamudan     if (s->use_linux_aio) {
578ed6e2161SNishanth Aravamudan         if (!(s->open_flags & O_DIRECT)) {
579c1bb86cdSEric Blake             error_setg(errp, "aio=native was specified, but it requires "
580c1bb86cdSEric Blake                              "cache.direct=on, which was not specified.");
581c1bb86cdSEric Blake             ret = -EINVAL;
582c1bb86cdSEric Blake             goto fail;
583c1bb86cdSEric Blake         }
584ed6e2161SNishanth Aravamudan         if (!aio_setup_linux_aio(bdrv_get_aio_context(bs), errp)) {
585ed6e2161SNishanth Aravamudan             error_prepend(errp, "Unable to use native AIO: ");
586ed6e2161SNishanth Aravamudan             goto fail;
587ed6e2161SNishanth Aravamudan         }
588ed6e2161SNishanth Aravamudan     }
589c1bb86cdSEric Blake #else
590c1bb86cdSEric Blake     if (s->use_linux_aio) {
591c1bb86cdSEric Blake         error_setg(errp, "aio=native was specified, but is not supported "
592c1bb86cdSEric Blake                          "in this build.");
593c1bb86cdSEric Blake         ret = -EINVAL;
594c1bb86cdSEric Blake         goto fail;
595c1bb86cdSEric Blake     }
596c1bb86cdSEric Blake #endif /* !defined(CONFIG_LINUX_AIO) */
597c1bb86cdSEric Blake 
598c1bb86cdSEric Blake     s->has_discard = true;
599c1bb86cdSEric Blake     s->has_write_zeroes = true;
600c1bb86cdSEric Blake     if ((bs->open_flags & BDRV_O_NOCACHE) != 0) {
601c1bb86cdSEric Blake         s->needs_alignment = true;
602c1bb86cdSEric Blake     }
603c1bb86cdSEric Blake 
604c1bb86cdSEric Blake     if (fstat(s->fd, &st) < 0) {
605c1bb86cdSEric Blake         ret = -errno;
606c1bb86cdSEric Blake         error_setg_errno(errp, errno, "Could not stat file");
607c1bb86cdSEric Blake         goto fail;
608c1bb86cdSEric Blake     }
609230ff739SJohn Snow 
610230ff739SJohn Snow     if (!device) {
611230ff739SJohn Snow         if (S_ISBLK(st.st_mode)) {
612230ff739SJohn Snow             warn_report("Opening a block device as a file using the '%s' "
613230ff739SJohn Snow                         "driver is deprecated", bs->drv->format_name);
614230ff739SJohn Snow         } else if (S_ISCHR(st.st_mode)) {
615230ff739SJohn Snow             warn_report("Opening a character device as a file using the '%s' "
616230ff739SJohn Snow                         "driver is deprecated", bs->drv->format_name);
617230ff739SJohn Snow         } else if (!S_ISREG(st.st_mode)) {
618230ff739SJohn Snow             error_setg(errp, "A regular file was expected by the '%s' driver, "
619230ff739SJohn Snow                        "but something else was given", bs->drv->format_name);
620230ff739SJohn Snow             ret = -EINVAL;
621230ff739SJohn Snow             goto fail;
622230ff739SJohn Snow         } else {
623c1bb86cdSEric Blake             s->discard_zeroes = true;
624c1bb86cdSEric Blake             s->has_fallocate = true;
625c1bb86cdSEric Blake         }
626230ff739SJohn Snow     } else {
627230ff739SJohn Snow         if (!(S_ISCHR(st.st_mode) || S_ISBLK(st.st_mode))) {
628230ff739SJohn Snow             error_setg(errp, "'%s' driver expects either "
629230ff739SJohn Snow                        "a character or block device", bs->drv->format_name);
630230ff739SJohn Snow             ret = -EINVAL;
631230ff739SJohn Snow             goto fail;
632230ff739SJohn Snow         }
633230ff739SJohn Snow     }
634230ff739SJohn Snow 
635c1bb86cdSEric Blake     if (S_ISBLK(st.st_mode)) {
636c1bb86cdSEric Blake #ifdef BLKDISCARDZEROES
637c1bb86cdSEric Blake         unsigned int arg;
638c1bb86cdSEric Blake         if (ioctl(s->fd, BLKDISCARDZEROES, &arg) == 0 && arg) {
639c1bb86cdSEric Blake             s->discard_zeroes = true;
640c1bb86cdSEric Blake         }
641c1bb86cdSEric Blake #endif
642c1bb86cdSEric Blake #ifdef __linux__
643c1bb86cdSEric Blake         /* On Linux 3.10, BLKDISCARD leaves stale data in the page cache.  Do
644c1bb86cdSEric Blake          * not rely on the contents of discarded blocks unless using O_DIRECT.
645c1bb86cdSEric Blake          * Same for BLKZEROOUT.
646c1bb86cdSEric Blake          */
647c1bb86cdSEric Blake         if (!(bs->open_flags & BDRV_O_NOCACHE)) {
648c1bb86cdSEric Blake             s->discard_zeroes = false;
649c1bb86cdSEric Blake             s->has_write_zeroes = false;
650c1bb86cdSEric Blake         }
651c1bb86cdSEric Blake #endif
652c1bb86cdSEric Blake     }
653c1bb86cdSEric Blake #ifdef __FreeBSD__
654c1bb86cdSEric Blake     if (S_ISCHR(st.st_mode)) {
655c1bb86cdSEric Blake         /*
656c1bb86cdSEric Blake          * The file is a char device (disk), which on FreeBSD isn't behind
657c1bb86cdSEric Blake          * a pager, so force all requests to be aligned. This is needed
658c1bb86cdSEric Blake          * so QEMU makes sure all IO operations on the device are aligned
659c1bb86cdSEric Blake          * to sector size, or else FreeBSD will reject them with EINVAL.
660c1bb86cdSEric Blake          */
661c1bb86cdSEric Blake         s->needs_alignment = true;
662c1bb86cdSEric Blake     }
663c1bb86cdSEric Blake #endif
664c1bb86cdSEric Blake 
665c1bb86cdSEric Blake #ifdef CONFIG_XFS
666c1bb86cdSEric Blake     if (platform_test_xfs_fd(s->fd)) {
667c1bb86cdSEric Blake         s->is_xfs = true;
668c1bb86cdSEric Blake     }
669c1bb86cdSEric Blake #endif
670c1bb86cdSEric Blake 
671738301e1SKevin Wolf     bs->supported_zero_flags = BDRV_REQ_MAY_UNMAP | BDRV_REQ_NO_FALLBACK;
672c1bb86cdSEric Blake     ret = 0;
673c1bb86cdSEric Blake fail:
674c1bb86cdSEric Blake     if (filename && (bdrv_flags & BDRV_O_TEMPORARY)) {
675c1bb86cdSEric Blake         unlink(filename);
676c1bb86cdSEric Blake     }
677c1bb86cdSEric Blake     qemu_opts_del(opts);
678c1bb86cdSEric Blake     return ret;
679c1bb86cdSEric Blake }
680c1bb86cdSEric Blake 
681c1bb86cdSEric Blake static int raw_open(BlockDriverState *bs, QDict *options, int flags,
682c1bb86cdSEric Blake                     Error **errp)
683c1bb86cdSEric Blake {
684c1bb86cdSEric Blake     BDRVRawState *s = bs->opaque;
685c1bb86cdSEric Blake 
686c1bb86cdSEric Blake     s->type = FTYPE_FILE;
687230ff739SJohn Snow     return raw_open_common(bs, options, flags, 0, false, errp);
688c1bb86cdSEric Blake }
689c1bb86cdSEric Blake 
690244a5668SFam Zheng typedef enum {
691244a5668SFam Zheng     RAW_PL_PREPARE,
692244a5668SFam Zheng     RAW_PL_COMMIT,
693244a5668SFam Zheng     RAW_PL_ABORT,
694244a5668SFam Zheng } RawPermLockOp;
695244a5668SFam Zheng 
696244a5668SFam Zheng #define PERM_FOREACH(i) \
697244a5668SFam Zheng     for ((i) = 0; (1ULL << (i)) <= BLK_PERM_ALL; i++)
698244a5668SFam Zheng 
699244a5668SFam Zheng /* Lock bytes indicated by @perm_lock_bits and @shared_perm_lock_bits in the
700244a5668SFam Zheng  * file; if @unlock == true, also unlock the unneeded bytes.
701244a5668SFam Zheng  * @shared_perm_lock_bits is the mask of all permissions that are NOT shared.
702244a5668SFam Zheng  */
7032996ffadSFam Zheng static int raw_apply_lock_bytes(BDRVRawState *s, int fd,
704244a5668SFam Zheng                                 uint64_t perm_lock_bits,
705244a5668SFam Zheng                                 uint64_t shared_perm_lock_bits,
706244a5668SFam Zheng                                 bool unlock, Error **errp)
707244a5668SFam Zheng {
708244a5668SFam Zheng     int ret;
709244a5668SFam Zheng     int i;
7102996ffadSFam Zheng     uint64_t locked_perm, locked_shared_perm;
7112996ffadSFam Zheng 
7122996ffadSFam Zheng     if (s) {
7132996ffadSFam Zheng         locked_perm = s->locked_perm;
7142996ffadSFam Zheng         locked_shared_perm = s->locked_shared_perm;
7152996ffadSFam Zheng     } else {
7162996ffadSFam Zheng         /*
7172996ffadSFam Zheng          * We don't have the previous bits, just lock/unlock for each of the
7182996ffadSFam Zheng          * requested bits.
7192996ffadSFam Zheng          */
7202996ffadSFam Zheng         if (unlock) {
7212996ffadSFam Zheng             locked_perm = BLK_PERM_ALL;
7222996ffadSFam Zheng             locked_shared_perm = BLK_PERM_ALL;
7232996ffadSFam Zheng         } else {
7242996ffadSFam Zheng             locked_perm = 0;
7252996ffadSFam Zheng             locked_shared_perm = 0;
7262996ffadSFam Zheng         }
7272996ffadSFam Zheng     }
728244a5668SFam Zheng 
729244a5668SFam Zheng     PERM_FOREACH(i) {
730244a5668SFam Zheng         int off = RAW_LOCK_PERM_BASE + i;
7312996ffadSFam Zheng         uint64_t bit = (1ULL << i);
7322996ffadSFam Zheng         if ((perm_lock_bits & bit) && !(locked_perm & bit)) {
733d0a96155SMax Reitz             ret = qemu_lock_fd(fd, off, 1, false);
734244a5668SFam Zheng             if (ret) {
735244a5668SFam Zheng                 error_setg(errp, "Failed to lock byte %d", off);
736244a5668SFam Zheng                 return ret;
7372996ffadSFam Zheng             } else if (s) {
7382996ffadSFam Zheng                 s->locked_perm |= bit;
739244a5668SFam Zheng             }
7402996ffadSFam Zheng         } else if (unlock && (locked_perm & bit) && !(perm_lock_bits & bit)) {
741d0a96155SMax Reitz             ret = qemu_unlock_fd(fd, off, 1);
742244a5668SFam Zheng             if (ret) {
743244a5668SFam Zheng                 error_setg(errp, "Failed to unlock byte %d", off);
744244a5668SFam Zheng                 return ret;
7452996ffadSFam Zheng             } else if (s) {
7462996ffadSFam Zheng                 s->locked_perm &= ~bit;
747244a5668SFam Zheng             }
748244a5668SFam Zheng         }
749244a5668SFam Zheng     }
750244a5668SFam Zheng     PERM_FOREACH(i) {
751244a5668SFam Zheng         int off = RAW_LOCK_SHARED_BASE + i;
7522996ffadSFam Zheng         uint64_t bit = (1ULL << i);
7532996ffadSFam Zheng         if ((shared_perm_lock_bits & bit) && !(locked_shared_perm & bit)) {
754d0a96155SMax Reitz             ret = qemu_lock_fd(fd, off, 1, false);
755244a5668SFam Zheng             if (ret) {
756244a5668SFam Zheng                 error_setg(errp, "Failed to lock byte %d", off);
757244a5668SFam Zheng                 return ret;
7582996ffadSFam Zheng             } else if (s) {
7592996ffadSFam Zheng                 s->locked_shared_perm |= bit;
760244a5668SFam Zheng             }
7612996ffadSFam Zheng         } else if (unlock && (locked_shared_perm & bit) &&
7622996ffadSFam Zheng                    !(shared_perm_lock_bits & bit)) {
763d0a96155SMax Reitz             ret = qemu_unlock_fd(fd, off, 1);
764244a5668SFam Zheng             if (ret) {
765244a5668SFam Zheng                 error_setg(errp, "Failed to unlock byte %d", off);
766244a5668SFam Zheng                 return ret;
7672996ffadSFam Zheng             } else if (s) {
7682996ffadSFam Zheng                 s->locked_shared_perm &= ~bit;
769244a5668SFam Zheng             }
770244a5668SFam Zheng         }
771244a5668SFam Zheng     }
772244a5668SFam Zheng     return 0;
773244a5668SFam Zheng }
774244a5668SFam Zheng 
775244a5668SFam Zheng /* Check "unshared" bytes implied by @perm and ~@shared_perm in the file. */
776d0a96155SMax Reitz static int raw_check_lock_bytes(int fd, uint64_t perm, uint64_t shared_perm,
777244a5668SFam Zheng                                 Error **errp)
778244a5668SFam Zheng {
779244a5668SFam Zheng     int ret;
780244a5668SFam Zheng     int i;
781244a5668SFam Zheng 
782244a5668SFam Zheng     PERM_FOREACH(i) {
783244a5668SFam Zheng         int off = RAW_LOCK_SHARED_BASE + i;
784244a5668SFam Zheng         uint64_t p = 1ULL << i;
785244a5668SFam Zheng         if (perm & p) {
786d0a96155SMax Reitz             ret = qemu_lock_fd_test(fd, off, 1, true);
787244a5668SFam Zheng             if (ret) {
788244a5668SFam Zheng                 char *perm_name = bdrv_perm_names(p);
789244a5668SFam Zheng                 error_setg(errp,
790244a5668SFam Zheng                            "Failed to get \"%s\" lock",
791244a5668SFam Zheng                            perm_name);
792244a5668SFam Zheng                 g_free(perm_name);
793244a5668SFam Zheng                 return ret;
794244a5668SFam Zheng             }
795244a5668SFam Zheng         }
796244a5668SFam Zheng     }
797244a5668SFam Zheng     PERM_FOREACH(i) {
798244a5668SFam Zheng         int off = RAW_LOCK_PERM_BASE + i;
799244a5668SFam Zheng         uint64_t p = 1ULL << i;
800244a5668SFam Zheng         if (!(shared_perm & p)) {
801d0a96155SMax Reitz             ret = qemu_lock_fd_test(fd, off, 1, true);
802244a5668SFam Zheng             if (ret) {
803244a5668SFam Zheng                 char *perm_name = bdrv_perm_names(p);
804244a5668SFam Zheng                 error_setg(errp,
805244a5668SFam Zheng                            "Failed to get shared \"%s\" lock",
806244a5668SFam Zheng                            perm_name);
807244a5668SFam Zheng                 g_free(perm_name);
808244a5668SFam Zheng                 return ret;
809244a5668SFam Zheng             }
810244a5668SFam Zheng         }
811244a5668SFam Zheng     }
812244a5668SFam Zheng     return 0;
813244a5668SFam Zheng }
814244a5668SFam Zheng 
815244a5668SFam Zheng static int raw_handle_perm_lock(BlockDriverState *bs,
816244a5668SFam Zheng                                 RawPermLockOp op,
817244a5668SFam Zheng                                 uint64_t new_perm, uint64_t new_shared,
818244a5668SFam Zheng                                 Error **errp)
819244a5668SFam Zheng {
820244a5668SFam Zheng     BDRVRawState *s = bs->opaque;
821244a5668SFam Zheng     int ret = 0;
822244a5668SFam Zheng     Error *local_err = NULL;
823244a5668SFam Zheng 
824244a5668SFam Zheng     if (!s->use_lock) {
825244a5668SFam Zheng         return 0;
826244a5668SFam Zheng     }
827244a5668SFam Zheng 
828244a5668SFam Zheng     if (bdrv_get_flags(bs) & BDRV_O_INACTIVE) {
829244a5668SFam Zheng         return 0;
830244a5668SFam Zheng     }
831244a5668SFam Zheng 
832244a5668SFam Zheng     switch (op) {
833244a5668SFam Zheng     case RAW_PL_PREPARE:
834696aaaedSVladimir Sementsov-Ogievskiy         if ((s->perm | new_perm) == s->perm &&
835696aaaedSVladimir Sementsov-Ogievskiy             (s->shared_perm & new_shared) == s->shared_perm)
836696aaaedSVladimir Sementsov-Ogievskiy         {
837696aaaedSVladimir Sementsov-Ogievskiy             /*
838696aaaedSVladimir Sementsov-Ogievskiy              * We are going to unlock bytes, it should not fail. If it fail due
839696aaaedSVladimir Sementsov-Ogievskiy              * to some fs-dependent permission-unrelated reasons (which occurs
840696aaaedSVladimir Sementsov-Ogievskiy              * sometimes on NFS and leads to abort in bdrv_replace_child) we
841696aaaedSVladimir Sementsov-Ogievskiy              * can't prevent such errors by any check here. And we ignore them
842696aaaedSVladimir Sementsov-Ogievskiy              * anyway in ABORT and COMMIT.
843696aaaedSVladimir Sementsov-Ogievskiy              */
844696aaaedSVladimir Sementsov-Ogievskiy             return 0;
845696aaaedSVladimir Sementsov-Ogievskiy         }
846f2e3af29SFam Zheng         ret = raw_apply_lock_bytes(s, s->fd, s->perm | new_perm,
847244a5668SFam Zheng                                    ~s->shared_perm | ~new_shared,
848244a5668SFam Zheng                                    false, errp);
849244a5668SFam Zheng         if (!ret) {
850f2e3af29SFam Zheng             ret = raw_check_lock_bytes(s->fd, new_perm, new_shared, errp);
851244a5668SFam Zheng             if (!ret) {
852244a5668SFam Zheng                 return 0;
853244a5668SFam Zheng             }
854b857431dSFam Zheng             error_append_hint(errp,
855b857431dSFam Zheng                               "Is another process using the image [%s]?\n",
856b857431dSFam Zheng                               bs->filename);
857244a5668SFam Zheng         }
858244a5668SFam Zheng         op = RAW_PL_ABORT;
859244a5668SFam Zheng         /* fall through to unlock bytes. */
860244a5668SFam Zheng     case RAW_PL_ABORT:
861f2e3af29SFam Zheng         raw_apply_lock_bytes(s, s->fd, s->perm, ~s->shared_perm,
862d0a96155SMax Reitz                              true, &local_err);
863244a5668SFam Zheng         if (local_err) {
864244a5668SFam Zheng             /* Theoretically the above call only unlocks bytes and it cannot
865244a5668SFam Zheng              * fail. Something weird happened, report it.
866244a5668SFam Zheng              */
867db0754dfSFam Zheng             warn_report_err(local_err);
868244a5668SFam Zheng         }
869244a5668SFam Zheng         break;
870244a5668SFam Zheng     case RAW_PL_COMMIT:
871f2e3af29SFam Zheng         raw_apply_lock_bytes(s, s->fd, new_perm, ~new_shared,
872d0a96155SMax Reitz                              true, &local_err);
873244a5668SFam Zheng         if (local_err) {
874244a5668SFam Zheng             /* Theoretically the above call only unlocks bytes and it cannot
875244a5668SFam Zheng              * fail. Something weird happened, report it.
876244a5668SFam Zheng              */
877db0754dfSFam Zheng             warn_report_err(local_err);
878244a5668SFam Zheng         }
879244a5668SFam Zheng         break;
880244a5668SFam Zheng     }
881244a5668SFam Zheng     return ret;
882244a5668SFam Zheng }
883244a5668SFam Zheng 
8845cec2870SKevin Wolf static int raw_reconfigure_getfd(BlockDriverState *bs, int flags,
88523dece19SKevin Wolf                                  int *open_flags, uint64_t perm, bool force_dup,
8866ceabe6fSKevin Wolf                                  Error **errp)
8875cec2870SKevin Wolf {
8885cec2870SKevin Wolf     BDRVRawState *s = bs->opaque;
8895cec2870SKevin Wolf     int fd = -1;
8905cec2870SKevin Wolf     int ret;
89123dece19SKevin Wolf     bool has_writers = perm &
89223dece19SKevin Wolf         (BLK_PERM_WRITE | BLK_PERM_WRITE_UNCHANGED | BLK_PERM_RESIZE);
8935cec2870SKevin Wolf     int fcntl_flags = O_APPEND | O_NONBLOCK;
8945cec2870SKevin Wolf #ifdef O_NOATIME
8955cec2870SKevin Wolf     fcntl_flags |= O_NOATIME;
8965cec2870SKevin Wolf #endif
8975cec2870SKevin Wolf 
8985cec2870SKevin Wolf     *open_flags = 0;
8995cec2870SKevin Wolf     if (s->type == FTYPE_CD) {
9005cec2870SKevin Wolf         *open_flags |= O_NONBLOCK;
9015cec2870SKevin Wolf     }
9025cec2870SKevin Wolf 
90323dece19SKevin Wolf     raw_parse_flags(flags, open_flags, has_writers);
9045cec2870SKevin Wolf 
9055cec2870SKevin Wolf #ifdef O_ASYNC
9065cec2870SKevin Wolf     /* Not all operating systems have O_ASYNC, and those that don't
9075cec2870SKevin Wolf      * will not let us track the state into rs->open_flags (typically
9085cec2870SKevin Wolf      * you achieve the same effect with an ioctl, for example I_SETSIG
9095cec2870SKevin Wolf      * on Solaris). But we do not use O_ASYNC, so that's fine.
9105cec2870SKevin Wolf      */
9115cec2870SKevin Wolf     assert((s->open_flags & O_ASYNC) == 0);
9125cec2870SKevin Wolf #endif
9135cec2870SKevin Wolf 
9146ceabe6fSKevin Wolf     if (!force_dup && *open_flags == s->open_flags) {
9156ceabe6fSKevin Wolf         /* We're lucky, the existing fd is fine */
9166ceabe6fSKevin Wolf         return s->fd;
9176ceabe6fSKevin Wolf     }
9186ceabe6fSKevin Wolf 
9195cec2870SKevin Wolf     if ((*open_flags & ~fcntl_flags) == (s->open_flags & ~fcntl_flags)) {
9205cec2870SKevin Wolf         /* dup the original fd */
9215cec2870SKevin Wolf         fd = qemu_dup(s->fd);
9225cec2870SKevin Wolf         if (fd >= 0) {
9235cec2870SKevin Wolf             ret = fcntl_setfl(fd, *open_flags);
9245cec2870SKevin Wolf             if (ret) {
9255cec2870SKevin Wolf                 qemu_close(fd);
9265cec2870SKevin Wolf                 fd = -1;
9275cec2870SKevin Wolf             }
9285cec2870SKevin Wolf         }
9295cec2870SKevin Wolf     }
9305cec2870SKevin Wolf 
9315cec2870SKevin Wolf     /* If we cannot use fcntl, or fcntl failed, fall back to qemu_open() */
9325cec2870SKevin Wolf     if (fd == -1) {
9335cec2870SKevin Wolf         const char *normalized_filename = bs->filename;
9345cec2870SKevin Wolf         ret = raw_normalize_devicepath(&normalized_filename, errp);
9355cec2870SKevin Wolf         if (ret >= 0) {
9365cec2870SKevin Wolf             assert(!(*open_flags & O_CREAT));
9375cec2870SKevin Wolf             fd = qemu_open(normalized_filename, *open_flags);
9385cec2870SKevin Wolf             if (fd == -1) {
9395cec2870SKevin Wolf                 error_setg_errno(errp, errno, "Could not reopen file");
9405cec2870SKevin Wolf                 return -1;
9415cec2870SKevin Wolf             }
9425cec2870SKevin Wolf         }
9435cec2870SKevin Wolf     }
9445cec2870SKevin Wolf 
9455cec2870SKevin Wolf     return fd;
9465cec2870SKevin Wolf }
9475cec2870SKevin Wolf 
948c1bb86cdSEric Blake static int raw_reopen_prepare(BDRVReopenState *state,
949c1bb86cdSEric Blake                               BlockReopenQueue *queue, Error **errp)
950c1bb86cdSEric Blake {
951c1bb86cdSEric Blake     BDRVRawState *s;
952c1bb86cdSEric Blake     BDRVRawReopenState *rs;
95331be8a2aSStefan Hajnoczi     QemuOpts *opts;
954a6aeca0cSKevin Wolf     int ret;
955c1bb86cdSEric Blake     Error *local_err = NULL;
956c1bb86cdSEric Blake 
957c1bb86cdSEric Blake     assert(state != NULL);
958c1bb86cdSEric Blake     assert(state->bs != NULL);
959c1bb86cdSEric Blake 
960c1bb86cdSEric Blake     s = state->bs->opaque;
961c1bb86cdSEric Blake 
962c1bb86cdSEric Blake     state->opaque = g_new0(BDRVRawReopenState, 1);
963c1bb86cdSEric Blake     rs = state->opaque;
96431be8a2aSStefan Hajnoczi 
96531be8a2aSStefan Hajnoczi     /* Handle options changes */
96631be8a2aSStefan Hajnoczi     opts = qemu_opts_create(&raw_runtime_opts, NULL, 0, &error_abort);
96731be8a2aSStefan Hajnoczi     qemu_opts_absorb_qdict(opts, state->options, &local_err);
96831be8a2aSStefan Hajnoczi     if (local_err) {
96931be8a2aSStefan Hajnoczi         error_propagate(errp, local_err);
97031be8a2aSStefan Hajnoczi         ret = -EINVAL;
97131be8a2aSStefan Hajnoczi         goto out;
97231be8a2aSStefan Hajnoczi     }
97331be8a2aSStefan Hajnoczi 
974f357fcd8SStefan Hajnoczi     rs->drop_cache = qemu_opt_get_bool_del(opts, "drop-cache", true);
9758d324575SAlberto Garcia     rs->check_cache_dropped =
9768d324575SAlberto Garcia         qemu_opt_get_bool_del(opts, "x-check-cache-dropped", false);
9778d324575SAlberto Garcia 
9788d324575SAlberto Garcia     /* This driver's reopen function doesn't currently allow changing
9798d324575SAlberto Garcia      * other options, so let's put them back in the original QDict and
9808d324575SAlberto Garcia      * bdrv_reopen_prepare() will detect changes and complain. */
9818d324575SAlberto Garcia     qemu_opts_to_qdict(opts, state->options);
982c1bb86cdSEric Blake 
9835cec2870SKevin Wolf     rs->fd = raw_reconfigure_getfd(state->bs, state->flags, &rs->open_flags,
98423dece19SKevin Wolf                                    state->perm, true, &local_err);
9855cec2870SKevin Wolf     if (local_err) {
9865cec2870SKevin Wolf         error_propagate(errp, local_err);
987c1bb86cdSEric Blake         ret = -1;
9885cec2870SKevin Wolf         goto out;
989c1bb86cdSEric Blake     }
990c1bb86cdSEric Blake 
991c1bb86cdSEric Blake     /* Fail already reopen_prepare() if we can't get a working O_DIRECT
992c1bb86cdSEric Blake      * alignment with the new fd. */
993c1bb86cdSEric Blake     if (rs->fd != -1) {
994c1bb86cdSEric Blake         raw_probe_alignment(state->bs, rs->fd, &local_err);
995c1bb86cdSEric Blake         if (local_err) {
996c1bb86cdSEric Blake             error_propagate(errp, local_err);
997c1bb86cdSEric Blake             ret = -EINVAL;
998a6aeca0cSKevin Wolf             goto out_fd;
999a6aeca0cSKevin Wolf         }
1000c1bb86cdSEric Blake     }
1001c1bb86cdSEric Blake 
1002e0c9cf3aSKevin Wolf     s->reopen_state = state;
1003a6aeca0cSKevin Wolf     ret = 0;
1004a6aeca0cSKevin Wolf out_fd:
1005a6aeca0cSKevin Wolf     if (ret < 0) {
1006a6aeca0cSKevin Wolf         qemu_close(rs->fd);
1007a6aeca0cSKevin Wolf         rs->fd = -1;
1008a6aeca0cSKevin Wolf     }
100931be8a2aSStefan Hajnoczi out:
101031be8a2aSStefan Hajnoczi     qemu_opts_del(opts);
1011c1bb86cdSEric Blake     return ret;
1012c1bb86cdSEric Blake }
1013c1bb86cdSEric Blake 
1014c1bb86cdSEric Blake static void raw_reopen_commit(BDRVReopenState *state)
1015c1bb86cdSEric Blake {
1016c1bb86cdSEric Blake     BDRVRawReopenState *rs = state->opaque;
1017c1bb86cdSEric Blake     BDRVRawState *s = state->bs->opaque;
1018c1bb86cdSEric Blake 
1019f357fcd8SStefan Hajnoczi     s->drop_cache = rs->drop_cache;
102031be8a2aSStefan Hajnoczi     s->check_cache_dropped = rs->check_cache_dropped;
1021c1bb86cdSEric Blake     s->open_flags = rs->open_flags;
1022c1bb86cdSEric Blake 
1023c1bb86cdSEric Blake     qemu_close(s->fd);
1024c1bb86cdSEric Blake     s->fd = rs->fd;
1025c1bb86cdSEric Blake 
1026c1bb86cdSEric Blake     g_free(state->opaque);
1027c1bb86cdSEric Blake     state->opaque = NULL;
1028e0c9cf3aSKevin Wolf 
1029e0c9cf3aSKevin Wolf     assert(s->reopen_state == state);
1030e0c9cf3aSKevin Wolf     s->reopen_state = NULL;
1031c1bb86cdSEric Blake }
1032c1bb86cdSEric Blake 
1033c1bb86cdSEric Blake 
1034c1bb86cdSEric Blake static void raw_reopen_abort(BDRVReopenState *state)
1035c1bb86cdSEric Blake {
1036c1bb86cdSEric Blake     BDRVRawReopenState *rs = state->opaque;
1037e0c9cf3aSKevin Wolf     BDRVRawState *s = state->bs->opaque;
1038c1bb86cdSEric Blake 
1039c1bb86cdSEric Blake      /* nothing to do if NULL, we didn't get far enough */
1040c1bb86cdSEric Blake     if (rs == NULL) {
1041c1bb86cdSEric Blake         return;
1042c1bb86cdSEric Blake     }
1043c1bb86cdSEric Blake 
1044c1bb86cdSEric Blake     if (rs->fd >= 0) {
1045c1bb86cdSEric Blake         qemu_close(rs->fd);
1046c1bb86cdSEric Blake         rs->fd = -1;
1047c1bb86cdSEric Blake     }
1048c1bb86cdSEric Blake     g_free(state->opaque);
1049c1bb86cdSEric Blake     state->opaque = NULL;
1050e0c9cf3aSKevin Wolf 
1051e0c9cf3aSKevin Wolf     assert(s->reopen_state == state);
1052e0c9cf3aSKevin Wolf     s->reopen_state = NULL;
1053c1bb86cdSEric Blake }
1054c1bb86cdSEric Blake 
1055867eccfeSMaxim Levitsky static int sg_get_max_transfer_length(int fd)
1056c1bb86cdSEric Blake {
1057c1bb86cdSEric Blake #ifdef BLKSECTGET
105848265250SEric Farman     int max_bytes = 0;
1059867eccfeSMaxim Levitsky 
1060867eccfeSMaxim Levitsky     if (ioctl(fd, BLKSECTGET, &max_bytes) == 0) {
106148265250SEric Farman         return max_bytes;
1062c1bb86cdSEric Blake     } else {
1063c1bb86cdSEric Blake         return -errno;
1064c1bb86cdSEric Blake     }
1065c1bb86cdSEric Blake #else
1066c1bb86cdSEric Blake     return -ENOSYS;
1067c1bb86cdSEric Blake #endif
1068c1bb86cdSEric Blake }
1069c1bb86cdSEric Blake 
1070867eccfeSMaxim Levitsky static int sg_get_max_segments(int fd)
10719103f1ceSFam Zheng {
10729103f1ceSFam Zheng #ifdef CONFIG_LINUX
10739103f1ceSFam Zheng     char buf[32];
10749103f1ceSFam Zheng     const char *end;
1075867eccfeSMaxim Levitsky     char *sysfspath = NULL;
10769103f1ceSFam Zheng     int ret;
1077867eccfeSMaxim Levitsky     int sysfd = -1;
10789103f1ceSFam Zheng     long max_segments;
1079867eccfeSMaxim Levitsky     struct stat st;
1080867eccfeSMaxim Levitsky 
1081867eccfeSMaxim Levitsky     if (fstat(fd, &st)) {
1082867eccfeSMaxim Levitsky         ret = -errno;
1083867eccfeSMaxim Levitsky         goto out;
1084867eccfeSMaxim Levitsky     }
10859103f1ceSFam Zheng 
10869103f1ceSFam Zheng     sysfspath = g_strdup_printf("/sys/dev/block/%u:%u/queue/max_segments",
1087867eccfeSMaxim Levitsky                                 major(st.st_rdev), minor(st.st_rdev));
1088867eccfeSMaxim Levitsky     sysfd = open(sysfspath, O_RDONLY);
1089867eccfeSMaxim Levitsky     if (sysfd == -1) {
10909103f1ceSFam Zheng         ret = -errno;
10919103f1ceSFam Zheng         goto out;
10929103f1ceSFam Zheng     }
10939103f1ceSFam Zheng     do {
1094867eccfeSMaxim Levitsky         ret = read(sysfd, buf, sizeof(buf) - 1);
10959103f1ceSFam Zheng     } while (ret == -1 && errno == EINTR);
10969103f1ceSFam Zheng     if (ret < 0) {
10979103f1ceSFam Zheng         ret = -errno;
10989103f1ceSFam Zheng         goto out;
10999103f1ceSFam Zheng     } else if (ret == 0) {
11009103f1ceSFam Zheng         ret = -EIO;
11019103f1ceSFam Zheng         goto out;
11029103f1ceSFam Zheng     }
11039103f1ceSFam Zheng     buf[ret] = 0;
11049103f1ceSFam Zheng     /* The file is ended with '\n', pass 'end' to accept that. */
11059103f1ceSFam Zheng     ret = qemu_strtol(buf, &end, 10, &max_segments);
11069103f1ceSFam Zheng     if (ret == 0 && end && *end == '\n') {
11079103f1ceSFam Zheng         ret = max_segments;
11089103f1ceSFam Zheng     }
11099103f1ceSFam Zheng 
11109103f1ceSFam Zheng out:
1111867eccfeSMaxim Levitsky     if (sysfd != -1) {
1112867eccfeSMaxim Levitsky         close(sysfd);
1113fed414dfSFam Zheng     }
11149103f1ceSFam Zheng     g_free(sysfspath);
11159103f1ceSFam Zheng     return ret;
11169103f1ceSFam Zheng #else
11179103f1ceSFam Zheng     return -ENOTSUP;
11189103f1ceSFam Zheng #endif
11199103f1ceSFam Zheng }
11209103f1ceSFam Zheng 
1121c1bb86cdSEric Blake static void raw_refresh_limits(BlockDriverState *bs, Error **errp)
1122c1bb86cdSEric Blake {
1123c1bb86cdSEric Blake     BDRVRawState *s = bs->opaque;
1124c1bb86cdSEric Blake 
1125867eccfeSMaxim Levitsky     if (bs->sg) {
1126867eccfeSMaxim Levitsky         int ret = sg_get_max_transfer_length(s->fd);
1127867eccfeSMaxim Levitsky 
112848265250SEric Farman         if (ret > 0 && ret <= BDRV_REQUEST_MAX_BYTES) {
112948265250SEric Farman             bs->bl.max_transfer = pow2floor(ret);
1130c1bb86cdSEric Blake         }
1131867eccfeSMaxim Levitsky 
1132867eccfeSMaxim Levitsky         ret = sg_get_max_segments(s->fd);
11339103f1ceSFam Zheng         if (ret > 0) {
1134867eccfeSMaxim Levitsky             bs->bl.max_transfer = MIN(bs->bl.max_transfer, ret * getpagesize());
1135c1bb86cdSEric Blake         }
1136c1bb86cdSEric Blake     }
1137c1bb86cdSEric Blake 
1138c1bb86cdSEric Blake     raw_probe_alignment(bs, s->fd, errp);
1139c1bb86cdSEric Blake     bs->bl.min_mem_alignment = s->buf_align;
1140c1bb86cdSEric Blake     bs->bl.opt_mem_alignment = MAX(s->buf_align, getpagesize());
1141c1bb86cdSEric Blake }
1142c1bb86cdSEric Blake 
1143c1bb86cdSEric Blake static int check_for_dasd(int fd)
1144c1bb86cdSEric Blake {
1145c1bb86cdSEric Blake #ifdef BIODASDINFO2
1146c1bb86cdSEric Blake     struct dasd_information2_t info = {0};
1147c1bb86cdSEric Blake 
1148c1bb86cdSEric Blake     return ioctl(fd, BIODASDINFO2, &info);
1149c1bb86cdSEric Blake #else
1150c1bb86cdSEric Blake     return -1;
1151c1bb86cdSEric Blake #endif
1152c1bb86cdSEric Blake }
1153c1bb86cdSEric Blake 
1154c1bb86cdSEric Blake /**
1155c1bb86cdSEric Blake  * Try to get @bs's logical and physical block size.
1156c1bb86cdSEric Blake  * On success, store them in @bsz and return zero.
1157c1bb86cdSEric Blake  * On failure, return negative errno.
1158c1bb86cdSEric Blake  */
1159c1bb86cdSEric Blake static int hdev_probe_blocksizes(BlockDriverState *bs, BlockSizes *bsz)
1160c1bb86cdSEric Blake {
1161c1bb86cdSEric Blake     BDRVRawState *s = bs->opaque;
1162c1bb86cdSEric Blake     int ret;
1163c1bb86cdSEric Blake 
1164c1bb86cdSEric Blake     /* If DASD, get blocksizes */
1165c1bb86cdSEric Blake     if (check_for_dasd(s->fd) < 0) {
1166c1bb86cdSEric Blake         return -ENOTSUP;
1167c1bb86cdSEric Blake     }
1168c1bb86cdSEric Blake     ret = probe_logical_blocksize(s->fd, &bsz->log);
1169c1bb86cdSEric Blake     if (ret < 0) {
1170c1bb86cdSEric Blake         return ret;
1171c1bb86cdSEric Blake     }
1172c1bb86cdSEric Blake     return probe_physical_blocksize(s->fd, &bsz->phys);
1173c1bb86cdSEric Blake }
1174c1bb86cdSEric Blake 
1175c1bb86cdSEric Blake /**
1176c1bb86cdSEric Blake  * Try to get @bs's geometry: cyls, heads, sectors.
1177c1bb86cdSEric Blake  * On success, store them in @geo and return 0.
1178c1bb86cdSEric Blake  * On failure return -errno.
1179c1bb86cdSEric Blake  * (Allows block driver to assign default geometry values that guest sees)
1180c1bb86cdSEric Blake  */
1181c1bb86cdSEric Blake #ifdef __linux__
1182c1bb86cdSEric Blake static int hdev_probe_geometry(BlockDriverState *bs, HDGeometry *geo)
1183c1bb86cdSEric Blake {
1184c1bb86cdSEric Blake     BDRVRawState *s = bs->opaque;
1185c1bb86cdSEric Blake     struct hd_geometry ioctl_geo = {0};
1186c1bb86cdSEric Blake 
1187c1bb86cdSEric Blake     /* If DASD, get its geometry */
1188c1bb86cdSEric Blake     if (check_for_dasd(s->fd) < 0) {
1189c1bb86cdSEric Blake         return -ENOTSUP;
1190c1bb86cdSEric Blake     }
1191c1bb86cdSEric Blake     if (ioctl(s->fd, HDIO_GETGEO, &ioctl_geo) < 0) {
1192c1bb86cdSEric Blake         return -errno;
1193c1bb86cdSEric Blake     }
1194c1bb86cdSEric Blake     /* HDIO_GETGEO may return success even though geo contains zeros
1195c1bb86cdSEric Blake        (e.g. certain multipath setups) */
1196c1bb86cdSEric Blake     if (!ioctl_geo.heads || !ioctl_geo.sectors || !ioctl_geo.cylinders) {
1197c1bb86cdSEric Blake         return -ENOTSUP;
1198c1bb86cdSEric Blake     }
1199c1bb86cdSEric Blake     /* Do not return a geometry for partition */
1200c1bb86cdSEric Blake     if (ioctl_geo.start != 0) {
1201c1bb86cdSEric Blake         return -ENOTSUP;
1202c1bb86cdSEric Blake     }
1203c1bb86cdSEric Blake     geo->heads = ioctl_geo.heads;
1204c1bb86cdSEric Blake     geo->sectors = ioctl_geo.sectors;
1205c1bb86cdSEric Blake     geo->cylinders = ioctl_geo.cylinders;
1206c1bb86cdSEric Blake 
1207c1bb86cdSEric Blake     return 0;
1208c1bb86cdSEric Blake }
1209c1bb86cdSEric Blake #else /* __linux__ */
1210c1bb86cdSEric Blake static int hdev_probe_geometry(BlockDriverState *bs, HDGeometry *geo)
1211c1bb86cdSEric Blake {
1212c1bb86cdSEric Blake     return -ENOTSUP;
1213c1bb86cdSEric Blake }
1214c1bb86cdSEric Blake #endif
1215c1bb86cdSEric Blake 
121603425671SKevin Wolf #if defined(__linux__)
121703425671SKevin Wolf static int handle_aiocb_ioctl(void *opaque)
1218c1bb86cdSEric Blake {
121903425671SKevin Wolf     RawPosixAIOData *aiocb = opaque;
1220c1bb86cdSEric Blake     int ret;
1221c1bb86cdSEric Blake 
1222d57c44d0SKevin Wolf     ret = ioctl(aiocb->aio_fildes, aiocb->ioctl.cmd, aiocb->ioctl.buf);
1223c1bb86cdSEric Blake     if (ret == -1) {
1224c1bb86cdSEric Blake         return -errno;
1225c1bb86cdSEric Blake     }
1226c1bb86cdSEric Blake 
1227c1bb86cdSEric Blake     return 0;
1228c1bb86cdSEric Blake }
122903425671SKevin Wolf #endif /* linux */
1230c1bb86cdSEric Blake 
123106dc9bd5SKevin Wolf static int handle_aiocb_flush(void *opaque)
1232c1bb86cdSEric Blake {
123306dc9bd5SKevin Wolf     RawPosixAIOData *aiocb = opaque;
1234e5bcf967SKevin Wolf     BDRVRawState *s = aiocb->bs->opaque;
1235c1bb86cdSEric Blake     int ret;
1236c1bb86cdSEric Blake 
1237e5bcf967SKevin Wolf     if (s->page_cache_inconsistent) {
1238e5bcf967SKevin Wolf         return -EIO;
1239e5bcf967SKevin Wolf     }
1240e5bcf967SKevin Wolf 
1241c1bb86cdSEric Blake     ret = qemu_fdatasync(aiocb->aio_fildes);
1242c1bb86cdSEric Blake     if (ret == -1) {
1243e5bcf967SKevin Wolf         /* There is no clear definition of the semantics of a failing fsync(),
1244e5bcf967SKevin Wolf          * so we may have to assume the worst. The sad truth is that this
1245e5bcf967SKevin Wolf          * assumption is correct for Linux. Some pages are now probably marked
1246e5bcf967SKevin Wolf          * clean in the page cache even though they are inconsistent with the
1247e5bcf967SKevin Wolf          * on-disk contents. The next fdatasync() call would succeed, but no
1248e5bcf967SKevin Wolf          * further writeback attempt will be made. We can't get back to a state
1249e5bcf967SKevin Wolf          * in which we know what is on disk (we would have to rewrite
1250e5bcf967SKevin Wolf          * everything that was touched since the last fdatasync() at least), so
1251e5bcf967SKevin Wolf          * make bdrv_flush() fail permanently. Given that the behaviour isn't
1252e5bcf967SKevin Wolf          * really defined, I have little hope that other OSes are doing better.
1253e5bcf967SKevin Wolf          *
1254e5bcf967SKevin Wolf          * Obviously, this doesn't affect O_DIRECT, which bypasses the page
1255e5bcf967SKevin Wolf          * cache. */
1256e5bcf967SKevin Wolf         if ((s->open_flags & O_DIRECT) == 0) {
1257e5bcf967SKevin Wolf             s->page_cache_inconsistent = true;
1258e5bcf967SKevin Wolf         }
1259c1bb86cdSEric Blake         return -errno;
1260c1bb86cdSEric Blake     }
1261c1bb86cdSEric Blake     return 0;
1262c1bb86cdSEric Blake }
1263c1bb86cdSEric Blake 
1264c1bb86cdSEric Blake #ifdef CONFIG_PREADV
1265c1bb86cdSEric Blake 
1266c1bb86cdSEric Blake static bool preadv_present = true;
1267c1bb86cdSEric Blake 
1268c1bb86cdSEric Blake static ssize_t
1269c1bb86cdSEric Blake qemu_preadv(int fd, const struct iovec *iov, int nr_iov, off_t offset)
1270c1bb86cdSEric Blake {
1271c1bb86cdSEric Blake     return preadv(fd, iov, nr_iov, offset);
1272c1bb86cdSEric Blake }
1273c1bb86cdSEric Blake 
1274c1bb86cdSEric Blake static ssize_t
1275c1bb86cdSEric Blake qemu_pwritev(int fd, const struct iovec *iov, int nr_iov, off_t offset)
1276c1bb86cdSEric Blake {
1277c1bb86cdSEric Blake     return pwritev(fd, iov, nr_iov, offset);
1278c1bb86cdSEric Blake }
1279c1bb86cdSEric Blake 
1280c1bb86cdSEric Blake #else
1281c1bb86cdSEric Blake 
1282c1bb86cdSEric Blake static bool preadv_present = false;
1283c1bb86cdSEric Blake 
1284c1bb86cdSEric Blake static ssize_t
1285c1bb86cdSEric Blake qemu_preadv(int fd, const struct iovec *iov, int nr_iov, off_t offset)
1286c1bb86cdSEric Blake {
1287c1bb86cdSEric Blake     return -ENOSYS;
1288c1bb86cdSEric Blake }
1289c1bb86cdSEric Blake 
1290c1bb86cdSEric Blake static ssize_t
1291c1bb86cdSEric Blake qemu_pwritev(int fd, const struct iovec *iov, int nr_iov, off_t offset)
1292c1bb86cdSEric Blake {
1293c1bb86cdSEric Blake     return -ENOSYS;
1294c1bb86cdSEric Blake }
1295c1bb86cdSEric Blake 
1296c1bb86cdSEric Blake #endif
1297c1bb86cdSEric Blake 
1298c1bb86cdSEric Blake static ssize_t handle_aiocb_rw_vector(RawPosixAIOData *aiocb)
1299c1bb86cdSEric Blake {
1300c1bb86cdSEric Blake     ssize_t len;
1301c1bb86cdSEric Blake 
1302c1bb86cdSEric Blake     do {
1303c1bb86cdSEric Blake         if (aiocb->aio_type & QEMU_AIO_WRITE)
1304c1bb86cdSEric Blake             len = qemu_pwritev(aiocb->aio_fildes,
1305d57c44d0SKevin Wolf                                aiocb->io.iov,
1306d57c44d0SKevin Wolf                                aiocb->io.niov,
1307c1bb86cdSEric Blake                                aiocb->aio_offset);
1308c1bb86cdSEric Blake          else
1309c1bb86cdSEric Blake             len = qemu_preadv(aiocb->aio_fildes,
1310d57c44d0SKevin Wolf                               aiocb->io.iov,
1311d57c44d0SKevin Wolf                               aiocb->io.niov,
1312c1bb86cdSEric Blake                               aiocb->aio_offset);
1313c1bb86cdSEric Blake     } while (len == -1 && errno == EINTR);
1314c1bb86cdSEric Blake 
1315c1bb86cdSEric Blake     if (len == -1) {
1316c1bb86cdSEric Blake         return -errno;
1317c1bb86cdSEric Blake     }
1318c1bb86cdSEric Blake     return len;
1319c1bb86cdSEric Blake }
1320c1bb86cdSEric Blake 
1321c1bb86cdSEric Blake /*
1322c1bb86cdSEric Blake  * Read/writes the data to/from a given linear buffer.
1323c1bb86cdSEric Blake  *
1324c1bb86cdSEric Blake  * Returns the number of bytes handles or -errno in case of an error. Short
1325c1bb86cdSEric Blake  * reads are only returned if the end of the file is reached.
1326c1bb86cdSEric Blake  */
1327c1bb86cdSEric Blake static ssize_t handle_aiocb_rw_linear(RawPosixAIOData *aiocb, char *buf)
1328c1bb86cdSEric Blake {
1329c1bb86cdSEric Blake     ssize_t offset = 0;
1330c1bb86cdSEric Blake     ssize_t len;
1331c1bb86cdSEric Blake 
1332c1bb86cdSEric Blake     while (offset < aiocb->aio_nbytes) {
1333c1bb86cdSEric Blake         if (aiocb->aio_type & QEMU_AIO_WRITE) {
1334c1bb86cdSEric Blake             len = pwrite(aiocb->aio_fildes,
1335c1bb86cdSEric Blake                          (const char *)buf + offset,
1336c1bb86cdSEric Blake                          aiocb->aio_nbytes - offset,
1337c1bb86cdSEric Blake                          aiocb->aio_offset + offset);
1338c1bb86cdSEric Blake         } else {
1339c1bb86cdSEric Blake             len = pread(aiocb->aio_fildes,
1340c1bb86cdSEric Blake                         buf + offset,
1341c1bb86cdSEric Blake                         aiocb->aio_nbytes - offset,
1342c1bb86cdSEric Blake                         aiocb->aio_offset + offset);
1343c1bb86cdSEric Blake         }
1344c1bb86cdSEric Blake         if (len == -1 && errno == EINTR) {
1345c1bb86cdSEric Blake             continue;
1346c1bb86cdSEric Blake         } else if (len == -1 && errno == EINVAL &&
1347c1bb86cdSEric Blake                    (aiocb->bs->open_flags & BDRV_O_NOCACHE) &&
1348c1bb86cdSEric Blake                    !(aiocb->aio_type & QEMU_AIO_WRITE) &&
1349c1bb86cdSEric Blake                    offset > 0) {
1350c1bb86cdSEric Blake             /* O_DIRECT pread() may fail with EINVAL when offset is unaligned
1351c1bb86cdSEric Blake              * after a short read.  Assume that O_DIRECT short reads only occur
1352c1bb86cdSEric Blake              * at EOF.  Therefore this is a short read, not an I/O error.
1353c1bb86cdSEric Blake              */
1354c1bb86cdSEric Blake             break;
1355c1bb86cdSEric Blake         } else if (len == -1) {
1356c1bb86cdSEric Blake             offset = -errno;
1357c1bb86cdSEric Blake             break;
1358c1bb86cdSEric Blake         } else if (len == 0) {
1359c1bb86cdSEric Blake             break;
1360c1bb86cdSEric Blake         }
1361c1bb86cdSEric Blake         offset += len;
1362c1bb86cdSEric Blake     }
1363c1bb86cdSEric Blake 
1364c1bb86cdSEric Blake     return offset;
1365c1bb86cdSEric Blake }
1366c1bb86cdSEric Blake 
1367999e6b69SKevin Wolf static int handle_aiocb_rw(void *opaque)
1368c1bb86cdSEric Blake {
1369999e6b69SKevin Wolf     RawPosixAIOData *aiocb = opaque;
1370c1bb86cdSEric Blake     ssize_t nbytes;
1371c1bb86cdSEric Blake     char *buf;
1372c1bb86cdSEric Blake 
1373c1bb86cdSEric Blake     if (!(aiocb->aio_type & QEMU_AIO_MISALIGNED)) {
1374c1bb86cdSEric Blake         /*
1375c1bb86cdSEric Blake          * If there is just a single buffer, and it is properly aligned
1376c1bb86cdSEric Blake          * we can just use plain pread/pwrite without any problems.
1377c1bb86cdSEric Blake          */
1378d57c44d0SKevin Wolf         if (aiocb->io.niov == 1) {
137954c7ca1bSKevin Wolf             nbytes = handle_aiocb_rw_linear(aiocb, aiocb->io.iov->iov_base);
138054c7ca1bSKevin Wolf             goto out;
1381c1bb86cdSEric Blake         }
1382c1bb86cdSEric Blake         /*
1383c1bb86cdSEric Blake          * We have more than one iovec, and all are properly aligned.
1384c1bb86cdSEric Blake          *
1385c1bb86cdSEric Blake          * Try preadv/pwritev first and fall back to linearizing the
1386c1bb86cdSEric Blake          * buffer if it's not supported.
1387c1bb86cdSEric Blake          */
1388c1bb86cdSEric Blake         if (preadv_present) {
1389c1bb86cdSEric Blake             nbytes = handle_aiocb_rw_vector(aiocb);
1390c1bb86cdSEric Blake             if (nbytes == aiocb->aio_nbytes ||
1391c1bb86cdSEric Blake                 (nbytes < 0 && nbytes != -ENOSYS)) {
139254c7ca1bSKevin Wolf                 goto out;
1393c1bb86cdSEric Blake             }
1394c1bb86cdSEric Blake             preadv_present = false;
1395c1bb86cdSEric Blake         }
1396c1bb86cdSEric Blake 
1397c1bb86cdSEric Blake         /*
1398c1bb86cdSEric Blake          * XXX(hch): short read/write.  no easy way to handle the reminder
1399c1bb86cdSEric Blake          * using these interfaces.  For now retry using plain
1400c1bb86cdSEric Blake          * pread/pwrite?
1401c1bb86cdSEric Blake          */
1402c1bb86cdSEric Blake     }
1403c1bb86cdSEric Blake 
1404c1bb86cdSEric Blake     /*
1405c1bb86cdSEric Blake      * Ok, we have to do it the hard way, copy all segments into
1406c1bb86cdSEric Blake      * a single aligned buffer.
1407c1bb86cdSEric Blake      */
1408c1bb86cdSEric Blake     buf = qemu_try_blockalign(aiocb->bs, aiocb->aio_nbytes);
1409c1bb86cdSEric Blake     if (buf == NULL) {
141054c7ca1bSKevin Wolf         nbytes = -ENOMEM;
141154c7ca1bSKevin Wolf         goto out;
1412c1bb86cdSEric Blake     }
1413c1bb86cdSEric Blake 
1414c1bb86cdSEric Blake     if (aiocb->aio_type & QEMU_AIO_WRITE) {
1415c1bb86cdSEric Blake         char *p = buf;
1416c1bb86cdSEric Blake         int i;
1417c1bb86cdSEric Blake 
1418d57c44d0SKevin Wolf         for (i = 0; i < aiocb->io.niov; ++i) {
1419d57c44d0SKevin Wolf             memcpy(p, aiocb->io.iov[i].iov_base, aiocb->io.iov[i].iov_len);
1420d57c44d0SKevin Wolf             p += aiocb->io.iov[i].iov_len;
1421c1bb86cdSEric Blake         }
1422c1bb86cdSEric Blake         assert(p - buf == aiocb->aio_nbytes);
1423c1bb86cdSEric Blake     }
1424c1bb86cdSEric Blake 
1425c1bb86cdSEric Blake     nbytes = handle_aiocb_rw_linear(aiocb, buf);
1426c1bb86cdSEric Blake     if (!(aiocb->aio_type & QEMU_AIO_WRITE)) {
1427c1bb86cdSEric Blake         char *p = buf;
1428c1bb86cdSEric Blake         size_t count = aiocb->aio_nbytes, copy;
1429c1bb86cdSEric Blake         int i;
1430c1bb86cdSEric Blake 
1431d57c44d0SKevin Wolf         for (i = 0; i < aiocb->io.niov && count; ++i) {
1432c1bb86cdSEric Blake             copy = count;
1433d57c44d0SKevin Wolf             if (copy > aiocb->io.iov[i].iov_len) {
1434d57c44d0SKevin Wolf                 copy = aiocb->io.iov[i].iov_len;
1435c1bb86cdSEric Blake             }
1436d57c44d0SKevin Wolf             memcpy(aiocb->io.iov[i].iov_base, p, copy);
1437c1bb86cdSEric Blake             assert(count >= copy);
1438c1bb86cdSEric Blake             p     += copy;
1439c1bb86cdSEric Blake             count -= copy;
1440c1bb86cdSEric Blake         }
1441c1bb86cdSEric Blake         assert(count == 0);
1442c1bb86cdSEric Blake     }
1443c1bb86cdSEric Blake     qemu_vfree(buf);
1444c1bb86cdSEric Blake 
144554c7ca1bSKevin Wolf out:
144654c7ca1bSKevin Wolf     if (nbytes == aiocb->aio_nbytes) {
144754c7ca1bSKevin Wolf         return 0;
144854c7ca1bSKevin Wolf     } else if (nbytes >= 0 && nbytes < aiocb->aio_nbytes) {
144954c7ca1bSKevin Wolf         if (aiocb->aio_type & QEMU_AIO_WRITE) {
145054c7ca1bSKevin Wolf             return -EINVAL;
145154c7ca1bSKevin Wolf         } else {
145254c7ca1bSKevin Wolf             iov_memset(aiocb->io.iov, aiocb->io.niov, nbytes,
145354c7ca1bSKevin Wolf                       0, aiocb->aio_nbytes - nbytes);
145454c7ca1bSKevin Wolf             return 0;
145554c7ca1bSKevin Wolf         }
145654c7ca1bSKevin Wolf     } else {
145754c7ca1bSKevin Wolf         assert(nbytes < 0);
1458c1bb86cdSEric Blake         return nbytes;
1459c1bb86cdSEric Blake     }
146054c7ca1bSKevin Wolf }
1461c1bb86cdSEric Blake 
1462c1bb86cdSEric Blake #ifdef CONFIG_XFS
1463c1bb86cdSEric Blake static int xfs_write_zeroes(BDRVRawState *s, int64_t offset, uint64_t bytes)
1464c1bb86cdSEric Blake {
146550ba5b2dSMax Reitz     int64_t len;
1466c1bb86cdSEric Blake     struct xfs_flock64 fl;
1467c1bb86cdSEric Blake     int err;
1468c1bb86cdSEric Blake 
146950ba5b2dSMax Reitz     len = lseek(s->fd, 0, SEEK_END);
147050ba5b2dSMax Reitz     if (len < 0) {
147150ba5b2dSMax Reitz         return -errno;
147250ba5b2dSMax Reitz     }
147350ba5b2dSMax Reitz 
147450ba5b2dSMax Reitz     if (offset + bytes > len) {
147550ba5b2dSMax Reitz         /* XFS_IOC_ZERO_RANGE does not increase the file length */
147650ba5b2dSMax Reitz         if (ftruncate(s->fd, offset + bytes) < 0) {
147750ba5b2dSMax Reitz             return -errno;
147850ba5b2dSMax Reitz         }
147950ba5b2dSMax Reitz     }
148050ba5b2dSMax Reitz 
1481c1bb86cdSEric Blake     memset(&fl, 0, sizeof(fl));
1482c1bb86cdSEric Blake     fl.l_whence = SEEK_SET;
1483c1bb86cdSEric Blake     fl.l_start = offset;
1484c1bb86cdSEric Blake     fl.l_len = bytes;
1485c1bb86cdSEric Blake 
1486c1bb86cdSEric Blake     if (xfsctl(NULL, s->fd, XFS_IOC_ZERO_RANGE, &fl) < 0) {
1487c1bb86cdSEric Blake         err = errno;
14884f7d28d7SLaurent Vivier         trace_file_xfs_write_zeroes(strerror(errno));
1489c1bb86cdSEric Blake         return -err;
1490c1bb86cdSEric Blake     }
1491c1bb86cdSEric Blake 
1492c1bb86cdSEric Blake     return 0;
1493c1bb86cdSEric Blake }
1494c1bb86cdSEric Blake 
1495c1bb86cdSEric Blake static int xfs_discard(BDRVRawState *s, int64_t offset, uint64_t bytes)
1496c1bb86cdSEric Blake {
1497c1bb86cdSEric Blake     struct xfs_flock64 fl;
1498c1bb86cdSEric Blake     int err;
1499c1bb86cdSEric Blake 
1500c1bb86cdSEric Blake     memset(&fl, 0, sizeof(fl));
1501c1bb86cdSEric Blake     fl.l_whence = SEEK_SET;
1502c1bb86cdSEric Blake     fl.l_start = offset;
1503c1bb86cdSEric Blake     fl.l_len = bytes;
1504c1bb86cdSEric Blake 
1505c1bb86cdSEric Blake     if (xfsctl(NULL, s->fd, XFS_IOC_UNRESVSP64, &fl) < 0) {
1506c1bb86cdSEric Blake         err = errno;
15074f7d28d7SLaurent Vivier         trace_file_xfs_discard(strerror(errno));
1508c1bb86cdSEric Blake         return -err;
1509c1bb86cdSEric Blake     }
1510c1bb86cdSEric Blake 
1511c1bb86cdSEric Blake     return 0;
1512c1bb86cdSEric Blake }
1513c1bb86cdSEric Blake #endif
1514c1bb86cdSEric Blake 
1515c1bb86cdSEric Blake static int translate_err(int err)
1516c1bb86cdSEric Blake {
1517c1bb86cdSEric Blake     if (err == -ENODEV || err == -ENOSYS || err == -EOPNOTSUPP ||
1518c1bb86cdSEric Blake         err == -ENOTTY) {
1519c1bb86cdSEric Blake         err = -ENOTSUP;
1520c1bb86cdSEric Blake     }
1521c1bb86cdSEric Blake     return err;
1522c1bb86cdSEric Blake }
1523c1bb86cdSEric Blake 
1524c1bb86cdSEric Blake #ifdef CONFIG_FALLOCATE
1525c1bb86cdSEric Blake static int do_fallocate(int fd, int mode, off_t offset, off_t len)
1526c1bb86cdSEric Blake {
1527c1bb86cdSEric Blake     do {
1528c1bb86cdSEric Blake         if (fallocate(fd, mode, offset, len) == 0) {
1529c1bb86cdSEric Blake             return 0;
1530c1bb86cdSEric Blake         }
1531c1bb86cdSEric Blake     } while (errno == EINTR);
1532c1bb86cdSEric Blake     return translate_err(-errno);
1533c1bb86cdSEric Blake }
1534c1bb86cdSEric Blake #endif
1535c1bb86cdSEric Blake 
1536c1bb86cdSEric Blake static ssize_t handle_aiocb_write_zeroes_block(RawPosixAIOData *aiocb)
1537c1bb86cdSEric Blake {
1538c1bb86cdSEric Blake     int ret = -ENOTSUP;
1539c1bb86cdSEric Blake     BDRVRawState *s = aiocb->bs->opaque;
1540c1bb86cdSEric Blake 
1541c1bb86cdSEric Blake     if (!s->has_write_zeroes) {
1542c1bb86cdSEric Blake         return -ENOTSUP;
1543c1bb86cdSEric Blake     }
1544c1bb86cdSEric Blake 
1545c1bb86cdSEric Blake #ifdef BLKZEROOUT
1546738301e1SKevin Wolf     /* The BLKZEROOUT implementation in the kernel doesn't set
1547738301e1SKevin Wolf      * BLKDEV_ZERO_NOFALLBACK, so we can't call this if we have to avoid slow
1548738301e1SKevin Wolf      * fallbacks. */
1549738301e1SKevin Wolf     if (!(aiocb->aio_type & QEMU_AIO_NO_FALLBACK)) {
1550c1bb86cdSEric Blake         do {
1551c1bb86cdSEric Blake             uint64_t range[2] = { aiocb->aio_offset, aiocb->aio_nbytes };
1552c1bb86cdSEric Blake             if (ioctl(aiocb->aio_fildes, BLKZEROOUT, range) == 0) {
1553c1bb86cdSEric Blake                 return 0;
1554c1bb86cdSEric Blake             }
1555c1bb86cdSEric Blake         } while (errno == EINTR);
1556c1bb86cdSEric Blake 
1557c1bb86cdSEric Blake         ret = translate_err(-errno);
1558738301e1SKevin Wolf     }
1559c1bb86cdSEric Blake #endif
1560c1bb86cdSEric Blake 
1561c1bb86cdSEric Blake     if (ret == -ENOTSUP) {
1562c1bb86cdSEric Blake         s->has_write_zeroes = false;
1563c1bb86cdSEric Blake     }
1564c1bb86cdSEric Blake     return ret;
1565c1bb86cdSEric Blake }
1566c1bb86cdSEric Blake 
15677154d8aeSKevin Wolf static int handle_aiocb_write_zeroes(void *opaque)
1568c1bb86cdSEric Blake {
15697154d8aeSKevin Wolf     RawPosixAIOData *aiocb = opaque;
1570c1bb86cdSEric Blake #if defined(CONFIG_FALLOCATE) || defined(CONFIG_XFS)
1571c1bb86cdSEric Blake     BDRVRawState *s = aiocb->bs->opaque;
1572c1bb86cdSEric Blake #endif
157370d9110bSDenis V. Lunev #ifdef CONFIG_FALLOCATE
157470d9110bSDenis V. Lunev     int64_t len;
157570d9110bSDenis V. Lunev #endif
1576c1bb86cdSEric Blake 
1577c1bb86cdSEric Blake     if (aiocb->aio_type & QEMU_AIO_BLKDEV) {
1578c1bb86cdSEric Blake         return handle_aiocb_write_zeroes_block(aiocb);
1579c1bb86cdSEric Blake     }
1580c1bb86cdSEric Blake 
1581c1bb86cdSEric Blake #ifdef CONFIG_XFS
1582c1bb86cdSEric Blake     if (s->is_xfs) {
1583c1bb86cdSEric Blake         return xfs_write_zeroes(s, aiocb->aio_offset, aiocb->aio_nbytes);
1584c1bb86cdSEric Blake     }
1585c1bb86cdSEric Blake #endif
1586c1bb86cdSEric Blake 
1587c1bb86cdSEric Blake #ifdef CONFIG_FALLOCATE_ZERO_RANGE
1588c1bb86cdSEric Blake     if (s->has_write_zeroes) {
1589c1bb86cdSEric Blake         int ret = do_fallocate(s->fd, FALLOC_FL_ZERO_RANGE,
1590c1bb86cdSEric Blake                                aiocb->aio_offset, aiocb->aio_nbytes);
1591*294682ccSAndrey Shinkevich         if (ret == -EINVAL) {
1592*294682ccSAndrey Shinkevich             /*
1593*294682ccSAndrey Shinkevich              * Allow falling back to pwrite for file systems that
1594*294682ccSAndrey Shinkevich              * do not support fallocate() for an unaligned byte range.
1595*294682ccSAndrey Shinkevich              */
1596*294682ccSAndrey Shinkevich             return -ENOTSUP;
1597*294682ccSAndrey Shinkevich         }
1598c1bb86cdSEric Blake         if (ret == 0 || ret != -ENOTSUP) {
1599c1bb86cdSEric Blake             return ret;
1600c1bb86cdSEric Blake         }
1601c1bb86cdSEric Blake         s->has_write_zeroes = false;
1602c1bb86cdSEric Blake     }
1603c1bb86cdSEric Blake #endif
1604c1bb86cdSEric Blake 
1605c1bb86cdSEric Blake #ifdef CONFIG_FALLOCATE_PUNCH_HOLE
1606c1bb86cdSEric Blake     if (s->has_discard && s->has_fallocate) {
1607c1bb86cdSEric Blake         int ret = do_fallocate(s->fd,
1608c1bb86cdSEric Blake                                FALLOC_FL_PUNCH_HOLE | FALLOC_FL_KEEP_SIZE,
1609c1bb86cdSEric Blake                                aiocb->aio_offset, aiocb->aio_nbytes);
1610c1bb86cdSEric Blake         if (ret == 0) {
1611c1bb86cdSEric Blake             ret = do_fallocate(s->fd, 0, aiocb->aio_offset, aiocb->aio_nbytes);
1612c1bb86cdSEric Blake             if (ret == 0 || ret != -ENOTSUP) {
1613c1bb86cdSEric Blake                 return ret;
1614c1bb86cdSEric Blake             }
1615c1bb86cdSEric Blake             s->has_fallocate = false;
1616c1bb86cdSEric Blake         } else if (ret != -ENOTSUP) {
1617c1bb86cdSEric Blake             return ret;
1618c1bb86cdSEric Blake         } else {
1619c1bb86cdSEric Blake             s->has_discard = false;
1620c1bb86cdSEric Blake         }
1621c1bb86cdSEric Blake     }
1622c1bb86cdSEric Blake #endif
1623c1bb86cdSEric Blake 
1624c1bb86cdSEric Blake #ifdef CONFIG_FALLOCATE
162570d9110bSDenis V. Lunev     /* Last resort: we are trying to extend the file with zeroed data. This
162670d9110bSDenis V. Lunev      * can be done via fallocate(fd, 0) */
162770d9110bSDenis V. Lunev     len = bdrv_getlength(aiocb->bs);
162870d9110bSDenis V. Lunev     if (s->has_fallocate && len >= 0 && aiocb->aio_offset >= len) {
1629c1bb86cdSEric Blake         int ret = do_fallocate(s->fd, 0, aiocb->aio_offset, aiocb->aio_nbytes);
1630c1bb86cdSEric Blake         if (ret == 0 || ret != -ENOTSUP) {
1631c1bb86cdSEric Blake             return ret;
1632c1bb86cdSEric Blake         }
1633c1bb86cdSEric Blake         s->has_fallocate = false;
1634c1bb86cdSEric Blake     }
1635c1bb86cdSEric Blake #endif
1636c1bb86cdSEric Blake 
1637c1bb86cdSEric Blake     return -ENOTSUP;
1638c1bb86cdSEric Blake }
1639c1bb86cdSEric Blake 
16407154d8aeSKevin Wolf static int handle_aiocb_write_zeroes_unmap(void *opaque)
164134fa110eSKevin Wolf {
16427154d8aeSKevin Wolf     RawPosixAIOData *aiocb = opaque;
164334fa110eSKevin Wolf     BDRVRawState *s G_GNUC_UNUSED = aiocb->bs->opaque;
164434fa110eSKevin Wolf     int ret;
164534fa110eSKevin Wolf 
164634fa110eSKevin Wolf     /* First try to write zeros and unmap at the same time */
164734fa110eSKevin Wolf 
164834fa110eSKevin Wolf #ifdef CONFIG_FALLOCATE_PUNCH_HOLE
164934fa110eSKevin Wolf     ret = do_fallocate(s->fd, FALLOC_FL_PUNCH_HOLE | FALLOC_FL_KEEP_SIZE,
165034fa110eSKevin Wolf                        aiocb->aio_offset, aiocb->aio_nbytes);
165134fa110eSKevin Wolf     if (ret != -ENOTSUP) {
165234fa110eSKevin Wolf         return ret;
165334fa110eSKevin Wolf     }
165434fa110eSKevin Wolf #endif
165534fa110eSKevin Wolf 
165634fa110eSKevin Wolf #ifdef CONFIG_XFS
165734fa110eSKevin Wolf     if (s->is_xfs) {
165834fa110eSKevin Wolf         /* xfs_discard() guarantees that the discarded area reads as all-zero
165934fa110eSKevin Wolf          * afterwards, so we can use it here. */
166034fa110eSKevin Wolf         return xfs_discard(s, aiocb->aio_offset, aiocb->aio_nbytes);
166134fa110eSKevin Wolf     }
166234fa110eSKevin Wolf #endif
166334fa110eSKevin Wolf 
166434fa110eSKevin Wolf     /* If we couldn't manage to unmap while guaranteed that the area reads as
166534fa110eSKevin Wolf      * all-zero afterwards, just write zeroes without unmapping */
166634fa110eSKevin Wolf     ret = handle_aiocb_write_zeroes(aiocb);
166734fa110eSKevin Wolf     return ret;
166834fa110eSKevin Wolf }
166934fa110eSKevin Wolf 
16701efad060SFam Zheng #ifndef HAVE_COPY_FILE_RANGE
16711efad060SFam Zheng static off_t copy_file_range(int in_fd, off_t *in_off, int out_fd,
16721efad060SFam Zheng                              off_t *out_off, size_t len, unsigned int flags)
16731efad060SFam Zheng {
16741efad060SFam Zheng #ifdef __NR_copy_file_range
16751efad060SFam Zheng     return syscall(__NR_copy_file_range, in_fd, in_off, out_fd,
16761efad060SFam Zheng                    out_off, len, flags);
16771efad060SFam Zheng #else
16781efad060SFam Zheng     errno = ENOSYS;
16791efad060SFam Zheng     return -1;
16801efad060SFam Zheng #endif
16811efad060SFam Zheng }
16821efad060SFam Zheng #endif
16831efad060SFam Zheng 
168458a209c4SKevin Wolf static int handle_aiocb_copy_range(void *opaque)
16851efad060SFam Zheng {
168658a209c4SKevin Wolf     RawPosixAIOData *aiocb = opaque;
16871efad060SFam Zheng     uint64_t bytes = aiocb->aio_nbytes;
16881efad060SFam Zheng     off_t in_off = aiocb->aio_offset;
1689d57c44d0SKevin Wolf     off_t out_off = aiocb->copy_range.aio_offset2;
16901efad060SFam Zheng 
16911efad060SFam Zheng     while (bytes) {
16921efad060SFam Zheng         ssize_t ret = copy_file_range(aiocb->aio_fildes, &in_off,
1693d57c44d0SKevin Wolf                                       aiocb->copy_range.aio_fd2, &out_off,
16941efad060SFam Zheng                                       bytes, 0);
1695ecc983a5SFam Zheng         trace_file_copy_file_range(aiocb->bs, aiocb->aio_fildes, in_off,
1696d57c44d0SKevin Wolf                                    aiocb->copy_range.aio_fd2, out_off, bytes,
1697d57c44d0SKevin Wolf                                    0, ret);
1698c436e3d0SFam Zheng         if (ret == 0) {
1699c436e3d0SFam Zheng             /* No progress (e.g. when beyond EOF), let the caller fall back to
1700c436e3d0SFam Zheng              * buffer I/O. */
1701c436e3d0SFam Zheng             return -ENOSPC;
17021efad060SFam Zheng         }
17031efad060SFam Zheng         if (ret < 0) {
1704c436e3d0SFam Zheng             switch (errno) {
1705c436e3d0SFam Zheng             case ENOSYS:
17061efad060SFam Zheng                 return -ENOTSUP;
1707c436e3d0SFam Zheng             case EINTR:
1708c436e3d0SFam Zheng                 continue;
1709c436e3d0SFam Zheng             default:
17101efad060SFam Zheng                 return -errno;
17111efad060SFam Zheng             }
17121efad060SFam Zheng         }
17131efad060SFam Zheng         bytes -= ret;
17141efad060SFam Zheng     }
17151efad060SFam Zheng     return 0;
17161efad060SFam Zheng }
17171efad060SFam Zheng 
171846ee0f46SKevin Wolf static int handle_aiocb_discard(void *opaque)
1719c1bb86cdSEric Blake {
172046ee0f46SKevin Wolf     RawPosixAIOData *aiocb = opaque;
1721c1bb86cdSEric Blake     int ret = -EOPNOTSUPP;
1722c1bb86cdSEric Blake     BDRVRawState *s = aiocb->bs->opaque;
1723c1bb86cdSEric Blake 
1724c1bb86cdSEric Blake     if (!s->has_discard) {
1725c1bb86cdSEric Blake         return -ENOTSUP;
1726c1bb86cdSEric Blake     }
1727c1bb86cdSEric Blake 
1728c1bb86cdSEric Blake     if (aiocb->aio_type & QEMU_AIO_BLKDEV) {
1729c1bb86cdSEric Blake #ifdef BLKDISCARD
1730c1bb86cdSEric Blake         do {
1731c1bb86cdSEric Blake             uint64_t range[2] = { aiocb->aio_offset, aiocb->aio_nbytes };
1732c1bb86cdSEric Blake             if (ioctl(aiocb->aio_fildes, BLKDISCARD, range) == 0) {
1733c1bb86cdSEric Blake                 return 0;
1734c1bb86cdSEric Blake             }
1735c1bb86cdSEric Blake         } while (errno == EINTR);
1736c1bb86cdSEric Blake 
1737c1bb86cdSEric Blake         ret = -errno;
1738c1bb86cdSEric Blake #endif
1739c1bb86cdSEric Blake     } else {
1740c1bb86cdSEric Blake #ifdef CONFIG_XFS
1741c1bb86cdSEric Blake         if (s->is_xfs) {
1742c1bb86cdSEric Blake             return xfs_discard(s, aiocb->aio_offset, aiocb->aio_nbytes);
1743c1bb86cdSEric Blake         }
1744c1bb86cdSEric Blake #endif
1745c1bb86cdSEric Blake 
1746c1bb86cdSEric Blake #ifdef CONFIG_FALLOCATE_PUNCH_HOLE
1747c1bb86cdSEric Blake         ret = do_fallocate(s->fd, FALLOC_FL_PUNCH_HOLE | FALLOC_FL_KEEP_SIZE,
1748c1bb86cdSEric Blake                            aiocb->aio_offset, aiocb->aio_nbytes);
1749c1bb86cdSEric Blake #endif
1750c1bb86cdSEric Blake     }
1751c1bb86cdSEric Blake 
1752c1bb86cdSEric Blake     ret = translate_err(ret);
1753c1bb86cdSEric Blake     if (ret == -ENOTSUP) {
1754c1bb86cdSEric Blake         s->has_discard = false;
1755c1bb86cdSEric Blake     }
1756c1bb86cdSEric Blake     return ret;
1757c1bb86cdSEric Blake }
1758c1bb86cdSEric Blake 
17593a20013fSNir Soffer /*
17603a20013fSNir Soffer  * Help alignment probing by allocating the first block.
17613a20013fSNir Soffer  *
17623a20013fSNir Soffer  * When reading with direct I/O from unallocated area on Gluster backed by XFS,
17633a20013fSNir Soffer  * reading succeeds regardless of request length. In this case we fallback to
17643a20013fSNir Soffer  * safe alignment which is not optimal. Allocating the first block avoids this
17653a20013fSNir Soffer  * fallback.
17663a20013fSNir Soffer  *
17673a20013fSNir Soffer  * fd may be opened with O_DIRECT, but we don't know the buffer alignment or
17683a20013fSNir Soffer  * request alignment, so we use safe values.
17693a20013fSNir Soffer  *
17703a20013fSNir Soffer  * Returns: 0 on success, -errno on failure. Since this is an optimization,
17713a20013fSNir Soffer  * caller may ignore failures.
17723a20013fSNir Soffer  */
17733a20013fSNir Soffer static int allocate_first_block(int fd, size_t max_size)
17743a20013fSNir Soffer {
17753a20013fSNir Soffer     size_t write_size = (max_size < MAX_BLOCKSIZE)
17763a20013fSNir Soffer         ? BDRV_SECTOR_SIZE
17773a20013fSNir Soffer         : MAX_BLOCKSIZE;
17783a20013fSNir Soffer     size_t max_align = MAX(MAX_BLOCKSIZE, getpagesize());
17793a20013fSNir Soffer     void *buf;
17803a20013fSNir Soffer     ssize_t n;
17813a20013fSNir Soffer     int ret;
17823a20013fSNir Soffer 
17833a20013fSNir Soffer     buf = qemu_memalign(max_align, write_size);
17843a20013fSNir Soffer     memset(buf, 0, write_size);
17853a20013fSNir Soffer 
17863a20013fSNir Soffer     do {
17873a20013fSNir Soffer         n = pwrite(fd, buf, write_size, 0);
17883a20013fSNir Soffer     } while (n == -1 && errno == EINTR);
17893a20013fSNir Soffer 
17903a20013fSNir Soffer     ret = (n == -1) ? -errno : 0;
17913a20013fSNir Soffer 
17923a20013fSNir Soffer     qemu_vfree(buf);
17933a20013fSNir Soffer     return ret;
17943a20013fSNir Soffer }
17953a20013fSNir Soffer 
179629cb4c01SKevin Wolf static int handle_aiocb_truncate(void *opaque)
179793f4e2ffSKevin Wolf {
179829cb4c01SKevin Wolf     RawPosixAIOData *aiocb = opaque;
179993f4e2ffSKevin Wolf     int result = 0;
180093f4e2ffSKevin Wolf     int64_t current_length = 0;
180193f4e2ffSKevin Wolf     char *buf = NULL;
180293f4e2ffSKevin Wolf     struct stat st;
180393f4e2ffSKevin Wolf     int fd = aiocb->aio_fildes;
180493f4e2ffSKevin Wolf     int64_t offset = aiocb->aio_offset;
1805d57c44d0SKevin Wolf     PreallocMode prealloc = aiocb->truncate.prealloc;
1806d57c44d0SKevin Wolf     Error **errp = aiocb->truncate.errp;
180793f4e2ffSKevin Wolf 
180893f4e2ffSKevin Wolf     if (fstat(fd, &st) < 0) {
180993f4e2ffSKevin Wolf         result = -errno;
181093f4e2ffSKevin Wolf         error_setg_errno(errp, -result, "Could not stat file");
181193f4e2ffSKevin Wolf         return result;
181293f4e2ffSKevin Wolf     }
181393f4e2ffSKevin Wolf 
181493f4e2ffSKevin Wolf     current_length = st.st_size;
1815d57c44d0SKevin Wolf     if (current_length > offset && prealloc != PREALLOC_MODE_OFF) {
181693f4e2ffSKevin Wolf         error_setg(errp, "Cannot use preallocation for shrinking files");
181793f4e2ffSKevin Wolf         return -ENOTSUP;
181893f4e2ffSKevin Wolf     }
181993f4e2ffSKevin Wolf 
1820d57c44d0SKevin Wolf     switch (prealloc) {
182193f4e2ffSKevin Wolf #ifdef CONFIG_POSIX_FALLOCATE
182293f4e2ffSKevin Wolf     case PREALLOC_MODE_FALLOC:
182393f4e2ffSKevin Wolf         /*
182493f4e2ffSKevin Wolf          * Truncating before posix_fallocate() makes it about twice slower on
182593f4e2ffSKevin Wolf          * file systems that do not support fallocate(), trying to check if a
182693f4e2ffSKevin Wolf          * block is allocated before allocating it, so don't do that here.
182793f4e2ffSKevin Wolf          */
182893f4e2ffSKevin Wolf         if (offset != current_length) {
182993f4e2ffSKevin Wolf             result = -posix_fallocate(fd, current_length,
183093f4e2ffSKevin Wolf                                       offset - current_length);
183193f4e2ffSKevin Wolf             if (result != 0) {
183293f4e2ffSKevin Wolf                 /* posix_fallocate() doesn't set errno. */
183393f4e2ffSKevin Wolf                 error_setg_errno(errp, -result,
183493f4e2ffSKevin Wolf                                  "Could not preallocate new data");
18353a20013fSNir Soffer             } else if (current_length == 0) {
18363a20013fSNir Soffer                 /*
18373a20013fSNir Soffer                  * posix_fallocate() uses fallocate() if the filesystem
18383a20013fSNir Soffer                  * supports it, or fallback to manually writing zeroes. If
18393a20013fSNir Soffer                  * fallocate() was used, unaligned reads from the fallocated
18403a20013fSNir Soffer                  * area in raw_probe_alignment() will succeed, hence we need to
18413a20013fSNir Soffer                  * allocate the first block.
18423a20013fSNir Soffer                  *
18433a20013fSNir Soffer                  * Optimize future alignment probing; ignore failures.
18443a20013fSNir Soffer                  */
18453a20013fSNir Soffer                 allocate_first_block(fd, offset);
184693f4e2ffSKevin Wolf             }
184793f4e2ffSKevin Wolf         } else {
184893f4e2ffSKevin Wolf             result = 0;
184993f4e2ffSKevin Wolf         }
185093f4e2ffSKevin Wolf         goto out;
185193f4e2ffSKevin Wolf #endif
185293f4e2ffSKevin Wolf     case PREALLOC_MODE_FULL:
185393f4e2ffSKevin Wolf     {
185493f4e2ffSKevin Wolf         int64_t num = 0, left = offset - current_length;
185593f4e2ffSKevin Wolf         off_t seek_result;
185693f4e2ffSKevin Wolf 
185793f4e2ffSKevin Wolf         /*
185893f4e2ffSKevin Wolf          * Knowing the final size from the beginning could allow the file
185993f4e2ffSKevin Wolf          * system driver to do less allocations and possibly avoid
186093f4e2ffSKevin Wolf          * fragmentation of the file.
186193f4e2ffSKevin Wolf          */
186293f4e2ffSKevin Wolf         if (ftruncate(fd, offset) != 0) {
186393f4e2ffSKevin Wolf             result = -errno;
186493f4e2ffSKevin Wolf             error_setg_errno(errp, -result, "Could not resize file");
186593f4e2ffSKevin Wolf             goto out;
186693f4e2ffSKevin Wolf         }
186793f4e2ffSKevin Wolf 
186893f4e2ffSKevin Wolf         buf = g_malloc0(65536);
186993f4e2ffSKevin Wolf 
187093f4e2ffSKevin Wolf         seek_result = lseek(fd, current_length, SEEK_SET);
187193f4e2ffSKevin Wolf         if (seek_result < 0) {
187293f4e2ffSKevin Wolf             result = -errno;
187393f4e2ffSKevin Wolf             error_setg_errno(errp, -result,
187493f4e2ffSKevin Wolf                              "Failed to seek to the old end of file");
187593f4e2ffSKevin Wolf             goto out;
187693f4e2ffSKevin Wolf         }
187793f4e2ffSKevin Wolf 
187893f4e2ffSKevin Wolf         while (left > 0) {
187993f4e2ffSKevin Wolf             num = MIN(left, 65536);
188093f4e2ffSKevin Wolf             result = write(fd, buf, num);
188193f4e2ffSKevin Wolf             if (result < 0) {
1882a1c81f4fSFam Zheng                 if (errno == EINTR) {
1883a1c81f4fSFam Zheng                     continue;
1884a1c81f4fSFam Zheng                 }
188593f4e2ffSKevin Wolf                 result = -errno;
188693f4e2ffSKevin Wolf                 error_setg_errno(errp, -result,
188793f4e2ffSKevin Wolf                                  "Could not write zeros for preallocation");
188893f4e2ffSKevin Wolf                 goto out;
188993f4e2ffSKevin Wolf             }
189093f4e2ffSKevin Wolf             left -= result;
189193f4e2ffSKevin Wolf         }
189293f4e2ffSKevin Wolf         if (result >= 0) {
189393f4e2ffSKevin Wolf             result = fsync(fd);
189493f4e2ffSKevin Wolf             if (result < 0) {
189593f4e2ffSKevin Wolf                 result = -errno;
189693f4e2ffSKevin Wolf                 error_setg_errno(errp, -result,
189793f4e2ffSKevin Wolf                                  "Could not flush file to disk");
189893f4e2ffSKevin Wolf                 goto out;
189993f4e2ffSKevin Wolf             }
190093f4e2ffSKevin Wolf         }
190193f4e2ffSKevin Wolf         goto out;
190293f4e2ffSKevin Wolf     }
190393f4e2ffSKevin Wolf     case PREALLOC_MODE_OFF:
190493f4e2ffSKevin Wolf         if (ftruncate(fd, offset) != 0) {
190593f4e2ffSKevin Wolf             result = -errno;
190693f4e2ffSKevin Wolf             error_setg_errno(errp, -result, "Could not resize file");
19073a20013fSNir Soffer         } else if (current_length == 0 && offset > current_length) {
19083a20013fSNir Soffer             /* Optimize future alignment probing; ignore failures. */
19093a20013fSNir Soffer             allocate_first_block(fd, offset);
191093f4e2ffSKevin Wolf         }
191193f4e2ffSKevin Wolf         return result;
191293f4e2ffSKevin Wolf     default:
191393f4e2ffSKevin Wolf         result = -ENOTSUP;
191493f4e2ffSKevin Wolf         error_setg(errp, "Unsupported preallocation mode: %s",
1915d57c44d0SKevin Wolf                    PreallocMode_str(prealloc));
191693f4e2ffSKevin Wolf         return result;
191793f4e2ffSKevin Wolf     }
191893f4e2ffSKevin Wolf 
191993f4e2ffSKevin Wolf out:
192093f4e2ffSKevin Wolf     if (result < 0) {
192193f4e2ffSKevin Wolf         if (ftruncate(fd, current_length) < 0) {
192293f4e2ffSKevin Wolf             error_report("Failed to restore old file length: %s",
192393f4e2ffSKevin Wolf                          strerror(errno));
192493f4e2ffSKevin Wolf         }
192593f4e2ffSKevin Wolf     }
192693f4e2ffSKevin Wolf 
192793f4e2ffSKevin Wolf     g_free(buf);
192893f4e2ffSKevin Wolf     return result;
192993f4e2ffSKevin Wolf }
193093f4e2ffSKevin Wolf 
19315d5de250SKevin Wolf static int coroutine_fn raw_thread_pool_submit(BlockDriverState *bs,
19325d5de250SKevin Wolf                                                ThreadPoolFunc func, void *arg)
19335d5de250SKevin Wolf {
19345d5de250SKevin Wolf     /* @bs can be NULL, bdrv_get_aio_context() returns the main context then */
19355d5de250SKevin Wolf     ThreadPool *pool = aio_get_thread_pool(bdrv_get_aio_context(bs));
19365d5de250SKevin Wolf     return thread_pool_submit_co(pool, func, arg);
19375d5de250SKevin Wolf }
19385d5de250SKevin Wolf 
1939c1bb86cdSEric Blake static int coroutine_fn raw_co_prw(BlockDriverState *bs, uint64_t offset,
1940c1bb86cdSEric Blake                                    uint64_t bytes, QEMUIOVector *qiov, int type)
1941c1bb86cdSEric Blake {
1942c1bb86cdSEric Blake     BDRVRawState *s = bs->opaque;
1943999e6b69SKevin Wolf     RawPosixAIOData acb;
1944c1bb86cdSEric Blake 
1945c1bb86cdSEric Blake     if (fd_open(bs) < 0)
1946c1bb86cdSEric Blake         return -EIO;
1947c1bb86cdSEric Blake 
1948c1bb86cdSEric Blake     /*
1949c1bb86cdSEric Blake      * Check if the underlying device requires requests to be aligned,
1950c1bb86cdSEric Blake      * and if the request we are trying to submit is aligned or not.
1951c1bb86cdSEric Blake      * If this is the case tell the low-level driver that it needs
1952c1bb86cdSEric Blake      * to copy the buffer.
1953c1bb86cdSEric Blake      */
1954c1bb86cdSEric Blake     if (s->needs_alignment) {
1955c1bb86cdSEric Blake         if (!bdrv_qiov_is_aligned(bs, qiov)) {
1956c1bb86cdSEric Blake             type |= QEMU_AIO_MISALIGNED;
1957c1bb86cdSEric Blake #ifdef CONFIG_LINUX_AIO
1958c1bb86cdSEric Blake         } else if (s->use_linux_aio) {
1959c1bb86cdSEric Blake             LinuxAioState *aio = aio_get_linux_aio(bdrv_get_aio_context(bs));
1960c1bb86cdSEric Blake             assert(qiov->size == bytes);
1961c1bb86cdSEric Blake             return laio_co_submit(bs, aio, s->fd, offset, qiov, type);
1962c1bb86cdSEric Blake #endif
1963c1bb86cdSEric Blake         }
1964c1bb86cdSEric Blake     }
1965c1bb86cdSEric Blake 
1966999e6b69SKevin Wolf     acb = (RawPosixAIOData) {
1967999e6b69SKevin Wolf         .bs             = bs,
1968999e6b69SKevin Wolf         .aio_fildes     = s->fd,
1969999e6b69SKevin Wolf         .aio_type       = type,
1970999e6b69SKevin Wolf         .aio_offset     = offset,
1971999e6b69SKevin Wolf         .aio_nbytes     = bytes,
1972999e6b69SKevin Wolf         .io             = {
1973999e6b69SKevin Wolf             .iov            = qiov->iov,
1974999e6b69SKevin Wolf             .niov           = qiov->niov,
1975999e6b69SKevin Wolf         },
1976999e6b69SKevin Wolf     };
1977999e6b69SKevin Wolf 
1978999e6b69SKevin Wolf     assert(qiov->size == bytes);
1979999e6b69SKevin Wolf     return raw_thread_pool_submit(bs, handle_aiocb_rw, &acb);
1980c1bb86cdSEric Blake }
1981c1bb86cdSEric Blake 
1982c1bb86cdSEric Blake static int coroutine_fn raw_co_preadv(BlockDriverState *bs, uint64_t offset,
1983c1bb86cdSEric Blake                                       uint64_t bytes, QEMUIOVector *qiov,
1984c1bb86cdSEric Blake                                       int flags)
1985c1bb86cdSEric Blake {
1986c1bb86cdSEric Blake     return raw_co_prw(bs, offset, bytes, qiov, QEMU_AIO_READ);
1987c1bb86cdSEric Blake }
1988c1bb86cdSEric Blake 
1989c1bb86cdSEric Blake static int coroutine_fn raw_co_pwritev(BlockDriverState *bs, uint64_t offset,
1990c1bb86cdSEric Blake                                        uint64_t bytes, QEMUIOVector *qiov,
1991c1bb86cdSEric Blake                                        int flags)
1992c1bb86cdSEric Blake {
1993c1bb86cdSEric Blake     assert(flags == 0);
1994c1bb86cdSEric Blake     return raw_co_prw(bs, offset, bytes, qiov, QEMU_AIO_WRITE);
1995c1bb86cdSEric Blake }
1996c1bb86cdSEric Blake 
1997c1bb86cdSEric Blake static void raw_aio_plug(BlockDriverState *bs)
1998c1bb86cdSEric Blake {
1999c1bb86cdSEric Blake #ifdef CONFIG_LINUX_AIO
2000c1bb86cdSEric Blake     BDRVRawState *s = bs->opaque;
2001c1bb86cdSEric Blake     if (s->use_linux_aio) {
2002c1bb86cdSEric Blake         LinuxAioState *aio = aio_get_linux_aio(bdrv_get_aio_context(bs));
2003c1bb86cdSEric Blake         laio_io_plug(bs, aio);
2004c1bb86cdSEric Blake     }
2005c1bb86cdSEric Blake #endif
2006c1bb86cdSEric Blake }
2007c1bb86cdSEric Blake 
2008c1bb86cdSEric Blake static void raw_aio_unplug(BlockDriverState *bs)
2009c1bb86cdSEric Blake {
2010c1bb86cdSEric Blake #ifdef CONFIG_LINUX_AIO
2011c1bb86cdSEric Blake     BDRVRawState *s = bs->opaque;
2012c1bb86cdSEric Blake     if (s->use_linux_aio) {
2013c1bb86cdSEric Blake         LinuxAioState *aio = aio_get_linux_aio(bdrv_get_aio_context(bs));
2014c1bb86cdSEric Blake         laio_io_unplug(bs, aio);
2015c1bb86cdSEric Blake     }
2016c1bb86cdSEric Blake #endif
2017c1bb86cdSEric Blake }
2018c1bb86cdSEric Blake 
201933d70fb6SKevin Wolf static int raw_co_flush_to_disk(BlockDriverState *bs)
2020c1bb86cdSEric Blake {
2021c1bb86cdSEric Blake     BDRVRawState *s = bs->opaque;
202206dc9bd5SKevin Wolf     RawPosixAIOData acb;
202333d70fb6SKevin Wolf     int ret;
2024c1bb86cdSEric Blake 
202533d70fb6SKevin Wolf     ret = fd_open(bs);
202633d70fb6SKevin Wolf     if (ret < 0) {
202733d70fb6SKevin Wolf         return ret;
202833d70fb6SKevin Wolf     }
2029c1bb86cdSEric Blake 
203006dc9bd5SKevin Wolf     acb = (RawPosixAIOData) {
203106dc9bd5SKevin Wolf         .bs             = bs,
203206dc9bd5SKevin Wolf         .aio_fildes     = s->fd,
203306dc9bd5SKevin Wolf         .aio_type       = QEMU_AIO_FLUSH,
203406dc9bd5SKevin Wolf     };
203506dc9bd5SKevin Wolf 
203606dc9bd5SKevin Wolf     return raw_thread_pool_submit(bs, handle_aiocb_flush, &acb);
2037c1bb86cdSEric Blake }
2038c1bb86cdSEric Blake 
2039ed6e2161SNishanth Aravamudan static void raw_aio_attach_aio_context(BlockDriverState *bs,
2040ed6e2161SNishanth Aravamudan                                        AioContext *new_context)
2041ed6e2161SNishanth Aravamudan {
2042ed6e2161SNishanth Aravamudan #ifdef CONFIG_LINUX_AIO
2043ed6e2161SNishanth Aravamudan     BDRVRawState *s = bs->opaque;
2044ed6e2161SNishanth Aravamudan     if (s->use_linux_aio) {
2045ed6e2161SNishanth Aravamudan         Error *local_err;
2046ed6e2161SNishanth Aravamudan         if (!aio_setup_linux_aio(new_context, &local_err)) {
2047ed6e2161SNishanth Aravamudan             error_reportf_err(local_err, "Unable to use native AIO, "
2048ed6e2161SNishanth Aravamudan                                          "falling back to thread pool: ");
2049ed6e2161SNishanth Aravamudan             s->use_linux_aio = false;
2050ed6e2161SNishanth Aravamudan         }
2051ed6e2161SNishanth Aravamudan     }
2052ed6e2161SNishanth Aravamudan #endif
2053ed6e2161SNishanth Aravamudan }
2054ed6e2161SNishanth Aravamudan 
2055c1bb86cdSEric Blake static void raw_close(BlockDriverState *bs)
2056c1bb86cdSEric Blake {
2057c1bb86cdSEric Blake     BDRVRawState *s = bs->opaque;
2058c1bb86cdSEric Blake 
2059c1bb86cdSEric Blake     if (s->fd >= 0) {
2060c1bb86cdSEric Blake         qemu_close(s->fd);
2061c1bb86cdSEric Blake         s->fd = -1;
2062c1bb86cdSEric Blake     }
2063c1bb86cdSEric Blake }
2064c1bb86cdSEric Blake 
2065d0bc9e5dSMax Reitz /**
2066d0bc9e5dSMax Reitz  * Truncates the given regular file @fd to @offset and, when growing, fills the
2067d0bc9e5dSMax Reitz  * new space according to @prealloc.
2068d0bc9e5dSMax Reitz  *
2069d0bc9e5dSMax Reitz  * Returns: 0 on success, -errno on failure.
2070d0bc9e5dSMax Reitz  */
207193f4e2ffSKevin Wolf static int coroutine_fn
207293f4e2ffSKevin Wolf raw_regular_truncate(BlockDriverState *bs, int fd, int64_t offset,
207393f4e2ffSKevin Wolf                      PreallocMode prealloc, Error **errp)
20749f63b07eSMax Reitz {
207529cb4c01SKevin Wolf     RawPosixAIOData acb;
2076d0bc9e5dSMax Reitz 
207729cb4c01SKevin Wolf     acb = (RawPosixAIOData) {
207893f4e2ffSKevin Wolf         .bs             = bs,
207993f4e2ffSKevin Wolf         .aio_fildes     = fd,
208093f4e2ffSKevin Wolf         .aio_type       = QEMU_AIO_TRUNCATE,
208193f4e2ffSKevin Wolf         .aio_offset     = offset,
2082d57c44d0SKevin Wolf         .truncate       = {
208393f4e2ffSKevin Wolf             .prealloc       = prealloc,
208493f4e2ffSKevin Wolf             .errp           = errp,
2085d57c44d0SKevin Wolf         },
208693f4e2ffSKevin Wolf     };
2087d0bc9e5dSMax Reitz 
208829cb4c01SKevin Wolf     return raw_thread_pool_submit(bs, handle_aiocb_truncate, &acb);
20899f63b07eSMax Reitz }
20909f63b07eSMax Reitz 
2091061ca8a3SKevin Wolf static int coroutine_fn raw_co_truncate(BlockDriverState *bs, int64_t offset,
20928243ccb7SMax Reitz                                         PreallocMode prealloc, Error **errp)
2093c1bb86cdSEric Blake {
2094c1bb86cdSEric Blake     BDRVRawState *s = bs->opaque;
2095c1bb86cdSEric Blake     struct stat st;
2096f59adb32SMax Reitz     int ret;
2097c1bb86cdSEric Blake 
2098c1bb86cdSEric Blake     if (fstat(s->fd, &st)) {
2099f59adb32SMax Reitz         ret = -errno;
2100f59adb32SMax Reitz         error_setg_errno(errp, -ret, "Failed to fstat() the file");
2101f59adb32SMax Reitz         return ret;
2102c1bb86cdSEric Blake     }
2103c1bb86cdSEric Blake 
2104c1bb86cdSEric Blake     if (S_ISREG(st.st_mode)) {
210593f4e2ffSKevin Wolf         return raw_regular_truncate(bs, s->fd, offset, prealloc, errp);
2106c1bb86cdSEric Blake     }
210735d72602SMax Reitz 
210835d72602SMax Reitz     if (prealloc != PREALLOC_MODE_OFF) {
210935d72602SMax Reitz         error_setg(errp, "Preallocation mode '%s' unsupported for this "
2110977c736fSMarkus Armbruster                    "non-regular file", PreallocMode_str(prealloc));
211135d72602SMax Reitz         return -ENOTSUP;
211235d72602SMax Reitz     }
211335d72602SMax Reitz 
211435d72602SMax Reitz     if (S_ISCHR(st.st_mode) || S_ISBLK(st.st_mode)) {
2115c1bb86cdSEric Blake         if (offset > raw_getlength(bs)) {
2116f59adb32SMax Reitz             error_setg(errp, "Cannot grow device files");
2117c1bb86cdSEric Blake             return -EINVAL;
2118c1bb86cdSEric Blake         }
2119c1bb86cdSEric Blake     } else {
2120f59adb32SMax Reitz         error_setg(errp, "Resizing this file is not supported");
2121c1bb86cdSEric Blake         return -ENOTSUP;
2122c1bb86cdSEric Blake     }
2123c1bb86cdSEric Blake 
2124c1bb86cdSEric Blake     return 0;
2125c1bb86cdSEric Blake }
2126c1bb86cdSEric Blake 
2127c1bb86cdSEric Blake #ifdef __OpenBSD__
2128c1bb86cdSEric Blake static int64_t raw_getlength(BlockDriverState *bs)
2129c1bb86cdSEric Blake {
2130c1bb86cdSEric Blake     BDRVRawState *s = bs->opaque;
2131c1bb86cdSEric Blake     int fd = s->fd;
2132c1bb86cdSEric Blake     struct stat st;
2133c1bb86cdSEric Blake 
2134c1bb86cdSEric Blake     if (fstat(fd, &st))
2135c1bb86cdSEric Blake         return -errno;
2136c1bb86cdSEric Blake     if (S_ISCHR(st.st_mode) || S_ISBLK(st.st_mode)) {
2137c1bb86cdSEric Blake         struct disklabel dl;
2138c1bb86cdSEric Blake 
2139c1bb86cdSEric Blake         if (ioctl(fd, DIOCGDINFO, &dl))
2140c1bb86cdSEric Blake             return -errno;
2141c1bb86cdSEric Blake         return (uint64_t)dl.d_secsize *
2142c1bb86cdSEric Blake             dl.d_partitions[DISKPART(st.st_rdev)].p_size;
2143c1bb86cdSEric Blake     } else
2144c1bb86cdSEric Blake         return st.st_size;
2145c1bb86cdSEric Blake }
2146c1bb86cdSEric Blake #elif defined(__NetBSD__)
2147c1bb86cdSEric Blake static int64_t raw_getlength(BlockDriverState *bs)
2148c1bb86cdSEric Blake {
2149c1bb86cdSEric Blake     BDRVRawState *s = bs->opaque;
2150c1bb86cdSEric Blake     int fd = s->fd;
2151c1bb86cdSEric Blake     struct stat st;
2152c1bb86cdSEric Blake 
2153c1bb86cdSEric Blake     if (fstat(fd, &st))
2154c1bb86cdSEric Blake         return -errno;
2155c1bb86cdSEric Blake     if (S_ISCHR(st.st_mode) || S_ISBLK(st.st_mode)) {
2156c1bb86cdSEric Blake         struct dkwedge_info dkw;
2157c1bb86cdSEric Blake 
2158c1bb86cdSEric Blake         if (ioctl(fd, DIOCGWEDGEINFO, &dkw) != -1) {
2159c1bb86cdSEric Blake             return dkw.dkw_size * 512;
2160c1bb86cdSEric Blake         } else {
2161c1bb86cdSEric Blake             struct disklabel dl;
2162c1bb86cdSEric Blake 
2163c1bb86cdSEric Blake             if (ioctl(fd, DIOCGDINFO, &dl))
2164c1bb86cdSEric Blake                 return -errno;
2165c1bb86cdSEric Blake             return (uint64_t)dl.d_secsize *
2166c1bb86cdSEric Blake                 dl.d_partitions[DISKPART(st.st_rdev)].p_size;
2167c1bb86cdSEric Blake         }
2168c1bb86cdSEric Blake     } else
2169c1bb86cdSEric Blake         return st.st_size;
2170c1bb86cdSEric Blake }
2171c1bb86cdSEric Blake #elif defined(__sun__)
2172c1bb86cdSEric Blake static int64_t raw_getlength(BlockDriverState *bs)
2173c1bb86cdSEric Blake {
2174c1bb86cdSEric Blake     BDRVRawState *s = bs->opaque;
2175c1bb86cdSEric Blake     struct dk_minfo minfo;
2176c1bb86cdSEric Blake     int ret;
2177c1bb86cdSEric Blake     int64_t size;
2178c1bb86cdSEric Blake 
2179c1bb86cdSEric Blake     ret = fd_open(bs);
2180c1bb86cdSEric Blake     if (ret < 0) {
2181c1bb86cdSEric Blake         return ret;
2182c1bb86cdSEric Blake     }
2183c1bb86cdSEric Blake 
2184c1bb86cdSEric Blake     /*
2185c1bb86cdSEric Blake      * Use the DKIOCGMEDIAINFO ioctl to read the size.
2186c1bb86cdSEric Blake      */
2187c1bb86cdSEric Blake     ret = ioctl(s->fd, DKIOCGMEDIAINFO, &minfo);
2188c1bb86cdSEric Blake     if (ret != -1) {
2189c1bb86cdSEric Blake         return minfo.dki_lbsize * minfo.dki_capacity;
2190c1bb86cdSEric Blake     }
2191c1bb86cdSEric Blake 
2192c1bb86cdSEric Blake     /*
2193c1bb86cdSEric Blake      * There are reports that lseek on some devices fails, but
2194c1bb86cdSEric Blake      * irc discussion said that contingency on contingency was overkill.
2195c1bb86cdSEric Blake      */
2196c1bb86cdSEric Blake     size = lseek(s->fd, 0, SEEK_END);
2197c1bb86cdSEric Blake     if (size < 0) {
2198c1bb86cdSEric Blake         return -errno;
2199c1bb86cdSEric Blake     }
2200c1bb86cdSEric Blake     return size;
2201c1bb86cdSEric Blake }
2202c1bb86cdSEric Blake #elif defined(CONFIG_BSD)
2203c1bb86cdSEric Blake static int64_t raw_getlength(BlockDriverState *bs)
2204c1bb86cdSEric Blake {
2205c1bb86cdSEric Blake     BDRVRawState *s = bs->opaque;
2206c1bb86cdSEric Blake     int fd = s->fd;
2207c1bb86cdSEric Blake     int64_t size;
2208c1bb86cdSEric Blake     struct stat sb;
2209c1bb86cdSEric Blake #if defined (__FreeBSD__) || defined(__FreeBSD_kernel__)
2210c1bb86cdSEric Blake     int reopened = 0;
2211c1bb86cdSEric Blake #endif
2212c1bb86cdSEric Blake     int ret;
2213c1bb86cdSEric Blake 
2214c1bb86cdSEric Blake     ret = fd_open(bs);
2215c1bb86cdSEric Blake     if (ret < 0)
2216c1bb86cdSEric Blake         return ret;
2217c1bb86cdSEric Blake 
2218c1bb86cdSEric Blake #if defined (__FreeBSD__) || defined(__FreeBSD_kernel__)
2219c1bb86cdSEric Blake again:
2220c1bb86cdSEric Blake #endif
2221c1bb86cdSEric Blake     if (!fstat(fd, &sb) && (S_IFCHR & sb.st_mode)) {
2222c1bb86cdSEric Blake #ifdef DIOCGMEDIASIZE
2223c1bb86cdSEric Blake         if (ioctl(fd, DIOCGMEDIASIZE, (off_t *)&size))
2224c1bb86cdSEric Blake #elif defined(DIOCGPART)
2225c1bb86cdSEric Blake         {
2226c1bb86cdSEric Blake                 struct partinfo pi;
2227c1bb86cdSEric Blake                 if (ioctl(fd, DIOCGPART, &pi) == 0)
2228c1bb86cdSEric Blake                         size = pi.media_size;
2229c1bb86cdSEric Blake                 else
2230c1bb86cdSEric Blake                         size = 0;
2231c1bb86cdSEric Blake         }
2232c1bb86cdSEric Blake         if (size == 0)
2233c1bb86cdSEric Blake #endif
2234c1bb86cdSEric Blake #if defined(__APPLE__) && defined(__MACH__)
2235c1bb86cdSEric Blake         {
2236c1bb86cdSEric Blake             uint64_t sectors = 0;
2237c1bb86cdSEric Blake             uint32_t sector_size = 0;
2238c1bb86cdSEric Blake 
2239c1bb86cdSEric Blake             if (ioctl(fd, DKIOCGETBLOCKCOUNT, &sectors) == 0
2240c1bb86cdSEric Blake                && ioctl(fd, DKIOCGETBLOCKSIZE, &sector_size) == 0) {
2241c1bb86cdSEric Blake                 size = sectors * sector_size;
2242c1bb86cdSEric Blake             } else {
2243c1bb86cdSEric Blake                 size = lseek(fd, 0LL, SEEK_END);
2244c1bb86cdSEric Blake                 if (size < 0) {
2245c1bb86cdSEric Blake                     return -errno;
2246c1bb86cdSEric Blake                 }
2247c1bb86cdSEric Blake             }
2248c1bb86cdSEric Blake         }
2249c1bb86cdSEric Blake #else
2250c1bb86cdSEric Blake         size = lseek(fd, 0LL, SEEK_END);
2251c1bb86cdSEric Blake         if (size < 0) {
2252c1bb86cdSEric Blake             return -errno;
2253c1bb86cdSEric Blake         }
2254c1bb86cdSEric Blake #endif
2255c1bb86cdSEric Blake #if defined(__FreeBSD__) || defined(__FreeBSD_kernel__)
2256c1bb86cdSEric Blake         switch(s->type) {
2257c1bb86cdSEric Blake         case FTYPE_CD:
2258c1bb86cdSEric Blake             /* XXX FreeBSD acd returns UINT_MAX sectors for an empty drive */
2259c1bb86cdSEric Blake             if (size == 2048LL * (unsigned)-1)
2260c1bb86cdSEric Blake                 size = 0;
2261c1bb86cdSEric Blake             /* XXX no disc?  maybe we need to reopen... */
2262c1bb86cdSEric Blake             if (size <= 0 && !reopened && cdrom_reopen(bs) >= 0) {
2263c1bb86cdSEric Blake                 reopened = 1;
2264c1bb86cdSEric Blake                 goto again;
2265c1bb86cdSEric Blake             }
2266c1bb86cdSEric Blake         }
2267c1bb86cdSEric Blake #endif
2268c1bb86cdSEric Blake     } else {
2269c1bb86cdSEric Blake         size = lseek(fd, 0, SEEK_END);
2270c1bb86cdSEric Blake         if (size < 0) {
2271c1bb86cdSEric Blake             return -errno;
2272c1bb86cdSEric Blake         }
2273c1bb86cdSEric Blake     }
2274c1bb86cdSEric Blake     return size;
2275c1bb86cdSEric Blake }
2276c1bb86cdSEric Blake #else
2277c1bb86cdSEric Blake static int64_t raw_getlength(BlockDriverState *bs)
2278c1bb86cdSEric Blake {
2279c1bb86cdSEric Blake     BDRVRawState *s = bs->opaque;
2280c1bb86cdSEric Blake     int ret;
2281c1bb86cdSEric Blake     int64_t size;
2282c1bb86cdSEric Blake 
2283c1bb86cdSEric Blake     ret = fd_open(bs);
2284c1bb86cdSEric Blake     if (ret < 0) {
2285c1bb86cdSEric Blake         return ret;
2286c1bb86cdSEric Blake     }
2287c1bb86cdSEric Blake 
2288c1bb86cdSEric Blake     size = lseek(s->fd, 0, SEEK_END);
2289c1bb86cdSEric Blake     if (size < 0) {
2290c1bb86cdSEric Blake         return -errno;
2291c1bb86cdSEric Blake     }
2292c1bb86cdSEric Blake     return size;
2293c1bb86cdSEric Blake }
2294c1bb86cdSEric Blake #endif
2295c1bb86cdSEric Blake 
2296c1bb86cdSEric Blake static int64_t raw_get_allocated_file_size(BlockDriverState *bs)
2297c1bb86cdSEric Blake {
2298c1bb86cdSEric Blake     struct stat st;
2299c1bb86cdSEric Blake     BDRVRawState *s = bs->opaque;
2300c1bb86cdSEric Blake 
2301c1bb86cdSEric Blake     if (fstat(s->fd, &st) < 0) {
2302c1bb86cdSEric Blake         return -errno;
2303c1bb86cdSEric Blake     }
2304c1bb86cdSEric Blake     return (int64_t)st.st_blocks * 512;
2305c1bb86cdSEric Blake }
2306c1bb86cdSEric Blake 
230793f4e2ffSKevin Wolf static int coroutine_fn
230893f4e2ffSKevin Wolf raw_co_create(BlockdevCreateOptions *options, Error **errp)
2309c1bb86cdSEric Blake {
2310927f11e1SKevin Wolf     BlockdevCreateOptionsFile *file_opts;
23117c20c808SMax Reitz     Error *local_err = NULL;
2312c1bb86cdSEric Blake     int fd;
2313d815efcaSMax Reitz     uint64_t perm, shared;
2314c1bb86cdSEric Blake     int result = 0;
2315c1bb86cdSEric Blake 
2316927f11e1SKevin Wolf     /* Validate options and set default values */
2317927f11e1SKevin Wolf     assert(options->driver == BLOCKDEV_DRIVER_FILE);
2318927f11e1SKevin Wolf     file_opts = &options->u.file;
2319c1bb86cdSEric Blake 
2320927f11e1SKevin Wolf     if (!file_opts->has_nocow) {
2321927f11e1SKevin Wolf         file_opts->nocow = false;
2322927f11e1SKevin Wolf     }
2323927f11e1SKevin Wolf     if (!file_opts->has_preallocation) {
2324927f11e1SKevin Wolf         file_opts->preallocation = PREALLOC_MODE_OFF;
2325c1bb86cdSEric Blake     }
2326c1bb86cdSEric Blake 
2327927f11e1SKevin Wolf     /* Create file */
2328b8cf1913SMax Reitz     fd = qemu_open(file_opts->filename, O_RDWR | O_CREAT | O_BINARY, 0644);
2329c1bb86cdSEric Blake     if (fd < 0) {
2330c1bb86cdSEric Blake         result = -errno;
2331c1bb86cdSEric Blake         error_setg_errno(errp, -result, "Could not create file");
2332c1bb86cdSEric Blake         goto out;
2333c1bb86cdSEric Blake     }
2334c1bb86cdSEric Blake 
2335b8cf1913SMax Reitz     /* Take permissions: We want to discard everything, so we need
2336b8cf1913SMax Reitz      * BLK_PERM_WRITE; and truncation to the desired size requires
2337b8cf1913SMax Reitz      * BLK_PERM_RESIZE.
2338b8cf1913SMax Reitz      * On the other hand, we cannot share the RESIZE permission
2339b8cf1913SMax Reitz      * because we promise that after this function, the file has the
2340b8cf1913SMax Reitz      * size given in the options.  If someone else were to resize it
2341b8cf1913SMax Reitz      * concurrently, we could not guarantee that.
2342b8cf1913SMax Reitz      * Note that after this function, we can no longer guarantee that
2343b8cf1913SMax Reitz      * the file is not touched by a third party, so it may be resized
2344b8cf1913SMax Reitz      * then. */
2345b8cf1913SMax Reitz     perm = BLK_PERM_WRITE | BLK_PERM_RESIZE;
2346b8cf1913SMax Reitz     shared = BLK_PERM_ALL & ~BLK_PERM_RESIZE;
2347b8cf1913SMax Reitz 
2348b8cf1913SMax Reitz     /* Step one: Take locks */
23492996ffadSFam Zheng     result = raw_apply_lock_bytes(NULL, fd, perm, ~shared, false, errp);
2350b8cf1913SMax Reitz     if (result < 0) {
2351b8cf1913SMax Reitz         goto out_close;
2352b8cf1913SMax Reitz     }
2353b8cf1913SMax Reitz 
2354b8cf1913SMax Reitz     /* Step two: Check that nobody else has taken conflicting locks */
2355b8cf1913SMax Reitz     result = raw_check_lock_bytes(fd, perm, shared, errp);
2356b8cf1913SMax Reitz     if (result < 0) {
2357b857431dSFam Zheng         error_append_hint(errp,
2358b857431dSFam Zheng                           "Is another process using the image [%s]?\n",
2359b857431dSFam Zheng                           file_opts->filename);
23607c20c808SMax Reitz         goto out_unlock;
2361b8cf1913SMax Reitz     }
2362b8cf1913SMax Reitz 
2363b8cf1913SMax Reitz     /* Clear the file by truncating it to 0 */
236493f4e2ffSKevin Wolf     result = raw_regular_truncate(NULL, fd, 0, PREALLOC_MODE_OFF, errp);
2365b8cf1913SMax Reitz     if (result < 0) {
23667c20c808SMax Reitz         goto out_unlock;
2367b8cf1913SMax Reitz     }
2368b8cf1913SMax Reitz 
2369927f11e1SKevin Wolf     if (file_opts->nocow) {
2370c1bb86cdSEric Blake #ifdef __linux__
2371c1bb86cdSEric Blake         /* Set NOCOW flag to solve performance issue on fs like btrfs.
2372c1bb86cdSEric Blake          * This is an optimisation. The FS_IOC_SETFLAGS ioctl return value
2373c1bb86cdSEric Blake          * will be ignored since any failure of this operation should not
2374c1bb86cdSEric Blake          * block the left work.
2375c1bb86cdSEric Blake          */
2376c1bb86cdSEric Blake         int attr;
2377c1bb86cdSEric Blake         if (ioctl(fd, FS_IOC_GETFLAGS, &attr) == 0) {
2378c1bb86cdSEric Blake             attr |= FS_NOCOW_FL;
2379c1bb86cdSEric Blake             ioctl(fd, FS_IOC_SETFLAGS, &attr);
2380c1bb86cdSEric Blake         }
2381c1bb86cdSEric Blake #endif
2382c1bb86cdSEric Blake     }
2383c1bb86cdSEric Blake 
2384b8cf1913SMax Reitz     /* Resize and potentially preallocate the file to the desired
2385b8cf1913SMax Reitz      * final size */
238693f4e2ffSKevin Wolf     result = raw_regular_truncate(NULL, fd, file_opts->size,
238793f4e2ffSKevin Wolf                                   file_opts->preallocation, errp);
23889f63b07eSMax Reitz     if (result < 0) {
23897c20c808SMax Reitz         goto out_unlock;
23907c20c808SMax Reitz     }
23917c20c808SMax Reitz 
23927c20c808SMax Reitz out_unlock:
23932996ffadSFam Zheng     raw_apply_lock_bytes(NULL, fd, 0, 0, true, &local_err);
23947c20c808SMax Reitz     if (local_err) {
23957c20c808SMax Reitz         /* The above call should not fail, and if it does, that does
23967c20c808SMax Reitz          * not mean the whole creation operation has failed.  So
23977c20c808SMax Reitz          * report it the user for their convenience, but do not report
23987c20c808SMax Reitz          * it to the caller. */
2399db0754dfSFam Zheng         warn_report_err(local_err);
24005a1dad9dSNir Soffer     }
24015a1dad9dSNir Soffer 
24025a1dad9dSNir Soffer out_close:
2403c1bb86cdSEric Blake     if (qemu_close(fd) != 0 && result == 0) {
2404c1bb86cdSEric Blake         result = -errno;
2405c1bb86cdSEric Blake         error_setg_errno(errp, -result, "Could not close the new file");
2406c1bb86cdSEric Blake     }
2407c1bb86cdSEric Blake out:
2408c1bb86cdSEric Blake     return result;
2409c1bb86cdSEric Blake }
2410c1bb86cdSEric Blake 
2411927f11e1SKevin Wolf static int coroutine_fn raw_co_create_opts(const char *filename, QemuOpts *opts,
2412927f11e1SKevin Wolf                                            Error **errp)
2413927f11e1SKevin Wolf {
2414927f11e1SKevin Wolf     BlockdevCreateOptions options;
2415927f11e1SKevin Wolf     int64_t total_size = 0;
2416927f11e1SKevin Wolf     bool nocow = false;
2417927f11e1SKevin Wolf     PreallocMode prealloc;
2418927f11e1SKevin Wolf     char *buf = NULL;
2419927f11e1SKevin Wolf     Error *local_err = NULL;
2420927f11e1SKevin Wolf 
2421927f11e1SKevin Wolf     /* Skip file: protocol prefix */
2422927f11e1SKevin Wolf     strstart(filename, "file:", &filename);
2423927f11e1SKevin Wolf 
2424927f11e1SKevin Wolf     /* Read out options */
2425927f11e1SKevin Wolf     total_size = ROUND_UP(qemu_opt_get_size_del(opts, BLOCK_OPT_SIZE, 0),
2426927f11e1SKevin Wolf                           BDRV_SECTOR_SIZE);
2427927f11e1SKevin Wolf     nocow = qemu_opt_get_bool(opts, BLOCK_OPT_NOCOW, false);
2428927f11e1SKevin Wolf     buf = qemu_opt_get_del(opts, BLOCK_OPT_PREALLOC);
2429927f11e1SKevin Wolf     prealloc = qapi_enum_parse(&PreallocMode_lookup, buf,
2430927f11e1SKevin Wolf                                PREALLOC_MODE_OFF, &local_err);
2431927f11e1SKevin Wolf     g_free(buf);
2432927f11e1SKevin Wolf     if (local_err) {
2433927f11e1SKevin Wolf         error_propagate(errp, local_err);
2434927f11e1SKevin Wolf         return -EINVAL;
2435927f11e1SKevin Wolf     }
2436927f11e1SKevin Wolf 
2437927f11e1SKevin Wolf     options = (BlockdevCreateOptions) {
2438927f11e1SKevin Wolf         .driver     = BLOCKDEV_DRIVER_FILE,
2439927f11e1SKevin Wolf         .u.file     = {
2440927f11e1SKevin Wolf             .filename           = (char *) filename,
2441927f11e1SKevin Wolf             .size               = total_size,
2442927f11e1SKevin Wolf             .has_preallocation  = true,
2443927f11e1SKevin Wolf             .preallocation      = prealloc,
2444927f11e1SKevin Wolf             .has_nocow          = true,
2445927f11e1SKevin Wolf             .nocow              = nocow,
2446927f11e1SKevin Wolf         },
2447927f11e1SKevin Wolf     };
2448927f11e1SKevin Wolf     return raw_co_create(&options, errp);
2449927f11e1SKevin Wolf }
2450927f11e1SKevin Wolf 
2451c1bb86cdSEric Blake /*
2452c1bb86cdSEric Blake  * Find allocation range in @bs around offset @start.
2453c1bb86cdSEric Blake  * May change underlying file descriptor's file offset.
2454c1bb86cdSEric Blake  * If @start is not in a hole, store @start in @data, and the
2455c1bb86cdSEric Blake  * beginning of the next hole in @hole, and return 0.
2456c1bb86cdSEric Blake  * If @start is in a non-trailing hole, store @start in @hole and the
2457c1bb86cdSEric Blake  * beginning of the next non-hole in @data, and return 0.
2458c1bb86cdSEric Blake  * If @start is in a trailing hole or beyond EOF, return -ENXIO.
2459c1bb86cdSEric Blake  * If we can't find out, return a negative errno other than -ENXIO.
2460c1bb86cdSEric Blake  */
2461c1bb86cdSEric Blake static int find_allocation(BlockDriverState *bs, off_t start,
2462c1bb86cdSEric Blake                            off_t *data, off_t *hole)
2463c1bb86cdSEric Blake {
2464c1bb86cdSEric Blake #if defined SEEK_HOLE && defined SEEK_DATA
2465c1bb86cdSEric Blake     BDRVRawState *s = bs->opaque;
2466c1bb86cdSEric Blake     off_t offs;
2467c1bb86cdSEric Blake 
2468c1bb86cdSEric Blake     /*
2469c1bb86cdSEric Blake      * SEEK_DATA cases:
2470c1bb86cdSEric Blake      * D1. offs == start: start is in data
2471c1bb86cdSEric Blake      * D2. offs > start: start is in a hole, next data at offs
2472c1bb86cdSEric Blake      * D3. offs < 0, errno = ENXIO: either start is in a trailing hole
2473c1bb86cdSEric Blake      *                              or start is beyond EOF
2474c1bb86cdSEric Blake      *     If the latter happens, the file has been truncated behind
2475c1bb86cdSEric Blake      *     our back since we opened it.  All bets are off then.
2476c1bb86cdSEric Blake      *     Treating like a trailing hole is simplest.
2477c1bb86cdSEric Blake      * D4. offs < 0, errno != ENXIO: we learned nothing
2478c1bb86cdSEric Blake      */
2479c1bb86cdSEric Blake     offs = lseek(s->fd, start, SEEK_DATA);
2480c1bb86cdSEric Blake     if (offs < 0) {
2481c1bb86cdSEric Blake         return -errno;          /* D3 or D4 */
2482c1bb86cdSEric Blake     }
2483a03083a0SJeff Cody 
2484a03083a0SJeff Cody     if (offs < start) {
2485a03083a0SJeff Cody         /* This is not a valid return by lseek().  We are safe to just return
2486a03083a0SJeff Cody          * -EIO in this case, and we'll treat it like D4. */
2487a03083a0SJeff Cody         return -EIO;
2488a03083a0SJeff Cody     }
2489c1bb86cdSEric Blake 
2490c1bb86cdSEric Blake     if (offs > start) {
2491c1bb86cdSEric Blake         /* D2: in hole, next data at offs */
2492c1bb86cdSEric Blake         *hole = start;
2493c1bb86cdSEric Blake         *data = offs;
2494c1bb86cdSEric Blake         return 0;
2495c1bb86cdSEric Blake     }
2496c1bb86cdSEric Blake 
2497c1bb86cdSEric Blake     /* D1: in data, end not yet known */
2498c1bb86cdSEric Blake 
2499c1bb86cdSEric Blake     /*
2500c1bb86cdSEric Blake      * SEEK_HOLE cases:
2501c1bb86cdSEric Blake      * H1. offs == start: start is in a hole
2502c1bb86cdSEric Blake      *     If this happens here, a hole has been dug behind our back
2503c1bb86cdSEric Blake      *     since the previous lseek().
2504c1bb86cdSEric Blake      * H2. offs > start: either start is in data, next hole at offs,
2505c1bb86cdSEric Blake      *                   or start is in trailing hole, EOF at offs
2506c1bb86cdSEric Blake      *     Linux treats trailing holes like any other hole: offs ==
2507c1bb86cdSEric Blake      *     start.  Solaris seeks to EOF instead: offs > start (blech).
2508c1bb86cdSEric Blake      *     If that happens here, a hole has been dug behind our back
2509c1bb86cdSEric Blake      *     since the previous lseek().
2510c1bb86cdSEric Blake      * H3. offs < 0, errno = ENXIO: start is beyond EOF
2511c1bb86cdSEric Blake      *     If this happens, the file has been truncated behind our
2512c1bb86cdSEric Blake      *     back since we opened it.  Treat it like a trailing hole.
2513c1bb86cdSEric Blake      * H4. offs < 0, errno != ENXIO: we learned nothing
2514c1bb86cdSEric Blake      *     Pretend we know nothing at all, i.e. "forget" about D1.
2515c1bb86cdSEric Blake      */
2516c1bb86cdSEric Blake     offs = lseek(s->fd, start, SEEK_HOLE);
2517c1bb86cdSEric Blake     if (offs < 0) {
2518c1bb86cdSEric Blake         return -errno;          /* D1 and (H3 or H4) */
2519c1bb86cdSEric Blake     }
2520a03083a0SJeff Cody 
2521a03083a0SJeff Cody     if (offs < start) {
2522a03083a0SJeff Cody         /* This is not a valid return by lseek().  We are safe to just return
2523a03083a0SJeff Cody          * -EIO in this case, and we'll treat it like H4. */
2524a03083a0SJeff Cody         return -EIO;
2525a03083a0SJeff Cody     }
2526c1bb86cdSEric Blake 
2527c1bb86cdSEric Blake     if (offs > start) {
2528c1bb86cdSEric Blake         /*
2529c1bb86cdSEric Blake          * D1 and H2: either in data, next hole at offs, or it was in
2530c1bb86cdSEric Blake          * data but is now in a trailing hole.  In the latter case,
2531c1bb86cdSEric Blake          * all bets are off.  Treating it as if it there was data all
2532c1bb86cdSEric Blake          * the way to EOF is safe, so simply do that.
2533c1bb86cdSEric Blake          */
2534c1bb86cdSEric Blake         *data = start;
2535c1bb86cdSEric Blake         *hole = offs;
2536c1bb86cdSEric Blake         return 0;
2537c1bb86cdSEric Blake     }
2538c1bb86cdSEric Blake 
2539c1bb86cdSEric Blake     /* D1 and H1 */
2540c1bb86cdSEric Blake     return -EBUSY;
2541c1bb86cdSEric Blake #else
2542c1bb86cdSEric Blake     return -ENOTSUP;
2543c1bb86cdSEric Blake #endif
2544c1bb86cdSEric Blake }
2545c1bb86cdSEric Blake 
2546c1bb86cdSEric Blake /*
2547a290f085SEric Blake  * Returns the allocation status of the specified offset.
2548c1bb86cdSEric Blake  *
2549a290f085SEric Blake  * The block layer guarantees 'offset' and 'bytes' are within bounds.
2550c1bb86cdSEric Blake  *
2551a290f085SEric Blake  * 'pnum' is set to the number of bytes (including and immediately following
2552a290f085SEric Blake  * the specified offset) that are known to be in the same
2553c1bb86cdSEric Blake  * allocated/unallocated state.
2554c1bb86cdSEric Blake  *
2555a290f085SEric Blake  * 'bytes' is the max value 'pnum' should be set to.
2556c1bb86cdSEric Blake  */
2557a290f085SEric Blake static int coroutine_fn raw_co_block_status(BlockDriverState *bs,
2558a290f085SEric Blake                                             bool want_zero,
2559a290f085SEric Blake                                             int64_t offset,
2560a290f085SEric Blake                                             int64_t bytes, int64_t *pnum,
2561a290f085SEric Blake                                             int64_t *map,
2562c1bb86cdSEric Blake                                             BlockDriverState **file)
2563c1bb86cdSEric Blake {
2564a290f085SEric Blake     off_t data = 0, hole = 0;
2565c1bb86cdSEric Blake     int ret;
2566c1bb86cdSEric Blake 
25679c3db310SMax Reitz     assert(QEMU_IS_ALIGNED(offset | bytes, bs->bl.request_alignment));
25689c3db310SMax Reitz 
2569c1bb86cdSEric Blake     ret = fd_open(bs);
2570c1bb86cdSEric Blake     if (ret < 0) {
2571c1bb86cdSEric Blake         return ret;
2572c1bb86cdSEric Blake     }
2573c1bb86cdSEric Blake 
2574a290f085SEric Blake     if (!want_zero) {
2575a290f085SEric Blake         *pnum = bytes;
2576a290f085SEric Blake         *map = offset;
2577a290f085SEric Blake         *file = bs;
2578a290f085SEric Blake         return BDRV_BLOCK_DATA | BDRV_BLOCK_OFFSET_VALID;
2579c1bb86cdSEric Blake     }
2580c1bb86cdSEric Blake 
2581a290f085SEric Blake     ret = find_allocation(bs, offset, &data, &hole);
2582c1bb86cdSEric Blake     if (ret == -ENXIO) {
2583c1bb86cdSEric Blake         /* Trailing hole */
2584a290f085SEric Blake         *pnum = bytes;
2585c1bb86cdSEric Blake         ret = BDRV_BLOCK_ZERO;
2586c1bb86cdSEric Blake     } else if (ret < 0) {
2587c1bb86cdSEric Blake         /* No info available, so pretend there are no holes */
2588a290f085SEric Blake         *pnum = bytes;
2589c1bb86cdSEric Blake         ret = BDRV_BLOCK_DATA;
2590a290f085SEric Blake     } else if (data == offset) {
2591a290f085SEric Blake         /* On a data extent, compute bytes to the end of the extent,
2592c1bb86cdSEric Blake          * possibly including a partial sector at EOF. */
2593a290f085SEric Blake         *pnum = MIN(bytes, hole - offset);
25949c3db310SMax Reitz 
25959c3db310SMax Reitz         /*
25969c3db310SMax Reitz          * We are not allowed to return partial sectors, though, so
25979c3db310SMax Reitz          * round up if necessary.
25989c3db310SMax Reitz          */
25999c3db310SMax Reitz         if (!QEMU_IS_ALIGNED(*pnum, bs->bl.request_alignment)) {
26009c3db310SMax Reitz             int64_t file_length = raw_getlength(bs);
26019c3db310SMax Reitz             if (file_length > 0) {
26029c3db310SMax Reitz                 /* Ignore errors, this is just a safeguard */
26039c3db310SMax Reitz                 assert(hole == file_length);
26049c3db310SMax Reitz             }
26059c3db310SMax Reitz             *pnum = ROUND_UP(*pnum, bs->bl.request_alignment);
26069c3db310SMax Reitz         }
26079c3db310SMax Reitz 
2608c1bb86cdSEric Blake         ret = BDRV_BLOCK_DATA;
2609c1bb86cdSEric Blake     } else {
2610a290f085SEric Blake         /* On a hole, compute bytes to the beginning of the next extent.  */
2611a290f085SEric Blake         assert(hole == offset);
2612a290f085SEric Blake         *pnum = MIN(bytes, data - offset);
2613c1bb86cdSEric Blake         ret = BDRV_BLOCK_ZERO;
2614c1bb86cdSEric Blake     }
2615a290f085SEric Blake     *map = offset;
2616c1bb86cdSEric Blake     *file = bs;
2617a290f085SEric Blake     return ret | BDRV_BLOCK_OFFSET_VALID;
2618c1bb86cdSEric Blake }
2619c1bb86cdSEric Blake 
262031be8a2aSStefan Hajnoczi #if defined(__linux__)
262131be8a2aSStefan Hajnoczi /* Verify that the file is not in the page cache */
262231be8a2aSStefan Hajnoczi static void check_cache_dropped(BlockDriverState *bs, Error **errp)
262331be8a2aSStefan Hajnoczi {
262431be8a2aSStefan Hajnoczi     const size_t window_size = 128 * 1024 * 1024;
262531be8a2aSStefan Hajnoczi     BDRVRawState *s = bs->opaque;
262631be8a2aSStefan Hajnoczi     void *window = NULL;
262731be8a2aSStefan Hajnoczi     size_t length = 0;
262831be8a2aSStefan Hajnoczi     unsigned char *vec;
262931be8a2aSStefan Hajnoczi     size_t page_size;
263031be8a2aSStefan Hajnoczi     off_t offset;
263131be8a2aSStefan Hajnoczi     off_t end;
263231be8a2aSStefan Hajnoczi 
263331be8a2aSStefan Hajnoczi     /* mincore(2) page status information requires 1 byte per page */
263431be8a2aSStefan Hajnoczi     page_size = sysconf(_SC_PAGESIZE);
263531be8a2aSStefan Hajnoczi     vec = g_malloc(DIV_ROUND_UP(window_size, page_size));
263631be8a2aSStefan Hajnoczi 
263731be8a2aSStefan Hajnoczi     end = raw_getlength(bs);
263831be8a2aSStefan Hajnoczi 
263931be8a2aSStefan Hajnoczi     for (offset = 0; offset < end; offset += window_size) {
264031be8a2aSStefan Hajnoczi         void *new_window;
264131be8a2aSStefan Hajnoczi         size_t new_length;
264231be8a2aSStefan Hajnoczi         size_t vec_end;
264331be8a2aSStefan Hajnoczi         size_t i;
264431be8a2aSStefan Hajnoczi         int ret;
264531be8a2aSStefan Hajnoczi 
264631be8a2aSStefan Hajnoczi         /* Unmap previous window if size has changed */
264731be8a2aSStefan Hajnoczi         new_length = MIN(end - offset, window_size);
264831be8a2aSStefan Hajnoczi         if (new_length != length) {
264931be8a2aSStefan Hajnoczi             munmap(window, length);
265031be8a2aSStefan Hajnoczi             window = NULL;
265131be8a2aSStefan Hajnoczi             length = 0;
265231be8a2aSStefan Hajnoczi         }
265331be8a2aSStefan Hajnoczi 
265431be8a2aSStefan Hajnoczi         new_window = mmap(window, new_length, PROT_NONE, MAP_PRIVATE,
265531be8a2aSStefan Hajnoczi                           s->fd, offset);
265631be8a2aSStefan Hajnoczi         if (new_window == MAP_FAILED) {
265731be8a2aSStefan Hajnoczi             error_setg_errno(errp, errno, "mmap failed");
265831be8a2aSStefan Hajnoczi             break;
265931be8a2aSStefan Hajnoczi         }
266031be8a2aSStefan Hajnoczi 
266131be8a2aSStefan Hajnoczi         window = new_window;
266231be8a2aSStefan Hajnoczi         length = new_length;
266331be8a2aSStefan Hajnoczi 
266431be8a2aSStefan Hajnoczi         ret = mincore(window, length, vec);
266531be8a2aSStefan Hajnoczi         if (ret < 0) {
266631be8a2aSStefan Hajnoczi             error_setg_errno(errp, errno, "mincore failed");
266731be8a2aSStefan Hajnoczi             break;
266831be8a2aSStefan Hajnoczi         }
266931be8a2aSStefan Hajnoczi 
267031be8a2aSStefan Hajnoczi         vec_end = DIV_ROUND_UP(length, page_size);
267131be8a2aSStefan Hajnoczi         for (i = 0; i < vec_end; i++) {
267231be8a2aSStefan Hajnoczi             if (vec[i] & 0x1) {
267331be8a2aSStefan Hajnoczi                 error_setg(errp, "page cache still in use!");
267431be8a2aSStefan Hajnoczi                 break;
267531be8a2aSStefan Hajnoczi             }
267631be8a2aSStefan Hajnoczi         }
267731be8a2aSStefan Hajnoczi     }
267831be8a2aSStefan Hajnoczi 
267931be8a2aSStefan Hajnoczi     if (window) {
268031be8a2aSStefan Hajnoczi         munmap(window, length);
268131be8a2aSStefan Hajnoczi     }
268231be8a2aSStefan Hajnoczi 
268331be8a2aSStefan Hajnoczi     g_free(vec);
268431be8a2aSStefan Hajnoczi }
268531be8a2aSStefan Hajnoczi #endif /* __linux__ */
268631be8a2aSStefan Hajnoczi 
2687dd577a26SStefan Hajnoczi static void coroutine_fn raw_co_invalidate_cache(BlockDriverState *bs,
2688dd577a26SStefan Hajnoczi                                                  Error **errp)
2689dd577a26SStefan Hajnoczi {
2690dd577a26SStefan Hajnoczi     BDRVRawState *s = bs->opaque;
2691dd577a26SStefan Hajnoczi     int ret;
2692dd577a26SStefan Hajnoczi 
2693dd577a26SStefan Hajnoczi     ret = fd_open(bs);
2694dd577a26SStefan Hajnoczi     if (ret < 0) {
2695dd577a26SStefan Hajnoczi         error_setg_errno(errp, -ret, "The file descriptor is not open");
2696dd577a26SStefan Hajnoczi         return;
2697dd577a26SStefan Hajnoczi     }
2698dd577a26SStefan Hajnoczi 
2699f357fcd8SStefan Hajnoczi     if (!s->drop_cache) {
2700f357fcd8SStefan Hajnoczi         return;
2701f357fcd8SStefan Hajnoczi     }
2702f357fcd8SStefan Hajnoczi 
2703dd577a26SStefan Hajnoczi     if (s->open_flags & O_DIRECT) {
2704dd577a26SStefan Hajnoczi         return; /* No host kernel page cache */
2705dd577a26SStefan Hajnoczi     }
2706dd577a26SStefan Hajnoczi 
2707dd577a26SStefan Hajnoczi #if defined(__linux__)
2708dd577a26SStefan Hajnoczi     /* This sets the scene for the next syscall... */
2709dd577a26SStefan Hajnoczi     ret = bdrv_co_flush(bs);
2710dd577a26SStefan Hajnoczi     if (ret < 0) {
2711dd577a26SStefan Hajnoczi         error_setg_errno(errp, -ret, "flush failed");
2712dd577a26SStefan Hajnoczi         return;
2713dd577a26SStefan Hajnoczi     }
2714dd577a26SStefan Hajnoczi 
2715dd577a26SStefan Hajnoczi     /* Linux does not invalidate pages that are dirty, locked, or mmapped by a
2716dd577a26SStefan Hajnoczi      * process.  These limitations are okay because we just fsynced the file,
2717dd577a26SStefan Hajnoczi      * we don't use mmap, and the file should not be in use by other processes.
2718dd577a26SStefan Hajnoczi      */
2719dd577a26SStefan Hajnoczi     ret = posix_fadvise(s->fd, 0, 0, POSIX_FADV_DONTNEED);
2720dd577a26SStefan Hajnoczi     if (ret != 0) { /* the return value is a positive errno */
2721dd577a26SStefan Hajnoczi         error_setg_errno(errp, ret, "fadvise failed");
2722dd577a26SStefan Hajnoczi         return;
2723dd577a26SStefan Hajnoczi     }
272431be8a2aSStefan Hajnoczi 
272531be8a2aSStefan Hajnoczi     if (s->check_cache_dropped) {
272631be8a2aSStefan Hajnoczi         check_cache_dropped(bs, errp);
272731be8a2aSStefan Hajnoczi     }
2728dd577a26SStefan Hajnoczi #else /* __linux__ */
2729dd577a26SStefan Hajnoczi     /* Do nothing.  Live migration to a remote host with cache.direct=off is
2730dd577a26SStefan Hajnoczi      * unsupported on other host operating systems.  Cache consistency issues
2731dd577a26SStefan Hajnoczi      * may occur but no error is reported here, partly because that's the
2732dd577a26SStefan Hajnoczi      * historical behavior and partly because it's hard to differentiate valid
2733dd577a26SStefan Hajnoczi      * configurations that should not cause errors.
2734dd577a26SStefan Hajnoczi      */
2735dd577a26SStefan Hajnoczi #endif /* !__linux__ */
2736dd577a26SStefan Hajnoczi }
2737dd577a26SStefan Hajnoczi 
273833d70fb6SKevin Wolf static coroutine_fn int
273946ee0f46SKevin Wolf raw_do_pdiscard(BlockDriverState *bs, int64_t offset, int bytes, bool blkdev)
2740c1bb86cdSEric Blake {
2741c1bb86cdSEric Blake     BDRVRawState *s = bs->opaque;
274246ee0f46SKevin Wolf     RawPosixAIOData acb;
2743c1bb86cdSEric Blake 
274446ee0f46SKevin Wolf     acb = (RawPosixAIOData) {
274546ee0f46SKevin Wolf         .bs             = bs,
274646ee0f46SKevin Wolf         .aio_fildes     = s->fd,
274746ee0f46SKevin Wolf         .aio_type       = QEMU_AIO_DISCARD,
274846ee0f46SKevin Wolf         .aio_offset     = offset,
274946ee0f46SKevin Wolf         .aio_nbytes     = bytes,
275046ee0f46SKevin Wolf     };
275146ee0f46SKevin Wolf 
275246ee0f46SKevin Wolf     if (blkdev) {
275346ee0f46SKevin Wolf         acb.aio_type |= QEMU_AIO_BLKDEV;
275446ee0f46SKevin Wolf     }
275546ee0f46SKevin Wolf 
275646ee0f46SKevin Wolf     return raw_thread_pool_submit(bs, handle_aiocb_discard, &acb);
275746ee0f46SKevin Wolf }
275846ee0f46SKevin Wolf 
275946ee0f46SKevin Wolf static coroutine_fn int
276046ee0f46SKevin Wolf raw_co_pdiscard(BlockDriverState *bs, int64_t offset, int bytes)
276146ee0f46SKevin Wolf {
276246ee0f46SKevin Wolf     return raw_do_pdiscard(bs, offset, bytes, false);
2763c1bb86cdSEric Blake }
2764c1bb86cdSEric Blake 
27657154d8aeSKevin Wolf static int coroutine_fn
27667154d8aeSKevin Wolf raw_do_pwrite_zeroes(BlockDriverState *bs, int64_t offset, int bytes,
27677154d8aeSKevin Wolf                      BdrvRequestFlags flags, bool blkdev)
27687154d8aeSKevin Wolf {
27697154d8aeSKevin Wolf     BDRVRawState *s = bs->opaque;
27707154d8aeSKevin Wolf     RawPosixAIOData acb;
27717154d8aeSKevin Wolf     ThreadPoolFunc *handler;
27727154d8aeSKevin Wolf 
27737154d8aeSKevin Wolf     acb = (RawPosixAIOData) {
27747154d8aeSKevin Wolf         .bs             = bs,
27757154d8aeSKevin Wolf         .aio_fildes     = s->fd,
27767154d8aeSKevin Wolf         .aio_type       = QEMU_AIO_WRITE_ZEROES,
27777154d8aeSKevin Wolf         .aio_offset     = offset,
27787154d8aeSKevin Wolf         .aio_nbytes     = bytes,
27797154d8aeSKevin Wolf     };
27807154d8aeSKevin Wolf 
27817154d8aeSKevin Wolf     if (blkdev) {
27827154d8aeSKevin Wolf         acb.aio_type |= QEMU_AIO_BLKDEV;
27837154d8aeSKevin Wolf     }
2784738301e1SKevin Wolf     if (flags & BDRV_REQ_NO_FALLBACK) {
2785738301e1SKevin Wolf         acb.aio_type |= QEMU_AIO_NO_FALLBACK;
2786738301e1SKevin Wolf     }
27877154d8aeSKevin Wolf 
27887154d8aeSKevin Wolf     if (flags & BDRV_REQ_MAY_UNMAP) {
27897154d8aeSKevin Wolf         acb.aio_type |= QEMU_AIO_DISCARD;
27907154d8aeSKevin Wolf         handler = handle_aiocb_write_zeroes_unmap;
27917154d8aeSKevin Wolf     } else {
27927154d8aeSKevin Wolf         handler = handle_aiocb_write_zeroes;
27937154d8aeSKevin Wolf     }
27947154d8aeSKevin Wolf 
27957154d8aeSKevin Wolf     return raw_thread_pool_submit(bs, handler, &acb);
27967154d8aeSKevin Wolf }
27977154d8aeSKevin Wolf 
2798c1bb86cdSEric Blake static int coroutine_fn raw_co_pwrite_zeroes(
2799c1bb86cdSEric Blake     BlockDriverState *bs, int64_t offset,
2800f5a5ca79SManos Pitsidianakis     int bytes, BdrvRequestFlags flags)
2801c1bb86cdSEric Blake {
28027154d8aeSKevin Wolf     return raw_do_pwrite_zeroes(bs, offset, bytes, flags, false);
2803c1bb86cdSEric Blake }
2804c1bb86cdSEric Blake 
2805c1bb86cdSEric Blake static int raw_get_info(BlockDriverState *bs, BlockDriverInfo *bdi)
2806c1bb86cdSEric Blake {
2807c1bb86cdSEric Blake     BDRVRawState *s = bs->opaque;
2808c1bb86cdSEric Blake 
2809c1bb86cdSEric Blake     bdi->unallocated_blocks_are_zero = s->discard_zeroes;
2810c1bb86cdSEric Blake     return 0;
2811c1bb86cdSEric Blake }
2812c1bb86cdSEric Blake 
2813c1bb86cdSEric Blake static QemuOptsList raw_create_opts = {
2814c1bb86cdSEric Blake     .name = "raw-create-opts",
2815c1bb86cdSEric Blake     .head = QTAILQ_HEAD_INITIALIZER(raw_create_opts.head),
2816c1bb86cdSEric Blake     .desc = {
2817c1bb86cdSEric Blake         {
2818c1bb86cdSEric Blake             .name = BLOCK_OPT_SIZE,
2819c1bb86cdSEric Blake             .type = QEMU_OPT_SIZE,
2820c1bb86cdSEric Blake             .help = "Virtual disk size"
2821c1bb86cdSEric Blake         },
2822c1bb86cdSEric Blake         {
2823c1bb86cdSEric Blake             .name = BLOCK_OPT_NOCOW,
2824c1bb86cdSEric Blake             .type = QEMU_OPT_BOOL,
2825c1bb86cdSEric Blake             .help = "Turn off copy-on-write (valid only on btrfs)"
2826c1bb86cdSEric Blake         },
2827c1bb86cdSEric Blake         {
2828c1bb86cdSEric Blake             .name = BLOCK_OPT_PREALLOC,
2829c1bb86cdSEric Blake             .type = QEMU_OPT_STRING,
2830abea0053SStefano Garzarella             .help = "Preallocation mode (allowed values: off"
2831abea0053SStefano Garzarella #ifdef CONFIG_POSIX_FALLOCATE
2832abea0053SStefano Garzarella                     ", falloc"
2833abea0053SStefano Garzarella #endif
2834abea0053SStefano Garzarella                     ", full)"
2835c1bb86cdSEric Blake         },
2836c1bb86cdSEric Blake         { /* end of list */ }
2837c1bb86cdSEric Blake     }
2838c1bb86cdSEric Blake };
2839c1bb86cdSEric Blake 
2840244a5668SFam Zheng static int raw_check_perm(BlockDriverState *bs, uint64_t perm, uint64_t shared,
2841244a5668SFam Zheng                           Error **errp)
2842244a5668SFam Zheng {
28436ceabe6fSKevin Wolf     BDRVRawState *s = bs->opaque;
28446ceabe6fSKevin Wolf     BDRVRawReopenState *rs = NULL;
28456ceabe6fSKevin Wolf     int open_flags;
28466ceabe6fSKevin Wolf     int ret;
28476ceabe6fSKevin Wolf 
28486ceabe6fSKevin Wolf     if (s->perm_change_fd) {
28496ceabe6fSKevin Wolf         /*
28506ceabe6fSKevin Wolf          * In the context of reopen, this function may be called several times
28516ceabe6fSKevin Wolf          * (directly and recursively while change permissions of the parent).
28526ceabe6fSKevin Wolf          * This is even true for children that don't inherit from the original
28536ceabe6fSKevin Wolf          * reopen node, so s->reopen_state is not set.
28546ceabe6fSKevin Wolf          *
28556ceabe6fSKevin Wolf          * Ignore all but the first call.
28566ceabe6fSKevin Wolf          */
28576ceabe6fSKevin Wolf         return 0;
28586ceabe6fSKevin Wolf     }
28596ceabe6fSKevin Wolf 
28606ceabe6fSKevin Wolf     if (s->reopen_state) {
28616ceabe6fSKevin Wolf         /* We already have a new file descriptor to set permissions for */
28626ceabe6fSKevin Wolf         assert(s->reopen_state->perm == perm);
28636ceabe6fSKevin Wolf         assert(s->reopen_state->shared_perm == shared);
28646ceabe6fSKevin Wolf         rs = s->reopen_state->opaque;
28656ceabe6fSKevin Wolf         s->perm_change_fd = rs->fd;
2866094e3639SMax Reitz         s->perm_change_flags = rs->open_flags;
28676ceabe6fSKevin Wolf     } else {
28686ceabe6fSKevin Wolf         /* We may need a new fd if auto-read-only switches the mode */
286923dece19SKevin Wolf         ret = raw_reconfigure_getfd(bs, bs->open_flags, &open_flags, perm,
28706ceabe6fSKevin Wolf                                     false, errp);
28716ceabe6fSKevin Wolf         if (ret < 0) {
28726ceabe6fSKevin Wolf             return ret;
28736ceabe6fSKevin Wolf         } else if (ret != s->fd) {
28746ceabe6fSKevin Wolf             s->perm_change_fd = ret;
2875094e3639SMax Reitz             s->perm_change_flags = open_flags;
28766ceabe6fSKevin Wolf         }
28776ceabe6fSKevin Wolf     }
28786ceabe6fSKevin Wolf 
28796ceabe6fSKevin Wolf     /* Prepare permissions on old fd to avoid conflicts between old and new,
28806ceabe6fSKevin Wolf      * but keep everything locked that new will need. */
28816ceabe6fSKevin Wolf     ret = raw_handle_perm_lock(bs, RAW_PL_PREPARE, perm, shared, errp);
28826ceabe6fSKevin Wolf     if (ret < 0) {
28836ceabe6fSKevin Wolf         goto fail;
28846ceabe6fSKevin Wolf     }
28856ceabe6fSKevin Wolf 
28866ceabe6fSKevin Wolf     /* Copy locks to the new fd */
28876ceabe6fSKevin Wolf     if (s->perm_change_fd) {
28886ceabe6fSKevin Wolf         ret = raw_apply_lock_bytes(NULL, s->perm_change_fd, perm, ~shared,
28896ceabe6fSKevin Wolf                                    false, errp);
28906ceabe6fSKevin Wolf         if (ret < 0) {
28916ceabe6fSKevin Wolf             raw_handle_perm_lock(bs, RAW_PL_ABORT, 0, 0, NULL);
28926ceabe6fSKevin Wolf             goto fail;
28936ceabe6fSKevin Wolf         }
28946ceabe6fSKevin Wolf     }
28956ceabe6fSKevin Wolf     return 0;
28966ceabe6fSKevin Wolf 
28976ceabe6fSKevin Wolf fail:
28986ceabe6fSKevin Wolf     if (s->perm_change_fd && !s->reopen_state) {
28996ceabe6fSKevin Wolf         qemu_close(s->perm_change_fd);
29006ceabe6fSKevin Wolf     }
29016ceabe6fSKevin Wolf     s->perm_change_fd = 0;
29026ceabe6fSKevin Wolf     return ret;
2903244a5668SFam Zheng }
2904244a5668SFam Zheng 
2905244a5668SFam Zheng static void raw_set_perm(BlockDriverState *bs, uint64_t perm, uint64_t shared)
2906244a5668SFam Zheng {
2907244a5668SFam Zheng     BDRVRawState *s = bs->opaque;
29086ceabe6fSKevin Wolf 
29096ceabe6fSKevin Wolf     /* For reopen, we have already switched to the new fd (.bdrv_set_perm is
29106ceabe6fSKevin Wolf      * called after .bdrv_reopen_commit) */
29116ceabe6fSKevin Wolf     if (s->perm_change_fd && s->fd != s->perm_change_fd) {
29126ceabe6fSKevin Wolf         qemu_close(s->fd);
29136ceabe6fSKevin Wolf         s->fd = s->perm_change_fd;
2914094e3639SMax Reitz         s->open_flags = s->perm_change_flags;
29156ceabe6fSKevin Wolf     }
29166ceabe6fSKevin Wolf     s->perm_change_fd = 0;
29176ceabe6fSKevin Wolf 
2918244a5668SFam Zheng     raw_handle_perm_lock(bs, RAW_PL_COMMIT, perm, shared, NULL);
2919244a5668SFam Zheng     s->perm = perm;
2920244a5668SFam Zheng     s->shared_perm = shared;
2921244a5668SFam Zheng }
2922244a5668SFam Zheng 
2923244a5668SFam Zheng static void raw_abort_perm_update(BlockDriverState *bs)
2924244a5668SFam Zheng {
29256ceabe6fSKevin Wolf     BDRVRawState *s = bs->opaque;
29266ceabe6fSKevin Wolf 
29276ceabe6fSKevin Wolf     /* For reopen, .bdrv_reopen_abort is called afterwards and will close
29286ceabe6fSKevin Wolf      * the file descriptor. */
29296ceabe6fSKevin Wolf     if (s->perm_change_fd && !s->reopen_state) {
29306ceabe6fSKevin Wolf         qemu_close(s->perm_change_fd);
29316ceabe6fSKevin Wolf     }
29326ceabe6fSKevin Wolf     s->perm_change_fd = 0;
29336ceabe6fSKevin Wolf 
2934244a5668SFam Zheng     raw_handle_perm_lock(bs, RAW_PL_ABORT, 0, 0, NULL);
2935244a5668SFam Zheng }
2936244a5668SFam Zheng 
293767b51fb9SVladimir Sementsov-Ogievskiy static int coroutine_fn raw_co_copy_range_from(
293867b51fb9SVladimir Sementsov-Ogievskiy         BlockDriverState *bs, BdrvChild *src, uint64_t src_offset,
293967b51fb9SVladimir Sementsov-Ogievskiy         BdrvChild *dst, uint64_t dst_offset, uint64_t bytes,
294067b51fb9SVladimir Sementsov-Ogievskiy         BdrvRequestFlags read_flags, BdrvRequestFlags write_flags)
29411efad060SFam Zheng {
294267b51fb9SVladimir Sementsov-Ogievskiy     return bdrv_co_copy_range_to(src, src_offset, dst, dst_offset, bytes,
294367b51fb9SVladimir Sementsov-Ogievskiy                                  read_flags, write_flags);
29441efad060SFam Zheng }
29451efad060SFam Zheng 
29461efad060SFam Zheng static int coroutine_fn raw_co_copy_range_to(BlockDriverState *bs,
294767b51fb9SVladimir Sementsov-Ogievskiy                                              BdrvChild *src,
294867b51fb9SVladimir Sementsov-Ogievskiy                                              uint64_t src_offset,
294967b51fb9SVladimir Sementsov-Ogievskiy                                              BdrvChild *dst,
295067b51fb9SVladimir Sementsov-Ogievskiy                                              uint64_t dst_offset,
295167b51fb9SVladimir Sementsov-Ogievskiy                                              uint64_t bytes,
295267b51fb9SVladimir Sementsov-Ogievskiy                                              BdrvRequestFlags read_flags,
295367b51fb9SVladimir Sementsov-Ogievskiy                                              BdrvRequestFlags write_flags)
29541efad060SFam Zheng {
295558a209c4SKevin Wolf     RawPosixAIOData acb;
29561efad060SFam Zheng     BDRVRawState *s = bs->opaque;
29571efad060SFam Zheng     BDRVRawState *src_s;
29581efad060SFam Zheng 
29591efad060SFam Zheng     assert(dst->bs == bs);
29601efad060SFam Zheng     if (src->bs->drv->bdrv_co_copy_range_to != raw_co_copy_range_to) {
29611efad060SFam Zheng         return -ENOTSUP;
29621efad060SFam Zheng     }
29631efad060SFam Zheng 
29641efad060SFam Zheng     src_s = src->bs->opaque;
29659f850f67SFam Zheng     if (fd_open(src->bs) < 0 || fd_open(dst->bs) < 0) {
29661efad060SFam Zheng         return -EIO;
29671efad060SFam Zheng     }
296858a209c4SKevin Wolf 
296958a209c4SKevin Wolf     acb = (RawPosixAIOData) {
297058a209c4SKevin Wolf         .bs             = bs,
297158a209c4SKevin Wolf         .aio_type       = QEMU_AIO_COPY_RANGE,
297258a209c4SKevin Wolf         .aio_fildes     = src_s->fd,
297358a209c4SKevin Wolf         .aio_offset     = src_offset,
297458a209c4SKevin Wolf         .aio_nbytes     = bytes,
297558a209c4SKevin Wolf         .copy_range     = {
297658a209c4SKevin Wolf             .aio_fd2        = s->fd,
297758a209c4SKevin Wolf             .aio_offset2    = dst_offset,
297858a209c4SKevin Wolf         },
297958a209c4SKevin Wolf     };
298058a209c4SKevin Wolf 
298158a209c4SKevin Wolf     return raw_thread_pool_submit(bs, handle_aiocb_copy_range, &acb);
29821efad060SFam Zheng }
29831efad060SFam Zheng 
2984c1bb86cdSEric Blake BlockDriver bdrv_file = {
2985c1bb86cdSEric Blake     .format_name = "file",
2986c1bb86cdSEric Blake     .protocol_name = "file",
2987c1bb86cdSEric Blake     .instance_size = sizeof(BDRVRawState),
2988c1bb86cdSEric Blake     .bdrv_needs_filename = true,
2989c1bb86cdSEric Blake     .bdrv_probe = NULL, /* no probe for protocols */
2990c1bb86cdSEric Blake     .bdrv_parse_filename = raw_parse_filename,
2991c1bb86cdSEric Blake     .bdrv_file_open = raw_open,
2992c1bb86cdSEric Blake     .bdrv_reopen_prepare = raw_reopen_prepare,
2993c1bb86cdSEric Blake     .bdrv_reopen_commit = raw_reopen_commit,
2994c1bb86cdSEric Blake     .bdrv_reopen_abort = raw_reopen_abort,
2995c1bb86cdSEric Blake     .bdrv_close = raw_close,
2996927f11e1SKevin Wolf     .bdrv_co_create = raw_co_create,
2997efc75e2aSStefan Hajnoczi     .bdrv_co_create_opts = raw_co_create_opts,
2998c1bb86cdSEric Blake     .bdrv_has_zero_init = bdrv_has_zero_init_1,
29991dcaf527SMax Reitz     .bdrv_has_zero_init_truncate = bdrv_has_zero_init_1,
3000a290f085SEric Blake     .bdrv_co_block_status = raw_co_block_status,
3001dd577a26SStefan Hajnoczi     .bdrv_co_invalidate_cache = raw_co_invalidate_cache,
3002c1bb86cdSEric Blake     .bdrv_co_pwrite_zeroes = raw_co_pwrite_zeroes,
3003c1bb86cdSEric Blake 
3004c1bb86cdSEric Blake     .bdrv_co_preadv         = raw_co_preadv,
3005c1bb86cdSEric Blake     .bdrv_co_pwritev        = raw_co_pwritev,
300633d70fb6SKevin Wolf     .bdrv_co_flush_to_disk  = raw_co_flush_to_disk,
300733d70fb6SKevin Wolf     .bdrv_co_pdiscard       = raw_co_pdiscard,
30081efad060SFam Zheng     .bdrv_co_copy_range_from = raw_co_copy_range_from,
30091efad060SFam Zheng     .bdrv_co_copy_range_to  = raw_co_copy_range_to,
3010c1bb86cdSEric Blake     .bdrv_refresh_limits = raw_refresh_limits,
3011c1bb86cdSEric Blake     .bdrv_io_plug = raw_aio_plug,
3012c1bb86cdSEric Blake     .bdrv_io_unplug = raw_aio_unplug,
3013ed6e2161SNishanth Aravamudan     .bdrv_attach_aio_context = raw_aio_attach_aio_context,
3014c1bb86cdSEric Blake 
3015061ca8a3SKevin Wolf     .bdrv_co_truncate = raw_co_truncate,
3016c1bb86cdSEric Blake     .bdrv_getlength = raw_getlength,
3017c1bb86cdSEric Blake     .bdrv_get_info = raw_get_info,
3018c1bb86cdSEric Blake     .bdrv_get_allocated_file_size
3019c1bb86cdSEric Blake                         = raw_get_allocated_file_size,
3020244a5668SFam Zheng     .bdrv_check_perm = raw_check_perm,
3021244a5668SFam Zheng     .bdrv_set_perm   = raw_set_perm,
3022244a5668SFam Zheng     .bdrv_abort_perm_update = raw_abort_perm_update,
3023c1bb86cdSEric Blake     .create_opts = &raw_create_opts,
30248a2ce0bcSAlberto Garcia     .mutable_opts = mutable_opts,
3025c1bb86cdSEric Blake };
3026c1bb86cdSEric Blake 
3027c1bb86cdSEric Blake /***********************************************/
3028c1bb86cdSEric Blake /* host device */
3029c1bb86cdSEric Blake 
3030c1bb86cdSEric Blake #if defined(__APPLE__) && defined(__MACH__)
3031c1bb86cdSEric Blake static kern_return_t GetBSDPath(io_iterator_t mediaIterator, char *bsdPath,
3032c1bb86cdSEric Blake                                 CFIndex maxPathSize, int flags);
3033c1bb86cdSEric Blake static char *FindEjectableOpticalMedia(io_iterator_t *mediaIterator)
3034c1bb86cdSEric Blake {
3035c1bb86cdSEric Blake     kern_return_t kernResult = KERN_FAILURE;
3036c1bb86cdSEric Blake     mach_port_t     masterPort;
3037c1bb86cdSEric Blake     CFMutableDictionaryRef  classesToMatch;
3038c1bb86cdSEric Blake     const char *matching_array[] = {kIODVDMediaClass, kIOCDMediaClass};
3039c1bb86cdSEric Blake     char *mediaType = NULL;
3040c1bb86cdSEric Blake 
3041c1bb86cdSEric Blake     kernResult = IOMasterPort( MACH_PORT_NULL, &masterPort );
3042c1bb86cdSEric Blake     if ( KERN_SUCCESS != kernResult ) {
3043c1bb86cdSEric Blake         printf( "IOMasterPort returned %d\n", kernResult );
3044c1bb86cdSEric Blake     }
3045c1bb86cdSEric Blake 
3046c1bb86cdSEric Blake     int index;
3047c1bb86cdSEric Blake     for (index = 0; index < ARRAY_SIZE(matching_array); index++) {
3048c1bb86cdSEric Blake         classesToMatch = IOServiceMatching(matching_array[index]);
3049c1bb86cdSEric Blake         if (classesToMatch == NULL) {
3050c1bb86cdSEric Blake             error_report("IOServiceMatching returned NULL for %s",
3051c1bb86cdSEric Blake                          matching_array[index]);
3052c1bb86cdSEric Blake             continue;
3053c1bb86cdSEric Blake         }
3054c1bb86cdSEric Blake         CFDictionarySetValue(classesToMatch, CFSTR(kIOMediaEjectableKey),
3055c1bb86cdSEric Blake                              kCFBooleanTrue);
3056c1bb86cdSEric Blake         kernResult = IOServiceGetMatchingServices(masterPort, classesToMatch,
3057c1bb86cdSEric Blake                                                   mediaIterator);
3058c1bb86cdSEric Blake         if (kernResult != KERN_SUCCESS) {
3059c1bb86cdSEric Blake             error_report("Note: IOServiceGetMatchingServices returned %d",
3060c1bb86cdSEric Blake                          kernResult);
3061c1bb86cdSEric Blake             continue;
3062c1bb86cdSEric Blake         }
3063c1bb86cdSEric Blake 
3064c1bb86cdSEric Blake         /* If a match was found, leave the loop */
3065c1bb86cdSEric Blake         if (*mediaIterator != 0) {
30664f7d28d7SLaurent Vivier             trace_file_FindEjectableOpticalMedia(matching_array[index]);
3067c1bb86cdSEric Blake             mediaType = g_strdup(matching_array[index]);
3068c1bb86cdSEric Blake             break;
3069c1bb86cdSEric Blake         }
3070c1bb86cdSEric Blake     }
3071c1bb86cdSEric Blake     return mediaType;
3072c1bb86cdSEric Blake }
3073c1bb86cdSEric Blake 
3074c1bb86cdSEric Blake kern_return_t GetBSDPath(io_iterator_t mediaIterator, char *bsdPath,
3075c1bb86cdSEric Blake                          CFIndex maxPathSize, int flags)
3076c1bb86cdSEric Blake {
3077c1bb86cdSEric Blake     io_object_t     nextMedia;
3078c1bb86cdSEric Blake     kern_return_t   kernResult = KERN_FAILURE;
3079c1bb86cdSEric Blake     *bsdPath = '\0';
3080c1bb86cdSEric Blake     nextMedia = IOIteratorNext( mediaIterator );
3081c1bb86cdSEric Blake     if ( nextMedia )
3082c1bb86cdSEric Blake     {
3083c1bb86cdSEric Blake         CFTypeRef   bsdPathAsCFString;
3084c1bb86cdSEric Blake     bsdPathAsCFString = IORegistryEntryCreateCFProperty( nextMedia, CFSTR( kIOBSDNameKey ), kCFAllocatorDefault, 0 );
3085c1bb86cdSEric Blake         if ( bsdPathAsCFString ) {
3086c1bb86cdSEric Blake             size_t devPathLength;
3087c1bb86cdSEric Blake             strcpy( bsdPath, _PATH_DEV );
3088c1bb86cdSEric Blake             if (flags & BDRV_O_NOCACHE) {
3089c1bb86cdSEric Blake                 strcat(bsdPath, "r");
3090c1bb86cdSEric Blake             }
3091c1bb86cdSEric Blake             devPathLength = strlen( bsdPath );
3092c1bb86cdSEric Blake             if ( CFStringGetCString( bsdPathAsCFString, bsdPath + devPathLength, maxPathSize - devPathLength, kCFStringEncodingASCII ) ) {
3093c1bb86cdSEric Blake                 kernResult = KERN_SUCCESS;
3094c1bb86cdSEric Blake             }
3095c1bb86cdSEric Blake             CFRelease( bsdPathAsCFString );
3096c1bb86cdSEric Blake         }
3097c1bb86cdSEric Blake         IOObjectRelease( nextMedia );
3098c1bb86cdSEric Blake     }
3099c1bb86cdSEric Blake 
3100c1bb86cdSEric Blake     return kernResult;
3101c1bb86cdSEric Blake }
3102c1bb86cdSEric Blake 
3103c1bb86cdSEric Blake /* Sets up a real cdrom for use in QEMU */
3104c1bb86cdSEric Blake static bool setup_cdrom(char *bsd_path, Error **errp)
3105c1bb86cdSEric Blake {
3106c1bb86cdSEric Blake     int index, num_of_test_partitions = 2, fd;
3107c1bb86cdSEric Blake     char test_partition[MAXPATHLEN];
3108c1bb86cdSEric Blake     bool partition_found = false;
3109c1bb86cdSEric Blake 
3110c1bb86cdSEric Blake     /* look for a working partition */
3111c1bb86cdSEric Blake     for (index = 0; index < num_of_test_partitions; index++) {
3112c1bb86cdSEric Blake         snprintf(test_partition, sizeof(test_partition), "%ss%d", bsd_path,
3113c1bb86cdSEric Blake                  index);
3114c1bb86cdSEric Blake         fd = qemu_open(test_partition, O_RDONLY | O_BINARY | O_LARGEFILE);
3115c1bb86cdSEric Blake         if (fd >= 0) {
3116c1bb86cdSEric Blake             partition_found = true;
3117c1bb86cdSEric Blake             qemu_close(fd);
3118c1bb86cdSEric Blake             break;
3119c1bb86cdSEric Blake         }
3120c1bb86cdSEric Blake     }
3121c1bb86cdSEric Blake 
3122c1bb86cdSEric Blake     /* if a working partition on the device was not found */
3123c1bb86cdSEric Blake     if (partition_found == false) {
3124c1bb86cdSEric Blake         error_setg(errp, "Failed to find a working partition on disc");
3125c1bb86cdSEric Blake     } else {
31264f7d28d7SLaurent Vivier         trace_file_setup_cdrom(test_partition);
3127c1bb86cdSEric Blake         pstrcpy(bsd_path, MAXPATHLEN, test_partition);
3128c1bb86cdSEric Blake     }
3129c1bb86cdSEric Blake     return partition_found;
3130c1bb86cdSEric Blake }
3131c1bb86cdSEric Blake 
3132c1bb86cdSEric Blake /* Prints directions on mounting and unmounting a device */
3133c1bb86cdSEric Blake static void print_unmounting_directions(const char *file_name)
3134c1bb86cdSEric Blake {
3135c1bb86cdSEric Blake     error_report("If device %s is mounted on the desktop, unmount"
3136c1bb86cdSEric Blake                  " it first before using it in QEMU", file_name);
3137c1bb86cdSEric Blake     error_report("Command to unmount device: diskutil unmountDisk %s",
3138c1bb86cdSEric Blake                  file_name);
3139c1bb86cdSEric Blake     error_report("Command to mount device: diskutil mountDisk %s", file_name);
3140c1bb86cdSEric Blake }
3141c1bb86cdSEric Blake 
3142c1bb86cdSEric Blake #endif /* defined(__APPLE__) && defined(__MACH__) */
3143c1bb86cdSEric Blake 
3144c1bb86cdSEric Blake static int hdev_probe_device(const char *filename)
3145c1bb86cdSEric Blake {
3146c1bb86cdSEric Blake     struct stat st;
3147c1bb86cdSEric Blake 
3148c1bb86cdSEric Blake     /* allow a dedicated CD-ROM driver to match with a higher priority */
3149c1bb86cdSEric Blake     if (strstart(filename, "/dev/cdrom", NULL))
3150c1bb86cdSEric Blake         return 50;
3151c1bb86cdSEric Blake 
3152c1bb86cdSEric Blake     if (stat(filename, &st) >= 0 &&
3153c1bb86cdSEric Blake             (S_ISCHR(st.st_mode) || S_ISBLK(st.st_mode))) {
3154c1bb86cdSEric Blake         return 100;
3155c1bb86cdSEric Blake     }
3156c1bb86cdSEric Blake 
3157c1bb86cdSEric Blake     return 0;
3158c1bb86cdSEric Blake }
3159c1bb86cdSEric Blake 
3160c1bb86cdSEric Blake static int check_hdev_writable(BDRVRawState *s)
3161c1bb86cdSEric Blake {
3162c1bb86cdSEric Blake #if defined(BLKROGET)
3163c1bb86cdSEric Blake     /* Linux block devices can be configured "read-only" using blockdev(8).
3164c1bb86cdSEric Blake      * This is independent of device node permissions and therefore open(2)
3165c1bb86cdSEric Blake      * with O_RDWR succeeds.  Actual writes fail with EPERM.
3166c1bb86cdSEric Blake      *
3167c1bb86cdSEric Blake      * bdrv_open() is supposed to fail if the disk is read-only.  Explicitly
3168c1bb86cdSEric Blake      * check for read-only block devices so that Linux block devices behave
3169c1bb86cdSEric Blake      * properly.
3170c1bb86cdSEric Blake      */
3171c1bb86cdSEric Blake     struct stat st;
3172c1bb86cdSEric Blake     int readonly = 0;
3173c1bb86cdSEric Blake 
3174c1bb86cdSEric Blake     if (fstat(s->fd, &st)) {
3175c1bb86cdSEric Blake         return -errno;
3176c1bb86cdSEric Blake     }
3177c1bb86cdSEric Blake 
3178c1bb86cdSEric Blake     if (!S_ISBLK(st.st_mode)) {
3179c1bb86cdSEric Blake         return 0;
3180c1bb86cdSEric Blake     }
3181c1bb86cdSEric Blake 
3182c1bb86cdSEric Blake     if (ioctl(s->fd, BLKROGET, &readonly) < 0) {
3183c1bb86cdSEric Blake         return -errno;
3184c1bb86cdSEric Blake     }
3185c1bb86cdSEric Blake 
3186c1bb86cdSEric Blake     if (readonly) {
3187c1bb86cdSEric Blake         return -EACCES;
3188c1bb86cdSEric Blake     }
3189c1bb86cdSEric Blake #endif /* defined(BLKROGET) */
3190c1bb86cdSEric Blake     return 0;
3191c1bb86cdSEric Blake }
3192c1bb86cdSEric Blake 
3193c1bb86cdSEric Blake static void hdev_parse_filename(const char *filename, QDict *options,
3194c1bb86cdSEric Blake                                 Error **errp)
3195c1bb86cdSEric Blake {
319603c320d8SMax Reitz     bdrv_parse_filename_strip_prefix(filename, "host_device:", options);
3197c1bb86cdSEric Blake }
3198c1bb86cdSEric Blake 
3199c1bb86cdSEric Blake static bool hdev_is_sg(BlockDriverState *bs)
3200c1bb86cdSEric Blake {
3201c1bb86cdSEric Blake 
3202c1bb86cdSEric Blake #if defined(__linux__)
3203c1bb86cdSEric Blake 
3204c1bb86cdSEric Blake     BDRVRawState *s = bs->opaque;
3205c1bb86cdSEric Blake     struct stat st;
3206c1bb86cdSEric Blake     struct sg_scsi_id scsiid;
3207c1bb86cdSEric Blake     int sg_version;
3208c1bb86cdSEric Blake     int ret;
3209c1bb86cdSEric Blake 
3210c1bb86cdSEric Blake     if (stat(bs->filename, &st) < 0 || !S_ISCHR(st.st_mode)) {
3211c1bb86cdSEric Blake         return false;
3212c1bb86cdSEric Blake     }
3213c1bb86cdSEric Blake 
3214c1bb86cdSEric Blake     ret = ioctl(s->fd, SG_GET_VERSION_NUM, &sg_version);
3215c1bb86cdSEric Blake     if (ret < 0) {
3216c1bb86cdSEric Blake         return false;
3217c1bb86cdSEric Blake     }
3218c1bb86cdSEric Blake 
3219c1bb86cdSEric Blake     ret = ioctl(s->fd, SG_GET_SCSI_ID, &scsiid);
3220c1bb86cdSEric Blake     if (ret >= 0) {
32214f7d28d7SLaurent Vivier         trace_file_hdev_is_sg(scsiid.scsi_type, sg_version);
3222c1bb86cdSEric Blake         return true;
3223c1bb86cdSEric Blake     }
3224c1bb86cdSEric Blake 
3225c1bb86cdSEric Blake #endif
3226c1bb86cdSEric Blake 
3227c1bb86cdSEric Blake     return false;
3228c1bb86cdSEric Blake }
3229c1bb86cdSEric Blake 
3230c1bb86cdSEric Blake static int hdev_open(BlockDriverState *bs, QDict *options, int flags,
3231c1bb86cdSEric Blake                      Error **errp)
3232c1bb86cdSEric Blake {
3233c1bb86cdSEric Blake     BDRVRawState *s = bs->opaque;
3234c1bb86cdSEric Blake     Error *local_err = NULL;
3235c1bb86cdSEric Blake     int ret;
3236c1bb86cdSEric Blake 
3237c1bb86cdSEric Blake #if defined(__APPLE__) && defined(__MACH__)
3238129c7d1cSMarkus Armbruster     /*
3239129c7d1cSMarkus Armbruster      * Caution: while qdict_get_str() is fine, getting non-string types
3240129c7d1cSMarkus Armbruster      * would require more care.  When @options come from -blockdev or
3241129c7d1cSMarkus Armbruster      * blockdev_add, its members are typed according to the QAPI
3242129c7d1cSMarkus Armbruster      * schema, but when they come from -drive, they're all QString.
3243129c7d1cSMarkus Armbruster      */
3244c1bb86cdSEric Blake     const char *filename = qdict_get_str(options, "filename");
3245c1bb86cdSEric Blake     char bsd_path[MAXPATHLEN] = "";
3246c1bb86cdSEric Blake     bool error_occurred = false;
3247c1bb86cdSEric Blake 
3248c1bb86cdSEric Blake     /* If using a real cdrom */
3249c1bb86cdSEric Blake     if (strcmp(filename, "/dev/cdrom") == 0) {
3250c1bb86cdSEric Blake         char *mediaType = NULL;
3251c1bb86cdSEric Blake         kern_return_t ret_val;
3252c1bb86cdSEric Blake         io_iterator_t mediaIterator = 0;
3253c1bb86cdSEric Blake 
3254c1bb86cdSEric Blake         mediaType = FindEjectableOpticalMedia(&mediaIterator);
3255c1bb86cdSEric Blake         if (mediaType == NULL) {
3256c1bb86cdSEric Blake             error_setg(errp, "Please make sure your CD/DVD is in the optical"
3257c1bb86cdSEric Blake                        " drive");
3258c1bb86cdSEric Blake             error_occurred = true;
3259c1bb86cdSEric Blake             goto hdev_open_Mac_error;
3260c1bb86cdSEric Blake         }
3261c1bb86cdSEric Blake 
3262c1bb86cdSEric Blake         ret_val = GetBSDPath(mediaIterator, bsd_path, sizeof(bsd_path), flags);
3263c1bb86cdSEric Blake         if (ret_val != KERN_SUCCESS) {
3264c1bb86cdSEric Blake             error_setg(errp, "Could not get BSD path for optical drive");
3265c1bb86cdSEric Blake             error_occurred = true;
3266c1bb86cdSEric Blake             goto hdev_open_Mac_error;
3267c1bb86cdSEric Blake         }
3268c1bb86cdSEric Blake 
3269c1bb86cdSEric Blake         /* If a real optical drive was not found */
3270c1bb86cdSEric Blake         if (bsd_path[0] == '\0') {
3271c1bb86cdSEric Blake             error_setg(errp, "Failed to obtain bsd path for optical drive");
3272c1bb86cdSEric Blake             error_occurred = true;
3273c1bb86cdSEric Blake             goto hdev_open_Mac_error;
3274c1bb86cdSEric Blake         }
3275c1bb86cdSEric Blake 
3276c1bb86cdSEric Blake         /* If using a cdrom disc and finding a partition on the disc failed */
3277c1bb86cdSEric Blake         if (strncmp(mediaType, kIOCDMediaClass, 9) == 0 &&
3278c1bb86cdSEric Blake             setup_cdrom(bsd_path, errp) == false) {
3279c1bb86cdSEric Blake             print_unmounting_directions(bsd_path);
3280c1bb86cdSEric Blake             error_occurred = true;
3281c1bb86cdSEric Blake             goto hdev_open_Mac_error;
3282c1bb86cdSEric Blake         }
3283c1bb86cdSEric Blake 
328446f5ac20SEric Blake         qdict_put_str(options, "filename", bsd_path);
3285c1bb86cdSEric Blake 
3286c1bb86cdSEric Blake hdev_open_Mac_error:
3287c1bb86cdSEric Blake         g_free(mediaType);
3288c1bb86cdSEric Blake         if (mediaIterator) {
3289c1bb86cdSEric Blake             IOObjectRelease(mediaIterator);
3290c1bb86cdSEric Blake         }
3291c1bb86cdSEric Blake         if (error_occurred) {
3292c1bb86cdSEric Blake             return -ENOENT;
3293c1bb86cdSEric Blake         }
3294c1bb86cdSEric Blake     }
3295c1bb86cdSEric Blake #endif /* defined(__APPLE__) && defined(__MACH__) */
3296c1bb86cdSEric Blake 
3297c1bb86cdSEric Blake     s->type = FTYPE_FILE;
3298c1bb86cdSEric Blake 
3299230ff739SJohn Snow     ret = raw_open_common(bs, options, flags, 0, true, &local_err);
3300c1bb86cdSEric Blake     if (ret < 0) {
3301c1bb86cdSEric Blake         error_propagate(errp, local_err);
3302c1bb86cdSEric Blake #if defined(__APPLE__) && defined(__MACH__)
3303c1bb86cdSEric Blake         if (*bsd_path) {
3304c1bb86cdSEric Blake             filename = bsd_path;
3305c1bb86cdSEric Blake         }
3306c1bb86cdSEric Blake         /* if a physical device experienced an error while being opened */
3307c1bb86cdSEric Blake         if (strncmp(filename, "/dev/", 5) == 0) {
3308c1bb86cdSEric Blake             print_unmounting_directions(filename);
3309c1bb86cdSEric Blake         }
3310c1bb86cdSEric Blake #endif /* defined(__APPLE__) && defined(__MACH__) */
3311c1bb86cdSEric Blake         return ret;
3312c1bb86cdSEric Blake     }
3313c1bb86cdSEric Blake 
3314c1bb86cdSEric Blake     /* Since this does ioctl the device must be already opened */
3315c1bb86cdSEric Blake     bs->sg = hdev_is_sg(bs);
3316c1bb86cdSEric Blake 
3317c1bb86cdSEric Blake     if (flags & BDRV_O_RDWR) {
3318c1bb86cdSEric Blake         ret = check_hdev_writable(s);
3319c1bb86cdSEric Blake         if (ret < 0) {
3320c1bb86cdSEric Blake             raw_close(bs);
3321c1bb86cdSEric Blake             error_setg_errno(errp, -ret, "The device is not writable");
3322c1bb86cdSEric Blake             return ret;
3323c1bb86cdSEric Blake         }
3324c1bb86cdSEric Blake     }
3325c1bb86cdSEric Blake 
3326c1bb86cdSEric Blake     return ret;
3327c1bb86cdSEric Blake }
3328c1bb86cdSEric Blake 
3329c1bb86cdSEric Blake #if defined(__linux__)
33302f3a7ab3SKevin Wolf static int coroutine_fn
33312f3a7ab3SKevin Wolf hdev_co_ioctl(BlockDriverState *bs, unsigned long int req, void *buf)
3332c1bb86cdSEric Blake {
3333c1bb86cdSEric Blake     BDRVRawState *s = bs->opaque;
333403425671SKevin Wolf     RawPosixAIOData acb;
33352f3a7ab3SKevin Wolf     int ret;
3336c1bb86cdSEric Blake 
33372f3a7ab3SKevin Wolf     ret = fd_open(bs);
33382f3a7ab3SKevin Wolf     if (ret < 0) {
33392f3a7ab3SKevin Wolf         return ret;
33402f3a7ab3SKevin Wolf     }
3341c1bb86cdSEric Blake 
33427c9e5276SPaolo Bonzini     if (req == SG_IO && s->pr_mgr) {
33437c9e5276SPaolo Bonzini         struct sg_io_hdr *io_hdr = buf;
33447c9e5276SPaolo Bonzini         if (io_hdr->cmdp[0] == PERSISTENT_RESERVE_OUT ||
33457c9e5276SPaolo Bonzini             io_hdr->cmdp[0] == PERSISTENT_RESERVE_IN) {
33467c9e5276SPaolo Bonzini             return pr_manager_execute(s->pr_mgr, bdrv_get_aio_context(bs),
33472f3a7ab3SKevin Wolf                                       s->fd, io_hdr);
33487c9e5276SPaolo Bonzini         }
33497c9e5276SPaolo Bonzini     }
33507c9e5276SPaolo Bonzini 
335103425671SKevin Wolf     acb = (RawPosixAIOData) {
335203425671SKevin Wolf         .bs         = bs,
335303425671SKevin Wolf         .aio_type   = QEMU_AIO_IOCTL,
335403425671SKevin Wolf         .aio_fildes = s->fd,
335503425671SKevin Wolf         .aio_offset = 0,
335603425671SKevin Wolf         .ioctl      = {
335703425671SKevin Wolf             .buf        = buf,
335803425671SKevin Wolf             .cmd        = req,
335903425671SKevin Wolf         },
336003425671SKevin Wolf     };
336103425671SKevin Wolf 
336203425671SKevin Wolf     return raw_thread_pool_submit(bs, handle_aiocb_ioctl, &acb);
3363c1bb86cdSEric Blake }
3364c1bb86cdSEric Blake #endif /* linux */
3365c1bb86cdSEric Blake 
3366c1bb86cdSEric Blake static int fd_open(BlockDriverState *bs)
3367c1bb86cdSEric Blake {
3368c1bb86cdSEric Blake     BDRVRawState *s = bs->opaque;
3369c1bb86cdSEric Blake 
3370c1bb86cdSEric Blake     /* this is just to ensure s->fd is sane (its called by io ops) */
3371c1bb86cdSEric Blake     if (s->fd >= 0)
3372c1bb86cdSEric Blake         return 0;
3373c1bb86cdSEric Blake     return -EIO;
3374c1bb86cdSEric Blake }
3375c1bb86cdSEric Blake 
337633d70fb6SKevin Wolf static coroutine_fn int
337733d70fb6SKevin Wolf hdev_co_pdiscard(BlockDriverState *bs, int64_t offset, int bytes)
3378c1bb86cdSEric Blake {
337933d70fb6SKevin Wolf     int ret;
3380c1bb86cdSEric Blake 
338133d70fb6SKevin Wolf     ret = fd_open(bs);
338233d70fb6SKevin Wolf     if (ret < 0) {
338333d70fb6SKevin Wolf         return ret;
3384c1bb86cdSEric Blake     }
338546ee0f46SKevin Wolf     return raw_do_pdiscard(bs, offset, bytes, true);
3386c1bb86cdSEric Blake }
3387c1bb86cdSEric Blake 
3388c1bb86cdSEric Blake static coroutine_fn int hdev_co_pwrite_zeroes(BlockDriverState *bs,
3389f5a5ca79SManos Pitsidianakis     int64_t offset, int bytes, BdrvRequestFlags flags)
3390c1bb86cdSEric Blake {
3391c1bb86cdSEric Blake     int rc;
3392c1bb86cdSEric Blake 
3393c1bb86cdSEric Blake     rc = fd_open(bs);
3394c1bb86cdSEric Blake     if (rc < 0) {
3395c1bb86cdSEric Blake         return rc;
3396c1bb86cdSEric Blake     }
339734fa110eSKevin Wolf 
33987154d8aeSKevin Wolf     return raw_do_pwrite_zeroes(bs, offset, bytes, flags, true);
3399c1bb86cdSEric Blake }
3400c1bb86cdSEric Blake 
3401efc75e2aSStefan Hajnoczi static int coroutine_fn hdev_co_create_opts(const char *filename, QemuOpts *opts,
3402c1bb86cdSEric Blake                                             Error **errp)
3403c1bb86cdSEric Blake {
3404c1bb86cdSEric Blake     int fd;
3405c1bb86cdSEric Blake     int ret = 0;
3406c1bb86cdSEric Blake     struct stat stat_buf;
3407c1bb86cdSEric Blake     int64_t total_size = 0;
3408c1bb86cdSEric Blake     bool has_prefix;
3409c1bb86cdSEric Blake 
3410c1bb86cdSEric Blake     /* This function is used by both protocol block drivers and therefore either
3411c1bb86cdSEric Blake      * of these prefixes may be given.
3412c1bb86cdSEric Blake      * The return value has to be stored somewhere, otherwise this is an error
3413c1bb86cdSEric Blake      * due to -Werror=unused-value. */
3414c1bb86cdSEric Blake     has_prefix =
3415c1bb86cdSEric Blake         strstart(filename, "host_device:", &filename) ||
3416c1bb86cdSEric Blake         strstart(filename, "host_cdrom:" , &filename);
3417c1bb86cdSEric Blake 
3418c1bb86cdSEric Blake     (void)has_prefix;
3419c1bb86cdSEric Blake 
3420db0754dfSFam Zheng     ret = raw_normalize_devicepath(&filename, errp);
3421c1bb86cdSEric Blake     if (ret < 0) {
3422c1bb86cdSEric Blake         return ret;
3423c1bb86cdSEric Blake     }
3424c1bb86cdSEric Blake 
3425c1bb86cdSEric Blake     /* Read out options */
3426c1bb86cdSEric Blake     total_size = ROUND_UP(qemu_opt_get_size_del(opts, BLOCK_OPT_SIZE, 0),
3427c1bb86cdSEric Blake                           BDRV_SECTOR_SIZE);
3428c1bb86cdSEric Blake 
3429c1bb86cdSEric Blake     fd = qemu_open(filename, O_WRONLY | O_BINARY);
3430c1bb86cdSEric Blake     if (fd < 0) {
3431c1bb86cdSEric Blake         ret = -errno;
3432c1bb86cdSEric Blake         error_setg_errno(errp, -ret, "Could not open device");
3433c1bb86cdSEric Blake         return ret;
3434c1bb86cdSEric Blake     }
3435c1bb86cdSEric Blake 
3436c1bb86cdSEric Blake     if (fstat(fd, &stat_buf) < 0) {
3437c1bb86cdSEric Blake         ret = -errno;
3438c1bb86cdSEric Blake         error_setg_errno(errp, -ret, "Could not stat device");
3439c1bb86cdSEric Blake     } else if (!S_ISBLK(stat_buf.st_mode) && !S_ISCHR(stat_buf.st_mode)) {
3440c1bb86cdSEric Blake         error_setg(errp,
3441c1bb86cdSEric Blake                    "The given file is neither a block nor a character device");
3442c1bb86cdSEric Blake         ret = -ENODEV;
3443c1bb86cdSEric Blake     } else if (lseek(fd, 0, SEEK_END) < total_size) {
3444c1bb86cdSEric Blake         error_setg(errp, "Device is too small");
3445c1bb86cdSEric Blake         ret = -ENOSPC;
3446c1bb86cdSEric Blake     }
3447c1bb86cdSEric Blake 
344897ec9117SFam Zheng     if (!ret && total_size) {
344997ec9117SFam Zheng         uint8_t buf[BDRV_SECTOR_SIZE] = { 0 };
345097ec9117SFam Zheng         int64_t zero_size = MIN(BDRV_SECTOR_SIZE, total_size);
345197ec9117SFam Zheng         if (lseek(fd, 0, SEEK_SET) == -1) {
345297ec9117SFam Zheng             ret = -errno;
345397ec9117SFam Zheng         } else {
345497ec9117SFam Zheng             ret = qemu_write_full(fd, buf, zero_size);
345597ec9117SFam Zheng             ret = ret == zero_size ? 0 : -errno;
345697ec9117SFam Zheng         }
345797ec9117SFam Zheng     }
3458c1bb86cdSEric Blake     qemu_close(fd);
3459c1bb86cdSEric Blake     return ret;
3460c1bb86cdSEric Blake }
3461c1bb86cdSEric Blake 
3462c1bb86cdSEric Blake static BlockDriver bdrv_host_device = {
3463c1bb86cdSEric Blake     .format_name        = "host_device",
3464c1bb86cdSEric Blake     .protocol_name        = "host_device",
3465c1bb86cdSEric Blake     .instance_size      = sizeof(BDRVRawState),
3466c1bb86cdSEric Blake     .bdrv_needs_filename = true,
3467c1bb86cdSEric Blake     .bdrv_probe_device  = hdev_probe_device,
3468c1bb86cdSEric Blake     .bdrv_parse_filename = hdev_parse_filename,
3469c1bb86cdSEric Blake     .bdrv_file_open     = hdev_open,
3470c1bb86cdSEric Blake     .bdrv_close         = raw_close,
3471c1bb86cdSEric Blake     .bdrv_reopen_prepare = raw_reopen_prepare,
3472c1bb86cdSEric Blake     .bdrv_reopen_commit  = raw_reopen_commit,
3473c1bb86cdSEric Blake     .bdrv_reopen_abort   = raw_reopen_abort,
3474efc75e2aSStefan Hajnoczi     .bdrv_co_create_opts = hdev_co_create_opts,
3475c1bb86cdSEric Blake     .create_opts         = &raw_create_opts,
34768a2ce0bcSAlberto Garcia     .mutable_opts        = mutable_opts,
3477dd577a26SStefan Hajnoczi     .bdrv_co_invalidate_cache = raw_co_invalidate_cache,
3478c1bb86cdSEric Blake     .bdrv_co_pwrite_zeroes = hdev_co_pwrite_zeroes,
3479c1bb86cdSEric Blake 
3480c1bb86cdSEric Blake     .bdrv_co_preadv         = raw_co_preadv,
3481c1bb86cdSEric Blake     .bdrv_co_pwritev        = raw_co_pwritev,
348233d70fb6SKevin Wolf     .bdrv_co_flush_to_disk  = raw_co_flush_to_disk,
348333d70fb6SKevin Wolf     .bdrv_co_pdiscard       = hdev_co_pdiscard,
34841efad060SFam Zheng     .bdrv_co_copy_range_from = raw_co_copy_range_from,
34851efad060SFam Zheng     .bdrv_co_copy_range_to  = raw_co_copy_range_to,
3486c1bb86cdSEric Blake     .bdrv_refresh_limits = raw_refresh_limits,
3487c1bb86cdSEric Blake     .bdrv_io_plug = raw_aio_plug,
3488c1bb86cdSEric Blake     .bdrv_io_unplug = raw_aio_unplug,
3489042b757cSNishanth Aravamudan     .bdrv_attach_aio_context = raw_aio_attach_aio_context,
3490c1bb86cdSEric Blake 
3491061ca8a3SKevin Wolf     .bdrv_co_truncate       = raw_co_truncate,
3492c1bb86cdSEric Blake     .bdrv_getlength	= raw_getlength,
3493c1bb86cdSEric Blake     .bdrv_get_info = raw_get_info,
3494c1bb86cdSEric Blake     .bdrv_get_allocated_file_size
3495c1bb86cdSEric Blake                         = raw_get_allocated_file_size,
3496244a5668SFam Zheng     .bdrv_check_perm = raw_check_perm,
3497244a5668SFam Zheng     .bdrv_set_perm   = raw_set_perm,
3498244a5668SFam Zheng     .bdrv_abort_perm_update = raw_abort_perm_update,
3499c1bb86cdSEric Blake     .bdrv_probe_blocksizes = hdev_probe_blocksizes,
3500c1bb86cdSEric Blake     .bdrv_probe_geometry = hdev_probe_geometry,
3501c1bb86cdSEric Blake 
3502c1bb86cdSEric Blake     /* generic scsi device */
3503c1bb86cdSEric Blake #ifdef __linux__
35042f3a7ab3SKevin Wolf     .bdrv_co_ioctl          = hdev_co_ioctl,
3505c1bb86cdSEric Blake #endif
3506c1bb86cdSEric Blake };
3507c1bb86cdSEric Blake 
3508c1bb86cdSEric Blake #if defined(__linux__) || defined(__FreeBSD__) || defined(__FreeBSD_kernel__)
3509c1bb86cdSEric Blake static void cdrom_parse_filename(const char *filename, QDict *options,
3510c1bb86cdSEric Blake                                  Error **errp)
3511c1bb86cdSEric Blake {
351203c320d8SMax Reitz     bdrv_parse_filename_strip_prefix(filename, "host_cdrom:", options);
3513c1bb86cdSEric Blake }
3514c1bb86cdSEric Blake #endif
3515c1bb86cdSEric Blake 
3516c1bb86cdSEric Blake #ifdef __linux__
3517c1bb86cdSEric Blake static int cdrom_open(BlockDriverState *bs, QDict *options, int flags,
3518c1bb86cdSEric Blake                       Error **errp)
3519c1bb86cdSEric Blake {
3520c1bb86cdSEric Blake     BDRVRawState *s = bs->opaque;
3521c1bb86cdSEric Blake 
3522c1bb86cdSEric Blake     s->type = FTYPE_CD;
3523c1bb86cdSEric Blake 
3524c1bb86cdSEric Blake     /* open will not fail even if no CD is inserted, so add O_NONBLOCK */
3525230ff739SJohn Snow     return raw_open_common(bs, options, flags, O_NONBLOCK, true, errp);
3526c1bb86cdSEric Blake }
3527c1bb86cdSEric Blake 
3528c1bb86cdSEric Blake static int cdrom_probe_device(const char *filename)
3529c1bb86cdSEric Blake {
3530c1bb86cdSEric Blake     int fd, ret;
3531c1bb86cdSEric Blake     int prio = 0;
3532c1bb86cdSEric Blake     struct stat st;
3533c1bb86cdSEric Blake 
3534c1bb86cdSEric Blake     fd = qemu_open(filename, O_RDONLY | O_NONBLOCK);
3535c1bb86cdSEric Blake     if (fd < 0) {
3536c1bb86cdSEric Blake         goto out;
3537c1bb86cdSEric Blake     }
3538c1bb86cdSEric Blake     ret = fstat(fd, &st);
3539c1bb86cdSEric Blake     if (ret == -1 || !S_ISBLK(st.st_mode)) {
3540c1bb86cdSEric Blake         goto outc;
3541c1bb86cdSEric Blake     }
3542c1bb86cdSEric Blake 
3543c1bb86cdSEric Blake     /* Attempt to detect via a CDROM specific ioctl */
3544c1bb86cdSEric Blake     ret = ioctl(fd, CDROM_DRIVE_STATUS, CDSL_CURRENT);
3545c1bb86cdSEric Blake     if (ret >= 0)
3546c1bb86cdSEric Blake         prio = 100;
3547c1bb86cdSEric Blake 
3548c1bb86cdSEric Blake outc:
3549c1bb86cdSEric Blake     qemu_close(fd);
3550c1bb86cdSEric Blake out:
3551c1bb86cdSEric Blake     return prio;
3552c1bb86cdSEric Blake }
3553c1bb86cdSEric Blake 
3554c1bb86cdSEric Blake static bool cdrom_is_inserted(BlockDriverState *bs)
3555c1bb86cdSEric Blake {
3556c1bb86cdSEric Blake     BDRVRawState *s = bs->opaque;
3557c1bb86cdSEric Blake     int ret;
3558c1bb86cdSEric Blake 
3559c1bb86cdSEric Blake     ret = ioctl(s->fd, CDROM_DRIVE_STATUS, CDSL_CURRENT);
3560c1bb86cdSEric Blake     return ret == CDS_DISC_OK;
3561c1bb86cdSEric Blake }
3562c1bb86cdSEric Blake 
3563c1bb86cdSEric Blake static void cdrom_eject(BlockDriverState *bs, bool eject_flag)
3564c1bb86cdSEric Blake {
3565c1bb86cdSEric Blake     BDRVRawState *s = bs->opaque;
3566c1bb86cdSEric Blake 
3567c1bb86cdSEric Blake     if (eject_flag) {
3568c1bb86cdSEric Blake         if (ioctl(s->fd, CDROMEJECT, NULL) < 0)
3569c1bb86cdSEric Blake             perror("CDROMEJECT");
3570c1bb86cdSEric Blake     } else {
3571c1bb86cdSEric Blake         if (ioctl(s->fd, CDROMCLOSETRAY, NULL) < 0)
3572c1bb86cdSEric Blake             perror("CDROMEJECT");
3573c1bb86cdSEric Blake     }
3574c1bb86cdSEric Blake }
3575c1bb86cdSEric Blake 
3576c1bb86cdSEric Blake static void cdrom_lock_medium(BlockDriverState *bs, bool locked)
3577c1bb86cdSEric Blake {
3578c1bb86cdSEric Blake     BDRVRawState *s = bs->opaque;
3579c1bb86cdSEric Blake 
3580c1bb86cdSEric Blake     if (ioctl(s->fd, CDROM_LOCKDOOR, locked) < 0) {
3581c1bb86cdSEric Blake         /*
3582c1bb86cdSEric Blake          * Note: an error can happen if the distribution automatically
3583c1bb86cdSEric Blake          * mounts the CD-ROM
3584c1bb86cdSEric Blake          */
3585c1bb86cdSEric Blake         /* perror("CDROM_LOCKDOOR"); */
3586c1bb86cdSEric Blake     }
3587c1bb86cdSEric Blake }
3588c1bb86cdSEric Blake 
3589c1bb86cdSEric Blake static BlockDriver bdrv_host_cdrom = {
3590c1bb86cdSEric Blake     .format_name        = "host_cdrom",
3591c1bb86cdSEric Blake     .protocol_name      = "host_cdrom",
3592c1bb86cdSEric Blake     .instance_size      = sizeof(BDRVRawState),
3593c1bb86cdSEric Blake     .bdrv_needs_filename = true,
3594c1bb86cdSEric Blake     .bdrv_probe_device	= cdrom_probe_device,
3595c1bb86cdSEric Blake     .bdrv_parse_filename = cdrom_parse_filename,
3596c1bb86cdSEric Blake     .bdrv_file_open     = cdrom_open,
3597c1bb86cdSEric Blake     .bdrv_close         = raw_close,
3598c1bb86cdSEric Blake     .bdrv_reopen_prepare = raw_reopen_prepare,
3599c1bb86cdSEric Blake     .bdrv_reopen_commit  = raw_reopen_commit,
3600c1bb86cdSEric Blake     .bdrv_reopen_abort   = raw_reopen_abort,
3601efc75e2aSStefan Hajnoczi     .bdrv_co_create_opts = hdev_co_create_opts,
3602c1bb86cdSEric Blake     .create_opts         = &raw_create_opts,
36038a2ce0bcSAlberto Garcia     .mutable_opts        = mutable_opts,
3604dd577a26SStefan Hajnoczi     .bdrv_co_invalidate_cache = raw_co_invalidate_cache,
3605c1bb86cdSEric Blake 
3606c1bb86cdSEric Blake 
3607c1bb86cdSEric Blake     .bdrv_co_preadv         = raw_co_preadv,
3608c1bb86cdSEric Blake     .bdrv_co_pwritev        = raw_co_pwritev,
360933d70fb6SKevin Wolf     .bdrv_co_flush_to_disk  = raw_co_flush_to_disk,
3610c1bb86cdSEric Blake     .bdrv_refresh_limits = raw_refresh_limits,
3611c1bb86cdSEric Blake     .bdrv_io_plug = raw_aio_plug,
3612c1bb86cdSEric Blake     .bdrv_io_unplug = raw_aio_unplug,
3613042b757cSNishanth Aravamudan     .bdrv_attach_aio_context = raw_aio_attach_aio_context,
3614c1bb86cdSEric Blake 
3615061ca8a3SKevin Wolf     .bdrv_co_truncate    = raw_co_truncate,
3616c1bb86cdSEric Blake     .bdrv_getlength      = raw_getlength,
3617c1bb86cdSEric Blake     .has_variable_length = true,
3618c1bb86cdSEric Blake     .bdrv_get_allocated_file_size
3619c1bb86cdSEric Blake                         = raw_get_allocated_file_size,
3620c1bb86cdSEric Blake 
3621c1bb86cdSEric Blake     /* removable device support */
3622c1bb86cdSEric Blake     .bdrv_is_inserted   = cdrom_is_inserted,
3623c1bb86cdSEric Blake     .bdrv_eject         = cdrom_eject,
3624c1bb86cdSEric Blake     .bdrv_lock_medium   = cdrom_lock_medium,
3625c1bb86cdSEric Blake 
3626c1bb86cdSEric Blake     /* generic scsi device */
36272f3a7ab3SKevin Wolf     .bdrv_co_ioctl      = hdev_co_ioctl,
3628c1bb86cdSEric Blake };
3629c1bb86cdSEric Blake #endif /* __linux__ */
3630c1bb86cdSEric Blake 
3631c1bb86cdSEric Blake #if defined (__FreeBSD__) || defined(__FreeBSD_kernel__)
3632c1bb86cdSEric Blake static int cdrom_open(BlockDriverState *bs, QDict *options, int flags,
3633c1bb86cdSEric Blake                       Error **errp)
3634c1bb86cdSEric Blake {
3635c1bb86cdSEric Blake     BDRVRawState *s = bs->opaque;
3636c1bb86cdSEric Blake     Error *local_err = NULL;
3637c1bb86cdSEric Blake     int ret;
3638c1bb86cdSEric Blake 
3639c1bb86cdSEric Blake     s->type = FTYPE_CD;
3640c1bb86cdSEric Blake 
3641230ff739SJohn Snow     ret = raw_open_common(bs, options, flags, 0, true, &local_err);
3642c1bb86cdSEric Blake     if (ret) {
3643c1bb86cdSEric Blake         error_propagate(errp, local_err);
3644c1bb86cdSEric Blake         return ret;
3645c1bb86cdSEric Blake     }
3646c1bb86cdSEric Blake 
3647c1bb86cdSEric Blake     /* make sure the door isn't locked at this time */
3648c1bb86cdSEric Blake     ioctl(s->fd, CDIOCALLOW);
3649c1bb86cdSEric Blake     return 0;
3650c1bb86cdSEric Blake }
3651c1bb86cdSEric Blake 
3652c1bb86cdSEric Blake static int cdrom_probe_device(const char *filename)
3653c1bb86cdSEric Blake {
3654c1bb86cdSEric Blake     if (strstart(filename, "/dev/cd", NULL) ||
3655c1bb86cdSEric Blake             strstart(filename, "/dev/acd", NULL))
3656c1bb86cdSEric Blake         return 100;
3657c1bb86cdSEric Blake     return 0;
3658c1bb86cdSEric Blake }
3659c1bb86cdSEric Blake 
3660c1bb86cdSEric Blake static int cdrom_reopen(BlockDriverState *bs)
3661c1bb86cdSEric Blake {
3662c1bb86cdSEric Blake     BDRVRawState *s = bs->opaque;
3663c1bb86cdSEric Blake     int fd;
3664c1bb86cdSEric Blake 
3665c1bb86cdSEric Blake     /*
3666c1bb86cdSEric Blake      * Force reread of possibly changed/newly loaded disc,
3667c1bb86cdSEric Blake      * FreeBSD seems to not notice sometimes...
3668c1bb86cdSEric Blake      */
3669c1bb86cdSEric Blake     if (s->fd >= 0)
3670c1bb86cdSEric Blake         qemu_close(s->fd);
3671c1bb86cdSEric Blake     fd = qemu_open(bs->filename, s->open_flags, 0644);
3672c1bb86cdSEric Blake     if (fd < 0) {
3673c1bb86cdSEric Blake         s->fd = -1;
3674c1bb86cdSEric Blake         return -EIO;
3675c1bb86cdSEric Blake     }
3676c1bb86cdSEric Blake     s->fd = fd;
3677c1bb86cdSEric Blake 
3678c1bb86cdSEric Blake     /* make sure the door isn't locked at this time */
3679c1bb86cdSEric Blake     ioctl(s->fd, CDIOCALLOW);
3680c1bb86cdSEric Blake     return 0;
3681c1bb86cdSEric Blake }
3682c1bb86cdSEric Blake 
3683c1bb86cdSEric Blake static bool cdrom_is_inserted(BlockDriverState *bs)
3684c1bb86cdSEric Blake {
3685c1bb86cdSEric Blake     return raw_getlength(bs) > 0;
3686c1bb86cdSEric Blake }
3687c1bb86cdSEric Blake 
3688c1bb86cdSEric Blake static void cdrom_eject(BlockDriverState *bs, bool eject_flag)
3689c1bb86cdSEric Blake {
3690c1bb86cdSEric Blake     BDRVRawState *s = bs->opaque;
3691c1bb86cdSEric Blake 
3692c1bb86cdSEric Blake     if (s->fd < 0)
3693c1bb86cdSEric Blake         return;
3694c1bb86cdSEric Blake 
3695c1bb86cdSEric Blake     (void) ioctl(s->fd, CDIOCALLOW);
3696c1bb86cdSEric Blake 
3697c1bb86cdSEric Blake     if (eject_flag) {
3698c1bb86cdSEric Blake         if (ioctl(s->fd, CDIOCEJECT) < 0)
3699c1bb86cdSEric Blake             perror("CDIOCEJECT");
3700c1bb86cdSEric Blake     } else {
3701c1bb86cdSEric Blake         if (ioctl(s->fd, CDIOCCLOSE) < 0)
3702c1bb86cdSEric Blake             perror("CDIOCCLOSE");
3703c1bb86cdSEric Blake     }
3704c1bb86cdSEric Blake 
3705c1bb86cdSEric Blake     cdrom_reopen(bs);
3706c1bb86cdSEric Blake }
3707c1bb86cdSEric Blake 
3708c1bb86cdSEric Blake static void cdrom_lock_medium(BlockDriverState *bs, bool locked)
3709c1bb86cdSEric Blake {
3710c1bb86cdSEric Blake     BDRVRawState *s = bs->opaque;
3711c1bb86cdSEric Blake 
3712c1bb86cdSEric Blake     if (s->fd < 0)
3713c1bb86cdSEric Blake         return;
3714c1bb86cdSEric Blake     if (ioctl(s->fd, (locked ? CDIOCPREVENT : CDIOCALLOW)) < 0) {
3715c1bb86cdSEric Blake         /*
3716c1bb86cdSEric Blake          * Note: an error can happen if the distribution automatically
3717c1bb86cdSEric Blake          * mounts the CD-ROM
3718c1bb86cdSEric Blake          */
3719c1bb86cdSEric Blake         /* perror("CDROM_LOCKDOOR"); */
3720c1bb86cdSEric Blake     }
3721c1bb86cdSEric Blake }
3722c1bb86cdSEric Blake 
3723c1bb86cdSEric Blake static BlockDriver bdrv_host_cdrom = {
3724c1bb86cdSEric Blake     .format_name        = "host_cdrom",
3725c1bb86cdSEric Blake     .protocol_name      = "host_cdrom",
3726c1bb86cdSEric Blake     .instance_size      = sizeof(BDRVRawState),
3727c1bb86cdSEric Blake     .bdrv_needs_filename = true,
3728c1bb86cdSEric Blake     .bdrv_probe_device	= cdrom_probe_device,
3729c1bb86cdSEric Blake     .bdrv_parse_filename = cdrom_parse_filename,
3730c1bb86cdSEric Blake     .bdrv_file_open     = cdrom_open,
3731c1bb86cdSEric Blake     .bdrv_close         = raw_close,
3732c1bb86cdSEric Blake     .bdrv_reopen_prepare = raw_reopen_prepare,
3733c1bb86cdSEric Blake     .bdrv_reopen_commit  = raw_reopen_commit,
3734c1bb86cdSEric Blake     .bdrv_reopen_abort   = raw_reopen_abort,
3735efc75e2aSStefan Hajnoczi     .bdrv_co_create_opts = hdev_co_create_opts,
3736c1bb86cdSEric Blake     .create_opts        = &raw_create_opts,
37378a2ce0bcSAlberto Garcia     .mutable_opts       = mutable_opts,
3738c1bb86cdSEric Blake 
3739c1bb86cdSEric Blake     .bdrv_co_preadv         = raw_co_preadv,
3740c1bb86cdSEric Blake     .bdrv_co_pwritev        = raw_co_pwritev,
374133d70fb6SKevin Wolf     .bdrv_co_flush_to_disk  = raw_co_flush_to_disk,
3742c1bb86cdSEric Blake     .bdrv_refresh_limits = raw_refresh_limits,
3743c1bb86cdSEric Blake     .bdrv_io_plug = raw_aio_plug,
3744c1bb86cdSEric Blake     .bdrv_io_unplug = raw_aio_unplug,
3745042b757cSNishanth Aravamudan     .bdrv_attach_aio_context = raw_aio_attach_aio_context,
3746c1bb86cdSEric Blake 
3747061ca8a3SKevin Wolf     .bdrv_co_truncate    = raw_co_truncate,
3748c1bb86cdSEric Blake     .bdrv_getlength      = raw_getlength,
3749c1bb86cdSEric Blake     .has_variable_length = true,
3750c1bb86cdSEric Blake     .bdrv_get_allocated_file_size
3751c1bb86cdSEric Blake                         = raw_get_allocated_file_size,
3752c1bb86cdSEric Blake 
3753c1bb86cdSEric Blake     /* removable device support */
3754c1bb86cdSEric Blake     .bdrv_is_inserted   = cdrom_is_inserted,
3755c1bb86cdSEric Blake     .bdrv_eject         = cdrom_eject,
3756c1bb86cdSEric Blake     .bdrv_lock_medium   = cdrom_lock_medium,
3757c1bb86cdSEric Blake };
3758c1bb86cdSEric Blake #endif /* __FreeBSD__ */
3759c1bb86cdSEric Blake 
3760c1bb86cdSEric Blake static void bdrv_file_init(void)
3761c1bb86cdSEric Blake {
3762c1bb86cdSEric Blake     /*
3763c1bb86cdSEric Blake      * Register all the drivers.  Note that order is important, the driver
3764c1bb86cdSEric Blake      * registered last will get probed first.
3765c1bb86cdSEric Blake      */
3766c1bb86cdSEric Blake     bdrv_register(&bdrv_file);
3767c1bb86cdSEric Blake     bdrv_register(&bdrv_host_device);
3768c1bb86cdSEric Blake #ifdef __linux__
3769c1bb86cdSEric Blake     bdrv_register(&bdrv_host_cdrom);
3770c1bb86cdSEric Blake #endif
3771c1bb86cdSEric Blake #if defined(__FreeBSD__) || defined(__FreeBSD_kernel__)
3772c1bb86cdSEric Blake     bdrv_register(&bdrv_host_cdrom);
3773c1bb86cdSEric Blake #endif
3774c1bb86cdSEric Blake }
3775c1bb86cdSEric Blake 
3776c1bb86cdSEric Blake block_init(bdrv_file_init);
3777