static LIST_HEAD(states);
 static DEFINE_SPINLOCK(leak_lock);
 
-static inline void btrfs_leak_debug_add(spinlock_t *lock,
-                                       struct list_head *new,
-                                       struct list_head *head)
+static inline void btrfs_leak_debug_add_eb(struct extent_buffer *eb)
+{
+       struct btrfs_fs_info *fs_info = eb->fs_info;
+       unsigned long flags;
+
+       spin_lock_irqsave(&fs_info->eb_leak_lock, flags);
+       list_add(&eb->leak_list, &fs_info->allocated_ebs);
+       spin_unlock_irqrestore(&fs_info->eb_leak_lock, flags);
+}
+
+static inline void btrfs_leak_debug_add_state(struct extent_state *state)
 {
        unsigned long flags;
 
-       spin_lock_irqsave(lock, flags);
-       list_add(new, head);
-       spin_unlock_irqrestore(lock, flags);
+       spin_lock_irqsave(&leak_lock, flags);
+       list_add(&state->leak_list, &states);
+       spin_unlock_irqrestore(&leak_lock, flags);
+}
+
+static inline void btrfs_leak_debug_del_eb(struct extent_buffer *eb)
+{
+       struct btrfs_fs_info *fs_info = eb->fs_info;
+       unsigned long flags;
+
+       spin_lock_irqsave(&fs_info->eb_leak_lock, flags);
+       list_del(&eb->leak_list);
+       spin_unlock_irqrestore(&fs_info->eb_leak_lock, flags);
 }
 
-static inline void btrfs_leak_debug_del(spinlock_t *lock,
-                                       struct list_head *entry)
+static inline void btrfs_leak_debug_del_state(struct extent_state *state)
 {
        unsigned long flags;
 
-       spin_lock_irqsave(lock, flags);
-       list_del(entry);
-       spin_unlock_irqrestore(lock, flags);
+       spin_lock_irqsave(&leak_lock, flags);
+       list_del(&state->leak_list);
+       spin_unlock_irqrestore(&leak_lock, flags);
 }
 
 void btrfs_extent_buffer_leak_debug_check(struct btrfs_fs_info *fs_info)
        }
 }
 #else
-#define btrfs_leak_debug_add(lock, new, head)  do {} while (0)
-#define btrfs_leak_debug_del(lock, entry)      do {} while (0)
-#define btrfs_extent_state_leak_debug_check()  do {} while (0)
+#define btrfs_leak_debug_add_eb(eb)                    do {} while (0)
+#define btrfs_leak_debug_add_state(state)              do {} while (0)
+#define btrfs_leak_debug_del_eb(eb)                    do {} while (0)
+#define btrfs_leak_debug_del_state(state)              do {} while (0)
+#define btrfs_extent_state_leak_debug_check()          do {} while (0)
 #define btrfs_debug_check_extent_io_range(c, s, e)     do {} while (0)
 #endif
 
                return state;
        state->state = 0;
        RB_CLEAR_NODE(&state->rb_node);
-       btrfs_leak_debug_add(&leak_lock, &state->leak_list, &states);
+       btrfs_leak_debug_add_state(state);
        refcount_set(&state->refs, 1);
        init_waitqueue_head(&state->wq);
        trace_alloc_extent_state(state, mask, _RET_IP_);
                return;
        if (refcount_dec_and_test(&state->refs)) {
                WARN_ON(extent_state_in_tree(state));
-               btrfs_leak_debug_del(&leak_lock, &state->leak_list);
+               btrfs_leak_debug_del_state(state);
                trace_free_extent_state(state, _RET_IP_);
                kmem_cache_free(extent_state_cache, state);
        }
 static inline void btrfs_release_extent_buffer(struct extent_buffer *eb)
 {
        btrfs_release_extent_buffer_pages(eb);
-       btrfs_leak_debug_del(&eb->fs_info->eb_leak_lock, &eb->leak_list);
+       btrfs_leak_debug_del_eb(eb);
        __free_extent_buffer(eb);
 }
 
        eb->bflags = 0;
        init_rwsem(&eb->lock);
 
-       btrfs_leak_debug_add(&fs_info->eb_leak_lock, &eb->leak_list,
-                            &fs_info->allocated_ebs);
+       btrfs_leak_debug_add_eb(eb);
        INIT_LIST_HEAD(&eb->release_list);
 
        spin_lock_init(&eb->refs_lock);
                        spin_unlock(&eb->refs_lock);
                }
 
-               btrfs_leak_debug_del(&eb->fs_info->eb_leak_lock, &eb->leak_list);
+               btrfs_leak_debug_del_eb(eb);
                /* Should be safe to release our pages at this point */
                btrfs_release_extent_buffer_pages(eb);
 #ifdef CONFIG_BTRFS_FS_RUN_SANITY_TESTS