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