52 	struct extent_map *em;  in alloc_extent_map()  local
53 	em = kmem_cache_zalloc(extent_map_cache, GFP_NOFS);  in alloc_extent_map()
54 	if (!em)  in alloc_extent_map()
56 	RB_CLEAR_NODE(&em->rb_node);  in alloc_extent_map()
57 	em->flags = 0;  in alloc_extent_map()
58 	em->compress_type = BTRFS_COMPRESS_NONE;  in alloc_extent_map()
59 	em->generation = 0;  in alloc_extent_map()
60 	refcount_set(&em->refs, 1);  in alloc_extent_map()
61 	INIT_LIST_HEAD(&em->list);  in alloc_extent_map()
62 	return em;  in alloc_extent_map()
72 void free_extent_map(struct extent_map *em)  in free_extent_map()  argument
74 	if (!em)  in free_extent_map()
76 	WARN_ON(refcount_read(&em->refs) == 0);  in free_extent_map()
77 	if (refcount_dec_and_test(&em->refs)) {  in free_extent_map()
78 		WARN_ON(extent_map_in_tree(em));  in free_extent_map()
79 		WARN_ON(!list_empty(&em->list));  in free_extent_map()
80 		if (test_bit(EXTENT_FLAG_FS_MAPPING, &em->flags))  in free_extent_map()
81 			kfree(em->map_lookup);  in free_extent_map()
82 		kmem_cache_free(extent_map_cache, em);  in free_extent_map()
94 static int tree_insert(struct rb_root_cached *root, struct extent_map *em)  in tree_insert()  argument
100 	u64 end = range_end(em->start, em->len);  in tree_insert()
107 		if (em->start < entry->start) {  in tree_insert()
109 		} else if (em->start >= extent_map_end(entry)) {  in tree_insert()
118 	while (parent && em->start >= extent_map_end(entry)) {  in tree_insert()
123 		if (end > entry->start && em->start < extent_map_end(entry))  in tree_insert()
128 	while (parent && em->start < entry->start) {  in tree_insert()
133 		if (end > entry->start && em->start < extent_map_end(entry))  in tree_insert()
136 	rb_link_node(&em->rb_node, orig_parent, p);  in tree_insert()
137 	rb_insert_color_cached(&em->rb_node, root, leftmost);  in tree_insert()
235 static void try_merge_map(struct extent_map_tree *tree, struct extent_map *em)  in try_merge_map()  argument
248 	if (refcount_read(&em->refs) > 2)  in try_merge_map()
251 	if (em->start != 0) {  in try_merge_map()
252 		rb = rb_prev(&em->rb_node);  in try_merge_map()
255 		if (rb && mergable_maps(merge, em)) {  in try_merge_map()
256 			em->start = merge->start;  in try_merge_map()
257 			em->orig_start = merge->orig_start;  in try_merge_map()
258 			em->len += merge->len;  in try_merge_map()
259 			em->block_len += merge->block_len;  in try_merge_map()
260 			em->block_start = merge->block_start;  in try_merge_map()
261 			em->mod_len = (em->mod_len + em->mod_start) - merge->mod_start;  in try_merge_map()
262 			em->mod_start = merge->mod_start;  in try_merge_map()
263 			em->generation = max(em->generation, merge->generation);  in try_merge_map()
271 	rb = rb_next(&em->rb_node);  in try_merge_map()
274 	if (rb && mergable_maps(em, merge)) {  in try_merge_map()
275 		em->len += merge->len;  in try_merge_map()
276 		em->block_len += merge->block_len;  in try_merge_map()
279 		em->mod_len = (merge->mod_start + merge->mod_len) - em->mod_start;  in try_merge_map()
280 		em->generation = max(em->generation, merge->generation);  in try_merge_map()
300 	struct extent_map *em;  in unpin_extent_cache()  local
304 	em = lookup_extent_mapping(tree, start, len);  in unpin_extent_cache()
306 	WARN_ON(!em || em->start != start);  in unpin_extent_cache()
308 	if (!em)  in unpin_extent_cache()
311 	em->generation = gen;  in unpin_extent_cache()
312 	clear_bit(EXTENT_FLAG_PINNED, &em->flags);  in unpin_extent_cache()
313 	em->mod_start = em->start;  in unpin_extent_cache()
314 	em->mod_len = em->len;  in unpin_extent_cache()
316 	if (test_bit(EXTENT_FLAG_FILLING, &em->flags)) {  in unpin_extent_cache()
318 		clear_bit(EXTENT_FLAG_FILLING, &em->flags);  in unpin_extent_cache()
321 	try_merge_map(tree, em);  in unpin_extent_cache()
324 		em->mod_start = em->start;  in unpin_extent_cache()
325 		em->mod_len = em->len;  in unpin_extent_cache()
328 	free_extent_map(em);  in unpin_extent_cache()
335 void clear_em_logging(struct extent_map_tree *tree, struct extent_map *em)  in clear_em_logging()  argument
337 	clear_bit(EXTENT_FLAG_LOGGING, &em->flags);  in clear_em_logging()
338 	if (extent_map_in_tree(em))  in clear_em_logging()
339 		try_merge_map(tree, em);  in clear_em_logging()
343 					struct extent_map *em,  in setup_extent_mapping()  argument
346 	refcount_inc(&em->refs);  in setup_extent_mapping()
347 	em->mod_start = em->start;  in setup_extent_mapping()
348 	em->mod_len = em->len;  in setup_extent_mapping()
351 		list_move(&em->list, &tree->modified_extents);  in setup_extent_mapping()
353 		try_merge_map(tree, em);  in setup_extent_mapping()
356 static void extent_map_device_set_bits(struct extent_map *em, unsigned bits)  in extent_map_device_set_bits()  argument
358 	struct map_lookup *map = em->map_lookup;  in extent_map_device_set_bits()
359 	u64 stripe_size = em->orig_block_len;  in extent_map_device_set_bits()
371 static void extent_map_device_clear_bits(struct extent_map *em, unsigned bits)  in extent_map_device_clear_bits()  argument
373 	struct map_lookup *map = em->map_lookup;  in extent_map_device_clear_bits()
374 	u64 stripe_size = em->orig_block_len;  in extent_map_device_clear_bits()
401 		       struct extent_map *em, int modified)  in add_extent_mapping()  argument
407 	ret = tree_insert(&tree->map, em);  in add_extent_mapping()
411 	setup_extent_mapping(tree, em, modified);  in add_extent_mapping()
412 	if (test_bit(EXTENT_FLAG_FS_MAPPING, &em->flags)) {  in add_extent_mapping()
413 		extent_map_device_set_bits(em, CHUNK_ALLOCATED);  in add_extent_mapping()
414 		extent_map_device_clear_bits(em, CHUNK_TRIMMED);  in add_extent_mapping()
424 	struct extent_map *em;  in __lookup_extent_mapping()  local
440 	em = rb_entry(rb_node, struct extent_map, rb_node);  in __lookup_extent_mapping()
442 	if (strict && !(end > em->start && start < extent_map_end(em)))  in __lookup_extent_mapping()
445 	refcount_inc(&em->refs);  in __lookup_extent_mapping()
446 	return em;  in __lookup_extent_mapping()
491 void remove_extent_mapping(struct extent_map_tree *tree, struct extent_map *em)  in remove_extent_mapping()  argument
493 	WARN_ON(test_bit(EXTENT_FLAG_PINNED, &em->flags));  in remove_extent_mapping()
494 	rb_erase_cached(&em->rb_node, &tree->map);  in remove_extent_mapping()
495 	if (!test_bit(EXTENT_FLAG_LOGGING, &em->flags))  in remove_extent_mapping()
496 		list_del_init(&em->list);  in remove_extent_mapping()
497 	if (test_bit(EXTENT_FLAG_FS_MAPPING, &em->flags))  in remove_extent_mapping()
498 		extent_map_device_clear_bits(em, CHUNK_ALLOCATED);  in remove_extent_mapping()
499 	RB_CLEAR_NODE(&em->rb_node);  in remove_extent_mapping()
517 static struct extent_map *next_extent_map(struct extent_map *em)  in next_extent_map()  argument
521 	next = rb_next(&em->rb_node);  in next_extent_map()
527 static struct extent_map *prev_extent_map(struct extent_map *em)  in prev_extent_map()  argument
531 	prev = rb_prev(&em->rb_node);  in prev_extent_map()
545 					 struct extent_map *em,  in merge_extent_mapping()  argument
554 	BUG_ON(map_start < em->start || map_start >= extent_map_end(em));  in merge_extent_mapping()
564 	start = prev ? extent_map_end(prev) : em->start;  in merge_extent_mapping()
565 	start = max_t(u64, start, em->start);  in merge_extent_mapping()
566 	end = next ? next->start : extent_map_end(em);  in merge_extent_mapping()
567 	end = min_t(u64, end, extent_map_end(em));  in merge_extent_mapping()
568 	start_diff = start - em->start;  in merge_extent_mapping()
569 	em->start = start;  in merge_extent_mapping()
570 	em->len = end - start;  in merge_extent_mapping()
571 	if (em->block_start < EXTENT_MAP_LAST_BYTE &&  in merge_extent_mapping()
572 	    !test_bit(EXTENT_FLAG_COMPRESSED, &em->flags)) {  in merge_extent_mapping()
573 		em->block_start += start_diff;  in merge_extent_mapping()
574 		em->block_len = em->len;  in merge_extent_mapping()
576 	return add_extent_mapping(em_tree, em, 0);  in merge_extent_mapping()
605 	struct extent_map *em = *em_in;  in btrfs_add_extent_mapping()  local
607 	ret = add_extent_mapping(em_tree, em, 0);  in btrfs_add_extent_mapping()
619 		trace_btrfs_handle_em_exist(fs_info, existing, em, start, len);  in btrfs_add_extent_mapping()
627 			free_extent_map(em);  in btrfs_add_extent_mapping()
631 			u64 orig_start = em->start;  in btrfs_add_extent_mapping()
632 			u64 orig_len = em->len;  in btrfs_add_extent_mapping()
639 						   em, start);  in btrfs_add_extent_mapping()
641 				free_extent_map(em);  in btrfs_add_extent_mapping()