ext4_bcache: remove buffer from LRU tree when it will be dropped.

[lwext4.git] / lwext4 / ext4_bcache.c

diff --git a/lwext4/ext4_bcache.c b/lwext4/ext4_bcache.c
index b4e247c6dce3df94bea306a65aa69b8e7d4dbbfa..628735ab659fe38b5aac1e7fda1f660210f6fafb 100644 (file)
--- a/lwext4/ext4_bcache.c
+++ b/lwext4/ext4_bcache.c
@@ -81,6 +81,15 @@ int ext4_bcache_init_dynamic(struct ext4_bcache *bc, uint32_t cnt,
        return EOK;
 }
 
+void ext4_bcache_cleanup(struct ext4_bcache *bc)
+{
+       struct ext4_buf *buf, *tmp;
+       RB_FOREACH_SAFE(buf, ext4_buf_lba, &bc->lba_root, tmp) {
+               ext4_block_flush_buf(bc->bdev, buf);
+               ext4_bcache_drop_buf(bc, buf);
+       }
+}
+
 int ext4_bcache_fini_dynamic(struct ext4_bcache *bc)
 {
        memset(bc, 0, sizeof(struct ext4_bcache));
@@ -154,10 +163,10 @@ void ext4_bcache_drop_buf(struct ext4_bcache *bc, struct ext4_buf *buf)
                ext4_dbg(DEBUG_BCACHE, DBG_WARN "Buffer is still referenced. "
                                "lba: %" PRIu64 ", refctr: %" PRIu32 "\n",
                                buf->lba, buf->refctr);
-       }
+       } else
+               RB_REMOVE(ext4_buf_lru, &bc->lru_root, buf);
 
        RB_REMOVE(ext4_buf_lba, &bc->lba_root, buf);
-       RB_REMOVE(ext4_buf_lru, &bc->lru_root, buf);
 
        /*Forcibly drop dirty buffer.*/
        if (ext4_bcache_test_flag(buf, BC_DIRTY))
@@ -167,11 +176,29 @@ void ext4_bcache_drop_buf(struct ext4_bcache *bc, struct ext4_buf *buf)
        bc->ref_blocks--;
 }
 
-int ext4_bcache_alloc(struct ext4_bcache *bc, struct ext4_block *b,
-                     bool *is_new)
+void ext4_bcache_invalidate_lba(struct ext4_bcache *bc,
+                               uint64_t from,
+                               uint32_t cnt)
 {
-       /* Try to search the buffer with exaxt LBA. */
-       struct ext4_buf *buf = ext4_buf_lookup(bc, b->lb_id);
+       uint64_t end = from + cnt - 1;
+       struct ext4_buf *tmp = ext4_buf_lookup(bc, from), *buf;
+       RB_FOREACH_FROM(buf, ext4_buf_lba, tmp) {
+               if (buf->lba > end)
+                       break;
+
+               /* Clear both dirty and up-to-date flags. */
+               if (ext4_bcache_test_flag(buf, BC_DIRTY))
+                       ext4_bcache_remove_dirty_node(bc, buf);
+
+               ext4_bcache_clear_dirty(buf);
+       }
+}
+
+struct ext4_buf *
+ext4_bcache_find_get(struct ext4_bcache *bc, struct ext4_block *b,
+                    uint64_t lba)
+{
+       struct ext4_buf *buf = ext4_buf_lookup(bc, lba);
        if (buf) {
                /* If buffer is not referenced. */
                if (!buf->refctr) {
@@ -188,7 +215,16 @@ int ext4_bcache_alloc(struct ext4_bcache *bc, struct ext4_block *b,
 
                b->buf = buf;
                b->data = buf->data;
+       }
+       return buf;
+}
 
+int ext4_bcache_alloc(struct ext4_bcache *bc, struct ext4_block *b,
+                     bool *is_new)
+{
+       /* Try to search the buffer with exaxt LBA. */
+       struct ext4_buf *buf = ext4_bcache_find_get(bc, b, b->lb_id);
+       if (buf) {
                *is_new = false;
                return EOK;
        }