]> git.sesse.net Git - bcachefs-tools-debian/blobdiff - libbcachefs/fs-io.c
Update bcachefs sources to 9b3aa5ec6c bcachefs: Add tabstops to printbufs
[bcachefs-tools-debian] / libbcachefs / fs-io.c
index c07755c6916d2a6b010679e55b13cd35f8bc4713..1d0871f63e4e71402874373a982262d74b00c24c 100644 (file)
@@ -223,6 +223,9 @@ static void i_sectors_acct(struct bch_fs *c, struct bch_inode_info *inode,
                return;
 
        mutex_lock(&inode->ei_quota_lock);
+       BUG_ON((s64) inode->v.i_blocks + sectors < 0);
+       inode->v.i_blocks += sectors;
+
 #ifdef CONFIG_BCACHEFS_QUOTA
        if (quota_res && sectors > 0) {
                BUG_ON(sectors > quota_res->sectors);
@@ -234,7 +237,6 @@ static void i_sectors_acct(struct bch_fs *c, struct bch_inode_info *inode,
                bch2_quota_acct(c, inode->ei_qid, Q_SPC, sectors, KEY_TYPE_QUOTA_WARN);
        }
 #endif
-       inode->v.i_blocks += sectors;
        mutex_unlock(&inode->ei_quota_lock);
 }
 
@@ -243,24 +245,26 @@ static void i_sectors_acct(struct bch_fs *c, struct bch_inode_info *inode,
 /* stored in page->private: */
 
 struct bch_page_sector {
-       /* Uncompressed, fully allocated replicas: */
-       unsigned                nr_replicas:3;
+       /* Uncompressed, fully allocated replicas (or on disk reservation): */
+       unsigned                nr_replicas:4;
 
-       /* Owns PAGE_SECTORS * replicas_reserved sized reservation: */
-       unsigned                replicas_reserved:3;
+       /* Owns PAGE_SECTORS * replicas_reserved sized in memory reservation: */
+       unsigned                replicas_reserved:4;
 
        /* i_sectors: */
        enum {
                SECTOR_UNALLOCATED,
                SECTOR_RESERVED,
                SECTOR_DIRTY,
+               SECTOR_DIRTY_RESERVED,
                SECTOR_ALLOCATED,
-       }                       state:2;
+       }                       state:8;
 };
 
 struct bch_page_state {
        spinlock_t              lock;
        atomic_t                write_count;
+       bool                    uptodate;
        struct bch_page_sector  s[PAGE_SECTORS];
 };
 
@@ -311,6 +315,212 @@ static struct bch_page_state *bch2_page_state_create(struct page *page,
        return bch2_page_state(page) ?: __bch2_page_state_create(page, gfp);
 }
 
+static unsigned bkey_to_sector_state(const struct bkey *k)
+{
+       if (k->type == KEY_TYPE_reservation)
+               return SECTOR_RESERVED;
+       if (bkey_extent_is_allocation(k))
+               return SECTOR_ALLOCATED;
+       return SECTOR_UNALLOCATED;
+}
+
+static void __bch2_page_state_set(struct page *page,
+                                 unsigned pg_offset, unsigned pg_len,
+                                 unsigned nr_ptrs, unsigned state)
+{
+       struct bch_page_state *s = bch2_page_state_create(page, __GFP_NOFAIL);
+       unsigned i;
+
+       BUG_ON(pg_offset >= PAGE_SECTORS);
+       BUG_ON(pg_offset + pg_len > PAGE_SECTORS);
+
+       spin_lock(&s->lock);
+
+       for (i = pg_offset; i < pg_offset + pg_len; i++) {
+               s->s[i].nr_replicas = nr_ptrs;
+               s->s[i].state = state;
+       }
+
+       if (i == PAGE_SECTORS)
+               s->uptodate = true;
+
+       spin_unlock(&s->lock);
+}
+
+static int bch2_page_state_set(struct bch_fs *c, subvol_inum inum,
+                              struct page **pages, unsigned nr_pages)
+{
+       struct btree_trans trans;
+       struct btree_iter iter;
+       struct bkey_s_c k;
+       u64 offset = pages[0]->index << PAGE_SECTORS_SHIFT;
+       unsigned pg_idx = 0;
+       u32 snapshot;
+       int ret;
+
+       bch2_trans_init(&trans, c, 0, 0);
+retry:
+       bch2_trans_begin(&trans);
+
+       ret = bch2_subvolume_get_snapshot(&trans, inum.subvol, &snapshot);
+       if (ret)
+               goto err;
+
+       for_each_btree_key_norestart(&trans, iter, BTREE_ID_extents,
+                          SPOS(inum.inum, offset, snapshot),
+                          BTREE_ITER_SLOTS, k, ret) {
+               unsigned nr_ptrs = bch2_bkey_nr_ptrs_fully_allocated(k);
+               unsigned state = bkey_to_sector_state(k.k);
+
+               while (pg_idx < nr_pages) {
+                       struct page *page = pages[pg_idx];
+                       u64 pg_start = page->index << PAGE_SECTORS_SHIFT;
+                       u64 pg_end = (page->index + 1) << PAGE_SECTORS_SHIFT;
+                       unsigned pg_offset = max(bkey_start_offset(k.k), pg_start) - pg_start;
+                       unsigned pg_len = min(k.k->p.offset, pg_end) - pg_offset - pg_start;
+
+                       BUG_ON(k.k->p.offset < pg_start);
+                       BUG_ON(bkey_start_offset(k.k) > pg_end);
+
+                       if (!bch2_page_state_create(page, __GFP_NOFAIL)->uptodate)
+                               __bch2_page_state_set(page, pg_offset, pg_len, nr_ptrs, state);
+
+                       if (k.k->p.offset < pg_end)
+                               break;
+                       pg_idx++;
+               }
+
+               if (pg_idx == nr_pages)
+                       break;
+       }
+
+       offset = iter.pos.offset;
+       bch2_trans_iter_exit(&trans, &iter);
+err:
+       if (ret == -EINTR)
+               goto retry;
+       bch2_trans_exit(&trans);
+
+       return ret;
+}
+
+static void bch2_bio_page_state_set(struct bio *bio, struct bkey_s_c k)
+{
+       struct bvec_iter iter;
+       struct bio_vec bv;
+       unsigned nr_ptrs = k.k->type == KEY_TYPE_reflink_v
+               ? 0 : bch2_bkey_nr_ptrs_fully_allocated(k);
+       unsigned state = bkey_to_sector_state(k.k);
+
+       bio_for_each_segment(bv, bio, iter)
+               __bch2_page_state_set(bv.bv_page, bv.bv_offset >> 9,
+                                     bv.bv_len >> 9, nr_ptrs, state);
+}
+
+static void mark_pagecache_unallocated(struct bch_inode_info *inode,
+                                      u64 start, u64 end)
+{
+       pgoff_t index = start >> PAGE_SECTORS_SHIFT;
+       pgoff_t end_index = (end - 1) >> PAGE_SECTORS_SHIFT;
+       struct pagevec pvec;
+
+       if (end <= start)
+               return;
+
+       pagevec_init(&pvec);
+
+       do {
+               unsigned nr_pages, i, j;
+
+               nr_pages = pagevec_lookup_range(&pvec, inode->v.i_mapping,
+                                               &index, end_index);
+               for (i = 0; i < nr_pages; i++) {
+                       struct page *page = pvec.pages[i];
+                       u64 pg_start = page->index << PAGE_SECTORS_SHIFT;
+                       u64 pg_end = (page->index + 1) << PAGE_SECTORS_SHIFT;
+                       unsigned pg_offset = max(start, pg_start) - pg_start;
+                       unsigned pg_len = min(end, pg_end) - pg_offset - pg_start;
+                       struct bch_page_state *s;
+
+                       BUG_ON(end <= pg_start);
+                       BUG_ON(pg_offset >= PAGE_SECTORS);
+                       BUG_ON(pg_offset + pg_len > PAGE_SECTORS);
+
+                       lock_page(page);
+                       s = bch2_page_state(page);
+
+                       if (s) {
+                               spin_lock(&s->lock);
+                               for (j = pg_offset; j < pg_offset + pg_len; j++)
+                                       s->s[j].nr_replicas = 0;
+                               spin_unlock(&s->lock);
+                       }
+
+                       unlock_page(page);
+               }
+               pagevec_release(&pvec);
+       } while (index <= end_index);
+}
+
+static void mark_pagecache_reserved(struct bch_inode_info *inode,
+                                   u64 start, u64 end)
+{
+       struct bch_fs *c = inode->v.i_sb->s_fs_info;
+       pgoff_t index = start >> PAGE_SECTORS_SHIFT;
+       pgoff_t end_index = (end - 1) >> PAGE_SECTORS_SHIFT;
+       struct pagevec pvec;
+       s64 i_sectors_delta = 0;
+
+       if (end <= start)
+               return;
+
+       pagevec_init(&pvec);
+
+       do {
+               unsigned nr_pages, i, j;
+
+               nr_pages = pagevec_lookup_range(&pvec, inode->v.i_mapping,
+                                               &index, end_index);
+               for (i = 0; i < nr_pages; i++) {
+                       struct page *page = pvec.pages[i];
+                       u64 pg_start = page->index << PAGE_SECTORS_SHIFT;
+                       u64 pg_end = (page->index + 1) << PAGE_SECTORS_SHIFT;
+                       unsigned pg_offset = max(start, pg_start) - pg_start;
+                       unsigned pg_len = min(end, pg_end) - pg_offset - pg_start;
+                       struct bch_page_state *s;
+
+                       BUG_ON(end <= pg_start);
+                       BUG_ON(pg_offset >= PAGE_SECTORS);
+                       BUG_ON(pg_offset + pg_len > PAGE_SECTORS);
+
+                       lock_page(page);
+                       s = bch2_page_state(page);
+
+                       if (s) {
+                               spin_lock(&s->lock);
+                               for (j = pg_offset; j < pg_offset + pg_len; j++)
+                                       switch (s->s[j].state) {
+                                       case SECTOR_UNALLOCATED:
+                                               s->s[j].state = SECTOR_RESERVED;
+                                               break;
+                                       case SECTOR_DIRTY:
+                                               s->s[j].state = SECTOR_DIRTY_RESERVED;
+                                               i_sectors_delta--;
+                                               break;
+                                       default:
+                                               break;
+                                       }
+                               spin_unlock(&s->lock);
+                       }
+
+                       unlock_page(page);
+               }
+               pagevec_release(&pvec);
+       } while (index <= end_index);
+
+       i_sectors_acct(c, inode, NULL, i_sectors_delta);
+}
+
 static inline unsigned inode_nr_replicas(struct bch_fs *c, struct bch_inode_info *inode)
 {
        /* XXX: this should not be open coded */
@@ -395,6 +605,8 @@ static int bch2_page_reservation_get(struct bch_fs *c,
        if (!s)
                return -ENOMEM;
 
+       BUG_ON(!s->uptodate);
+
        for (i = round_down(offset, block_bytes(c)) >> 9;
             i < round_up(offset + len, block_bytes(c)) >> 9;
             i++) {
@@ -449,16 +661,22 @@ static void bch2_clear_page_bits(struct page *page)
                disk_res.sectors += s->s[i].replicas_reserved;
                s->s[i].replicas_reserved = 0;
 
-               if (s->s[i].state == SECTOR_DIRTY) {
-                       dirty_sectors++;
+               switch (s->s[i].state) {
+               case SECTOR_DIRTY:
                        s->s[i].state = SECTOR_UNALLOCATED;
+                       --dirty_sectors;
+                       break;
+               case SECTOR_DIRTY_RESERVED:
+                       s->s[i].state = SECTOR_RESERVED;
+                       break;
+               default:
+                       break;
                }
        }
 
        bch2_disk_reservation_put(c, &disk_res);
 
-       if (dirty_sectors)
-               i_sectors_acct(c, inode, NULL, -dirty_sectors);
+       i_sectors_acct(c, inode, NULL, dirty_sectors);
 
        bch2_page_state_release(page);
 }
@@ -491,16 +709,22 @@ static void bch2_set_page_dirty(struct bch_fs *c,
                s->s[i].replicas_reserved += sectors;
                res->disk.sectors -= sectors;
 
-               if (s->s[i].state == SECTOR_UNALLOCATED)
+               switch (s->s[i].state) {
+               case SECTOR_UNALLOCATED:
+                       s->s[i].state = SECTOR_DIRTY;
                        dirty_sectors++;
-
-               s->s[i].state = max_t(unsigned, s->s[i].state, SECTOR_DIRTY);
+                       break;
+               case SECTOR_RESERVED:
+                       s->s[i].state = SECTOR_DIRTY_RESERVED;
+                       break;
+               default:
+                       break;
+               }
        }
 
        spin_unlock(&s->lock);
 
-       if (dirty_sectors)
-               i_sectors_acct(c, inode, &res->quota, dirty_sectors);
+       i_sectors_acct(c, inode, &res->quota, dirty_sectors);
 
        if (!PageDirty(page))
                __set_page_dirty_nobuffers(page);
@@ -554,7 +778,7 @@ vm_fault_t bch2_page_mkwrite(struct vm_fault *vmf)
        struct bch2_page_reservation res;
        unsigned len;
        loff_t isize;
-       int ret = VM_FAULT_LOCKED;
+       int ret;
 
        bch2_page_reservation_init(c, inode, &res);
 
@@ -580,6 +804,14 @@ vm_fault_t bch2_page_mkwrite(struct vm_fault *vmf)
 
        len = min_t(loff_t, PAGE_SIZE, isize - page_offset(page));
 
+       if (!bch2_page_state_create(page, __GFP_NOFAIL)->uptodate) {
+               if (bch2_page_state_set(c, inode_inum(inode), &page, 1)) {
+                       unlock_page(page);
+                       ret = VM_FAULT_SIGBUS;
+                       goto out;
+               }
+       }
+
        if (bch2_page_reservation_get(c, inode, page, &res, 0, len, true)) {
                unlock_page(page);
                ret = VM_FAULT_SIGBUS;
@@ -590,6 +822,7 @@ vm_fault_t bch2_page_mkwrite(struct vm_fault *vmf)
        bch2_page_reservation_put(c, inode, &res);
 
        wait_for_stable_page(page);
+       ret = VM_FAULT_LOCKED;
 out:
        bch2_pagecache_add_put(&inode->ei_pagecache_lock);
        sb_end_pagefault(inode->v.i_sb);
@@ -703,29 +936,6 @@ static inline struct page *readpage_iter_next(struct readpages_iter *iter)
        return iter->pages[iter->idx];
 }
 
-static void bch2_add_page_sectors(struct bio *bio, struct bkey_s_c k)
-{
-       struct bvec_iter iter;
-       struct bio_vec bv;
-       unsigned nr_ptrs = k.k->type == KEY_TYPE_reflink_v
-               ? 0 : bch2_bkey_nr_ptrs_fully_allocated(k);
-       unsigned state = k.k->type == KEY_TYPE_reservation
-               ? SECTOR_RESERVED
-               : SECTOR_ALLOCATED;
-
-       bio_for_each_segment(bv, bio, iter) {
-               struct bch_page_state *s = bch2_page_state(bv.bv_page);
-               unsigned i;
-
-               for (i = bv.bv_offset >> 9;
-                    i < (bv.bv_offset + bv.bv_len) >> 9;
-                    i++) {
-                       s->s[i].nr_replicas = nr_ptrs;
-                       s->s[i].state = state;
-               }
-       }
-}
-
 static bool extent_partial_reads_expensive(struct bkey_s_c k)
 {
        struct bkey_ptrs_c ptrs = bch2_bkey_ptrs_c(k);
@@ -745,7 +955,7 @@ static void readpage_bio_extend(struct readpages_iter *iter,
 {
        while (bio_sectors(bio) < sectors_this_extent &&
               bio->bi_vcnt < bio->bi_max_vecs) {
-               pgoff_t page_offset = bio_end_sector(bio) >> PAGE_SECTOR_SHIFT;
+               pgoff_t page_offset = bio_end_sector(bio) >> PAGE_SECTORS_SHIFT;
                struct page *page = readpage_iter_next(iter);
                int ret;
 
@@ -814,7 +1024,7 @@ retry:
 
        bch2_trans_iter_init(trans, &iter, BTREE_ID_extents,
                             SPOS(inum.inum, rbio->bio.bi_iter.bi_sector, snapshot),
-                            BTREE_ITER_SLOTS|BTREE_ITER_FILTER_SNAPSHOTS);
+                            BTREE_ITER_SLOTS);
        while (1) {
                struct bkey_s_c k;
                unsigned bytes, sectors, offset_into_extent;
@@ -852,8 +1062,6 @@ retry:
 
                sectors = min(sectors, k.k->size - offset_into_extent);
 
-               bch2_trans_unlock(trans);
-
                if (readpages_iter)
                        readpage_bio_extend(readpages_iter, &rbio->bio, sectors,
                                            extent_partial_reads_expensive(k));
@@ -864,8 +1072,7 @@ retry:
                if (rbio->bio.bi_iter.bi_size == bytes)
                        flags |= BCH_READ_LAST_FRAGMENT;
 
-               if (bkey_extent_is_allocation(k.k))
-                       bch2_add_page_sectors(&rbio->bio, k);
+               bch2_bio_page_state_set(&rbio->bio, k);
 
                bch2_read_extent(trans, rbio, iter.pos,
                                 data_btree, k, offset_into_extent, flags);
@@ -875,6 +1082,10 @@ retry:
 
                swap(rbio->bio.bi_iter.bi_size, bytes);
                bio_advance(&rbio->bio, bytes);
+
+               ret = btree_trans_too_many_iters(trans);
+               if (ret)
+                       break;
        }
 err:
        bch2_trans_iter_exit(trans, &iter);
@@ -922,7 +1133,7 @@ void bch2_readahead(struct readahead_control *ractl)
                readpages_iter.idx++;
 
                bio_set_op_attrs(&rbio->bio, REQ_OP_READ, 0);
-               rbio->bio.bi_iter.bi_sector = (sector_t) index << PAGE_SECTOR_SHIFT;
+               rbio->bio.bi_iter.bi_sector = (sector_t) index << PAGE_SECTORS_SHIFT;
                rbio->bio.bi_end_io = bch2_readpages_end_io;
                BUG_ON(!bio_add_page(&rbio->bio, page, PAGE_SIZE, 0));
 
@@ -945,7 +1156,7 @@ static void __bchfs_readpage(struct bch_fs *c, struct bch_read_bio *rbio,
 
        bio_set_op_attrs(&rbio->bio, REQ_OP_READ, REQ_SYNC);
        rbio->bio.bi_iter.bi_sector =
-               (sector_t) page->index << PAGE_SECTOR_SHIFT;
+               (sector_t) page->index << PAGE_SECTORS_SHIFT;
        BUG_ON(!bio_add_page(&rbio->bio, page, PAGE_SIZE, 0));
 
        bch2_trans_init(&trans, c, 0, 0);
@@ -1067,7 +1278,7 @@ static void bch2_writepage_io_done(struct closure *cl)
         * racing with fallocate can cause us to add fewer sectors than
         * expected - but we shouldn't add more sectors than expected:
         */
-       BUG_ON(io->op.i_sectors_delta > 0);
+       WARN_ON(io->op.i_sectors_delta > 0);
 
        /*
         * (error (due to going RO) halfway through a page can screw that up
@@ -1126,7 +1337,6 @@ static void bch2_writepage_io_alloc(struct bch_fs *c,
        op                      = &w->io->op;
        bch2_write_op_init(op, c, w->opts);
        op->target              = w->opts.foreground_target;
-       op_journal_seq_set(op, &inode->ei_journal_seq);
        op->nr_replicas         = nr_replicas;
        op->res.nr_replicas     = nr_replicas;
        op->write_point         = writepoint_hashed(inode->ei_last_dirtied);
@@ -1173,16 +1383,16 @@ static int __bch2_writepage(struct page *page,
 do_io:
        s = bch2_page_state_create(page, __GFP_NOFAIL);
 
-       ret = bch2_get_page_disk_reservation(c, inode, page, true);
-       if (ret) {
-               SetPageError(page);
-               mapping_set_error(page->mapping, ret);
-               unlock_page(page);
-               return 0;
-       }
+       /*
+        * Things get really hairy with errors during writeback:
+        */
+       ret = bch2_get_page_disk_reservation(c, inode, page, false);
+       BUG_ON(ret);
 
        /* Before unlocking the page, get copy of reservations: */
+       spin_lock(&s->lock);
        orig = *s;
+       spin_unlock(&s->lock);
 
        for (i = 0; i < PAGE_SECTORS; i++) {
                if (s->s[i].state < SECTOR_DIRTY)
@@ -1215,7 +1425,7 @@ do_io:
 
        offset = 0;
        while (1) {
-               unsigned sectors = 1, dirty_sectors = 0, reserved_sectors = 0;
+               unsigned sectors = 0, dirty_sectors = 0, reserved_sectors = 0;
                u64 sector;
 
                while (offset < PAGE_SECTORS &&
@@ -1225,16 +1435,15 @@ do_io:
                if (offset == PAGE_SECTORS)
                        break;
 
-               sector = ((u64) page->index << PAGE_SECTOR_SHIFT) + offset;
-
                while (offset + sectors < PAGE_SECTORS &&
-                      orig.s[offset + sectors].state >= SECTOR_DIRTY)
+                      orig.s[offset + sectors].state >= SECTOR_DIRTY) {
+                       reserved_sectors += orig.s[offset + sectors].replicas_reserved;
+                       dirty_sectors += orig.s[offset + sectors].state == SECTOR_DIRTY;
                        sectors++;
-
-               for (i = offset; i < offset + sectors; i++) {
-                       reserved_sectors += orig.s[i].replicas_reserved;
-                       dirty_sectors += orig.s[i].state == SECTOR_DIRTY;
                }
+               BUG_ON(!sectors);
+
+               sector = ((u64) page->index << PAGE_SECTORS_SHIFT) + offset;
 
                if (w->io &&
                    (w->io->op.res.nr_replicas != nr_replicas_this_write ||
@@ -1351,6 +1560,12 @@ readpage:
        if (ret)
                goto err;
 out:
+       if (!bch2_page_state_create(page, __GFP_NOFAIL)->uptodate) {
+               ret = bch2_page_state_set(c, inode_inum(inode), &page, 1);
+               if (ret)
+                       goto out;
+       }
+
        ret = bch2_page_reservation_get(c, inode, page, res,
                                        offset, len, true);
        if (ret) {
@@ -1480,20 +1695,21 @@ static int __bch2_buffered_write(struct bch_inode_info *inode,
        }
 
        while (reserved < len) {
-               struct page *page = pages[(offset + reserved) >> PAGE_SHIFT];
+               unsigned i = (offset + reserved) >> PAGE_SHIFT;
+               struct page *page = pages[i];
                unsigned pg_offset = (offset + reserved) & (PAGE_SIZE - 1);
                unsigned pg_len = min_t(unsigned, len - reserved,
                                        PAGE_SIZE - pg_offset);
-retry_reservation:
-               ret = bch2_page_reservation_get(c, inode, page, &res,
-                                               pg_offset, pg_len, true);
 
-               if (ret && !PageUptodate(page)) {
-                       ret = bch2_read_single_page(page, mapping);
-                       if (!ret)
-                               goto retry_reservation;
+               if (!bch2_page_state_create(page, __GFP_NOFAIL)->uptodate) {
+                       ret = bch2_page_state_set(c, inode_inum(inode),
+                                                 pages + i, nr_pages - i);
+                       if (ret)
+                               goto out;
                }
 
+               ret = bch2_page_reservation_get(c, inode, page, &res,
+                                               pg_offset, pg_len, true);
                if (ret)
                        goto out;
 
@@ -1509,8 +1725,8 @@ retry_reservation:
                unsigned pg_offset = (offset + copied) & (PAGE_SIZE - 1);
                unsigned pg_len = min_t(unsigned, len - copied,
                                        PAGE_SIZE - pg_offset);
-               unsigned pg_copied = iov_iter_copy_from_user_atomic(page,
-                                               iter, pg_offset, pg_len);
+               unsigned pg_copied = copy_page_from_iter_atomic(page,
+                                               pg_offset, pg_len,iter);
 
                if (!pg_copied)
                        break;
@@ -1523,7 +1739,6 @@ retry_reservation:
                }
 
                flush_dcache_page(page);
-               iov_iter_advance(iter, pg_copied);
                copied += pg_copied;
 
                if (pg_copied != pg_len)
@@ -1838,7 +2053,7 @@ retry:
        if (err)
                goto err;
 
-       for_each_btree_key(&trans, iter, BTREE_ID_extents,
+       for_each_btree_key_norestart(&trans, iter, BTREE_ID_extents,
                           SPOS(inum.inum, offset, snapshot),
                           BTREE_ITER_SLOTS, k, err) {
                if (bkey_cmp(bkey_start_pos(k.k), POS(inum.inum, end)) >= 0)
@@ -1937,7 +2152,6 @@ static long bch2_dio_write_loop(struct dio_write *dio)
                bch2_write_op_init(&dio->op, c, io_opts(c, &inode->ei_inode));
                dio->op.end_io          = bch2_dio_write_loop_async;
                dio->op.target          = dio->op.opts.foreground_target;
-               op_journal_seq_set(&dio->op, &inode->ei_journal_seq);
                dio->op.write_point     = writepoint_hashed((unsigned long) current);
                dio->op.nr_replicas     = dio->op.opts.data_replicas;
                dio->op.subvol          = inode->ei_subvol;
@@ -2169,29 +2383,36 @@ unlock:
 
 /* fsync: */
 
-int bch2_fsync(struct file *file, loff_t start, loff_t end, int datasync)
+/*
+ * inode->ei_inode.bi_journal_seq won't be up to date since it's set in an
+ * insert trigger: look up the btree inode instead
+ */
+static int bch2_flush_inode(struct bch_fs *c, subvol_inum inum)
 {
-       struct bch_inode_info *inode = file_bch_inode(file);
-       struct bch_fs *c = inode->v.i_sb->s_fs_info;
-       int ret, ret2;
+       struct bch_inode_unpacked inode;
+       int ret;
 
-       ret = file_write_and_wait_range(file, start, end);
+       if (c->opts.journal_flush_disabled)
+               return 0;
+
+       ret = bch2_inode_find_by_inum(c, inum, &inode);
        if (ret)
                return ret;
 
-       if (datasync && !(inode->v.i_state & I_DIRTY_DATASYNC))
-               goto out;
+       return bch2_journal_flush_seq(&c->journal, inode.bi_journal_seq);
+}
 
-       ret = sync_inode_metadata(&inode->v, 1);
-       if (ret)
-               return ret;
-out:
-       if (!c->opts.journal_flush_disabled)
-               ret = bch2_journal_flush_seq(&c->journal,
-                                            inode->ei_journal_seq);
-       ret2 = file_check_and_advance_wb_err(file);
+int bch2_fsync(struct file *file, loff_t start, loff_t end, int datasync)
+{
+       struct bch_inode_info *inode = file_bch_inode(file);
+       struct bch_fs *c = inode->v.i_sb->s_fs_info;
+       int ret, ret2, ret3;
 
-       return ret ?: ret2;
+       ret = file_write_and_wait_range(file, start, end);
+       ret2 = sync_inode_metadata(&inode->v, 1);
+       ret3 = bch2_flush_inode(c, inode_inum(inode));
+
+       return ret ?: ret2 ?: ret3;
 }
 
 /* truncate: */
@@ -2213,7 +2434,7 @@ retry:
        if (ret)
                goto err;
 
-       for_each_btree_key(&trans, iter, BTREE_ID_extents, start, 0, k, ret) {
+       for_each_btree_key_norestart(&trans, iter, BTREE_ID_extents, start, 0, k, ret) {
                if (bkey_cmp(bkey_start_pos(k.k), end) >= 0)
                        break;
 
@@ -2228,7 +2449,8 @@ err:
        if (ret == -EINTR)
                goto retry;
 
-       return bch2_trans_exit(&trans) ?: ret;
+       bch2_trans_exit(&trans);
+       return ret;
 }
 
 static int __bch2_truncate_page(struct bch_inode_info *inode,
@@ -2241,6 +2463,7 @@ static int __bch2_truncate_page(struct bch_inode_info *inode,
        unsigned end_offset = ((end - 1) & (PAGE_SIZE - 1)) + 1;
        unsigned i;
        struct page *page;
+       s64 i_sectors_delta = 0;
        int ret = 0;
 
        /* Page boundary? Nothing to do */
@@ -2259,8 +2482,8 @@ static int __bch2_truncate_page(struct bch_inode_info *inode,
                 * page
                 */
                ret = range_has_data(c, inode->ei_subvol,
-                               POS(inode->v.i_ino, index << PAGE_SECTOR_SHIFT),
-                               POS(inode->v.i_ino, (index + 1) << PAGE_SECTOR_SHIFT));
+                               POS(inode->v.i_ino, index << PAGE_SECTORS_SHIFT),
+                               POS(inode->v.i_ino, (index + 1) << PAGE_SECTORS_SHIFT));
                if (ret <= 0)
                        return ret;
 
@@ -2292,9 +2515,21 @@ static int __bch2_truncate_page(struct bch_inode_info *inode,
             i < round_down(end_offset, block_bytes(c)) >> 9;
             i++) {
                s->s[i].nr_replicas     = 0;
+               if (s->s[i].state == SECTOR_DIRTY)
+                       i_sectors_delta--;
                s->s[i].state           = SECTOR_UNALLOCATED;
        }
 
+       i_sectors_acct(c, inode, NULL, i_sectors_delta);
+
+       /*
+        * Caller needs to know whether this page will be written out by
+        * writeback - doing an i_size update if necessary - or whether it will
+        * be responsible for the i_size update:
+        */
+       ret = s->s[(min_t(u64, inode->v.i_size - (index << PAGE_SHIFT),
+                         PAGE_SIZE) - 1) >> 9].state >= SECTOR_DIRTY;
+
        zero_user_segment(page, start_offset, end_offset);
 
        /*
@@ -2303,8 +2538,7 @@ static int __bch2_truncate_page(struct bch_inode_info *inode,
         * XXX: because we aren't currently tracking whether the page has actual
         * data in it (vs. just 0s, or only partially written) this wrong. ick.
         */
-       ret = bch2_get_page_disk_reservation(c, inode, page, false);
-       BUG_ON(ret);
+       BUG_ON(bch2_get_page_disk_reservation(c, inode, page, false));
 
        /*
         * This removes any writeable userspace mappings; we need to force
@@ -2326,6 +2560,20 @@ static int bch2_truncate_page(struct bch_inode_info *inode, loff_t from)
                                    from, round_up(from, PAGE_SIZE));
 }
 
+static int bch2_truncate_pages(struct bch_inode_info *inode,
+                              loff_t start, loff_t end)
+{
+       int ret = __bch2_truncate_page(inode, start >> PAGE_SHIFT,
+                                      start, end);
+
+       if (ret >= 0 &&
+           start >> PAGE_SHIFT != end >> PAGE_SHIFT)
+               ret = __bch2_truncate_page(inode,
+                                          end >> PAGE_SHIFT,
+                                          start, end);
+       return ret;
+}
+
 static int bch2_extend(struct user_namespace *mnt_userns,
                       struct bch_inode_info *inode,
                       struct bch_inode_unpacked *inode_u,
@@ -2416,7 +2664,7 @@ int bch2_truncate(struct user_namespace *mnt_userns,
        iattr->ia_valid &= ~ATTR_SIZE;
 
        ret = bch2_truncate_page(inode, iattr->ia_size);
-       if (unlikely(ret))
+       if (unlikely(ret < 0))
                goto err;
 
        /*
@@ -2452,9 +2700,12 @@ int bch2_truncate(struct user_namespace *mnt_userns,
 
        ret = bch2_fpunch(c, inode_inum(inode),
                        round_up(iattr->ia_size, block_bytes(c)) >> 9,
-                       U64_MAX, &inode->ei_journal_seq, &i_sectors_delta);
+                       U64_MAX, &i_sectors_delta);
        i_sectors_acct(c, inode, NULL, i_sectors_delta);
 
+       WARN_ON(!inode->v.i_size && inode->v.i_blocks &&
+               !bch2_journal_error(&c->journal));
+
        if (unlikely(ret))
                goto err;
 
@@ -2482,49 +2733,39 @@ static int inode_update_times_fn(struct bch_inode_info *inode,
 static long bchfs_fpunch(struct bch_inode_info *inode, loff_t offset, loff_t len)
 {
        struct bch_fs *c = inode->v.i_sb->s_fs_info;
-       u64 discard_start = round_up(offset, block_bytes(c)) >> 9;
-       u64 discard_end = round_down(offset + len, block_bytes(c)) >> 9;
+       u64 end         = offset + len;
+       u64 block_start = round_up(offset, block_bytes(c));
+       u64 block_end   = round_down(end, block_bytes(c));
+       bool truncated_last_page;
        int ret = 0;
 
-       inode_lock(&inode->v);
-       inode_dio_wait(&inode->v);
-       bch2_pagecache_block_get(&inode->ei_pagecache_lock);
-
-       ret = __bch2_truncate_page(inode,
-                                  offset >> PAGE_SHIFT,
-                                  offset, offset + len);
-       if (unlikely(ret))
+       ret = bch2_truncate_pages(inode, offset, end);
+       if (unlikely(ret < 0))
                goto err;
 
-       if (offset >> PAGE_SHIFT !=
-           (offset + len) >> PAGE_SHIFT) {
-               ret = __bch2_truncate_page(inode,
-                                          (offset + len) >> PAGE_SHIFT,
-                                          offset, offset + len);
-               if (unlikely(ret))
-                       goto err;
-       }
+       truncated_last_page = ret;
 
-       truncate_pagecache_range(&inode->v, offset, offset + len - 1);
+       truncate_pagecache_range(&inode->v, offset, end - 1);
 
-       if (discard_start < discard_end) {
+       if (block_start < block_end ) {
                s64 i_sectors_delta = 0;
 
                ret = bch2_fpunch(c, inode_inum(inode),
-                                 discard_start, discard_end,
-                                 &inode->ei_journal_seq,
+                                 block_start >> 9, block_end >> 9,
                                  &i_sectors_delta);
                i_sectors_acct(c, inode, NULL, i_sectors_delta);
        }
 
        mutex_lock(&inode->ei_update_lock);
-       ret = bch2_write_inode(c, inode, inode_update_times_fn, NULL,
-                              ATTR_MTIME|ATTR_CTIME) ?: ret;
+       if (end >= inode->v.i_size && !truncated_last_page) {
+               ret = bch2_write_inode_size(c, inode, inode->v.i_size,
+                                           ATTR_MTIME|ATTR_CTIME);
+       } else {
+               ret = bch2_write_inode(c, inode, inode_update_times_fn, NULL,
+                                      ATTR_MTIME|ATTR_CTIME);
+       }
        mutex_unlock(&inode->ei_update_lock);
 err:
-       bch2_pagecache_block_put(&inode->ei_pagecache_lock);
-       inode_unlock(&inode->v);
-
        return ret;
 }
 
@@ -2544,31 +2785,18 @@ static long bchfs_fcollapse_finsert(struct bch_inode_info *inode,
        if ((offset | len) & (block_bytes(c) - 1))
                return -EINVAL;
 
-       /*
-        * We need i_mutex to keep the page cache consistent with the extents
-        * btree, and the btree consistent with i_size - we don't need outside
-        * locking for the extents btree itself, because we're using linked
-        * iterators
-        */
-       inode_lock(&inode->v);
-       inode_dio_wait(&inode->v);
-       bch2_pagecache_block_get(&inode->ei_pagecache_lock);
-
        if (insert) {
-               ret = -EFBIG;
                if (inode->v.i_sb->s_maxbytes - inode->v.i_size < len)
-                       goto err;
+                       return -EFBIG;
 
-               ret = -EINVAL;
                if (offset >= inode->v.i_size)
-                       goto err;
+                       return -EINVAL;
 
                src_start       = U64_MAX;
                shift           = len;
        } else {
-               ret = -EINVAL;
                if (offset + len >= inode->v.i_size)
-                       goto err;
+                       return -EINVAL;
 
                src_start       = offset + len;
                shift           = -len;
@@ -2578,7 +2806,7 @@ static long bchfs_fcollapse_finsert(struct bch_inode_info *inode,
 
        ret = write_invalidate_inode_pages_range(mapping, offset, LLONG_MAX);
        if (ret)
-               goto err;
+               return ret;
 
        if (insert) {
                i_size_write(&inode->v, new_size);
@@ -2591,12 +2819,11 @@ static long bchfs_fcollapse_finsert(struct bch_inode_info *inode,
 
                ret = bch2_fpunch(c, inode_inum(inode),
                                  offset >> 9, (offset + len) >> 9,
-                                 &inode->ei_journal_seq,
                                  &i_sectors_delta);
                i_sectors_acct(c, inode, NULL, i_sectors_delta);
 
                if (ret)
-                       goto err;
+                       return ret;
        }
 
        bch2_bkey_buf_init(&copy);
@@ -2695,8 +2922,7 @@ reassemble:
                ret =   bch2_btree_iter_traverse(&del) ?:
                        bch2_trans_update(&trans, &del, &delete, trigger_flags) ?:
                        bch2_trans_update(&trans, &dst, copy.k, trigger_flags) ?:
-                       bch2_trans_commit(&trans, &disk_res,
-                                         &inode->ei_journal_seq,
+                       bch2_trans_commit(&trans, &disk_res, NULL,
                                          BTREE_INSERT_NOFAIL);
                bch2_disk_reservation_put(c, &disk_res);
 
@@ -2710,18 +2936,19 @@ reassemble:
        bch2_bkey_buf_exit(&copy, c);
 
        if (ret)
-               goto err;
+               return ret;
 
+       mutex_lock(&inode->ei_update_lock);
        if (!insert) {
                i_size_write(&inode->v, new_size);
-               mutex_lock(&inode->ei_update_lock);
                ret = bch2_write_inode_size(c, inode, new_size,
                                            ATTR_MTIME|ATTR_CTIME);
-               mutex_unlock(&inode->ei_update_lock);
+       } else {
+               /* We need an inode update to update bi_journal_seq for fsync: */
+               ret = bch2_write_inode(c, inode, inode_update_times_fn, NULL,
+                                      ATTR_MTIME|ATTR_CTIME);
        }
-err:
-       bch2_pagecache_block_put(&inode->ei_pagecache_lock);
-       inode_unlock(&inode->v);
+       mutex_unlock(&inode->ei_update_lock);
        return ret;
 }
 
@@ -2807,8 +3034,10 @@ static int __bchfs_fallocate(struct bch_inode_info *inode, int mode,
 
                ret = bch2_extent_update(&trans, inode_inum(inode), &iter,
                                         &reservation.k_i,
-                               &disk_res, &inode->ei_journal_seq,
+                               &disk_res, NULL,
                                0, &i_sectors_delta, true);
+               if (ret)
+                       goto bkey_err;
                i_sectors_acct(c, inode, &quota_res, i_sectors_delta);
 bkey_err:
                bch2_quota_reservation_put(c, inode, &quota_res);
@@ -2816,6 +3045,20 @@ bkey_err:
                if (ret == -EINTR)
                        ret = 0;
        }
+
+       bch2_trans_unlock(&trans); /* lock ordering, before taking pagecache locks: */
+       mark_pagecache_reserved(inode, start_sector, iter.pos.offset);
+
+       if (ret == -ENOSPC && (mode & FALLOC_FL_ZERO_RANGE)) {
+               struct quota_res quota_res = { 0 };
+               s64 i_sectors_delta = 0;
+
+               bch2_fpunch_at(&trans, &iter, inode_inum(inode),
+                              end_sector, &i_sectors_delta);
+               i_sectors_acct(c, inode, &quota_res, i_sectors_delta);
+               bch2_quota_reservation_put(c, inode, &quota_res);
+       }
+
        bch2_trans_iter_exit(&trans, &iter);
        bch2_trans_exit(&trans);
        return ret;
@@ -2824,77 +3067,58 @@ bkey_err:
 static long bchfs_fallocate(struct bch_inode_info *inode, int mode,
                            loff_t offset, loff_t len)
 {
-       struct address_space *mapping = inode->v.i_mapping;
        struct bch_fs *c = inode->v.i_sb->s_fs_info;
-       loff_t end              = offset + len;
-       loff_t block_start      = round_down(offset,    block_bytes(c));
-       loff_t block_end        = round_up(end,         block_bytes(c));
-       int ret;
-
-       inode_lock(&inode->v);
-       inode_dio_wait(&inode->v);
-       bch2_pagecache_block_get(&inode->ei_pagecache_lock);
+       u64 end         = offset + len;
+       u64 block_start = round_down(offset,    block_bytes(c));
+       u64 block_end   = round_up(end,         block_bytes(c));
+       bool truncated_last_page = false;
+       int ret, ret2 = 0;
 
        if (!(mode & FALLOC_FL_KEEP_SIZE) && end > inode->v.i_size) {
                ret = inode_newsize_ok(&inode->v, end);
                if (ret)
-                       goto err;
+                       return ret;
        }
 
        if (mode & FALLOC_FL_ZERO_RANGE) {
-               ret = __bch2_truncate_page(inode,
-                                          offset >> PAGE_SHIFT,
-                                          offset, end);
-
-               if (!ret &&
-                   offset >> PAGE_SHIFT != end >> PAGE_SHIFT)
-                       ret = __bch2_truncate_page(inode,
-                                                  end >> PAGE_SHIFT,
-                                                  offset, end);
+               ret = bch2_truncate_pages(inode, offset, end);
+               if (unlikely(ret < 0))
+                       return ret;
 
-               if (unlikely(ret))
-                       goto err;
+               truncated_last_page = ret;
 
                truncate_pagecache_range(&inode->v, offset, end - 1);
+
+               block_start     = round_up(offset,      block_bytes(c));
+               block_end       = round_down(end,       block_bytes(c));
        }
 
        ret = __bchfs_fallocate(inode, mode, block_start >> 9, block_end >> 9);
-       if (ret)
-               goto err;
 
        /*
-        * Do we need to extend the file?
-        *
-        * If we zeroed up to the end of the file, we dropped whatever writes
-        * were going to write out the current i_size, so we have to extend
-        * manually even if FL_KEEP_SIZE was set:
+        * On -ENOSPC in ZERO_RANGE mode, we still want to do the inode update,
+        * so that the VFS cache i_size is consistent with the btree i_size:
         */
-       if (end >= inode->v.i_size &&
-           (!(mode & FALLOC_FL_KEEP_SIZE) ||
-            (mode & FALLOC_FL_ZERO_RANGE))) {
+       if (ret &&
+           !(ret == -ENOSPC && (mode & FALLOC_FL_ZERO_RANGE)))
+               return ret;
 
-               /*
-                * Sync existing appends before extending i_size,
-                * as in bch2_extend():
-                */
-               ret = filemap_write_and_wait_range(mapping,
-                                       inode->ei_inode.bi_size, S64_MAX);
-               if (ret)
-                       goto err;
+       if (mode & FALLOC_FL_KEEP_SIZE && end > inode->v.i_size)
+               end = inode->v.i_size;
 
-               if (mode & FALLOC_FL_KEEP_SIZE)
-                       end = inode->v.i_size;
-               else
-                       i_size_write(&inode->v, end);
+       if (end >= inode->v.i_size &&
+           (((mode & FALLOC_FL_ZERO_RANGE) && !truncated_last_page) ||
+            !(mode & FALLOC_FL_KEEP_SIZE))) {
+               spin_lock(&inode->v.i_lock);
+               i_size_write(&inode->v, end);
+               spin_unlock(&inode->v.i_lock);
 
                mutex_lock(&inode->ei_update_lock);
-               ret = bch2_write_inode_size(c, inode, end, 0);
+               ret2 = bch2_write_inode_size(c, inode, end, 0);
                mutex_unlock(&inode->ei_update_lock);
        }
-err:
-       bch2_pagecache_block_put(&inode->ei_pagecache_lock);
-       inode_unlock(&inode->v);
-       return ret;
+
+       return ret ?: ret2;
 }
 
 long bch2_fallocate_dispatch(struct file *file, int mode,
@@ -2907,6 +3131,10 @@ long bch2_fallocate_dispatch(struct file *file, int mode,
        if (!percpu_ref_tryget(&c->writes))
                return -EROFS;
 
+       inode_lock(&inode->v);
+       inode_dio_wait(&inode->v);
+       bch2_pagecache_block_get(&inode->ei_pagecache_lock);
+
        if (!(mode & ~(FALLOC_FL_KEEP_SIZE|FALLOC_FL_ZERO_RANGE)))
                ret = bchfs_fallocate(inode, mode, offset, len);
        else if (mode == (FALLOC_FL_PUNCH_HOLE|FALLOC_FL_KEEP_SIZE))
@@ -2918,48 +3146,14 @@ long bch2_fallocate_dispatch(struct file *file, int mode,
        else
                ret = -EOPNOTSUPP;
 
+
+       bch2_pagecache_block_put(&inode->ei_pagecache_lock);
+       inode_unlock(&inode->v);
        percpu_ref_put(&c->writes);
 
        return ret;
 }
 
-static void mark_range_unallocated(struct bch_inode_info *inode,
-                                  loff_t start, loff_t end)
-{
-       pgoff_t index = start >> PAGE_SHIFT;
-       pgoff_t end_index = (end - 1) >> PAGE_SHIFT;
-       struct pagevec pvec;
-
-       pagevec_init(&pvec);
-
-       do {
-               unsigned nr_pages, i, j;
-
-               nr_pages = pagevec_lookup_range(&pvec, inode->v.i_mapping,
-                                               &index, end_index);
-               if (nr_pages == 0)
-                       break;
-
-               for (i = 0; i < nr_pages; i++) {
-                       struct page *page = pvec.pages[i];
-                       struct bch_page_state *s;
-
-                       lock_page(page);
-                       s = bch2_page_state(page);
-
-                       if (s) {
-                               spin_lock(&s->lock);
-                               for (j = 0; j < PAGE_SECTORS; j++)
-                                       s->s[j].nr_replicas = 0;
-                               spin_unlock(&s->lock);
-                       }
-
-                       unlock_page(page);
-               }
-               pagevec_release(&pvec);
-       } while (index <= end_index);
-}
-
 loff_t bch2_remap_file_range(struct file *file_src, loff_t pos_src,
                             struct file *file_dst, loff_t pos_dst,
                             loff_t len, unsigned remap_flags)
@@ -3005,13 +3199,13 @@ loff_t bch2_remap_file_range(struct file *file_src, loff_t pos_src,
        if (ret)
                goto err;
 
-       mark_range_unallocated(src, pos_src, pos_src + aligned_len);
+       mark_pagecache_unallocated(src, pos_src >> 9,
+                                  (pos_src + aligned_len) >> 9);
 
        ret = bch2_remap_range(c,
                               inode_inum(dst), pos_dst >> 9,
                               inode_inum(src), pos_src >> 9,
                               aligned_len >> 9,
-                              &dst->ei_journal_seq,
                               pos_dst + len, &i_sectors_delta);
        if (ret < 0)
                goto err;
@@ -3029,10 +3223,9 @@ loff_t bch2_remap_file_range(struct file *file_src, loff_t pos_src,
                i_size_write(&dst->v, pos_dst + ret);
        spin_unlock(&dst->v.i_lock);
 
-       if (((file_dst->f_flags & (__O_SYNC | O_DSYNC)) ||
-            IS_SYNC(file_inode(file_dst))) &&
-           !c->opts.journal_flush_disabled)
-               ret = bch2_journal_flush_seq(&c->journal, dst->ei_journal_seq);
+       if ((file_dst->f_flags & (__O_SYNC | O_DSYNC)) ||
+           IS_SYNC(file_inode(file_dst)))
+               ret = bch2_flush_inode(c, inode_inum(dst));
 err:
        bch2_unlock_inodes(INODE_LOCK|INODE_PAGECACHE_BLOCK, src, dst);
 
@@ -3117,7 +3310,7 @@ retry:
        if (ret)
                goto err;
 
-       for_each_btree_key(&trans, iter, BTREE_ID_extents,
+       for_each_btree_key_norestart(&trans, iter, BTREE_ID_extents,
                           SPOS(inode->v.i_ino, offset >> 9, snapshot), 0, k, ret) {
                if (k.k->p.inode != inode->v.i_ino) {
                        break;
@@ -3132,7 +3325,7 @@ err:
        if (ret == -EINTR)
                goto retry;
 
-       ret = bch2_trans_exit(&trans) ?: ret;
+       bch2_trans_exit(&trans);
        if (ret)
                return ret;
 
@@ -3224,7 +3417,7 @@ retry:
        if (ret)
                goto err;
 
-       for_each_btree_key(&trans, iter, BTREE_ID_extents,
+       for_each_btree_key_norestart(&trans, iter, BTREE_ID_extents,
                           SPOS(inode->v.i_ino, offset >> 9, snapshot),
                           BTREE_ITER_SLOTS, k, ret) {
                if (k.k->p.inode != inode->v.i_ino) {
@@ -3247,7 +3440,7 @@ err:
        if (ret == -EINTR)
                goto retry;
 
-       ret = bch2_trans_exit(&trans) ?: ret;
+       bch2_trans_exit(&trans);
        if (ret)
                return ret;