static struct kmem_cache *ext4_es_cachep;
 
-static int __es_insert_extent(struct ext4_es_tree *tree,
-                             struct extent_status *newes);
-static int __es_remove_extent(struct ext4_es_tree *tree, ext4_lblk_t lblk,
+static int __es_insert_extent(struct inode *inode, struct extent_status *newes);
+static int __es_remove_extent(struct inode *inode, ext4_lblk_t lblk,
                              ext4_lblk_t end);
 
 int __init ext4_init_es(void)
 }
 
 static struct extent_status *
-ext4_es_alloc_extent(ext4_lblk_t lblk, ext4_lblk_t len, ext4_fsblk_t pblk)
+ext4_es_alloc_extent(struct inode *inode, ext4_lblk_t lblk, ext4_lblk_t len,
+                    ext4_fsblk_t pblk)
 {
        struct extent_status *es;
        es = kmem_cache_alloc(ext4_es_cachep, GFP_ATOMIC);
        return es;
 }
 
-static void ext4_es_free_extent(struct extent_status *es)
+static void ext4_es_free_extent(struct inode *inode, struct extent_status *es)
 {
        kmem_cache_free(ext4_es_cachep, es);
 }
 }
 
 static struct extent_status *
-ext4_es_try_to_merge_left(struct ext4_es_tree *tree, struct extent_status *es)
+ext4_es_try_to_merge_left(struct inode *inode, struct extent_status *es)
 {
+       struct ext4_es_tree *tree = &EXT4_I(inode)->i_es_tree;
        struct extent_status *es1;
        struct rb_node *node;
 
        if (ext4_es_can_be_merged(es1, es)) {
                es1->es_len += es->es_len;
                rb_erase(&es->rb_node, &tree->root);
-               ext4_es_free_extent(es);
+               ext4_es_free_extent(inode, es);
                es = es1;
        }
 
 }
 
 static struct extent_status *
-ext4_es_try_to_merge_right(struct ext4_es_tree *tree, struct extent_status *es)
+ext4_es_try_to_merge_right(struct inode *inode, struct extent_status *es)
 {
+       struct ext4_es_tree *tree = &EXT4_I(inode)->i_es_tree;
        struct extent_status *es1;
        struct rb_node *node;
 
        if (ext4_es_can_be_merged(es, es1)) {
                es->es_len += es1->es_len;
                rb_erase(node, &tree->root);
-               ext4_es_free_extent(es1);
+               ext4_es_free_extent(inode, es1);
        }
 
        return es;
 }
 
-static int __es_insert_extent(struct ext4_es_tree *tree,
-                             struct extent_status *newes)
+static int __es_insert_extent(struct inode *inode, struct extent_status *newes)
 {
+       struct ext4_es_tree *tree = &EXT4_I(inode)->i_es_tree;
        struct rb_node **p = &tree->root.rb_node;
        struct rb_node *parent = NULL;
        struct extent_status *es;
                                    ext4_es_is_unwritten(es))
                                        ext4_es_store_pblock(es,
                                                             newes->es_pblk);
-                               es = ext4_es_try_to_merge_left(tree, es);
+                               es = ext4_es_try_to_merge_left(inode, es);
                                goto out;
                        }
                        p = &(*p)->rb_left;
                } else if (newes->es_lblk > ext4_es_end(es)) {
                        if (ext4_es_can_be_merged(es, newes)) {
                                es->es_len += newes->es_len;
-                               es = ext4_es_try_to_merge_right(tree, es);
+                               es = ext4_es_try_to_merge_right(inode, es);
                                goto out;
                        }
                        p = &(*p)->rb_right;
                }
        }
 
-       es = ext4_es_alloc_extent(newes->es_lblk, newes->es_len,
+       es = ext4_es_alloc_extent(inode, newes->es_lblk, newes->es_len,
                                  newes->es_pblk);
        if (!es)
                return -ENOMEM;
                          ext4_lblk_t len, ext4_fsblk_t pblk,
                          unsigned long long status)
 {
-       struct ext4_es_tree *tree;
        struct extent_status newes;
        ext4_lblk_t end = lblk + len - 1;
        int err = 0;
        trace_ext4_es_insert_extent(inode, &newes);
 
        write_lock(&EXT4_I(inode)->i_es_lock);
-       tree = &EXT4_I(inode)->i_es_tree;
-       err = __es_remove_extent(tree, lblk, end);
+       err = __es_remove_extent(inode, lblk, end);
        if (err != 0)
                goto error;
-       err = __es_insert_extent(tree, &newes);
+       err = __es_insert_extent(inode, &newes);
 
 error:
        write_unlock(&EXT4_I(inode)->i_es_lock);
        return found;
 }
 
-static int __es_remove_extent(struct ext4_es_tree *tree, ext4_lblk_t lblk,
-                                ext4_lblk_t end)
+static int __es_remove_extent(struct inode *inode, ext4_lblk_t lblk,
+                             ext4_lblk_t end)
 {
+       struct ext4_es_tree *tree = &EXT4_I(inode)->i_es_tree;
        struct rb_node *node;
        struct extent_status *es;
        struct extent_status orig_es;
                                ext4_es_store_pblock(&newes, block);
                        }
                        ext4_es_store_status(&newes, ext4_es_status(&orig_es));
-                       err = __es_insert_extent(tree, &newes);
+                       err = __es_insert_extent(inode, &newes);
                        if (err) {
                                es->es_lblk = orig_es.es_lblk;
                                es->es_len = orig_es.es_len;
        while (es && ext4_es_end(es) <= end) {
                node = rb_next(&es->rb_node);
                rb_erase(&es->rb_node, &tree->root);
-               ext4_es_free_extent(es);
+               ext4_es_free_extent(inode, es);
                if (!node) {
                        es = NULL;
                        break;
 int ext4_es_remove_extent(struct inode *inode, ext4_lblk_t lblk,
                          ext4_lblk_t len)
 {
-       struct ext4_es_tree *tree;
        ext4_lblk_t end;
        int err = 0;
 
        end = lblk + len - 1;
        BUG_ON(end < lblk);
 
-       tree = &EXT4_I(inode)->i_es_tree;
-
        write_lock(&EXT4_I(inode)->i_es_lock);
-       err = __es_remove_extent(tree, lblk, end);
+       err = __es_remove_extent(inode, lblk, end);
        write_unlock(&EXT4_I(inode)->i_es_lock);
        ext4_es_print_tree(inode);
        return err;