xref: /linux/fs/bcachefs/fs-io.c (revision 67e0dd8f0d8b4bf09098c4692abcb43a20089dff)
11c6fdbd8SKent Overstreet // SPDX-License-Identifier: GPL-2.0
21c6fdbd8SKent Overstreet #ifndef NO_BCACHEFS_FS
31c6fdbd8SKent Overstreet 
41c6fdbd8SKent Overstreet #include "bcachefs.h"
57b3f84eaSKent Overstreet #include "alloc_foreground.h"
607a1006aSKent Overstreet #include "bkey_buf.h"
71c6fdbd8SKent Overstreet #include "btree_update.h"
81c6fdbd8SKent Overstreet #include "buckets.h"
91c6fdbd8SKent Overstreet #include "clock.h"
101c6fdbd8SKent Overstreet #include "error.h"
11e2d9912cSKent Overstreet #include "extents.h"
1208c07feaSKent Overstreet #include "extent_update.h"
131c6fdbd8SKent Overstreet #include "fs.h"
141c6fdbd8SKent Overstreet #include "fs-io.h"
151c6fdbd8SKent Overstreet #include "fsck.h"
161c6fdbd8SKent Overstreet #include "inode.h"
171c6fdbd8SKent Overstreet #include "journal.h"
181c6fdbd8SKent Overstreet #include "io.h"
191c6fdbd8SKent Overstreet #include "keylist.h"
201c6fdbd8SKent Overstreet #include "quota.h"
2176426098SKent Overstreet #include "reflink.h"
221c6fdbd8SKent Overstreet #include "trace.h"
231c6fdbd8SKent Overstreet 
241c6fdbd8SKent Overstreet #include <linux/aio.h>
251c6fdbd8SKent Overstreet #include <linux/backing-dev.h>
261c6fdbd8SKent Overstreet #include <linux/falloc.h>
271c6fdbd8SKent Overstreet #include <linux/migrate.h>
281c6fdbd8SKent Overstreet #include <linux/mmu_context.h>
291c6fdbd8SKent Overstreet #include <linux/pagevec.h>
309ba2eb25SKent Overstreet #include <linux/rmap.h>
311c6fdbd8SKent Overstreet #include <linux/sched/signal.h>
321c6fdbd8SKent Overstreet #include <linux/task_io_accounting_ops.h>
331c6fdbd8SKent Overstreet #include <linux/uio.h>
341c6fdbd8SKent Overstreet #include <linux/writeback.h>
351c6fdbd8SKent Overstreet 
361c6fdbd8SKent Overstreet #include <trace/events/writeback.h>
371c6fdbd8SKent Overstreet 
387f5e31e1SKent Overstreet static inline bool bio_full(struct bio *bio, unsigned len)
397f5e31e1SKent Overstreet {
407f5e31e1SKent Overstreet 	if (bio->bi_vcnt >= bio->bi_max_vecs)
417f5e31e1SKent Overstreet 		return true;
427f5e31e1SKent Overstreet 	if (bio->bi_iter.bi_size > UINT_MAX - len)
437f5e31e1SKent Overstreet 		return true;
447f5e31e1SKent Overstreet 	return false;
457f5e31e1SKent Overstreet }
467f5e31e1SKent Overstreet 
47eb8e6e9cSKent Overstreet static inline struct address_space *faults_disabled_mapping(void)
48eb8e6e9cSKent Overstreet {
49eb8e6e9cSKent Overstreet 	return (void *) (((unsigned long) current->faults_disabled_mapping) & ~1UL);
50eb8e6e9cSKent Overstreet }
51eb8e6e9cSKent Overstreet 
52eb8e6e9cSKent Overstreet static inline void set_fdm_dropped_locks(void)
53eb8e6e9cSKent Overstreet {
54eb8e6e9cSKent Overstreet 	current->faults_disabled_mapping =
55eb8e6e9cSKent Overstreet 		(void *) (((unsigned long) current->faults_disabled_mapping)|1);
56eb8e6e9cSKent Overstreet }
57eb8e6e9cSKent Overstreet 
58eb8e6e9cSKent Overstreet static inline bool fdm_dropped_locks(void)
59eb8e6e9cSKent Overstreet {
60eb8e6e9cSKent Overstreet 	return ((unsigned long) current->faults_disabled_mapping) & 1;
61eb8e6e9cSKent Overstreet }
62eb8e6e9cSKent Overstreet 
631c6fdbd8SKent Overstreet struct quota_res {
641c6fdbd8SKent Overstreet 	u64				sectors;
651c6fdbd8SKent Overstreet };
661c6fdbd8SKent Overstreet 
679a3df993SKent Overstreet struct bch_writepage_io {
681c6fdbd8SKent Overstreet 	struct bch_inode_info		*inode;
691c6fdbd8SKent Overstreet 
701c6fdbd8SKent Overstreet 	/* must be last: */
711c6fdbd8SKent Overstreet 	struct bch_write_op		op;
721c6fdbd8SKent Overstreet };
731c6fdbd8SKent Overstreet 
741c6fdbd8SKent Overstreet struct dio_write {
75f8f30863SKent Overstreet 	struct completion		done;
761c6fdbd8SKent Overstreet 	struct kiocb			*req;
77ed484030SKent Overstreet 	struct mm_struct		*mm;
781c6fdbd8SKent Overstreet 	unsigned			loop:1,
791c6fdbd8SKent Overstreet 					sync:1,
801c6fdbd8SKent Overstreet 					free_iov:1;
811c6fdbd8SKent Overstreet 	struct quota_res		quota_res;
82042a1f26SKent Overstreet 	u64				written;
831c6fdbd8SKent Overstreet 
841c6fdbd8SKent Overstreet 	struct iov_iter			iter;
851c6fdbd8SKent Overstreet 	struct iovec			inline_vecs[2];
861c6fdbd8SKent Overstreet 
871c6fdbd8SKent Overstreet 	/* must be last: */
889a3df993SKent Overstreet 	struct bch_write_op		op;
891c6fdbd8SKent Overstreet };
901c6fdbd8SKent Overstreet 
911c6fdbd8SKent Overstreet struct dio_read {
921c6fdbd8SKent Overstreet 	struct closure			cl;
931c6fdbd8SKent Overstreet 	struct kiocb			*req;
941c6fdbd8SKent Overstreet 	long				ret;
95b4725cc1SKent Overstreet 	bool				should_dirty;
961c6fdbd8SKent Overstreet 	struct bch_read_bio		rbio;
971c6fdbd8SKent Overstreet };
981c6fdbd8SKent Overstreet 
991c6fdbd8SKent Overstreet /* pagecache_block must be held */
100a023127aSKent Overstreet static noinline int write_invalidate_inode_pages_range(struct address_space *mapping,
1011c6fdbd8SKent Overstreet 					      loff_t start, loff_t end)
1021c6fdbd8SKent Overstreet {
1031c6fdbd8SKent Overstreet 	int ret;
1041c6fdbd8SKent Overstreet 
1051c6fdbd8SKent Overstreet 	/*
1061c6fdbd8SKent Overstreet 	 * XXX: the way this is currently implemented, we can spin if a process
1071c6fdbd8SKent Overstreet 	 * is continually redirtying a specific page
1081c6fdbd8SKent Overstreet 	 */
1091c6fdbd8SKent Overstreet 	do {
1101c6fdbd8SKent Overstreet 		if (!mapping->nrpages)
1111c6fdbd8SKent Overstreet 			return 0;
1121c6fdbd8SKent Overstreet 
1131c6fdbd8SKent Overstreet 		ret = filemap_write_and_wait_range(mapping, start, end);
1141c6fdbd8SKent Overstreet 		if (ret)
1151c6fdbd8SKent Overstreet 			break;
1161c6fdbd8SKent Overstreet 
1171c6fdbd8SKent Overstreet 		if (!mapping->nrpages)
1181c6fdbd8SKent Overstreet 			return 0;
1191c6fdbd8SKent Overstreet 
1201c6fdbd8SKent Overstreet 		ret = invalidate_inode_pages2_range(mapping,
1211c6fdbd8SKent Overstreet 				start >> PAGE_SHIFT,
1221c6fdbd8SKent Overstreet 				end >> PAGE_SHIFT);
1231c6fdbd8SKent Overstreet 	} while (ret == -EBUSY);
1241c6fdbd8SKent Overstreet 
1251c6fdbd8SKent Overstreet 	return ret;
1261c6fdbd8SKent Overstreet }
1271c6fdbd8SKent Overstreet 
1281c6fdbd8SKent Overstreet /* quotas */
1291c6fdbd8SKent Overstreet 
1301c6fdbd8SKent Overstreet #ifdef CONFIG_BCACHEFS_QUOTA
1311c6fdbd8SKent Overstreet 
1321c6fdbd8SKent Overstreet static void bch2_quota_reservation_put(struct bch_fs *c,
1331c6fdbd8SKent Overstreet 				       struct bch_inode_info *inode,
1341c6fdbd8SKent Overstreet 				       struct quota_res *res)
1351c6fdbd8SKent Overstreet {
1361c6fdbd8SKent Overstreet 	if (!res->sectors)
1371c6fdbd8SKent Overstreet 		return;
1381c6fdbd8SKent Overstreet 
1391c6fdbd8SKent Overstreet 	mutex_lock(&inode->ei_quota_lock);
1401c6fdbd8SKent Overstreet 	BUG_ON(res->sectors > inode->ei_quota_reserved);
1411c6fdbd8SKent Overstreet 
1421c6fdbd8SKent Overstreet 	bch2_quota_acct(c, inode->ei_qid, Q_SPC,
14326609b61SKent Overstreet 			-((s64) res->sectors), KEY_TYPE_QUOTA_PREALLOC);
1441c6fdbd8SKent Overstreet 	inode->ei_quota_reserved -= res->sectors;
1451c6fdbd8SKent Overstreet 	mutex_unlock(&inode->ei_quota_lock);
1461c6fdbd8SKent Overstreet 
1471c6fdbd8SKent Overstreet 	res->sectors = 0;
1481c6fdbd8SKent Overstreet }
1491c6fdbd8SKent Overstreet 
1501c6fdbd8SKent Overstreet static int bch2_quota_reservation_add(struct bch_fs *c,
1511c6fdbd8SKent Overstreet 				      struct bch_inode_info *inode,
1521c6fdbd8SKent Overstreet 				      struct quota_res *res,
1531c6fdbd8SKent Overstreet 				      unsigned sectors,
1541c6fdbd8SKent Overstreet 				      bool check_enospc)
1551c6fdbd8SKent Overstreet {
1561c6fdbd8SKent Overstreet 	int ret;
1571c6fdbd8SKent Overstreet 
1581c6fdbd8SKent Overstreet 	mutex_lock(&inode->ei_quota_lock);
1591c6fdbd8SKent Overstreet 	ret = bch2_quota_acct(c, inode->ei_qid, Q_SPC, sectors,
16026609b61SKent Overstreet 			      check_enospc ? KEY_TYPE_QUOTA_PREALLOC : KEY_TYPE_QUOTA_NOCHECK);
1611c6fdbd8SKent Overstreet 	if (likely(!ret)) {
1621c6fdbd8SKent Overstreet 		inode->ei_quota_reserved += sectors;
1631c6fdbd8SKent Overstreet 		res->sectors += sectors;
1641c6fdbd8SKent Overstreet 	}
1651c6fdbd8SKent Overstreet 	mutex_unlock(&inode->ei_quota_lock);
1661c6fdbd8SKent Overstreet 
1671c6fdbd8SKent Overstreet 	return ret;
1681c6fdbd8SKent Overstreet }
1691c6fdbd8SKent Overstreet 
1701c6fdbd8SKent Overstreet #else
1711c6fdbd8SKent Overstreet 
1721c6fdbd8SKent Overstreet static void bch2_quota_reservation_put(struct bch_fs *c,
1731c6fdbd8SKent Overstreet 				       struct bch_inode_info *inode,
1741c6fdbd8SKent Overstreet 				       struct quota_res *res)
1751c6fdbd8SKent Overstreet {
1761c6fdbd8SKent Overstreet }
1771c6fdbd8SKent Overstreet 
1781c6fdbd8SKent Overstreet static int bch2_quota_reservation_add(struct bch_fs *c,
1791c6fdbd8SKent Overstreet 				      struct bch_inode_info *inode,
1801c6fdbd8SKent Overstreet 				      struct quota_res *res,
1811c6fdbd8SKent Overstreet 				      unsigned sectors,
1821c6fdbd8SKent Overstreet 				      bool check_enospc)
1831c6fdbd8SKent Overstreet {
1841c6fdbd8SKent Overstreet 	return 0;
1851c6fdbd8SKent Overstreet }
1861c6fdbd8SKent Overstreet 
1871c6fdbd8SKent Overstreet #endif
1881c6fdbd8SKent Overstreet 
1891c6fdbd8SKent Overstreet /* i_size updates: */
1901c6fdbd8SKent Overstreet 
1912ea90048SKent Overstreet struct inode_new_size {
1922ea90048SKent Overstreet 	loff_t		new_size;
1932ea90048SKent Overstreet 	u64		now;
1942ea90048SKent Overstreet 	unsigned	fields;
1952ea90048SKent Overstreet };
1962ea90048SKent Overstreet 
1971c6fdbd8SKent Overstreet static int inode_set_size(struct bch_inode_info *inode,
1981c6fdbd8SKent Overstreet 			  struct bch_inode_unpacked *bi,
1991c6fdbd8SKent Overstreet 			  void *p)
2001c6fdbd8SKent Overstreet {
2012ea90048SKent Overstreet 	struct inode_new_size *s = p;
2021c6fdbd8SKent Overstreet 
2032ea90048SKent Overstreet 	bi->bi_size = s->new_size;
2042ea90048SKent Overstreet 	if (s->fields & ATTR_ATIME)
2052ea90048SKent Overstreet 		bi->bi_atime = s->now;
2062ea90048SKent Overstreet 	if (s->fields & ATTR_MTIME)
2072ea90048SKent Overstreet 		bi->bi_mtime = s->now;
2082ea90048SKent Overstreet 	if (s->fields & ATTR_CTIME)
2092ea90048SKent Overstreet 		bi->bi_ctime = s->now;
2101c6fdbd8SKent Overstreet 
2111c6fdbd8SKent Overstreet 	return 0;
2121c6fdbd8SKent Overstreet }
2131c6fdbd8SKent Overstreet 
21476426098SKent Overstreet int __must_check bch2_write_inode_size(struct bch_fs *c,
2151c6fdbd8SKent Overstreet 				       struct bch_inode_info *inode,
2162ea90048SKent Overstreet 				       loff_t new_size, unsigned fields)
2171c6fdbd8SKent Overstreet {
2182ea90048SKent Overstreet 	struct inode_new_size s = {
2192ea90048SKent Overstreet 		.new_size	= new_size,
2202ea90048SKent Overstreet 		.now		= bch2_current_time(c),
2212ea90048SKent Overstreet 		.fields		= fields,
2222ea90048SKent Overstreet 	};
2232ea90048SKent Overstreet 
2242ea90048SKent Overstreet 	return bch2_write_inode(c, inode, inode_set_size, &s, fields);
2251c6fdbd8SKent Overstreet }
2261c6fdbd8SKent Overstreet 
2271c6fdbd8SKent Overstreet static void i_sectors_acct(struct bch_fs *c, struct bch_inode_info *inode,
228190fa7afSKent Overstreet 			   struct quota_res *quota_res, s64 sectors)
2291c6fdbd8SKent Overstreet {
230190fa7afSKent Overstreet 	if (!sectors)
231190fa7afSKent Overstreet 		return;
232190fa7afSKent Overstreet 
2331c6fdbd8SKent Overstreet 	mutex_lock(&inode->ei_quota_lock);
2341c6fdbd8SKent Overstreet #ifdef CONFIG_BCACHEFS_QUOTA
2351c6fdbd8SKent Overstreet 	if (quota_res && sectors > 0) {
2361c6fdbd8SKent Overstreet 		BUG_ON(sectors > quota_res->sectors);
2371c6fdbd8SKent Overstreet 		BUG_ON(sectors > inode->ei_quota_reserved);
2381c6fdbd8SKent Overstreet 
2391c6fdbd8SKent Overstreet 		quota_res->sectors -= sectors;
2401c6fdbd8SKent Overstreet 		inode->ei_quota_reserved -= sectors;
2411c6fdbd8SKent Overstreet 	} else {
24226609b61SKent Overstreet 		bch2_quota_acct(c, inode->ei_qid, Q_SPC, sectors, KEY_TYPE_QUOTA_WARN);
2431c6fdbd8SKent Overstreet 	}
2441c6fdbd8SKent Overstreet #endif
2451c6fdbd8SKent Overstreet 	inode->v.i_blocks += sectors;
2461c6fdbd8SKent Overstreet 	mutex_unlock(&inode->ei_quota_lock);
2471c6fdbd8SKent Overstreet }
2481c6fdbd8SKent Overstreet 
2491c6fdbd8SKent Overstreet /* page state: */
2501c6fdbd8SKent Overstreet 
2511c6fdbd8SKent Overstreet /* stored in page->private: */
2521c6fdbd8SKent Overstreet 
253f57a6a5dSKent Overstreet struct bch_page_sector {
254f81b648dSKent Overstreet 	/* Uncompressed, fully allocated replicas: */
255f57a6a5dSKent Overstreet 	unsigned		nr_replicas:3;
256f81b648dSKent Overstreet 
257f81b648dSKent Overstreet 	/* Owns PAGE_SECTORS * replicas_reserved sized reservation: */
258f57a6a5dSKent Overstreet 	unsigned		replicas_reserved:3;
2591c6fdbd8SKent Overstreet 
260f57a6a5dSKent Overstreet 	/* i_sectors: */
261f57a6a5dSKent Overstreet 	enum {
262f57a6a5dSKent Overstreet 		SECTOR_UNALLOCATED,
2632ba5d38bSKent Overstreet 		SECTOR_RESERVED,
264f57a6a5dSKent Overstreet 		SECTOR_DIRTY,
265f57a6a5dSKent Overstreet 		SECTOR_ALLOCATED,
266f57a6a5dSKent Overstreet 	}			state:2;
2671c6fdbd8SKent Overstreet };
2681c6fdbd8SKent Overstreet 
269f57a6a5dSKent Overstreet struct bch_page_state {
2703826ee0bSKent Overstreet 	spinlock_t		lock;
2717f5e31e1SKent Overstreet 	atomic_t		write_count;
272f57a6a5dSKent Overstreet 	struct bch_page_sector	s[PAGE_SECTORS];
273f57a6a5dSKent Overstreet };
274f57a6a5dSKent Overstreet 
275f57a6a5dSKent Overstreet static inline struct bch_page_state *__bch2_page_state(struct page *page)
2761c6fdbd8SKent Overstreet {
277f57a6a5dSKent Overstreet 	return page_has_private(page)
278f57a6a5dSKent Overstreet 		? (struct bch_page_state *) page_private(page)
279f57a6a5dSKent Overstreet 		: NULL;
280f57a6a5dSKent Overstreet }
2811c6fdbd8SKent Overstreet 
282f57a6a5dSKent Overstreet static inline struct bch_page_state *bch2_page_state(struct page *page)
283f57a6a5dSKent Overstreet {
284e1036a2aSKent Overstreet 	EBUG_ON(!PageLocked(page));
2851c6fdbd8SKent Overstreet 
286f57a6a5dSKent Overstreet 	return __bch2_page_state(page);
287f57a6a5dSKent Overstreet }
288f57a6a5dSKent Overstreet 
289f57a6a5dSKent Overstreet /* for newly allocated pages: */
290f57a6a5dSKent Overstreet static void __bch2_page_state_release(struct page *page)
291f57a6a5dSKent Overstreet {
29200276f9fSMatthew Wilcox (Oracle) 	kfree(detach_page_private(page));
293f57a6a5dSKent Overstreet }
294f57a6a5dSKent Overstreet 
295f57a6a5dSKent Overstreet static void bch2_page_state_release(struct page *page)
296f57a6a5dSKent Overstreet {
29700276f9fSMatthew Wilcox (Oracle) 	EBUG_ON(!PageLocked(page));
29800276f9fSMatthew Wilcox (Oracle) 	__bch2_page_state_release(page);
299f57a6a5dSKent Overstreet }
300f57a6a5dSKent Overstreet 
301f57a6a5dSKent Overstreet /* for newly allocated pages: */
302f57a6a5dSKent Overstreet static struct bch_page_state *__bch2_page_state_create(struct page *page,
303f57a6a5dSKent Overstreet 						       gfp_t gfp)
304f57a6a5dSKent Overstreet {
305f57a6a5dSKent Overstreet 	struct bch_page_state *s;
306f57a6a5dSKent Overstreet 
307f57a6a5dSKent Overstreet 	s = kzalloc(sizeof(*s), GFP_NOFS|gfp);
308f57a6a5dSKent Overstreet 	if (!s)
309f57a6a5dSKent Overstreet 		return NULL;
310f57a6a5dSKent Overstreet 
3113826ee0bSKent Overstreet 	spin_lock_init(&s->lock);
31200276f9fSMatthew Wilcox (Oracle) 	attach_page_private(page, s);
3131c6fdbd8SKent Overstreet 	return s;
3141c6fdbd8SKent Overstreet }
3151c6fdbd8SKent Overstreet 
316f57a6a5dSKent Overstreet static struct bch_page_state *bch2_page_state_create(struct page *page,
317f57a6a5dSKent Overstreet 						     gfp_t gfp)
318f57a6a5dSKent Overstreet {
319f57a6a5dSKent Overstreet 	return bch2_page_state(page) ?: __bch2_page_state_create(page, gfp);
320f57a6a5dSKent Overstreet }
321f57a6a5dSKent Overstreet 
322e1036a2aSKent Overstreet static inline unsigned inode_nr_replicas(struct bch_fs *c, struct bch_inode_info *inode)
323e1036a2aSKent Overstreet {
324e1036a2aSKent Overstreet 	/* XXX: this should not be open coded */
325e1036a2aSKent Overstreet 	return inode->ei_inode.bi_data_replicas
326e1036a2aSKent Overstreet 		? inode->ei_inode.bi_data_replicas - 1
327e1036a2aSKent Overstreet 		: c->opts.data_replicas;
328e1036a2aSKent Overstreet }
329e1036a2aSKent Overstreet 
330f57a6a5dSKent Overstreet static inline unsigned sectors_to_reserve(struct bch_page_sector *s,
331f57a6a5dSKent Overstreet 						  unsigned nr_replicas)
332f57a6a5dSKent Overstreet {
333f57a6a5dSKent Overstreet 	return max(0, (int) nr_replicas -
334f57a6a5dSKent Overstreet 		   s->nr_replicas -
335f57a6a5dSKent Overstreet 		   s->replicas_reserved);
336f57a6a5dSKent Overstreet }
337f57a6a5dSKent Overstreet 
338f57a6a5dSKent Overstreet static int bch2_get_page_disk_reservation(struct bch_fs *c,
339f57a6a5dSKent Overstreet 				struct bch_inode_info *inode,
3401c6fdbd8SKent Overstreet 				struct page *page, bool check_enospc)
3411c6fdbd8SKent Overstreet {
342f57a6a5dSKent Overstreet 	struct bch_page_state *s = bch2_page_state_create(page, 0);
343e1036a2aSKent Overstreet 	unsigned nr_replicas = inode_nr_replicas(c, inode);
344f57a6a5dSKent Overstreet 	struct disk_reservation disk_res = { 0 };
345f57a6a5dSKent Overstreet 	unsigned i, disk_res_sectors = 0;
346f81b648dSKent Overstreet 	int ret;
3471c6fdbd8SKent Overstreet 
348f57a6a5dSKent Overstreet 	if (!s)
349f57a6a5dSKent Overstreet 		return -ENOMEM;
3501c6fdbd8SKent Overstreet 
351f57a6a5dSKent Overstreet 	for (i = 0; i < ARRAY_SIZE(s->s); i++)
352f57a6a5dSKent Overstreet 		disk_res_sectors += sectors_to_reserve(&s->s[i], nr_replicas);
353f57a6a5dSKent Overstreet 
354f57a6a5dSKent Overstreet 	if (!disk_res_sectors)
355f57a6a5dSKent Overstreet 		return 0;
356f57a6a5dSKent Overstreet 
357f57a6a5dSKent Overstreet 	ret = bch2_disk_reservation_get(c, &disk_res,
358f57a6a5dSKent Overstreet 					disk_res_sectors, 1,
359f57a6a5dSKent Overstreet 					!check_enospc
360f57a6a5dSKent Overstreet 					? BCH_DISK_RESERVATION_NOFAIL
361f57a6a5dSKent Overstreet 					: 0);
3621c6fdbd8SKent Overstreet 	if (unlikely(ret))
363f81b648dSKent Overstreet 		return ret;
364f81b648dSKent Overstreet 
365f57a6a5dSKent Overstreet 	for (i = 0; i < ARRAY_SIZE(s->s); i++)
366f57a6a5dSKent Overstreet 		s->s[i].replicas_reserved +=
367f57a6a5dSKent Overstreet 			sectors_to_reserve(&s->s[i], nr_replicas);
368f57a6a5dSKent Overstreet 
369f57a6a5dSKent Overstreet 	return 0;
3701c6fdbd8SKent Overstreet }
3711c6fdbd8SKent Overstreet 
372d1542e03SKent Overstreet struct bch2_page_reservation {
373d1542e03SKent Overstreet 	struct disk_reservation	disk;
374d1542e03SKent Overstreet 	struct quota_res	quota;
375d1542e03SKent Overstreet };
376d1542e03SKent Overstreet 
377d1542e03SKent Overstreet static void bch2_page_reservation_init(struct bch_fs *c,
378f57a6a5dSKent Overstreet 			struct bch_inode_info *inode,
379d1542e03SKent Overstreet 			struct bch2_page_reservation *res)
380d1542e03SKent Overstreet {
381d1542e03SKent Overstreet 	memset(res, 0, sizeof(*res));
382d1542e03SKent Overstreet 
383d1542e03SKent Overstreet 	res->disk.nr_replicas = inode_nr_replicas(c, inode);
384d1542e03SKent Overstreet }
385d1542e03SKent Overstreet 
386d1542e03SKent Overstreet static void bch2_page_reservation_put(struct bch_fs *c,
387d1542e03SKent Overstreet 			struct bch_inode_info *inode,
388d1542e03SKent Overstreet 			struct bch2_page_reservation *res)
389d1542e03SKent Overstreet {
390d1542e03SKent Overstreet 	bch2_disk_reservation_put(c, &res->disk);
391d1542e03SKent Overstreet 	bch2_quota_reservation_put(c, inode, &res->quota);
392d1542e03SKent Overstreet }
393d1542e03SKent Overstreet 
394d1542e03SKent Overstreet static int bch2_page_reservation_get(struct bch_fs *c,
395d1542e03SKent Overstreet 			struct bch_inode_info *inode, struct page *page,
396d1542e03SKent Overstreet 			struct bch2_page_reservation *res,
397d1542e03SKent Overstreet 			unsigned offset, unsigned len, bool check_enospc)
398f57a6a5dSKent Overstreet {
399f57a6a5dSKent Overstreet 	struct bch_page_state *s = bch2_page_state_create(page, 0);
400d1542e03SKent Overstreet 	unsigned i, disk_sectors = 0, quota_sectors = 0;
401f57a6a5dSKent Overstreet 	int ret;
402f57a6a5dSKent Overstreet 
403f57a6a5dSKent Overstreet 	if (!s)
404f57a6a5dSKent Overstreet 		return -ENOMEM;
405f57a6a5dSKent Overstreet 
4064b0a66d5SKent Overstreet 	for (i = round_down(offset, block_bytes(c)) >> 9;
4074b0a66d5SKent Overstreet 	     i < round_up(offset + len, block_bytes(c)) >> 9;
408d1542e03SKent Overstreet 	     i++) {
409d1542e03SKent Overstreet 		disk_sectors += sectors_to_reserve(&s->s[i],
410d1542e03SKent Overstreet 						res->disk.nr_replicas);
411d1542e03SKent Overstreet 		quota_sectors += s->s[i].state == SECTOR_UNALLOCATED;
4121c6fdbd8SKent Overstreet 	}
4131c6fdbd8SKent Overstreet 
414d1542e03SKent Overstreet 	if (disk_sectors) {
415d1542e03SKent Overstreet 		ret = bch2_disk_reservation_add(c, &res->disk,
416d1542e03SKent Overstreet 						disk_sectors,
417d1542e03SKent Overstreet 						!check_enospc
418d1542e03SKent Overstreet 						? BCH_DISK_RESERVATION_NOFAIL
419d1542e03SKent Overstreet 						: 0);
420d1542e03SKent Overstreet 		if (unlikely(ret))
421d1542e03SKent Overstreet 			return ret;
422d1542e03SKent Overstreet 	}
423d1542e03SKent Overstreet 
424d1542e03SKent Overstreet 	if (quota_sectors) {
425d1542e03SKent Overstreet 		ret = bch2_quota_reservation_add(c, inode, &res->quota,
426d1542e03SKent Overstreet 						 quota_sectors,
427d1542e03SKent Overstreet 						 check_enospc);
428d1542e03SKent Overstreet 		if (unlikely(ret)) {
429d1542e03SKent Overstreet 			struct disk_reservation tmp = {
430d1542e03SKent Overstreet 				.sectors = disk_sectors
431d1542e03SKent Overstreet 			};
432d1542e03SKent Overstreet 
433d1542e03SKent Overstreet 			bch2_disk_reservation_put(c, &tmp);
434d1542e03SKent Overstreet 			res->disk.sectors -= disk_sectors;
435d1542e03SKent Overstreet 			return ret;
436d1542e03SKent Overstreet 		}
437d1542e03SKent Overstreet 	}
438d1542e03SKent Overstreet 
439d1542e03SKent Overstreet 	return 0;
440f57a6a5dSKent Overstreet }
441f57a6a5dSKent Overstreet 
4421c6fdbd8SKent Overstreet static void bch2_clear_page_bits(struct page *page)
4431c6fdbd8SKent Overstreet {
4441c6fdbd8SKent Overstreet 	struct bch_inode_info *inode = to_bch_ei(page->mapping->host);
4451c6fdbd8SKent Overstreet 	struct bch_fs *c = inode->v.i_sb->s_fs_info;
446f57a6a5dSKent Overstreet 	struct bch_page_state *s = bch2_page_state(page);
447d1542e03SKent Overstreet 	struct disk_reservation disk_res = { 0 };
448f57a6a5dSKent Overstreet 	int i, dirty_sectors = 0;
4491c6fdbd8SKent Overstreet 
450f57a6a5dSKent Overstreet 	if (!s)
4511c6fdbd8SKent Overstreet 		return;
4521c6fdbd8SKent Overstreet 
4533826ee0bSKent Overstreet 	EBUG_ON(!PageLocked(page));
4543826ee0bSKent Overstreet 	EBUG_ON(PageWriteback(page));
4553826ee0bSKent Overstreet 
456f57a6a5dSKent Overstreet 	for (i = 0; i < ARRAY_SIZE(s->s); i++) {
457d1542e03SKent Overstreet 		disk_res.sectors += s->s[i].replicas_reserved;
458d1542e03SKent Overstreet 		s->s[i].replicas_reserved = 0;
459d1542e03SKent Overstreet 
460f57a6a5dSKent Overstreet 		if (s->s[i].state == SECTOR_DIRTY) {
461f57a6a5dSKent Overstreet 			dirty_sectors++;
462f57a6a5dSKent Overstreet 			s->s[i].state = SECTOR_UNALLOCATED;
463f57a6a5dSKent Overstreet 		}
464f57a6a5dSKent Overstreet 	}
465adfcfaf0SKent Overstreet 
466d1542e03SKent Overstreet 	bch2_disk_reservation_put(c, &disk_res);
467d1542e03SKent Overstreet 
468f57a6a5dSKent Overstreet 	if (dirty_sectors)
469f57a6a5dSKent Overstreet 		i_sectors_acct(c, inode, NULL, -dirty_sectors);
470adfcfaf0SKent Overstreet 
471f57a6a5dSKent Overstreet 	bch2_page_state_release(page);
4721c6fdbd8SKent Overstreet }
4731c6fdbd8SKent Overstreet 
474d1542e03SKent Overstreet static void bch2_set_page_dirty(struct bch_fs *c,
475d1542e03SKent Overstreet 			struct bch_inode_info *inode, struct page *page,
476d1542e03SKent Overstreet 			struct bch2_page_reservation *res,
477d1542e03SKent Overstreet 			unsigned offset, unsigned len)
4781c6fdbd8SKent Overstreet {
479d1542e03SKent Overstreet 	struct bch_page_state *s = bch2_page_state(page);
480f57a6a5dSKent Overstreet 	unsigned i, dirty_sectors = 0;
4811c6fdbd8SKent Overstreet 
482877dfb34SKent Overstreet 	WARN_ON((u64) page_offset(page) + offset + len >
483877dfb34SKent Overstreet 		round_up((u64) i_size_read(&inode->v), block_bytes(c)));
484fb472ac5SKent Overstreet 
4853826ee0bSKent Overstreet 	spin_lock(&s->lock);
4863826ee0bSKent Overstreet 
4874b0a66d5SKent Overstreet 	for (i = round_down(offset, block_bytes(c)) >> 9;
4884b0a66d5SKent Overstreet 	     i < round_up(offset + len, block_bytes(c)) >> 9;
489d1542e03SKent Overstreet 	     i++) {
490d1542e03SKent Overstreet 		unsigned sectors = sectors_to_reserve(&s->s[i],
491d1542e03SKent Overstreet 						res->disk.nr_replicas);
4921c6fdbd8SKent Overstreet 
493406d6d5aSKent Overstreet 		/*
494406d6d5aSKent Overstreet 		 * This can happen if we race with the error path in
495406d6d5aSKent Overstreet 		 * bch2_writepage_io_done():
496406d6d5aSKent Overstreet 		 */
497406d6d5aSKent Overstreet 		sectors = min_t(unsigned, sectors, res->disk.sectors);
498406d6d5aSKent Overstreet 
499d1542e03SKent Overstreet 		s->s[i].replicas_reserved += sectors;
500d1542e03SKent Overstreet 		res->disk.sectors -= sectors;
501adfcfaf0SKent Overstreet 
5022ba5d38bSKent Overstreet 		if (s->s[i].state == SECTOR_UNALLOCATED)
503f57a6a5dSKent Overstreet 			dirty_sectors++;
5042ba5d38bSKent Overstreet 
5052ba5d38bSKent Overstreet 		s->s[i].state = max_t(unsigned, s->s[i].state, SECTOR_DIRTY);
506f57a6a5dSKent Overstreet 	}
507f57a6a5dSKent Overstreet 
5083826ee0bSKent Overstreet 	spin_unlock(&s->lock);
5093826ee0bSKent Overstreet 
510f57a6a5dSKent Overstreet 	if (dirty_sectors)
511d1542e03SKent Overstreet 		i_sectors_acct(c, inode, &res->quota, dirty_sectors);
5121c6fdbd8SKent Overstreet 
513d1542e03SKent Overstreet 	if (!PageDirty(page))
514d1542e03SKent Overstreet 		filemap_dirty_folio(inode->v.i_mapping, page_folio(page));
5151c6fdbd8SKent Overstreet }
5161c6fdbd8SKent Overstreet 
5171c6fdbd8SKent Overstreet vm_fault_t bch2_page_fault(struct vm_fault *vmf)
5181c6fdbd8SKent Overstreet {
5191c6fdbd8SKent Overstreet 	struct file *file = vmf->vma->vm_file;
520eb8e6e9cSKent Overstreet 	struct address_space *mapping = file->f_mapping;
521eb8e6e9cSKent Overstreet 	struct address_space *fdm = faults_disabled_mapping();
5221c6fdbd8SKent Overstreet 	struct bch_inode_info *inode = file_bch_inode(file);
5231c6fdbd8SKent Overstreet 	int ret;
5241c6fdbd8SKent Overstreet 
525eb8e6e9cSKent Overstreet 	if (fdm == mapping)
526eb8e6e9cSKent Overstreet 		return VM_FAULT_SIGBUS;
527eb8e6e9cSKent Overstreet 
528eb8e6e9cSKent Overstreet 	/* Lock ordering: */
529eb8e6e9cSKent Overstreet 	if (fdm > mapping) {
530eb8e6e9cSKent Overstreet 		struct bch_inode_info *fdm_host = to_bch_ei(fdm->host);
531eb8e6e9cSKent Overstreet 
532eb8e6e9cSKent Overstreet 		if (bch2_pagecache_add_tryget(&inode->ei_pagecache_lock))
533eb8e6e9cSKent Overstreet 			goto got_lock;
534eb8e6e9cSKent Overstreet 
535eb8e6e9cSKent Overstreet 		bch2_pagecache_block_put(&fdm_host->ei_pagecache_lock);
536eb8e6e9cSKent Overstreet 
5371c6fdbd8SKent Overstreet 		bch2_pagecache_add_get(&inode->ei_pagecache_lock);
538eb8e6e9cSKent Overstreet 		bch2_pagecache_add_put(&inode->ei_pagecache_lock);
539eb8e6e9cSKent Overstreet 
540eb8e6e9cSKent Overstreet 		bch2_pagecache_block_get(&fdm_host->ei_pagecache_lock);
541eb8e6e9cSKent Overstreet 
542eb8e6e9cSKent Overstreet 		/* Signal that lock has been dropped: */
543eb8e6e9cSKent Overstreet 		set_fdm_dropped_locks();
544eb8e6e9cSKent Overstreet 		return VM_FAULT_SIGBUS;
545eb8e6e9cSKent Overstreet 	}
546eb8e6e9cSKent Overstreet 
547eb8e6e9cSKent Overstreet 	bch2_pagecache_add_get(&inode->ei_pagecache_lock);
548eb8e6e9cSKent Overstreet got_lock:
5491c6fdbd8SKent Overstreet 	ret = filemap_fault(vmf);
5501c6fdbd8SKent Overstreet 	bch2_pagecache_add_put(&inode->ei_pagecache_lock);
5511c6fdbd8SKent Overstreet 
5521c6fdbd8SKent Overstreet 	return ret;
5531c6fdbd8SKent Overstreet }
5541c6fdbd8SKent Overstreet 
5551c6fdbd8SKent Overstreet vm_fault_t bch2_page_mkwrite(struct vm_fault *vmf)
5561c6fdbd8SKent Overstreet {
5571c6fdbd8SKent Overstreet 	struct page *page = vmf->page;
5581c6fdbd8SKent Overstreet 	struct file *file = vmf->vma->vm_file;
5591c6fdbd8SKent Overstreet 	struct bch_inode_info *inode = file_bch_inode(file);
5601c6fdbd8SKent Overstreet 	struct address_space *mapping = file->f_mapping;
5611c6fdbd8SKent Overstreet 	struct bch_fs *c = inode->v.i_sb->s_fs_info;
562d1542e03SKent Overstreet 	struct bch2_page_reservation res;
5636cc3535dSKent Overstreet 	unsigned len;
5646cc3535dSKent Overstreet 	loff_t isize;
5651c6fdbd8SKent Overstreet 	int ret = VM_FAULT_LOCKED;
5661c6fdbd8SKent Overstreet 
567d1542e03SKent Overstreet 	bch2_page_reservation_init(c, inode, &res);
568d1542e03SKent Overstreet 
5691c6fdbd8SKent Overstreet 	sb_start_pagefault(inode->v.i_sb);
5701c6fdbd8SKent Overstreet 	file_update_time(file);
5711c6fdbd8SKent Overstreet 
5721c6fdbd8SKent Overstreet 	/*
5731c6fdbd8SKent Overstreet 	 * Not strictly necessary, but helps avoid dio writes livelocking in
5741c6fdbd8SKent Overstreet 	 * write_invalidate_inode_pages_range() - can drop this if/when we get
5751c6fdbd8SKent Overstreet 	 * a write_invalidate_inode_pages_range() that works without dropping
5761c6fdbd8SKent Overstreet 	 * page lock before invalidating page
5771c6fdbd8SKent Overstreet 	 */
5781c6fdbd8SKent Overstreet 	bch2_pagecache_add_get(&inode->ei_pagecache_lock);
5791c6fdbd8SKent Overstreet 
5801c6fdbd8SKent Overstreet 	lock_page(page);
5816cc3535dSKent Overstreet 	isize = i_size_read(&inode->v);
5826cc3535dSKent Overstreet 
5836cc3535dSKent Overstreet 	if (page->mapping != mapping || page_offset(page) >= isize) {
5841c6fdbd8SKent Overstreet 		unlock_page(page);
5851c6fdbd8SKent Overstreet 		ret = VM_FAULT_NOPAGE;
5861c6fdbd8SKent Overstreet 		goto out;
5871c6fdbd8SKent Overstreet 	}
5881c6fdbd8SKent Overstreet 
589d55460bbSKent Overstreet 	len = min_t(loff_t, PAGE_SIZE, isize - page_offset(page));
5906cc3535dSKent Overstreet 
5916cc3535dSKent Overstreet 	if (bch2_page_reservation_get(c, inode, page, &res, 0, len, true)) {
5921c6fdbd8SKent Overstreet 		unlock_page(page);
5931c6fdbd8SKent Overstreet 		ret = VM_FAULT_SIGBUS;
5941c6fdbd8SKent Overstreet 		goto out;
5951c6fdbd8SKent Overstreet 	}
5961c6fdbd8SKent Overstreet 
5976cc3535dSKent Overstreet 	bch2_set_page_dirty(c, inode, page, &res, 0, len);
5981b783a69SKent Overstreet 	bch2_page_reservation_put(c, inode, &res);
5991b783a69SKent Overstreet 
6001c6fdbd8SKent Overstreet 	wait_for_stable_page(page);
6011c6fdbd8SKent Overstreet out:
6021c6fdbd8SKent Overstreet 	bch2_pagecache_add_put(&inode->ei_pagecache_lock);
6031c6fdbd8SKent Overstreet 	sb_end_pagefault(inode->v.i_sb);
604d1542e03SKent Overstreet 
6051c6fdbd8SKent Overstreet 	return ret;
6061c6fdbd8SKent Overstreet }
6071c6fdbd8SKent Overstreet 
6081c6fdbd8SKent Overstreet void bch2_invalidate_folio(struct folio *folio, size_t offset, size_t length)
6091c6fdbd8SKent Overstreet {
6101c6fdbd8SKent Overstreet 	if (offset || length < folio_size(folio))
6111c6fdbd8SKent Overstreet 		return;
6121c6fdbd8SKent Overstreet 
6131c6fdbd8SKent Overstreet 	bch2_clear_page_bits(&folio->page);
6141c6fdbd8SKent Overstreet }
6151c6fdbd8SKent Overstreet 
6161c6fdbd8SKent Overstreet bool bch2_release_folio(struct folio *folio, gfp_t gfp_mask)
6171c6fdbd8SKent Overstreet {
618a6d90385SKent Overstreet 	if (folio_test_dirty(folio) || folio_test_writeback(folio))
6191c6fdbd8SKent Overstreet 		return false;
6201c6fdbd8SKent Overstreet 
6211c6fdbd8SKent Overstreet 	bch2_clear_page_bits(&folio->page);
6221c6fdbd8SKent Overstreet 	return true;
6231c6fdbd8SKent Overstreet }
6241c6fdbd8SKent Overstreet 
6251c6fdbd8SKent Overstreet /* readpage(s): */
6261c6fdbd8SKent Overstreet 
6271c6fdbd8SKent Overstreet static void bch2_readpages_end_io(struct bio *bio)
6281c6fdbd8SKent Overstreet {
6291c6fdbd8SKent Overstreet 	struct bvec_iter_all iter;
6301c6fdbd8SKent Overstreet 	struct bio_vec *bv;
6311c6fdbd8SKent Overstreet 
6321c6fdbd8SKent Overstreet 	bio_for_each_segment_all(bv, bio, iter) {
6331c6fdbd8SKent Overstreet 		struct page *page = bv->bv_page;
6341c6fdbd8SKent Overstreet 
6351c6fdbd8SKent Overstreet 		if (!bio->bi_status) {
6361c6fdbd8SKent Overstreet 			SetPageUptodate(page);
6371c6fdbd8SKent Overstreet 		} else {
6381c6fdbd8SKent Overstreet 			ClearPageUptodate(page);
6391c6fdbd8SKent Overstreet 			SetPageError(page);
6401c6fdbd8SKent Overstreet 		}
6411c6fdbd8SKent Overstreet 		unlock_page(page);
6421c6fdbd8SKent Overstreet 	}
6431c6fdbd8SKent Overstreet 
6441c6fdbd8SKent Overstreet 	bio_put(bio);
6451c6fdbd8SKent Overstreet }
6461c6fdbd8SKent Overstreet 
6471c6fdbd8SKent Overstreet struct readpages_iter {
6481c6fdbd8SKent Overstreet 	struct address_space	*mapping;
6491c6fdbd8SKent Overstreet 	struct page		**pages;
6501c6fdbd8SKent Overstreet 	unsigned		nr_pages;
6511c6fdbd8SKent Overstreet 	unsigned		idx;
6521c6fdbd8SKent Overstreet 	pgoff_t			offset;
6531c6fdbd8SKent Overstreet };
6541c6fdbd8SKent Overstreet 
6551c6fdbd8SKent Overstreet static int readpages_iter_init(struct readpages_iter *iter,
6561c6fdbd8SKent Overstreet 			       struct readahead_control *ractl)
6571c6fdbd8SKent Overstreet {
6581c6fdbd8SKent Overstreet 	unsigned i, nr_pages = readahead_count(ractl);
6591c6fdbd8SKent Overstreet 
6601c6fdbd8SKent Overstreet 	memset(iter, 0, sizeof(*iter));
6611c6fdbd8SKent Overstreet 
6621c6fdbd8SKent Overstreet 	iter->mapping	= ractl->mapping;
6631c6fdbd8SKent Overstreet 	iter->offset	= readahead_index(ractl);
6641c6fdbd8SKent Overstreet 	iter->nr_pages	= nr_pages;
6651c6fdbd8SKent Overstreet 
6661c6fdbd8SKent Overstreet 	iter->pages = kmalloc_array(nr_pages, sizeof(struct page *), GFP_NOFS);
6671c6fdbd8SKent Overstreet 	if (!iter->pages)
6681c6fdbd8SKent Overstreet 		return -ENOMEM;
6691c6fdbd8SKent Overstreet 
67089931472SKent Overstreet 	nr_pages = __readahead_batch(ractl, iter->pages, nr_pages);
6711c6fdbd8SKent Overstreet 	for (i = 0; i < nr_pages; i++) {
672f57a6a5dSKent Overstreet 		__bch2_page_state_create(iter->pages[i], __GFP_NOFAIL);
6731c6fdbd8SKent Overstreet 		put_page(iter->pages[i]);
6741c6fdbd8SKent Overstreet 	}
6751c6fdbd8SKent Overstreet 
6761c6fdbd8SKent Overstreet 	return 0;
6771c6fdbd8SKent Overstreet }
6781c6fdbd8SKent Overstreet 
6791c6fdbd8SKent Overstreet static inline struct page *readpage_iter_next(struct readpages_iter *iter)
6801c6fdbd8SKent Overstreet {
6811c6fdbd8SKent Overstreet 	if (iter->idx >= iter->nr_pages)
6821c6fdbd8SKent Overstreet 		return NULL;
6831c6fdbd8SKent Overstreet 
6841c6fdbd8SKent Overstreet 	EBUG_ON(iter->pages[iter->idx]->index != iter->offset + iter->idx);
6851c6fdbd8SKent Overstreet 
6861c6fdbd8SKent Overstreet 	return iter->pages[iter->idx];
6871c6fdbd8SKent Overstreet }
6881c6fdbd8SKent Overstreet 
6891c6fdbd8SKent Overstreet static void bch2_add_page_sectors(struct bio *bio, struct bkey_s_c k)
6901c6fdbd8SKent Overstreet {
6911c6fdbd8SKent Overstreet 	struct bvec_iter iter;
6921c6fdbd8SKent Overstreet 	struct bio_vec bv;
69376426098SKent Overstreet 	unsigned nr_ptrs = k.k->type == KEY_TYPE_reflink_v
6944de77495SKent Overstreet 		? 0 : bch2_bkey_nr_ptrs_fully_allocated(k);
6952ba5d38bSKent Overstreet 	unsigned state = k.k->type == KEY_TYPE_reservation
6962ba5d38bSKent Overstreet 		? SECTOR_RESERVED
6972ba5d38bSKent Overstreet 		: SECTOR_ALLOCATED;
6981c6fdbd8SKent Overstreet 
6991c6fdbd8SKent Overstreet 	bio_for_each_segment(bv, bio, iter) {
700f57a6a5dSKent Overstreet 		struct bch_page_state *s = bch2_page_state(bv.bv_page);
701f57a6a5dSKent Overstreet 		unsigned i;
702f81b648dSKent Overstreet 
703f57a6a5dSKent Overstreet 		for (i = bv.bv_offset >> 9;
704f57a6a5dSKent Overstreet 		     i < (bv.bv_offset + bv.bv_len) >> 9;
705f57a6a5dSKent Overstreet 		     i++) {
706f57a6a5dSKent Overstreet 			s->s[i].nr_replicas = nr_ptrs;
7072ba5d38bSKent Overstreet 			s->s[i].state = state;
708f57a6a5dSKent Overstreet 		}
7091c6fdbd8SKent Overstreet 	}
7101c6fdbd8SKent Overstreet }
7111c6fdbd8SKent Overstreet 
71235189e09SKent Overstreet static bool extent_partial_reads_expensive(struct bkey_s_c k)
71335189e09SKent Overstreet {
71435189e09SKent Overstreet 	struct bkey_ptrs_c ptrs = bch2_bkey_ptrs_c(k);
71535189e09SKent Overstreet 	struct bch_extent_crc_unpacked crc;
71635189e09SKent Overstreet 	const union bch_extent_entry *i;
71735189e09SKent Overstreet 
71835189e09SKent Overstreet 	bkey_for_each_crc(k.k, ptrs, crc, i)
71935189e09SKent Overstreet 		if (crc.csum_type || crc.compression_type)
72035189e09SKent Overstreet 			return true;
72135189e09SKent Overstreet 	return false;
72235189e09SKent Overstreet }
72335189e09SKent Overstreet 
7241c6fdbd8SKent Overstreet static void readpage_bio_extend(struct readpages_iter *iter,
72576426098SKent Overstreet 				struct bio *bio,
72676426098SKent Overstreet 				unsigned sectors_this_extent,
7271c6fdbd8SKent Overstreet 				bool get_more)
7281c6fdbd8SKent Overstreet {
72976426098SKent Overstreet 	while (bio_sectors(bio) < sectors_this_extent &&
7301c6fdbd8SKent Overstreet 	       bio->bi_vcnt < bio->bi_max_vecs) {
7311c6fdbd8SKent Overstreet 		pgoff_t page_offset = bio_end_sector(bio) >> PAGE_SECTOR_SHIFT;
7321c6fdbd8SKent Overstreet 		struct page *page = readpage_iter_next(iter);
7331c6fdbd8SKent Overstreet 		int ret;
7341c6fdbd8SKent Overstreet 
7351c6fdbd8SKent Overstreet 		if (page) {
7361c6fdbd8SKent Overstreet 			if (iter->offset + iter->idx != page_offset)
7371c6fdbd8SKent Overstreet 				break;
7381c6fdbd8SKent Overstreet 
7391c6fdbd8SKent Overstreet 			iter->idx++;
7401c6fdbd8SKent Overstreet 		} else {
7411c6fdbd8SKent Overstreet 			if (!get_more)
7421c6fdbd8SKent Overstreet 				break;
7431c6fdbd8SKent Overstreet 
7441c6fdbd8SKent Overstreet 			page = xa_load(&iter->mapping->i_pages, page_offset);
7451c6fdbd8SKent Overstreet 			if (page && !xa_is_value(page))
7461c6fdbd8SKent Overstreet 				break;
7471c6fdbd8SKent Overstreet 
7481c6fdbd8SKent Overstreet 			page = __page_cache_alloc(readahead_gfp_mask(iter->mapping));
7491c6fdbd8SKent Overstreet 			if (!page)
7501c6fdbd8SKent Overstreet 				break;
7511c6fdbd8SKent Overstreet 
752f57a6a5dSKent Overstreet 			if (!__bch2_page_state_create(page, 0)) {
753f57a6a5dSKent Overstreet 				put_page(page);
754f57a6a5dSKent Overstreet 				break;
755f57a6a5dSKent Overstreet 			}
7561c6fdbd8SKent Overstreet 
7571c6fdbd8SKent Overstreet 			ret = add_to_page_cache_lru(page, iter->mapping,
7581c6fdbd8SKent Overstreet 						    page_offset, GFP_NOFS);
7591c6fdbd8SKent Overstreet 			if (ret) {
760f57a6a5dSKent Overstreet 				__bch2_page_state_release(page);
7611c6fdbd8SKent Overstreet 				put_page(page);
7621c6fdbd8SKent Overstreet 				break;
7631c6fdbd8SKent Overstreet 			}
7641c6fdbd8SKent Overstreet 
7651c6fdbd8SKent Overstreet 			put_page(page);
7661c6fdbd8SKent Overstreet 		}
7671c6fdbd8SKent Overstreet 
768885678f6SKent Overstreet 		BUG_ON(!bio_add_page(bio, page, PAGE_SIZE, 0));
7691c6fdbd8SKent Overstreet 	}
7701c6fdbd8SKent Overstreet }
7711c6fdbd8SKent Overstreet 
7720f238367SKent Overstreet static void bchfs_read(struct btree_trans *trans, struct btree_iter *iter,
7731c6fdbd8SKent Overstreet 		       struct bch_read_bio *rbio, u64 inum,
7741c6fdbd8SKent Overstreet 		       struct readpages_iter *readpages_iter)
7751c6fdbd8SKent Overstreet {
7760f238367SKent Overstreet 	struct bch_fs *c = trans->c;
77707a1006aSKent Overstreet 	struct bkey_buf sk;
7781c6fdbd8SKent Overstreet 	int flags = BCH_READ_RETRY_IF_STALE|
7791c6fdbd8SKent Overstreet 		BCH_READ_MAY_PROMOTE;
78076426098SKent Overstreet 	int ret = 0;
7811c6fdbd8SKent Overstreet 
7821c6fdbd8SKent Overstreet 	rbio->c = c;
7831c6fdbd8SKent Overstreet 	rbio->start_time = local_clock();
78435189e09SKent Overstreet 
78507a1006aSKent Overstreet 	bch2_bkey_buf_init(&sk);
78676426098SKent Overstreet retry:
787700c25b3SKent Overstreet 	bch2_trans_begin(trans);
788700c25b3SKent Overstreet 
7891c6fdbd8SKent Overstreet 	while (1) {
7901c6fdbd8SKent Overstreet 		struct bkey_s_c k;
79176426098SKent Overstreet 		unsigned bytes, sectors, offset_into_extent;
7925ff75ccbSKent Overstreet 		enum btree_id data_btree = BTREE_ID_extents;
7931c6fdbd8SKent Overstreet 
7943737e0ddSKent Overstreet 		/*
7953737e0ddSKent Overstreet 		 * read_extent -> io_time_reset may cause a transaction restart
7963737e0ddSKent Overstreet 		 * without returning an error, we need to check for that here:
7973737e0ddSKent Overstreet 		 */
7983737e0ddSKent Overstreet 		if (!bch2_trans_relock(trans)) {
7993737e0ddSKent Overstreet 			ret = -EINTR;
8003737e0ddSKent Overstreet 			break;
8013737e0ddSKent Overstreet 		}
8023737e0ddSKent Overstreet 
80306ed8558SKent Overstreet 		bch2_btree_iter_set_pos(iter,
80406ed8558SKent Overstreet 				POS(inum, rbio->bio.bi_iter.bi_sector));
8051c6fdbd8SKent Overstreet 
8061c6fdbd8SKent Overstreet 		k = bch2_btree_iter_peek_slot(iter);
80776426098SKent Overstreet 		ret = bkey_err(k);
80876426098SKent Overstreet 		if (ret)
80976426098SKent Overstreet 			break;
8101c6fdbd8SKent Overstreet 
81106ed8558SKent Overstreet 		offset_into_extent = iter->pos.offset -
81206ed8558SKent Overstreet 			bkey_start_offset(k.k);
81376426098SKent Overstreet 		sectors = k.k->size - offset_into_extent;
81476426098SKent Overstreet 
81507a1006aSKent Overstreet 		bch2_bkey_buf_reassemble(&sk, c, k);
81613dcd4abSKent Overstreet 
8175ff75ccbSKent Overstreet 		ret = bch2_read_indirect_extent(trans, &data_btree,
81822d8a33dSYuxuan Shui 					&offset_into_extent, &sk);
81976426098SKent Overstreet 		if (ret)
82076426098SKent Overstreet 			break;
82176426098SKent Overstreet 
82213dcd4abSKent Overstreet 		k = bkey_i_to_s_c(sk.k);
82313dcd4abSKent Overstreet 
82476426098SKent Overstreet 		sectors = min(sectors, k.k->size - offset_into_extent);
82576426098SKent Overstreet 
82676426098SKent Overstreet 		bch2_trans_unlock(trans);
82706ed8558SKent Overstreet 
82835189e09SKent Overstreet 		if (readpages_iter)
82935189e09SKent Overstreet 			readpage_bio_extend(readpages_iter, &rbio->bio, sectors,
83035189e09SKent Overstreet 					    extent_partial_reads_expensive(k));
8311c6fdbd8SKent Overstreet 
83276426098SKent Overstreet 		bytes = min(sectors, bio_sectors(&rbio->bio)) << 9;
83306ed8558SKent Overstreet 		swap(rbio->bio.bi_iter.bi_size, bytes);
8341c6fdbd8SKent Overstreet 
83506ed8558SKent Overstreet 		if (rbio->bio.bi_iter.bi_size == bytes)
8361c6fdbd8SKent Overstreet 			flags |= BCH_READ_LAST_FRAGMENT;
8371c6fdbd8SKent Overstreet 
8381c6fdbd8SKent Overstreet 		if (bkey_extent_is_allocation(k.k))
83906ed8558SKent Overstreet 			bch2_add_page_sectors(&rbio->bio, k);
8401c6fdbd8SKent Overstreet 
8415ff75ccbSKent Overstreet 		bch2_read_extent(trans, rbio, iter->pos,
8425ff75ccbSKent Overstreet 				 data_btree, k, offset_into_extent, flags);
8431c6fdbd8SKent Overstreet 
8441c6fdbd8SKent Overstreet 		if (flags & BCH_READ_LAST_FRAGMENT)
84535189e09SKent Overstreet 			break;
8461c6fdbd8SKent Overstreet 
84706ed8558SKent Overstreet 		swap(rbio->bio.bi_iter.bi_size, bytes);
84806ed8558SKent Overstreet 		bio_advance(&rbio->bio, bytes);
8491c6fdbd8SKent Overstreet 	}
85076426098SKent Overstreet 
85176426098SKent Overstreet 	if (ret == -EINTR)
85276426098SKent Overstreet 		goto retry;
85376426098SKent Overstreet 
85435189e09SKent Overstreet 	if (ret) {
8550fefe8d8SKent Overstreet 		bch_err_inum_ratelimited(c, inum,
8560fefe8d8SKent Overstreet 				"read error %i from btree lookup", ret);
8570fefe8d8SKent Overstreet 		rbio->bio.bi_status = BLK_STS_IOERR;
85876426098SKent Overstreet 		bio_endio(&rbio->bio);
8591c6fdbd8SKent Overstreet 	}
8601c6fdbd8SKent Overstreet 
86107a1006aSKent Overstreet 	bch2_bkey_buf_exit(&sk, c);
86235189e09SKent Overstreet }
86335189e09SKent Overstreet 
8641c6fdbd8SKent Overstreet void bch2_readahead(struct readahead_control *ractl)
8651c6fdbd8SKent Overstreet {
8661c6fdbd8SKent Overstreet 	struct bch_inode_info *inode = to_bch_ei(ractl->mapping->host);
8671c6fdbd8SKent Overstreet 	struct bch_fs *c = inode->v.i_sb->s_fs_info;
8689a3df993SKent Overstreet 	struct bch_io_opts opts = io_opts(c, &inode->ei_inode);
869424eb881SKent Overstreet 	struct btree_trans trans;
870*67e0dd8fSKent Overstreet 	struct btree_iter iter;
8711c6fdbd8SKent Overstreet 	struct page *page;
8721c6fdbd8SKent Overstreet 	struct readpages_iter readpages_iter;
8731c6fdbd8SKent Overstreet 	int ret;
8741c6fdbd8SKent Overstreet 
8751c6fdbd8SKent Overstreet 	ret = readpages_iter_init(&readpages_iter, ractl);
8761c6fdbd8SKent Overstreet 	BUG_ON(ret);
8771c6fdbd8SKent Overstreet 
87820bceecbSKent Overstreet 	bch2_trans_init(&trans, c, 0, 0);
879*67e0dd8fSKent Overstreet 	bch2_trans_iter_init(&trans, &iter, BTREE_ID_extents, POS_MIN,
8801c6fdbd8SKent Overstreet 			     BTREE_ITER_SLOTS);
8811c6fdbd8SKent Overstreet 
8821c6fdbd8SKent Overstreet 	bch2_pagecache_add_get(&inode->ei_pagecache_lock);
8831c6fdbd8SKent Overstreet 
8841c6fdbd8SKent Overstreet 	while ((page = readpage_iter_next(&readpages_iter))) {
8851c6fdbd8SKent Overstreet 		pgoff_t index = readpages_iter.offset + readpages_iter.idx;
8861c6fdbd8SKent Overstreet 		unsigned n = min_t(unsigned,
8871c6fdbd8SKent Overstreet 				   readpages_iter.nr_pages -
8881c6fdbd8SKent Overstreet 				   readpages_iter.idx,
8891c6fdbd8SKent Overstreet 				   BIO_MAX_VECS);
8901c6fdbd8SKent Overstreet 		struct bch_read_bio *rbio =
8911c6fdbd8SKent Overstreet 			rbio_init(bio_alloc_bioset(NULL, n, REQ_OP_READ,
8921c6fdbd8SKent Overstreet 						   GFP_NOFS, &c->bio_read),
8931c6fdbd8SKent Overstreet 				  opts);
8941c6fdbd8SKent Overstreet 
8951c6fdbd8SKent Overstreet 		readpages_iter.idx++;
8961c6fdbd8SKent Overstreet 
8971c6fdbd8SKent Overstreet 		rbio->bio.bi_iter.bi_sector = (sector_t) index << PAGE_SECTOR_SHIFT;
8981c6fdbd8SKent Overstreet 		rbio->bio.bi_end_io = bch2_readpages_end_io;
899885678f6SKent Overstreet 		BUG_ON(!bio_add_page(&rbio->bio, page, PAGE_SIZE, 0));
9001c6fdbd8SKent Overstreet 
901*67e0dd8fSKent Overstreet 		bchfs_read(&trans, &iter, rbio, inode->v.i_ino,
9020f238367SKent Overstreet 			   &readpages_iter);
9031c6fdbd8SKent Overstreet 	}
9041c6fdbd8SKent Overstreet 
9051c6fdbd8SKent Overstreet 	bch2_pagecache_add_put(&inode->ei_pagecache_lock);
906424eb881SKent Overstreet 
907*67e0dd8fSKent Overstreet 	bch2_trans_iter_exit(&trans, &iter);
908424eb881SKent Overstreet 	bch2_trans_exit(&trans);
9091c6fdbd8SKent Overstreet 	kfree(readpages_iter.pages);
9101c6fdbd8SKent Overstreet }
9111c6fdbd8SKent Overstreet 
9121c6fdbd8SKent Overstreet static void __bchfs_readpage(struct bch_fs *c, struct bch_read_bio *rbio,
9131c6fdbd8SKent Overstreet 			     u64 inum, struct page *page)
9141c6fdbd8SKent Overstreet {
915424eb881SKent Overstreet 	struct btree_trans trans;
916*67e0dd8fSKent Overstreet 	struct btree_iter iter;
9171c6fdbd8SKent Overstreet 
918f57a6a5dSKent Overstreet 	bch2_page_state_create(page, __GFP_NOFAIL);
9191c6fdbd8SKent Overstreet 
9201c6fdbd8SKent Overstreet 	rbio->bio.bi_opf = REQ_OP_READ|REQ_SYNC;
9217f5e31e1SKent Overstreet 	rbio->bio.bi_iter.bi_sector =
9227f5e31e1SKent Overstreet 		(sector_t) page->index << PAGE_SECTOR_SHIFT;
9237f5e31e1SKent Overstreet 	BUG_ON(!bio_add_page(&rbio->bio, page, PAGE_SIZE, 0));
9241c6fdbd8SKent Overstreet 
92520bceecbSKent Overstreet 	bch2_trans_init(&trans, c, 0, 0);
926*67e0dd8fSKent Overstreet 	bch2_trans_iter_init(&trans, &iter, BTREE_ID_extents, POS_MIN,
9271c6fdbd8SKent Overstreet 			     BTREE_ITER_SLOTS);
928424eb881SKent Overstreet 
929*67e0dd8fSKent Overstreet 	bchfs_read(&trans, &iter, rbio, inum, NULL);
930424eb881SKent Overstreet 
931*67e0dd8fSKent Overstreet 	bch2_trans_iter_exit(&trans, &iter);
932424eb881SKent Overstreet 	bch2_trans_exit(&trans);
9331c6fdbd8SKent Overstreet }
9341c6fdbd8SKent Overstreet 
9351c6fdbd8SKent Overstreet static void bch2_read_single_page_end_io(struct bio *bio)
9361c6fdbd8SKent Overstreet {
9371c6fdbd8SKent Overstreet 	complete(bio->bi_private);
9381c6fdbd8SKent Overstreet }
9391c6fdbd8SKent Overstreet 
9401c6fdbd8SKent Overstreet static int bch2_read_single_page(struct page *page,
9411c6fdbd8SKent Overstreet 				 struct address_space *mapping)
9421c6fdbd8SKent Overstreet {
9431c6fdbd8SKent Overstreet 	struct bch_inode_info *inode = to_bch_ei(mapping->host);
9441c6fdbd8SKent Overstreet 	struct bch_fs *c = inode->v.i_sb->s_fs_info;
9451c6fdbd8SKent Overstreet 	struct bch_read_bio *rbio;
9461c6fdbd8SKent Overstreet 	int ret;
9471c6fdbd8SKent Overstreet 	DECLARE_COMPLETION_ONSTACK(done);
9481c6fdbd8SKent Overstreet 
9491c6fdbd8SKent Overstreet 	rbio = rbio_init(bio_alloc_bioset(NULL, 1, REQ_OP_READ, GFP_NOFS, &c->bio_read),
9509a3df993SKent Overstreet 			 io_opts(c, &inode->ei_inode));
9511c6fdbd8SKent Overstreet 	rbio->bio.bi_private = &done;
9521c6fdbd8SKent Overstreet 	rbio->bio.bi_end_io = bch2_read_single_page_end_io;
9531c6fdbd8SKent Overstreet 
9541c6fdbd8SKent Overstreet 	__bchfs_readpage(c, rbio, inode->v.i_ino, page);
9551c6fdbd8SKent Overstreet 	wait_for_completion(&done);
9561c6fdbd8SKent Overstreet 
9571c6fdbd8SKent Overstreet 	ret = blk_status_to_errno(rbio->bio.bi_status);
9581c6fdbd8SKent Overstreet 	bio_put(&rbio->bio);
9591c6fdbd8SKent Overstreet 
9601c6fdbd8SKent Overstreet 	if (ret < 0)
9611c6fdbd8SKent Overstreet 		return ret;
9621c6fdbd8SKent Overstreet 
9631c6fdbd8SKent Overstreet 	SetPageUptodate(page);
9641c6fdbd8SKent Overstreet 	return 0;
9651c6fdbd8SKent Overstreet }
9661c6fdbd8SKent Overstreet 
9671c6fdbd8SKent Overstreet int bch2_read_folio(struct file *file, struct folio *folio)
9681c6fdbd8SKent Overstreet {
9691c6fdbd8SKent Overstreet 	struct page *page = &folio->page;
9701c6fdbd8SKent Overstreet 	int ret;
9711c6fdbd8SKent Overstreet 
9721c6fdbd8SKent Overstreet 	ret = bch2_read_single_page(page, page->mapping);
9731c6fdbd8SKent Overstreet 	folio_unlock(folio);
9741c6fdbd8SKent Overstreet 	return ret;
9751c6fdbd8SKent Overstreet }
9761c6fdbd8SKent Overstreet 
9771c6fdbd8SKent Overstreet /* writepages: */
9781c6fdbd8SKent Overstreet 
9791c6fdbd8SKent Overstreet struct bch_writepage_state {
9801c6fdbd8SKent Overstreet 	struct bch_writepage_io	*io;
9811c6fdbd8SKent Overstreet 	struct bch_io_opts	opts;
9821c6fdbd8SKent Overstreet };
9831c6fdbd8SKent Overstreet 
9841c6fdbd8SKent Overstreet static inline struct bch_writepage_state bch_writepage_state_init(struct bch_fs *c,
9851c6fdbd8SKent Overstreet 								  struct bch_inode_info *inode)
9861c6fdbd8SKent Overstreet {
9879a3df993SKent Overstreet 	return (struct bch_writepage_state) {
9889a3df993SKent Overstreet 		.opts = io_opts(c, &inode->ei_inode)
9899a3df993SKent Overstreet 	};
9901c6fdbd8SKent Overstreet }
9911c6fdbd8SKent Overstreet 
9929f311f21SKent Overstreet static void bch2_writepage_io_done(struct bch_write_op *op)
9931c6fdbd8SKent Overstreet {
9949f311f21SKent Overstreet 	struct bch_writepage_io *io =
9959f311f21SKent Overstreet 		container_of(op, struct bch_writepage_io, op);
9969a3df993SKent Overstreet 	struct bch_fs *c = io->op.c;
9979a3df993SKent Overstreet 	struct bio *bio = &io->op.wbio.bio;
9981c6fdbd8SKent Overstreet 	struct bvec_iter_all iter;
9991c6fdbd8SKent Overstreet 	struct bio_vec *bvec;
1000b3fce09cSKent Overstreet 	unsigned i;
10011c6fdbd8SKent Overstreet 
1002ef1b2092SKent Overstreet 	up(&io->op.c->io_in_flight);
1003ef1b2092SKent Overstreet 
10049a3df993SKent Overstreet 	if (io->op.error) {
100533c74e41SKent Overstreet 		set_bit(EI_INODE_ERROR, &io->inode->ei_flags);
100633c74e41SKent Overstreet 
100775812e70SKent Overstreet 		bio_for_each_segment_all(bvec, bio, iter) {
1008b3fce09cSKent Overstreet 			struct bch_page_state *s;
1009b3fce09cSKent Overstreet 
10101c6fdbd8SKent Overstreet 			SetPageError(bvec->bv_page);
101175812e70SKent Overstreet 			mapping_set_error(bvec->bv_page->mapping, -EIO);
1012b3fce09cSKent Overstreet 
10133826ee0bSKent Overstreet 			s = __bch2_page_state(bvec->bv_page);
10143826ee0bSKent Overstreet 			spin_lock(&s->lock);
1015b3fce09cSKent Overstreet 			for (i = 0; i < PAGE_SECTORS; i++)
1016b3fce09cSKent Overstreet 				s->s[i].nr_replicas = 0;
10173826ee0bSKent Overstreet 			spin_unlock(&s->lock);
101875812e70SKent Overstreet 		}
10191c6fdbd8SKent Overstreet 	}
10201c6fdbd8SKent Overstreet 
10214be1a412SKent Overstreet 	if (io->op.flags & BCH_WRITE_WROTE_DATA_INLINE) {
10224be1a412SKent Overstreet 		bio_for_each_segment_all(bvec, bio, iter) {
10234be1a412SKent Overstreet 			struct bch_page_state *s;
10244be1a412SKent Overstreet 
10254be1a412SKent Overstreet 			s = __bch2_page_state(bvec->bv_page);
10264be1a412SKent Overstreet 			spin_lock(&s->lock);
10274be1a412SKent Overstreet 			for (i = 0; i < PAGE_SECTORS; i++)
10284be1a412SKent Overstreet 				s->s[i].nr_replicas = 0;
10294be1a412SKent Overstreet 			spin_unlock(&s->lock);
10304be1a412SKent Overstreet 		}
10314be1a412SKent Overstreet 	}
10324be1a412SKent Overstreet 
10331c6fdbd8SKent Overstreet 	/*
10341c6fdbd8SKent Overstreet 	 * racing with fallocate can cause us to add fewer sectors than
10351c6fdbd8SKent Overstreet 	 * expected - but we shouldn't add more sectors than expected:
10361c6fdbd8SKent Overstreet 	 */
10379a3df993SKent Overstreet 	BUG_ON(io->op.i_sectors_delta > 0);
10381c6fdbd8SKent Overstreet 
10391c6fdbd8SKent Overstreet 	/*
10401c6fdbd8SKent Overstreet 	 * (error (due to going RO) halfway through a page can screw that up
10411c6fdbd8SKent Overstreet 	 * slightly)
10421c6fdbd8SKent Overstreet 	 * XXX wtf?
10439a3df993SKent Overstreet 	   BUG_ON(io->op.op.i_sectors_delta >= PAGE_SECTORS);
10441c6fdbd8SKent Overstreet 	 */
10451c6fdbd8SKent Overstreet 
10461c6fdbd8SKent Overstreet 	/*
10471c6fdbd8SKent Overstreet 	 * PageWriteback is effectively our ref on the inode - fixup i_blocks
10481c6fdbd8SKent Overstreet 	 * before calling end_page_writeback:
10491c6fdbd8SKent Overstreet 	 */
10509a3df993SKent Overstreet 	i_sectors_acct(c, io->inode, NULL, io->op.i_sectors_delta);
10511c6fdbd8SKent Overstreet 
10527f5e31e1SKent Overstreet 	bio_for_each_segment_all(bvec, bio, iter) {
10537f5e31e1SKent Overstreet 		struct bch_page_state *s = __bch2_page_state(bvec->bv_page);
10547f5e31e1SKent Overstreet 
10557f5e31e1SKent Overstreet 		if (atomic_dec_and_test(&s->write_count))
10561c6fdbd8SKent Overstreet 			end_page_writeback(bvec->bv_page);
10577f5e31e1SKent Overstreet 	}
10581c6fdbd8SKent Overstreet 
10599f311f21SKent Overstreet 	bio_put(&io->op.wbio.bio);
10601c6fdbd8SKent Overstreet }
10611c6fdbd8SKent Overstreet 
10621c6fdbd8SKent Overstreet static void bch2_writepage_do_io(struct bch_writepage_state *w)
10631c6fdbd8SKent Overstreet {
10641c6fdbd8SKent Overstreet 	struct bch_writepage_io *io = w->io;
10651c6fdbd8SKent Overstreet 
1066ef1b2092SKent Overstreet 	down(&io->op.c->io_in_flight);
1067ef1b2092SKent Overstreet 
10681c6fdbd8SKent Overstreet 	w->io = NULL;
10699f311f21SKent Overstreet 	closure_call(&io->op.cl, bch2_write, NULL, NULL);
10701c6fdbd8SKent Overstreet }
10711c6fdbd8SKent Overstreet 
10721c6fdbd8SKent Overstreet /*
10731c6fdbd8SKent Overstreet  * Get a bch_writepage_io and add @page to it - appending to an existing one if
10741c6fdbd8SKent Overstreet  * possible, else allocating a new one:
10751c6fdbd8SKent Overstreet  */
10761c6fdbd8SKent Overstreet static void bch2_writepage_io_alloc(struct bch_fs *c,
107750fe5bd6SKent Overstreet 				    struct writeback_control *wbc,
10781c6fdbd8SKent Overstreet 				    struct bch_writepage_state *w,
10791c6fdbd8SKent Overstreet 				    struct bch_inode_info *inode,
10807f5e31e1SKent Overstreet 				    u64 sector,
10811c6fdbd8SKent Overstreet 				    unsigned nr_replicas)
10821c6fdbd8SKent Overstreet {
10831c6fdbd8SKent Overstreet 	struct bch_write_op *op;
10841c6fdbd8SKent Overstreet 
10851c6fdbd8SKent Overstreet 	w->io = container_of(bio_alloc_bioset(NULL, BIO_MAX_VECS,
10861c6fdbd8SKent Overstreet 					      REQ_OP_WRITE,
10871c6fdbd8SKent Overstreet 					      GFP_NOFS,
10881c6fdbd8SKent Overstreet 					      &c->writepage_bioset),
10899a3df993SKent Overstreet 			     struct bch_writepage_io, op.wbio.bio);
10901c6fdbd8SKent Overstreet 
10919a3df993SKent Overstreet 	w->io->inode		= inode;
10929a3df993SKent Overstreet 	op			= &w->io->op;
10939a3df993SKent Overstreet 	bch2_write_op_init(op, c, w->opts);
10949a3df993SKent Overstreet 	op->target		= w->opts.foreground_target;
10959a3df993SKent Overstreet 	op_journal_seq_set(op, &inode->ei_journal_seq);
10961c6fdbd8SKent Overstreet 	op->nr_replicas		= nr_replicas;
10971c6fdbd8SKent Overstreet 	op->res.nr_replicas	= nr_replicas;
10981c6fdbd8SKent Overstreet 	op->write_point		= writepoint_hashed(inode->ei_last_dirtied);
10997f5e31e1SKent Overstreet 	op->pos			= POS(inode->v.i_ino, sector);
11009f311f21SKent Overstreet 	op->end_io		= bch2_writepage_io_done;
11017f5e31e1SKent Overstreet 	op->wbio.bio.bi_iter.bi_sector = sector;
110250fe5bd6SKent Overstreet 	op->wbio.bio.bi_opf	= wbc_to_write_flags(wbc);
11031c6fdbd8SKent Overstreet }
11041c6fdbd8SKent Overstreet 
11051c6fdbd8SKent Overstreet static int __bch2_writepage(struct folio *folio,
11061c6fdbd8SKent Overstreet 			    struct writeback_control *wbc,
11071c6fdbd8SKent Overstreet 			    void *data)
11081c6fdbd8SKent Overstreet {
11091c6fdbd8SKent Overstreet 	struct page *page = &folio->page;
11101c6fdbd8SKent Overstreet 	struct bch_inode_info *inode = to_bch_ei(page->mapping->host);
11111c6fdbd8SKent Overstreet 	struct bch_fs *c = inode->v.i_sb->s_fs_info;
11121c6fdbd8SKent Overstreet 	struct bch_writepage_state *w = data;
11137f5e31e1SKent Overstreet 	struct bch_page_state *s, orig;
11147f5e31e1SKent Overstreet 	unsigned i, offset, nr_replicas_this_write = U32_MAX;
11151c6fdbd8SKent Overstreet 	loff_t i_size = i_size_read(&inode->v);
11161c6fdbd8SKent Overstreet 	pgoff_t end_index = i_size >> PAGE_SHIFT;
1117e1036a2aSKent Overstreet 	int ret;
11181c6fdbd8SKent Overstreet 
11191c6fdbd8SKent Overstreet 	EBUG_ON(!PageUptodate(page));
11201c6fdbd8SKent Overstreet 
11211c6fdbd8SKent Overstreet 	/* Is the page fully inside i_size? */
11221c6fdbd8SKent Overstreet 	if (page->index < end_index)
11231c6fdbd8SKent Overstreet 		goto do_io;
11241c6fdbd8SKent Overstreet 
11251c6fdbd8SKent Overstreet 	/* Is the page fully outside i_size? (truncate in progress) */
11261c6fdbd8SKent Overstreet 	offset = i_size & (PAGE_SIZE - 1);
11271c6fdbd8SKent Overstreet 	if (page->index > end_index || !offset) {
11281c6fdbd8SKent Overstreet 		unlock_page(page);
11291c6fdbd8SKent Overstreet 		return 0;
11301c6fdbd8SKent Overstreet 	}
11311c6fdbd8SKent Overstreet 
11321c6fdbd8SKent Overstreet 	/*
11331c6fdbd8SKent Overstreet 	 * The page straddles i_size.  It must be zeroed out on each and every
11341c6fdbd8SKent Overstreet 	 * writepage invocation because it may be mmapped.  "A file is mapped
11351c6fdbd8SKent Overstreet 	 * in multiples of the page size.  For a file that is not a multiple of
11361c6fdbd8SKent Overstreet 	 * the  page size, the remaining memory is zeroed when mapped, and
11371c6fdbd8SKent Overstreet 	 * writes to that region are not written out to the file."
11381c6fdbd8SKent Overstreet 	 */
11391c6fdbd8SKent Overstreet 	zero_user_segment(page, offset, PAGE_SIZE);
11401c6fdbd8SKent Overstreet do_io:
1141f57a6a5dSKent Overstreet 	s = bch2_page_state_create(page, __GFP_NOFAIL);
1142f81b648dSKent Overstreet 
1143f57a6a5dSKent Overstreet 	ret = bch2_get_page_disk_reservation(c, inode, page, true);
1144e1036a2aSKent Overstreet 	if (ret) {
1145e1036a2aSKent Overstreet 		SetPageError(page);
1146e1036a2aSKent Overstreet 		mapping_set_error(page->mapping, ret);
1147e1036a2aSKent Overstreet 		unlock_page(page);
1148e1036a2aSKent Overstreet 		return 0;
1149e1036a2aSKent Overstreet 	}
1150e1036a2aSKent Overstreet 
11517f5e31e1SKent Overstreet 	/* Before unlocking the page, get copy of reservations: */
11527f5e31e1SKent Overstreet 	orig = *s;
11537f5e31e1SKent Overstreet 
11547f5e31e1SKent Overstreet 	for (i = 0; i < PAGE_SECTORS; i++) {
11552ba5d38bSKent Overstreet 		if (s->s[i].state < SECTOR_DIRTY)
11567f5e31e1SKent Overstreet 			continue;
11577f5e31e1SKent Overstreet 
1158f81b648dSKent Overstreet 		nr_replicas_this_write =
1159f57a6a5dSKent Overstreet 			min_t(unsigned, nr_replicas_this_write,
1160f57a6a5dSKent Overstreet 			      s->s[i].nr_replicas +
1161f57a6a5dSKent Overstreet 			      s->s[i].replicas_reserved);
11627f5e31e1SKent Overstreet 	}
11631c6fdbd8SKent Overstreet 
1164f57a6a5dSKent Overstreet 	for (i = 0; i < PAGE_SECTORS; i++) {
11652ba5d38bSKent Overstreet 		if (s->s[i].state < SECTOR_DIRTY)
11667f5e31e1SKent Overstreet 			continue;
11677f5e31e1SKent Overstreet 
1168f57a6a5dSKent Overstreet 		s->s[i].nr_replicas = w->opts.compression
1169f57a6a5dSKent Overstreet 			? 0 : nr_replicas_this_write;
1170e1036a2aSKent Overstreet 
1171f57a6a5dSKent Overstreet 		s->s[i].replicas_reserved = 0;
1172f57a6a5dSKent Overstreet 		s->s[i].state = SECTOR_ALLOCATED;
1173f57a6a5dSKent Overstreet 	}
11741c6fdbd8SKent Overstreet 
11757f5e31e1SKent Overstreet 	BUG_ON(atomic_read(&s->write_count));
11767f5e31e1SKent Overstreet 	atomic_set(&s->write_count, 1);
11777f5e31e1SKent Overstreet 
11781c6fdbd8SKent Overstreet 	BUG_ON(PageWriteback(page));
11791c6fdbd8SKent Overstreet 	set_page_writeback(page);
11807f5e31e1SKent Overstreet 
11811c6fdbd8SKent Overstreet 	unlock_page(page);
11821c6fdbd8SKent Overstreet 
11837f5e31e1SKent Overstreet 	offset = 0;
11847f5e31e1SKent Overstreet 	while (1) {
11857f5e31e1SKent Overstreet 		unsigned sectors = 1, dirty_sectors = 0, reserved_sectors = 0;
11867f5e31e1SKent Overstreet 		u64 sector;
11877f5e31e1SKent Overstreet 
11887f5e31e1SKent Overstreet 		while (offset < PAGE_SECTORS &&
11892ba5d38bSKent Overstreet 		       orig.s[offset].state < SECTOR_DIRTY)
11907f5e31e1SKent Overstreet 			offset++;
11917f5e31e1SKent Overstreet 
11927f5e31e1SKent Overstreet 		if (offset == PAGE_SECTORS)
11937f5e31e1SKent Overstreet 			break;
11947f5e31e1SKent Overstreet 
11957f5e31e1SKent Overstreet 		sector = ((u64) page->index << PAGE_SECTOR_SHIFT) + offset;
11967f5e31e1SKent Overstreet 
11977f5e31e1SKent Overstreet 		while (offset + sectors < PAGE_SECTORS &&
11982ba5d38bSKent Overstreet 		       orig.s[offset + sectors].state >= SECTOR_DIRTY)
11997f5e31e1SKent Overstreet 			sectors++;
12007f5e31e1SKent Overstreet 
12017f5e31e1SKent Overstreet 		for (i = offset; i < offset + sectors; i++) {
12027f5e31e1SKent Overstreet 			reserved_sectors += orig.s[i].replicas_reserved;
12037f5e31e1SKent Overstreet 			dirty_sectors += orig.s[i].state == SECTOR_DIRTY;
12047f5e31e1SKent Overstreet 		}
12057f5e31e1SKent Overstreet 
12061c6fdbd8SKent Overstreet 		if (w->io &&
12079a3df993SKent Overstreet 		    (w->io->op.res.nr_replicas != nr_replicas_this_write ||
12089a3df993SKent Overstreet 		     bio_full(&w->io->op.wbio.bio, PAGE_SIZE) ||
1209f59b3464SKent Overstreet 		     w->io->op.wbio.bio.bi_iter.bi_size + (sectors << 9) >=
1210f59b3464SKent Overstreet 		     (BIO_MAX_VECS * PAGE_SIZE) ||
12119a3df993SKent Overstreet 		     bio_end_sector(&w->io->op.wbio.bio) != sector))
12121c6fdbd8SKent Overstreet 			bch2_writepage_do_io(w);
12131c6fdbd8SKent Overstreet 
12141c6fdbd8SKent Overstreet 		if (!w->io)
121550fe5bd6SKent Overstreet 			bch2_writepage_io_alloc(c, wbc, w, inode, sector,
1216f81b648dSKent Overstreet 						nr_replicas_this_write);
12171c6fdbd8SKent Overstreet 
12187f5e31e1SKent Overstreet 		atomic_inc(&s->write_count);
12197f5e31e1SKent Overstreet 
12209a3df993SKent Overstreet 		BUG_ON(inode != w->io->inode);
12219a3df993SKent Overstreet 		BUG_ON(!bio_add_page(&w->io->op.wbio.bio, page,
12227f5e31e1SKent Overstreet 				     sectors << 9, offset << 9));
12231c6fdbd8SKent Overstreet 
12246cc3535dSKent Overstreet 		/* Check for writing past i_size: */
12259a3df993SKent Overstreet 		WARN_ON((bio_end_sector(&w->io->op.wbio.bio) << 9) >
12266cc3535dSKent Overstreet 			round_up(i_size, block_bytes(c)));
12276cc3535dSKent Overstreet 
12289a3df993SKent Overstreet 		w->io->op.res.sectors += reserved_sectors;
12299a3df993SKent Overstreet 		w->io->op.i_sectors_delta -= dirty_sectors;
12301c6fdbd8SKent Overstreet 		w->io->op.new_i_size = i_size;
12311c6fdbd8SKent Overstreet 
12327f5e31e1SKent Overstreet 		offset += sectors;
12337f5e31e1SKent Overstreet 	}
12347f5e31e1SKent Overstreet 
12357f5e31e1SKent Overstreet 	if (atomic_dec_and_test(&s->write_count))
12367f5e31e1SKent Overstreet 		end_page_writeback(page);
12377f5e31e1SKent Overstreet 
12381c6fdbd8SKent Overstreet 	return 0;
12391c6fdbd8SKent Overstreet }
12401c6fdbd8SKent Overstreet 
12411c6fdbd8SKent Overstreet int bch2_writepages(struct address_space *mapping, struct writeback_control *wbc)
12421c6fdbd8SKent Overstreet {
12431c6fdbd8SKent Overstreet 	struct bch_fs *c = mapping->host->i_sb->s_fs_info;
12441c6fdbd8SKent Overstreet 	struct bch_writepage_state w =
12451c6fdbd8SKent Overstreet 		bch_writepage_state_init(c, to_bch_ei(mapping->host));
12461c6fdbd8SKent Overstreet 	struct blk_plug plug;
12471c6fdbd8SKent Overstreet 	int ret;
12481c6fdbd8SKent Overstreet 
12491c6fdbd8SKent Overstreet 	blk_start_plug(&plug);
12501c6fdbd8SKent Overstreet 	ret = write_cache_pages(mapping, wbc, __bch2_writepage, &w);
12511c6fdbd8SKent Overstreet 	if (w.io)
12521c6fdbd8SKent Overstreet 		bch2_writepage_do_io(&w);
12531c6fdbd8SKent Overstreet 	blk_finish_plug(&plug);
12541c6fdbd8SKent Overstreet 	return ret;
12551c6fdbd8SKent Overstreet }
12561c6fdbd8SKent Overstreet 
12571c6fdbd8SKent Overstreet int bch2_writepage(struct page *page, struct writeback_control *wbc)
12581c6fdbd8SKent Overstreet {
12591c6fdbd8SKent Overstreet 	struct bch_fs *c = page->mapping->host->i_sb->s_fs_info;
12601c6fdbd8SKent Overstreet 	struct bch_writepage_state w =
12611c6fdbd8SKent Overstreet 		bch_writepage_state_init(c, to_bch_ei(page->mapping->host));
12621c6fdbd8SKent Overstreet 	int ret;
12631c6fdbd8SKent Overstreet 
12641c6fdbd8SKent Overstreet 	ret = __bch2_writepage(page_folio(page), wbc, &w);
12651c6fdbd8SKent Overstreet 	if (w.io)
12661c6fdbd8SKent Overstreet 		bch2_writepage_do_io(&w);
12671c6fdbd8SKent Overstreet 
12681c6fdbd8SKent Overstreet 	return ret;
12691c6fdbd8SKent Overstreet }
12701c6fdbd8SKent Overstreet 
12711c6fdbd8SKent Overstreet /* buffered writes: */
12721c6fdbd8SKent Overstreet 
12731c6fdbd8SKent Overstreet int bch2_write_begin(struct file *file, struct address_space *mapping,
12741c6fdbd8SKent Overstreet 		     loff_t pos, unsigned len,
12751c6fdbd8SKent Overstreet 		     struct page **pagep, void **fsdata)
12761c6fdbd8SKent Overstreet {
12771c6fdbd8SKent Overstreet 	struct bch_inode_info *inode = to_bch_ei(mapping->host);
12781c6fdbd8SKent Overstreet 	struct bch_fs *c = inode->v.i_sb->s_fs_info;
1279d1542e03SKent Overstreet 	struct bch2_page_reservation *res;
12801c6fdbd8SKent Overstreet 	pgoff_t index = pos >> PAGE_SHIFT;
12811c6fdbd8SKent Overstreet 	unsigned offset = pos & (PAGE_SIZE - 1);
12821c6fdbd8SKent Overstreet 	struct page *page;
12831c6fdbd8SKent Overstreet 	int ret = -ENOMEM;
12841c6fdbd8SKent Overstreet 
1285d1542e03SKent Overstreet 	res = kmalloc(sizeof(*res), GFP_KERNEL);
1286d1542e03SKent Overstreet 	if (!res)
1287d1542e03SKent Overstreet 		return -ENOMEM;
1288d1542e03SKent Overstreet 
1289d1542e03SKent Overstreet 	bch2_page_reservation_init(c, inode, res);
1290d1542e03SKent Overstreet 	*fsdata = res;
12911c6fdbd8SKent Overstreet 
12921c6fdbd8SKent Overstreet 	bch2_pagecache_add_get(&inode->ei_pagecache_lock);
12931c6fdbd8SKent Overstreet 
12941c6fdbd8SKent Overstreet 	page = grab_cache_page_write_begin(mapping, index);
12951c6fdbd8SKent Overstreet 	if (!page)
12961c6fdbd8SKent Overstreet 		goto err_unlock;
12971c6fdbd8SKent Overstreet 
12981c6fdbd8SKent Overstreet 	if (PageUptodate(page))
12991c6fdbd8SKent Overstreet 		goto out;
13001c6fdbd8SKent Overstreet 
13011c6fdbd8SKent Overstreet 	/* If we're writing entire page, don't need to read it in first: */
13021c6fdbd8SKent Overstreet 	if (len == PAGE_SIZE)
13031c6fdbd8SKent Overstreet 		goto out;
13041c6fdbd8SKent Overstreet 
13051c6fdbd8SKent Overstreet 	if (!offset && pos + len >= inode->v.i_size) {
13061c6fdbd8SKent Overstreet 		zero_user_segment(page, len, PAGE_SIZE);
13071c6fdbd8SKent Overstreet 		flush_dcache_page(page);
13081c6fdbd8SKent Overstreet 		goto out;
13091c6fdbd8SKent Overstreet 	}
13101c6fdbd8SKent Overstreet 
13111c6fdbd8SKent Overstreet 	if (index > inode->v.i_size >> PAGE_SHIFT) {
13121c6fdbd8SKent Overstreet 		zero_user_segments(page, 0, offset, offset + len, PAGE_SIZE);
13131c6fdbd8SKent Overstreet 		flush_dcache_page(page);
13141c6fdbd8SKent Overstreet 		goto out;
13151c6fdbd8SKent Overstreet 	}
13161c6fdbd8SKent Overstreet readpage:
13171c6fdbd8SKent Overstreet 	ret = bch2_read_single_page(page, mapping);
13181c6fdbd8SKent Overstreet 	if (ret)
13191c6fdbd8SKent Overstreet 		goto err;
13201c6fdbd8SKent Overstreet out:
1321d1542e03SKent Overstreet 	ret = bch2_page_reservation_get(c, inode, page, res,
1322d1542e03SKent Overstreet 					offset, len, true);
13231c6fdbd8SKent Overstreet 	if (ret) {
13241c6fdbd8SKent Overstreet 		if (!PageUptodate(page)) {
13251c6fdbd8SKent Overstreet 			/*
13261c6fdbd8SKent Overstreet 			 * If the page hasn't been read in, we won't know if we
13271c6fdbd8SKent Overstreet 			 * actually need a reservation - we don't actually need
13281c6fdbd8SKent Overstreet 			 * to read here, we just need to check if the page is
13291c6fdbd8SKent Overstreet 			 * fully backed by uncompressed data:
13301c6fdbd8SKent Overstreet 			 */
13311c6fdbd8SKent Overstreet 			goto readpage;
13321c6fdbd8SKent Overstreet 		}
13331c6fdbd8SKent Overstreet 
13341c6fdbd8SKent Overstreet 		goto err;
13351c6fdbd8SKent Overstreet 	}
13361c6fdbd8SKent Overstreet 
13371c6fdbd8SKent Overstreet 	*pagep = page;
13381c6fdbd8SKent Overstreet 	return 0;
13391c6fdbd8SKent Overstreet err:
13401c6fdbd8SKent Overstreet 	unlock_page(page);
13411c6fdbd8SKent Overstreet 	put_page(page);
13421c6fdbd8SKent Overstreet 	*pagep = NULL;
13431c6fdbd8SKent Overstreet err_unlock:
13441c6fdbd8SKent Overstreet 	bch2_pagecache_add_put(&inode->ei_pagecache_lock);
1345d1542e03SKent Overstreet 	kfree(res);
1346d1542e03SKent Overstreet 	*fsdata = NULL;
13471c6fdbd8SKent Overstreet 	return ret;
13481c6fdbd8SKent Overstreet }
13491c6fdbd8SKent Overstreet 
13501c6fdbd8SKent Overstreet int bch2_write_end(struct file *file, struct address_space *mapping,
13511c6fdbd8SKent Overstreet 		   loff_t pos, unsigned len, unsigned copied,
13521c6fdbd8SKent Overstreet 		   struct page *page, void *fsdata)
13531c6fdbd8SKent Overstreet {
13541c6fdbd8SKent Overstreet 	struct bch_inode_info *inode = to_bch_ei(mapping->host);
13551c6fdbd8SKent Overstreet 	struct bch_fs *c = inode->v.i_sb->s_fs_info;
1356d1542e03SKent Overstreet 	struct bch2_page_reservation *res = fsdata;
1357d1542e03SKent Overstreet 	unsigned offset = pos & (PAGE_SIZE - 1);
13581c6fdbd8SKent Overstreet 
13591c6fdbd8SKent Overstreet 	lockdep_assert_held(&inode->v.i_rwsem);
13601c6fdbd8SKent Overstreet 
13611c6fdbd8SKent Overstreet 	if (unlikely(copied < len && !PageUptodate(page))) {
13621c6fdbd8SKent Overstreet 		/*
13631c6fdbd8SKent Overstreet 		 * The page needs to be read in, but that would destroy
13641c6fdbd8SKent Overstreet 		 * our partial write - simplest thing is to just force
13651c6fdbd8SKent Overstreet 		 * userspace to redo the write:
13661c6fdbd8SKent Overstreet 		 */
13671c6fdbd8SKent Overstreet 		zero_user(page, 0, PAGE_SIZE);
13681c6fdbd8SKent Overstreet 		flush_dcache_page(page);
13691c6fdbd8SKent Overstreet 		copied = 0;
13701c6fdbd8SKent Overstreet 	}
13711c6fdbd8SKent Overstreet 
13721c6fdbd8SKent Overstreet 	spin_lock(&inode->v.i_lock);
13731c6fdbd8SKent Overstreet 	if (pos + copied > inode->v.i_size)
13741c6fdbd8SKent Overstreet 		i_size_write(&inode->v, pos + copied);
13751c6fdbd8SKent Overstreet 	spin_unlock(&inode->v.i_lock);
13761c6fdbd8SKent Overstreet 
13771c6fdbd8SKent Overstreet 	if (copied) {
13781c6fdbd8SKent Overstreet 		if (!PageUptodate(page))
13791c6fdbd8SKent Overstreet 			SetPageUptodate(page);
1380d1542e03SKent Overstreet 
1381d1542e03SKent Overstreet 		bch2_set_page_dirty(c, inode, page, res, offset, copied);
13821c6fdbd8SKent Overstreet 
13831c6fdbd8SKent Overstreet 		inode->ei_last_dirtied = (unsigned long) current;
13841c6fdbd8SKent Overstreet 	}
13851c6fdbd8SKent Overstreet 
13861c6fdbd8SKent Overstreet 	unlock_page(page);
13871c6fdbd8SKent Overstreet 	put_page(page);
13881c6fdbd8SKent Overstreet 	bch2_pagecache_add_put(&inode->ei_pagecache_lock);
13891c6fdbd8SKent Overstreet 
1390d1542e03SKent Overstreet 	bch2_page_reservation_put(c, inode, res);
1391d1542e03SKent Overstreet 	kfree(res);
1392d1542e03SKent Overstreet 
13931c6fdbd8SKent Overstreet 	return copied;
13941c6fdbd8SKent Overstreet }
13951c6fdbd8SKent Overstreet 
13961c6fdbd8SKent Overstreet #define WRITE_BATCH_PAGES	32
13971c6fdbd8SKent Overstreet 
13981c6fdbd8SKent Overstreet static int __bch2_buffered_write(struct bch_inode_info *inode,
13991c6fdbd8SKent Overstreet 				 struct address_space *mapping,
14001c6fdbd8SKent Overstreet 				 struct iov_iter *iter,
14011c6fdbd8SKent Overstreet 				 loff_t pos, unsigned len)
14021c6fdbd8SKent Overstreet {
14031c6fdbd8SKent Overstreet 	struct bch_fs *c = inode->v.i_sb->s_fs_info;
14041c6fdbd8SKent Overstreet 	struct page *pages[WRITE_BATCH_PAGES];
1405d1542e03SKent Overstreet 	struct bch2_page_reservation res;
14061c6fdbd8SKent Overstreet 	unsigned long index = pos >> PAGE_SHIFT;
14071c6fdbd8SKent Overstreet 	unsigned offset = pos & (PAGE_SIZE - 1);
14081c6fdbd8SKent Overstreet 	unsigned nr_pages = DIV_ROUND_UP(offset + len, PAGE_SIZE);
1409d1542e03SKent Overstreet 	unsigned i, reserved = 0, set_dirty = 0;
1410d1542e03SKent Overstreet 	unsigned copied = 0, nr_pages_copied = 0;
14111c6fdbd8SKent Overstreet 	int ret = 0;
14121c6fdbd8SKent Overstreet 
14131c6fdbd8SKent Overstreet 	BUG_ON(!len);
14141c6fdbd8SKent Overstreet 	BUG_ON(nr_pages > ARRAY_SIZE(pages));
14151c6fdbd8SKent Overstreet 
1416d1542e03SKent Overstreet 	bch2_page_reservation_init(c, inode, &res);
1417d1542e03SKent Overstreet 
14181c6fdbd8SKent Overstreet 	for (i = 0; i < nr_pages; i++) {
14191c6fdbd8SKent Overstreet 		pages[i] = grab_cache_page_write_begin(mapping, index + i);
14201c6fdbd8SKent Overstreet 		if (!pages[i]) {
14211c6fdbd8SKent Overstreet 			nr_pages = i;
14228de819f8SKent Overstreet 			if (!i) {
14231c6fdbd8SKent Overstreet 				ret = -ENOMEM;
14241c6fdbd8SKent Overstreet 				goto out;
14251c6fdbd8SKent Overstreet 			}
14268de819f8SKent Overstreet 			len = min_t(unsigned, len,
14278de819f8SKent Overstreet 				    nr_pages * PAGE_SIZE - offset);
14288de819f8SKent Overstreet 			break;
14298de819f8SKent Overstreet 		}
14301c6fdbd8SKent Overstreet 	}
14311c6fdbd8SKent Overstreet 
14321c6fdbd8SKent Overstreet 	if (offset && !PageUptodate(pages[0])) {
14331c6fdbd8SKent Overstreet 		ret = bch2_read_single_page(pages[0], mapping);
14341c6fdbd8SKent Overstreet 		if (ret)
14351c6fdbd8SKent Overstreet 			goto out;
14361c6fdbd8SKent Overstreet 	}
14371c6fdbd8SKent Overstreet 
14381c6fdbd8SKent Overstreet 	if ((pos + len) & (PAGE_SIZE - 1) &&
14391c6fdbd8SKent Overstreet 	    !PageUptodate(pages[nr_pages - 1])) {
14401c6fdbd8SKent Overstreet 		if ((index + nr_pages - 1) << PAGE_SHIFT >= inode->v.i_size) {
14411c6fdbd8SKent Overstreet 			zero_user(pages[nr_pages - 1], 0, PAGE_SIZE);
14421c6fdbd8SKent Overstreet 		} else {
14431c6fdbd8SKent Overstreet 			ret = bch2_read_single_page(pages[nr_pages - 1], mapping);
14441c6fdbd8SKent Overstreet 			if (ret)
14451c6fdbd8SKent Overstreet 				goto out;
14461c6fdbd8SKent Overstreet 		}
14471c6fdbd8SKent Overstreet 	}
14481c6fdbd8SKent Overstreet 
1449d1542e03SKent Overstreet 	while (reserved < len) {
1450d1542e03SKent Overstreet 		struct page *page = pages[(offset + reserved) >> PAGE_SHIFT];
1451d1542e03SKent Overstreet 		unsigned pg_offset = (offset + reserved) & (PAGE_SIZE - 1);
1452d1542e03SKent Overstreet 		unsigned pg_len = min_t(unsigned, len - reserved,
1453d1542e03SKent Overstreet 					PAGE_SIZE - pg_offset);
1454d1542e03SKent Overstreet retry_reservation:
1455d1542e03SKent Overstreet 		ret = bch2_page_reservation_get(c, inode, page, &res,
1456d1542e03SKent Overstreet 						pg_offset, pg_len, true);
14571c6fdbd8SKent Overstreet 
1458d1542e03SKent Overstreet 		if (ret && !PageUptodate(page)) {
1459d1542e03SKent Overstreet 			ret = bch2_read_single_page(page, mapping);
1460d1542e03SKent Overstreet 			if (!ret)
1461d1542e03SKent Overstreet 				goto retry_reservation;
14621c6fdbd8SKent Overstreet 		}
14631c6fdbd8SKent Overstreet 
14641c6fdbd8SKent Overstreet 		if (ret)
14651c6fdbd8SKent Overstreet 			goto out;
1466d1542e03SKent Overstreet 
1467d1542e03SKent Overstreet 		reserved += pg_len;
14681c6fdbd8SKent Overstreet 	}
14691c6fdbd8SKent Overstreet 
14701c6fdbd8SKent Overstreet 	if (mapping_writably_mapped(mapping))
14711c6fdbd8SKent Overstreet 		for (i = 0; i < nr_pages; i++)
14721c6fdbd8SKent Overstreet 			flush_dcache_page(pages[i]);
14731c6fdbd8SKent Overstreet 
14741c6fdbd8SKent Overstreet 	while (copied < len) {
14751c6fdbd8SKent Overstreet 		struct page *page = pages[(offset + copied) >> PAGE_SHIFT];
14761c6fdbd8SKent Overstreet 		unsigned pg_offset = (offset + copied) & (PAGE_SIZE - 1);
1477d1542e03SKent Overstreet 		unsigned pg_len = min_t(unsigned, len - copied,
14781c6fdbd8SKent Overstreet 					PAGE_SIZE - pg_offset);
14791c6fdbd8SKent Overstreet 		unsigned pg_copied = copy_page_from_iter_atomic(page,
1480d1542e03SKent Overstreet 						pg_offset, pg_len, iter);
1481d1542e03SKent Overstreet 
1482d1542e03SKent Overstreet 		if (!pg_copied)
1483d1542e03SKent Overstreet 			break;
14841c6fdbd8SKent Overstreet 
1485912bdf17SKent Overstreet 		if (!PageUptodate(page) &&
1486912bdf17SKent Overstreet 		    pg_copied != PAGE_SIZE &&
1487912bdf17SKent Overstreet 		    pos + copied + pg_copied < inode->v.i_size) {
1488912bdf17SKent Overstreet 			zero_user(page, 0, PAGE_SIZE);
1489912bdf17SKent Overstreet 			break;
1490912bdf17SKent Overstreet 		}
1491912bdf17SKent Overstreet 
14921c6fdbd8SKent Overstreet 		flush_dcache_page(page);
14931c6fdbd8SKent Overstreet 		copied += pg_copied;
1494912bdf17SKent Overstreet 
1495912bdf17SKent Overstreet 		if (pg_copied != pg_len)
1496912bdf17SKent Overstreet 			break;
14971c6fdbd8SKent Overstreet 	}
14981c6fdbd8SKent Overstreet 
14991c6fdbd8SKent Overstreet 	if (!copied)
15001c6fdbd8SKent Overstreet 		goto out;
15011c6fdbd8SKent Overstreet 
1502877dfb34SKent Overstreet 	spin_lock(&inode->v.i_lock);
1503877dfb34SKent Overstreet 	if (pos + copied > inode->v.i_size)
1504877dfb34SKent Overstreet 		i_size_write(&inode->v, pos + copied);
1505877dfb34SKent Overstreet 	spin_unlock(&inode->v.i_lock);
1506877dfb34SKent Overstreet 
1507d1542e03SKent Overstreet 	while (set_dirty < copied) {
1508d1542e03SKent Overstreet 		struct page *page = pages[(offset + set_dirty) >> PAGE_SHIFT];
1509d1542e03SKent Overstreet 		unsigned pg_offset = (offset + set_dirty) & (PAGE_SIZE - 1);
1510d1542e03SKent Overstreet 		unsigned pg_len = min_t(unsigned, copied - set_dirty,
1511d1542e03SKent Overstreet 					PAGE_SIZE - pg_offset);
1512d1542e03SKent Overstreet 
1513d1542e03SKent Overstreet 		if (!PageUptodate(page))
1514d1542e03SKent Overstreet 			SetPageUptodate(page);
1515d1542e03SKent Overstreet 
1516d1542e03SKent Overstreet 		bch2_set_page_dirty(c, inode, page, &res, pg_offset, pg_len);
1517d1542e03SKent Overstreet 		unlock_page(page);
1518d1542e03SKent Overstreet 		put_page(page);
1519d1542e03SKent Overstreet 
1520d1542e03SKent Overstreet 		set_dirty += pg_len;
1521d1542e03SKent Overstreet 	}
1522877dfb34SKent Overstreet 
1523877dfb34SKent Overstreet 	nr_pages_copied = DIV_ROUND_UP(offset + copied, PAGE_SIZE);
1524877dfb34SKent Overstreet 	inode->ei_last_dirtied = (unsigned long) current;
15251c6fdbd8SKent Overstreet out:
1526d1542e03SKent Overstreet 	for (i = nr_pages_copied; i < nr_pages; i++) {
15271c6fdbd8SKent Overstreet 		unlock_page(pages[i]);
15281c6fdbd8SKent Overstreet 		put_page(pages[i]);
15291c6fdbd8SKent Overstreet 	}
15301c6fdbd8SKent Overstreet 
1531d1542e03SKent Overstreet 	bch2_page_reservation_put(c, inode, &res);
15321c6fdbd8SKent Overstreet 
15331c6fdbd8SKent Overstreet 	return copied ?: ret;
15341c6fdbd8SKent Overstreet }
15351c6fdbd8SKent Overstreet 
15361c6fdbd8SKent Overstreet static ssize_t bch2_buffered_write(struct kiocb *iocb, struct iov_iter *iter)
15371c6fdbd8SKent Overstreet {
15381c6fdbd8SKent Overstreet 	struct file *file = iocb->ki_filp;
15391c6fdbd8SKent Overstreet 	struct address_space *mapping = file->f_mapping;
15401c6fdbd8SKent Overstreet 	struct bch_inode_info *inode = file_bch_inode(file);
15411c6fdbd8SKent Overstreet 	loff_t pos = iocb->ki_pos;
15421c6fdbd8SKent Overstreet 	ssize_t written = 0;
15431c6fdbd8SKent Overstreet 	int ret = 0;
15441c6fdbd8SKent Overstreet 
15451c6fdbd8SKent Overstreet 	bch2_pagecache_add_get(&inode->ei_pagecache_lock);
15461c6fdbd8SKent Overstreet 
15471c6fdbd8SKent Overstreet 	do {
15481c6fdbd8SKent Overstreet 		unsigned offset = pos & (PAGE_SIZE - 1);
15491c6fdbd8SKent Overstreet 		unsigned bytes = min_t(unsigned long, iov_iter_count(iter),
15501c6fdbd8SKent Overstreet 			      PAGE_SIZE * WRITE_BATCH_PAGES - offset);
15511c6fdbd8SKent Overstreet again:
15521c6fdbd8SKent Overstreet 		/*
15531c6fdbd8SKent Overstreet 		 * Bring in the user page that we will copy from _first_.
15541c6fdbd8SKent Overstreet 		 * Otherwise there's a nasty deadlock on copying from the
15551c6fdbd8SKent Overstreet 		 * same page as we're writing to, without it being marked
15561c6fdbd8SKent Overstreet 		 * up-to-date.
15571c6fdbd8SKent Overstreet 		 *
15581c6fdbd8SKent Overstreet 		 * Not only is this an optimisation, but it is also required
15591c6fdbd8SKent Overstreet 		 * to check that the address is actually valid, when atomic
15601c6fdbd8SKent Overstreet 		 * usercopies are used, below.
15611c6fdbd8SKent Overstreet 		 */
15621c6fdbd8SKent Overstreet 		if (unlikely(fault_in_iov_iter_readable(iter, bytes))) {
15631c6fdbd8SKent Overstreet 			bytes = min_t(unsigned long, iov_iter_count(iter),
15641c6fdbd8SKent Overstreet 				      PAGE_SIZE - offset);
15651c6fdbd8SKent Overstreet 
15661c6fdbd8SKent Overstreet 			if (unlikely(fault_in_iov_iter_readable(iter, bytes))) {
15671c6fdbd8SKent Overstreet 				ret = -EFAULT;
15681c6fdbd8SKent Overstreet 				break;
15691c6fdbd8SKent Overstreet 			}
15701c6fdbd8SKent Overstreet 		}
15711c6fdbd8SKent Overstreet 
15721c6fdbd8SKent Overstreet 		if (unlikely(fatal_signal_pending(current))) {
15731c6fdbd8SKent Overstreet 			ret = -EINTR;
15741c6fdbd8SKent Overstreet 			break;
15751c6fdbd8SKent Overstreet 		}
15761c6fdbd8SKent Overstreet 
15771c6fdbd8SKent Overstreet 		ret = __bch2_buffered_write(inode, mapping, iter, pos, bytes);
15781c6fdbd8SKent Overstreet 		if (unlikely(ret < 0))
15791c6fdbd8SKent Overstreet 			break;
15801c6fdbd8SKent Overstreet 
15811c6fdbd8SKent Overstreet 		cond_resched();
15821c6fdbd8SKent Overstreet 
15831c6fdbd8SKent Overstreet 		if (unlikely(ret == 0)) {
15841c6fdbd8SKent Overstreet 			/*
15851c6fdbd8SKent Overstreet 			 * If we were unable to copy any data at all, we must
15861c6fdbd8SKent Overstreet 			 * fall back to a single segment length write.
15871c6fdbd8SKent Overstreet 			 *
15881c6fdbd8SKent Overstreet 			 * If we didn't fallback here, we could livelock
15891c6fdbd8SKent Overstreet 			 * because not all segments in the iov can be copied at
15901c6fdbd8SKent Overstreet 			 * once without a pagefault.
15911c6fdbd8SKent Overstreet 			 */
15921c6fdbd8SKent Overstreet 			bytes = min_t(unsigned long, PAGE_SIZE - offset,
15931c6fdbd8SKent Overstreet 				      iov_iter_single_seg_count(iter));
15941c6fdbd8SKent Overstreet 			goto again;
15951c6fdbd8SKent Overstreet 		}
15961c6fdbd8SKent Overstreet 		pos += ret;
15971c6fdbd8SKent Overstreet 		written += ret;
1598912bdf17SKent Overstreet 		ret = 0;
15991c6fdbd8SKent Overstreet 
16001c6fdbd8SKent Overstreet 		balance_dirty_pages_ratelimited(mapping);
16011c6fdbd8SKent Overstreet 	} while (iov_iter_count(iter));
16021c6fdbd8SKent Overstreet 
16031c6fdbd8SKent Overstreet 	bch2_pagecache_add_put(&inode->ei_pagecache_lock);
16041c6fdbd8SKent Overstreet 
16051c6fdbd8SKent Overstreet 	return written ? written : ret;
16061c6fdbd8SKent Overstreet }
16071c6fdbd8SKent Overstreet 
16081c6fdbd8SKent Overstreet /* O_DIRECT reads */
16091c6fdbd8SKent Overstreet 
1610b4725cc1SKent Overstreet static void bio_check_or_release(struct bio *bio, bool check_dirty)
1611b4725cc1SKent Overstreet {
1612b4725cc1SKent Overstreet 	if (check_dirty) {
1613b4725cc1SKent Overstreet 		bio_check_pages_dirty(bio);
1614b4725cc1SKent Overstreet 	} else {
1615b4725cc1SKent Overstreet 		bio_release_pages(bio, false);
1616b4725cc1SKent Overstreet 		bio_put(bio);
1617b4725cc1SKent Overstreet 	}
1618b4725cc1SKent Overstreet }
1619b4725cc1SKent Overstreet 
16201c6fdbd8SKent Overstreet static void bch2_dio_read_complete(struct closure *cl)
16211c6fdbd8SKent Overstreet {
16221c6fdbd8SKent Overstreet 	struct dio_read *dio = container_of(cl, struct dio_read, cl);
16231c6fdbd8SKent Overstreet 
16241c6fdbd8SKent Overstreet 	dio->req->ki_complete(dio->req, dio->ret);
1625b4725cc1SKent Overstreet 	bio_check_or_release(&dio->rbio.bio, dio->should_dirty);
16261c6fdbd8SKent Overstreet }
16271c6fdbd8SKent Overstreet 
16281c6fdbd8SKent Overstreet static void bch2_direct_IO_read_endio(struct bio *bio)
16291c6fdbd8SKent Overstreet {
16301c6fdbd8SKent Overstreet 	struct dio_read *dio = bio->bi_private;
16311c6fdbd8SKent Overstreet 
16321c6fdbd8SKent Overstreet 	if (bio->bi_status)
16331c6fdbd8SKent Overstreet 		dio->ret = blk_status_to_errno(bio->bi_status);
16341c6fdbd8SKent Overstreet 
16351c6fdbd8SKent Overstreet 	closure_put(&dio->cl);
16361c6fdbd8SKent Overstreet }
16371c6fdbd8SKent Overstreet 
16381c6fdbd8SKent Overstreet static void bch2_direct_IO_read_split_endio(struct bio *bio)
16391c6fdbd8SKent Overstreet {
1640b4725cc1SKent Overstreet 	struct dio_read *dio = bio->bi_private;
1641b4725cc1SKent Overstreet 	bool should_dirty = dio->should_dirty;
1642b4725cc1SKent Overstreet 
16431c6fdbd8SKent Overstreet 	bch2_direct_IO_read_endio(bio);
1644b4725cc1SKent Overstreet 	bio_check_or_release(bio, should_dirty);
16451c6fdbd8SKent Overstreet }
16461c6fdbd8SKent Overstreet 
16471c6fdbd8SKent Overstreet static int bch2_direct_IO_read(struct kiocb *req, struct iov_iter *iter)
16481c6fdbd8SKent Overstreet {
16491c6fdbd8SKent Overstreet 	struct file *file = req->ki_filp;
16501c6fdbd8SKent Overstreet 	struct bch_inode_info *inode = file_bch_inode(file);
16511c6fdbd8SKent Overstreet 	struct bch_fs *c = inode->v.i_sb->s_fs_info;
16529a3df993SKent Overstreet 	struct bch_io_opts opts = io_opts(c, &inode->ei_inode);
16531c6fdbd8SKent Overstreet 	struct dio_read *dio;
16541c6fdbd8SKent Overstreet 	struct bio *bio;
16551c6fdbd8SKent Overstreet 	loff_t offset = req->ki_pos;
16561c6fdbd8SKent Overstreet 	bool sync = is_sync_kiocb(req);
16571c6fdbd8SKent Overstreet 	size_t shorten;
16581c6fdbd8SKent Overstreet 	ssize_t ret;
16591c6fdbd8SKent Overstreet 
16601c6fdbd8SKent Overstreet 	if ((offset|iter->count) & (block_bytes(c) - 1))
16611c6fdbd8SKent Overstreet 		return -EINVAL;
16621c6fdbd8SKent Overstreet 
16631c6fdbd8SKent Overstreet 	ret = min_t(loff_t, iter->count,
16641c6fdbd8SKent Overstreet 		    max_t(loff_t, 0, i_size_read(&inode->v) - offset));
16651c6fdbd8SKent Overstreet 
16661c6fdbd8SKent Overstreet 	if (!ret)
16671c6fdbd8SKent Overstreet 		return ret;
16681c6fdbd8SKent Overstreet 
16691c6fdbd8SKent Overstreet 	shorten = iov_iter_count(iter) - round_up(ret, block_bytes(c));
16701c6fdbd8SKent Overstreet 	iter->count -= shorten;
16711c6fdbd8SKent Overstreet 
16721c6fdbd8SKent Overstreet 	bio = bio_alloc_bioset(NULL,
16731c6fdbd8SKent Overstreet 			       iov_iter_npages(iter, BIO_MAX_VECS),
16741c6fdbd8SKent Overstreet 			       REQ_OP_READ,
16751c6fdbd8SKent Overstreet 			       GFP_KERNEL,
16761c6fdbd8SKent Overstreet 			       &c->dio_read_bioset);
16771c6fdbd8SKent Overstreet 
16781c6fdbd8SKent Overstreet 	bio->bi_end_io = bch2_direct_IO_read_endio;
16791c6fdbd8SKent Overstreet 
16801c6fdbd8SKent Overstreet 	dio = container_of(bio, struct dio_read, rbio.bio);
16811c6fdbd8SKent Overstreet 	closure_init(&dio->cl, NULL);
16821c6fdbd8SKent Overstreet 
16831c6fdbd8SKent Overstreet 	/*
16841c6fdbd8SKent Overstreet 	 * this is a _really_ horrible hack just to avoid an atomic sub at the
16851c6fdbd8SKent Overstreet 	 * end:
16861c6fdbd8SKent Overstreet 	 */
16871c6fdbd8SKent Overstreet 	if (!sync) {
16881c6fdbd8SKent Overstreet 		set_closure_fn(&dio->cl, bch2_dio_read_complete, NULL);
16891c6fdbd8SKent Overstreet 		atomic_set(&dio->cl.remaining,
16901c6fdbd8SKent Overstreet 			   CLOSURE_REMAINING_INITIALIZER -
16911c6fdbd8SKent Overstreet 			   CLOSURE_RUNNING +
16921c6fdbd8SKent Overstreet 			   CLOSURE_DESTRUCTOR);
16931c6fdbd8SKent Overstreet 	} else {
16941c6fdbd8SKent Overstreet 		atomic_set(&dio->cl.remaining,
16951c6fdbd8SKent Overstreet 			   CLOSURE_REMAINING_INITIALIZER + 1);
16961c6fdbd8SKent Overstreet 	}
16971c6fdbd8SKent Overstreet 
16981c6fdbd8SKent Overstreet 	dio->req	= req;
16991c6fdbd8SKent Overstreet 	dio->ret	= ret;
1700b4725cc1SKent Overstreet 	/*
1701b4725cc1SKent Overstreet 	 * This is one of the sketchier things I've encountered: we have to skip
1702b4725cc1SKent Overstreet 	 * the dirtying of requests that are internal from the kernel (i.e. from
1703b4725cc1SKent Overstreet 	 * loopback), because we'll deadlock on page_lock.
1704b4725cc1SKent Overstreet 	 */
1705b4725cc1SKent Overstreet 	dio->should_dirty = iter_is_iovec(iter);
17061c6fdbd8SKent Overstreet 
17071c6fdbd8SKent Overstreet 	goto start;
17081c6fdbd8SKent Overstreet 	while (iter->count) {
17091c6fdbd8SKent Overstreet 		bio = bio_alloc_bioset(NULL,
17101c6fdbd8SKent Overstreet 				       iov_iter_npages(iter, BIO_MAX_VECS),
17111c6fdbd8SKent Overstreet 				       REQ_OP_READ,
17121c6fdbd8SKent Overstreet 				       GFP_KERNEL,
17131c6fdbd8SKent Overstreet 				       &c->bio_read);
17141c6fdbd8SKent Overstreet 		bio->bi_end_io		= bch2_direct_IO_read_split_endio;
17151c6fdbd8SKent Overstreet start:
17161c6fdbd8SKent Overstreet 		bio->bi_opf		= REQ_OP_READ|REQ_SYNC;
17171c6fdbd8SKent Overstreet 		bio->bi_iter.bi_sector	= offset >> 9;
17181c6fdbd8SKent Overstreet 		bio->bi_private		= dio;
17191c6fdbd8SKent Overstreet 
17201c6fdbd8SKent Overstreet 		ret = bio_iov_iter_get_pages(bio, iter);
17211c6fdbd8SKent Overstreet 		if (ret < 0) {
17221c6fdbd8SKent Overstreet 			/* XXX: fault inject this path */
17231c6fdbd8SKent Overstreet 			bio->bi_status = BLK_STS_RESOURCE;
17241c6fdbd8SKent Overstreet 			bio_endio(bio);
17251c6fdbd8SKent Overstreet 			break;
17261c6fdbd8SKent Overstreet 		}
17271c6fdbd8SKent Overstreet 
17281c6fdbd8SKent Overstreet 		offset += bio->bi_iter.bi_size;
1729b4725cc1SKent Overstreet 
1730b4725cc1SKent Overstreet 		if (dio->should_dirty)
17311c6fdbd8SKent Overstreet 			bio_set_pages_dirty(bio);
17321c6fdbd8SKent Overstreet 
17331c6fdbd8SKent Overstreet 		if (iter->count)
17341c6fdbd8SKent Overstreet 			closure_get(&dio->cl);
17351c6fdbd8SKent Overstreet 
17361c6fdbd8SKent Overstreet 		bch2_read(c, rbio_init(bio, opts), inode->v.i_ino);
17371c6fdbd8SKent Overstreet 	}
17381c6fdbd8SKent Overstreet 
17391c6fdbd8SKent Overstreet 	iter->count += shorten;
17401c6fdbd8SKent Overstreet 
17411c6fdbd8SKent Overstreet 	if (sync) {
17421c6fdbd8SKent Overstreet 		closure_sync(&dio->cl);
17431c6fdbd8SKent Overstreet 		closure_debug_destroy(&dio->cl);
17441c6fdbd8SKent Overstreet 		ret = dio->ret;
1745b4725cc1SKent Overstreet 		bio_check_or_release(&dio->rbio.bio, dio->should_dirty);
17461c6fdbd8SKent Overstreet 		return ret;
17471c6fdbd8SKent Overstreet 	} else {
17481c6fdbd8SKent Overstreet 		return -EIOCBQUEUED;
17491c6fdbd8SKent Overstreet 	}
17501c6fdbd8SKent Overstreet }
17511c6fdbd8SKent Overstreet 
17521c6fdbd8SKent Overstreet ssize_t bch2_read_iter(struct kiocb *iocb, struct iov_iter *iter)
17531c6fdbd8SKent Overstreet {
17541c6fdbd8SKent Overstreet 	struct file *file = iocb->ki_filp;
17551c6fdbd8SKent Overstreet 	struct bch_inode_info *inode = file_bch_inode(file);
17561c6fdbd8SKent Overstreet 	struct address_space *mapping = file->f_mapping;
17571c6fdbd8SKent Overstreet 	size_t count = iov_iter_count(iter);
17581c6fdbd8SKent Overstreet 	ssize_t ret;
17591c6fdbd8SKent Overstreet 
17601c6fdbd8SKent Overstreet 	if (!count)
17611c6fdbd8SKent Overstreet 		return 0; /* skip atime */
17621c6fdbd8SKent Overstreet 
17631c6fdbd8SKent Overstreet 	if (iocb->ki_flags & IOCB_DIRECT) {
17641c6fdbd8SKent Overstreet 		struct blk_plug plug;
17651c6fdbd8SKent Overstreet 
1766a023127aSKent Overstreet 		if (unlikely(mapping->nrpages)) {
17671c6fdbd8SKent Overstreet 			ret = filemap_write_and_wait_range(mapping,
17681c6fdbd8SKent Overstreet 						iocb->ki_pos,
17691c6fdbd8SKent Overstreet 						iocb->ki_pos + count - 1);
17701c6fdbd8SKent Overstreet 			if (ret < 0)
17711c6fdbd8SKent Overstreet 				return ret;
1772a023127aSKent Overstreet 		}
17731c6fdbd8SKent Overstreet 
17741c6fdbd8SKent Overstreet 		file_accessed(file);
17751c6fdbd8SKent Overstreet 
17761c6fdbd8SKent Overstreet 		blk_start_plug(&plug);
17771c6fdbd8SKent Overstreet 		ret = bch2_direct_IO_read(iocb, iter);
17781c6fdbd8SKent Overstreet 		blk_finish_plug(&plug);
17791c6fdbd8SKent Overstreet 
17801c6fdbd8SKent Overstreet 		if (ret >= 0)
17811c6fdbd8SKent Overstreet 			iocb->ki_pos += ret;
17821c6fdbd8SKent Overstreet 	} else {
17831c6fdbd8SKent Overstreet 		bch2_pagecache_add_get(&inode->ei_pagecache_lock);
17841c6fdbd8SKent Overstreet 		ret = generic_file_read_iter(iocb, iter);
17851c6fdbd8SKent Overstreet 		bch2_pagecache_add_put(&inode->ei_pagecache_lock);
17861c6fdbd8SKent Overstreet 	}
17871c6fdbd8SKent Overstreet 
17881c6fdbd8SKent Overstreet 	return ret;
17891c6fdbd8SKent Overstreet }
17901c6fdbd8SKent Overstreet 
17911c6fdbd8SKent Overstreet /* O_DIRECT writes */
17921c6fdbd8SKent Overstreet 
17931c6fdbd8SKent Overstreet /*
17941c6fdbd8SKent Overstreet  * We're going to return -EIOCBQUEUED, but we haven't finished consuming the
17951c6fdbd8SKent Overstreet  * iov_iter yet, so we need to stash a copy of the iovec: it might be on the
17961c6fdbd8SKent Overstreet  * caller's stack, we're not guaranteed that it will live for the duration of
17971c6fdbd8SKent Overstreet  * the IO:
17981c6fdbd8SKent Overstreet  */
17991c6fdbd8SKent Overstreet static noinline int bch2_dio_write_copy_iov(struct dio_write *dio)
18001c6fdbd8SKent Overstreet {
18011c6fdbd8SKent Overstreet 	struct iovec *iov = dio->inline_vecs;
18021c6fdbd8SKent Overstreet 
18031c6fdbd8SKent Overstreet 	/*
18041c6fdbd8SKent Overstreet 	 * iov_iter has a single embedded iovec - nothing to do:
18051c6fdbd8SKent Overstreet 	 */
18061c6fdbd8SKent Overstreet 	if (iter_is_ubuf(&dio->iter))
18071c6fdbd8SKent Overstreet 		return 0;
18081c6fdbd8SKent Overstreet 
18091c6fdbd8SKent Overstreet 	/*
18101c6fdbd8SKent Overstreet 	 * We don't currently handle non-iovec iov_iters here - return an error,
18111c6fdbd8SKent Overstreet 	 * and we'll fall back to doing the IO synchronously:
18121c6fdbd8SKent Overstreet 	 */
18131c6fdbd8SKent Overstreet 	if (!iter_is_iovec(&dio->iter))
18141c6fdbd8SKent Overstreet 		return -1;
18151c6fdbd8SKent Overstreet 
18161c6fdbd8SKent Overstreet 	if (dio->iter.nr_segs > ARRAY_SIZE(dio->inline_vecs)) {
18171c6fdbd8SKent Overstreet 		iov = kmalloc_array(dio->iter.nr_segs, sizeof(*iov),
18181c6fdbd8SKent Overstreet 				    GFP_KERNEL);
18191c6fdbd8SKent Overstreet 		if (unlikely(!iov))
18201c6fdbd8SKent Overstreet 			return -ENOMEM;
18211c6fdbd8SKent Overstreet 
18221c6fdbd8SKent Overstreet 		dio->free_iov = true;
18231c6fdbd8SKent Overstreet 	}
18241c6fdbd8SKent Overstreet 
18251c6fdbd8SKent Overstreet 	memcpy(iov, dio->iter.__iov, dio->iter.nr_segs * sizeof(*iov));
18261c6fdbd8SKent Overstreet 	dio->iter.__iov = iov;
18271c6fdbd8SKent Overstreet 	return 0;
18281c6fdbd8SKent Overstreet }
18291c6fdbd8SKent Overstreet 
1830042a1f26SKent Overstreet static void bch2_dio_write_loop_async(struct bch_write_op *);
1831042a1f26SKent Overstreet 
18321c6fdbd8SKent Overstreet static long bch2_dio_write_loop(struct dio_write *dio)
18331c6fdbd8SKent Overstreet {
1834ed484030SKent Overstreet 	bool kthread = (current->flags & PF_KTHREAD) != 0;
18351c6fdbd8SKent Overstreet 	struct kiocb *req = dio->req;
18361c6fdbd8SKent Overstreet 	struct address_space *mapping = req->ki_filp->f_mapping;
18379a3df993SKent Overstreet 	struct bch_inode_info *inode = file_bch_inode(req->ki_filp);
1838042a1f26SKent Overstreet 	struct bch_fs *c = inode->v.i_sb->s_fs_info;
18399a3df993SKent Overstreet 	struct bio *bio = &dio->op.wbio.bio;
1840eb8e6e9cSKent Overstreet 	unsigned unaligned, iter_count;
1841eb8e6e9cSKent Overstreet 	bool sync = dio->sync, dropped_locks;
18421c6fdbd8SKent Overstreet 	long ret;
18431c6fdbd8SKent Overstreet 
18441c6fdbd8SKent Overstreet 	if (dio->loop)
18451c6fdbd8SKent Overstreet 		goto loop;
18461c6fdbd8SKent Overstreet 
18471c6fdbd8SKent Overstreet 	while (1) {
1848eb8e6e9cSKent Overstreet 		iter_count = dio->iter.count;
1849eb8e6e9cSKent Overstreet 
1850ed484030SKent Overstreet 		if (kthread)
1851ed484030SKent Overstreet 			kthread_use_mm(dio->mm);
18521c6fdbd8SKent Overstreet 		BUG_ON(current->faults_disabled_mapping);
18531c6fdbd8SKent Overstreet 		current->faults_disabled_mapping = mapping;
18541c6fdbd8SKent Overstreet 
18551c6fdbd8SKent Overstreet 		ret = bio_iov_iter_get_pages(bio, &dio->iter);
18561c6fdbd8SKent Overstreet 
1857eb8e6e9cSKent Overstreet 		dropped_locks = fdm_dropped_locks();
1858eb8e6e9cSKent Overstreet 
18591c6fdbd8SKent Overstreet 		current->faults_disabled_mapping = NULL;
1860ed484030SKent Overstreet 		if (kthread)
1861ed484030SKent Overstreet 			kthread_unuse_mm(dio->mm);
18621c6fdbd8SKent Overstreet 
1863eb8e6e9cSKent Overstreet 		/*
1864eb8e6e9cSKent Overstreet 		 * If the fault handler returned an error but also signalled
1865eb8e6e9cSKent Overstreet 		 * that it dropped & retook ei_pagecache_lock, we just need to
1866eb8e6e9cSKent Overstreet 		 * re-shoot down the page cache and retry:
1867eb8e6e9cSKent Overstreet 		 */
1868eb8e6e9cSKent Overstreet 		if (dropped_locks && ret)
1869eb8e6e9cSKent Overstreet 			ret = 0;
1870eb8e6e9cSKent Overstreet 
18711c6fdbd8SKent Overstreet 		if (unlikely(ret < 0))
18721c6fdbd8SKent Overstreet 			goto err;
18731c6fdbd8SKent Overstreet 
1874eb8e6e9cSKent Overstreet 		if (unlikely(dropped_locks)) {
1875eb8e6e9cSKent Overstreet 			ret = write_invalidate_inode_pages_range(mapping,
1876eb8e6e9cSKent Overstreet 					req->ki_pos,
1877eb8e6e9cSKent Overstreet 					req->ki_pos + iter_count - 1);
1878eb8e6e9cSKent Overstreet 			if (unlikely(ret))
1879eb8e6e9cSKent Overstreet 				goto err;
1880eb8e6e9cSKent Overstreet 
1881eb8e6e9cSKent Overstreet 			if (!bio->bi_iter.bi_size)
1882eb8e6e9cSKent Overstreet 				continue;
1883eb8e6e9cSKent Overstreet 		}
1884eb8e6e9cSKent Overstreet 
18850a426c32SKent Overstreet 		unaligned = bio->bi_iter.bi_size & (block_bytes(c) - 1);
18860a426c32SKent Overstreet 		bio->bi_iter.bi_size -= unaligned;
18870a426c32SKent Overstreet 		iov_iter_revert(&dio->iter, unaligned);
18880a426c32SKent Overstreet 
18890a426c32SKent Overstreet 		if (!bio->bi_iter.bi_size) {
18900a426c32SKent Overstreet 			/*
18910a426c32SKent Overstreet 			 * bio_iov_iter_get_pages was only able to get <
18920a426c32SKent Overstreet 			 * blocksize worth of pages:
18930a426c32SKent Overstreet 			 */
18940a426c32SKent Overstreet 			ret = -EFAULT;
18950a426c32SKent Overstreet 			goto err;
18960a426c32SKent Overstreet 		}
18970a426c32SKent Overstreet 
1898042a1f26SKent Overstreet 		bch2_write_op_init(&dio->op, c, io_opts(c, &inode->ei_inode));
1899042a1f26SKent Overstreet 		dio->op.end_io		= bch2_dio_write_loop_async;
1900042a1f26SKent Overstreet 		dio->op.target		= dio->op.opts.foreground_target;
1901042a1f26SKent Overstreet 		op_journal_seq_set(&dio->op, &inode->ei_journal_seq);
1902042a1f26SKent Overstreet 		dio->op.write_point	= writepoint_hashed((unsigned long) current);
1903042a1f26SKent Overstreet 		dio->op.nr_replicas	= dio->op.opts.data_replicas;
1904042a1f26SKent Overstreet 		dio->op.pos		= POS(inode->v.i_ino, (u64) req->ki_pos >> 9);
1905042a1f26SKent Overstreet 
1906042a1f26SKent Overstreet 		if ((req->ki_flags & IOCB_DSYNC) &&
1907042a1f26SKent Overstreet 		    !c->opts.journal_flush_disabled)
1908042a1f26SKent Overstreet 			dio->op.flags |= BCH_WRITE_FLUSH;
1909a6336910SKent Overstreet 		dio->op.flags |= BCH_WRITE_CHECK_ENOSPC;
1910042a1f26SKent Overstreet 
1911042a1f26SKent Overstreet 		ret = bch2_disk_reservation_get(c, &dio->op.res, bio_sectors(bio),
1912042a1f26SKent Overstreet 						dio->op.opts.data_replicas, 0);
1913042a1f26SKent Overstreet 		if (unlikely(ret) &&
1914042a1f26SKent Overstreet 		    !bch2_check_range_allocated(c, dio->op.pos,
191535a067b4SKent Overstreet 				bio_sectors(bio),
191635a067b4SKent Overstreet 				dio->op.opts.data_replicas,
191735a067b4SKent Overstreet 				dio->op.opts.compression != 0))
1918042a1f26SKent Overstreet 			goto err;
19191c6fdbd8SKent Overstreet 
19201c6fdbd8SKent Overstreet 		task_io_account_write(bio->bi_iter.bi_size);
19211c6fdbd8SKent Overstreet 
19221c6fdbd8SKent Overstreet 		if (!dio->sync && !dio->loop && dio->iter.count) {
19231c6fdbd8SKent Overstreet 			if (bch2_dio_write_copy_iov(dio)) {
1924286d8ad0SKent Overstreet 				dio->sync = sync = true;
1925f8f30863SKent Overstreet 				goto do_io;
19261c6fdbd8SKent Overstreet 			}
19271c6fdbd8SKent Overstreet 		}
1928f8f30863SKent Overstreet do_io:
19291c6fdbd8SKent Overstreet 		dio->loop = true;
1930f8f30863SKent Overstreet 		closure_call(&dio->op.cl, bch2_write, NULL, NULL);
19311c6fdbd8SKent Overstreet 
1932286d8ad0SKent Overstreet 		if (sync)
1933f8f30863SKent Overstreet 			wait_for_completion(&dio->done);
1934f8f30863SKent Overstreet 		else
19351c6fdbd8SKent Overstreet 			return -EIOCBQUEUED;
19361c6fdbd8SKent Overstreet loop:
19379a3df993SKent Overstreet 		i_sectors_acct(c, inode, &dio->quota_res,
19389a3df993SKent Overstreet 			       dio->op.i_sectors_delta);
1939042a1f26SKent Overstreet 		req->ki_pos += (u64) dio->op.written << 9;
1940042a1f26SKent Overstreet 		dio->written += dio->op.written;
19419a3df993SKent Overstreet 
19429a3df993SKent Overstreet 		spin_lock(&inode->v.i_lock);
1943042a1f26SKent Overstreet 		if (req->ki_pos > inode->v.i_size)
1944042a1f26SKent Overstreet 			i_size_write(&inode->v, req->ki_pos);
19459a3df993SKent Overstreet 		spin_unlock(&inode->v.i_lock);
19469a3df993SKent Overstreet 
1947032ac32cSKent Overstreet 		bio_release_pages(bio, false);
19485468f119SKent Overstreet 		bio->bi_vcnt = 0;
194933c74e41SKent Overstreet 
195033c74e41SKent Overstreet 		if (dio->op.error) {
195133c74e41SKent Overstreet 			set_bit(EI_INODE_ERROR, &inode->ei_flags);
195233c74e41SKent Overstreet 			break;
195333c74e41SKent Overstreet 		}
195433c74e41SKent Overstreet 
195533c74e41SKent Overstreet 		if (!dio->iter.count)
19561c6fdbd8SKent Overstreet 			break;
1957f8f30863SKent Overstreet 
19581c6fdbd8SKent Overstreet 		bio_reset(bio, NULL, REQ_OP_WRITE);
1959f8f30863SKent Overstreet 		reinit_completion(&dio->done);
19601c6fdbd8SKent Overstreet 	}
19611c6fdbd8SKent Overstreet 
1962042a1f26SKent Overstreet 	ret = dio->op.error ?: ((long) dio->written << 9);
19631c6fdbd8SKent Overstreet err:
19641c6fdbd8SKent Overstreet 	bch2_pagecache_block_put(&inode->ei_pagecache_lock);
19650a426c32SKent Overstreet 	bch2_quota_reservation_put(c, inode, &dio->quota_res);
19661c6fdbd8SKent Overstreet 
19671c6fdbd8SKent Overstreet 	if (dio->free_iov)
19681c6fdbd8SKent Overstreet 		kfree(dio->iter.__iov);
19691c6fdbd8SKent Overstreet 
19705468f119SKent Overstreet 	bio_release_pages(bio, false);
19711c6fdbd8SKent Overstreet 	bio_put(bio);
19721c6fdbd8SKent Overstreet 
19731c6fdbd8SKent Overstreet 	/* inode->i_dio_count is our ref on inode and thus bch_fs */
19741c6fdbd8SKent Overstreet 	inode_dio_end(&inode->v);
19751c6fdbd8SKent Overstreet 
19761c6fdbd8SKent Overstreet 	if (!sync) {
19771c6fdbd8SKent Overstreet 		req->ki_complete(req, ret);
19781c6fdbd8SKent Overstreet 		ret = -EIOCBQUEUED;
19791c6fdbd8SKent Overstreet 	}
19801c6fdbd8SKent Overstreet 	return ret;
19811c6fdbd8SKent Overstreet }
19821c6fdbd8SKent Overstreet 
1983f8f30863SKent Overstreet static void bch2_dio_write_loop_async(struct bch_write_op *op)
19841c6fdbd8SKent Overstreet {
1985f8f30863SKent Overstreet 	struct dio_write *dio = container_of(op, struct dio_write, op);
19861c6fdbd8SKent Overstreet 
1987f8f30863SKent Overstreet 	if (dio->sync)
1988f8f30863SKent Overstreet 		complete(&dio->done);
1989f8f30863SKent Overstreet 	else
19901c6fdbd8SKent Overstreet 		bch2_dio_write_loop(dio);
19911c6fdbd8SKent Overstreet }
19921c6fdbd8SKent Overstreet 
19931c6fdbd8SKent Overstreet static noinline
19941c6fdbd8SKent Overstreet ssize_t bch2_direct_write(struct kiocb *req, struct iov_iter *iter)
19951c6fdbd8SKent Overstreet {
19961c6fdbd8SKent Overstreet 	struct file *file = req->ki_filp;
199754847d25SKent Overstreet 	struct address_space *mapping = file->f_mapping;
19981c6fdbd8SKent Overstreet 	struct bch_inode_info *inode = file_bch_inode(file);
19991c6fdbd8SKent Overstreet 	struct bch_fs *c = inode->v.i_sb->s_fs_info;
20001c6fdbd8SKent Overstreet 	struct dio_write *dio;
20011c6fdbd8SKent Overstreet 	struct bio *bio;
20027edcfbfeSKent Overstreet 	bool locked = true, extending;
20031c6fdbd8SKent Overstreet 	ssize_t ret;
20041c6fdbd8SKent Overstreet 
20057edcfbfeSKent Overstreet 	prefetch(&c->opts);
20067edcfbfeSKent Overstreet 	prefetch((void *) &c->opts + 64);
20077edcfbfeSKent Overstreet 	prefetch(&inode->ei_inode);
20087edcfbfeSKent Overstreet 	prefetch((void *) &inode->ei_inode + 64);
20091c6fdbd8SKent Overstreet 
20107edcfbfeSKent Overstreet 	inode_lock(&inode->v);
20117edcfbfeSKent Overstreet 
20127edcfbfeSKent Overstreet 	ret = generic_write_checks(req, iter);
20137edcfbfeSKent Overstreet 	if (unlikely(ret <= 0))
20147edcfbfeSKent Overstreet 		goto err;
20157edcfbfeSKent Overstreet 
20167edcfbfeSKent Overstreet 	ret = file_remove_privs(file);
20177edcfbfeSKent Overstreet 	if (unlikely(ret))
20187edcfbfeSKent Overstreet 		goto err;
20197edcfbfeSKent Overstreet 
20207edcfbfeSKent Overstreet 	ret = file_update_time(file);
20217edcfbfeSKent Overstreet 	if (unlikely(ret))
20227edcfbfeSKent Overstreet 		goto err;
20231c6fdbd8SKent Overstreet 
2024919dbbd1SKent Overstreet 	if (unlikely((req->ki_pos|iter->count) & (block_bytes(c) - 1)))
20257edcfbfeSKent Overstreet 		goto err;
20267edcfbfeSKent Overstreet 
20277edcfbfeSKent Overstreet 	inode_dio_begin(&inode->v);
20287edcfbfeSKent Overstreet 	bch2_pagecache_block_get(&inode->ei_pagecache_lock);
20297edcfbfeSKent Overstreet 
20307edcfbfeSKent Overstreet 	extending = req->ki_pos + iter->count > inode->v.i_size;
20317edcfbfeSKent Overstreet 	if (!extending) {
20327edcfbfeSKent Overstreet 		inode_unlock(&inode->v);
20337edcfbfeSKent Overstreet 		locked = false;
20347edcfbfeSKent Overstreet 	}
20351c6fdbd8SKent Overstreet 
20361c6fdbd8SKent Overstreet 	bio = bio_alloc_bioset(NULL,
20372ed5cd50SKent Overstreet 			       iov_iter_is_bvec(iter)
20382ed5cd50SKent Overstreet 			       ? 0
20392ed5cd50SKent Overstreet 			       : iov_iter_npages(iter, BIO_MAX_VECS),
20401c6fdbd8SKent Overstreet 			       REQ_OP_WRITE,
20411c6fdbd8SKent Overstreet 			       GFP_KERNEL,
20421c6fdbd8SKent Overstreet 			       &c->dio_write_bioset);
20439a3df993SKent Overstreet 	dio = container_of(bio, struct dio_write, op.wbio.bio);
2044f8f30863SKent Overstreet 	init_completion(&dio->done);
20451c6fdbd8SKent Overstreet 	dio->req		= req;
2046ed484030SKent Overstreet 	dio->mm			= current->mm;
20471c6fdbd8SKent Overstreet 	dio->loop		= false;
20487edcfbfeSKent Overstreet 	dio->sync		= is_sync_kiocb(req) || extending;
20491c6fdbd8SKent Overstreet 	dio->free_iov		= false;
20501c6fdbd8SKent Overstreet 	dio->quota_res.sectors	= 0;
2051042a1f26SKent Overstreet 	dio->written		= 0;
20521c6fdbd8SKent Overstreet 	dio->iter		= *iter;
20539a3df993SKent Overstreet 
20541c6fdbd8SKent Overstreet 	ret = bch2_quota_reservation_add(c, inode, &dio->quota_res,
20551c6fdbd8SKent Overstreet 					 iter->count >> 9, true);
20561c6fdbd8SKent Overstreet 	if (unlikely(ret))
20577edcfbfeSKent Overstreet 		goto err_put_bio;
20581c6fdbd8SKent Overstreet 
2059a023127aSKent Overstreet 	if (unlikely(mapping->nrpages)) {
206054847d25SKent Overstreet 		ret = write_invalidate_inode_pages_range(mapping,
206154847d25SKent Overstreet 						req->ki_pos,
206254847d25SKent Overstreet 						req->ki_pos + iter->count - 1);
206354847d25SKent Overstreet 		if (unlikely(ret))
206454847d25SKent Overstreet 			goto err_put_bio;
2065a023127aSKent Overstreet 	}
206654847d25SKent Overstreet 
20677edcfbfeSKent Overstreet 	ret = bch2_dio_write_loop(dio);
20681c6fdbd8SKent Overstreet err:
20697edcfbfeSKent Overstreet 	if (locked)
20707edcfbfeSKent Overstreet 		inode_unlock(&inode->v);
20717edcfbfeSKent Overstreet 	return ret;
20727edcfbfeSKent Overstreet err_put_bio:
20737edcfbfeSKent Overstreet 	bch2_pagecache_block_put(&inode->ei_pagecache_lock);
20741c6fdbd8SKent Overstreet 	bch2_quota_reservation_put(c, inode, &dio->quota_res);
20751c6fdbd8SKent Overstreet 	bio_put(bio);
20767edcfbfeSKent Overstreet 	inode_dio_end(&inode->v);
20777edcfbfeSKent Overstreet 	goto err;
20781c6fdbd8SKent Overstreet }
20791c6fdbd8SKent Overstreet 
20807edcfbfeSKent Overstreet ssize_t bch2_write_iter(struct kiocb *iocb, struct iov_iter *from)
20811c6fdbd8SKent Overstreet {
20821c6fdbd8SKent Overstreet 	struct file *file = iocb->ki_filp;
20837edcfbfeSKent Overstreet 	struct bch_inode_info *inode = file_bch_inode(file);
20841c6fdbd8SKent Overstreet 	ssize_t	ret;
20851c6fdbd8SKent Overstreet 
20861c6fdbd8SKent Overstreet 	if (iocb->ki_flags & IOCB_DIRECT)
20871c6fdbd8SKent Overstreet 		return bch2_direct_write(iocb, from);
20881c6fdbd8SKent Overstreet 
20897edcfbfeSKent Overstreet 	inode_lock(&inode->v);
20907edcfbfeSKent Overstreet 
20917edcfbfeSKent Overstreet 	ret = generic_write_checks(iocb, from);
20927edcfbfeSKent Overstreet 	if (ret <= 0)
20937edcfbfeSKent Overstreet 		goto unlock;
20947edcfbfeSKent Overstreet 
20951c6fdbd8SKent Overstreet 	ret = file_remove_privs(file);
20961c6fdbd8SKent Overstreet 	if (ret)
20977edcfbfeSKent Overstreet 		goto unlock;
20981c6fdbd8SKent Overstreet 
20991c6fdbd8SKent Overstreet 	ret = file_update_time(file);
21001c6fdbd8SKent Overstreet 	if (ret)
21017edcfbfeSKent Overstreet 		goto unlock;
21021c6fdbd8SKent Overstreet 
21037edcfbfeSKent Overstreet 	ret = bch2_buffered_write(iocb, from);
21041c6fdbd8SKent Overstreet 	if (likely(ret > 0))
21051c6fdbd8SKent Overstreet 		iocb->ki_pos += ret;
21067edcfbfeSKent Overstreet unlock:
21071c6fdbd8SKent Overstreet 	inode_unlock(&inode->v);
21081c6fdbd8SKent Overstreet 
21097edcfbfeSKent Overstreet 	if (ret > 0)
21101c6fdbd8SKent Overstreet 		ret = generic_write_sync(iocb, ret);
21111c6fdbd8SKent Overstreet 
21121c6fdbd8SKent Overstreet 	return ret;
21131c6fdbd8SKent Overstreet }
21141c6fdbd8SKent Overstreet 
21151c6fdbd8SKent Overstreet /* fsync: */
21161c6fdbd8SKent Overstreet 
21171c6fdbd8SKent Overstreet int bch2_fsync(struct file *file, loff_t start, loff_t end, int datasync)
21181c6fdbd8SKent Overstreet {
21191c6fdbd8SKent Overstreet 	struct bch_inode_info *inode = file_bch_inode(file);
21201c6fdbd8SKent Overstreet 	struct bch_fs *c = inode->v.i_sb->s_fs_info;
2121bb1b3658SKent Overstreet 	int ret, ret2;
21221c6fdbd8SKent Overstreet 
21231c6fdbd8SKent Overstreet 	ret = file_write_and_wait_range(file, start, end);
21241c6fdbd8SKent Overstreet 	if (ret)
21251c6fdbd8SKent Overstreet 		return ret;
21261c6fdbd8SKent Overstreet 
21271c6fdbd8SKent Overstreet 	if (datasync && !(inode->v.i_state & I_DIRTY_DATASYNC))
21281c6fdbd8SKent Overstreet 		goto out;
21291c6fdbd8SKent Overstreet 
21301c6fdbd8SKent Overstreet 	ret = sync_inode_metadata(&inode->v, 1);
21311c6fdbd8SKent Overstreet 	if (ret)
21321c6fdbd8SKent Overstreet 		return ret;
21331c6fdbd8SKent Overstreet out:
213475812e70SKent Overstreet 	if (!c->opts.journal_flush_disabled)
213575812e70SKent Overstreet 		ret = bch2_journal_flush_seq(&c->journal,
213675812e70SKent Overstreet 					     inode->ei_journal_seq);
2137bb1b3658SKent Overstreet 	ret2 = file_check_and_advance_wb_err(file);
2138bb1b3658SKent Overstreet 
2139bb1b3658SKent Overstreet 	return ret ?: ret2;
21401c6fdbd8SKent Overstreet }
21411c6fdbd8SKent Overstreet 
21421c6fdbd8SKent Overstreet /* truncate: */
21431c6fdbd8SKent Overstreet 
21441c6fdbd8SKent Overstreet static inline int range_has_data(struct bch_fs *c,
21451c6fdbd8SKent Overstreet 				  struct bpos start,
21461c6fdbd8SKent Overstreet 				  struct bpos end)
21471c6fdbd8SKent Overstreet {
2148424eb881SKent Overstreet 	struct btree_trans trans;
2149*67e0dd8fSKent Overstreet 	struct btree_iter iter;
21501c6fdbd8SKent Overstreet 	struct bkey_s_c k;
21511c6fdbd8SKent Overstreet 	int ret = 0;
21521c6fdbd8SKent Overstreet 
215320bceecbSKent Overstreet 	bch2_trans_init(&trans, c, 0, 0);
2154424eb881SKent Overstreet 
215541f8b09eSKent Overstreet 	for_each_btree_key(&trans, iter, BTREE_ID_extents, start, 0, k, ret) {
21561c6fdbd8SKent Overstreet 		if (bkey_cmp(bkey_start_pos(k.k), end) >= 0)
21571c6fdbd8SKent Overstreet 			break;
21581c6fdbd8SKent Overstreet 
21591c6fdbd8SKent Overstreet 		if (bkey_extent_is_data(k.k)) {
21601c6fdbd8SKent Overstreet 			ret = 1;
21611c6fdbd8SKent Overstreet 			break;
21621c6fdbd8SKent Overstreet 		}
21631c6fdbd8SKent Overstreet 	}
2164*67e0dd8fSKent Overstreet 	bch2_trans_iter_exit(&trans, &iter);
21651c6fdbd8SKent Overstreet 
2166424eb881SKent Overstreet 	return bch2_trans_exit(&trans) ?: ret;
21671c6fdbd8SKent Overstreet }
21681c6fdbd8SKent Overstreet 
21691c6fdbd8SKent Overstreet static int __bch2_truncate_page(struct bch_inode_info *inode,
21701c6fdbd8SKent Overstreet 				pgoff_t index, loff_t start, loff_t end)
21711c6fdbd8SKent Overstreet {
21721c6fdbd8SKent Overstreet 	struct bch_fs *c = inode->v.i_sb->s_fs_info;
21731c6fdbd8SKent Overstreet 	struct address_space *mapping = inode->v.i_mapping;
2174a99b1cafSKent Overstreet 	struct bch_page_state *s;
21751c6fdbd8SKent Overstreet 	unsigned start_offset = start & (PAGE_SIZE - 1);
21761c6fdbd8SKent Overstreet 	unsigned end_offset = ((end - 1) & (PAGE_SIZE - 1)) + 1;
2177a99b1cafSKent Overstreet 	unsigned i;
21781c6fdbd8SKent Overstreet 	struct page *page;
21791c6fdbd8SKent Overstreet 	int ret = 0;
21801c6fdbd8SKent Overstreet 
21811c6fdbd8SKent Overstreet 	/* Page boundary? Nothing to do */
21821c6fdbd8SKent Overstreet 	if (!((index == start >> PAGE_SHIFT && start_offset) ||
21831c6fdbd8SKent Overstreet 	      (index == end >> PAGE_SHIFT && end_offset != PAGE_SIZE)))
21841c6fdbd8SKent Overstreet 		return 0;
21851c6fdbd8SKent Overstreet 
21861c6fdbd8SKent Overstreet 	/* Above i_size? */
21871c6fdbd8SKent Overstreet 	if (index << PAGE_SHIFT >= inode->v.i_size)
21881c6fdbd8SKent Overstreet 		return 0;
21891c6fdbd8SKent Overstreet 
21901c6fdbd8SKent Overstreet 	page = find_lock_page(mapping, index);
21911c6fdbd8SKent Overstreet 	if (!page) {
21921c6fdbd8SKent Overstreet 		/*
21931c6fdbd8SKent Overstreet 		 * XXX: we're doing two index lookups when we end up reading the
21941c6fdbd8SKent Overstreet 		 * page
21951c6fdbd8SKent Overstreet 		 */
21961c6fdbd8SKent Overstreet 		ret = range_has_data(c,
21971c6fdbd8SKent Overstreet 				POS(inode->v.i_ino, index << PAGE_SECTOR_SHIFT),
21981c6fdbd8SKent Overstreet 				POS(inode->v.i_ino, (index + 1) << PAGE_SECTOR_SHIFT));
21991c6fdbd8SKent Overstreet 		if (ret <= 0)
22001c6fdbd8SKent Overstreet 			return ret;
22011c6fdbd8SKent Overstreet 
22021c6fdbd8SKent Overstreet 		page = find_or_create_page(mapping, index, GFP_KERNEL);
22031c6fdbd8SKent Overstreet 		if (unlikely(!page)) {
22041c6fdbd8SKent Overstreet 			ret = -ENOMEM;
22051c6fdbd8SKent Overstreet 			goto out;
22061c6fdbd8SKent Overstreet 		}
22071c6fdbd8SKent Overstreet 	}
22081c6fdbd8SKent Overstreet 
2209a99b1cafSKent Overstreet 	s = bch2_page_state_create(page, 0);
2210a99b1cafSKent Overstreet 	if (!s) {
2211a99b1cafSKent Overstreet 		ret = -ENOMEM;
2212a99b1cafSKent Overstreet 		goto unlock;
2213a99b1cafSKent Overstreet 	}
2214a99b1cafSKent Overstreet 
22151c6fdbd8SKent Overstreet 	if (!PageUptodate(page)) {
22161c6fdbd8SKent Overstreet 		ret = bch2_read_single_page(page, mapping);
22171c6fdbd8SKent Overstreet 		if (ret)
22181c6fdbd8SKent Overstreet 			goto unlock;
22191c6fdbd8SKent Overstreet 	}
22201c6fdbd8SKent Overstreet 
2221a99b1cafSKent Overstreet 	if (index != start >> PAGE_SHIFT)
2222a99b1cafSKent Overstreet 		start_offset = 0;
2223a99b1cafSKent Overstreet 	if (index != end >> PAGE_SHIFT)
2224a99b1cafSKent Overstreet 		end_offset = PAGE_SIZE;
2225a99b1cafSKent Overstreet 
2226a99b1cafSKent Overstreet 	for (i = round_up(start_offset, block_bytes(c)) >> 9;
2227a99b1cafSKent Overstreet 	     i < round_down(end_offset, block_bytes(c)) >> 9;
2228a99b1cafSKent Overstreet 	     i++) {
2229a99b1cafSKent Overstreet 		s->s[i].nr_replicas	= 0;
2230a99b1cafSKent Overstreet 		s->s[i].state		= SECTOR_UNALLOCATED;
2231a99b1cafSKent Overstreet 	}
2232a99b1cafSKent Overstreet 
2233a99b1cafSKent Overstreet 	zero_user_segment(page, start_offset, end_offset);
2234a99b1cafSKent Overstreet 
22351c6fdbd8SKent Overstreet 	/*
22361c6fdbd8SKent Overstreet 	 * Bit of a hack - we don't want truncate to fail due to -ENOSPC.
22371c6fdbd8SKent Overstreet 	 *
22381c6fdbd8SKent Overstreet 	 * XXX: because we aren't currently tracking whether the page has actual
22391c6fdbd8SKent Overstreet 	 * data in it (vs. just 0s, or only partially written) this wrong. ick.
22401c6fdbd8SKent Overstreet 	 */
2241d1542e03SKent Overstreet 	ret = bch2_get_page_disk_reservation(c, inode, page, false);
22421c6fdbd8SKent Overstreet 	BUG_ON(ret);
22431c6fdbd8SKent Overstreet 
22449ba2eb25SKent Overstreet 	/*
22459ba2eb25SKent Overstreet 	 * This removes any writeable userspace mappings; we need to force
22469ba2eb25SKent Overstreet 	 * .page_mkwrite to be called again before any mmapped writes, to
22479ba2eb25SKent Overstreet 	 * redirty the full page:
22489ba2eb25SKent Overstreet 	 */
22499ba2eb25SKent Overstreet 	page_mkclean(page);
2250d1542e03SKent Overstreet 	filemap_dirty_folio(mapping, page_folio(page));
22511c6fdbd8SKent Overstreet unlock:
22521c6fdbd8SKent Overstreet 	unlock_page(page);
22531c6fdbd8SKent Overstreet 	put_page(page);
22541c6fdbd8SKent Overstreet out:
22551c6fdbd8SKent Overstreet 	return ret;
22561c6fdbd8SKent Overstreet }
22571c6fdbd8SKent Overstreet 
22581c6fdbd8SKent Overstreet static int bch2_truncate_page(struct bch_inode_info *inode, loff_t from)
22591c6fdbd8SKent Overstreet {
22601c6fdbd8SKent Overstreet 	return __bch2_truncate_page(inode, from >> PAGE_SHIFT,
2261a99b1cafSKent Overstreet 				    from, round_up(from, PAGE_SIZE));
22621c6fdbd8SKent Overstreet }
22631c6fdbd8SKent Overstreet 
226468a507a2SKent Overstreet static int bch2_extend(struct mnt_idmap *idmap,
226568a507a2SKent Overstreet 		       struct bch_inode_info *inode,
2266e0541a93SKent Overstreet 		       struct bch_inode_unpacked *inode_u,
2267e0541a93SKent Overstreet 		       struct iattr *iattr)
22681c6fdbd8SKent Overstreet {
22691c6fdbd8SKent Overstreet 	struct address_space *mapping = inode->v.i_mapping;
22701c6fdbd8SKent Overstreet 	int ret;
22711c6fdbd8SKent Overstreet 
2272e0541a93SKent Overstreet 	/*
2273e0541a93SKent Overstreet 	 * sync appends:
22742925fc49SKent Overstreet 	 *
22752925fc49SKent Overstreet 	 * this has to be done _before_ extending i_size:
2276e0541a93SKent Overstreet 	 */
2277e0541a93SKent Overstreet 	ret = filemap_write_and_wait_range(mapping, inode_u->bi_size, S64_MAX);
22781c6fdbd8SKent Overstreet 	if (ret)
22791c6fdbd8SKent Overstreet 		return ret;
22801c6fdbd8SKent Overstreet 
22811c6fdbd8SKent Overstreet 	truncate_setsize(&inode->v, iattr->ia_size);
22821c6fdbd8SKent Overstreet 
228368a507a2SKent Overstreet 	return bch2_setattr_nonsize(idmap, inode, iattr);
22841c6fdbd8SKent Overstreet }
22851c6fdbd8SKent Overstreet 
228654e2264eSKent Overstreet static int bch2_truncate_finish_fn(struct bch_inode_info *inode,
228754e2264eSKent Overstreet 				   struct bch_inode_unpacked *bi,
228854e2264eSKent Overstreet 				   void *p)
228954e2264eSKent Overstreet {
229054e2264eSKent Overstreet 	bi->bi_flags &= ~BCH_INODE_I_SIZE_DIRTY;
229154e2264eSKent Overstreet 	return 0;
229254e2264eSKent Overstreet }
229354e2264eSKent Overstreet 
229454e2264eSKent Overstreet static int bch2_truncate_start_fn(struct bch_inode_info *inode,
229554e2264eSKent Overstreet 				  struct bch_inode_unpacked *bi, void *p)
229654e2264eSKent Overstreet {
229754e2264eSKent Overstreet 	u64 *new_i_size = p;
229854e2264eSKent Overstreet 
229954e2264eSKent Overstreet 	bi->bi_flags |= BCH_INODE_I_SIZE_DIRTY;
230054e2264eSKent Overstreet 	bi->bi_size = *new_i_size;
230154e2264eSKent Overstreet 	return 0;
230254e2264eSKent Overstreet }
230354e2264eSKent Overstreet 
230468a507a2SKent Overstreet int bch2_truncate(struct mnt_idmap *idmap,
230568a507a2SKent Overstreet 		  struct bch_inode_info *inode, struct iattr *iattr)
23061c6fdbd8SKent Overstreet {
23071c6fdbd8SKent Overstreet 	struct bch_fs *c = inode->v.i_sb->s_fs_info;
23081c6fdbd8SKent Overstreet 	struct address_space *mapping = inode->v.i_mapping;
2309e0541a93SKent Overstreet 	struct bch_inode_unpacked inode_u;
231054e2264eSKent Overstreet 	u64 new_i_size = iattr->ia_size;
23112e87eae1SKent Overstreet 	s64 i_sectors_delta = 0;
23121c6fdbd8SKent Overstreet 	int ret = 0;
23131c6fdbd8SKent Overstreet 
231468a507a2SKent Overstreet 	/*
231578d66ab1SDan Robertson 	 * If the truncate call with change the size of the file, the
231678d66ab1SDan Robertson 	 * cmtimes should be updated. If the size will not change, we
231778d66ab1SDan Robertson 	 * do not need to update the cmtimes.
231868a507a2SKent Overstreet 	 */
231978d66ab1SDan Robertson 	if (iattr->ia_size != inode->v.i_size) {
232068a507a2SKent Overstreet 		if (!(iattr->ia_valid & ATTR_MTIME))
232168a507a2SKent Overstreet 			ktime_get_coarse_real_ts64(&iattr->ia_mtime);
232268a507a2SKent Overstreet 		if (!(iattr->ia_valid & ATTR_CTIME))
232368a507a2SKent Overstreet 			ktime_get_coarse_real_ts64(&iattr->ia_ctime);
232468a507a2SKent Overstreet 		iattr->ia_valid |= ATTR_MTIME|ATTR_CTIME;
232578d66ab1SDan Robertson 	}
232668a507a2SKent Overstreet 
23271c6fdbd8SKent Overstreet 	inode_dio_wait(&inode->v);
23281c6fdbd8SKent Overstreet 	bch2_pagecache_block_get(&inode->ei_pagecache_lock);
23291c6fdbd8SKent Overstreet 
2330b97bbd4eSKent Overstreet 	ret = bch2_inode_find_by_inum(c, inode->v.i_ino, &inode_u);
2331e0541a93SKent Overstreet 	if (ret)
2332e0541a93SKent Overstreet 		goto err;
23331c6fdbd8SKent Overstreet 
2334c45d473dSKent Overstreet 	/*
2335c45d473dSKent Overstreet 	 * check this before next assertion; on filesystem error our normal
2336c45d473dSKent Overstreet 	 * invariants are a bit broken (truncate has to truncate the page cache
2337c45d473dSKent Overstreet 	 * before the inode).
2338c45d473dSKent Overstreet 	 */
2339c45d473dSKent Overstreet 	ret = bch2_journal_error(&c->journal);
2340c45d473dSKent Overstreet 	if (ret)
2341c45d473dSKent Overstreet 		goto err;
2342c45d473dSKent Overstreet 
234333c74e41SKent Overstreet 	WARN_ON(!test_bit(EI_INODE_ERROR, &inode->ei_flags) &&
234433c74e41SKent Overstreet 		inode->v.i_size < inode_u.bi_size);
2345e0541a93SKent Overstreet 
2346e0541a93SKent Overstreet 	if (iattr->ia_size > inode->v.i_size) {
234768a507a2SKent Overstreet 		ret = bch2_extend(idmap, inode, &inode_u, iattr);
234854e2264eSKent Overstreet 		goto err;
23491c6fdbd8SKent Overstreet 	}
23501c6fdbd8SKent Overstreet 
235168a507a2SKent Overstreet 	iattr->ia_valid &= ~ATTR_SIZE;
235268a507a2SKent Overstreet 
23531c6fdbd8SKent Overstreet 	ret = bch2_truncate_page(inode, iattr->ia_size);
23541c6fdbd8SKent Overstreet 	if (unlikely(ret))
235554e2264eSKent Overstreet 		goto err;
23561c6fdbd8SKent Overstreet 
23576cc3535dSKent Overstreet 	/*
23586cc3535dSKent Overstreet 	 * When extending, we're going to write the new i_size to disk
23596cc3535dSKent Overstreet 	 * immediately so we need to flush anything above the current on disk
23606cc3535dSKent Overstreet 	 * i_size first:
23616cc3535dSKent Overstreet 	 *
23626cc3535dSKent Overstreet 	 * Also, when extending we need to flush the page that i_size currently
23636cc3535dSKent Overstreet 	 * straddles - if it's mapped to userspace, we need to ensure that
23646cc3535dSKent Overstreet 	 * userspace has to redirty it and call .mkwrite -> set_page_dirty
23656cc3535dSKent Overstreet 	 * again to allocate the part of the page that was extended.
23666cc3535dSKent Overstreet 	 */
2367e0541a93SKent Overstreet 	if (iattr->ia_size > inode_u.bi_size)
23681c6fdbd8SKent Overstreet 		ret = filemap_write_and_wait_range(mapping,
2369e0541a93SKent Overstreet 				inode_u.bi_size,
23701c6fdbd8SKent Overstreet 				iattr->ia_size - 1);
23711c6fdbd8SKent Overstreet 	else if (iattr->ia_size & (PAGE_SIZE - 1))
23721c6fdbd8SKent Overstreet 		ret = filemap_write_and_wait_range(mapping,
23731c6fdbd8SKent Overstreet 				round_down(iattr->ia_size, PAGE_SIZE),
23741c6fdbd8SKent Overstreet 				iattr->ia_size - 1);
23751c6fdbd8SKent Overstreet 	if (ret)
237654e2264eSKent Overstreet 		goto err;
23771c6fdbd8SKent Overstreet 
237854e2264eSKent Overstreet 	mutex_lock(&inode->ei_update_lock);
237954e2264eSKent Overstreet 	ret = bch2_write_inode(c, inode, bch2_truncate_start_fn,
238054e2264eSKent Overstreet 			       &new_i_size, 0);
238154e2264eSKent Overstreet 	mutex_unlock(&inode->ei_update_lock);
23821c6fdbd8SKent Overstreet 
23831c6fdbd8SKent Overstreet 	if (unlikely(ret))
238454e2264eSKent Overstreet 		goto err;
23851c6fdbd8SKent Overstreet 
23861c6fdbd8SKent Overstreet 	truncate_setsize(&inode->v, iattr->ia_size);
23871c6fdbd8SKent Overstreet 
23882e87eae1SKent Overstreet 	ret = bch2_fpunch(c, inode->v.i_ino,
2389a99b1cafSKent Overstreet 			round_up(iattr->ia_size, block_bytes(c)) >> 9,
23902e87eae1SKent Overstreet 			U64_MAX, &inode->ei_journal_seq, &i_sectors_delta);
23912e87eae1SKent Overstreet 	i_sectors_acct(c, inode, NULL, i_sectors_delta);
23922e87eae1SKent Overstreet 
23931c6fdbd8SKent Overstreet 	if (unlikely(ret))
239454e2264eSKent Overstreet 		goto err;
23951c6fdbd8SKent Overstreet 
239654e2264eSKent Overstreet 	mutex_lock(&inode->ei_update_lock);
239768a507a2SKent Overstreet 	ret = bch2_write_inode(c, inode, bch2_truncate_finish_fn, NULL, 0);
239854e2264eSKent Overstreet 	mutex_unlock(&inode->ei_update_lock);
239968a507a2SKent Overstreet 
240068a507a2SKent Overstreet 	ret = bch2_setattr_nonsize(idmap, inode, iattr);
240154e2264eSKent Overstreet err:
24021c6fdbd8SKent Overstreet 	bch2_pagecache_block_put(&inode->ei_pagecache_lock);
24031c6fdbd8SKent Overstreet 	return ret;
24041c6fdbd8SKent Overstreet }
24051c6fdbd8SKent Overstreet 
24061c6fdbd8SKent Overstreet /* fallocate: */
24071c6fdbd8SKent Overstreet 
2408050197b1SKent Overstreet static int inode_update_times_fn(struct bch_inode_info *inode,
2409050197b1SKent Overstreet 				 struct bch_inode_unpacked *bi, void *p)
2410050197b1SKent Overstreet {
2411050197b1SKent Overstreet 	struct bch_fs *c = inode->v.i_sb->s_fs_info;
2412050197b1SKent Overstreet 
2413050197b1SKent Overstreet 	bi->bi_mtime = bi->bi_ctime = bch2_current_time(c);
2414050197b1SKent Overstreet 	return 0;
2415050197b1SKent Overstreet }
2416050197b1SKent Overstreet 
24172e87eae1SKent Overstreet static long bchfs_fpunch(struct bch_inode_info *inode, loff_t offset, loff_t len)
24181c6fdbd8SKent Overstreet {
24191c6fdbd8SKent Overstreet 	struct bch_fs *c = inode->v.i_sb->s_fs_info;
2420a99b1cafSKent Overstreet 	u64 discard_start = round_up(offset, block_bytes(c)) >> 9;
2421a99b1cafSKent Overstreet 	u64 discard_end = round_down(offset + len, block_bytes(c)) >> 9;
24221c6fdbd8SKent Overstreet 	int ret = 0;
24231c6fdbd8SKent Overstreet 
24241c6fdbd8SKent Overstreet 	inode_lock(&inode->v);
24251c6fdbd8SKent Overstreet 	inode_dio_wait(&inode->v);
24261c6fdbd8SKent Overstreet 	bch2_pagecache_block_get(&inode->ei_pagecache_lock);
24271c6fdbd8SKent Overstreet 
24281c6fdbd8SKent Overstreet 	ret = __bch2_truncate_page(inode,
24291c6fdbd8SKent Overstreet 				   offset >> PAGE_SHIFT,
24301c6fdbd8SKent Overstreet 				   offset, offset + len);
24311c6fdbd8SKent Overstreet 	if (unlikely(ret))
24321c6fdbd8SKent Overstreet 		goto err;
24331c6fdbd8SKent Overstreet 
24341c6fdbd8SKent Overstreet 	if (offset >> PAGE_SHIFT !=
24351c6fdbd8SKent Overstreet 	    (offset + len) >> PAGE_SHIFT) {
24361c6fdbd8SKent Overstreet 		ret = __bch2_truncate_page(inode,
24371c6fdbd8SKent Overstreet 					   (offset + len) >> PAGE_SHIFT,
24381c6fdbd8SKent Overstreet 					   offset, offset + len);
24391c6fdbd8SKent Overstreet 		if (unlikely(ret))
24401c6fdbd8SKent Overstreet 			goto err;
24411c6fdbd8SKent Overstreet 	}
24421c6fdbd8SKent Overstreet 
24431c6fdbd8SKent Overstreet 	truncate_pagecache_range(&inode->v, offset, offset + len - 1);
24441c6fdbd8SKent Overstreet 
24452e87eae1SKent Overstreet 	if (discard_start < discard_end) {
24462e87eae1SKent Overstreet 		s64 i_sectors_delta = 0;
24472e87eae1SKent Overstreet 
24482e87eae1SKent Overstreet 		ret = bch2_fpunch(c, inode->v.i_ino,
24492e87eae1SKent Overstreet 				  discard_start, discard_end,
24502e87eae1SKent Overstreet 				  &inode->ei_journal_seq,
24512e87eae1SKent Overstreet 				  &i_sectors_delta);
24522e87eae1SKent Overstreet 		i_sectors_acct(c, inode, NULL, i_sectors_delta);
24532e87eae1SKent Overstreet 	}
2454050197b1SKent Overstreet 
2455050197b1SKent Overstreet 	mutex_lock(&inode->ei_update_lock);
2456050197b1SKent Overstreet 	ret = bch2_write_inode(c, inode, inode_update_times_fn, NULL,
2457050197b1SKent Overstreet 			       ATTR_MTIME|ATTR_CTIME) ?: ret;
2458050197b1SKent Overstreet 	mutex_unlock(&inode->ei_update_lock);
24591c6fdbd8SKent Overstreet err:
24601c6fdbd8SKent Overstreet 	bch2_pagecache_block_put(&inode->ei_pagecache_lock);
24611c6fdbd8SKent Overstreet 	inode_unlock(&inode->v);
24621c6fdbd8SKent Overstreet 
24631c6fdbd8SKent Overstreet 	return ret;
24641c6fdbd8SKent Overstreet }
24651c6fdbd8SKent Overstreet 
24662e87eae1SKent Overstreet static long bchfs_fcollapse_finsert(struct bch_inode_info *inode,
24675f786787SKent Overstreet 				   loff_t offset, loff_t len,
24685f786787SKent Overstreet 				   bool insert)
24691c6fdbd8SKent Overstreet {
24701c6fdbd8SKent Overstreet 	struct bch_fs *c = inode->v.i_sb->s_fs_info;
24711c6fdbd8SKent Overstreet 	struct address_space *mapping = inode->v.i_mapping;
247207a1006aSKent Overstreet 	struct bkey_buf copy;
2473d69f41d6SKent Overstreet 	struct btree_trans trans;
2474*67e0dd8fSKent Overstreet 	struct btree_iter src, dst, del;
24755f786787SKent Overstreet 	loff_t shift, new_size;
24765f786787SKent Overstreet 	u64 src_start;
247750dc0f69SKent Overstreet 	int ret = 0;
24781c6fdbd8SKent Overstreet 
24791c6fdbd8SKent Overstreet 	if ((offset | len) & (block_bytes(c) - 1))
24801c6fdbd8SKent Overstreet 		return -EINVAL;
24811c6fdbd8SKent Overstreet 
24821c6fdbd8SKent Overstreet 	/*
24831c6fdbd8SKent Overstreet 	 * We need i_mutex to keep the page cache consistent with the extents
24841c6fdbd8SKent Overstreet 	 * btree, and the btree consistent with i_size - we don't need outside
24851c6fdbd8SKent Overstreet 	 * locking for the extents btree itself, because we're using linked
24861c6fdbd8SKent Overstreet 	 * iterators
24871c6fdbd8SKent Overstreet 	 */
24881c6fdbd8SKent Overstreet 	inode_lock(&inode->v);
24891c6fdbd8SKent Overstreet 	inode_dio_wait(&inode->v);
24901c6fdbd8SKent Overstreet 	bch2_pagecache_block_get(&inode->ei_pagecache_lock);
24911c6fdbd8SKent Overstreet 
24925f786787SKent Overstreet 	if (insert) {
24935f786787SKent Overstreet 		ret = -EFBIG;
24945f786787SKent Overstreet 		if (inode->v.i_sb->s_maxbytes - inode->v.i_size < len)
24955f786787SKent Overstreet 			goto err;
24965f786787SKent Overstreet 
24975f786787SKent Overstreet 		ret = -EINVAL;
24985f786787SKent Overstreet 		if (offset >= inode->v.i_size)
24995f786787SKent Overstreet 			goto err;
25005f786787SKent Overstreet 
25015f786787SKent Overstreet 		src_start	= U64_MAX;
25025f786787SKent Overstreet 		shift		= len;
25035f786787SKent Overstreet 	} else {
25041c6fdbd8SKent Overstreet 		ret = -EINVAL;
25051c6fdbd8SKent Overstreet 		if (offset + len >= inode->v.i_size)
25061c6fdbd8SKent Overstreet 			goto err;
25071c6fdbd8SKent Overstreet 
25085f786787SKent Overstreet 		src_start	= offset + len;
25095f786787SKent Overstreet 		shift		= -len;
25105f786787SKent Overstreet 	}
25111c6fdbd8SKent Overstreet 
25125f786787SKent Overstreet 	new_size = inode->v.i_size + shift;
25131c6fdbd8SKent Overstreet 
25141c6fdbd8SKent Overstreet 	ret = write_invalidate_inode_pages_range(mapping, offset, LLONG_MAX);
25151c6fdbd8SKent Overstreet 	if (ret)
25161c6fdbd8SKent Overstreet 		goto err;
25171c6fdbd8SKent Overstreet 
25185f786787SKent Overstreet 	if (insert) {
25195f786787SKent Overstreet 		i_size_write(&inode->v, new_size);
25205f786787SKent Overstreet 		mutex_lock(&inode->ei_update_lock);
25215f786787SKent Overstreet 		ret = bch2_write_inode_size(c, inode, new_size,
25225f786787SKent Overstreet 					    ATTR_MTIME|ATTR_CTIME);
25235f786787SKent Overstreet 		mutex_unlock(&inode->ei_update_lock);
25245f786787SKent Overstreet 	} else {
25252e87eae1SKent Overstreet 		s64 i_sectors_delta = 0;
25262e87eae1SKent Overstreet 
25272e87eae1SKent Overstreet 		ret = bch2_fpunch(c, inode->v.i_ino,
25282e87eae1SKent Overstreet 				  offset >> 9, (offset + len) >> 9,
25292e87eae1SKent Overstreet 				  &inode->ei_journal_seq,
25302e87eae1SKent Overstreet 				  &i_sectors_delta);
25312e87eae1SKent Overstreet 		i_sectors_acct(c, inode, NULL, i_sectors_delta);
25322e87eae1SKent Overstreet 
253363095894SKent Overstreet 		if (ret)
253463095894SKent Overstreet 			goto err;
25355f786787SKent Overstreet 	}
25368ef231bdSKent Overstreet 
253750dc0f69SKent Overstreet 	bch2_bkey_buf_init(&copy);
2538f7beb4caSKent Overstreet 	bch2_trans_init(&trans, c, BTREE_ITER_MAX, 1024);
2539*67e0dd8fSKent Overstreet 	bch2_trans_iter_init(&trans, &src, BTREE_ID_extents,
25405f786787SKent Overstreet 			POS(inode->v.i_ino, src_start >> 9),
254163095894SKent Overstreet 			BTREE_ITER_INTENT);
2542*67e0dd8fSKent Overstreet 	bch2_trans_copy_iter(&dst, &src);
2543*67e0dd8fSKent Overstreet 	bch2_trans_copy_iter(&del, &src);
25445f786787SKent Overstreet 
254550dc0f69SKent Overstreet 	while (ret == 0 || ret == -EINTR) {
254663095894SKent Overstreet 		struct disk_reservation disk_res =
254763095894SKent Overstreet 			bch2_disk_reservation_init(c, 0);
254863095894SKent Overstreet 		struct bkey_i delete;
254963095894SKent Overstreet 		struct bkey_s_c k;
255063095894SKent Overstreet 		struct bpos next_pos;
25515f786787SKent Overstreet 		struct bpos move_pos = POS(inode->v.i_ino, offset >> 9);
25525f786787SKent Overstreet 		struct bpos atomic_end;
25532d594dfbSKent Overstreet 		unsigned trigger_flags = 0;
255463095894SKent Overstreet 
2555700c25b3SKent Overstreet 		bch2_trans_begin(&trans);
2556700c25b3SKent Overstreet 
25575f786787SKent Overstreet 		k = insert
2558*67e0dd8fSKent Overstreet 			? bch2_btree_iter_peek_prev(&src)
2559*67e0dd8fSKent Overstreet 			: bch2_btree_iter_peek(&src);
256063095894SKent Overstreet 		if ((ret = bkey_err(k)))
256150dc0f69SKent Overstreet 			continue;
256263095894SKent Overstreet 
256363095894SKent Overstreet 		if (!k.k || k.k->p.inode != inode->v.i_ino)
256463095894SKent Overstreet 			break;
256563095894SKent Overstreet 
25665f786787SKent Overstreet 		if (insert &&
25675f786787SKent Overstreet 		    bkey_cmp(k.k->p, POS(inode->v.i_ino, offset >> 9)) <= 0)
25685f786787SKent Overstreet 			break;
25695f786787SKent Overstreet reassemble:
257007a1006aSKent Overstreet 		bch2_bkey_buf_reassemble(&copy, c, k);
25715f786787SKent Overstreet 
25725f786787SKent Overstreet 		if (insert &&
2573283eda57SKent Overstreet 		    bkey_cmp(bkey_start_pos(k.k), move_pos) < 0)
257435189e09SKent Overstreet 			bch2_cut_front(move_pos, copy.k);
25755f786787SKent Overstreet 
257635189e09SKent Overstreet 		copy.k->k.p.offset += shift >> 9;
2577*67e0dd8fSKent Overstreet 		bch2_btree_iter_set_pos(&dst, bkey_start_pos(&copy.k->k));
25781c6fdbd8SKent Overstreet 
2579*67e0dd8fSKent Overstreet 		ret = bch2_extent_atomic_end(&trans, &dst, copy.k, &atomic_end);
25803c7f3b7aSKent Overstreet 		if (ret)
258150dc0f69SKent Overstreet 			continue;
2582e2d9912cSKent Overstreet 
258335189e09SKent Overstreet 		if (bkey_cmp(atomic_end, copy.k->k.p)) {
25845f786787SKent Overstreet 			if (insert) {
25855f786787SKent Overstreet 				move_pos = atomic_end;
25865f786787SKent Overstreet 				move_pos.offset -= shift >> 9;
25875f786787SKent Overstreet 				goto reassemble;
25885f786787SKent Overstreet 			} else {
2589085ab693SKent Overstreet 				bch2_cut_back(atomic_end, copy.k);
25905f786787SKent Overstreet 			}
25915f786787SKent Overstreet 		}
25925f786787SKent Overstreet 
259363095894SKent Overstreet 		bkey_init(&delete.k);
2594283eda57SKent Overstreet 		delete.k.p = copy.k->k.p;
2595283eda57SKent Overstreet 		delete.k.size = copy.k->k.size;
2596283eda57SKent Overstreet 		delete.k.p.offset -= shift >> 9;
2597*67e0dd8fSKent Overstreet 		bch2_btree_iter_set_pos(&del, bkey_start_pos(&delete.k));
25981c6fdbd8SKent Overstreet 
25995f786787SKent Overstreet 		next_pos = insert ? bkey_start_pos(&delete.k) : delete.k.p;
260063095894SKent Overstreet 
260135189e09SKent Overstreet 		if (copy.k->k.size == k.k->size) {
260263095894SKent Overstreet 			/*
260363095894SKent Overstreet 			 * If we're moving the entire extent, we can skip
260463095894SKent Overstreet 			 * running triggers:
260563095894SKent Overstreet 			 */
26062d594dfbSKent Overstreet 			trigger_flags |= BTREE_TRIGGER_NORUN;
260763095894SKent Overstreet 		} else {
260863095894SKent Overstreet 			/* We might end up splitting compressed extents: */
260963095894SKent Overstreet 			unsigned nr_ptrs =
26104de77495SKent Overstreet 				bch2_bkey_nr_ptrs_allocated(bkey_i_to_s_c(copy.k));
261163095894SKent Overstreet 
261263095894SKent Overstreet 			ret = bch2_disk_reservation_get(c, &disk_res,
261335189e09SKent Overstreet 					copy.k->k.size, nr_ptrs,
26141c6fdbd8SKent Overstreet 					BCH_DISK_RESERVATION_NOFAIL);
26151c6fdbd8SKent Overstreet 			BUG_ON(ret);
261663095894SKent Overstreet 		}
26171c6fdbd8SKent Overstreet 
2618*67e0dd8fSKent Overstreet 		ret =   bch2_btree_iter_traverse(&del) ?:
2619*67e0dd8fSKent Overstreet 			bch2_trans_update(&trans, &del, &delete, trigger_flags) ?:
2620*67e0dd8fSKent Overstreet 			bch2_trans_update(&trans, &dst, copy.k, trigger_flags) ?:
262124326cd1SKent Overstreet 			bch2_trans_commit(&trans, &disk_res,
262263095894SKent Overstreet 					  &inode->ei_journal_seq,
26232d594dfbSKent Overstreet 					  BTREE_INSERT_NOFAIL);
26241c6fdbd8SKent Overstreet 		bch2_disk_reservation_put(c, &disk_res);
262550dc0f69SKent Overstreet 
262663095894SKent Overstreet 		if (!ret)
2627*67e0dd8fSKent Overstreet 			bch2_btree_iter_set_pos(&src, next_pos);
262850dc0f69SKent Overstreet 	}
2629*67e0dd8fSKent Overstreet 	bch2_trans_iter_exit(&trans, &del);
2630*67e0dd8fSKent Overstreet 	bch2_trans_iter_exit(&trans, &dst);
2631*67e0dd8fSKent Overstreet 	bch2_trans_iter_exit(&trans, &src);
263250dc0f69SKent Overstreet 	bch2_trans_exit(&trans);
263350dc0f69SKent Overstreet 	bch2_bkey_buf_exit(&copy, c);
263463095894SKent Overstreet 
26358ef231bdSKent Overstreet 	if (ret)
26368ef231bdSKent Overstreet 		goto err;
26371c6fdbd8SKent Overstreet 
26385f786787SKent Overstreet 	if (!insert) {
26398ef231bdSKent Overstreet 		i_size_write(&inode->v, new_size);
26408ef231bdSKent Overstreet 		mutex_lock(&inode->ei_update_lock);
26418ef231bdSKent Overstreet 		ret = bch2_write_inode_size(c, inode, new_size,
26428ef231bdSKent Overstreet 					    ATTR_MTIME|ATTR_CTIME);
26438ef231bdSKent Overstreet 		mutex_unlock(&inode->ei_update_lock);
26445f786787SKent Overstreet 	}
26451c6fdbd8SKent Overstreet err:
26461c6fdbd8SKent Overstreet 	bch2_pagecache_block_put(&inode->ei_pagecache_lock);
26471c6fdbd8SKent Overstreet 	inode_unlock(&inode->v);
26481c6fdbd8SKent Overstreet 	return ret;
26491c6fdbd8SKent Overstreet }
26501c6fdbd8SKent Overstreet 
2651694015c2SKent Overstreet static int __bchfs_fallocate(struct bch_inode_info *inode, int mode,
2652694015c2SKent Overstreet 			     u64 start_sector, u64 end_sector)
26531c6fdbd8SKent Overstreet {
26541c6fdbd8SKent Overstreet 	struct bch_fs *c = inode->v.i_sb->s_fs_info;
2655190fa7afSKent Overstreet 	struct btree_trans trans;
2656*67e0dd8fSKent Overstreet 	struct btree_iter iter;
2657694015c2SKent Overstreet 	struct bpos end_pos = POS(inode->v.i_ino, end_sector);
26589a3df993SKent Overstreet 	unsigned replicas = io_opts(c, &inode->ei_inode).data_replicas;
2659694015c2SKent Overstreet 	int ret = 0;
26601c6fdbd8SKent Overstreet 
2661f7beb4caSKent Overstreet 	bch2_trans_init(&trans, c, BTREE_ITER_MAX, 512);
26621c6fdbd8SKent Overstreet 
2663*67e0dd8fSKent Overstreet 	bch2_trans_iter_init(&trans, &iter, BTREE_ID_extents,
2664694015c2SKent Overstreet 			POS(inode->v.i_ino, start_sector),
2665190fa7afSKent Overstreet 			BTREE_ITER_SLOTS|BTREE_ITER_INTENT);
26661c6fdbd8SKent Overstreet 
2667*67e0dd8fSKent Overstreet 	while (!ret && bkey_cmp(iter.pos, end_pos) < 0) {
26682e87eae1SKent Overstreet 		s64 i_sectors_delta = 0;
26691c6fdbd8SKent Overstreet 		struct disk_reservation disk_res = { 0 };
2670190fa7afSKent Overstreet 		struct quota_res quota_res = { 0 };
26711c6fdbd8SKent Overstreet 		struct bkey_i_reservation reservation;
26721c6fdbd8SKent Overstreet 		struct bkey_s_c k;
2673694015c2SKent Overstreet 		unsigned sectors;
26741c6fdbd8SKent Overstreet 
2675163e885aSKent Overstreet 		bch2_trans_begin(&trans);
2676a8abd3a7SKent Overstreet 
2677*67e0dd8fSKent Overstreet 		k = bch2_btree_iter_peek_slot(&iter);
26780f238367SKent Overstreet 		if ((ret = bkey_err(k)))
26790f238367SKent Overstreet 			goto bkey_err;
26801c6fdbd8SKent Overstreet 
26811c6fdbd8SKent Overstreet 		/* already reserved */
268226609b61SKent Overstreet 		if (k.k->type == KEY_TYPE_reservation &&
26831c6fdbd8SKent Overstreet 		    bkey_s_c_to_reservation(k).v->nr_replicas >= replicas) {
2684*67e0dd8fSKent Overstreet 			bch2_btree_iter_advance(&iter);
26851c6fdbd8SKent Overstreet 			continue;
26861c6fdbd8SKent Overstreet 		}
26871c6fdbd8SKent Overstreet 
2688190fa7afSKent Overstreet 		if (bkey_extent_is_data(k.k) &&
2689190fa7afSKent Overstreet 		    !(mode & FALLOC_FL_ZERO_RANGE)) {
2690*67e0dd8fSKent Overstreet 			bch2_btree_iter_advance(&iter);
26911c6fdbd8SKent Overstreet 			continue;
26921c6fdbd8SKent Overstreet 		}
26931c6fdbd8SKent Overstreet 
26941c6fdbd8SKent Overstreet 		bkey_reservation_init(&reservation.k_i);
269526609b61SKent Overstreet 		reservation.k.type	= KEY_TYPE_reservation;
26961c6fdbd8SKent Overstreet 		reservation.k.p		= k.k->p;
26971c6fdbd8SKent Overstreet 		reservation.k.size	= k.k->size;
26981c6fdbd8SKent Overstreet 
2699*67e0dd8fSKent Overstreet 		bch2_cut_front(iter.pos,	&reservation.k_i);
2700085ab693SKent Overstreet 		bch2_cut_back(end_pos,		&reservation.k_i);
27011c6fdbd8SKent Overstreet 
27021c6fdbd8SKent Overstreet 		sectors = reservation.k.size;
27034de77495SKent Overstreet 		reservation.v.nr_replicas = bch2_bkey_nr_ptrs_allocated(k);
27041c6fdbd8SKent Overstreet 
27051c6fdbd8SKent Overstreet 		if (!bkey_extent_is_allocation(k.k)) {
27061c6fdbd8SKent Overstreet 			ret = bch2_quota_reservation_add(c, inode,
2707190fa7afSKent Overstreet 					&quota_res,
27081c6fdbd8SKent Overstreet 					sectors, true);
27091c6fdbd8SKent Overstreet 			if (unlikely(ret))
27100f238367SKent Overstreet 				goto bkey_err;
27111c6fdbd8SKent Overstreet 		}
27121c6fdbd8SKent Overstreet 
27131c6fdbd8SKent Overstreet 		if (reservation.v.nr_replicas < replicas ||
27144de77495SKent Overstreet 		    bch2_bkey_sectors_compressed(k)) {
27151c6fdbd8SKent Overstreet 			ret = bch2_disk_reservation_get(c, &disk_res, sectors,
27161c6fdbd8SKent Overstreet 							replicas, 0);
27171c6fdbd8SKent Overstreet 			if (unlikely(ret))
27180f238367SKent Overstreet 				goto bkey_err;
27191c6fdbd8SKent Overstreet 
27201c6fdbd8SKent Overstreet 			reservation.v.nr_replicas = disk_res.nr_replicas;
27211c6fdbd8SKent Overstreet 		}
27221c6fdbd8SKent Overstreet 
2723*67e0dd8fSKent Overstreet 		ret = bch2_extent_update(&trans, &iter, &reservation.k_i,
27242e87eae1SKent Overstreet 				&disk_res, &inode->ei_journal_seq,
2725a6336910SKent Overstreet 				0, &i_sectors_delta, true);
27262e87eae1SKent Overstreet 		i_sectors_acct(c, inode, &quota_res, i_sectors_delta);
27270f238367SKent Overstreet bkey_err:
2728190fa7afSKent Overstreet 		bch2_quota_reservation_put(c, inode, &quota_res);
27291c6fdbd8SKent Overstreet 		bch2_disk_reservation_put(c, &disk_res);
27301c6fdbd8SKent Overstreet 		if (ret == -EINTR)
27311c6fdbd8SKent Overstreet 			ret = 0;
273250dc0f69SKent Overstreet 	}
2733*67e0dd8fSKent Overstreet 	bch2_trans_iter_exit(&trans, &iter);
2734694015c2SKent Overstreet 	bch2_trans_exit(&trans);
2735694015c2SKent Overstreet 	return ret;
2736694015c2SKent Overstreet }
273750dc0f69SKent Overstreet 
2738694015c2SKent Overstreet static long bchfs_fallocate(struct bch_inode_info *inode, int mode,
2739694015c2SKent Overstreet 			    loff_t offset, loff_t len)
2740694015c2SKent Overstreet {
2741694015c2SKent Overstreet 	struct address_space *mapping = inode->v.i_mapping;
2742694015c2SKent Overstreet 	struct bch_fs *c = inode->v.i_sb->s_fs_info;
2743694015c2SKent Overstreet 	loff_t end		= offset + len;
2744694015c2SKent Overstreet 	loff_t block_start	= round_down(offset,	block_bytes(c));
2745694015c2SKent Overstreet 	loff_t block_end	= round_up(end,		block_bytes(c));
2746694015c2SKent Overstreet 	int ret;
2747694015c2SKent Overstreet 
2748694015c2SKent Overstreet 	inode_lock(&inode->v);
2749694015c2SKent Overstreet 	inode_dio_wait(&inode->v);
2750694015c2SKent Overstreet 	bch2_pagecache_block_get(&inode->ei_pagecache_lock);
2751694015c2SKent Overstreet 
2752694015c2SKent Overstreet 	if (!(mode & FALLOC_FL_KEEP_SIZE) && end > inode->v.i_size) {
2753694015c2SKent Overstreet 		ret = inode_newsize_ok(&inode->v, end);
2754694015c2SKent Overstreet 		if (ret)
2755694015c2SKent Overstreet 			goto err;
2756694015c2SKent Overstreet 	}
2757694015c2SKent Overstreet 
2758694015c2SKent Overstreet 	if (mode & FALLOC_FL_ZERO_RANGE) {
2759694015c2SKent Overstreet 		ret = __bch2_truncate_page(inode,
2760694015c2SKent Overstreet 					   offset >> PAGE_SHIFT,
2761694015c2SKent Overstreet 					   offset, end);
2762694015c2SKent Overstreet 
2763694015c2SKent Overstreet 		if (!ret &&
2764694015c2SKent Overstreet 		    offset >> PAGE_SHIFT != end >> PAGE_SHIFT)
2765694015c2SKent Overstreet 			ret = __bch2_truncate_page(inode,
2766694015c2SKent Overstreet 						   end >> PAGE_SHIFT,
2767694015c2SKent Overstreet 						   offset, end);
2768694015c2SKent Overstreet 
2769694015c2SKent Overstreet 		if (unlikely(ret))
2770694015c2SKent Overstreet 			goto err;
2771694015c2SKent Overstreet 
2772694015c2SKent Overstreet 		truncate_pagecache_range(&inode->v, offset, end - 1);
2773694015c2SKent Overstreet 	}
2774694015c2SKent Overstreet 
2775694015c2SKent Overstreet 	ret = __bchfs_fallocate(inode, mode, block_start >> 9, block_end >> 9);
2776190fa7afSKent Overstreet 	if (ret)
2777190fa7afSKent Overstreet 		goto err;
2778e0541a93SKent Overstreet 
2779e0541a93SKent Overstreet 	/*
2780e0541a93SKent Overstreet 	 * Do we need to extend the file?
2781e0541a93SKent Overstreet 	 *
2782e0541a93SKent Overstreet 	 * If we zeroed up to the end of the file, we dropped whatever writes
2783e0541a93SKent Overstreet 	 * were going to write out the current i_size, so we have to extend
2784e0541a93SKent Overstreet 	 * manually even if FL_KEEP_SIZE was set:
2785e0541a93SKent Overstreet 	 */
2786e0541a93SKent Overstreet 	if (end >= inode->v.i_size &&
2787e0541a93SKent Overstreet 	    (!(mode & FALLOC_FL_KEEP_SIZE) ||
2788e0541a93SKent Overstreet 	     (mode & FALLOC_FL_ZERO_RANGE))) {
27891c6fdbd8SKent Overstreet 
2790e0541a93SKent Overstreet 		/*
2791e0541a93SKent Overstreet 		 * Sync existing appends before extending i_size,
2792e0541a93SKent Overstreet 		 * as in bch2_extend():
2793e0541a93SKent Overstreet 		 */
2794e0541a93SKent Overstreet 		ret = filemap_write_and_wait_range(mapping,
2795694015c2SKent Overstreet 					inode->ei_inode.bi_size, S64_MAX);
2796e0541a93SKent Overstreet 		if (ret)
2797e0541a93SKent Overstreet 			goto err;
2798e0541a93SKent Overstreet 
2799e0541a93SKent Overstreet 		if (mode & FALLOC_FL_KEEP_SIZE)
2800e0541a93SKent Overstreet 			end = inode->v.i_size;
2801e0541a93SKent Overstreet 		else
2802e0541a93SKent Overstreet 			i_size_write(&inode->v, end);
2803e0541a93SKent Overstreet 
28041c6fdbd8SKent Overstreet 		mutex_lock(&inode->ei_update_lock);
2805e0541a93SKent Overstreet 		ret = bch2_write_inode_size(c, inode, end, 0);
28061c6fdbd8SKent Overstreet 		mutex_unlock(&inode->ei_update_lock);
28071c6fdbd8SKent Overstreet 	}
28081c6fdbd8SKent Overstreet err:
28091c6fdbd8SKent Overstreet 	bch2_pagecache_block_put(&inode->ei_pagecache_lock);
28101c6fdbd8SKent Overstreet 	inode_unlock(&inode->v);
28111c6fdbd8SKent Overstreet 	return ret;
28121c6fdbd8SKent Overstreet }
28131c6fdbd8SKent Overstreet 
28141c6fdbd8SKent Overstreet long bch2_fallocate_dispatch(struct file *file, int mode,
28151c6fdbd8SKent Overstreet 			     loff_t offset, loff_t len)
28161c6fdbd8SKent Overstreet {
28171c6fdbd8SKent Overstreet 	struct bch_inode_info *inode = file_bch_inode(file);
28182a9101a9SKent Overstreet 	struct bch_fs *c = inode->v.i_sb->s_fs_info;
28192a9101a9SKent Overstreet 	long ret;
28202a9101a9SKent Overstreet 
28212a9101a9SKent Overstreet 	if (!percpu_ref_tryget(&c->writes))
28222a9101a9SKent Overstreet 		return -EROFS;
28231c6fdbd8SKent Overstreet 
28241c6fdbd8SKent Overstreet 	if (!(mode & ~(FALLOC_FL_KEEP_SIZE|FALLOC_FL_ZERO_RANGE)))
28252a9101a9SKent Overstreet 		ret = bchfs_fallocate(inode, mode, offset, len);
28262a9101a9SKent Overstreet 	else if (mode == (FALLOC_FL_PUNCH_HOLE|FALLOC_FL_KEEP_SIZE))
28272a9101a9SKent Overstreet 		ret = bchfs_fpunch(inode, offset, len);
28282a9101a9SKent Overstreet 	else if (mode == FALLOC_FL_INSERT_RANGE)
28292a9101a9SKent Overstreet 		ret = bchfs_fcollapse_finsert(inode, offset, len, true);
28302a9101a9SKent Overstreet 	else if (mode == FALLOC_FL_COLLAPSE_RANGE)
28312a9101a9SKent Overstreet 		ret = bchfs_fcollapse_finsert(inode, offset, len, false);
28322a9101a9SKent Overstreet 	else
28332a9101a9SKent Overstreet 		ret = -EOPNOTSUPP;
28341c6fdbd8SKent Overstreet 
28352a9101a9SKent Overstreet 	percpu_ref_put(&c->writes);
28361c6fdbd8SKent Overstreet 
28372a9101a9SKent Overstreet 	return ret;
28381c6fdbd8SKent Overstreet }
28391c6fdbd8SKent Overstreet 
284076426098SKent Overstreet static void mark_range_unallocated(struct bch_inode_info *inode,
284176426098SKent Overstreet 				   loff_t start, loff_t end)
284276426098SKent Overstreet {
284376426098SKent Overstreet 	pgoff_t index = start >> PAGE_SHIFT;
284476426098SKent Overstreet 	pgoff_t end_index = (end - 1) >> PAGE_SHIFT;
284576426098SKent Overstreet 	struct folio_batch fbatch;
284676426098SKent Overstreet 	unsigned i, j;
284776426098SKent Overstreet 
284876426098SKent Overstreet 	folio_batch_init(&fbatch);
284976426098SKent Overstreet 
285076426098SKent Overstreet 	while (filemap_get_folios(inode->v.i_mapping,
285176426098SKent Overstreet 				  &index, end_index, &fbatch)) {
285276426098SKent Overstreet 		for (i = 0; i < folio_batch_count(&fbatch); i++) {
285376426098SKent Overstreet 			struct folio *folio = fbatch.folios[i];
285476426098SKent Overstreet 			struct bch_page_state *s;
285576426098SKent Overstreet 
285676426098SKent Overstreet 			folio_lock(folio);
285776426098SKent Overstreet 			s = bch2_page_state(&folio->page);
285876426098SKent Overstreet 
28593826ee0bSKent Overstreet 			if (s) {
28603826ee0bSKent Overstreet 				spin_lock(&s->lock);
286176426098SKent Overstreet 				for (j = 0; j < PAGE_SECTORS; j++)
286276426098SKent Overstreet 					s->s[j].nr_replicas = 0;
28633826ee0bSKent Overstreet 				spin_unlock(&s->lock);
28643826ee0bSKent Overstreet 			}
286576426098SKent Overstreet 
286676426098SKent Overstreet 			folio_unlock(folio);
286776426098SKent Overstreet 		}
286876426098SKent Overstreet 		folio_batch_release(&fbatch);
286976426098SKent Overstreet 		cond_resched();
287076426098SKent Overstreet 	}
287176426098SKent Overstreet }
287276426098SKent Overstreet 
287376426098SKent Overstreet loff_t bch2_remap_file_range(struct file *file_src, loff_t pos_src,
287476426098SKent Overstreet 			     struct file *file_dst, loff_t pos_dst,
287576426098SKent Overstreet 			     loff_t len, unsigned remap_flags)
287676426098SKent Overstreet {
287776426098SKent Overstreet 	struct bch_inode_info *src = file_bch_inode(file_src);
287876426098SKent Overstreet 	struct bch_inode_info *dst = file_bch_inode(file_dst);
287976426098SKent Overstreet 	struct bch_fs *c = src->v.i_sb->s_fs_info;
28802e87eae1SKent Overstreet 	s64 i_sectors_delta = 0;
2881677fc056SKent Overstreet 	u64 aligned_len;
288276426098SKent Overstreet 	loff_t ret = 0;
288376426098SKent Overstreet 
288476426098SKent Overstreet 	if (remap_flags & ~(REMAP_FILE_DEDUP|REMAP_FILE_ADVISORY))
288576426098SKent Overstreet 		return -EINVAL;
288676426098SKent Overstreet 
288776426098SKent Overstreet 	if (remap_flags & REMAP_FILE_DEDUP)
288876426098SKent Overstreet 		return -EOPNOTSUPP;
288976426098SKent Overstreet 
289076426098SKent Overstreet 	if ((pos_src & (block_bytes(c) - 1)) ||
289176426098SKent Overstreet 	    (pos_dst & (block_bytes(c) - 1)))
289276426098SKent Overstreet 		return -EINVAL;
289376426098SKent Overstreet 
289476426098SKent Overstreet 	if (src == dst &&
289576426098SKent Overstreet 	    abs(pos_src - pos_dst) < len)
289676426098SKent Overstreet 		return -EINVAL;
289776426098SKent Overstreet 
289876426098SKent Overstreet 	bch2_lock_inodes(INODE_LOCK|INODE_PAGECACHE_BLOCK, src, dst);
289976426098SKent Overstreet 
29002e87eae1SKent Overstreet 	file_update_time(file_dst);
29012e87eae1SKent Overstreet 
290276426098SKent Overstreet 	inode_dio_wait(&src->v);
290376426098SKent Overstreet 	inode_dio_wait(&dst->v);
290476426098SKent Overstreet 
290576426098SKent Overstreet 	ret = generic_remap_file_range_prep(file_src, pos_src,
290676426098SKent Overstreet 					    file_dst, pos_dst,
290776426098SKent Overstreet 					    &len, remap_flags);
290876426098SKent Overstreet 	if (ret < 0 || len == 0)
29092e87eae1SKent Overstreet 		goto err;
291076426098SKent Overstreet 
2911677fc056SKent Overstreet 	aligned_len = round_up((u64) len, block_bytes(c));
291276426098SKent Overstreet 
291376426098SKent Overstreet 	ret = write_invalidate_inode_pages_range(dst->v.i_mapping,
2914677fc056SKent Overstreet 				pos_dst, pos_dst + len - 1);
291576426098SKent Overstreet 	if (ret)
29162e87eae1SKent Overstreet 		goto err;
291776426098SKent Overstreet 
291876426098SKent Overstreet 	mark_range_unallocated(src, pos_src, pos_src + aligned_len);
291976426098SKent Overstreet 
29202e87eae1SKent Overstreet 	ret = bch2_remap_range(c,
292176426098SKent Overstreet 			       POS(dst->v.i_ino, pos_dst >> 9),
292276426098SKent Overstreet 			       POS(src->v.i_ino, pos_src >> 9),
292376426098SKent Overstreet 			       aligned_len >> 9,
29242e87eae1SKent Overstreet 			       &dst->ei_journal_seq,
29252e87eae1SKent Overstreet 			       pos_dst + len, &i_sectors_delta);
29262e87eae1SKent Overstreet 	if (ret < 0)
29272e87eae1SKent Overstreet 		goto err;
292876426098SKent Overstreet 
29292e87eae1SKent Overstreet 	/*
29302e87eae1SKent Overstreet 	 * due to alignment, we might have remapped slightly more than requsted
29312e87eae1SKent Overstreet 	 */
2932677fc056SKent Overstreet 	ret = min((u64) ret << 9, (u64) len);
29332e87eae1SKent Overstreet 
29342e87eae1SKent Overstreet 	/* XXX get a quota reservation */
29352e87eae1SKent Overstreet 	i_sectors_acct(c, dst, NULL, i_sectors_delta);
29362e87eae1SKent Overstreet 
29372e87eae1SKent Overstreet 	spin_lock(&dst->v.i_lock);
2938677fc056SKent Overstreet 	if (pos_dst + ret > dst->v.i_size)
2939677fc056SKent Overstreet 		i_size_write(&dst->v, pos_dst + ret);
29402e87eae1SKent Overstreet 	spin_unlock(&dst->v.i_lock);
2941e7084c9cSKent Overstreet 
2942e7084c9cSKent Overstreet 	if (((file_dst->f_flags & (__O_SYNC | O_DSYNC)) ||
2943e7084c9cSKent Overstreet 	     IS_SYNC(file_inode(file_dst))) &&
2944e7084c9cSKent Overstreet 	    !c->opts.journal_flush_disabled)
2945e7084c9cSKent Overstreet 		ret = bch2_journal_flush_seq(&c->journal, dst->ei_journal_seq);
29462e87eae1SKent Overstreet err:
294776426098SKent Overstreet 	bch2_unlock_inodes(INODE_LOCK|INODE_PAGECACHE_BLOCK, src, dst);
294876426098SKent Overstreet 
294976426098SKent Overstreet 	return ret;
295076426098SKent Overstreet }
295176426098SKent Overstreet 
29521c6fdbd8SKent Overstreet /* fseek: */
29531c6fdbd8SKent Overstreet 
2954543ef2ebSKent Overstreet static int folio_data_offset(struct folio *folio, unsigned offset)
29551c6fdbd8SKent Overstreet {
2956f57a6a5dSKent Overstreet 	struct bch_page_state *s = bch2_page_state(&folio->page);
2957f57a6a5dSKent Overstreet 	unsigned i;
2958f81b648dSKent Overstreet 
2959543ef2ebSKent Overstreet 	if (s)
2960543ef2ebSKent Overstreet 		for (i = offset >> 9; i < PAGE_SECTORS; i++)
2961f57a6a5dSKent Overstreet 			if (s->s[i].state >= SECTOR_DIRTY)
2962543ef2ebSKent Overstreet 				return i << 9;
2963f57a6a5dSKent Overstreet 
2964543ef2ebSKent Overstreet 	return -1;
29651c6fdbd8SKent Overstreet }
29661c6fdbd8SKent Overstreet 
2967543ef2ebSKent Overstreet static loff_t bch2_seek_pagecache_data(struct inode *vinode,
29681c6fdbd8SKent Overstreet 				       loff_t start_offset,
29691c6fdbd8SKent Overstreet 				       loff_t end_offset)
29701c6fdbd8SKent Overstreet {
29711c6fdbd8SKent Overstreet 	struct folio_batch fbatch;
29721c6fdbd8SKent Overstreet 	pgoff_t start_index	= start_offset >> PAGE_SHIFT;
29731c6fdbd8SKent Overstreet 	pgoff_t end_index	= end_offset >> PAGE_SHIFT;
29741c6fdbd8SKent Overstreet 	pgoff_t index		= start_index;
29751c6fdbd8SKent Overstreet 	unsigned i;
2976543ef2ebSKent Overstreet 	loff_t ret;
2977543ef2ebSKent Overstreet 	int offset;
29781c6fdbd8SKent Overstreet 
29791c6fdbd8SKent Overstreet 	folio_batch_init(&fbatch);
29801c6fdbd8SKent Overstreet 
29811c6fdbd8SKent Overstreet 	while (filemap_get_folios(vinode->i_mapping,
29821c6fdbd8SKent Overstreet 				  &index, end_index, &fbatch)) {
29831c6fdbd8SKent Overstreet 		for (i = 0; i < folio_batch_count(&fbatch); i++) {
29841c6fdbd8SKent Overstreet 			struct folio *folio = fbatch.folios[i];
29851c6fdbd8SKent Overstreet 
29861c6fdbd8SKent Overstreet 			folio_lock(folio);
2987543ef2ebSKent Overstreet 			offset = folio_data_offset(folio,
2988543ef2ebSKent Overstreet 					folio->index == start_index
2989543ef2ebSKent Overstreet 					? start_offset & (PAGE_SIZE - 1)
2990543ef2ebSKent Overstreet 					: 0);
2991543ef2ebSKent Overstreet 			if (offset >= 0) {
2992543ef2ebSKent Overstreet 				ret = clamp(((loff_t) folio->index << PAGE_SHIFT) +
2993543ef2ebSKent Overstreet 					    offset,
2994543ef2ebSKent Overstreet 					    start_offset, end_offset);
29951c6fdbd8SKent Overstreet 				folio_unlock(folio);
29961c6fdbd8SKent Overstreet 				folio_batch_release(&fbatch);
2997543ef2ebSKent Overstreet 				return ret;
29981c6fdbd8SKent Overstreet 			}
29991c6fdbd8SKent Overstreet 			folio_unlock(folio);
30001c6fdbd8SKent Overstreet 		}
30011c6fdbd8SKent Overstreet 		folio_batch_release(&fbatch);
30021c6fdbd8SKent Overstreet 		cond_resched();
30031c6fdbd8SKent Overstreet 	}
30041c6fdbd8SKent Overstreet 
30051c6fdbd8SKent Overstreet 	return end_offset;
30061c6fdbd8SKent Overstreet }
30071c6fdbd8SKent Overstreet 
30081c6fdbd8SKent Overstreet static loff_t bch2_seek_data(struct file *file, u64 offset)
30091c6fdbd8SKent Overstreet {
30101c6fdbd8SKent Overstreet 	struct bch_inode_info *inode = file_bch_inode(file);
30111c6fdbd8SKent Overstreet 	struct bch_fs *c = inode->v.i_sb->s_fs_info;
3012424eb881SKent Overstreet 	struct btree_trans trans;
3013*67e0dd8fSKent Overstreet 	struct btree_iter iter;
30141c6fdbd8SKent Overstreet 	struct bkey_s_c k;
30151c6fdbd8SKent Overstreet 	u64 isize, next_data = MAX_LFS_FILESIZE;
30161c6fdbd8SKent Overstreet 	int ret;
30171c6fdbd8SKent Overstreet 
30181c6fdbd8SKent Overstreet 	isize = i_size_read(&inode->v);
30191c6fdbd8SKent Overstreet 	if (offset >= isize)
30201c6fdbd8SKent Overstreet 		return -ENXIO;
30211c6fdbd8SKent Overstreet 
302220bceecbSKent Overstreet 	bch2_trans_init(&trans, c, 0, 0);
3023424eb881SKent Overstreet 
302441f8b09eSKent Overstreet 	for_each_btree_key(&trans, iter, BTREE_ID_extents,
302594f651e2SKent Overstreet 			   POS(inode->v.i_ino, offset >> 9), 0, k, ret) {
30261c6fdbd8SKent Overstreet 		if (k.k->p.inode != inode->v.i_ino) {
30271c6fdbd8SKent Overstreet 			break;
30281c6fdbd8SKent Overstreet 		} else if (bkey_extent_is_data(k.k)) {
30291c6fdbd8SKent Overstreet 			next_data = max(offset, bkey_start_offset(k.k) << 9);
30301c6fdbd8SKent Overstreet 			break;
30311c6fdbd8SKent Overstreet 		} else if (k.k->p.offset >> 9 > isize)
30321c6fdbd8SKent Overstreet 			break;
30331c6fdbd8SKent Overstreet 	}
3034*67e0dd8fSKent Overstreet 	bch2_trans_iter_exit(&trans, &iter);
30351c6fdbd8SKent Overstreet 
303694f651e2SKent Overstreet 	ret = bch2_trans_exit(&trans) ?: ret;
30371c6fdbd8SKent Overstreet 	if (ret)
30381c6fdbd8SKent Overstreet 		return ret;
30391c6fdbd8SKent Overstreet 
30401c6fdbd8SKent Overstreet 	if (next_data > offset)
3041543ef2ebSKent Overstreet 		next_data = bch2_seek_pagecache_data(&inode->v,
30421c6fdbd8SKent Overstreet 						     offset, next_data);
30431c6fdbd8SKent Overstreet 
3044e10d3094SKent Overstreet 	if (next_data >= isize)
30451c6fdbd8SKent Overstreet 		return -ENXIO;
30461c6fdbd8SKent Overstreet 
30471c6fdbd8SKent Overstreet 	return vfs_setpos(file, next_data, MAX_LFS_FILESIZE);
30481c6fdbd8SKent Overstreet }
30491c6fdbd8SKent Overstreet 
3050543ef2ebSKent Overstreet static int __page_hole_offset(struct page *page, unsigned offset)
30511c6fdbd8SKent Overstreet {
3052543ef2ebSKent Overstreet 	struct bch_page_state *s = bch2_page_state(page);
3053543ef2ebSKent Overstreet 	unsigned i;
3054543ef2ebSKent Overstreet 
3055543ef2ebSKent Overstreet 	if (!s)
3056543ef2ebSKent Overstreet 		return 0;
3057543ef2ebSKent Overstreet 
3058543ef2ebSKent Overstreet 	for (i = offset >> 9; i < PAGE_SECTORS; i++)
3059543ef2ebSKent Overstreet 		if (s->s[i].state < SECTOR_DIRTY)
3060543ef2ebSKent Overstreet 			return i << 9;
3061543ef2ebSKent Overstreet 
3062543ef2ebSKent Overstreet 	return -1;
3063543ef2ebSKent Overstreet }
3064543ef2ebSKent Overstreet 
3065543ef2ebSKent Overstreet static loff_t page_hole_offset(struct address_space *mapping, loff_t offset)
3066543ef2ebSKent Overstreet {
3067543ef2ebSKent Overstreet 	pgoff_t index = offset >> PAGE_SHIFT;
30681c6fdbd8SKent Overstreet 	struct page *page;
3069543ef2ebSKent Overstreet 	int pg_offset;
3070543ef2ebSKent Overstreet 	loff_t ret = -1;
30711c6fdbd8SKent Overstreet 
30721c6fdbd8SKent Overstreet 	page = find_lock_page(mapping, index);
30731c6fdbd8SKent Overstreet 	if (!page)
3074543ef2ebSKent Overstreet 		return offset;
30751c6fdbd8SKent Overstreet 
3076543ef2ebSKent Overstreet 	pg_offset = __page_hole_offset(page, offset & (PAGE_SIZE - 1));
3077543ef2ebSKent Overstreet 	if (pg_offset >= 0)
3078543ef2ebSKent Overstreet 		ret = ((loff_t) index << PAGE_SHIFT) + pg_offset;
3079543ef2ebSKent Overstreet 
30801c6fdbd8SKent Overstreet 	unlock_page(page);
30811c6fdbd8SKent Overstreet 
30821c6fdbd8SKent Overstreet 	return ret;
30831c6fdbd8SKent Overstreet }
30841c6fdbd8SKent Overstreet 
3085543ef2ebSKent Overstreet static loff_t bch2_seek_pagecache_hole(struct inode *vinode,
30861c6fdbd8SKent Overstreet 				       loff_t start_offset,
30871c6fdbd8SKent Overstreet 				       loff_t end_offset)
30881c6fdbd8SKent Overstreet {
30891c6fdbd8SKent Overstreet 	struct address_space *mapping = vinode->i_mapping;
3090543ef2ebSKent Overstreet 	loff_t offset = start_offset, hole;
30911c6fdbd8SKent Overstreet 
3092543ef2ebSKent Overstreet 	while (offset < end_offset) {
3093543ef2ebSKent Overstreet 		hole = page_hole_offset(mapping, offset);
3094543ef2ebSKent Overstreet 		if (hole >= 0 && hole <= end_offset)
3095543ef2ebSKent Overstreet 			return max(start_offset, hole);
3096543ef2ebSKent Overstreet 
3097543ef2ebSKent Overstreet 		offset += PAGE_SIZE;
3098543ef2ebSKent Overstreet 		offset &= PAGE_MASK;
3099543ef2ebSKent Overstreet 	}
31001c6fdbd8SKent Overstreet 
31011c6fdbd8SKent Overstreet 	return end_offset;
31021c6fdbd8SKent Overstreet }
31031c6fdbd8SKent Overstreet 
31041c6fdbd8SKent Overstreet static loff_t bch2_seek_hole(struct file *file, u64 offset)
31051c6fdbd8SKent Overstreet {
31061c6fdbd8SKent Overstreet 	struct bch_inode_info *inode = file_bch_inode(file);
31071c6fdbd8SKent Overstreet 	struct bch_fs *c = inode->v.i_sb->s_fs_info;
3108424eb881SKent Overstreet 	struct btree_trans trans;
3109*67e0dd8fSKent Overstreet 	struct btree_iter iter;
31101c6fdbd8SKent Overstreet 	struct bkey_s_c k;
31111c6fdbd8SKent Overstreet 	u64 isize, next_hole = MAX_LFS_FILESIZE;
31121c6fdbd8SKent Overstreet 	int ret;
31131c6fdbd8SKent Overstreet 
31141c6fdbd8SKent Overstreet 	isize = i_size_read(&inode->v);
31151c6fdbd8SKent Overstreet 	if (offset >= isize)
31161c6fdbd8SKent Overstreet 		return -ENXIO;
31171c6fdbd8SKent Overstreet 
311820bceecbSKent Overstreet 	bch2_trans_init(&trans, c, 0, 0);
3119424eb881SKent Overstreet 
312041f8b09eSKent Overstreet 	for_each_btree_key(&trans, iter, BTREE_ID_extents,
31211c6fdbd8SKent Overstreet 			   POS(inode->v.i_ino, offset >> 9),
312294f651e2SKent Overstreet 			   BTREE_ITER_SLOTS, k, ret) {
31231c6fdbd8SKent Overstreet 		if (k.k->p.inode != inode->v.i_ino) {
3124543ef2ebSKent Overstreet 			next_hole = bch2_seek_pagecache_hole(&inode->v,
31251c6fdbd8SKent Overstreet 					offset, MAX_LFS_FILESIZE);
31261c6fdbd8SKent Overstreet 			break;
31271c6fdbd8SKent Overstreet 		} else if (!bkey_extent_is_data(k.k)) {
3128543ef2ebSKent Overstreet 			next_hole = bch2_seek_pagecache_hole(&inode->v,
31291c6fdbd8SKent Overstreet 					max(offset, bkey_start_offset(k.k) << 9),
31301c6fdbd8SKent Overstreet 					k.k->p.offset << 9);
31311c6fdbd8SKent Overstreet 
31321c6fdbd8SKent Overstreet 			if (next_hole < k.k->p.offset << 9)
31331c6fdbd8SKent Overstreet 				break;
31341c6fdbd8SKent Overstreet 		} else {
31351c6fdbd8SKent Overstreet 			offset = max(offset, bkey_start_offset(k.k) << 9);
31361c6fdbd8SKent Overstreet 		}
31371c6fdbd8SKent Overstreet 	}
3138*67e0dd8fSKent Overstreet 	bch2_trans_iter_exit(&trans, &iter);
31391c6fdbd8SKent Overstreet 
314094f651e2SKent Overstreet 	ret = bch2_trans_exit(&trans) ?: ret;
31411c6fdbd8SKent Overstreet 	if (ret)
31421c6fdbd8SKent Overstreet 		return ret;
31431c6fdbd8SKent Overstreet 
31441c6fdbd8SKent Overstreet 	if (next_hole > isize)
31451c6fdbd8SKent Overstreet 		next_hole = isize;
31461c6fdbd8SKent Overstreet 
31471c6fdbd8SKent Overstreet 	return vfs_setpos(file, next_hole, MAX_LFS_FILESIZE);
31481c6fdbd8SKent Overstreet }
31491c6fdbd8SKent Overstreet 
31501c6fdbd8SKent Overstreet loff_t bch2_llseek(struct file *file, loff_t offset, int whence)
31511c6fdbd8SKent Overstreet {
31521c6fdbd8SKent Overstreet 	switch (whence) {
31531c6fdbd8SKent Overstreet 	case SEEK_SET:
31541c6fdbd8SKent Overstreet 	case SEEK_CUR:
31551c6fdbd8SKent Overstreet 	case SEEK_END:
31561c6fdbd8SKent Overstreet 		return generic_file_llseek(file, offset, whence);
31571c6fdbd8SKent Overstreet 	case SEEK_DATA:
31581c6fdbd8SKent Overstreet 		return bch2_seek_data(file, offset);
31591c6fdbd8SKent Overstreet 	case SEEK_HOLE:
31601c6fdbd8SKent Overstreet 		return bch2_seek_hole(file, offset);
31611c6fdbd8SKent Overstreet 	}
31621c6fdbd8SKent Overstreet 
31631c6fdbd8SKent Overstreet 	return -EINVAL;
31641c6fdbd8SKent Overstreet }
31651c6fdbd8SKent Overstreet 
31661c6fdbd8SKent Overstreet void bch2_fs_fsio_exit(struct bch_fs *c)
31671c6fdbd8SKent Overstreet {
31681c6fdbd8SKent Overstreet 	bioset_exit(&c->dio_write_bioset);
31691c6fdbd8SKent Overstreet 	bioset_exit(&c->dio_read_bioset);
31701c6fdbd8SKent Overstreet 	bioset_exit(&c->writepage_bioset);
31711c6fdbd8SKent Overstreet }
31721c6fdbd8SKent Overstreet 
31731c6fdbd8SKent Overstreet int bch2_fs_fsio_init(struct bch_fs *c)
31741c6fdbd8SKent Overstreet {
31751c6fdbd8SKent Overstreet 	int ret = 0;
31761c6fdbd8SKent Overstreet 
31771c6fdbd8SKent Overstreet 	pr_verbose_init(c->opts, "");
31781c6fdbd8SKent Overstreet 
31791c6fdbd8SKent Overstreet 	if (bioset_init(&c->writepage_bioset,
31809a3df993SKent Overstreet 			4, offsetof(struct bch_writepage_io, op.wbio.bio),
31811c6fdbd8SKent Overstreet 			BIOSET_NEED_BVECS) ||
31821c6fdbd8SKent Overstreet 	    bioset_init(&c->dio_read_bioset,
31831c6fdbd8SKent Overstreet 			4, offsetof(struct dio_read, rbio.bio),
31841c6fdbd8SKent Overstreet 			BIOSET_NEED_BVECS) ||
31851c6fdbd8SKent Overstreet 	    bioset_init(&c->dio_write_bioset,
31869a3df993SKent Overstreet 			4, offsetof(struct dio_write, op.wbio.bio),
31871c6fdbd8SKent Overstreet 			BIOSET_NEED_BVECS))
31881c6fdbd8SKent Overstreet 		ret = -ENOMEM;
31891c6fdbd8SKent Overstreet 
31901c6fdbd8SKent Overstreet 	pr_verbose_init(c->opts, "ret %i", ret);
31911c6fdbd8SKent Overstreet 	return ret;
31921c6fdbd8SKent Overstreet }
31931c6fdbd8SKent Overstreet 
31941c6fdbd8SKent Overstreet #endif /* NO_BCACHEFS_FS */
3195