xref: /linux-6.15/block/fops.c (revision de4c7bef)
1cd82cca7SChristoph Hellwig // SPDX-License-Identifier: GPL-2.0-only
2cd82cca7SChristoph Hellwig /*
3cd82cca7SChristoph Hellwig  * Copyright (C) 1991, 1992  Linus Torvalds
4cd82cca7SChristoph Hellwig  * Copyright (C) 2001  Andrea Arcangeli <[email protected]> SuSE
5cd82cca7SChristoph Hellwig  * Copyright (C) 2016 - 2020 Christoph Hellwig
6cd82cca7SChristoph Hellwig  */
7cd82cca7SChristoph Hellwig #include <linux/init.h>
8cd82cca7SChristoph Hellwig #include <linux/mm.h>
9cd82cca7SChristoph Hellwig #include <linux/blkdev.h>
10cd82cca7SChristoph Hellwig #include <linux/buffer_head.h>
11cd82cca7SChristoph Hellwig #include <linux/mpage.h>
12cd82cca7SChristoph Hellwig #include <linux/uio.h>
13cd82cca7SChristoph Hellwig #include <linux/namei.h>
14cd82cca7SChristoph Hellwig #include <linux/task_io_accounting_ops.h>
15cd82cca7SChristoph Hellwig #include <linux/falloc.h>
16cd82cca7SChristoph Hellwig #include <linux/suspend.h>
17f278eb3dSMing Lei #include <linux/fs.h>
18487c607dSChristoph Hellwig #include <linux/iomap.h>
198581fd40SJakub Kicinski #include <linux/module.h>
20cd82cca7SChristoph Hellwig #include "blk.h"
21cd82cca7SChristoph Hellwig 
22fac7c6d5SPavel Begunkov static inline struct inode *bdev_file_inode(struct file *file)
23cd82cca7SChristoph Hellwig {
24cd82cca7SChristoph Hellwig 	return file->f_mapping->host;
25cd82cca7SChristoph Hellwig }
26cd82cca7SChristoph Hellwig 
2716458cf3SBart Van Assche static blk_opf_t dio_bio_write_op(struct kiocb *iocb)
28cd82cca7SChristoph Hellwig {
2916458cf3SBart Van Assche 	blk_opf_t opf = REQ_OP_WRITE | REQ_SYNC | REQ_IDLE;
30cd82cca7SChristoph Hellwig 
31cd82cca7SChristoph Hellwig 	/* avoid the need for a I/O completion work item */
3291b94c5dSAl Viro 	if (iocb_is_dsync(iocb))
3316458cf3SBart Van Assche 		opf |= REQ_FUA;
3416458cf3SBart Van Assche 	return opf;
35cd82cca7SChristoph Hellwig }
36cd82cca7SChristoph Hellwig 
3737fee2e4SKeith Busch static bool blkdev_dio_unaligned(struct block_device *bdev, loff_t pos,
3837fee2e4SKeith Busch 			      struct iov_iter *iter)
3937fee2e4SKeith Busch {
40b1a000d3SKeith Busch 	return pos & (bdev_logical_block_size(bdev) - 1) ||
41b1a000d3SKeith Busch 		!bdev_iter_is_aligned(bdev, iter);
4237fee2e4SKeith Busch }
4337fee2e4SKeith Busch 
44cd82cca7SChristoph Hellwig #define DIO_INLINE_BIO_VECS 4
45cd82cca7SChristoph Hellwig 
46cd82cca7SChristoph Hellwig static ssize_t __blkdev_direct_IO_simple(struct kiocb *iocb,
47*de4c7befSJohn Garry 		struct iov_iter *iter, struct block_device *bdev,
48*de4c7befSJohn Garry 		unsigned int nr_pages)
49cd82cca7SChristoph Hellwig {
50cd82cca7SChristoph Hellwig 	struct bio_vec inline_vecs[DIO_INLINE_BIO_VECS], *vecs;
51cd82cca7SChristoph Hellwig 	loff_t pos = iocb->ki_pos;
52cd82cca7SChristoph Hellwig 	bool should_dirty = false;
53cd82cca7SChristoph Hellwig 	struct bio bio;
54cd82cca7SChristoph Hellwig 	ssize_t ret;
55cd82cca7SChristoph Hellwig 
56cd82cca7SChristoph Hellwig 	if (nr_pages <= DIO_INLINE_BIO_VECS)
57cd82cca7SChristoph Hellwig 		vecs = inline_vecs;
58cd82cca7SChristoph Hellwig 	else {
59cd82cca7SChristoph Hellwig 		vecs = kmalloc_array(nr_pages, sizeof(struct bio_vec),
60cd82cca7SChristoph Hellwig 				     GFP_KERNEL);
61cd82cca7SChristoph Hellwig 		if (!vecs)
62cd82cca7SChristoph Hellwig 			return -ENOMEM;
63cd82cca7SChristoph Hellwig 	}
64cd82cca7SChristoph Hellwig 
6549add496SChristoph Hellwig 	if (iov_iter_rw(iter) == READ) {
6649add496SChristoph Hellwig 		bio_init(&bio, bdev, vecs, nr_pages, REQ_OP_READ);
67fcb14cb1SAl Viro 		if (user_backed_iter(iter))
6849add496SChristoph Hellwig 			should_dirty = true;
6949add496SChristoph Hellwig 	} else {
7049add496SChristoph Hellwig 		bio_init(&bio, bdev, vecs, nr_pages, dio_bio_write_op(iocb));
7149add496SChristoph Hellwig 	}
726549a874SPavel Begunkov 	bio.bi_iter.bi_sector = pos >> SECTOR_SHIFT;
7344981351SBart Van Assche 	bio.bi_write_hint = file_inode(iocb->ki_filp)->i_write_hint;
74cd82cca7SChristoph Hellwig 	bio.bi_ioprio = iocb->ki_ioprio;
75cd82cca7SChristoph Hellwig 
76cd82cca7SChristoph Hellwig 	ret = bio_iov_iter_get_pages(&bio, iter);
77cd82cca7SChristoph Hellwig 	if (unlikely(ret))
78cd82cca7SChristoph Hellwig 		goto out;
79cd82cca7SChristoph Hellwig 	ret = bio.bi_iter.bi_size;
80cd82cca7SChristoph Hellwig 
8149add496SChristoph Hellwig 	if (iov_iter_rw(iter) == WRITE)
82cd82cca7SChristoph Hellwig 		task_io_account_write(ret);
8349add496SChristoph Hellwig 
84cd82cca7SChristoph Hellwig 	if (iocb->ki_flags & IOCB_NOWAIT)
85cd82cca7SChristoph Hellwig 		bio.bi_opf |= REQ_NOWAIT;
86cd82cca7SChristoph Hellwig 
879650b453SMing Lei 	submit_bio_wait(&bio);
88cd82cca7SChristoph Hellwig 
89cd82cca7SChristoph Hellwig 	bio_release_pages(&bio, should_dirty);
90cd82cca7SChristoph Hellwig 	if (unlikely(bio.bi_status))
91cd82cca7SChristoph Hellwig 		ret = blk_status_to_errno(bio.bi_status);
92cd82cca7SChristoph Hellwig 
93cd82cca7SChristoph Hellwig out:
94cd82cca7SChristoph Hellwig 	if (vecs != inline_vecs)
95cd82cca7SChristoph Hellwig 		kfree(vecs);
96cd82cca7SChristoph Hellwig 
97cd82cca7SChristoph Hellwig 	bio_uninit(&bio);
98cd82cca7SChristoph Hellwig 
99cd82cca7SChristoph Hellwig 	return ret;
100cd82cca7SChristoph Hellwig }
101cd82cca7SChristoph Hellwig 
10209ce8744SJens Axboe enum {
103e71aa913SPavel Begunkov 	DIO_SHOULD_DIRTY	= 1,
104e71aa913SPavel Begunkov 	DIO_IS_SYNC		= 2,
10509ce8744SJens Axboe };
10609ce8744SJens Axboe 
107cd82cca7SChristoph Hellwig struct blkdev_dio {
108cd82cca7SChristoph Hellwig 	union {
109cd82cca7SChristoph Hellwig 		struct kiocb		*iocb;
110cd82cca7SChristoph Hellwig 		struct task_struct	*waiter;
111cd82cca7SChristoph Hellwig 	};
112cd82cca7SChristoph Hellwig 	size_t			size;
113cd82cca7SChristoph Hellwig 	atomic_t		ref;
11409ce8744SJens Axboe 	unsigned int		flags;
1156155631aSJens Axboe 	struct bio		bio ____cacheline_aligned_in_smp;
116cd82cca7SChristoph Hellwig };
117cd82cca7SChristoph Hellwig 
118cd82cca7SChristoph Hellwig static struct bio_set blkdev_dio_pool;
119cd82cca7SChristoph Hellwig 
120cd82cca7SChristoph Hellwig static void blkdev_bio_end_io(struct bio *bio)
121cd82cca7SChristoph Hellwig {
122cd82cca7SChristoph Hellwig 	struct blkdev_dio *dio = bio->bi_private;
12309ce8744SJens Axboe 	bool should_dirty = dio->flags & DIO_SHOULD_DIRTY;
124cd82cca7SChristoph Hellwig 
125cd82cca7SChristoph Hellwig 	if (bio->bi_status && !dio->bio.bi_status)
126cd82cca7SChristoph Hellwig 		dio->bio.bi_status = bio->bi_status;
127cd82cca7SChristoph Hellwig 
128e71aa913SPavel Begunkov 	if (atomic_dec_and_test(&dio->ref)) {
12909ce8744SJens Axboe 		if (!(dio->flags & DIO_IS_SYNC)) {
130cd82cca7SChristoph Hellwig 			struct kiocb *iocb = dio->iocb;
131cd82cca7SChristoph Hellwig 			ssize_t ret;
132cd82cca7SChristoph Hellwig 
1333e08773cSChristoph Hellwig 			WRITE_ONCE(iocb->private, NULL);
1343e08773cSChristoph Hellwig 
135cd82cca7SChristoph Hellwig 			if (likely(!dio->bio.bi_status)) {
136cd82cca7SChristoph Hellwig 				ret = dio->size;
137cd82cca7SChristoph Hellwig 				iocb->ki_pos += ret;
138cd82cca7SChristoph Hellwig 			} else {
139cd82cca7SChristoph Hellwig 				ret = blk_status_to_errno(dio->bio.bi_status);
140cd82cca7SChristoph Hellwig 			}
141cd82cca7SChristoph Hellwig 
1426b19b766SJens Axboe 			dio->iocb->ki_complete(iocb, ret);
143cd82cca7SChristoph Hellwig 			bio_put(&dio->bio);
144cd82cca7SChristoph Hellwig 		} else {
145cd82cca7SChristoph Hellwig 			struct task_struct *waiter = dio->waiter;
146cd82cca7SChristoph Hellwig 
147cd82cca7SChristoph Hellwig 			WRITE_ONCE(dio->waiter, NULL);
148cd82cca7SChristoph Hellwig 			blk_wake_io_task(waiter);
149cd82cca7SChristoph Hellwig 		}
150cd82cca7SChristoph Hellwig 	}
151cd82cca7SChristoph Hellwig 
152cd82cca7SChristoph Hellwig 	if (should_dirty) {
153cd82cca7SChristoph Hellwig 		bio_check_pages_dirty(bio);
154cd82cca7SChristoph Hellwig 	} else {
155cd82cca7SChristoph Hellwig 		bio_release_pages(bio, false);
156cd82cca7SChristoph Hellwig 		bio_put(bio);
157cd82cca7SChristoph Hellwig 	}
158cd82cca7SChristoph Hellwig }
159cd82cca7SChristoph Hellwig 
160cd82cca7SChristoph Hellwig static ssize_t __blkdev_direct_IO(struct kiocb *iocb, struct iov_iter *iter,
161*de4c7befSJohn Garry 		struct block_device *bdev, unsigned int nr_pages)
162cd82cca7SChristoph Hellwig {
163cd82cca7SChristoph Hellwig 	struct blk_plug plug;
164cd82cca7SChristoph Hellwig 	struct blkdev_dio *dio;
165cd82cca7SChristoph Hellwig 	struct bio *bio;
166cd82cca7SChristoph Hellwig 	bool is_read = (iov_iter_rw(iter) == READ), is_sync;
16716458cf3SBart Van Assche 	blk_opf_t opf = is_read ? REQ_OP_READ : dio_bio_write_op(iocb);
168cd82cca7SChristoph Hellwig 	loff_t pos = iocb->ki_pos;
169cd82cca7SChristoph Hellwig 	int ret = 0;
170cd82cca7SChristoph Hellwig 
1710df71650SMike Snitzer 	if (iocb->ki_flags & IOCB_ALLOC_CACHE)
1720df71650SMike Snitzer 		opf |= REQ_ALLOC_CACHE;
1730df71650SMike Snitzer 	bio = bio_alloc_bioset(bdev, nr_pages, opf, GFP_KERNEL,
1740df71650SMike Snitzer 			       &blkdev_dio_pool);
175cd82cca7SChristoph Hellwig 	dio = container_of(bio, struct blkdev_dio, bio);
176e71aa913SPavel Begunkov 	atomic_set(&dio->ref, 1);
177e71aa913SPavel Begunkov 	/*
178e71aa913SPavel Begunkov 	 * Grab an extra reference to ensure the dio structure which is embedded
179e71aa913SPavel Begunkov 	 * into the first bio stays around.
180e71aa913SPavel Begunkov 	 */
181e71aa913SPavel Begunkov 	bio_get(bio);
182e71aa913SPavel Begunkov 
18309ce8744SJens Axboe 	is_sync = is_sync_kiocb(iocb);
18409ce8744SJens Axboe 	if (is_sync) {
18509ce8744SJens Axboe 		dio->flags = DIO_IS_SYNC;
186cd82cca7SChristoph Hellwig 		dio->waiter = current;
187cd82cca7SChristoph Hellwig 	} else {
18809ce8744SJens Axboe 		dio->flags = 0;
189cd82cca7SChristoph Hellwig 		dio->iocb = iocb;
190cd82cca7SChristoph Hellwig 	}
191cd82cca7SChristoph Hellwig 
192cd82cca7SChristoph Hellwig 	dio->size = 0;
193fcb14cb1SAl Viro 	if (is_read && user_backed_iter(iter))
19409ce8744SJens Axboe 		dio->flags |= DIO_SHOULD_DIRTY;
195cd82cca7SChristoph Hellwig 
196cd82cca7SChristoph Hellwig 	blk_start_plug(&plug);
197cd82cca7SChristoph Hellwig 
198cd82cca7SChristoph Hellwig 	for (;;) {
1996549a874SPavel Begunkov 		bio->bi_iter.bi_sector = pos >> SECTOR_SHIFT;
20044981351SBart Van Assche 		bio->bi_write_hint = file_inode(iocb->ki_filp)->i_write_hint;
201cd82cca7SChristoph Hellwig 		bio->bi_private = dio;
202cd82cca7SChristoph Hellwig 		bio->bi_end_io = blkdev_bio_end_io;
203cd82cca7SChristoph Hellwig 		bio->bi_ioprio = iocb->ki_ioprio;
204cd82cca7SChristoph Hellwig 
205cd82cca7SChristoph Hellwig 		ret = bio_iov_iter_get_pages(bio, iter);
206cd82cca7SChristoph Hellwig 		if (unlikely(ret)) {
207cd82cca7SChristoph Hellwig 			bio->bi_status = BLK_STS_IOERR;
208cd82cca7SChristoph Hellwig 			bio_endio(bio);
209cd82cca7SChristoph Hellwig 			break;
210cd82cca7SChristoph Hellwig 		}
21167d59247SJens Axboe 		if (iocb->ki_flags & IOCB_NOWAIT) {
21267d59247SJens Axboe 			/*
21367d59247SJens Axboe 			 * This is nonblocking IO, and we need to allocate
21467d59247SJens Axboe 			 * another bio if we have data left to map. As we
21567d59247SJens Axboe 			 * cannot guarantee that one of the sub bios will not
21667d59247SJens Axboe 			 * fail getting issued FOR NOWAIT and as error results
21767d59247SJens Axboe 			 * are coalesced across all of them, be safe and ask for
21867d59247SJens Axboe 			 * a retry of this from blocking context.
21967d59247SJens Axboe 			 */
22067d59247SJens Axboe 			if (unlikely(iov_iter_count(iter))) {
22167d59247SJens Axboe 				bio_release_pages(bio, false);
22267d59247SJens Axboe 				bio_clear_flag(bio, BIO_REFFED);
22367d59247SJens Axboe 				bio_put(bio);
22467d59247SJens Axboe 				blk_finish_plug(&plug);
22567d59247SJens Axboe 				return -EAGAIN;
22667d59247SJens Axboe 			}
22767d59247SJens Axboe 			bio->bi_opf |= REQ_NOWAIT;
22867d59247SJens Axboe 		}
229cd82cca7SChristoph Hellwig 
230cd82cca7SChristoph Hellwig 		if (is_read) {
23109ce8744SJens Axboe 			if (dio->flags & DIO_SHOULD_DIRTY)
232cd82cca7SChristoph Hellwig 				bio_set_pages_dirty(bio);
233cd82cca7SChristoph Hellwig 		} else {
234cd82cca7SChristoph Hellwig 			task_io_account_write(bio->bi_iter.bi_size);
235cd82cca7SChristoph Hellwig 		}
236cd82cca7SChristoph Hellwig 		dio->size += bio->bi_iter.bi_size;
237cd82cca7SChristoph Hellwig 		pos += bio->bi_iter.bi_size;
238cd82cca7SChristoph Hellwig 
239cd82cca7SChristoph Hellwig 		nr_pages = bio_iov_vecs_to_alloc(iter, BIO_MAX_VECS);
240cd82cca7SChristoph Hellwig 		if (!nr_pages) {
2413e08773cSChristoph Hellwig 			submit_bio(bio);
242cd82cca7SChristoph Hellwig 			break;
243cd82cca7SChristoph Hellwig 		}
244cd82cca7SChristoph Hellwig 		atomic_inc(&dio->ref);
245cd82cca7SChristoph Hellwig 		submit_bio(bio);
24607888c66SChristoph Hellwig 		bio = bio_alloc(bdev, nr_pages, opf, GFP_KERNEL);
247cd82cca7SChristoph Hellwig 	}
248cd82cca7SChristoph Hellwig 
249cd82cca7SChristoph Hellwig 	blk_finish_plug(&plug);
250cd82cca7SChristoph Hellwig 
251cd82cca7SChristoph Hellwig 	if (!is_sync)
252cd82cca7SChristoph Hellwig 		return -EIOCBQUEUED;
253cd82cca7SChristoph Hellwig 
254cd82cca7SChristoph Hellwig 	for (;;) {
255cd82cca7SChristoph Hellwig 		set_current_state(TASK_UNINTERRUPTIBLE);
256cd82cca7SChristoph Hellwig 		if (!READ_ONCE(dio->waiter))
257cd82cca7SChristoph Hellwig 			break;
258cd82cca7SChristoph Hellwig 		blk_io_schedule();
259cd82cca7SChristoph Hellwig 	}
260cd82cca7SChristoph Hellwig 	__set_current_state(TASK_RUNNING);
261cd82cca7SChristoph Hellwig 
262cd82cca7SChristoph Hellwig 	if (!ret)
263cd82cca7SChristoph Hellwig 		ret = blk_status_to_errno(dio->bio.bi_status);
264cd82cca7SChristoph Hellwig 	if (likely(!ret))
265cd82cca7SChristoph Hellwig 		ret = dio->size;
266cd82cca7SChristoph Hellwig 
267cd82cca7SChristoph Hellwig 	bio_put(&dio->bio);
268cd82cca7SChristoph Hellwig 	return ret;
269cd82cca7SChristoph Hellwig }
270cd82cca7SChristoph Hellwig 
27154a88eb8SPavel Begunkov static void blkdev_bio_end_io_async(struct bio *bio)
27254a88eb8SPavel Begunkov {
27354a88eb8SPavel Begunkov 	struct blkdev_dio *dio = container_of(bio, struct blkdev_dio, bio);
27454a88eb8SPavel Begunkov 	struct kiocb *iocb = dio->iocb;
27554a88eb8SPavel Begunkov 	ssize_t ret;
27654a88eb8SPavel Begunkov 
277bb49c6faSStefano Garzarella 	WRITE_ONCE(iocb->private, NULL);
278bb49c6faSStefano Garzarella 
27954a88eb8SPavel Begunkov 	if (likely(!bio->bi_status)) {
28054a88eb8SPavel Begunkov 		ret = dio->size;
28154a88eb8SPavel Begunkov 		iocb->ki_pos += ret;
28254a88eb8SPavel Begunkov 	} else {
28354a88eb8SPavel Begunkov 		ret = blk_status_to_errno(bio->bi_status);
28454a88eb8SPavel Begunkov 	}
28554a88eb8SPavel Begunkov 
286b6773cdbSLinus Torvalds 	iocb->ki_complete(iocb, ret);
28754a88eb8SPavel Begunkov 
28854a88eb8SPavel Begunkov 	if (dio->flags & DIO_SHOULD_DIRTY) {
28954a88eb8SPavel Begunkov 		bio_check_pages_dirty(bio);
29054a88eb8SPavel Begunkov 	} else {
29154a88eb8SPavel Begunkov 		bio_release_pages(bio, false);
29254a88eb8SPavel Begunkov 		bio_put(bio);
29354a88eb8SPavel Begunkov 	}
29454a88eb8SPavel Begunkov }
29554a88eb8SPavel Begunkov 
29654a88eb8SPavel Begunkov static ssize_t __blkdev_direct_IO_async(struct kiocb *iocb,
29754a88eb8SPavel Begunkov 					struct iov_iter *iter,
298*de4c7befSJohn Garry 					struct block_device *bdev,
29954a88eb8SPavel Begunkov 					unsigned int nr_pages)
30054a88eb8SPavel Begunkov {
301b77c88c2SChristoph Hellwig 	bool is_read = iov_iter_rw(iter) == READ;
30216458cf3SBart Van Assche 	blk_opf_t opf = is_read ? REQ_OP_READ : dio_bio_write_op(iocb);
30354a88eb8SPavel Begunkov 	struct blkdev_dio *dio;
30454a88eb8SPavel Begunkov 	struct bio *bio;
30554a88eb8SPavel Begunkov 	loff_t pos = iocb->ki_pos;
30654a88eb8SPavel Begunkov 	int ret = 0;
30754a88eb8SPavel Begunkov 
3080df71650SMike Snitzer 	if (iocb->ki_flags & IOCB_ALLOC_CACHE)
3090df71650SMike Snitzer 		opf |= REQ_ALLOC_CACHE;
3100df71650SMike Snitzer 	bio = bio_alloc_bioset(bdev, nr_pages, opf, GFP_KERNEL,
3110df71650SMike Snitzer 			       &blkdev_dio_pool);
31254a88eb8SPavel Begunkov 	dio = container_of(bio, struct blkdev_dio, bio);
31354a88eb8SPavel Begunkov 	dio->flags = 0;
31454a88eb8SPavel Begunkov 	dio->iocb = iocb;
31554a88eb8SPavel Begunkov 	bio->bi_iter.bi_sector = pos >> SECTOR_SHIFT;
31644981351SBart Van Assche 	bio->bi_write_hint = file_inode(iocb->ki_filp)->i_write_hint;
31754a88eb8SPavel Begunkov 	bio->bi_end_io = blkdev_bio_end_io_async;
31854a88eb8SPavel Begunkov 	bio->bi_ioprio = iocb->ki_ioprio;
31954a88eb8SPavel Begunkov 
3201bb6b810SPavel Begunkov 	if (iov_iter_is_bvec(iter)) {
3211bb6b810SPavel Begunkov 		/*
3221bb6b810SPavel Begunkov 		 * Users don't rely on the iterator being in any particular
3231bb6b810SPavel Begunkov 		 * state for async I/O returning -EIOCBQUEUED, hence we can
3241bb6b810SPavel Begunkov 		 * avoid expensive iov_iter_advance(). Bypass
3251bb6b810SPavel Begunkov 		 * bio_iov_iter_get_pages() and set the bvec directly.
3261bb6b810SPavel Begunkov 		 */
3271bb6b810SPavel Begunkov 		bio_iov_bvec_set(bio, iter);
3281bb6b810SPavel Begunkov 	} else {
32954a88eb8SPavel Begunkov 		ret = bio_iov_iter_get_pages(bio, iter);
33054a88eb8SPavel Begunkov 		if (unlikely(ret)) {
33175feae73SPavel Begunkov 			bio_put(bio);
33254a88eb8SPavel Begunkov 			return ret;
33354a88eb8SPavel Begunkov 		}
3341bb6b810SPavel Begunkov 	}
33554a88eb8SPavel Begunkov 	dio->size = bio->bi_iter.bi_size;
33654a88eb8SPavel Begunkov 
337b77c88c2SChristoph Hellwig 	if (is_read) {
338fcb14cb1SAl Viro 		if (user_backed_iter(iter)) {
33954a88eb8SPavel Begunkov 			dio->flags |= DIO_SHOULD_DIRTY;
34054a88eb8SPavel Begunkov 			bio_set_pages_dirty(bio);
34154a88eb8SPavel Begunkov 		}
34254a88eb8SPavel Begunkov 	} else {
34354a88eb8SPavel Begunkov 		task_io_account_write(bio->bi_iter.bi_size);
34454a88eb8SPavel Begunkov 	}
34554a88eb8SPavel Begunkov 
3462bc05769SJens Axboe 	if (iocb->ki_flags & IOCB_NOWAIT)
3472bc05769SJens Axboe 		bio->bi_opf |= REQ_NOWAIT;
3482bc05769SJens Axboe 
34954a88eb8SPavel Begunkov 	if (iocb->ki_flags & IOCB_HIPRI) {
3502bc05769SJens Axboe 		bio->bi_opf |= REQ_POLLED;
35154a88eb8SPavel Begunkov 		submit_bio(bio);
35254a88eb8SPavel Begunkov 		WRITE_ONCE(iocb->private, bio);
35354a88eb8SPavel Begunkov 	} else {
35454a88eb8SPavel Begunkov 		submit_bio(bio);
35554a88eb8SPavel Begunkov 	}
35654a88eb8SPavel Begunkov 	return -EIOCBQUEUED;
35754a88eb8SPavel Begunkov }
35854a88eb8SPavel Begunkov 
359cd82cca7SChristoph Hellwig static ssize_t blkdev_direct_IO(struct kiocb *iocb, struct iov_iter *iter)
360cd82cca7SChristoph Hellwig {
361*de4c7befSJohn Garry 	struct block_device *bdev = I_BDEV(iocb->ki_filp->f_mapping->host);
362cd82cca7SChristoph Hellwig 	unsigned int nr_pages;
363cd82cca7SChristoph Hellwig 
364cd82cca7SChristoph Hellwig 	if (!iov_iter_count(iter))
365cd82cca7SChristoph Hellwig 		return 0;
366cd82cca7SChristoph Hellwig 
367*de4c7befSJohn Garry 	if (blkdev_dio_unaligned(bdev, iocb->ki_pos, iter))
368*de4c7befSJohn Garry 		return -EINVAL;
369*de4c7befSJohn Garry 
370cd82cca7SChristoph Hellwig 	nr_pages = bio_iov_vecs_to_alloc(iter, BIO_MAX_VECS + 1);
37154a88eb8SPavel Begunkov 	if (likely(nr_pages <= BIO_MAX_VECS)) {
37254a88eb8SPavel Begunkov 		if (is_sync_kiocb(iocb))
373*de4c7befSJohn Garry 			return __blkdev_direct_IO_simple(iocb, iter, bdev,
374*de4c7befSJohn Garry 							nr_pages);
375*de4c7befSJohn Garry 		return __blkdev_direct_IO_async(iocb, iter, bdev, nr_pages);
37654a88eb8SPavel Begunkov 	}
377*de4c7befSJohn Garry 	return __blkdev_direct_IO(iocb, iter, bdev, bio_max_segs(nr_pages));
378cd82cca7SChristoph Hellwig }
379cd82cca7SChristoph Hellwig 
380487c607dSChristoph Hellwig static int blkdev_iomap_begin(struct inode *inode, loff_t offset, loff_t length,
381487c607dSChristoph Hellwig 		unsigned int flags, struct iomap *iomap, struct iomap *srcmap)
382487c607dSChristoph Hellwig {
383487c607dSChristoph Hellwig 	struct block_device *bdev = I_BDEV(inode);
384487c607dSChristoph Hellwig 	loff_t isize = i_size_read(inode);
385487c607dSChristoph Hellwig 
386487c607dSChristoph Hellwig 	iomap->bdev = bdev;
387487c607dSChristoph Hellwig 	iomap->offset = ALIGN_DOWN(offset, bdev_logical_block_size(bdev));
388487c607dSChristoph Hellwig 	if (iomap->offset >= isize)
389487c607dSChristoph Hellwig 		return -EIO;
390487c607dSChristoph Hellwig 	iomap->type = IOMAP_MAPPED;
391487c607dSChristoph Hellwig 	iomap->addr = iomap->offset;
392487c607dSChristoph Hellwig 	iomap->length = isize - iomap->offset;
393925c86a1SChristoph Hellwig 	iomap->flags |= IOMAP_F_BUFFER_HEAD; /* noop for !CONFIG_BUFFER_HEAD */
394487c607dSChristoph Hellwig 	return 0;
395487c607dSChristoph Hellwig }
396487c607dSChristoph Hellwig 
397487c607dSChristoph Hellwig static const struct iomap_ops blkdev_iomap_ops = {
398487c607dSChristoph Hellwig 	.iomap_begin		= blkdev_iomap_begin,
399487c607dSChristoph Hellwig };
400487c607dSChristoph Hellwig 
401925c86a1SChristoph Hellwig #ifdef CONFIG_BUFFER_HEAD
402925c86a1SChristoph Hellwig static int blkdev_get_block(struct inode *inode, sector_t iblock,
403925c86a1SChristoph Hellwig 		struct buffer_head *bh, int create)
404925c86a1SChristoph Hellwig {
405925c86a1SChristoph Hellwig 	bh->b_bdev = I_BDEV(inode);
406925c86a1SChristoph Hellwig 	bh->b_blocknr = iblock;
407925c86a1SChristoph Hellwig 	set_buffer_mapped(bh);
408925c86a1SChristoph Hellwig 	return 0;
409925c86a1SChristoph Hellwig }
410925c86a1SChristoph Hellwig 
41117bf23a9SMatthew Wilcox (Oracle) /*
41217bf23a9SMatthew Wilcox (Oracle)  * We cannot call mpage_writepages() as it does not take the buffer lock.
41317bf23a9SMatthew Wilcox (Oracle)  * We must use block_write_full_folio() directly which holds the buffer
41417bf23a9SMatthew Wilcox (Oracle)  * lock.  The buffer lock provides the synchronisation with writeback
41517bf23a9SMatthew Wilcox (Oracle)  * that filesystems rely on when they use the blockdev's mapping.
41617bf23a9SMatthew Wilcox (Oracle)  */
41717bf23a9SMatthew Wilcox (Oracle) static int blkdev_writepages(struct address_space *mapping,
41817bf23a9SMatthew Wilcox (Oracle) 		struct writeback_control *wbc)
419cd82cca7SChristoph Hellwig {
42017bf23a9SMatthew Wilcox (Oracle) 	struct blk_plug plug;
42117bf23a9SMatthew Wilcox (Oracle) 	int err;
42217bf23a9SMatthew Wilcox (Oracle) 
42317bf23a9SMatthew Wilcox (Oracle) 	blk_start_plug(&plug);
42417bf23a9SMatthew Wilcox (Oracle) 	err = write_cache_pages(mapping, wbc, block_write_full_folio,
42517bf23a9SMatthew Wilcox (Oracle) 			blkdev_get_block);
42617bf23a9SMatthew Wilcox (Oracle) 	blk_finish_plug(&plug);
42717bf23a9SMatthew Wilcox (Oracle) 
42817bf23a9SMatthew Wilcox (Oracle) 	return err;
429cd82cca7SChristoph Hellwig }
430cd82cca7SChristoph Hellwig 
4312c69e205SMatthew Wilcox (Oracle) static int blkdev_read_folio(struct file *file, struct folio *folio)
432cd82cca7SChristoph Hellwig {
4332c69e205SMatthew Wilcox (Oracle) 	return block_read_full_folio(folio, blkdev_get_block);
434cd82cca7SChristoph Hellwig }
435cd82cca7SChristoph Hellwig 
436cd82cca7SChristoph Hellwig static void blkdev_readahead(struct readahead_control *rac)
437cd82cca7SChristoph Hellwig {
438cd82cca7SChristoph Hellwig 	mpage_readahead(rac, blkdev_get_block);
439cd82cca7SChristoph Hellwig }
440cd82cca7SChristoph Hellwig 
441cd82cca7SChristoph Hellwig static int blkdev_write_begin(struct file *file, struct address_space *mapping,
4429d6b0cd7SMatthew Wilcox (Oracle) 		loff_t pos, unsigned len, struct page **pagep, void **fsdata)
443cd82cca7SChristoph Hellwig {
444b3992d1eSMatthew Wilcox (Oracle) 	return block_write_begin(mapping, pos, len, pagep, blkdev_get_block);
445cd82cca7SChristoph Hellwig }
446cd82cca7SChristoph Hellwig 
447cd82cca7SChristoph Hellwig static int blkdev_write_end(struct file *file, struct address_space *mapping,
448cd82cca7SChristoph Hellwig 		loff_t pos, unsigned len, unsigned copied, struct page *page,
449cd82cca7SChristoph Hellwig 		void *fsdata)
450cd82cca7SChristoph Hellwig {
451cd82cca7SChristoph Hellwig 	int ret;
452cd82cca7SChristoph Hellwig 	ret = block_write_end(file, mapping, pos, len, copied, page, fsdata);
453cd82cca7SChristoph Hellwig 
454cd82cca7SChristoph Hellwig 	unlock_page(page);
455cd82cca7SChristoph Hellwig 	put_page(page);
456cd82cca7SChristoph Hellwig 
457cd82cca7SChristoph Hellwig 	return ret;
458cd82cca7SChristoph Hellwig }
459cd82cca7SChristoph Hellwig 
460cd82cca7SChristoph Hellwig const struct address_space_operations def_blk_aops = {
461e621900aSMatthew Wilcox (Oracle) 	.dirty_folio	= block_dirty_folio,
4627ba13abbSMatthew Wilcox (Oracle) 	.invalidate_folio = block_invalidate_folio,
4632c69e205SMatthew Wilcox (Oracle) 	.read_folio	= blkdev_read_folio,
464cd82cca7SChristoph Hellwig 	.readahead	= blkdev_readahead,
46517bf23a9SMatthew Wilcox (Oracle) 	.writepages	= blkdev_writepages,
466cd82cca7SChristoph Hellwig 	.write_begin	= blkdev_write_begin,
467cd82cca7SChristoph Hellwig 	.write_end	= blkdev_write_end,
46867235182SMatthew Wilcox (Oracle) 	.migrate_folio	= buffer_migrate_folio_norefs,
469cd82cca7SChristoph Hellwig 	.is_dirty_writeback = buffer_check_dirty_writeback,
470cd82cca7SChristoph Hellwig };
471925c86a1SChristoph Hellwig #else /* CONFIG_BUFFER_HEAD */
472925c86a1SChristoph Hellwig static int blkdev_read_folio(struct file *file, struct folio *folio)
473925c86a1SChristoph Hellwig {
474925c86a1SChristoph Hellwig 	return iomap_read_folio(folio, &blkdev_iomap_ops);
475925c86a1SChristoph Hellwig }
476925c86a1SChristoph Hellwig 
477925c86a1SChristoph Hellwig static void blkdev_readahead(struct readahead_control *rac)
478925c86a1SChristoph Hellwig {
479925c86a1SChristoph Hellwig 	iomap_readahead(rac, &blkdev_iomap_ops);
480925c86a1SChristoph Hellwig }
481925c86a1SChristoph Hellwig 
482925c86a1SChristoph Hellwig static int blkdev_map_blocks(struct iomap_writepage_ctx *wpc,
48319871b5cSChristoph Hellwig 		struct inode *inode, loff_t offset, unsigned int len)
484925c86a1SChristoph Hellwig {
485925c86a1SChristoph Hellwig 	loff_t isize = i_size_read(inode);
486925c86a1SChristoph Hellwig 
487925c86a1SChristoph Hellwig 	if (WARN_ON_ONCE(offset >= isize))
488925c86a1SChristoph Hellwig 		return -EIO;
489925c86a1SChristoph Hellwig 	if (offset >= wpc->iomap.offset &&
490925c86a1SChristoph Hellwig 	    offset < wpc->iomap.offset + wpc->iomap.length)
491925c86a1SChristoph Hellwig 		return 0;
492925c86a1SChristoph Hellwig 	return blkdev_iomap_begin(inode, offset, isize - offset,
493925c86a1SChristoph Hellwig 				  IOMAP_WRITE, &wpc->iomap, NULL);
494925c86a1SChristoph Hellwig }
495925c86a1SChristoph Hellwig 
496925c86a1SChristoph Hellwig static const struct iomap_writeback_ops blkdev_writeback_ops = {
497925c86a1SChristoph Hellwig 	.map_blocks		= blkdev_map_blocks,
498925c86a1SChristoph Hellwig };
499925c86a1SChristoph Hellwig 
500925c86a1SChristoph Hellwig static int blkdev_writepages(struct address_space *mapping,
501925c86a1SChristoph Hellwig 		struct writeback_control *wbc)
502925c86a1SChristoph Hellwig {
503925c86a1SChristoph Hellwig 	struct iomap_writepage_ctx wpc = { };
504925c86a1SChristoph Hellwig 
505925c86a1SChristoph Hellwig 	return iomap_writepages(mapping, wbc, &wpc, &blkdev_writeback_ops);
506925c86a1SChristoph Hellwig }
507925c86a1SChristoph Hellwig 
508925c86a1SChristoph Hellwig const struct address_space_operations def_blk_aops = {
509925c86a1SChristoph Hellwig 	.dirty_folio	= filemap_dirty_folio,
510925c86a1SChristoph Hellwig 	.release_folio		= iomap_release_folio,
511925c86a1SChristoph Hellwig 	.invalidate_folio	= iomap_invalidate_folio,
512925c86a1SChristoph Hellwig 	.read_folio		= blkdev_read_folio,
513925c86a1SChristoph Hellwig 	.readahead		= blkdev_readahead,
514925c86a1SChristoph Hellwig 	.writepages		= blkdev_writepages,
515925c86a1SChristoph Hellwig 	.is_partially_uptodate  = iomap_is_partially_uptodate,
516af7628d6SMatthew Wilcox (Oracle) 	.error_remove_folio	= generic_error_remove_folio,
517925c86a1SChristoph Hellwig 	.migrate_folio		= filemap_migrate_folio,
518925c86a1SChristoph Hellwig };
519925c86a1SChristoph Hellwig #endif /* CONFIG_BUFFER_HEAD */
520cd82cca7SChristoph Hellwig 
521cd82cca7SChristoph Hellwig /*
522cd82cca7SChristoph Hellwig  * for a block special file file_inode(file)->i_size is zero
523cd82cca7SChristoph Hellwig  * so we compute the size by hand (just as in block_read/write above)
524cd82cca7SChristoph Hellwig  */
525cd82cca7SChristoph Hellwig static loff_t blkdev_llseek(struct file *file, loff_t offset, int whence)
526cd82cca7SChristoph Hellwig {
527cd82cca7SChristoph Hellwig 	struct inode *bd_inode = bdev_file_inode(file);
528cd82cca7SChristoph Hellwig 	loff_t retval;
529cd82cca7SChristoph Hellwig 
530cd82cca7SChristoph Hellwig 	inode_lock(bd_inode);
531cd82cca7SChristoph Hellwig 	retval = fixed_size_llseek(file, offset, whence, i_size_read(bd_inode));
532cd82cca7SChristoph Hellwig 	inode_unlock(bd_inode);
533cd82cca7SChristoph Hellwig 	return retval;
534cd82cca7SChristoph Hellwig }
535cd82cca7SChristoph Hellwig 
536cd82cca7SChristoph Hellwig static int blkdev_fsync(struct file *filp, loff_t start, loff_t end,
537cd82cca7SChristoph Hellwig 		int datasync)
538cd82cca7SChristoph Hellwig {
5394e762d86SChristoph Hellwig 	struct block_device *bdev = I_BDEV(filp->f_mapping->host);
540cd82cca7SChristoph Hellwig 	int error;
541cd82cca7SChristoph Hellwig 
542cd82cca7SChristoph Hellwig 	error = file_write_and_wait_range(filp, start, end);
543cd82cca7SChristoph Hellwig 	if (error)
544cd82cca7SChristoph Hellwig 		return error;
545cd82cca7SChristoph Hellwig 
546cd82cca7SChristoph Hellwig 	/*
547cd82cca7SChristoph Hellwig 	 * There is no need to serialise calls to blkdev_issue_flush with
548cd82cca7SChristoph Hellwig 	 * i_mutex and doing so causes performance issues with concurrent
549cd82cca7SChristoph Hellwig 	 * O_SYNC writers to a block device.
550cd82cca7SChristoph Hellwig 	 */
551cd82cca7SChristoph Hellwig 	error = blkdev_issue_flush(bdev);
552cd82cca7SChristoph Hellwig 	if (error == -EOPNOTSUPP)
553cd82cca7SChristoph Hellwig 		error = 0;
554cd82cca7SChristoph Hellwig 
555cd82cca7SChristoph Hellwig 	return error;
556cd82cca7SChristoph Hellwig }
557cd82cca7SChristoph Hellwig 
558841dd789SJan Kara /**
559841dd789SJan Kara  * file_to_blk_mode - get block open flags from file flags
560841dd789SJan Kara  * @file: file whose open flags should be converted
561841dd789SJan Kara  *
562841dd789SJan Kara  * Look at file open flags and generate corresponding block open flags from
563841dd789SJan Kara  * them. The function works both for file just being open (e.g. during ->open
564841dd789SJan Kara  * callback) and for file that is already open. This is actually non-trivial
565841dd789SJan Kara  * (see comment in the function).
566841dd789SJan Kara  */
56705bdb996SChristoph Hellwig blk_mode_t file_to_blk_mode(struct file *file)
56805bdb996SChristoph Hellwig {
56905bdb996SChristoph Hellwig 	blk_mode_t mode = 0;
57005bdb996SChristoph Hellwig 
57105bdb996SChristoph Hellwig 	if (file->f_mode & FMODE_READ)
57205bdb996SChristoph Hellwig 		mode |= BLK_OPEN_READ;
57305bdb996SChristoph Hellwig 	if (file->f_mode & FMODE_WRITE)
57405bdb996SChristoph Hellwig 		mode |= BLK_OPEN_WRITE;
575841dd789SJan Kara 	/*
576ab838b3fSChristian Brauner 	 * do_dentry_open() clears O_EXCL from f_flags, use file->private_data
577ab838b3fSChristian Brauner 	 * to determine whether the open was exclusive for already open files.
578841dd789SJan Kara 	 */
579ab838b3fSChristian Brauner 	if (file->private_data)
580ab838b3fSChristian Brauner 		mode |= BLK_OPEN_EXCL;
581841dd789SJan Kara 	else if (file->f_flags & O_EXCL)
58205bdb996SChristoph Hellwig 		mode |= BLK_OPEN_EXCL;
58305bdb996SChristoph Hellwig 	if (file->f_flags & O_NDELAY)
58405bdb996SChristoph Hellwig 		mode |= BLK_OPEN_NDELAY;
58505bdb996SChristoph Hellwig 
58605bdb996SChristoph Hellwig 	/*
58705bdb996SChristoph Hellwig 	 * If all bits in O_ACCMODE set (aka O_RDWR | O_WRONLY), the floppy
58805bdb996SChristoph Hellwig 	 * driver has historically allowed ioctls as if the file was opened for
58905bdb996SChristoph Hellwig 	 * writing, but does not allow and actual reads or writes.
59005bdb996SChristoph Hellwig 	 */
59105bdb996SChristoph Hellwig 	if ((file->f_flags & O_ACCMODE) == (O_RDWR | O_WRONLY))
59205bdb996SChristoph Hellwig 		mode |= BLK_OPEN_WRITE_IOCTL;
59305bdb996SChristoph Hellwig 
59405bdb996SChristoph Hellwig 	return mode;
59505bdb996SChristoph Hellwig }
59605bdb996SChristoph Hellwig 
597cd82cca7SChristoph Hellwig static int blkdev_open(struct inode *inode, struct file *filp)
598cd82cca7SChristoph Hellwig {
599a56aefcaSChristian Brauner 	struct block_device *bdev;
600841dd789SJan Kara 	blk_mode_t mode;
601a56aefcaSChristian Brauner 	int ret;
602cd82cca7SChristoph Hellwig 
603841dd789SJan Kara 	mode = file_to_blk_mode(filp);
604ab838b3fSChristian Brauner 	/* Use the file as the holder. */
605ab838b3fSChristian Brauner 	if (mode & BLK_OPEN_EXCL)
606ab838b3fSChristian Brauner 		filp->private_data = filp;
607ab838b3fSChristian Brauner 	ret = bdev_permission(inode->i_rdev, mode, filp->private_data);
608a56aefcaSChristian Brauner 	if (ret)
609a56aefcaSChristian Brauner 		return ret;
610cd82cca7SChristoph Hellwig 
611a56aefcaSChristian Brauner 	bdev = blkdev_get_no_open(inode->i_rdev);
612a56aefcaSChristian Brauner 	if (!bdev)
613a56aefcaSChristian Brauner 		return -ENXIO;
614e9833d87SJens Axboe 
615ab838b3fSChristian Brauner 	ret = bdev_open(bdev, mode, filp->private_data, NULL, filp);
616a56aefcaSChristian Brauner 	if (ret)
617a56aefcaSChristian Brauner 		blkdev_put_no_open(bdev);
618a56aefcaSChristian Brauner 	return ret;
619cd82cca7SChristoph Hellwig }
620cd82cca7SChristoph Hellwig 
6217ee34cbcSChristoph Hellwig static int blkdev_release(struct inode *inode, struct file *filp)
622cd82cca7SChristoph Hellwig {
6237c09a4edSChristian Brauner 	bdev_release(filp);
624cd82cca7SChristoph Hellwig 	return 0;
625cd82cca7SChristoph Hellwig }
626cd82cca7SChristoph Hellwig 
627727cfe97SChristoph Hellwig static ssize_t
628727cfe97SChristoph Hellwig blkdev_direct_write(struct kiocb *iocb, struct iov_iter *from)
629727cfe97SChristoph Hellwig {
630727cfe97SChristoph Hellwig 	size_t count = iov_iter_count(from);
631727cfe97SChristoph Hellwig 	ssize_t written;
632727cfe97SChristoph Hellwig 
633727cfe97SChristoph Hellwig 	written = kiocb_invalidate_pages(iocb, count);
634727cfe97SChristoph Hellwig 	if (written) {
635727cfe97SChristoph Hellwig 		if (written == -EBUSY)
636727cfe97SChristoph Hellwig 			return 0;
637727cfe97SChristoph Hellwig 		return written;
638727cfe97SChristoph Hellwig 	}
639727cfe97SChristoph Hellwig 
640727cfe97SChristoph Hellwig 	written = blkdev_direct_IO(iocb, from);
641727cfe97SChristoph Hellwig 	if (written > 0) {
642727cfe97SChristoph Hellwig 		kiocb_invalidate_post_direct_write(iocb, count);
643727cfe97SChristoph Hellwig 		iocb->ki_pos += written;
644727cfe97SChristoph Hellwig 		count -= written;
645727cfe97SChristoph Hellwig 	}
646727cfe97SChristoph Hellwig 	if (written != -EIOCBQUEUED)
647727cfe97SChristoph Hellwig 		iov_iter_revert(from, count - iov_iter_count(from));
648727cfe97SChristoph Hellwig 	return written;
649727cfe97SChristoph Hellwig }
650727cfe97SChristoph Hellwig 
651487c607dSChristoph Hellwig static ssize_t blkdev_buffered_write(struct kiocb *iocb, struct iov_iter *from)
652487c607dSChristoph Hellwig {
653487c607dSChristoph Hellwig 	return iomap_file_buffered_write(iocb, from, &blkdev_iomap_ops);
654487c607dSChristoph Hellwig }
655487c607dSChristoph Hellwig 
656cd82cca7SChristoph Hellwig /*
657cd82cca7SChristoph Hellwig  * Write data to the block device.  Only intended for the block device itself
658cd82cca7SChristoph Hellwig  * and the raw driver which basically is a fake block device.
659cd82cca7SChristoph Hellwig  *
660cd82cca7SChristoph Hellwig  * Does not take i_mutex for the write and thus is not for general purpose
661cd82cca7SChristoph Hellwig  * use.
662cd82cca7SChristoph Hellwig  */
663cd82cca7SChristoph Hellwig static ssize_t blkdev_write_iter(struct kiocb *iocb, struct iov_iter *from)
664cd82cca7SChristoph Hellwig {
665727cfe97SChristoph Hellwig 	struct file *file = iocb->ki_filp;
666727cfe97SChristoph Hellwig 	struct block_device *bdev = I_BDEV(file->f_mapping->host);
667fac7c6d5SPavel Begunkov 	struct inode *bd_inode = bdev->bd_inode;
668138c1a38SJens Axboe 	loff_t size = bdev_nr_bytes(bdev);
669cd82cca7SChristoph Hellwig 	size_t shorted = 0;
670cd82cca7SChristoph Hellwig 	ssize_t ret;
671cd82cca7SChristoph Hellwig 
672fac7c6d5SPavel Begunkov 	if (bdev_read_only(bdev))
673cd82cca7SChristoph Hellwig 		return -EPERM;
674cd82cca7SChristoph Hellwig 
675cd82cca7SChristoph Hellwig 	if (IS_SWAPFILE(bd_inode) && !is_hibernate_resume_dev(bd_inode->i_rdev))
676cd82cca7SChristoph Hellwig 		return -ETXTBSY;
677cd82cca7SChristoph Hellwig 
678cd82cca7SChristoph Hellwig 	if (!iov_iter_count(from))
679cd82cca7SChristoph Hellwig 		return 0;
680cd82cca7SChristoph Hellwig 
681cd82cca7SChristoph Hellwig 	if (iocb->ki_pos >= size)
682cd82cca7SChristoph Hellwig 		return -ENOSPC;
683cd82cca7SChristoph Hellwig 
684cd82cca7SChristoph Hellwig 	if ((iocb->ki_flags & (IOCB_NOWAIT | IOCB_DIRECT)) == IOCB_NOWAIT)
685cd82cca7SChristoph Hellwig 		return -EOPNOTSUPP;
686cd82cca7SChristoph Hellwig 
687cd82cca7SChristoph Hellwig 	size -= iocb->ki_pos;
688cd82cca7SChristoph Hellwig 	if (iov_iter_count(from) > size) {
689cd82cca7SChristoph Hellwig 		shorted = iov_iter_count(from) - size;
690cd82cca7SChristoph Hellwig 		iov_iter_truncate(from, size);
691cd82cca7SChristoph Hellwig 	}
692cd82cca7SChristoph Hellwig 
693727cfe97SChristoph Hellwig 	ret = file_update_time(file);
694727cfe97SChristoph Hellwig 	if (ret)
695727cfe97SChristoph Hellwig 		return ret;
696727cfe97SChristoph Hellwig 
697727cfe97SChristoph Hellwig 	if (iocb->ki_flags & IOCB_DIRECT) {
698727cfe97SChristoph Hellwig 		ret = blkdev_direct_write(iocb, from);
699727cfe97SChristoph Hellwig 		if (ret >= 0 && iov_iter_count(from))
700727cfe97SChristoph Hellwig 			ret = direct_write_fallback(iocb, from, ret,
701487c607dSChristoph Hellwig 					blkdev_buffered_write(iocb, from));
702727cfe97SChristoph Hellwig 	} else {
703487c607dSChristoph Hellwig 		ret = blkdev_buffered_write(iocb, from);
704727cfe97SChristoph Hellwig 	}
705727cfe97SChristoph Hellwig 
706cd82cca7SChristoph Hellwig 	if (ret > 0)
707cd82cca7SChristoph Hellwig 		ret = generic_write_sync(iocb, ret);
708cd82cca7SChristoph Hellwig 	iov_iter_reexpand(from, iov_iter_count(from) + shorted);
709cd82cca7SChristoph Hellwig 	return ret;
710cd82cca7SChristoph Hellwig }
711cd82cca7SChristoph Hellwig 
712cd82cca7SChristoph Hellwig static ssize_t blkdev_read_iter(struct kiocb *iocb, struct iov_iter *to)
713cd82cca7SChristoph Hellwig {
7144e762d86SChristoph Hellwig 	struct block_device *bdev = I_BDEV(iocb->ki_filp->f_mapping->host);
715138c1a38SJens Axboe 	loff_t size = bdev_nr_bytes(bdev);
716cd82cca7SChristoph Hellwig 	loff_t pos = iocb->ki_pos;
717cd82cca7SChristoph Hellwig 	size_t shorted = 0;
718ceaa7625SJens Axboe 	ssize_t ret = 0;
7193e1f941dSIlya Dryomov 	size_t count;
720cd82cca7SChristoph Hellwig 
7213e1f941dSIlya Dryomov 	if (unlikely(pos + iov_iter_count(to) > size)) {
722cd82cca7SChristoph Hellwig 		if (pos >= size)
723cd82cca7SChristoph Hellwig 			return 0;
724cd82cca7SChristoph Hellwig 		size -= pos;
7253e1f941dSIlya Dryomov 		shorted = iov_iter_count(to) - size;
726cd82cca7SChristoph Hellwig 		iov_iter_truncate(to, size);
727cd82cca7SChristoph Hellwig 	}
7283e1f941dSIlya Dryomov 
7293e1f941dSIlya Dryomov 	count = iov_iter_count(to);
7303e1f941dSIlya Dryomov 	if (!count)
7313e1f941dSIlya Dryomov 		goto reexpand; /* skip atime */
732cd82cca7SChristoph Hellwig 
733ceaa7625SJens Axboe 	if (iocb->ki_flags & IOCB_DIRECT) {
7343c435a0fSChristoph Hellwig 		ret = kiocb_write_and_wait(iocb, count);
735ceaa7625SJens Axboe 		if (ret < 0)
7363e1f941dSIlya Dryomov 			goto reexpand;
737ceaa7625SJens Axboe 		file_accessed(iocb->ki_filp);
738ceaa7625SJens Axboe 
739ceaa7625SJens Axboe 		ret = blkdev_direct_IO(iocb, to);
740ceaa7625SJens Axboe 		if (ret >= 0) {
741ceaa7625SJens Axboe 			iocb->ki_pos += ret;
742ceaa7625SJens Axboe 			count -= ret;
743ceaa7625SJens Axboe 		}
7443e1f941dSIlya Dryomov 		iov_iter_revert(to, count - iov_iter_count(to));
745ceaa7625SJens Axboe 		if (ret < 0 || !count)
7463e1f941dSIlya Dryomov 			goto reexpand;
747ceaa7625SJens Axboe 	}
748ceaa7625SJens Axboe 
749ceaa7625SJens Axboe 	ret = filemap_read(iocb, to, ret);
7506450fe1fSPavel Begunkov 
7513e1f941dSIlya Dryomov reexpand:
7526450fe1fSPavel Begunkov 	if (unlikely(shorted))
753cd82cca7SChristoph Hellwig 		iov_iter_reexpand(to, iov_iter_count(to) + shorted);
754cd82cca7SChristoph Hellwig 	return ret;
755cd82cca7SChristoph Hellwig }
756cd82cca7SChristoph Hellwig 
757cd82cca7SChristoph Hellwig #define	BLKDEV_FALLOC_FL_SUPPORTED					\
758cd82cca7SChristoph Hellwig 		(FALLOC_FL_KEEP_SIZE | FALLOC_FL_PUNCH_HOLE |		\
759cd82cca7SChristoph Hellwig 		 FALLOC_FL_ZERO_RANGE | FALLOC_FL_NO_HIDE_STALE)
760cd82cca7SChristoph Hellwig 
761cd82cca7SChristoph Hellwig static long blkdev_fallocate(struct file *file, int mode, loff_t start,
762cd82cca7SChristoph Hellwig 			     loff_t len)
763cd82cca7SChristoph Hellwig {
764f278eb3dSMing Lei 	struct inode *inode = bdev_file_inode(file);
765f278eb3dSMing Lei 	struct block_device *bdev = I_BDEV(inode);
766cd82cca7SChristoph Hellwig 	loff_t end = start + len - 1;
767cd82cca7SChristoph Hellwig 	loff_t isize;
768cd82cca7SChristoph Hellwig 	int error;
769cd82cca7SChristoph Hellwig 
770cd82cca7SChristoph Hellwig 	/* Fail if we don't recognize the flags. */
771cd82cca7SChristoph Hellwig 	if (mode & ~BLKDEV_FALLOC_FL_SUPPORTED)
772cd82cca7SChristoph Hellwig 		return -EOPNOTSUPP;
773cd82cca7SChristoph Hellwig 
774cd82cca7SChristoph Hellwig 	/* Don't go off the end of the device. */
7752a93ad8fSChristoph Hellwig 	isize = bdev_nr_bytes(bdev);
776cd82cca7SChristoph Hellwig 	if (start >= isize)
777cd82cca7SChristoph Hellwig 		return -EINVAL;
778cd82cca7SChristoph Hellwig 	if (end >= isize) {
779cd82cca7SChristoph Hellwig 		if (mode & FALLOC_FL_KEEP_SIZE) {
780cd82cca7SChristoph Hellwig 			len = isize - start;
781cd82cca7SChristoph Hellwig 			end = start + len - 1;
782cd82cca7SChristoph Hellwig 		} else
783cd82cca7SChristoph Hellwig 			return -EINVAL;
784cd82cca7SChristoph Hellwig 	}
785cd82cca7SChristoph Hellwig 
786cd82cca7SChristoph Hellwig 	/*
787cd82cca7SChristoph Hellwig 	 * Don't allow IO that isn't aligned to logical block size.
788cd82cca7SChristoph Hellwig 	 */
789cd82cca7SChristoph Hellwig 	if ((start | len) & (bdev_logical_block_size(bdev) - 1))
790cd82cca7SChristoph Hellwig 		return -EINVAL;
791cd82cca7SChristoph Hellwig 
792f278eb3dSMing Lei 	filemap_invalidate_lock(inode->i_mapping);
793f278eb3dSMing Lei 
7941364a3c3SSarthak Kukreti 	/*
7951364a3c3SSarthak Kukreti 	 * Invalidate the page cache, including dirty pages, for valid
7961364a3c3SSarthak Kukreti 	 * de-allocate mode calls to fallocate().
7971364a3c3SSarthak Kukreti 	 */
7981364a3c3SSarthak Kukreti 	switch (mode) {
7991364a3c3SSarthak Kukreti 	case FALLOC_FL_ZERO_RANGE:
8001364a3c3SSarthak Kukreti 	case FALLOC_FL_ZERO_RANGE | FALLOC_FL_KEEP_SIZE:
80105bdb996SChristoph Hellwig 		error = truncate_bdev_range(bdev, file_to_blk_mode(file), start, end);
802cd82cca7SChristoph Hellwig 		if (error)
803f278eb3dSMing Lei 			goto fail;
804cd82cca7SChristoph Hellwig 
8056549a874SPavel Begunkov 		error = blkdev_issue_zeroout(bdev, start >> SECTOR_SHIFT,
8066549a874SPavel Begunkov 					     len >> SECTOR_SHIFT, GFP_KERNEL,
8076549a874SPavel Begunkov 					     BLKDEV_ZERO_NOUNMAP);
808cd82cca7SChristoph Hellwig 		break;
809cd82cca7SChristoph Hellwig 	case FALLOC_FL_PUNCH_HOLE | FALLOC_FL_KEEP_SIZE:
8101364a3c3SSarthak Kukreti 		error = truncate_bdev_range(bdev, file_to_blk_mode(file), start, end);
8111364a3c3SSarthak Kukreti 		if (error)
8121364a3c3SSarthak Kukreti 			goto fail;
8131364a3c3SSarthak Kukreti 
8146549a874SPavel Begunkov 		error = blkdev_issue_zeroout(bdev, start >> SECTOR_SHIFT,
8156549a874SPavel Begunkov 					     len >> SECTOR_SHIFT, GFP_KERNEL,
8166549a874SPavel Begunkov 					     BLKDEV_ZERO_NOFALLBACK);
817cd82cca7SChristoph Hellwig 		break;
818cd82cca7SChristoph Hellwig 	case FALLOC_FL_PUNCH_HOLE | FALLOC_FL_KEEP_SIZE | FALLOC_FL_NO_HIDE_STALE:
8191364a3c3SSarthak Kukreti 		error = truncate_bdev_range(bdev, file_to_blk_mode(file), start, end);
8201364a3c3SSarthak Kukreti 		if (error)
8211364a3c3SSarthak Kukreti 			goto fail;
8221364a3c3SSarthak Kukreti 
8236549a874SPavel Begunkov 		error = blkdev_issue_discard(bdev, start >> SECTOR_SHIFT,
82444abff2cSChristoph Hellwig 					     len >> SECTOR_SHIFT, GFP_KERNEL);
825cd82cca7SChristoph Hellwig 		break;
826cd82cca7SChristoph Hellwig 	default:
827f278eb3dSMing Lei 		error = -EOPNOTSUPP;
828cd82cca7SChristoph Hellwig 	}
829cd82cca7SChristoph Hellwig 
830f278eb3dSMing Lei  fail:
831f278eb3dSMing Lei 	filemap_invalidate_unlock(inode->i_mapping);
832f278eb3dSMing Lei 	return error;
833cd82cca7SChristoph Hellwig }
834cd82cca7SChristoph Hellwig 
83569baa3a6SLoic Poulain static int blkdev_mmap(struct file *file, struct vm_area_struct *vma)
83669baa3a6SLoic Poulain {
83769baa3a6SLoic Poulain 	struct inode *bd_inode = bdev_file_inode(file);
83869baa3a6SLoic Poulain 
83969baa3a6SLoic Poulain 	if (bdev_read_only(I_BDEV(bd_inode)))
84069baa3a6SLoic Poulain 		return generic_file_readonly_mmap(file, vma);
84169baa3a6SLoic Poulain 
84269baa3a6SLoic Poulain 	return generic_file_mmap(file, vma);
84369baa3a6SLoic Poulain }
84469baa3a6SLoic Poulain 
845cd82cca7SChristoph Hellwig const struct file_operations def_blk_fops = {
846cd82cca7SChristoph Hellwig 	.open		= blkdev_open,
8477ee34cbcSChristoph Hellwig 	.release	= blkdev_release,
848cd82cca7SChristoph Hellwig 	.llseek		= blkdev_llseek,
849cd82cca7SChristoph Hellwig 	.read_iter	= blkdev_read_iter,
850cd82cca7SChristoph Hellwig 	.write_iter	= blkdev_write_iter,
8513e08773cSChristoph Hellwig 	.iopoll		= iocb_bio_iopoll,
85269baa3a6SLoic Poulain 	.mmap		= blkdev_mmap,
853cd82cca7SChristoph Hellwig 	.fsync		= blkdev_fsync,
8548a709512SChristoph Hellwig 	.unlocked_ioctl	= blkdev_ioctl,
855cd82cca7SChristoph Hellwig #ifdef CONFIG_COMPAT
856cd82cca7SChristoph Hellwig 	.compat_ioctl	= compat_blkdev_ioctl,
857cd82cca7SChristoph Hellwig #endif
8582cb1e089SDavid Howells 	.splice_read	= filemap_splice_read,
859cd82cca7SChristoph Hellwig 	.splice_write	= iter_file_splice_write,
860cd82cca7SChristoph Hellwig 	.fallocate	= blkdev_fallocate,
861cd82cca7SChristoph Hellwig };
862cd82cca7SChristoph Hellwig 
863cd82cca7SChristoph Hellwig static __init int blkdev_init(void)
864cd82cca7SChristoph Hellwig {
865cd82cca7SChristoph Hellwig 	return bioset_init(&blkdev_dio_pool, 4,
866cd82cca7SChristoph Hellwig 				offsetof(struct blkdev_dio, bio),
867cd82cca7SChristoph Hellwig 				BIOSET_NEED_BVECS|BIOSET_PERCPU_CACHE);
868cd82cca7SChristoph Hellwig }
869cd82cca7SChristoph Hellwig module_init(blkdev_init);
870