iomap: Protect read_bytes_pending with the state_lock

Perform one atomic operation (acquiring the spinlock) instead of
two (spinlock & atomic_sub) per read completion.

Signed-off-by: Matthew Wilcox (Oracle) <willy@infradead.org>

diff --git a/fs/iomap/buffered-io.c b/fs/iomap/buffered-io.c
index 6e780ca64ce3430d4f2aa51c15c69e8e2af86a64..4a996c5327efdef42718fba0a27309ef0a4ab08a 100644 (file)
--- a/fs/iomap/buffered-io.c
+++ b/fs/iomap/buffered-io.c
@@ -29,9 +29,9 @@ typedef int (*iomap_punch_t)(struct inode *inode, loff_t offset, loff_t length);
  * and I/O completions.
  */
 struct iomap_folio_state {
-       atomic_t                read_bytes_pending;
-       atomic_t                write_bytes_pending;
        spinlock_t              state_lock;
+       unsigned int            read_bytes_pending;
+       atomic_t                write_bytes_pending;
 
        /*
         * Each block has two bits in this bitmap:
@@ -183,7 +183,7 @@ static void ifs_free(struct folio *folio)
 
        if (!ifs)
                return;
-       WARN_ON_ONCE(atomic_read(&ifs->read_bytes_pending));
+       WARN_ON_ONCE(ifs->read_bytes_pending != 0);
        WARN_ON_ONCE(atomic_read(&ifs->write_bytes_pending));
        WARN_ON_ONCE(ifs_is_fully_uptodate(folio, ifs) !=
                        folio_test_uptodate(folio));
@@ -250,19 +250,29 @@ static void iomap_adjust_read_range(struct inode *inode, struct folio *folio,
        *lenp = plen;
 }
 
-static void iomap_finish_folio_read(struct folio *folio, size_t offset,
+static void iomap_finish_folio_read(struct folio *folio, size_t off,
                size_t len, int error)
 {
        struct iomap_folio_state *ifs = folio->private;
+       bool uptodate = !error;
+       bool finished = true;
 
-       if (unlikely(error)) {
-               folio_clear_uptodate(folio);
-               folio_set_error(folio);
-       } else {
-               iomap_set_range_uptodate(folio, offset, len);
+       if (ifs) {
+               unsigned long flags;
+
+               spin_lock_irqsave(&ifs->state_lock, flags);
+               if (!error)
+                       uptodate = ifs_set_range_uptodate(folio, ifs, off, len);
+               ifs->read_bytes_pending -= len;
+               finished = !ifs->read_bytes_pending;
+               spin_unlock_irqrestore(&ifs->state_lock, flags);
        }
 
-       if (!ifs || atomic_sub_and_test(len, &ifs->read_bytes_pending))
+       if (error)
+               folio_set_error(folio);
+       if (uptodate)
+               folio_mark_uptodate(folio);
+       if (finished)
                folio_unlock(folio);
 }
 
@@ -360,8 +370,11 @@ static loff_t iomap_readpage_iter(const struct iomap_iter *iter,
        }
 
        ctx->cur_folio_in_bio = true;
-       if (ifs)
-               atomic_add(plen, &ifs->read_bytes_pending);
+       if (ifs) {
+               spin_lock_irq(&ifs->state_lock);
+               ifs->read_bytes_pending += plen;
+               spin_unlock_irq(&ifs->state_lock);
+       }
 
        sector = iomap_sector(iomap, pos);
        if (!ctx->bio ||