blob: 27fcdd2b2607ebdf2e427a5ac5bce5135fc51764 [file] [log] [blame]
Zheng Liu654598b2012-11-08 21:57:20 -05001/*
2 * fs/ext4/extents_status.c
3 *
4 * Written by Yongqiang Yang <xiaoqiangnk@gmail.com>
5 * Modified by
6 * Allison Henderson <achender@linux.vnet.ibm.com>
7 * Hugh Dickins <hughd@google.com>
8 * Zheng Liu <wenqing.lz@taobao.com>
9 *
10 * Ext4 extents status tree core functions.
11 */
12#include <linux/rbtree.h>
13#include "ext4.h"
14#include "extents_status.h"
15#include "ext4_extents.h"
16
Zheng Liu992e9fd2012-11-08 21:57:33 -050017#include <trace/events/ext4.h>
18
Zheng Liu654598b2012-11-08 21:57:20 -050019/*
20 * According to previous discussion in Ext4 Developer Workshop, we
21 * will introduce a new structure called io tree to track all extent
22 * status in order to solve some problems that we have met
23 * (e.g. Reservation space warning), and provide extent-level locking.
24 * Delay extent tree is the first step to achieve this goal. It is
25 * original built by Yongqiang Yang. At that time it is called delay
Zheng Liu06b0c882013-02-18 00:26:51 -050026 * extent tree, whose goal is only track delayed extents in memory to
Zheng Liu654598b2012-11-08 21:57:20 -050027 * simplify the implementation of fiemap and bigalloc, and introduce
28 * lseek SEEK_DATA/SEEK_HOLE support. That is why it is still called
Zheng Liu06b0c882013-02-18 00:26:51 -050029 * delay extent tree at the first commit. But for better understand
30 * what it does, it has been rename to extent status tree.
Zheng Liu654598b2012-11-08 21:57:20 -050031 *
Zheng Liu06b0c882013-02-18 00:26:51 -050032 * Step1:
33 * Currently the first step has been done. All delayed extents are
34 * tracked in the tree. It maintains the delayed extent when a delayed
35 * allocation is issued, and the delayed extent is written out or
Zheng Liu654598b2012-11-08 21:57:20 -050036 * invalidated. Therefore the implementation of fiemap and bigalloc
37 * are simplified, and SEEK_DATA/SEEK_HOLE are introduced.
38 *
39 * The following comment describes the implemenmtation of extent
40 * status tree and future works.
Zheng Liu06b0c882013-02-18 00:26:51 -050041 *
42 * Step2:
43 * In this step all extent status are tracked by extent status tree.
44 * Thus, we can first try to lookup a block mapping in this tree before
45 * finding it in extent tree. Hence, single extent cache can be removed
46 * because extent status tree can do a better job. Extents in status
47 * tree are loaded on-demand. Therefore, the extent status tree may not
48 * contain all of the extents in a file. Meanwhile we define a shrinker
49 * to reclaim memory from extent status tree because fragmented extent
50 * tree will make status tree cost too much memory. written/unwritten/-
51 * hole extents in the tree will be reclaimed by this shrinker when we
52 * are under high memory pressure. Delayed extents will not be
53 * reclimed because fiemap, bigalloc, and seek_data/hole need it.
Zheng Liu654598b2012-11-08 21:57:20 -050054 */
55
56/*
Zheng Liu06b0c882013-02-18 00:26:51 -050057 * Extent status tree implementation for ext4.
Zheng Liu654598b2012-11-08 21:57:20 -050058 *
59 *
60 * ==========================================================================
Zheng Liu06b0c882013-02-18 00:26:51 -050061 * Extent status tree tracks all extent status.
Zheng Liu654598b2012-11-08 21:57:20 -050062 *
Zheng Liu06b0c882013-02-18 00:26:51 -050063 * 1. Why we need to implement extent status tree?
Zheng Liu654598b2012-11-08 21:57:20 -050064 *
Zheng Liu06b0c882013-02-18 00:26:51 -050065 * Without extent status tree, ext4 identifies a delayed extent by looking
Zheng Liu654598b2012-11-08 21:57:20 -050066 * up page cache, this has several deficiencies - complicated, buggy,
67 * and inefficient code.
68 *
Zheng Liu06b0c882013-02-18 00:26:51 -050069 * FIEMAP, SEEK_HOLE/DATA, bigalloc, and writeout all need to know if a
70 * block or a range of blocks are belonged to a delayed extent.
Zheng Liu654598b2012-11-08 21:57:20 -050071 *
Zheng Liu06b0c882013-02-18 00:26:51 -050072 * Let us have a look at how they do without extent status tree.
Zheng Liu654598b2012-11-08 21:57:20 -050073 * -- FIEMAP
74 * FIEMAP looks up page cache to identify delayed allocations from holes.
75 *
76 * -- SEEK_HOLE/DATA
77 * SEEK_HOLE/DATA has the same problem as FIEMAP.
78 *
79 * -- bigalloc
80 * bigalloc looks up page cache to figure out if a block is
81 * already under delayed allocation or not to determine whether
82 * quota reserving is needed for the cluster.
83 *
Zheng Liu654598b2012-11-08 21:57:20 -050084 * -- writeout
85 * Writeout looks up whole page cache to see if a buffer is
86 * mapped, If there are not very many delayed buffers, then it is
87 * time comsuming.
88 *
Zheng Liu06b0c882013-02-18 00:26:51 -050089 * With extent status tree implementation, FIEMAP, SEEK_HOLE/DATA,
Zheng Liu654598b2012-11-08 21:57:20 -050090 * bigalloc and writeout can figure out if a block or a range of
91 * blocks is under delayed allocation(belonged to a delayed extent) or
Zheng Liu06b0c882013-02-18 00:26:51 -050092 * not by searching the extent tree.
Zheng Liu654598b2012-11-08 21:57:20 -050093 *
94 *
95 * ==========================================================================
Zheng Liu06b0c882013-02-18 00:26:51 -050096 * 2. Ext4 extent status tree impelmentation
Zheng Liu654598b2012-11-08 21:57:20 -050097 *
Zheng Liu06b0c882013-02-18 00:26:51 -050098 * -- extent
99 * A extent is a range of blocks which are contiguous logically and
100 * physically. Unlike extent in extent tree, this extent in ext4 is
101 * a in-memory struct, there is no corresponding on-disk data. There
102 * is no limit on length of extent, so an extent can contain as many
103 * blocks as they are contiguous logically and physically.
Zheng Liu654598b2012-11-08 21:57:20 -0500104 *
Zheng Liu06b0c882013-02-18 00:26:51 -0500105 * -- extent status tree
106 * Every inode has an extent status tree and all allocation blocks
107 * are added to the tree with different status. The extent in the
108 * tree are ordered by logical block no.
Zheng Liu654598b2012-11-08 21:57:20 -0500109 *
Zheng Liu06b0c882013-02-18 00:26:51 -0500110 * -- operations on a extent status tree
111 * There are three important operations on a delayed extent tree: find
112 * next extent, adding a extent(a range of blocks) and removing a extent.
Zheng Liu654598b2012-11-08 21:57:20 -0500113 *
Zheng Liu06b0c882013-02-18 00:26:51 -0500114 * -- race on a extent status tree
115 * Extent status tree is protected by inode->i_es_lock.
116 *
117 * -- memory consumption
118 * Fragmented extent tree will make extent status tree cost too much
119 * memory. Hence, we will reclaim written/unwritten/hole extents from
120 * the tree under a heavy memory pressure.
Zheng Liu654598b2012-11-08 21:57:20 -0500121 *
122 *
123 * ==========================================================================
Zheng Liu06b0c882013-02-18 00:26:51 -0500124 * 3. Performance analysis
125 *
Zheng Liu654598b2012-11-08 21:57:20 -0500126 * -- overhead
127 * 1. There is a cache extent for write access, so if writes are
128 * not very random, adding space operaions are in O(1) time.
129 *
130 * -- gain
131 * 2. Code is much simpler, more readable, more maintainable and
132 * more efficient.
133 *
134 *
135 * ==========================================================================
136 * 4. TODO list
Zheng Liu654598b2012-11-08 21:57:20 -0500137 *
Zheng Liu06b0c882013-02-18 00:26:51 -0500138 * -- Refactor delayed space reservation
Zheng Liu654598b2012-11-08 21:57:20 -0500139 *
140 * -- Extent-level locking
141 */
142
143static struct kmem_cache *ext4_es_cachep;
144
Zheng Liubdedbb72013-02-18 00:32:02 -0500145static int __es_insert_extent(struct inode *inode, struct extent_status *newes);
146static int __es_remove_extent(struct inode *inode, ext4_lblk_t lblk,
Zheng Liu06b0c882013-02-18 00:26:51 -0500147 ext4_lblk_t end);
Zheng Liu74cd15c2013-02-18 00:32:55 -0500148static int __es_try_to_reclaim_extents(struct ext4_inode_info *ei,
149 int nr_to_scan);
Zheng Liu06b0c882013-02-18 00:26:51 -0500150
Zheng Liu654598b2012-11-08 21:57:20 -0500151int __init ext4_init_es(void)
152{
Theodore Ts'o24630772013-02-28 23:58:56 -0500153 ext4_es_cachep = kmem_cache_create("ext4_extent_status",
154 sizeof(struct extent_status),
155 0, (SLAB_RECLAIM_ACCOUNT), NULL);
Zheng Liu654598b2012-11-08 21:57:20 -0500156 if (ext4_es_cachep == NULL)
157 return -ENOMEM;
158 return 0;
159}
160
161void ext4_exit_es(void)
162{
163 if (ext4_es_cachep)
164 kmem_cache_destroy(ext4_es_cachep);
165}
166
167void ext4_es_init_tree(struct ext4_es_tree *tree)
168{
169 tree->root = RB_ROOT;
170 tree->cache_es = NULL;
171}
172
173#ifdef ES_DEBUG__
174static void ext4_es_print_tree(struct inode *inode)
175{
176 struct ext4_es_tree *tree;
177 struct rb_node *node;
178
179 printk(KERN_DEBUG "status extents for inode %lu:", inode->i_ino);
180 tree = &EXT4_I(inode)->i_es_tree;
181 node = rb_first(&tree->root);
182 while (node) {
183 struct extent_status *es;
184 es = rb_entry(node, struct extent_status, rb_node);
Zheng Liufdc02122013-02-18 00:26:51 -0500185 printk(KERN_DEBUG " [%u/%u) %llu %llx",
186 es->es_lblk, es->es_len,
187 ext4_es_pblock(es), ext4_es_status(es));
Zheng Liu654598b2012-11-08 21:57:20 -0500188 node = rb_next(node);
189 }
190 printk(KERN_DEBUG "\n");
191}
192#else
193#define ext4_es_print_tree(inode)
194#endif
195
Zheng Liu06b0c882013-02-18 00:26:51 -0500196static inline ext4_lblk_t ext4_es_end(struct extent_status *es)
Zheng Liu654598b2012-11-08 21:57:20 -0500197{
Zheng Liu06b0c882013-02-18 00:26:51 -0500198 BUG_ON(es->es_lblk + es->es_len < es->es_lblk);
199 return es->es_lblk + es->es_len - 1;
Zheng Liu654598b2012-11-08 21:57:20 -0500200}
201
202/*
203 * search through the tree for an delayed extent with a given offset. If
204 * it can't be found, try to find next extent.
205 */
206static struct extent_status *__es_tree_search(struct rb_root *root,
Zheng Liu06b0c882013-02-18 00:26:51 -0500207 ext4_lblk_t lblk)
Zheng Liu654598b2012-11-08 21:57:20 -0500208{
209 struct rb_node *node = root->rb_node;
210 struct extent_status *es = NULL;
211
212 while (node) {
213 es = rb_entry(node, struct extent_status, rb_node);
Zheng Liu06b0c882013-02-18 00:26:51 -0500214 if (lblk < es->es_lblk)
Zheng Liu654598b2012-11-08 21:57:20 -0500215 node = node->rb_left;
Zheng Liu06b0c882013-02-18 00:26:51 -0500216 else if (lblk > ext4_es_end(es))
Zheng Liu654598b2012-11-08 21:57:20 -0500217 node = node->rb_right;
218 else
219 return es;
220 }
221
Zheng Liu06b0c882013-02-18 00:26:51 -0500222 if (es && lblk < es->es_lblk)
Zheng Liu654598b2012-11-08 21:57:20 -0500223 return es;
224
Zheng Liu06b0c882013-02-18 00:26:51 -0500225 if (es && lblk > ext4_es_end(es)) {
Zheng Liu654598b2012-11-08 21:57:20 -0500226 node = rb_next(&es->rb_node);
227 return node ? rb_entry(node, struct extent_status, rb_node) :
228 NULL;
229 }
230
231 return NULL;
232}
233
234/*
Zheng Liube401362013-02-18 00:27:26 -0500235 * ext4_es_find_delayed_extent: find the 1st delayed extent covering @es->lblk
Zheng Liu06b0c882013-02-18 00:26:51 -0500236 * if it exists, otherwise, the next extent after @es->lblk.
Zheng Liu654598b2012-11-08 21:57:20 -0500237 *
238 * @inode: the inode which owns delayed extents
Zheng Liube401362013-02-18 00:27:26 -0500239 * @lblk: the offset where we start to search
Zheng Liu654598b2012-11-08 21:57:20 -0500240 * @es: delayed extent that we found
Zheng Liu654598b2012-11-08 21:57:20 -0500241 */
Zheng Liube401362013-02-18 00:27:26 -0500242void ext4_es_find_delayed_extent(struct inode *inode, ext4_lblk_t lblk,
243 struct extent_status *es)
Zheng Liu654598b2012-11-08 21:57:20 -0500244{
245 struct ext4_es_tree *tree = NULL;
246 struct extent_status *es1 = NULL;
247 struct rb_node *node;
Zheng Liu654598b2012-11-08 21:57:20 -0500248
Zheng Liube401362013-02-18 00:27:26 -0500249 BUG_ON(es == NULL);
250 trace_ext4_es_find_delayed_extent_enter(inode, lblk);
Zheng Liu992e9fd2012-11-08 21:57:33 -0500251
Zheng Liu654598b2012-11-08 21:57:20 -0500252 read_lock(&EXT4_I(inode)->i_es_lock);
253 tree = &EXT4_I(inode)->i_es_tree;
254
Zheng Liufdc02122013-02-18 00:26:51 -0500255 /* find extent in cache firstly */
Zheng Liube401362013-02-18 00:27:26 -0500256 es->es_lblk = es->es_len = es->es_pblk = 0;
Zheng Liu654598b2012-11-08 21:57:20 -0500257 if (tree->cache_es) {
258 es1 = tree->cache_es;
Zheng Liube401362013-02-18 00:27:26 -0500259 if (in_range(lblk, es1->es_lblk, es1->es_len)) {
Zheng Liufdc02122013-02-18 00:26:51 -0500260 es_debug("%u cached by [%u/%u) %llu %llx\n",
Zheng Liube401362013-02-18 00:27:26 -0500261 lblk, es1->es_lblk, es1->es_len,
Zheng Liufdc02122013-02-18 00:26:51 -0500262 ext4_es_pblock(es1), ext4_es_status(es1));
Zheng Liu654598b2012-11-08 21:57:20 -0500263 goto out;
264 }
265 }
266
Zheng Liube401362013-02-18 00:27:26 -0500267 es1 = __es_tree_search(&tree->root, lblk);
Zheng Liu654598b2012-11-08 21:57:20 -0500268
269out:
Zheng Liube401362013-02-18 00:27:26 -0500270 if (es1 && !ext4_es_is_delayed(es1)) {
271 while ((node = rb_next(&es1->rb_node)) != NULL) {
272 es1 = rb_entry(node, struct extent_status, rb_node);
273 if (ext4_es_is_delayed(es1))
274 break;
275 }
276 }
277
278 if (es1 && ext4_es_is_delayed(es1)) {
Zheng Liu654598b2012-11-08 21:57:20 -0500279 tree->cache_es = es1;
Zheng Liu06b0c882013-02-18 00:26:51 -0500280 es->es_lblk = es1->es_lblk;
281 es->es_len = es1->es_len;
Zheng Liufdc02122013-02-18 00:26:51 -0500282 es->es_pblk = es1->es_pblk;
Zheng Liu654598b2012-11-08 21:57:20 -0500283 }
284
285 read_unlock(&EXT4_I(inode)->i_es_lock);
Zheng Liu992e9fd2012-11-08 21:57:33 -0500286
Zheng Liu74cd15c2013-02-18 00:32:55 -0500287 ext4_es_lru_add(inode);
Zheng Liube401362013-02-18 00:27:26 -0500288 trace_ext4_es_find_delayed_extent_exit(inode, es);
Zheng Liu654598b2012-11-08 21:57:20 -0500289}
290
291static struct extent_status *
Zheng Liubdedbb72013-02-18 00:32:02 -0500292ext4_es_alloc_extent(struct inode *inode, ext4_lblk_t lblk, ext4_lblk_t len,
293 ext4_fsblk_t pblk)
Zheng Liu654598b2012-11-08 21:57:20 -0500294{
295 struct extent_status *es;
296 es = kmem_cache_alloc(ext4_es_cachep, GFP_ATOMIC);
297 if (es == NULL)
298 return NULL;
Zheng Liu06b0c882013-02-18 00:26:51 -0500299 es->es_lblk = lblk;
300 es->es_len = len;
Zheng Liufdc02122013-02-18 00:26:51 -0500301 es->es_pblk = pblk;
Zheng Liu74cd15c2013-02-18 00:32:55 -0500302
303 /*
304 * We don't count delayed extent because we never try to reclaim them
305 */
Theodore Ts'o24630772013-02-28 23:58:56 -0500306 if (!ext4_es_is_delayed(es)) {
Zheng Liu74cd15c2013-02-18 00:32:55 -0500307 EXT4_I(inode)->i_es_lru_nr++;
Theodore Ts'o24630772013-02-28 23:58:56 -0500308 atomic_inc(&EXT4_SB(inode->i_sb)->s_extent_cache_cnt);
309 }
Zheng Liu74cd15c2013-02-18 00:32:55 -0500310
Zheng Liu654598b2012-11-08 21:57:20 -0500311 return es;
312}
313
Zheng Liubdedbb72013-02-18 00:32:02 -0500314static void ext4_es_free_extent(struct inode *inode, struct extent_status *es)
Zheng Liu654598b2012-11-08 21:57:20 -0500315{
Zheng Liu74cd15c2013-02-18 00:32:55 -0500316 /* Decrease the lru counter when this es is not delayed */
317 if (!ext4_es_is_delayed(es)) {
318 BUG_ON(EXT4_I(inode)->i_es_lru_nr == 0);
319 EXT4_I(inode)->i_es_lru_nr--;
Theodore Ts'o24630772013-02-28 23:58:56 -0500320 atomic_dec(&EXT4_SB(inode->i_sb)->s_extent_cache_cnt);
Zheng Liu74cd15c2013-02-18 00:32:55 -0500321 }
322
Zheng Liu654598b2012-11-08 21:57:20 -0500323 kmem_cache_free(ext4_es_cachep, es);
324}
325
Zheng Liu06b0c882013-02-18 00:26:51 -0500326/*
327 * Check whether or not two extents can be merged
328 * Condition:
329 * - logical block number is contiguous
Zheng Liufdc02122013-02-18 00:26:51 -0500330 * - physical block number is contiguous
331 * - status is equal
Zheng Liu06b0c882013-02-18 00:26:51 -0500332 */
333static int ext4_es_can_be_merged(struct extent_status *es1,
334 struct extent_status *es2)
335{
336 if (es1->es_lblk + es1->es_len != es2->es_lblk)
337 return 0;
338
Zheng Liufdc02122013-02-18 00:26:51 -0500339 if (ext4_es_status(es1) != ext4_es_status(es2))
340 return 0;
341
342 if ((ext4_es_is_written(es1) || ext4_es_is_unwritten(es1)) &&
343 (ext4_es_pblock(es1) + es1->es_len != ext4_es_pblock(es2)))
344 return 0;
345
Zheng Liu06b0c882013-02-18 00:26:51 -0500346 return 1;
347}
348
Zheng Liu654598b2012-11-08 21:57:20 -0500349static struct extent_status *
Zheng Liubdedbb72013-02-18 00:32:02 -0500350ext4_es_try_to_merge_left(struct inode *inode, struct extent_status *es)
Zheng Liu654598b2012-11-08 21:57:20 -0500351{
Zheng Liubdedbb72013-02-18 00:32:02 -0500352 struct ext4_es_tree *tree = &EXT4_I(inode)->i_es_tree;
Zheng Liu654598b2012-11-08 21:57:20 -0500353 struct extent_status *es1;
354 struct rb_node *node;
355
356 node = rb_prev(&es->rb_node);
357 if (!node)
358 return es;
359
360 es1 = rb_entry(node, struct extent_status, rb_node);
Zheng Liu06b0c882013-02-18 00:26:51 -0500361 if (ext4_es_can_be_merged(es1, es)) {
362 es1->es_len += es->es_len;
Zheng Liu654598b2012-11-08 21:57:20 -0500363 rb_erase(&es->rb_node, &tree->root);
Zheng Liubdedbb72013-02-18 00:32:02 -0500364 ext4_es_free_extent(inode, es);
Zheng Liu654598b2012-11-08 21:57:20 -0500365 es = es1;
366 }
367
368 return es;
369}
370
371static struct extent_status *
Zheng Liubdedbb72013-02-18 00:32:02 -0500372ext4_es_try_to_merge_right(struct inode *inode, struct extent_status *es)
Zheng Liu654598b2012-11-08 21:57:20 -0500373{
Zheng Liubdedbb72013-02-18 00:32:02 -0500374 struct ext4_es_tree *tree = &EXT4_I(inode)->i_es_tree;
Zheng Liu654598b2012-11-08 21:57:20 -0500375 struct extent_status *es1;
376 struct rb_node *node;
377
378 node = rb_next(&es->rb_node);
379 if (!node)
380 return es;
381
382 es1 = rb_entry(node, struct extent_status, rb_node);
Zheng Liu06b0c882013-02-18 00:26:51 -0500383 if (ext4_es_can_be_merged(es, es1)) {
384 es->es_len += es1->es_len;
Zheng Liu654598b2012-11-08 21:57:20 -0500385 rb_erase(node, &tree->root);
Zheng Liubdedbb72013-02-18 00:32:02 -0500386 ext4_es_free_extent(inode, es1);
Zheng Liu654598b2012-11-08 21:57:20 -0500387 }
388
389 return es;
390}
391
Zheng Liubdedbb72013-02-18 00:32:02 -0500392static int __es_insert_extent(struct inode *inode, struct extent_status *newes)
Zheng Liu654598b2012-11-08 21:57:20 -0500393{
Zheng Liubdedbb72013-02-18 00:32:02 -0500394 struct ext4_es_tree *tree = &EXT4_I(inode)->i_es_tree;
Zheng Liu654598b2012-11-08 21:57:20 -0500395 struct rb_node **p = &tree->root.rb_node;
396 struct rb_node *parent = NULL;
397 struct extent_status *es;
Zheng Liu654598b2012-11-08 21:57:20 -0500398
399 while (*p) {
400 parent = *p;
401 es = rb_entry(parent, struct extent_status, rb_node);
402
Zheng Liu06b0c882013-02-18 00:26:51 -0500403 if (newes->es_lblk < es->es_lblk) {
404 if (ext4_es_can_be_merged(newes, es)) {
405 /*
406 * Here we can modify es_lblk directly
407 * because it isn't overlapped.
408 */
409 es->es_lblk = newes->es_lblk;
410 es->es_len += newes->es_len;
Zheng Liufdc02122013-02-18 00:26:51 -0500411 if (ext4_es_is_written(es) ||
412 ext4_es_is_unwritten(es))
413 ext4_es_store_pblock(es,
414 newes->es_pblk);
Zheng Liubdedbb72013-02-18 00:32:02 -0500415 es = ext4_es_try_to_merge_left(inode, es);
Zheng Liu654598b2012-11-08 21:57:20 -0500416 goto out;
417 }
418 p = &(*p)->rb_left;
Zheng Liu06b0c882013-02-18 00:26:51 -0500419 } else if (newes->es_lblk > ext4_es_end(es)) {
420 if (ext4_es_can_be_merged(es, newes)) {
421 es->es_len += newes->es_len;
Zheng Liubdedbb72013-02-18 00:32:02 -0500422 es = ext4_es_try_to_merge_right(inode, es);
Zheng Liu654598b2012-11-08 21:57:20 -0500423 goto out;
424 }
425 p = &(*p)->rb_right;
426 } else {
Zheng Liu06b0c882013-02-18 00:26:51 -0500427 BUG_ON(1);
428 return -EINVAL;
Zheng Liu654598b2012-11-08 21:57:20 -0500429 }
430 }
431
Zheng Liubdedbb72013-02-18 00:32:02 -0500432 es = ext4_es_alloc_extent(inode, newes->es_lblk, newes->es_len,
Zheng Liufdc02122013-02-18 00:26:51 -0500433 newes->es_pblk);
Zheng Liu654598b2012-11-08 21:57:20 -0500434 if (!es)
435 return -ENOMEM;
436 rb_link_node(&es->rb_node, parent, p);
437 rb_insert_color(&es->rb_node, &tree->root);
438
439out:
440 tree->cache_es = es;
441 return 0;
442}
443
444/*
Zheng Liu06b0c882013-02-18 00:26:51 -0500445 * ext4_es_insert_extent() adds a space to a extent status tree.
Zheng Liu654598b2012-11-08 21:57:20 -0500446 *
447 * ext4_es_insert_extent is called by ext4_da_write_begin and
448 * ext4_es_remove_extent.
449 *
450 * Return 0 on success, error code on failure.
451 */
Zheng Liu06b0c882013-02-18 00:26:51 -0500452int ext4_es_insert_extent(struct inode *inode, ext4_lblk_t lblk,
Zheng Liufdc02122013-02-18 00:26:51 -0500453 ext4_lblk_t len, ext4_fsblk_t pblk,
454 unsigned long long status)
Zheng Liu654598b2012-11-08 21:57:20 -0500455{
Zheng Liu06b0c882013-02-18 00:26:51 -0500456 struct extent_status newes;
457 ext4_lblk_t end = lblk + len - 1;
Zheng Liu654598b2012-11-08 21:57:20 -0500458 int err = 0;
459
Zheng Liufdc02122013-02-18 00:26:51 -0500460 es_debug("add [%u/%u) %llu %llx to extent status tree of inode %lu\n",
461 lblk, len, pblk, status, inode->i_ino);
Zheng Liu06b0c882013-02-18 00:26:51 -0500462
Eryu Guand4381472013-02-22 15:27:47 -0500463 if (!len)
464 return 0;
465
Zheng Liu06b0c882013-02-18 00:26:51 -0500466 BUG_ON(end < lblk);
467
468 newes.es_lblk = lblk;
469 newes.es_len = len;
Zheng Liufdc02122013-02-18 00:26:51 -0500470 ext4_es_store_pblock(&newes, pblk);
471 ext4_es_store_status(&newes, status);
472 trace_ext4_es_insert_extent(inode, &newes);
Zheng Liu654598b2012-11-08 21:57:20 -0500473
474 write_lock(&EXT4_I(inode)->i_es_lock);
Zheng Liubdedbb72013-02-18 00:32:02 -0500475 err = __es_remove_extent(inode, lblk, end);
Zheng Liu06b0c882013-02-18 00:26:51 -0500476 if (err != 0)
477 goto error;
Zheng Liubdedbb72013-02-18 00:32:02 -0500478 err = __es_insert_extent(inode, &newes);
Zheng Liu06b0c882013-02-18 00:26:51 -0500479
480error:
Zheng Liu654598b2012-11-08 21:57:20 -0500481 write_unlock(&EXT4_I(inode)->i_es_lock);
482
Zheng Liu74cd15c2013-02-18 00:32:55 -0500483 ext4_es_lru_add(inode);
Zheng Liu654598b2012-11-08 21:57:20 -0500484 ext4_es_print_tree(inode);
485
486 return err;
487}
488
Zheng Liud100eef2013-02-18 00:29:59 -0500489/*
490 * ext4_es_lookup_extent() looks up an extent in extent status tree.
491 *
492 * ext4_es_lookup_extent is called by ext4_map_blocks/ext4_da_map_blocks.
493 *
494 * Return: 1 on found, 0 on not
495 */
496int ext4_es_lookup_extent(struct inode *inode, ext4_lblk_t lblk,
497 struct extent_status *es)
498{
499 struct ext4_es_tree *tree;
500 struct extent_status *es1 = NULL;
501 struct rb_node *node;
502 int found = 0;
503
504 trace_ext4_es_lookup_extent_enter(inode, lblk);
505 es_debug("lookup extent in block %u\n", lblk);
506
507 tree = &EXT4_I(inode)->i_es_tree;
508 read_lock(&EXT4_I(inode)->i_es_lock);
509
510 /* find extent in cache firstly */
511 es->es_lblk = es->es_len = es->es_pblk = 0;
512 if (tree->cache_es) {
513 es1 = tree->cache_es;
514 if (in_range(lblk, es1->es_lblk, es1->es_len)) {
515 es_debug("%u cached by [%u/%u)\n",
516 lblk, es1->es_lblk, es1->es_len);
517 found = 1;
518 goto out;
519 }
520 }
521
522 node = tree->root.rb_node;
523 while (node) {
524 es1 = rb_entry(node, struct extent_status, rb_node);
525 if (lblk < es1->es_lblk)
526 node = node->rb_left;
527 else if (lblk > ext4_es_end(es1))
528 node = node->rb_right;
529 else {
530 found = 1;
531 break;
532 }
533 }
534
535out:
536 if (found) {
537 BUG_ON(!es1);
538 es->es_lblk = es1->es_lblk;
539 es->es_len = es1->es_len;
540 es->es_pblk = es1->es_pblk;
541 }
542
543 read_unlock(&EXT4_I(inode)->i_es_lock);
544
Zheng Liu74cd15c2013-02-18 00:32:55 -0500545 ext4_es_lru_add(inode);
Zheng Liud100eef2013-02-18 00:29:59 -0500546 trace_ext4_es_lookup_extent_exit(inode, es, found);
547 return found;
548}
549
Zheng Liubdedbb72013-02-18 00:32:02 -0500550static int __es_remove_extent(struct inode *inode, ext4_lblk_t lblk,
551 ext4_lblk_t end)
Zheng Liu654598b2012-11-08 21:57:20 -0500552{
Zheng Liubdedbb72013-02-18 00:32:02 -0500553 struct ext4_es_tree *tree = &EXT4_I(inode)->i_es_tree;
Zheng Liu654598b2012-11-08 21:57:20 -0500554 struct rb_node *node;
Zheng Liu654598b2012-11-08 21:57:20 -0500555 struct extent_status *es;
556 struct extent_status orig_es;
Zheng Liu06b0c882013-02-18 00:26:51 -0500557 ext4_lblk_t len1, len2;
Zheng Liufdc02122013-02-18 00:26:51 -0500558 ext4_fsblk_t block;
Zheng Liu654598b2012-11-08 21:57:20 -0500559 int err = 0;
560
Zheng Liu06b0c882013-02-18 00:26:51 -0500561 es = __es_tree_search(&tree->root, lblk);
Zheng Liu654598b2012-11-08 21:57:20 -0500562 if (!es)
563 goto out;
Zheng Liu06b0c882013-02-18 00:26:51 -0500564 if (es->es_lblk > end)
Zheng Liu654598b2012-11-08 21:57:20 -0500565 goto out;
566
567 /* Simply invalidate cache_es. */
568 tree->cache_es = NULL;
569
Zheng Liu06b0c882013-02-18 00:26:51 -0500570 orig_es.es_lblk = es->es_lblk;
571 orig_es.es_len = es->es_len;
Zheng Liufdc02122013-02-18 00:26:51 -0500572 orig_es.es_pblk = es->es_pblk;
573
Zheng Liu06b0c882013-02-18 00:26:51 -0500574 len1 = lblk > es->es_lblk ? lblk - es->es_lblk : 0;
575 len2 = ext4_es_end(es) > end ? ext4_es_end(es) - end : 0;
Zheng Liu654598b2012-11-08 21:57:20 -0500576 if (len1 > 0)
Zheng Liu06b0c882013-02-18 00:26:51 -0500577 es->es_len = len1;
Zheng Liu654598b2012-11-08 21:57:20 -0500578 if (len2 > 0) {
579 if (len1 > 0) {
Zheng Liu06b0c882013-02-18 00:26:51 -0500580 struct extent_status newes;
581
582 newes.es_lblk = end + 1;
583 newes.es_len = len2;
Zheng Liufdc02122013-02-18 00:26:51 -0500584 if (ext4_es_is_written(&orig_es) ||
585 ext4_es_is_unwritten(&orig_es)) {
586 block = ext4_es_pblock(&orig_es) +
587 orig_es.es_len - len2;
588 ext4_es_store_pblock(&newes, block);
589 }
590 ext4_es_store_status(&newes, ext4_es_status(&orig_es));
Zheng Liubdedbb72013-02-18 00:32:02 -0500591 err = __es_insert_extent(inode, &newes);
Zheng Liu654598b2012-11-08 21:57:20 -0500592 if (err) {
Zheng Liu06b0c882013-02-18 00:26:51 -0500593 es->es_lblk = orig_es.es_lblk;
594 es->es_len = orig_es.es_len;
Zheng Liu654598b2012-11-08 21:57:20 -0500595 goto out;
596 }
597 } else {
Zheng Liu06b0c882013-02-18 00:26:51 -0500598 es->es_lblk = end + 1;
599 es->es_len = len2;
Zheng Liufdc02122013-02-18 00:26:51 -0500600 if (ext4_es_is_written(es) ||
601 ext4_es_is_unwritten(es)) {
602 block = orig_es.es_pblk + orig_es.es_len - len2;
603 ext4_es_store_pblock(es, block);
604 }
Zheng Liu654598b2012-11-08 21:57:20 -0500605 }
606 goto out;
607 }
608
609 if (len1 > 0) {
610 node = rb_next(&es->rb_node);
611 if (node)
612 es = rb_entry(node, struct extent_status, rb_node);
613 else
614 es = NULL;
615 }
616
Zheng Liu06b0c882013-02-18 00:26:51 -0500617 while (es && ext4_es_end(es) <= end) {
Zheng Liu654598b2012-11-08 21:57:20 -0500618 node = rb_next(&es->rb_node);
619 rb_erase(&es->rb_node, &tree->root);
Zheng Liubdedbb72013-02-18 00:32:02 -0500620 ext4_es_free_extent(inode, es);
Zheng Liu654598b2012-11-08 21:57:20 -0500621 if (!node) {
622 es = NULL;
623 break;
624 }
625 es = rb_entry(node, struct extent_status, rb_node);
626 }
627
Zheng Liu06b0c882013-02-18 00:26:51 -0500628 if (es && es->es_lblk < end + 1) {
Zheng Liufdc02122013-02-18 00:26:51 -0500629 ext4_lblk_t orig_len = es->es_len;
630
Zheng Liu06b0c882013-02-18 00:26:51 -0500631 len1 = ext4_es_end(es) - end;
632 es->es_lblk = end + 1;
633 es->es_len = len1;
Zheng Liufdc02122013-02-18 00:26:51 -0500634 if (ext4_es_is_written(es) || ext4_es_is_unwritten(es)) {
635 block = es->es_pblk + orig_len - len1;
636 ext4_es_store_pblock(es, block);
637 }
Zheng Liu654598b2012-11-08 21:57:20 -0500638 }
639
640out:
Zheng Liu06b0c882013-02-18 00:26:51 -0500641 return err;
642}
643
644/*
645 * ext4_es_remove_extent() removes a space from a extent status tree.
646 *
647 * Return 0 on success, error code on failure.
648 */
649int ext4_es_remove_extent(struct inode *inode, ext4_lblk_t lblk,
650 ext4_lblk_t len)
651{
Zheng Liu06b0c882013-02-18 00:26:51 -0500652 ext4_lblk_t end;
653 int err = 0;
654
655 trace_ext4_es_remove_extent(inode, lblk, len);
656 es_debug("remove [%u/%u) from extent status tree of inode %lu\n",
657 lblk, len, inode->i_ino);
658
Eryu Guand4381472013-02-22 15:27:47 -0500659 if (!len)
660 return err;
661
Zheng Liu06b0c882013-02-18 00:26:51 -0500662 end = lblk + len - 1;
663 BUG_ON(end < lblk);
664
Zheng Liu06b0c882013-02-18 00:26:51 -0500665 write_lock(&EXT4_I(inode)->i_es_lock);
Zheng Liubdedbb72013-02-18 00:32:02 -0500666 err = __es_remove_extent(inode, lblk, end);
Zheng Liu654598b2012-11-08 21:57:20 -0500667 write_unlock(&EXT4_I(inode)->i_es_lock);
668 ext4_es_print_tree(inode);
669 return err;
670}
Zheng Liu74cd15c2013-02-18 00:32:55 -0500671
672static int ext4_es_shrink(struct shrinker *shrink, struct shrink_control *sc)
673{
674 struct ext4_sb_info *sbi = container_of(shrink,
675 struct ext4_sb_info, s_es_shrinker);
676 struct ext4_inode_info *ei;
677 struct list_head *cur, *tmp, scanned;
678 int nr_to_scan = sc->nr_to_scan;
679 int ret, nr_shrunk = 0;
680
Theodore Ts'o24630772013-02-28 23:58:56 -0500681 ret = atomic_read(&sbi->s_extent_cache_cnt);
682 trace_ext4_es_shrink_enter(sbi->s_sb, nr_to_scan, ret);
Zheng Liu74cd15c2013-02-18 00:32:55 -0500683
684 if (!nr_to_scan)
Theodore Ts'o24630772013-02-28 23:58:56 -0500685 return ret;
Zheng Liu74cd15c2013-02-18 00:32:55 -0500686
687 INIT_LIST_HEAD(&scanned);
688
689 spin_lock(&sbi->s_es_lru_lock);
690 list_for_each_safe(cur, tmp, &sbi->s_es_lru) {
691 list_move_tail(cur, &scanned);
692
693 ei = list_entry(cur, struct ext4_inode_info, i_es_lru);
694
695 read_lock(&ei->i_es_lock);
696 if (ei->i_es_lru_nr == 0) {
697 read_unlock(&ei->i_es_lock);
698 continue;
699 }
700 read_unlock(&ei->i_es_lock);
701
702 write_lock(&ei->i_es_lock);
703 ret = __es_try_to_reclaim_extents(ei, nr_to_scan);
704 write_unlock(&ei->i_es_lock);
705
706 nr_shrunk += ret;
707 nr_to_scan -= ret;
708 if (nr_to_scan == 0)
709 break;
710 }
711 list_splice_tail(&scanned, &sbi->s_es_lru);
712 spin_unlock(&sbi->s_es_lru_lock);
Zheng Liu74cd15c2013-02-18 00:32:55 -0500713
Theodore Ts'o24630772013-02-28 23:58:56 -0500714 ret = atomic_read(&sbi->s_extent_cache_cnt);
715 trace_ext4_es_shrink_exit(sbi->s_sb, nr_shrunk, ret);
716 return ret;
Zheng Liu74cd15c2013-02-18 00:32:55 -0500717}
718
719void ext4_es_register_shrinker(struct super_block *sb)
720{
721 struct ext4_sb_info *sbi;
722
723 sbi = EXT4_SB(sb);
724 INIT_LIST_HEAD(&sbi->s_es_lru);
725 spin_lock_init(&sbi->s_es_lru_lock);
726 sbi->s_es_shrinker.shrink = ext4_es_shrink;
727 sbi->s_es_shrinker.seeks = DEFAULT_SEEKS;
728 register_shrinker(&sbi->s_es_shrinker);
729}
730
731void ext4_es_unregister_shrinker(struct super_block *sb)
732{
733 unregister_shrinker(&EXT4_SB(sb)->s_es_shrinker);
734}
735
736void ext4_es_lru_add(struct inode *inode)
737{
738 struct ext4_inode_info *ei = EXT4_I(inode);
739 struct ext4_sb_info *sbi = EXT4_SB(inode->i_sb);
740
741 spin_lock(&sbi->s_es_lru_lock);
742 if (list_empty(&ei->i_es_lru))
743 list_add_tail(&ei->i_es_lru, &sbi->s_es_lru);
744 else
745 list_move_tail(&ei->i_es_lru, &sbi->s_es_lru);
746 spin_unlock(&sbi->s_es_lru_lock);
747}
748
749void ext4_es_lru_del(struct inode *inode)
750{
751 struct ext4_inode_info *ei = EXT4_I(inode);
752 struct ext4_sb_info *sbi = EXT4_SB(inode->i_sb);
753
754 spin_lock(&sbi->s_es_lru_lock);
755 if (!list_empty(&ei->i_es_lru))
756 list_del_init(&ei->i_es_lru);
757 spin_unlock(&sbi->s_es_lru_lock);
758}
759
Zheng Liu74cd15c2013-02-18 00:32:55 -0500760static int __es_try_to_reclaim_extents(struct ext4_inode_info *ei,
761 int nr_to_scan)
762{
763 struct inode *inode = &ei->vfs_inode;
764 struct ext4_es_tree *tree = &ei->i_es_tree;
765 struct rb_node *node;
766 struct extent_status *es;
767 int nr_shrunk = 0;
768
769 if (ei->i_es_lru_nr == 0)
770 return 0;
771
772 node = rb_first(&tree->root);
773 while (node != NULL) {
774 es = rb_entry(node, struct extent_status, rb_node);
775 node = rb_next(&es->rb_node);
776 /*
777 * We can't reclaim delayed extent from status tree because
778 * fiemap, bigallic, and seek_data/hole need to use it.
779 */
780 if (!ext4_es_is_delayed(es)) {
781 rb_erase(&es->rb_node, &tree->root);
782 ext4_es_free_extent(inode, es);
783 nr_shrunk++;
784 if (--nr_to_scan == 0)
785 break;
786 }
787 }
788 tree->cache_es = NULL;
789 return nr_shrunk;
790}