2 * Copyright (C) 2009 Oracle. All rights reserved.
4 * This program is free software; you can redistribute it and/or
5 * modify it under the terms of the GNU General Public
6 * License v2 as published by the Free Software Foundation.
8 * This program is distributed in the hope that it will be useful,
9 * but WITHOUT ANY WARRANTY; without even the implied warranty of
10 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU
11 * General Public License for more details.
13 * You should have received a copy of the GNU General Public
14 * License along with this program; if not, write to the
15 * Free Software Foundation, Inc., 59 Temple Place - Suite 330,
16 * Boston, MA 021110-1307, USA.
19 #include <linux/sched.h>
20 #include <linux/pagemap.h>
21 #include <linux/writeback.h>
22 #include <linux/blkdev.h>
23 #include <linux/rbtree.h>
24 #include <linux/slab.h>
27 #include "transaction.h"
30 #include "btrfs_inode.h"
31 #include "async-thread.h"
32 #include "free-space-cache.h"
33 #include "inode-map.h"
37 * backref_node, mapping_node and tree_block start with this
40 struct rb_node rb_node
;
45 * present a tree block in the backref cache
48 struct rb_node rb_node
;
52 /* objectid of tree block owner, can be not uptodate */
54 /* link to pending, changed or detached list */
55 struct list_head list
;
56 /* list of upper level blocks reference this block */
57 struct list_head upper
;
58 /* list of child blocks in the cache */
59 struct list_head lower
;
60 /* NULL if this node is not tree root */
61 struct btrfs_root
*root
;
62 /* extent buffer got by COW the block */
63 struct extent_buffer
*eb
;
64 /* level of tree block */
66 /* is the block in non-reference counted tree */
67 unsigned int cowonly
:1;
68 /* 1 if no child node in the cache */
69 unsigned int lowest
:1;
70 /* is the extent buffer locked */
71 unsigned int locked
:1;
72 /* has the block been processed */
73 unsigned int processed
:1;
74 /* have backrefs of this block been checked */
75 unsigned int checked
:1;
77 * 1 if corresponding block has been cowed but some upper
78 * level block pointers may not point to the new location
80 unsigned int pending
:1;
82 * 1 if the backref node isn't connected to any other
85 unsigned int detached
:1;
89 * present a block pointer in the backref cache
92 struct list_head list
[2];
93 struct backref_node
*node
[2];
98 #define RELOCATION_RESERVED_NODES 256
100 struct backref_cache
{
101 /* red black tree of all backref nodes in the cache */
102 struct rb_root rb_root
;
103 /* for passing backref nodes to btrfs_reloc_cow_block */
104 struct backref_node
*path
[BTRFS_MAX_LEVEL
];
106 * list of blocks that have been cowed but some block
107 * pointers in upper level blocks may not reflect the
110 struct list_head pending
[BTRFS_MAX_LEVEL
];
111 /* list of backref nodes with no child node */
112 struct list_head leaves
;
113 /* list of blocks that have been cowed in current transaction */
114 struct list_head changed
;
115 /* list of detached backref node. */
116 struct list_head detached
;
125 * map address of tree root to tree
127 struct mapping_node
{
128 struct rb_node rb_node
;
133 struct mapping_tree
{
134 struct rb_root rb_root
;
139 * present a tree block to process
142 struct rb_node rb_node
;
144 struct btrfs_key key
;
145 unsigned int level
:8;
146 unsigned int key_ready
:1;
149 #define MAX_EXTENTS 128
151 struct file_extent_cluster
{
154 u64 boundary
[MAX_EXTENTS
];
158 struct reloc_control
{
159 /* block group to relocate */
160 struct btrfs_block_group_cache
*block_group
;
162 struct btrfs_root
*extent_root
;
163 /* inode for moving data */
164 struct inode
*data_inode
;
166 struct btrfs_block_rsv
*block_rsv
;
168 struct backref_cache backref_cache
;
170 struct file_extent_cluster cluster
;
171 /* tree blocks have been processed */
172 struct extent_io_tree processed_blocks
;
173 /* map start of tree root to corresponding reloc tree */
174 struct mapping_tree reloc_root_tree
;
175 /* list of reloc trees */
176 struct list_head reloc_roots
;
177 /* size of metadata reservation for merging reloc trees */
178 u64 merging_rsv_size
;
179 /* size of relocated tree nodes */
181 /* reserved size for block group relocation*/
187 unsigned int stage
:8;
188 unsigned int create_reloc_tree
:1;
189 unsigned int merge_reloc_tree
:1;
190 unsigned int found_file_extent
:1;
193 /* stages of data relocation */
194 #define MOVE_DATA_EXTENTS 0
195 #define UPDATE_DATA_PTRS 1
197 static void remove_backref_node(struct backref_cache
*cache
,
198 struct backref_node
*node
);
199 static void __mark_block_processed(struct reloc_control
*rc
,
200 struct backref_node
*node
);
202 static void mapping_tree_init(struct mapping_tree
*tree
)
204 tree
->rb_root
= RB_ROOT
;
205 spin_lock_init(&tree
->lock
);
208 static void backref_cache_init(struct backref_cache
*cache
)
211 cache
->rb_root
= RB_ROOT
;
212 for (i
= 0; i
< BTRFS_MAX_LEVEL
; i
++)
213 INIT_LIST_HEAD(&cache
->pending
[i
]);
214 INIT_LIST_HEAD(&cache
->changed
);
215 INIT_LIST_HEAD(&cache
->detached
);
216 INIT_LIST_HEAD(&cache
->leaves
);
219 static void backref_cache_cleanup(struct backref_cache
*cache
)
221 struct backref_node
*node
;
224 while (!list_empty(&cache
->detached
)) {
225 node
= list_entry(cache
->detached
.next
,
226 struct backref_node
, list
);
227 remove_backref_node(cache
, node
);
230 while (!list_empty(&cache
->leaves
)) {
231 node
= list_entry(cache
->leaves
.next
,
232 struct backref_node
, lower
);
233 remove_backref_node(cache
, node
);
236 cache
->last_trans
= 0;
238 for (i
= 0; i
< BTRFS_MAX_LEVEL
; i
++)
239 ASSERT(list_empty(&cache
->pending
[i
]));
240 ASSERT(list_empty(&cache
->changed
));
241 ASSERT(list_empty(&cache
->detached
));
242 ASSERT(RB_EMPTY_ROOT(&cache
->rb_root
));
243 ASSERT(!cache
->nr_nodes
);
244 ASSERT(!cache
->nr_edges
);
247 static struct backref_node
*alloc_backref_node(struct backref_cache
*cache
)
249 struct backref_node
*node
;
251 node
= kzalloc(sizeof(*node
), GFP_NOFS
);
253 INIT_LIST_HEAD(&node
->list
);
254 INIT_LIST_HEAD(&node
->upper
);
255 INIT_LIST_HEAD(&node
->lower
);
256 RB_CLEAR_NODE(&node
->rb_node
);
262 static void free_backref_node(struct backref_cache
*cache
,
263 struct backref_node
*node
)
271 static struct backref_edge
*alloc_backref_edge(struct backref_cache
*cache
)
273 struct backref_edge
*edge
;
275 edge
= kzalloc(sizeof(*edge
), GFP_NOFS
);
281 static void free_backref_edge(struct backref_cache
*cache
,
282 struct backref_edge
*edge
)
290 static struct rb_node
*tree_insert(struct rb_root
*root
, u64 bytenr
,
291 struct rb_node
*node
)
293 struct rb_node
**p
= &root
->rb_node
;
294 struct rb_node
*parent
= NULL
;
295 struct tree_entry
*entry
;
299 entry
= rb_entry(parent
, struct tree_entry
, rb_node
);
301 if (bytenr
< entry
->bytenr
)
303 else if (bytenr
> entry
->bytenr
)
309 rb_link_node(node
, parent
, p
);
310 rb_insert_color(node
, root
);
314 static struct rb_node
*tree_search(struct rb_root
*root
, u64 bytenr
)
316 struct rb_node
*n
= root
->rb_node
;
317 struct tree_entry
*entry
;
320 entry
= rb_entry(n
, struct tree_entry
, rb_node
);
322 if (bytenr
< entry
->bytenr
)
324 else if (bytenr
> entry
->bytenr
)
332 static void backref_tree_panic(struct rb_node
*rb_node
, int errno
, u64 bytenr
)
335 struct btrfs_fs_info
*fs_info
= NULL
;
336 struct backref_node
*bnode
= rb_entry(rb_node
, struct backref_node
,
339 fs_info
= bnode
->root
->fs_info
;
340 btrfs_panic(fs_info
, errno
,
341 "Inconsistency in backref cache found at offset %llu",
346 * walk up backref nodes until reach node presents tree root
348 static struct backref_node
*walk_up_backref(struct backref_node
*node
,
349 struct backref_edge
*edges
[],
352 struct backref_edge
*edge
;
355 while (!list_empty(&node
->upper
)) {
356 edge
= list_entry(node
->upper
.next
,
357 struct backref_edge
, list
[LOWER
]);
359 node
= edge
->node
[UPPER
];
361 BUG_ON(node
->detached
);
367 * walk down backref nodes to find start of next reference path
369 static struct backref_node
*walk_down_backref(struct backref_edge
*edges
[],
372 struct backref_edge
*edge
;
373 struct backref_node
*lower
;
377 edge
= edges
[idx
- 1];
378 lower
= edge
->node
[LOWER
];
379 if (list_is_last(&edge
->list
[LOWER
], &lower
->upper
)) {
383 edge
= list_entry(edge
->list
[LOWER
].next
,
384 struct backref_edge
, list
[LOWER
]);
385 edges
[idx
- 1] = edge
;
387 return edge
->node
[UPPER
];
393 static void unlock_node_buffer(struct backref_node
*node
)
396 btrfs_tree_unlock(node
->eb
);
401 static void drop_node_buffer(struct backref_node
*node
)
404 unlock_node_buffer(node
);
405 free_extent_buffer(node
->eb
);
410 static void drop_backref_node(struct backref_cache
*tree
,
411 struct backref_node
*node
)
413 BUG_ON(!list_empty(&node
->upper
));
415 drop_node_buffer(node
);
416 list_del(&node
->list
);
417 list_del(&node
->lower
);
418 if (!RB_EMPTY_NODE(&node
->rb_node
))
419 rb_erase(&node
->rb_node
, &tree
->rb_root
);
420 free_backref_node(tree
, node
);
424 * remove a backref node from the backref cache
426 static void remove_backref_node(struct backref_cache
*cache
,
427 struct backref_node
*node
)
429 struct backref_node
*upper
;
430 struct backref_edge
*edge
;
435 BUG_ON(!node
->lowest
&& !node
->detached
);
436 while (!list_empty(&node
->upper
)) {
437 edge
= list_entry(node
->upper
.next
, struct backref_edge
,
439 upper
= edge
->node
[UPPER
];
440 list_del(&edge
->list
[LOWER
]);
441 list_del(&edge
->list
[UPPER
]);
442 free_backref_edge(cache
, edge
);
444 if (RB_EMPTY_NODE(&upper
->rb_node
)) {
445 BUG_ON(!list_empty(&node
->upper
));
446 drop_backref_node(cache
, node
);
452 * add the node to leaf node list if no other
453 * child block cached.
455 if (list_empty(&upper
->lower
)) {
456 list_add_tail(&upper
->lower
, &cache
->leaves
);
461 drop_backref_node(cache
, node
);
464 static void update_backref_node(struct backref_cache
*cache
,
465 struct backref_node
*node
, u64 bytenr
)
467 struct rb_node
*rb_node
;
468 rb_erase(&node
->rb_node
, &cache
->rb_root
);
469 node
->bytenr
= bytenr
;
470 rb_node
= tree_insert(&cache
->rb_root
, node
->bytenr
, &node
->rb_node
);
472 backref_tree_panic(rb_node
, -EEXIST
, bytenr
);
476 * update backref cache after a transaction commit
478 static int update_backref_cache(struct btrfs_trans_handle
*trans
,
479 struct backref_cache
*cache
)
481 struct backref_node
*node
;
484 if (cache
->last_trans
== 0) {
485 cache
->last_trans
= trans
->transid
;
489 if (cache
->last_trans
== trans
->transid
)
493 * detached nodes are used to avoid unnecessary backref
494 * lookup. transaction commit changes the extent tree.
495 * so the detached nodes are no longer useful.
497 while (!list_empty(&cache
->detached
)) {
498 node
= list_entry(cache
->detached
.next
,
499 struct backref_node
, list
);
500 remove_backref_node(cache
, node
);
503 while (!list_empty(&cache
->changed
)) {
504 node
= list_entry(cache
->changed
.next
,
505 struct backref_node
, list
);
506 list_del_init(&node
->list
);
507 BUG_ON(node
->pending
);
508 update_backref_node(cache
, node
, node
->new_bytenr
);
512 * some nodes can be left in the pending list if there were
513 * errors during processing the pending nodes.
515 for (level
= 0; level
< BTRFS_MAX_LEVEL
; level
++) {
516 list_for_each_entry(node
, &cache
->pending
[level
], list
) {
517 BUG_ON(!node
->pending
);
518 if (node
->bytenr
== node
->new_bytenr
)
520 update_backref_node(cache
, node
, node
->new_bytenr
);
524 cache
->last_trans
= 0;
529 static int should_ignore_root(struct btrfs_root
*root
)
531 struct btrfs_root
*reloc_root
;
533 if (!test_bit(BTRFS_ROOT_REF_COWS
, &root
->state
))
536 reloc_root
= root
->reloc_root
;
540 if (btrfs_root_last_snapshot(&reloc_root
->root_item
) ==
541 root
->fs_info
->running_transaction
->transid
- 1)
544 * if there is reloc tree and it was created in previous
545 * transaction backref lookup can find the reloc tree,
546 * so backref node for the fs tree root is useless for
552 * find reloc tree by address of tree root
554 static struct btrfs_root
*find_reloc_root(struct reloc_control
*rc
,
557 struct rb_node
*rb_node
;
558 struct mapping_node
*node
;
559 struct btrfs_root
*root
= NULL
;
561 spin_lock(&rc
->reloc_root_tree
.lock
);
562 rb_node
= tree_search(&rc
->reloc_root_tree
.rb_root
, bytenr
);
564 node
= rb_entry(rb_node
, struct mapping_node
, rb_node
);
565 root
= (struct btrfs_root
*)node
->data
;
567 spin_unlock(&rc
->reloc_root_tree
.lock
);
571 static int is_cowonly_root(u64 root_objectid
)
573 if (root_objectid
== BTRFS_ROOT_TREE_OBJECTID
||
574 root_objectid
== BTRFS_EXTENT_TREE_OBJECTID
||
575 root_objectid
== BTRFS_CHUNK_TREE_OBJECTID
||
576 root_objectid
== BTRFS_DEV_TREE_OBJECTID
||
577 root_objectid
== BTRFS_TREE_LOG_OBJECTID
||
578 root_objectid
== BTRFS_CSUM_TREE_OBJECTID
||
579 root_objectid
== BTRFS_UUID_TREE_OBJECTID
||
580 root_objectid
== BTRFS_QUOTA_TREE_OBJECTID
||
581 root_objectid
== BTRFS_FREE_SPACE_TREE_OBJECTID
)
586 static struct btrfs_root
*read_fs_root(struct btrfs_fs_info
*fs_info
,
589 struct btrfs_key key
;
591 key
.objectid
= root_objectid
;
592 key
.type
= BTRFS_ROOT_ITEM_KEY
;
593 if (is_cowonly_root(root_objectid
))
596 key
.offset
= (u64
)-1;
598 return btrfs_get_fs_root(fs_info
, &key
, false);
601 #ifdef BTRFS_COMPAT_EXTENT_TREE_V0
602 static noinline_for_stack
603 struct btrfs_root
*find_tree_root(struct reloc_control
*rc
,
604 struct extent_buffer
*leaf
,
605 struct btrfs_extent_ref_v0
*ref0
)
607 struct btrfs_root
*root
;
608 u64 root_objectid
= btrfs_ref_root_v0(leaf
, ref0
);
609 u64 generation
= btrfs_ref_generation_v0(leaf
, ref0
);
611 BUG_ON(root_objectid
== BTRFS_TREE_RELOC_OBJECTID
);
613 root
= read_fs_root(rc
->extent_root
->fs_info
, root_objectid
);
614 BUG_ON(IS_ERR(root
));
616 if (test_bit(BTRFS_ROOT_REF_COWS
, &root
->state
) &&
617 generation
!= btrfs_root_generation(&root
->root_item
))
624 static noinline_for_stack
625 int find_inline_backref(struct extent_buffer
*leaf
, int slot
,
626 unsigned long *ptr
, unsigned long *end
)
628 struct btrfs_key key
;
629 struct btrfs_extent_item
*ei
;
630 struct btrfs_tree_block_info
*bi
;
633 btrfs_item_key_to_cpu(leaf
, &key
, slot
);
635 item_size
= btrfs_item_size_nr(leaf
, slot
);
636 #ifdef BTRFS_COMPAT_EXTENT_TREE_V0
637 if (item_size
< sizeof(*ei
)) {
638 WARN_ON(item_size
!= sizeof(struct btrfs_extent_item_v0
));
642 ei
= btrfs_item_ptr(leaf
, slot
, struct btrfs_extent_item
);
643 WARN_ON(!(btrfs_extent_flags(leaf
, ei
) &
644 BTRFS_EXTENT_FLAG_TREE_BLOCK
));
646 if (key
.type
== BTRFS_EXTENT_ITEM_KEY
&&
647 item_size
<= sizeof(*ei
) + sizeof(*bi
)) {
648 WARN_ON(item_size
< sizeof(*ei
) + sizeof(*bi
));
651 if (key
.type
== BTRFS_METADATA_ITEM_KEY
&&
652 item_size
<= sizeof(*ei
)) {
653 WARN_ON(item_size
< sizeof(*ei
));
657 if (key
.type
== BTRFS_EXTENT_ITEM_KEY
) {
658 bi
= (struct btrfs_tree_block_info
*)(ei
+ 1);
659 *ptr
= (unsigned long)(bi
+ 1);
661 *ptr
= (unsigned long)(ei
+ 1);
663 *end
= (unsigned long)ei
+ item_size
;
668 * build backref tree for a given tree block. root of the backref tree
669 * corresponds the tree block, leaves of the backref tree correspond
670 * roots of b-trees that reference the tree block.
672 * the basic idea of this function is check backrefs of a given block
673 * to find upper level blocks that reference the block, and then check
674 * backrefs of these upper level blocks recursively. the recursion stop
675 * when tree root is reached or backrefs for the block is cached.
677 * NOTE: if we find backrefs for a block are cached, we know backrefs
678 * for all upper level blocks that directly/indirectly reference the
679 * block are also cached.
681 static noinline_for_stack
682 struct backref_node
*build_backref_tree(struct reloc_control
*rc
,
683 struct btrfs_key
*node_key
,
684 int level
, u64 bytenr
)
686 struct backref_cache
*cache
= &rc
->backref_cache
;
687 struct btrfs_path
*path1
;
688 struct btrfs_path
*path2
;
689 struct extent_buffer
*eb
;
690 struct btrfs_root
*root
;
691 struct backref_node
*cur
;
692 struct backref_node
*upper
;
693 struct backref_node
*lower
;
694 struct backref_node
*node
= NULL
;
695 struct backref_node
*exist
= NULL
;
696 struct backref_edge
*edge
;
697 struct rb_node
*rb_node
;
698 struct btrfs_key key
;
706 bool need_check
= true;
708 path1
= btrfs_alloc_path();
709 path2
= btrfs_alloc_path();
710 if (!path1
|| !path2
) {
714 path1
->reada
= READA_FORWARD
;
715 path2
->reada
= READA_FORWARD
;
717 node
= alloc_backref_node(cache
);
723 node
->bytenr
= bytenr
;
730 key
.objectid
= cur
->bytenr
;
731 key
.type
= BTRFS_METADATA_ITEM_KEY
;
732 key
.offset
= (u64
)-1;
734 path1
->search_commit_root
= 1;
735 path1
->skip_locking
= 1;
736 ret
= btrfs_search_slot(NULL
, rc
->extent_root
, &key
, path1
,
743 ASSERT(path1
->slots
[0]);
747 WARN_ON(cur
->checked
);
748 if (!list_empty(&cur
->upper
)) {
750 * the backref was added previously when processing
751 * backref of type BTRFS_TREE_BLOCK_REF_KEY
753 ASSERT(list_is_singular(&cur
->upper
));
754 edge
= list_entry(cur
->upper
.next
, struct backref_edge
,
756 ASSERT(list_empty(&edge
->list
[UPPER
]));
757 exist
= edge
->node
[UPPER
];
759 * add the upper level block to pending list if we need
763 list_add_tail(&edge
->list
[UPPER
], &list
);
770 eb
= path1
->nodes
[0];
773 if (path1
->slots
[0] >= btrfs_header_nritems(eb
)) {
774 ret
= btrfs_next_leaf(rc
->extent_root
, path1
);
781 eb
= path1
->nodes
[0];
784 btrfs_item_key_to_cpu(eb
, &key
, path1
->slots
[0]);
785 if (key
.objectid
!= cur
->bytenr
) {
790 if (key
.type
== BTRFS_EXTENT_ITEM_KEY
||
791 key
.type
== BTRFS_METADATA_ITEM_KEY
) {
792 ret
= find_inline_backref(eb
, path1
->slots
[0],
800 /* update key for inline back ref */
801 struct btrfs_extent_inline_ref
*iref
;
802 iref
= (struct btrfs_extent_inline_ref
*)ptr
;
803 key
.type
= btrfs_extent_inline_ref_type(eb
, iref
);
804 key
.offset
= btrfs_extent_inline_ref_offset(eb
, iref
);
805 WARN_ON(key
.type
!= BTRFS_TREE_BLOCK_REF_KEY
&&
806 key
.type
!= BTRFS_SHARED_BLOCK_REF_KEY
);
810 ((key
.type
== BTRFS_TREE_BLOCK_REF_KEY
&&
811 exist
->owner
== key
.offset
) ||
812 (key
.type
== BTRFS_SHARED_BLOCK_REF_KEY
&&
813 exist
->bytenr
== key
.offset
))) {
818 #ifdef BTRFS_COMPAT_EXTENT_TREE_V0
819 if (key
.type
== BTRFS_SHARED_BLOCK_REF_KEY
||
820 key
.type
== BTRFS_EXTENT_REF_V0_KEY
) {
821 if (key
.type
== BTRFS_EXTENT_REF_V0_KEY
) {
822 struct btrfs_extent_ref_v0
*ref0
;
823 ref0
= btrfs_item_ptr(eb
, path1
->slots
[0],
824 struct btrfs_extent_ref_v0
);
825 if (key
.objectid
== key
.offset
) {
826 root
= find_tree_root(rc
, eb
, ref0
);
827 if (root
&& !should_ignore_root(root
))
830 list_add(&cur
->list
, &useless
);
833 if (is_cowonly_root(btrfs_ref_root_v0(eb
,
838 ASSERT(key
.type
!= BTRFS_EXTENT_REF_V0_KEY
);
839 if (key
.type
== BTRFS_SHARED_BLOCK_REF_KEY
) {
841 if (key
.objectid
== key
.offset
) {
843 * only root blocks of reloc trees use
844 * backref of this type.
846 root
= find_reloc_root(rc
, cur
->bytenr
);
852 edge
= alloc_backref_edge(cache
);
857 rb_node
= tree_search(&cache
->rb_root
, key
.offset
);
859 upper
= alloc_backref_node(cache
);
861 free_backref_edge(cache
, edge
);
865 upper
->bytenr
= key
.offset
;
866 upper
->level
= cur
->level
+ 1;
868 * backrefs for the upper level block isn't
869 * cached, add the block to pending list
871 list_add_tail(&edge
->list
[UPPER
], &list
);
873 upper
= rb_entry(rb_node
, struct backref_node
,
875 ASSERT(upper
->checked
);
876 INIT_LIST_HEAD(&edge
->list
[UPPER
]);
878 list_add_tail(&edge
->list
[LOWER
], &cur
->upper
);
879 edge
->node
[LOWER
] = cur
;
880 edge
->node
[UPPER
] = upper
;
883 } else if (key
.type
!= BTRFS_TREE_BLOCK_REF_KEY
) {
887 /* key.type == BTRFS_TREE_BLOCK_REF_KEY */
888 root
= read_fs_root(rc
->extent_root
->fs_info
, key
.offset
);
894 if (!test_bit(BTRFS_ROOT_REF_COWS
, &root
->state
))
897 if (btrfs_root_level(&root
->root_item
) == cur
->level
) {
899 ASSERT(btrfs_root_bytenr(&root
->root_item
) ==
901 if (should_ignore_root(root
))
902 list_add(&cur
->list
, &useless
);
908 level
= cur
->level
+ 1;
911 * searching the tree to find upper level blocks
912 * reference the block.
914 path2
->search_commit_root
= 1;
915 path2
->skip_locking
= 1;
916 path2
->lowest_level
= level
;
917 ret
= btrfs_search_slot(NULL
, root
, node_key
, path2
, 0, 0);
918 path2
->lowest_level
= 0;
923 if (ret
> 0 && path2
->slots
[level
] > 0)
924 path2
->slots
[level
]--;
926 eb
= path2
->nodes
[level
];
927 if (btrfs_node_blockptr(eb
, path2
->slots
[level
]) !=
929 btrfs_err(root
->fs_info
,
930 "couldn't find block (%llu) (level %d) in tree (%llu) with key (%llu %u %llu)",
931 cur
->bytenr
, level
- 1, root
->objectid
,
932 node_key
->objectid
, node_key
->type
,
939 for (; level
< BTRFS_MAX_LEVEL
; level
++) {
940 if (!path2
->nodes
[level
]) {
941 ASSERT(btrfs_root_bytenr(&root
->root_item
) ==
943 if (should_ignore_root(root
))
944 list_add(&lower
->list
, &useless
);
950 edge
= alloc_backref_edge(cache
);
956 eb
= path2
->nodes
[level
];
957 rb_node
= tree_search(&cache
->rb_root
, eb
->start
);
959 upper
= alloc_backref_node(cache
);
961 free_backref_edge(cache
, edge
);
965 upper
->bytenr
= eb
->start
;
966 upper
->owner
= btrfs_header_owner(eb
);
967 upper
->level
= lower
->level
+ 1;
968 if (!test_bit(BTRFS_ROOT_REF_COWS
,
973 * if we know the block isn't shared
974 * we can void checking its backrefs.
976 if (btrfs_block_can_be_shared(root
, eb
))
982 * add the block to pending list if we
983 * need check its backrefs, we only do this once
984 * while walking up a tree as we will catch
985 * anything else later on.
987 if (!upper
->checked
&& need_check
) {
989 list_add_tail(&edge
->list
[UPPER
],
994 INIT_LIST_HEAD(&edge
->list
[UPPER
]);
997 upper
= rb_entry(rb_node
, struct backref_node
,
999 ASSERT(upper
->checked
);
1000 INIT_LIST_HEAD(&edge
->list
[UPPER
]);
1002 upper
->owner
= btrfs_header_owner(eb
);
1004 list_add_tail(&edge
->list
[LOWER
], &lower
->upper
);
1005 edge
->node
[LOWER
] = lower
;
1006 edge
->node
[UPPER
] = upper
;
1013 btrfs_release_path(path2
);
1016 ptr
+= btrfs_extent_inline_ref_size(key
.type
);
1026 btrfs_release_path(path1
);
1031 /* the pending list isn't empty, take the first block to process */
1032 if (!list_empty(&list
)) {
1033 edge
= list_entry(list
.next
, struct backref_edge
, list
[UPPER
]);
1034 list_del_init(&edge
->list
[UPPER
]);
1035 cur
= edge
->node
[UPPER
];
1040 * everything goes well, connect backref nodes and insert backref nodes
1043 ASSERT(node
->checked
);
1044 cowonly
= node
->cowonly
;
1046 rb_node
= tree_insert(&cache
->rb_root
, node
->bytenr
,
1049 backref_tree_panic(rb_node
, -EEXIST
, node
->bytenr
);
1050 list_add_tail(&node
->lower
, &cache
->leaves
);
1053 list_for_each_entry(edge
, &node
->upper
, list
[LOWER
])
1054 list_add_tail(&edge
->list
[UPPER
], &list
);
1056 while (!list_empty(&list
)) {
1057 edge
= list_entry(list
.next
, struct backref_edge
, list
[UPPER
]);
1058 list_del_init(&edge
->list
[UPPER
]);
1059 upper
= edge
->node
[UPPER
];
1060 if (upper
->detached
) {
1061 list_del(&edge
->list
[LOWER
]);
1062 lower
= edge
->node
[LOWER
];
1063 free_backref_edge(cache
, edge
);
1064 if (list_empty(&lower
->upper
))
1065 list_add(&lower
->list
, &useless
);
1069 if (!RB_EMPTY_NODE(&upper
->rb_node
)) {
1070 if (upper
->lowest
) {
1071 list_del_init(&upper
->lower
);
1075 list_add_tail(&edge
->list
[UPPER
], &upper
->lower
);
1079 if (!upper
->checked
) {
1081 * Still want to blow up for developers since this is a
1088 if (cowonly
!= upper
->cowonly
) {
1095 rb_node
= tree_insert(&cache
->rb_root
, upper
->bytenr
,
1098 backref_tree_panic(rb_node
, -EEXIST
,
1102 list_add_tail(&edge
->list
[UPPER
], &upper
->lower
);
1104 list_for_each_entry(edge
, &upper
->upper
, list
[LOWER
])
1105 list_add_tail(&edge
->list
[UPPER
], &list
);
1108 * process useless backref nodes. backref nodes for tree leaves
1109 * are deleted from the cache. backref nodes for upper level
1110 * tree blocks are left in the cache to avoid unnecessary backref
1113 while (!list_empty(&useless
)) {
1114 upper
= list_entry(useless
.next
, struct backref_node
, list
);
1115 list_del_init(&upper
->list
);
1116 ASSERT(list_empty(&upper
->upper
));
1119 if (upper
->lowest
) {
1120 list_del_init(&upper
->lower
);
1123 while (!list_empty(&upper
->lower
)) {
1124 edge
= list_entry(upper
->lower
.next
,
1125 struct backref_edge
, list
[UPPER
]);
1126 list_del(&edge
->list
[UPPER
]);
1127 list_del(&edge
->list
[LOWER
]);
1128 lower
= edge
->node
[LOWER
];
1129 free_backref_edge(cache
, edge
);
1131 if (list_empty(&lower
->upper
))
1132 list_add(&lower
->list
, &useless
);
1134 __mark_block_processed(rc
, upper
);
1135 if (upper
->level
> 0) {
1136 list_add(&upper
->list
, &cache
->detached
);
1137 upper
->detached
= 1;
1139 rb_erase(&upper
->rb_node
, &cache
->rb_root
);
1140 free_backref_node(cache
, upper
);
1144 btrfs_free_path(path1
);
1145 btrfs_free_path(path2
);
1147 while (!list_empty(&useless
)) {
1148 lower
= list_entry(useless
.next
,
1149 struct backref_node
, list
);
1150 list_del_init(&lower
->list
);
1152 while (!list_empty(&list
)) {
1153 edge
= list_first_entry(&list
, struct backref_edge
,
1155 list_del(&edge
->list
[UPPER
]);
1156 list_del(&edge
->list
[LOWER
]);
1157 lower
= edge
->node
[LOWER
];
1158 upper
= edge
->node
[UPPER
];
1159 free_backref_edge(cache
, edge
);
1162 * Lower is no longer linked to any upper backref nodes
1163 * and isn't in the cache, we can free it ourselves.
1165 if (list_empty(&lower
->upper
) &&
1166 RB_EMPTY_NODE(&lower
->rb_node
))
1167 list_add(&lower
->list
, &useless
);
1169 if (!RB_EMPTY_NODE(&upper
->rb_node
))
1172 /* Add this guy's upper edges to the list to process */
1173 list_for_each_entry(edge
, &upper
->upper
, list
[LOWER
])
1174 list_add_tail(&edge
->list
[UPPER
], &list
);
1175 if (list_empty(&upper
->upper
))
1176 list_add(&upper
->list
, &useless
);
1179 while (!list_empty(&useless
)) {
1180 lower
= list_entry(useless
.next
,
1181 struct backref_node
, list
);
1182 list_del_init(&lower
->list
);
1185 free_backref_node(cache
, lower
);
1188 free_backref_node(cache
, node
);
1189 return ERR_PTR(err
);
1191 ASSERT(!node
|| !node
->detached
);
1196 * helper to add backref node for the newly created snapshot.
1197 * the backref node is created by cloning backref node that
1198 * corresponds to root of source tree
1200 static int clone_backref_node(struct btrfs_trans_handle
*trans
,
1201 struct reloc_control
*rc
,
1202 struct btrfs_root
*src
,
1203 struct btrfs_root
*dest
)
1205 struct btrfs_root
*reloc_root
= src
->reloc_root
;
1206 struct backref_cache
*cache
= &rc
->backref_cache
;
1207 struct backref_node
*node
= NULL
;
1208 struct backref_node
*new_node
;
1209 struct backref_edge
*edge
;
1210 struct backref_edge
*new_edge
;
1211 struct rb_node
*rb_node
;
1213 if (cache
->last_trans
> 0)
1214 update_backref_cache(trans
, cache
);
1216 rb_node
= tree_search(&cache
->rb_root
, src
->commit_root
->start
);
1218 node
= rb_entry(rb_node
, struct backref_node
, rb_node
);
1222 BUG_ON(node
->new_bytenr
!= reloc_root
->node
->start
);
1226 rb_node
= tree_search(&cache
->rb_root
,
1227 reloc_root
->commit_root
->start
);
1229 node
= rb_entry(rb_node
, struct backref_node
,
1231 BUG_ON(node
->detached
);
1238 new_node
= alloc_backref_node(cache
);
1242 new_node
->bytenr
= dest
->node
->start
;
1243 new_node
->level
= node
->level
;
1244 new_node
->lowest
= node
->lowest
;
1245 new_node
->checked
= 1;
1246 new_node
->root
= dest
;
1248 if (!node
->lowest
) {
1249 list_for_each_entry(edge
, &node
->lower
, list
[UPPER
]) {
1250 new_edge
= alloc_backref_edge(cache
);
1254 new_edge
->node
[UPPER
] = new_node
;
1255 new_edge
->node
[LOWER
] = edge
->node
[LOWER
];
1256 list_add_tail(&new_edge
->list
[UPPER
],
1260 list_add_tail(&new_node
->lower
, &cache
->leaves
);
1263 rb_node
= tree_insert(&cache
->rb_root
, new_node
->bytenr
,
1264 &new_node
->rb_node
);
1266 backref_tree_panic(rb_node
, -EEXIST
, new_node
->bytenr
);
1268 if (!new_node
->lowest
) {
1269 list_for_each_entry(new_edge
, &new_node
->lower
, list
[UPPER
]) {
1270 list_add_tail(&new_edge
->list
[LOWER
],
1271 &new_edge
->node
[LOWER
]->upper
);
1276 while (!list_empty(&new_node
->lower
)) {
1277 new_edge
= list_entry(new_node
->lower
.next
,
1278 struct backref_edge
, list
[UPPER
]);
1279 list_del(&new_edge
->list
[UPPER
]);
1280 free_backref_edge(cache
, new_edge
);
1282 free_backref_node(cache
, new_node
);
1287 * helper to add 'address of tree root -> reloc tree' mapping
1289 static int __must_check
__add_reloc_root(struct btrfs_root
*root
)
1291 struct btrfs_fs_info
*fs_info
= root
->fs_info
;
1292 struct rb_node
*rb_node
;
1293 struct mapping_node
*node
;
1294 struct reloc_control
*rc
= fs_info
->reloc_ctl
;
1296 node
= kmalloc(sizeof(*node
), GFP_NOFS
);
1300 node
->bytenr
= root
->node
->start
;
1303 spin_lock(&rc
->reloc_root_tree
.lock
);
1304 rb_node
= tree_insert(&rc
->reloc_root_tree
.rb_root
,
1305 node
->bytenr
, &node
->rb_node
);
1306 spin_unlock(&rc
->reloc_root_tree
.lock
);
1308 btrfs_panic(fs_info
, -EEXIST
,
1309 "Duplicate root found for start=%llu while inserting into relocation tree",
1315 list_add_tail(&root
->root_list
, &rc
->reloc_roots
);
1320 * helper to delete the 'address of tree root -> reloc tree'
1323 static void __del_reloc_root(struct btrfs_root
*root
)
1325 struct btrfs_fs_info
*fs_info
= root
->fs_info
;
1326 struct rb_node
*rb_node
;
1327 struct mapping_node
*node
= NULL
;
1328 struct reloc_control
*rc
= fs_info
->reloc_ctl
;
1330 spin_lock(&rc
->reloc_root_tree
.lock
);
1331 rb_node
= tree_search(&rc
->reloc_root_tree
.rb_root
,
1334 node
= rb_entry(rb_node
, struct mapping_node
, rb_node
);
1335 rb_erase(&node
->rb_node
, &rc
->reloc_root_tree
.rb_root
);
1337 spin_unlock(&rc
->reloc_root_tree
.lock
);
1341 BUG_ON((struct btrfs_root
*)node
->data
!= root
);
1343 spin_lock(&fs_info
->trans_lock
);
1344 list_del_init(&root
->root_list
);
1345 spin_unlock(&fs_info
->trans_lock
);
1350 * helper to update the 'address of tree root -> reloc tree'
1353 static int __update_reloc_root(struct btrfs_root
*root
, u64 new_bytenr
)
1355 struct btrfs_fs_info
*fs_info
= root
->fs_info
;
1356 struct rb_node
*rb_node
;
1357 struct mapping_node
*node
= NULL
;
1358 struct reloc_control
*rc
= fs_info
->reloc_ctl
;
1360 spin_lock(&rc
->reloc_root_tree
.lock
);
1361 rb_node
= tree_search(&rc
->reloc_root_tree
.rb_root
,
1364 node
= rb_entry(rb_node
, struct mapping_node
, rb_node
);
1365 rb_erase(&node
->rb_node
, &rc
->reloc_root_tree
.rb_root
);
1367 spin_unlock(&rc
->reloc_root_tree
.lock
);
1371 BUG_ON((struct btrfs_root
*)node
->data
!= root
);
1373 spin_lock(&rc
->reloc_root_tree
.lock
);
1374 node
->bytenr
= new_bytenr
;
1375 rb_node
= tree_insert(&rc
->reloc_root_tree
.rb_root
,
1376 node
->bytenr
, &node
->rb_node
);
1377 spin_unlock(&rc
->reloc_root_tree
.lock
);
1379 backref_tree_panic(rb_node
, -EEXIST
, node
->bytenr
);
1383 static struct btrfs_root
*create_reloc_root(struct btrfs_trans_handle
*trans
,
1384 struct btrfs_root
*root
, u64 objectid
)
1386 struct btrfs_fs_info
*fs_info
= root
->fs_info
;
1387 struct btrfs_root
*reloc_root
;
1388 struct extent_buffer
*eb
;
1389 struct btrfs_root_item
*root_item
;
1390 struct btrfs_key root_key
;
1394 root_item
= kmalloc(sizeof(*root_item
), GFP_NOFS
);
1397 root_key
.objectid
= BTRFS_TREE_RELOC_OBJECTID
;
1398 root_key
.type
= BTRFS_ROOT_ITEM_KEY
;
1399 root_key
.offset
= objectid
;
1401 if (root
->root_key
.objectid
== objectid
) {
1402 /* called by btrfs_init_reloc_root */
1403 ret
= btrfs_copy_root(trans
, root
, root
->commit_root
, &eb
,
1404 BTRFS_TREE_RELOC_OBJECTID
);
1407 last_snap
= btrfs_root_last_snapshot(&root
->root_item
);
1408 btrfs_set_root_last_snapshot(&root
->root_item
,
1409 trans
->transid
- 1);
1412 * called by btrfs_reloc_post_snapshot_hook.
1413 * the source tree is a reloc tree, all tree blocks
1414 * modified after it was created have RELOC flag
1415 * set in their headers. so it's OK to not update
1416 * the 'last_snapshot'.
1418 ret
= btrfs_copy_root(trans
, root
, root
->node
, &eb
,
1419 BTRFS_TREE_RELOC_OBJECTID
);
1423 memcpy(root_item
, &root
->root_item
, sizeof(*root_item
));
1424 btrfs_set_root_bytenr(root_item
, eb
->start
);
1425 btrfs_set_root_level(root_item
, btrfs_header_level(eb
));
1426 btrfs_set_root_generation(root_item
, trans
->transid
);
1428 if (root
->root_key
.objectid
== objectid
) {
1429 btrfs_set_root_refs(root_item
, 0);
1430 memset(&root_item
->drop_progress
, 0,
1431 sizeof(struct btrfs_disk_key
));
1432 root_item
->drop_level
= 0;
1434 * abuse rtransid, it is safe because it is impossible to
1435 * receive data into a relocation tree.
1437 btrfs_set_root_rtransid(root_item
, last_snap
);
1438 btrfs_set_root_otransid(root_item
, trans
->transid
);
1441 btrfs_tree_unlock(eb
);
1442 free_extent_buffer(eb
);
1444 ret
= btrfs_insert_root(trans
, fs_info
->tree_root
,
1445 &root_key
, root_item
);
1449 reloc_root
= btrfs_read_fs_root(fs_info
->tree_root
, &root_key
);
1450 BUG_ON(IS_ERR(reloc_root
));
1451 reloc_root
->last_trans
= trans
->transid
;
1456 * create reloc tree for a given fs tree. reloc tree is just a
1457 * snapshot of the fs tree with special root objectid.
1459 int btrfs_init_reloc_root(struct btrfs_trans_handle
*trans
,
1460 struct btrfs_root
*root
)
1462 struct btrfs_fs_info
*fs_info
= root
->fs_info
;
1463 struct btrfs_root
*reloc_root
;
1464 struct reloc_control
*rc
= fs_info
->reloc_ctl
;
1465 struct btrfs_block_rsv
*rsv
;
1469 if (root
->reloc_root
) {
1470 reloc_root
= root
->reloc_root
;
1471 reloc_root
->last_trans
= trans
->transid
;
1475 if (!rc
|| !rc
->create_reloc_tree
||
1476 root
->root_key
.objectid
== BTRFS_TREE_RELOC_OBJECTID
)
1479 if (!trans
->reloc_reserved
) {
1480 rsv
= trans
->block_rsv
;
1481 trans
->block_rsv
= rc
->block_rsv
;
1484 reloc_root
= create_reloc_root(trans
, root
, root
->root_key
.objectid
);
1486 trans
->block_rsv
= rsv
;
1488 ret
= __add_reloc_root(reloc_root
);
1490 root
->reloc_root
= reloc_root
;
1495 * update root item of reloc tree
1497 int btrfs_update_reloc_root(struct btrfs_trans_handle
*trans
,
1498 struct btrfs_root
*root
)
1500 struct btrfs_fs_info
*fs_info
= root
->fs_info
;
1501 struct btrfs_root
*reloc_root
;
1502 struct btrfs_root_item
*root_item
;
1505 if (!root
->reloc_root
)
1508 reloc_root
= root
->reloc_root
;
1509 root_item
= &reloc_root
->root_item
;
1511 if (fs_info
->reloc_ctl
->merge_reloc_tree
&&
1512 btrfs_root_refs(root_item
) == 0) {
1513 root
->reloc_root
= NULL
;
1514 __del_reloc_root(reloc_root
);
1517 if (reloc_root
->commit_root
!= reloc_root
->node
) {
1518 btrfs_set_root_node(root_item
, reloc_root
->node
);
1519 free_extent_buffer(reloc_root
->commit_root
);
1520 reloc_root
->commit_root
= btrfs_root_node(reloc_root
);
1523 ret
= btrfs_update_root(trans
, fs_info
->tree_root
,
1524 &reloc_root
->root_key
, root_item
);
1532 * helper to find first cached inode with inode number >= objectid
1535 static struct inode
*find_next_inode(struct btrfs_root
*root
, u64 objectid
)
1537 struct rb_node
*node
;
1538 struct rb_node
*prev
;
1539 struct btrfs_inode
*entry
;
1540 struct inode
*inode
;
1542 spin_lock(&root
->inode_lock
);
1544 node
= root
->inode_tree
.rb_node
;
1548 entry
= rb_entry(node
, struct btrfs_inode
, rb_node
);
1550 if (objectid
< btrfs_ino(&entry
->vfs_inode
))
1551 node
= node
->rb_left
;
1552 else if (objectid
> btrfs_ino(&entry
->vfs_inode
))
1553 node
= node
->rb_right
;
1559 entry
= rb_entry(prev
, struct btrfs_inode
, rb_node
);
1560 if (objectid
<= btrfs_ino(&entry
->vfs_inode
)) {
1564 prev
= rb_next(prev
);
1568 entry
= rb_entry(node
, struct btrfs_inode
, rb_node
);
1569 inode
= igrab(&entry
->vfs_inode
);
1571 spin_unlock(&root
->inode_lock
);
1575 objectid
= btrfs_ino(&entry
->vfs_inode
) + 1;
1576 if (cond_resched_lock(&root
->inode_lock
))
1579 node
= rb_next(node
);
1581 spin_unlock(&root
->inode_lock
);
1585 static int in_block_group(u64 bytenr
,
1586 struct btrfs_block_group_cache
*block_group
)
1588 if (bytenr
>= block_group
->key
.objectid
&&
1589 bytenr
< block_group
->key
.objectid
+ block_group
->key
.offset
)
1595 * get new location of data
1597 static int get_new_location(struct inode
*reloc_inode
, u64
*new_bytenr
,
1598 u64 bytenr
, u64 num_bytes
)
1600 struct btrfs_root
*root
= BTRFS_I(reloc_inode
)->root
;
1601 struct btrfs_path
*path
;
1602 struct btrfs_file_extent_item
*fi
;
1603 struct extent_buffer
*leaf
;
1606 path
= btrfs_alloc_path();
1610 bytenr
-= BTRFS_I(reloc_inode
)->index_cnt
;
1611 ret
= btrfs_lookup_file_extent(NULL
, root
, path
, btrfs_ino(reloc_inode
),
1620 leaf
= path
->nodes
[0];
1621 fi
= btrfs_item_ptr(leaf
, path
->slots
[0],
1622 struct btrfs_file_extent_item
);
1624 BUG_ON(btrfs_file_extent_offset(leaf
, fi
) ||
1625 btrfs_file_extent_compression(leaf
, fi
) ||
1626 btrfs_file_extent_encryption(leaf
, fi
) ||
1627 btrfs_file_extent_other_encoding(leaf
, fi
));
1629 if (num_bytes
!= btrfs_file_extent_disk_num_bytes(leaf
, fi
)) {
1634 *new_bytenr
= btrfs_file_extent_disk_bytenr(leaf
, fi
);
1637 btrfs_free_path(path
);
1642 * update file extent items in the tree leaf to point to
1643 * the new locations.
1645 static noinline_for_stack
1646 int replace_file_extents(struct btrfs_trans_handle
*trans
,
1647 struct reloc_control
*rc
,
1648 struct btrfs_root
*root
,
1649 struct extent_buffer
*leaf
)
1651 struct btrfs_fs_info
*fs_info
= root
->fs_info
;
1652 struct btrfs_key key
;
1653 struct btrfs_file_extent_item
*fi
;
1654 struct inode
*inode
= NULL
;
1666 if (rc
->stage
!= UPDATE_DATA_PTRS
)
1669 /* reloc trees always use full backref */
1670 if (root
->root_key
.objectid
== BTRFS_TREE_RELOC_OBJECTID
)
1671 parent
= leaf
->start
;
1675 nritems
= btrfs_header_nritems(leaf
);
1676 for (i
= 0; i
< nritems
; i
++) {
1678 btrfs_item_key_to_cpu(leaf
, &key
, i
);
1679 if (key
.type
!= BTRFS_EXTENT_DATA_KEY
)
1681 fi
= btrfs_item_ptr(leaf
, i
, struct btrfs_file_extent_item
);
1682 if (btrfs_file_extent_type(leaf
, fi
) ==
1683 BTRFS_FILE_EXTENT_INLINE
)
1685 bytenr
= btrfs_file_extent_disk_bytenr(leaf
, fi
);
1686 num_bytes
= btrfs_file_extent_disk_num_bytes(leaf
, fi
);
1689 if (!in_block_group(bytenr
, rc
->block_group
))
1693 * if we are modifying block in fs tree, wait for readpage
1694 * to complete and drop the extent cache
1696 if (root
->root_key
.objectid
!= BTRFS_TREE_RELOC_OBJECTID
) {
1698 inode
= find_next_inode(root
, key
.objectid
);
1700 } else if (inode
&& btrfs_ino(inode
) < key
.objectid
) {
1701 btrfs_add_delayed_iput(inode
);
1702 inode
= find_next_inode(root
, key
.objectid
);
1704 if (inode
&& btrfs_ino(inode
) == key
.objectid
) {
1706 btrfs_file_extent_num_bytes(leaf
, fi
);
1707 WARN_ON(!IS_ALIGNED(key
.offset
,
1708 fs_info
->sectorsize
));
1709 WARN_ON(!IS_ALIGNED(end
, fs_info
->sectorsize
));
1711 ret
= try_lock_extent(&BTRFS_I(inode
)->io_tree
,
1716 btrfs_drop_extent_cache(inode
, key
.offset
, end
,
1718 unlock_extent(&BTRFS_I(inode
)->io_tree
,
1723 ret
= get_new_location(rc
->data_inode
, &new_bytenr
,
1727 * Don't have to abort since we've not changed anything
1728 * in the file extent yet.
1733 btrfs_set_file_extent_disk_bytenr(leaf
, fi
, new_bytenr
);
1736 key
.offset
-= btrfs_file_extent_offset(leaf
, fi
);
1737 ret
= btrfs_inc_extent_ref(trans
, root
, new_bytenr
,
1739 btrfs_header_owner(leaf
),
1740 key
.objectid
, key
.offset
);
1742 btrfs_abort_transaction(trans
, ret
);
1746 ret
= btrfs_free_extent(trans
, root
, bytenr
, num_bytes
,
1747 parent
, btrfs_header_owner(leaf
),
1748 key
.objectid
, key
.offset
);
1750 btrfs_abort_transaction(trans
, ret
);
1755 btrfs_mark_buffer_dirty(leaf
);
1757 btrfs_add_delayed_iput(inode
);
1761 static noinline_for_stack
1762 int memcmp_node_keys(struct extent_buffer
*eb
, int slot
,
1763 struct btrfs_path
*path
, int level
)
1765 struct btrfs_disk_key key1
;
1766 struct btrfs_disk_key key2
;
1767 btrfs_node_key(eb
, &key1
, slot
);
1768 btrfs_node_key(path
->nodes
[level
], &key2
, path
->slots
[level
]);
1769 return memcmp(&key1
, &key2
, sizeof(key1
));
1773 * try to replace tree blocks in fs tree with the new blocks
1774 * in reloc tree. tree blocks haven't been modified since the
1775 * reloc tree was create can be replaced.
1777 * if a block was replaced, level of the block + 1 is returned.
1778 * if no block got replaced, 0 is returned. if there are other
1779 * errors, a negative error number is returned.
1781 static noinline_for_stack
1782 int replace_path(struct btrfs_trans_handle
*trans
,
1783 struct btrfs_root
*dest
, struct btrfs_root
*src
,
1784 struct btrfs_path
*path
, struct btrfs_key
*next_key
,
1785 int lowest_level
, int max_level
)
1787 struct btrfs_fs_info
*fs_info
= dest
->fs_info
;
1788 struct extent_buffer
*eb
;
1789 struct extent_buffer
*parent
;
1790 struct btrfs_key key
;
1802 BUG_ON(src
->root_key
.objectid
!= BTRFS_TREE_RELOC_OBJECTID
);
1803 BUG_ON(dest
->root_key
.objectid
== BTRFS_TREE_RELOC_OBJECTID
);
1805 last_snapshot
= btrfs_root_last_snapshot(&src
->root_item
);
1807 slot
= path
->slots
[lowest_level
];
1808 btrfs_node_key_to_cpu(path
->nodes
[lowest_level
], &key
, slot
);
1810 eb
= btrfs_lock_root_node(dest
);
1811 btrfs_set_lock_blocking(eb
);
1812 level
= btrfs_header_level(eb
);
1814 if (level
< lowest_level
) {
1815 btrfs_tree_unlock(eb
);
1816 free_extent_buffer(eb
);
1821 ret
= btrfs_cow_block(trans
, dest
, eb
, NULL
, 0, &eb
);
1824 btrfs_set_lock_blocking(eb
);
1827 next_key
->objectid
= (u64
)-1;
1828 next_key
->type
= (u8
)-1;
1829 next_key
->offset
= (u64
)-1;
1834 level
= btrfs_header_level(parent
);
1835 BUG_ON(level
< lowest_level
);
1837 ret
= btrfs_bin_search(parent
, &key
, level
, &slot
);
1838 if (ret
&& slot
> 0)
1841 if (next_key
&& slot
+ 1 < btrfs_header_nritems(parent
))
1842 btrfs_node_key_to_cpu(parent
, next_key
, slot
+ 1);
1844 old_bytenr
= btrfs_node_blockptr(parent
, slot
);
1845 blocksize
= fs_info
->nodesize
;
1846 old_ptr_gen
= btrfs_node_ptr_generation(parent
, slot
);
1848 if (level
<= max_level
) {
1849 eb
= path
->nodes
[level
];
1850 new_bytenr
= btrfs_node_blockptr(eb
,
1851 path
->slots
[level
]);
1852 new_ptr_gen
= btrfs_node_ptr_generation(eb
,
1853 path
->slots
[level
]);
1859 if (WARN_ON(new_bytenr
> 0 && new_bytenr
== old_bytenr
)) {
1864 if (new_bytenr
== 0 || old_ptr_gen
> last_snapshot
||
1865 memcmp_node_keys(parent
, slot
, path
, level
)) {
1866 if (level
<= lowest_level
) {
1871 eb
= read_tree_block(dest
, old_bytenr
, old_ptr_gen
);
1875 } else if (!extent_buffer_uptodate(eb
)) {
1877 free_extent_buffer(eb
);
1880 btrfs_tree_lock(eb
);
1882 ret
= btrfs_cow_block(trans
, dest
, eb
, parent
,
1886 btrfs_set_lock_blocking(eb
);
1888 btrfs_tree_unlock(parent
);
1889 free_extent_buffer(parent
);
1896 btrfs_tree_unlock(parent
);
1897 free_extent_buffer(parent
);
1902 btrfs_node_key_to_cpu(path
->nodes
[level
], &key
,
1903 path
->slots
[level
]);
1904 btrfs_release_path(path
);
1906 path
->lowest_level
= level
;
1907 ret
= btrfs_search_slot(trans
, src
, &key
, path
, 0, 1);
1908 path
->lowest_level
= 0;
1912 * Info qgroup to trace both subtrees.
1914 * We must trace both trees.
1915 * 1) Tree reloc subtree
1916 * If not traced, we will leak data numbers
1918 * If not traced, we will double count old data
1919 * and tree block numbers, if current trans doesn't free
1920 * data reloc tree inode.
1922 ret
= btrfs_qgroup_trace_subtree(trans
, src
, parent
,
1923 btrfs_header_generation(parent
),
1924 btrfs_header_level(parent
));
1927 ret
= btrfs_qgroup_trace_subtree(trans
, dest
,
1929 btrfs_header_generation(path
->nodes
[level
]),
1930 btrfs_header_level(path
->nodes
[level
]));
1935 * swap blocks in fs tree and reloc tree.
1937 btrfs_set_node_blockptr(parent
, slot
, new_bytenr
);
1938 btrfs_set_node_ptr_generation(parent
, slot
, new_ptr_gen
);
1939 btrfs_mark_buffer_dirty(parent
);
1941 btrfs_set_node_blockptr(path
->nodes
[level
],
1942 path
->slots
[level
], old_bytenr
);
1943 btrfs_set_node_ptr_generation(path
->nodes
[level
],
1944 path
->slots
[level
], old_ptr_gen
);
1945 btrfs_mark_buffer_dirty(path
->nodes
[level
]);
1947 ret
= btrfs_inc_extent_ref(trans
, src
, old_bytenr
, blocksize
,
1948 path
->nodes
[level
]->start
,
1949 src
->root_key
.objectid
, level
- 1, 0);
1951 ret
= btrfs_inc_extent_ref(trans
, dest
, new_bytenr
, blocksize
,
1952 0, dest
->root_key
.objectid
, level
- 1,
1956 ret
= btrfs_free_extent(trans
, src
, new_bytenr
, blocksize
,
1957 path
->nodes
[level
]->start
,
1958 src
->root_key
.objectid
, level
- 1, 0);
1961 ret
= btrfs_free_extent(trans
, dest
, old_bytenr
, blocksize
,
1962 0, dest
->root_key
.objectid
, level
- 1,
1966 btrfs_unlock_up_safe(path
, 0);
1971 btrfs_tree_unlock(parent
);
1972 free_extent_buffer(parent
);
1977 * helper to find next relocated block in reloc tree
1979 static noinline_for_stack
1980 int walk_up_reloc_tree(struct btrfs_root
*root
, struct btrfs_path
*path
,
1983 struct extent_buffer
*eb
;
1988 last_snapshot
= btrfs_root_last_snapshot(&root
->root_item
);
1990 for (i
= 0; i
< *level
; i
++) {
1991 free_extent_buffer(path
->nodes
[i
]);
1992 path
->nodes
[i
] = NULL
;
1995 for (i
= *level
; i
< BTRFS_MAX_LEVEL
&& path
->nodes
[i
]; i
++) {
1996 eb
= path
->nodes
[i
];
1997 nritems
= btrfs_header_nritems(eb
);
1998 while (path
->slots
[i
] + 1 < nritems
) {
2000 if (btrfs_node_ptr_generation(eb
, path
->slots
[i
]) <=
2007 free_extent_buffer(path
->nodes
[i
]);
2008 path
->nodes
[i
] = NULL
;
2014 * walk down reloc tree to find relocated block of lowest level
2016 static noinline_for_stack
2017 int walk_down_reloc_tree(struct btrfs_root
*root
, struct btrfs_path
*path
,
2020 struct extent_buffer
*eb
= NULL
;
2027 last_snapshot
= btrfs_root_last_snapshot(&root
->root_item
);
2029 for (i
= *level
; i
> 0; i
--) {
2030 eb
= path
->nodes
[i
];
2031 nritems
= btrfs_header_nritems(eb
);
2032 while (path
->slots
[i
] < nritems
) {
2033 ptr_gen
= btrfs_node_ptr_generation(eb
, path
->slots
[i
]);
2034 if (ptr_gen
> last_snapshot
)
2038 if (path
->slots
[i
] >= nritems
) {
2049 bytenr
= btrfs_node_blockptr(eb
, path
->slots
[i
]);
2050 eb
= read_tree_block(root
, bytenr
, ptr_gen
);
2053 } else if (!extent_buffer_uptodate(eb
)) {
2054 free_extent_buffer(eb
);
2057 BUG_ON(btrfs_header_level(eb
) != i
- 1);
2058 path
->nodes
[i
- 1] = eb
;
2059 path
->slots
[i
- 1] = 0;
2065 * invalidate extent cache for file extents whose key in range of
2066 * [min_key, max_key)
2068 static int invalidate_extent_cache(struct btrfs_root
*root
,
2069 struct btrfs_key
*min_key
,
2070 struct btrfs_key
*max_key
)
2072 struct btrfs_fs_info
*fs_info
= root
->fs_info
;
2073 struct inode
*inode
= NULL
;
2078 objectid
= min_key
->objectid
;
2083 if (objectid
> max_key
->objectid
)
2086 inode
= find_next_inode(root
, objectid
);
2089 ino
= btrfs_ino(inode
);
2091 if (ino
> max_key
->objectid
) {
2097 if (!S_ISREG(inode
->i_mode
))
2100 if (unlikely(min_key
->objectid
== ino
)) {
2101 if (min_key
->type
> BTRFS_EXTENT_DATA_KEY
)
2103 if (min_key
->type
< BTRFS_EXTENT_DATA_KEY
)
2106 start
= min_key
->offset
;
2107 WARN_ON(!IS_ALIGNED(start
, fs_info
->sectorsize
));
2113 if (unlikely(max_key
->objectid
== ino
)) {
2114 if (max_key
->type
< BTRFS_EXTENT_DATA_KEY
)
2116 if (max_key
->type
> BTRFS_EXTENT_DATA_KEY
) {
2119 if (max_key
->offset
== 0)
2121 end
= max_key
->offset
;
2122 WARN_ON(!IS_ALIGNED(end
, fs_info
->sectorsize
));
2129 /* the lock_extent waits for readpage to complete */
2130 lock_extent(&BTRFS_I(inode
)->io_tree
, start
, end
);
2131 btrfs_drop_extent_cache(inode
, start
, end
, 1);
2132 unlock_extent(&BTRFS_I(inode
)->io_tree
, start
, end
);
2137 static int find_next_key(struct btrfs_path
*path
, int level
,
2138 struct btrfs_key
*key
)
2141 while (level
< BTRFS_MAX_LEVEL
) {
2142 if (!path
->nodes
[level
])
2144 if (path
->slots
[level
] + 1 <
2145 btrfs_header_nritems(path
->nodes
[level
])) {
2146 btrfs_node_key_to_cpu(path
->nodes
[level
], key
,
2147 path
->slots
[level
] + 1);
2156 * merge the relocated tree blocks in reloc tree with corresponding
2159 static noinline_for_stack
int merge_reloc_root(struct reloc_control
*rc
,
2160 struct btrfs_root
*root
)
2162 struct btrfs_fs_info
*fs_info
= rc
->extent_root
->fs_info
;
2163 LIST_HEAD(inode_list
);
2164 struct btrfs_key key
;
2165 struct btrfs_key next_key
;
2166 struct btrfs_trans_handle
*trans
= NULL
;
2167 struct btrfs_root
*reloc_root
;
2168 struct btrfs_root_item
*root_item
;
2169 struct btrfs_path
*path
;
2170 struct extent_buffer
*leaf
;
2178 path
= btrfs_alloc_path();
2181 path
->reada
= READA_FORWARD
;
2183 reloc_root
= root
->reloc_root
;
2184 root_item
= &reloc_root
->root_item
;
2186 if (btrfs_disk_key_objectid(&root_item
->drop_progress
) == 0) {
2187 level
= btrfs_root_level(root_item
);
2188 extent_buffer_get(reloc_root
->node
);
2189 path
->nodes
[level
] = reloc_root
->node
;
2190 path
->slots
[level
] = 0;
2192 btrfs_disk_key_to_cpu(&key
, &root_item
->drop_progress
);
2194 level
= root_item
->drop_level
;
2196 path
->lowest_level
= level
;
2197 ret
= btrfs_search_slot(NULL
, reloc_root
, &key
, path
, 0, 0);
2198 path
->lowest_level
= 0;
2200 btrfs_free_path(path
);
2204 btrfs_node_key_to_cpu(path
->nodes
[level
], &next_key
,
2205 path
->slots
[level
]);
2206 WARN_ON(memcmp(&key
, &next_key
, sizeof(key
)));
2208 btrfs_unlock_up_safe(path
, 0);
2211 min_reserved
= fs_info
->nodesize
* (BTRFS_MAX_LEVEL
- 1) * 2;
2212 memset(&next_key
, 0, sizeof(next_key
));
2215 ret
= btrfs_block_rsv_refill(root
, rc
->block_rsv
, min_reserved
,
2216 BTRFS_RESERVE_FLUSH_ALL
);
2221 trans
= btrfs_start_transaction(root
, 0);
2222 if (IS_ERR(trans
)) {
2223 err
= PTR_ERR(trans
);
2227 trans
->block_rsv
= rc
->block_rsv
;
2232 ret
= walk_down_reloc_tree(reloc_root
, path
, &level
);
2240 if (!find_next_key(path
, level
, &key
) &&
2241 btrfs_comp_cpu_keys(&next_key
, &key
) >= 0) {
2244 ret
= replace_path(trans
, root
, reloc_root
, path
,
2245 &next_key
, level
, max_level
);
2254 btrfs_node_key_to_cpu(path
->nodes
[level
], &key
,
2255 path
->slots
[level
]);
2259 ret
= walk_up_reloc_tree(reloc_root
, path
, &level
);
2265 * save the merging progress in the drop_progress.
2266 * this is OK since root refs == 1 in this case.
2268 btrfs_node_key(path
->nodes
[level
], &root_item
->drop_progress
,
2269 path
->slots
[level
]);
2270 root_item
->drop_level
= level
;
2272 btrfs_end_transaction_throttle(trans
, root
);
2275 btrfs_btree_balance_dirty(root
);
2277 if (replaced
&& rc
->stage
== UPDATE_DATA_PTRS
)
2278 invalidate_extent_cache(root
, &key
, &next_key
);
2282 * handle the case only one block in the fs tree need to be
2283 * relocated and the block is tree root.
2285 leaf
= btrfs_lock_root_node(root
);
2286 ret
= btrfs_cow_block(trans
, root
, leaf
, NULL
, 0, &leaf
);
2287 btrfs_tree_unlock(leaf
);
2288 free_extent_buffer(leaf
);
2292 btrfs_free_path(path
);
2295 memset(&root_item
->drop_progress
, 0,
2296 sizeof(root_item
->drop_progress
));
2297 root_item
->drop_level
= 0;
2298 btrfs_set_root_refs(root_item
, 0);
2299 btrfs_update_reloc_root(trans
, root
);
2303 btrfs_end_transaction_throttle(trans
, root
);
2305 btrfs_btree_balance_dirty(root
);
2307 if (replaced
&& rc
->stage
== UPDATE_DATA_PTRS
)
2308 invalidate_extent_cache(root
, &key
, &next_key
);
2313 static noinline_for_stack
2314 int prepare_to_merge(struct reloc_control
*rc
, int err
)
2316 struct btrfs_root
*root
= rc
->extent_root
;
2317 struct btrfs_fs_info
*fs_info
= root
->fs_info
;
2318 struct btrfs_root
*reloc_root
;
2319 struct btrfs_trans_handle
*trans
;
2320 LIST_HEAD(reloc_roots
);
2324 mutex_lock(&fs_info
->reloc_mutex
);
2325 rc
->merging_rsv_size
+= fs_info
->nodesize
* (BTRFS_MAX_LEVEL
- 1) * 2;
2326 rc
->merging_rsv_size
+= rc
->nodes_relocated
* 2;
2327 mutex_unlock(&fs_info
->reloc_mutex
);
2331 num_bytes
= rc
->merging_rsv_size
;
2332 ret
= btrfs_block_rsv_add(root
, rc
->block_rsv
, num_bytes
,
2333 BTRFS_RESERVE_FLUSH_ALL
);
2338 trans
= btrfs_join_transaction(rc
->extent_root
);
2339 if (IS_ERR(trans
)) {
2341 btrfs_block_rsv_release(rc
->extent_root
,
2342 rc
->block_rsv
, num_bytes
);
2343 return PTR_ERR(trans
);
2347 if (num_bytes
!= rc
->merging_rsv_size
) {
2348 btrfs_end_transaction(trans
, rc
->extent_root
);
2349 btrfs_block_rsv_release(rc
->extent_root
,
2350 rc
->block_rsv
, num_bytes
);
2355 rc
->merge_reloc_tree
= 1;
2357 while (!list_empty(&rc
->reloc_roots
)) {
2358 reloc_root
= list_entry(rc
->reloc_roots
.next
,
2359 struct btrfs_root
, root_list
);
2360 list_del_init(&reloc_root
->root_list
);
2362 root
= read_fs_root(fs_info
, reloc_root
->root_key
.offset
);
2363 BUG_ON(IS_ERR(root
));
2364 BUG_ON(root
->reloc_root
!= reloc_root
);
2367 * set reference count to 1, so btrfs_recover_relocation
2368 * knows it should resumes merging
2371 btrfs_set_root_refs(&reloc_root
->root_item
, 1);
2372 btrfs_update_reloc_root(trans
, root
);
2374 list_add(&reloc_root
->root_list
, &reloc_roots
);
2377 list_splice(&reloc_roots
, &rc
->reloc_roots
);
2380 btrfs_commit_transaction(trans
, rc
->extent_root
);
2382 btrfs_end_transaction(trans
, rc
->extent_root
);
2386 static noinline_for_stack
2387 void free_reloc_roots(struct list_head
*list
)
2389 struct btrfs_root
*reloc_root
;
2391 while (!list_empty(list
)) {
2392 reloc_root
= list_entry(list
->next
, struct btrfs_root
,
2394 free_extent_buffer(reloc_root
->node
);
2395 free_extent_buffer(reloc_root
->commit_root
);
2396 reloc_root
->node
= NULL
;
2397 reloc_root
->commit_root
= NULL
;
2398 __del_reloc_root(reloc_root
);
2402 static noinline_for_stack
2403 void merge_reloc_roots(struct reloc_control
*rc
)
2405 struct btrfs_fs_info
*fs_info
= rc
->extent_root
->fs_info
;
2406 struct btrfs_root
*root
;
2407 struct btrfs_root
*reloc_root
;
2411 LIST_HEAD(reloc_roots
);
2415 root
= rc
->extent_root
;
2418 * this serializes us with btrfs_record_root_in_transaction,
2419 * we have to make sure nobody is in the middle of
2420 * adding their roots to the list while we are
2423 mutex_lock(&fs_info
->reloc_mutex
);
2424 list_splice_init(&rc
->reloc_roots
, &reloc_roots
);
2425 mutex_unlock(&fs_info
->reloc_mutex
);
2427 while (!list_empty(&reloc_roots
)) {
2429 reloc_root
= list_entry(reloc_roots
.next
,
2430 struct btrfs_root
, root_list
);
2432 if (btrfs_root_refs(&reloc_root
->root_item
) > 0) {
2433 root
= read_fs_root(fs_info
,
2434 reloc_root
->root_key
.offset
);
2435 BUG_ON(IS_ERR(root
));
2436 BUG_ON(root
->reloc_root
!= reloc_root
);
2438 ret
= merge_reloc_root(rc
, root
);
2440 if (list_empty(&reloc_root
->root_list
))
2441 list_add_tail(&reloc_root
->root_list
,
2446 list_del_init(&reloc_root
->root_list
);
2450 * we keep the old last snapshot transid in rtranid when we
2451 * created the relocation tree.
2453 last_snap
= btrfs_root_rtransid(&reloc_root
->root_item
);
2454 otransid
= btrfs_root_otransid(&reloc_root
->root_item
);
2455 objectid
= reloc_root
->root_key
.offset
;
2457 ret
= btrfs_drop_snapshot(reloc_root
, rc
->block_rsv
, 0, 1);
2459 if (list_empty(&reloc_root
->root_list
))
2460 list_add_tail(&reloc_root
->root_list
,
2472 btrfs_handle_fs_error(fs_info
, ret
, NULL
);
2473 if (!list_empty(&reloc_roots
))
2474 free_reloc_roots(&reloc_roots
);
2476 /* new reloc root may be added */
2477 mutex_lock(&fs_info
->reloc_mutex
);
2478 list_splice_init(&rc
->reloc_roots
, &reloc_roots
);
2479 mutex_unlock(&fs_info
->reloc_mutex
);
2480 if (!list_empty(&reloc_roots
))
2481 free_reloc_roots(&reloc_roots
);
2484 BUG_ON(!RB_EMPTY_ROOT(&rc
->reloc_root_tree
.rb_root
));
2487 static void free_block_list(struct rb_root
*blocks
)
2489 struct tree_block
*block
;
2490 struct rb_node
*rb_node
;
2491 while ((rb_node
= rb_first(blocks
))) {
2492 block
= rb_entry(rb_node
, struct tree_block
, rb_node
);
2493 rb_erase(rb_node
, blocks
);
2498 static int record_reloc_root_in_trans(struct btrfs_trans_handle
*trans
,
2499 struct btrfs_root
*reloc_root
)
2501 struct btrfs_fs_info
*fs_info
= reloc_root
->fs_info
;
2502 struct btrfs_root
*root
;
2504 if (reloc_root
->last_trans
== trans
->transid
)
2507 root
= read_fs_root(fs_info
, reloc_root
->root_key
.offset
);
2508 BUG_ON(IS_ERR(root
));
2509 BUG_ON(root
->reloc_root
!= reloc_root
);
2511 return btrfs_record_root_in_trans(trans
, root
);
2514 static noinline_for_stack
2515 struct btrfs_root
*select_reloc_root(struct btrfs_trans_handle
*trans
,
2516 struct reloc_control
*rc
,
2517 struct backref_node
*node
,
2518 struct backref_edge
*edges
[])
2520 struct backref_node
*next
;
2521 struct btrfs_root
*root
;
2527 next
= walk_up_backref(next
, edges
, &index
);
2530 BUG_ON(!test_bit(BTRFS_ROOT_REF_COWS
, &root
->state
));
2532 if (root
->root_key
.objectid
== BTRFS_TREE_RELOC_OBJECTID
) {
2533 record_reloc_root_in_trans(trans
, root
);
2537 btrfs_record_root_in_trans(trans
, root
);
2538 root
= root
->reloc_root
;
2540 if (next
->new_bytenr
!= root
->node
->start
) {
2541 BUG_ON(next
->new_bytenr
);
2542 BUG_ON(!list_empty(&next
->list
));
2543 next
->new_bytenr
= root
->node
->start
;
2545 list_add_tail(&next
->list
,
2546 &rc
->backref_cache
.changed
);
2547 __mark_block_processed(rc
, next
);
2553 next
= walk_down_backref(edges
, &index
);
2554 if (!next
|| next
->level
<= node
->level
)
2561 /* setup backref node path for btrfs_reloc_cow_block */
2563 rc
->backref_cache
.path
[next
->level
] = next
;
2566 next
= edges
[index
]->node
[UPPER
];
2572 * select a tree root for relocation. return NULL if the block
2573 * is reference counted. we should use do_relocation() in this
2574 * case. return a tree root pointer if the block isn't reference
2575 * counted. return -ENOENT if the block is root of reloc tree.
2577 static noinline_for_stack
2578 struct btrfs_root
*select_one_root(struct backref_node
*node
)
2580 struct backref_node
*next
;
2581 struct btrfs_root
*root
;
2582 struct btrfs_root
*fs_root
= NULL
;
2583 struct backref_edge
*edges
[BTRFS_MAX_LEVEL
- 1];
2589 next
= walk_up_backref(next
, edges
, &index
);
2593 /* no other choice for non-references counted tree */
2594 if (!test_bit(BTRFS_ROOT_REF_COWS
, &root
->state
))
2597 if (root
->root_key
.objectid
!= BTRFS_TREE_RELOC_OBJECTID
)
2603 next
= walk_down_backref(edges
, &index
);
2604 if (!next
|| next
->level
<= node
->level
)
2609 return ERR_PTR(-ENOENT
);
2613 static noinline_for_stack
2614 u64
calcu_metadata_size(struct reloc_control
*rc
,
2615 struct backref_node
*node
, int reserve
)
2617 struct btrfs_fs_info
*fs_info
= rc
->extent_root
->fs_info
;
2618 struct backref_node
*next
= node
;
2619 struct backref_edge
*edge
;
2620 struct backref_edge
*edges
[BTRFS_MAX_LEVEL
- 1];
2624 BUG_ON(reserve
&& node
->processed
);
2629 if (next
->processed
&& (reserve
|| next
!= node
))
2632 num_bytes
+= fs_info
->nodesize
;
2634 if (list_empty(&next
->upper
))
2637 edge
= list_entry(next
->upper
.next
,
2638 struct backref_edge
, list
[LOWER
]);
2639 edges
[index
++] = edge
;
2640 next
= edge
->node
[UPPER
];
2642 next
= walk_down_backref(edges
, &index
);
2647 static int reserve_metadata_space(struct btrfs_trans_handle
*trans
,
2648 struct reloc_control
*rc
,
2649 struct backref_node
*node
)
2651 struct btrfs_root
*root
= rc
->extent_root
;
2652 struct btrfs_fs_info
*fs_info
= root
->fs_info
;
2657 num_bytes
= calcu_metadata_size(rc
, node
, 1) * 2;
2659 trans
->block_rsv
= rc
->block_rsv
;
2660 rc
->reserved_bytes
+= num_bytes
;
2663 * We are under a transaction here so we can only do limited flushing.
2664 * If we get an enospc just kick back -EAGAIN so we know to drop the
2665 * transaction and try to refill when we can flush all the things.
2667 ret
= btrfs_block_rsv_refill(root
, rc
->block_rsv
, num_bytes
,
2668 BTRFS_RESERVE_FLUSH_LIMIT
);
2670 tmp
= fs_info
->nodesize
* RELOCATION_RESERVED_NODES
;
2671 while (tmp
<= rc
->reserved_bytes
)
2674 * only one thread can access block_rsv at this point,
2675 * so we don't need hold lock to protect block_rsv.
2676 * we expand more reservation size here to allow enough
2677 * space for relocation and we will return eailer in
2680 rc
->block_rsv
->size
= tmp
+ fs_info
->nodesize
*
2681 RELOCATION_RESERVED_NODES
;
2689 * relocate a block tree, and then update pointers in upper level
2690 * blocks that reference the block to point to the new location.
2692 * if called by link_to_upper, the block has already been relocated.
2693 * in that case this function just updates pointers.
2695 static int do_relocation(struct btrfs_trans_handle
*trans
,
2696 struct reloc_control
*rc
,
2697 struct backref_node
*node
,
2698 struct btrfs_key
*key
,
2699 struct btrfs_path
*path
, int lowest
)
2701 struct backref_node
*upper
;
2702 struct backref_edge
*edge
;
2703 struct backref_edge
*edges
[BTRFS_MAX_LEVEL
- 1];
2704 struct btrfs_root
*root
;
2705 struct extent_buffer
*eb
;
2713 BUG_ON(lowest
&& node
->eb
);
2715 path
->lowest_level
= node
->level
+ 1;
2716 rc
->backref_cache
.path
[node
->level
] = node
;
2717 list_for_each_entry(edge
, &node
->upper
, list
[LOWER
]) {
2720 upper
= edge
->node
[UPPER
];
2721 root
= select_reloc_root(trans
, rc
, upper
, edges
);
2724 if (upper
->eb
&& !upper
->locked
) {
2726 ret
= btrfs_bin_search(upper
->eb
, key
,
2727 upper
->level
, &slot
);
2729 bytenr
= btrfs_node_blockptr(upper
->eb
, slot
);
2730 if (node
->eb
->start
== bytenr
)
2733 drop_node_buffer(upper
);
2737 ret
= btrfs_search_slot(trans
, root
, key
, path
, 0, 1);
2744 btrfs_release_path(path
);
2749 upper
->eb
= path
->nodes
[upper
->level
];
2750 path
->nodes
[upper
->level
] = NULL
;
2752 BUG_ON(upper
->eb
!= path
->nodes
[upper
->level
]);
2756 path
->locks
[upper
->level
] = 0;
2758 slot
= path
->slots
[upper
->level
];
2759 btrfs_release_path(path
);
2761 ret
= btrfs_bin_search(upper
->eb
, key
, upper
->level
,
2766 bytenr
= btrfs_node_blockptr(upper
->eb
, slot
);
2768 if (bytenr
!= node
->bytenr
) {
2769 btrfs_err(root
->fs_info
,
2770 "lowest leaf/node mismatch: bytenr %llu node->bytenr %llu slot %d upper %llu",
2771 bytenr
, node
->bytenr
, slot
,
2777 if (node
->eb
->start
== bytenr
)
2781 blocksize
= root
->fs_info
->nodesize
;
2782 generation
= btrfs_node_ptr_generation(upper
->eb
, slot
);
2783 eb
= read_tree_block(root
, bytenr
, generation
);
2787 } else if (!extent_buffer_uptodate(eb
)) {
2788 free_extent_buffer(eb
);
2792 btrfs_tree_lock(eb
);
2793 btrfs_set_lock_blocking(eb
);
2796 ret
= btrfs_cow_block(trans
, root
, eb
, upper
->eb
,
2798 btrfs_tree_unlock(eb
);
2799 free_extent_buffer(eb
);
2804 BUG_ON(node
->eb
!= eb
);
2806 btrfs_set_node_blockptr(upper
->eb
, slot
,
2808 btrfs_set_node_ptr_generation(upper
->eb
, slot
,
2810 btrfs_mark_buffer_dirty(upper
->eb
);
2812 ret
= btrfs_inc_extent_ref(trans
, root
,
2813 node
->eb
->start
, blocksize
,
2815 btrfs_header_owner(upper
->eb
),
2819 ret
= btrfs_drop_subtree(trans
, root
, eb
, upper
->eb
);
2823 if (!upper
->pending
)
2824 drop_node_buffer(upper
);
2826 unlock_node_buffer(upper
);
2831 if (!err
&& node
->pending
) {
2832 drop_node_buffer(node
);
2833 list_move_tail(&node
->list
, &rc
->backref_cache
.changed
);
2837 path
->lowest_level
= 0;
2838 BUG_ON(err
== -ENOSPC
);
2842 static int link_to_upper(struct btrfs_trans_handle
*trans
,
2843 struct reloc_control
*rc
,
2844 struct backref_node
*node
,
2845 struct btrfs_path
*path
)
2847 struct btrfs_key key
;
2849 btrfs_node_key_to_cpu(node
->eb
, &key
, 0);
2850 return do_relocation(trans
, rc
, node
, &key
, path
, 0);
2853 static int finish_pending_nodes(struct btrfs_trans_handle
*trans
,
2854 struct reloc_control
*rc
,
2855 struct btrfs_path
*path
, int err
)
2858 struct backref_cache
*cache
= &rc
->backref_cache
;
2859 struct backref_node
*node
;
2863 for (level
= 0; level
< BTRFS_MAX_LEVEL
; level
++) {
2864 while (!list_empty(&cache
->pending
[level
])) {
2865 node
= list_entry(cache
->pending
[level
].next
,
2866 struct backref_node
, list
);
2867 list_move_tail(&node
->list
, &list
);
2868 BUG_ON(!node
->pending
);
2871 ret
= link_to_upper(trans
, rc
, node
, path
);
2876 list_splice_init(&list
, &cache
->pending
[level
]);
2881 static void mark_block_processed(struct reloc_control
*rc
,
2882 u64 bytenr
, u32 blocksize
)
2884 set_extent_bits(&rc
->processed_blocks
, bytenr
, bytenr
+ blocksize
- 1,
2888 static void __mark_block_processed(struct reloc_control
*rc
,
2889 struct backref_node
*node
)
2892 if (node
->level
== 0 ||
2893 in_block_group(node
->bytenr
, rc
->block_group
)) {
2894 blocksize
= rc
->extent_root
->fs_info
->nodesize
;
2895 mark_block_processed(rc
, node
->bytenr
, blocksize
);
2897 node
->processed
= 1;
2901 * mark a block and all blocks directly/indirectly reference the block
2904 static void update_processed_blocks(struct reloc_control
*rc
,
2905 struct backref_node
*node
)
2907 struct backref_node
*next
= node
;
2908 struct backref_edge
*edge
;
2909 struct backref_edge
*edges
[BTRFS_MAX_LEVEL
- 1];
2915 if (next
->processed
)
2918 __mark_block_processed(rc
, next
);
2920 if (list_empty(&next
->upper
))
2923 edge
= list_entry(next
->upper
.next
,
2924 struct backref_edge
, list
[LOWER
]);
2925 edges
[index
++] = edge
;
2926 next
= edge
->node
[UPPER
];
2928 next
= walk_down_backref(edges
, &index
);
2932 static int tree_block_processed(u64 bytenr
, struct reloc_control
*rc
)
2934 u32 blocksize
= rc
->extent_root
->fs_info
->nodesize
;
2936 if (test_range_bit(&rc
->processed_blocks
, bytenr
,
2937 bytenr
+ blocksize
- 1, EXTENT_DIRTY
, 1, NULL
))
2942 static int get_tree_block_key(struct reloc_control
*rc
,
2943 struct tree_block
*block
)
2945 struct extent_buffer
*eb
;
2947 BUG_ON(block
->key_ready
);
2948 eb
= read_tree_block(rc
->extent_root
, block
->bytenr
,
2952 } else if (!extent_buffer_uptodate(eb
)) {
2953 free_extent_buffer(eb
);
2956 WARN_ON(btrfs_header_level(eb
) != block
->level
);
2957 if (block
->level
== 0)
2958 btrfs_item_key_to_cpu(eb
, &block
->key
, 0);
2960 btrfs_node_key_to_cpu(eb
, &block
->key
, 0);
2961 free_extent_buffer(eb
);
2962 block
->key_ready
= 1;
2967 * helper function to relocate a tree block
2969 static int relocate_tree_block(struct btrfs_trans_handle
*trans
,
2970 struct reloc_control
*rc
,
2971 struct backref_node
*node
,
2972 struct btrfs_key
*key
,
2973 struct btrfs_path
*path
)
2975 struct btrfs_root
*root
;
2981 BUG_ON(node
->processed
);
2982 root
= select_one_root(node
);
2983 if (root
== ERR_PTR(-ENOENT
)) {
2984 update_processed_blocks(rc
, node
);
2988 if (!root
|| test_bit(BTRFS_ROOT_REF_COWS
, &root
->state
)) {
2989 ret
= reserve_metadata_space(trans
, rc
, node
);
2995 if (test_bit(BTRFS_ROOT_REF_COWS
, &root
->state
)) {
2996 BUG_ON(node
->new_bytenr
);
2997 BUG_ON(!list_empty(&node
->list
));
2998 btrfs_record_root_in_trans(trans
, root
);
2999 root
= root
->reloc_root
;
3000 node
->new_bytenr
= root
->node
->start
;
3002 list_add_tail(&node
->list
, &rc
->backref_cache
.changed
);
3004 path
->lowest_level
= node
->level
;
3005 ret
= btrfs_search_slot(trans
, root
, key
, path
, 0, 1);
3006 btrfs_release_path(path
);
3011 update_processed_blocks(rc
, node
);
3013 ret
= do_relocation(trans
, rc
, node
, key
, path
, 1);
3016 if (ret
|| node
->level
== 0 || node
->cowonly
)
3017 remove_backref_node(&rc
->backref_cache
, node
);
3022 * relocate a list of blocks
3024 static noinline_for_stack
3025 int relocate_tree_blocks(struct btrfs_trans_handle
*trans
,
3026 struct reloc_control
*rc
, struct rb_root
*blocks
)
3028 struct backref_node
*node
;
3029 struct btrfs_path
*path
;
3030 struct tree_block
*block
;
3031 struct rb_node
*rb_node
;
3035 path
= btrfs_alloc_path();
3038 goto out_free_blocks
;
3041 rb_node
= rb_first(blocks
);
3043 block
= rb_entry(rb_node
, struct tree_block
, rb_node
);
3044 if (!block
->key_ready
)
3045 readahead_tree_block(rc
->extent_root
, block
->bytenr
);
3046 rb_node
= rb_next(rb_node
);
3049 rb_node
= rb_first(blocks
);
3051 block
= rb_entry(rb_node
, struct tree_block
, rb_node
);
3052 if (!block
->key_ready
) {
3053 err
= get_tree_block_key(rc
, block
);
3057 rb_node
= rb_next(rb_node
);
3060 rb_node
= rb_first(blocks
);
3062 block
= rb_entry(rb_node
, struct tree_block
, rb_node
);
3064 node
= build_backref_tree(rc
, &block
->key
,
3065 block
->level
, block
->bytenr
);
3067 err
= PTR_ERR(node
);
3071 ret
= relocate_tree_block(trans
, rc
, node
, &block
->key
,
3074 if (ret
!= -EAGAIN
|| rb_node
== rb_first(blocks
))
3078 rb_node
= rb_next(rb_node
);
3081 err
= finish_pending_nodes(trans
, rc
, path
, err
);
3084 btrfs_free_path(path
);
3086 free_block_list(blocks
);
3090 static noinline_for_stack
3091 int prealloc_file_extent_cluster(struct inode
*inode
,
3092 struct file_extent_cluster
*cluster
)
3097 u64 offset
= BTRFS_I(inode
)->index_cnt
;
3101 u64 prealloc_start
= cluster
->start
- offset
;
3102 u64 prealloc_end
= cluster
->end
- offset
;
3105 BUG_ON(cluster
->start
!= cluster
->boundary
[0]);
3108 ret
= btrfs_check_data_free_space(inode
, prealloc_start
,
3109 prealloc_end
+ 1 - prealloc_start
);
3113 cur_offset
= prealloc_start
;
3114 while (nr
< cluster
->nr
) {
3115 start
= cluster
->boundary
[nr
] - offset
;
3116 if (nr
+ 1 < cluster
->nr
)
3117 end
= cluster
->boundary
[nr
+ 1] - 1 - offset
;
3119 end
= cluster
->end
- offset
;
3121 lock_extent(&BTRFS_I(inode
)->io_tree
, start
, end
);
3122 num_bytes
= end
+ 1 - start
;
3123 if (cur_offset
< start
)
3124 btrfs_free_reserved_data_space(inode
, cur_offset
,
3125 start
- cur_offset
);
3126 ret
= btrfs_prealloc_file_range(inode
, 0, start
,
3127 num_bytes
, num_bytes
,
3128 end
+ 1, &alloc_hint
);
3129 cur_offset
= end
+ 1;
3130 unlock_extent(&BTRFS_I(inode
)->io_tree
, start
, end
);
3135 if (cur_offset
< prealloc_end
)
3136 btrfs_free_reserved_data_space(inode
, cur_offset
,
3137 prealloc_end
+ 1 - cur_offset
);
3139 inode_unlock(inode
);
3143 static noinline_for_stack
3144 int setup_extent_mapping(struct inode
*inode
, u64 start
, u64 end
,
3147 struct btrfs_fs_info
*fs_info
= btrfs_sb(inode
->i_sb
);
3148 struct extent_map_tree
*em_tree
= &BTRFS_I(inode
)->extent_tree
;
3149 struct extent_map
*em
;
3152 em
= alloc_extent_map();
3157 em
->len
= end
+ 1 - start
;
3158 em
->block_len
= em
->len
;
3159 em
->block_start
= block_start
;
3160 em
->bdev
= fs_info
->fs_devices
->latest_bdev
;
3161 set_bit(EXTENT_FLAG_PINNED
, &em
->flags
);
3163 lock_extent(&BTRFS_I(inode
)->io_tree
, start
, end
);
3165 write_lock(&em_tree
->lock
);
3166 ret
= add_extent_mapping(em_tree
, em
, 0);
3167 write_unlock(&em_tree
->lock
);
3168 if (ret
!= -EEXIST
) {
3169 free_extent_map(em
);
3172 btrfs_drop_extent_cache(inode
, start
, end
, 0);
3174 unlock_extent(&BTRFS_I(inode
)->io_tree
, start
, end
);
3178 static int relocate_file_extent_cluster(struct inode
*inode
,
3179 struct file_extent_cluster
*cluster
)
3183 u64 offset
= BTRFS_I(inode
)->index_cnt
;
3184 unsigned long index
;
3185 unsigned long last_index
;
3187 struct file_ra_state
*ra
;
3188 gfp_t mask
= btrfs_alloc_write_mask(inode
->i_mapping
);
3195 ra
= kzalloc(sizeof(*ra
), GFP_NOFS
);
3199 ret
= prealloc_file_extent_cluster(inode
, cluster
);
3203 file_ra_state_init(ra
, inode
->i_mapping
);
3205 ret
= setup_extent_mapping(inode
, cluster
->start
- offset
,
3206 cluster
->end
- offset
, cluster
->start
);
3210 index
= (cluster
->start
- offset
) >> PAGE_SHIFT
;
3211 last_index
= (cluster
->end
- offset
) >> PAGE_SHIFT
;
3212 while (index
<= last_index
) {
3213 ret
= btrfs_delalloc_reserve_metadata(inode
, PAGE_SIZE
);
3217 page
= find_lock_page(inode
->i_mapping
, index
);
3219 page_cache_sync_readahead(inode
->i_mapping
,
3221 last_index
+ 1 - index
);
3222 page
= find_or_create_page(inode
->i_mapping
, index
,
3225 btrfs_delalloc_release_metadata(inode
,
3232 if (PageReadahead(page
)) {
3233 page_cache_async_readahead(inode
->i_mapping
,
3234 ra
, NULL
, page
, index
,
3235 last_index
+ 1 - index
);
3238 if (!PageUptodate(page
)) {
3239 btrfs_readpage(NULL
, page
);
3241 if (!PageUptodate(page
)) {
3244 btrfs_delalloc_release_metadata(inode
,
3251 page_start
= page_offset(page
);
3252 page_end
= page_start
+ PAGE_SIZE
- 1;
3254 lock_extent(&BTRFS_I(inode
)->io_tree
, page_start
, page_end
);
3256 set_page_extent_mapped(page
);
3258 if (nr
< cluster
->nr
&&
3259 page_start
+ offset
== cluster
->boundary
[nr
]) {
3260 set_extent_bits(&BTRFS_I(inode
)->io_tree
,
3261 page_start
, page_end
,
3266 btrfs_set_extent_delalloc(inode
, page_start
, page_end
, NULL
, 0);
3267 set_page_dirty(page
);
3269 unlock_extent(&BTRFS_I(inode
)->io_tree
,
3270 page_start
, page_end
);
3275 balance_dirty_pages_ratelimited(inode
->i_mapping
);
3276 btrfs_throttle(BTRFS_I(inode
)->root
);
3278 WARN_ON(nr
!= cluster
->nr
);
3284 static noinline_for_stack
3285 int relocate_data_extent(struct inode
*inode
, struct btrfs_key
*extent_key
,
3286 struct file_extent_cluster
*cluster
)
3290 if (cluster
->nr
> 0 && extent_key
->objectid
!= cluster
->end
+ 1) {
3291 ret
= relocate_file_extent_cluster(inode
, cluster
);
3298 cluster
->start
= extent_key
->objectid
;
3300 BUG_ON(cluster
->nr
>= MAX_EXTENTS
);
3301 cluster
->end
= extent_key
->objectid
+ extent_key
->offset
- 1;
3302 cluster
->boundary
[cluster
->nr
] = extent_key
->objectid
;
3305 if (cluster
->nr
>= MAX_EXTENTS
) {
3306 ret
= relocate_file_extent_cluster(inode
, cluster
);
3314 #ifdef BTRFS_COMPAT_EXTENT_TREE_V0
3315 static int get_ref_objectid_v0(struct reloc_control
*rc
,
3316 struct btrfs_path
*path
,
3317 struct btrfs_key
*extent_key
,
3318 u64
*ref_objectid
, int *path_change
)
3320 struct btrfs_key key
;
3321 struct extent_buffer
*leaf
;
3322 struct btrfs_extent_ref_v0
*ref0
;
3326 leaf
= path
->nodes
[0];
3327 slot
= path
->slots
[0];
3329 if (slot
>= btrfs_header_nritems(leaf
)) {
3330 ret
= btrfs_next_leaf(rc
->extent_root
, path
);
3334 leaf
= path
->nodes
[0];
3335 slot
= path
->slots
[0];
3339 btrfs_item_key_to_cpu(leaf
, &key
, slot
);
3340 if (key
.objectid
!= extent_key
->objectid
)
3343 if (key
.type
!= BTRFS_EXTENT_REF_V0_KEY
) {
3347 ref0
= btrfs_item_ptr(leaf
, slot
,
3348 struct btrfs_extent_ref_v0
);
3349 *ref_objectid
= btrfs_ref_objectid_v0(leaf
, ref0
);
3357 * helper to add a tree block to the list.
3358 * the major work is getting the generation and level of the block
3360 static int add_tree_block(struct reloc_control
*rc
,
3361 struct btrfs_key
*extent_key
,
3362 struct btrfs_path
*path
,
3363 struct rb_root
*blocks
)
3365 struct extent_buffer
*eb
;
3366 struct btrfs_extent_item
*ei
;
3367 struct btrfs_tree_block_info
*bi
;
3368 struct tree_block
*block
;
3369 struct rb_node
*rb_node
;
3374 eb
= path
->nodes
[0];
3375 item_size
= btrfs_item_size_nr(eb
, path
->slots
[0]);
3377 if (extent_key
->type
== BTRFS_METADATA_ITEM_KEY
||
3378 item_size
>= sizeof(*ei
) + sizeof(*bi
)) {
3379 ei
= btrfs_item_ptr(eb
, path
->slots
[0],
3380 struct btrfs_extent_item
);
3381 if (extent_key
->type
== BTRFS_EXTENT_ITEM_KEY
) {
3382 bi
= (struct btrfs_tree_block_info
*)(ei
+ 1);
3383 level
= btrfs_tree_block_level(eb
, bi
);
3385 level
= (int)extent_key
->offset
;
3387 generation
= btrfs_extent_generation(eb
, ei
);
3389 #ifdef BTRFS_COMPAT_EXTENT_TREE_V0
3393 BUG_ON(item_size
!= sizeof(struct btrfs_extent_item_v0
));
3394 ret
= get_ref_objectid_v0(rc
, path
, extent_key
,
3398 BUG_ON(ref_owner
>= BTRFS_MAX_LEVEL
);
3399 level
= (int)ref_owner
;
3400 /* FIXME: get real generation */
3407 btrfs_release_path(path
);
3409 BUG_ON(level
== -1);
3411 block
= kmalloc(sizeof(*block
), GFP_NOFS
);
3415 block
->bytenr
= extent_key
->objectid
;
3416 block
->key
.objectid
= rc
->extent_root
->fs_info
->nodesize
;
3417 block
->key
.offset
= generation
;
3418 block
->level
= level
;
3419 block
->key_ready
= 0;
3421 rb_node
= tree_insert(blocks
, block
->bytenr
, &block
->rb_node
);
3423 backref_tree_panic(rb_node
, -EEXIST
, block
->bytenr
);
3429 * helper to add tree blocks for backref of type BTRFS_SHARED_DATA_REF_KEY
3431 static int __add_tree_block(struct reloc_control
*rc
,
3432 u64 bytenr
, u32 blocksize
,
3433 struct rb_root
*blocks
)
3435 struct btrfs_fs_info
*fs_info
= rc
->extent_root
->fs_info
;
3436 struct btrfs_path
*path
;
3437 struct btrfs_key key
;
3439 bool skinny
= btrfs_fs_incompat(fs_info
, SKINNY_METADATA
);
3441 if (tree_block_processed(bytenr
, rc
))
3444 if (tree_search(blocks
, bytenr
))
3447 path
= btrfs_alloc_path();
3451 key
.objectid
= bytenr
;
3453 key
.type
= BTRFS_METADATA_ITEM_KEY
;
3454 key
.offset
= (u64
)-1;
3456 key
.type
= BTRFS_EXTENT_ITEM_KEY
;
3457 key
.offset
= blocksize
;
3460 path
->search_commit_root
= 1;
3461 path
->skip_locking
= 1;
3462 ret
= btrfs_search_slot(NULL
, rc
->extent_root
, &key
, path
, 0, 0);
3466 if (ret
> 0 && skinny
) {
3467 if (path
->slots
[0]) {
3469 btrfs_item_key_to_cpu(path
->nodes
[0], &key
,
3471 if (key
.objectid
== bytenr
&&
3472 (key
.type
== BTRFS_METADATA_ITEM_KEY
||
3473 (key
.type
== BTRFS_EXTENT_ITEM_KEY
&&
3474 key
.offset
== blocksize
)))
3480 btrfs_release_path(path
);
3486 ret
= add_tree_block(rc
, &key
, path
, blocks
);
3488 btrfs_free_path(path
);
3493 * helper to check if the block use full backrefs for pointers in it
3495 static int block_use_full_backref(struct reloc_control
*rc
,
3496 struct extent_buffer
*eb
)
3501 if (btrfs_header_flag(eb
, BTRFS_HEADER_FLAG_RELOC
) ||
3502 btrfs_header_backref_rev(eb
) < BTRFS_MIXED_BACKREF_REV
)
3505 ret
= btrfs_lookup_extent_info(NULL
, rc
->extent_root
,
3506 eb
->start
, btrfs_header_level(eb
), 1,
3510 if (flags
& BTRFS_BLOCK_FLAG_FULL_BACKREF
)
3517 static int delete_block_group_cache(struct btrfs_fs_info
*fs_info
,
3518 struct btrfs_block_group_cache
*block_group
,
3519 struct inode
*inode
,
3522 struct btrfs_key key
;
3523 struct btrfs_root
*root
= fs_info
->tree_root
;
3524 struct btrfs_trans_handle
*trans
;
3531 key
.type
= BTRFS_INODE_ITEM_KEY
;
3534 inode
= btrfs_iget(fs_info
->sb
, &key
, root
, NULL
);
3535 if (IS_ERR(inode
) || is_bad_inode(inode
)) {
3542 ret
= btrfs_check_trunc_cache_free_space(root
,
3543 &fs_info
->global_block_rsv
);
3547 trans
= btrfs_join_transaction(root
);
3548 if (IS_ERR(trans
)) {
3549 ret
= PTR_ERR(trans
);
3553 ret
= btrfs_truncate_free_space_cache(root
, trans
, block_group
, inode
);
3555 btrfs_end_transaction(trans
, root
);
3556 btrfs_btree_balance_dirty(root
);
3563 * helper to add tree blocks for backref of type BTRFS_EXTENT_DATA_REF_KEY
3564 * this function scans fs tree to find blocks reference the data extent
3566 static int find_data_references(struct reloc_control
*rc
,
3567 struct btrfs_key
*extent_key
,
3568 struct extent_buffer
*leaf
,
3569 struct btrfs_extent_data_ref
*ref
,
3570 struct rb_root
*blocks
)
3572 struct btrfs_fs_info
*fs_info
= rc
->extent_root
->fs_info
;
3573 struct btrfs_path
*path
;
3574 struct tree_block
*block
;
3575 struct btrfs_root
*root
;
3576 struct btrfs_file_extent_item
*fi
;
3577 struct rb_node
*rb_node
;
3578 struct btrfs_key key
;
3589 ref_root
= btrfs_extent_data_ref_root(leaf
, ref
);
3590 ref_objectid
= btrfs_extent_data_ref_objectid(leaf
, ref
);
3591 ref_offset
= btrfs_extent_data_ref_offset(leaf
, ref
);
3592 ref_count
= btrfs_extent_data_ref_count(leaf
, ref
);
3595 * This is an extent belonging to the free space cache, lets just delete
3596 * it and redo the search.
3598 if (ref_root
== BTRFS_ROOT_TREE_OBJECTID
) {
3599 ret
= delete_block_group_cache(fs_info
, rc
->block_group
,
3600 NULL
, ref_objectid
);
3606 path
= btrfs_alloc_path();
3609 path
->reada
= READA_FORWARD
;
3611 root
= read_fs_root(fs_info
, ref_root
);
3613 err
= PTR_ERR(root
);
3617 key
.objectid
= ref_objectid
;
3618 key
.type
= BTRFS_EXTENT_DATA_KEY
;
3619 if (ref_offset
> ((u64
)-1 << 32))
3622 key
.offset
= ref_offset
;
3624 path
->search_commit_root
= 1;
3625 path
->skip_locking
= 1;
3626 ret
= btrfs_search_slot(NULL
, root
, &key
, path
, 0, 0);
3632 leaf
= path
->nodes
[0];
3633 nritems
= btrfs_header_nritems(leaf
);
3635 * the references in tree blocks that use full backrefs
3636 * are not counted in
3638 if (block_use_full_backref(rc
, leaf
))
3642 rb_node
= tree_search(blocks
, leaf
->start
);
3647 path
->slots
[0] = nritems
;
3650 while (ref_count
> 0) {
3651 while (path
->slots
[0] >= nritems
) {
3652 ret
= btrfs_next_leaf(root
, path
);
3657 if (WARN_ON(ret
> 0))
3660 leaf
= path
->nodes
[0];
3661 nritems
= btrfs_header_nritems(leaf
);
3664 if (block_use_full_backref(rc
, leaf
))
3668 rb_node
= tree_search(blocks
, leaf
->start
);
3673 path
->slots
[0] = nritems
;
3677 btrfs_item_key_to_cpu(leaf
, &key
, path
->slots
[0]);
3678 if (WARN_ON(key
.objectid
!= ref_objectid
||
3679 key
.type
!= BTRFS_EXTENT_DATA_KEY
))
3682 fi
= btrfs_item_ptr(leaf
, path
->slots
[0],
3683 struct btrfs_file_extent_item
);
3685 if (btrfs_file_extent_type(leaf
, fi
) ==
3686 BTRFS_FILE_EXTENT_INLINE
)
3689 if (btrfs_file_extent_disk_bytenr(leaf
, fi
) !=
3690 extent_key
->objectid
)
3693 key
.offset
-= btrfs_file_extent_offset(leaf
, fi
);
3694 if (key
.offset
!= ref_offset
)
3702 if (!tree_block_processed(leaf
->start
, rc
)) {
3703 block
= kmalloc(sizeof(*block
), GFP_NOFS
);
3708 block
->bytenr
= leaf
->start
;
3709 btrfs_item_key_to_cpu(leaf
, &block
->key
, 0);
3711 block
->key_ready
= 1;
3712 rb_node
= tree_insert(blocks
, block
->bytenr
,
3715 backref_tree_panic(rb_node
, -EEXIST
,
3721 path
->slots
[0] = nritems
;
3727 btrfs_free_path(path
);
3732 * helper to find all tree blocks that reference a given data extent
3734 static noinline_for_stack
3735 int add_data_references(struct reloc_control
*rc
,
3736 struct btrfs_key
*extent_key
,
3737 struct btrfs_path
*path
,
3738 struct rb_root
*blocks
)
3740 struct btrfs_key key
;
3741 struct extent_buffer
*eb
;
3742 struct btrfs_extent_data_ref
*dref
;
3743 struct btrfs_extent_inline_ref
*iref
;
3746 u32 blocksize
= rc
->extent_root
->fs_info
->nodesize
;
3750 eb
= path
->nodes
[0];
3751 ptr
= btrfs_item_ptr_offset(eb
, path
->slots
[0]);
3752 end
= ptr
+ btrfs_item_size_nr(eb
, path
->slots
[0]);
3753 #ifdef BTRFS_COMPAT_EXTENT_TREE_V0
3754 if (ptr
+ sizeof(struct btrfs_extent_item_v0
) == end
)
3758 ptr
+= sizeof(struct btrfs_extent_item
);
3761 iref
= (struct btrfs_extent_inline_ref
*)ptr
;
3762 key
.type
= btrfs_extent_inline_ref_type(eb
, iref
);
3763 if (key
.type
== BTRFS_SHARED_DATA_REF_KEY
) {
3764 key
.offset
= btrfs_extent_inline_ref_offset(eb
, iref
);
3765 ret
= __add_tree_block(rc
, key
.offset
, blocksize
,
3767 } else if (key
.type
== BTRFS_EXTENT_DATA_REF_KEY
) {
3768 dref
= (struct btrfs_extent_data_ref
*)(&iref
->offset
);
3769 ret
= find_data_references(rc
, extent_key
,
3778 ptr
+= btrfs_extent_inline_ref_size(key
.type
);
3784 eb
= path
->nodes
[0];
3785 if (path
->slots
[0] >= btrfs_header_nritems(eb
)) {
3786 ret
= btrfs_next_leaf(rc
->extent_root
, path
);
3793 eb
= path
->nodes
[0];
3796 btrfs_item_key_to_cpu(eb
, &key
, path
->slots
[0]);
3797 if (key
.objectid
!= extent_key
->objectid
)
3800 #ifdef BTRFS_COMPAT_EXTENT_TREE_V0
3801 if (key
.type
== BTRFS_SHARED_DATA_REF_KEY
||
3802 key
.type
== BTRFS_EXTENT_REF_V0_KEY
) {
3804 BUG_ON(key
.type
== BTRFS_EXTENT_REF_V0_KEY
);
3805 if (key
.type
== BTRFS_SHARED_DATA_REF_KEY
) {
3807 ret
= __add_tree_block(rc
, key
.offset
, blocksize
,
3809 } else if (key
.type
== BTRFS_EXTENT_DATA_REF_KEY
) {
3810 dref
= btrfs_item_ptr(eb
, path
->slots
[0],
3811 struct btrfs_extent_data_ref
);
3812 ret
= find_data_references(rc
, extent_key
,
3824 btrfs_release_path(path
);
3826 free_block_list(blocks
);
3831 * helper to find next unprocessed extent
3833 static noinline_for_stack
3834 int find_next_extent(struct reloc_control
*rc
, struct btrfs_path
*path
,
3835 struct btrfs_key
*extent_key
)
3837 struct btrfs_fs_info
*fs_info
= rc
->extent_root
->fs_info
;
3838 struct btrfs_key key
;
3839 struct extent_buffer
*leaf
;
3840 u64 start
, end
, last
;
3843 last
= rc
->block_group
->key
.objectid
+ rc
->block_group
->key
.offset
;
3846 if (rc
->search_start
>= last
) {
3851 key
.objectid
= rc
->search_start
;
3852 key
.type
= BTRFS_EXTENT_ITEM_KEY
;
3855 path
->search_commit_root
= 1;
3856 path
->skip_locking
= 1;
3857 ret
= btrfs_search_slot(NULL
, rc
->extent_root
, &key
, path
,
3862 leaf
= path
->nodes
[0];
3863 if (path
->slots
[0] >= btrfs_header_nritems(leaf
)) {
3864 ret
= btrfs_next_leaf(rc
->extent_root
, path
);
3867 leaf
= path
->nodes
[0];
3870 btrfs_item_key_to_cpu(leaf
, &key
, path
->slots
[0]);
3871 if (key
.objectid
>= last
) {
3876 if (key
.type
!= BTRFS_EXTENT_ITEM_KEY
&&
3877 key
.type
!= BTRFS_METADATA_ITEM_KEY
) {
3882 if (key
.type
== BTRFS_EXTENT_ITEM_KEY
&&
3883 key
.objectid
+ key
.offset
<= rc
->search_start
) {
3888 if (key
.type
== BTRFS_METADATA_ITEM_KEY
&&
3889 key
.objectid
+ fs_info
->nodesize
<=
3895 ret
= find_first_extent_bit(&rc
->processed_blocks
,
3896 key
.objectid
, &start
, &end
,
3897 EXTENT_DIRTY
, NULL
);
3899 if (ret
== 0 && start
<= key
.objectid
) {
3900 btrfs_release_path(path
);
3901 rc
->search_start
= end
+ 1;
3903 if (key
.type
== BTRFS_EXTENT_ITEM_KEY
)
3904 rc
->search_start
= key
.objectid
+ key
.offset
;
3906 rc
->search_start
= key
.objectid
+
3908 memcpy(extent_key
, &key
, sizeof(key
));
3912 btrfs_release_path(path
);
3916 static void set_reloc_control(struct reloc_control
*rc
)
3918 struct btrfs_fs_info
*fs_info
= rc
->extent_root
->fs_info
;
3920 mutex_lock(&fs_info
->reloc_mutex
);
3921 fs_info
->reloc_ctl
= rc
;
3922 mutex_unlock(&fs_info
->reloc_mutex
);
3925 static void unset_reloc_control(struct reloc_control
*rc
)
3927 struct btrfs_fs_info
*fs_info
= rc
->extent_root
->fs_info
;
3929 mutex_lock(&fs_info
->reloc_mutex
);
3930 fs_info
->reloc_ctl
= NULL
;
3931 mutex_unlock(&fs_info
->reloc_mutex
);
3934 static int check_extent_flags(u64 flags
)
3936 if ((flags
& BTRFS_EXTENT_FLAG_DATA
) &&
3937 (flags
& BTRFS_EXTENT_FLAG_TREE_BLOCK
))
3939 if (!(flags
& BTRFS_EXTENT_FLAG_DATA
) &&
3940 !(flags
& BTRFS_EXTENT_FLAG_TREE_BLOCK
))
3942 if ((flags
& BTRFS_EXTENT_FLAG_DATA
) &&
3943 (flags
& BTRFS_BLOCK_FLAG_FULL_BACKREF
))
3948 static noinline_for_stack
3949 int prepare_to_relocate(struct reloc_control
*rc
)
3951 struct btrfs_trans_handle
*trans
;
3954 rc
->block_rsv
= btrfs_alloc_block_rsv(rc
->extent_root
,
3955 BTRFS_BLOCK_RSV_TEMP
);
3959 memset(&rc
->cluster
, 0, sizeof(rc
->cluster
));
3960 rc
->search_start
= rc
->block_group
->key
.objectid
;
3961 rc
->extents_found
= 0;
3962 rc
->nodes_relocated
= 0;
3963 rc
->merging_rsv_size
= 0;
3964 rc
->reserved_bytes
= 0;
3965 rc
->block_rsv
->size
= rc
->extent_root
->fs_info
->nodesize
*
3966 RELOCATION_RESERVED_NODES
;
3967 ret
= btrfs_block_rsv_refill(rc
->extent_root
,
3968 rc
->block_rsv
, rc
->block_rsv
->size
,
3969 BTRFS_RESERVE_FLUSH_ALL
);
3973 rc
->create_reloc_tree
= 1;
3974 set_reloc_control(rc
);
3976 trans
= btrfs_join_transaction(rc
->extent_root
);
3977 if (IS_ERR(trans
)) {
3978 unset_reloc_control(rc
);
3980 * extent tree is not a ref_cow tree and has no reloc_root to
3981 * cleanup. And callers are responsible to free the above
3984 return PTR_ERR(trans
);
3986 btrfs_commit_transaction(trans
, rc
->extent_root
);
3990 static noinline_for_stack
int relocate_block_group(struct reloc_control
*rc
)
3992 struct rb_root blocks
= RB_ROOT
;
3993 struct btrfs_key key
;
3994 struct btrfs_trans_handle
*trans
= NULL
;
3995 struct btrfs_path
*path
;
3996 struct btrfs_extent_item
*ei
;
4003 path
= btrfs_alloc_path();
4006 path
->reada
= READA_FORWARD
;
4008 ret
= prepare_to_relocate(rc
);
4015 rc
->reserved_bytes
= 0;
4016 ret
= btrfs_block_rsv_refill(rc
->extent_root
,
4017 rc
->block_rsv
, rc
->block_rsv
->size
,
4018 BTRFS_RESERVE_FLUSH_ALL
);
4024 trans
= btrfs_start_transaction(rc
->extent_root
, 0);
4025 if (IS_ERR(trans
)) {
4026 err
= PTR_ERR(trans
);
4031 if (update_backref_cache(trans
, &rc
->backref_cache
)) {
4032 btrfs_end_transaction(trans
, rc
->extent_root
);
4036 ret
= find_next_extent(rc
, path
, &key
);
4042 rc
->extents_found
++;
4044 ei
= btrfs_item_ptr(path
->nodes
[0], path
->slots
[0],
4045 struct btrfs_extent_item
);
4046 item_size
= btrfs_item_size_nr(path
->nodes
[0], path
->slots
[0]);
4047 if (item_size
>= sizeof(*ei
)) {
4048 flags
= btrfs_extent_flags(path
->nodes
[0], ei
);
4049 ret
= check_extent_flags(flags
);
4053 #ifdef BTRFS_COMPAT_EXTENT_TREE_V0
4055 int path_change
= 0;
4058 sizeof(struct btrfs_extent_item_v0
));
4059 ret
= get_ref_objectid_v0(rc
, path
, &key
, &ref_owner
,
4065 if (ref_owner
< BTRFS_FIRST_FREE_OBJECTID
)
4066 flags
= BTRFS_EXTENT_FLAG_TREE_BLOCK
;
4068 flags
= BTRFS_EXTENT_FLAG_DATA
;
4071 btrfs_release_path(path
);
4073 path
->search_commit_root
= 1;
4074 path
->skip_locking
= 1;
4075 ret
= btrfs_search_slot(NULL
, rc
->extent_root
,
4088 if (flags
& BTRFS_EXTENT_FLAG_TREE_BLOCK
) {
4089 ret
= add_tree_block(rc
, &key
, path
, &blocks
);
4090 } else if (rc
->stage
== UPDATE_DATA_PTRS
&&
4091 (flags
& BTRFS_EXTENT_FLAG_DATA
)) {
4092 ret
= add_data_references(rc
, &key
, path
, &blocks
);
4094 btrfs_release_path(path
);
4102 if (!RB_EMPTY_ROOT(&blocks
)) {
4103 ret
= relocate_tree_blocks(trans
, rc
, &blocks
);
4106 * if we fail to relocate tree blocks, force to update
4107 * backref cache when committing transaction.
4109 rc
->backref_cache
.last_trans
= trans
->transid
- 1;
4111 if (ret
!= -EAGAIN
) {
4115 rc
->extents_found
--;
4116 rc
->search_start
= key
.objectid
;
4120 btrfs_end_transaction_throttle(trans
, rc
->extent_root
);
4121 btrfs_btree_balance_dirty(rc
->extent_root
);
4124 if (rc
->stage
== MOVE_DATA_EXTENTS
&&
4125 (flags
& BTRFS_EXTENT_FLAG_DATA
)) {
4126 rc
->found_file_extent
= 1;
4127 ret
= relocate_data_extent(rc
->data_inode
,
4128 &key
, &rc
->cluster
);
4135 if (trans
&& progress
&& err
== -ENOSPC
) {
4136 ret
= btrfs_force_chunk_alloc(trans
, rc
->extent_root
,
4137 rc
->block_group
->flags
);
4145 btrfs_release_path(path
);
4146 clear_extent_bits(&rc
->processed_blocks
, 0, (u64
)-1, EXTENT_DIRTY
);
4149 btrfs_end_transaction_throttle(trans
, rc
->extent_root
);
4150 btrfs_btree_balance_dirty(rc
->extent_root
);
4154 ret
= relocate_file_extent_cluster(rc
->data_inode
,
4160 rc
->create_reloc_tree
= 0;
4161 set_reloc_control(rc
);
4163 backref_cache_cleanup(&rc
->backref_cache
);
4164 btrfs_block_rsv_release(rc
->extent_root
, rc
->block_rsv
, (u64
)-1);
4166 err
= prepare_to_merge(rc
, err
);
4168 merge_reloc_roots(rc
);
4170 rc
->merge_reloc_tree
= 0;
4171 unset_reloc_control(rc
);
4172 btrfs_block_rsv_release(rc
->extent_root
, rc
->block_rsv
, (u64
)-1);
4174 /* get rid of pinned extents */
4175 trans
= btrfs_join_transaction(rc
->extent_root
);
4176 if (IS_ERR(trans
)) {
4177 err
= PTR_ERR(trans
);
4180 btrfs_commit_transaction(trans
, rc
->extent_root
);
4182 btrfs_free_block_rsv(rc
->extent_root
, rc
->block_rsv
);
4183 btrfs_free_path(path
);
4187 static int __insert_orphan_inode(struct btrfs_trans_handle
*trans
,
4188 struct btrfs_root
*root
, u64 objectid
)
4190 struct btrfs_path
*path
;
4191 struct btrfs_inode_item
*item
;
4192 struct extent_buffer
*leaf
;
4195 path
= btrfs_alloc_path();
4199 ret
= btrfs_insert_empty_inode(trans
, root
, path
, objectid
);
4203 leaf
= path
->nodes
[0];
4204 item
= btrfs_item_ptr(leaf
, path
->slots
[0], struct btrfs_inode_item
);
4205 memzero_extent_buffer(leaf
, (unsigned long)item
, sizeof(*item
));
4206 btrfs_set_inode_generation(leaf
, item
, 1);
4207 btrfs_set_inode_size(leaf
, item
, 0);
4208 btrfs_set_inode_mode(leaf
, item
, S_IFREG
| 0600);
4209 btrfs_set_inode_flags(leaf
, item
, BTRFS_INODE_NOCOMPRESS
|
4210 BTRFS_INODE_PREALLOC
);
4211 btrfs_mark_buffer_dirty(leaf
);
4213 btrfs_free_path(path
);
4218 * helper to create inode for data relocation.
4219 * the inode is in data relocation tree and its link count is 0
4221 static noinline_for_stack
4222 struct inode
*create_reloc_inode(struct btrfs_fs_info
*fs_info
,
4223 struct btrfs_block_group_cache
*group
)
4225 struct inode
*inode
= NULL
;
4226 struct btrfs_trans_handle
*trans
;
4227 struct btrfs_root
*root
;
4228 struct btrfs_key key
;
4232 root
= read_fs_root(fs_info
, BTRFS_DATA_RELOC_TREE_OBJECTID
);
4234 return ERR_CAST(root
);
4236 trans
= btrfs_start_transaction(root
, 6);
4238 return ERR_CAST(trans
);
4240 err
= btrfs_find_free_objectid(root
, &objectid
);
4244 err
= __insert_orphan_inode(trans
, root
, objectid
);
4247 key
.objectid
= objectid
;
4248 key
.type
= BTRFS_INODE_ITEM_KEY
;
4250 inode
= btrfs_iget(fs_info
->sb
, &key
, root
, NULL
);
4251 BUG_ON(IS_ERR(inode
) || is_bad_inode(inode
));
4252 BTRFS_I(inode
)->index_cnt
= group
->key
.objectid
;
4254 err
= btrfs_orphan_add(trans
, inode
);
4256 btrfs_end_transaction(trans
, root
);
4257 btrfs_btree_balance_dirty(root
);
4261 inode
= ERR_PTR(err
);
4266 static struct reloc_control
*alloc_reloc_control(struct btrfs_fs_info
*fs_info
)
4268 struct reloc_control
*rc
;
4270 rc
= kzalloc(sizeof(*rc
), GFP_NOFS
);
4274 INIT_LIST_HEAD(&rc
->reloc_roots
);
4275 backref_cache_init(&rc
->backref_cache
);
4276 mapping_tree_init(&rc
->reloc_root_tree
);
4277 extent_io_tree_init(&rc
->processed_blocks
,
4278 fs_info
->btree_inode
->i_mapping
);
4283 * Print the block group being relocated
4285 static void describe_relocation(struct btrfs_fs_info
*fs_info
,
4286 struct btrfs_block_group_cache
*block_group
)
4288 char buf
[128]; /* prefixed by a '|' that'll be dropped */
4289 u64 flags
= block_group
->flags
;
4291 /* Shouldn't happen */
4293 strcpy(buf
, "|NONE");
4297 #define DESCRIBE_FLAG(f, d) \
4298 if (flags & BTRFS_BLOCK_GROUP_##f) { \
4299 bp += snprintf(bp, buf - bp + sizeof(buf), "|%s", d); \
4300 flags &= ~BTRFS_BLOCK_GROUP_##f; \
4302 DESCRIBE_FLAG(DATA
, "data");
4303 DESCRIBE_FLAG(SYSTEM
, "system");
4304 DESCRIBE_FLAG(METADATA
, "metadata");
4305 DESCRIBE_FLAG(RAID0
, "raid0");
4306 DESCRIBE_FLAG(RAID1
, "raid1");
4307 DESCRIBE_FLAG(DUP
, "dup");
4308 DESCRIBE_FLAG(RAID10
, "raid10");
4309 DESCRIBE_FLAG(RAID5
, "raid5");
4310 DESCRIBE_FLAG(RAID6
, "raid6");
4312 snprintf(buf
, buf
- bp
+ sizeof(buf
), "|0x%llx", flags
);
4313 #undef DESCRIBE_FLAG
4317 "relocating block group %llu flags %s",
4318 block_group
->key
.objectid
, buf
+ 1);
4322 * function to relocate all extents in a block group.
4324 int btrfs_relocate_block_group(struct btrfs_fs_info
*fs_info
, u64 group_start
)
4326 struct btrfs_root
*extent_root
= fs_info
->extent_root
;
4327 struct reloc_control
*rc
;
4328 struct inode
*inode
;
4329 struct btrfs_path
*path
;
4334 rc
= alloc_reloc_control(fs_info
);
4338 rc
->extent_root
= extent_root
;
4340 rc
->block_group
= btrfs_lookup_block_group(fs_info
, group_start
);
4341 BUG_ON(!rc
->block_group
);
4343 ret
= btrfs_inc_block_group_ro(extent_root
, rc
->block_group
);
4350 path
= btrfs_alloc_path();
4356 inode
= lookup_free_space_inode(fs_info
->tree_root
, rc
->block_group
,
4358 btrfs_free_path(path
);
4361 ret
= delete_block_group_cache(fs_info
, rc
->block_group
, inode
, 0);
4363 ret
= PTR_ERR(inode
);
4365 if (ret
&& ret
!= -ENOENT
) {
4370 rc
->data_inode
= create_reloc_inode(fs_info
, rc
->block_group
);
4371 if (IS_ERR(rc
->data_inode
)) {
4372 err
= PTR_ERR(rc
->data_inode
);
4373 rc
->data_inode
= NULL
;
4377 describe_relocation(fs_info
, rc
->block_group
);
4379 btrfs_wait_block_group_reservations(rc
->block_group
);
4380 btrfs_wait_nocow_writers(rc
->block_group
);
4381 btrfs_wait_ordered_roots(fs_info
, -1,
4382 rc
->block_group
->key
.objectid
,
4383 rc
->block_group
->key
.offset
);
4386 mutex_lock(&fs_info
->cleaner_mutex
);
4387 ret
= relocate_block_group(rc
);
4388 mutex_unlock(&fs_info
->cleaner_mutex
);
4394 if (rc
->extents_found
== 0)
4397 btrfs_info(fs_info
, "found %llu extents", rc
->extents_found
);
4399 if (rc
->stage
== MOVE_DATA_EXTENTS
&& rc
->found_file_extent
) {
4400 ret
= btrfs_wait_ordered_range(rc
->data_inode
, 0,
4406 invalidate_mapping_pages(rc
->data_inode
->i_mapping
,
4408 rc
->stage
= UPDATE_DATA_PTRS
;
4412 WARN_ON(rc
->block_group
->pinned
> 0);
4413 WARN_ON(rc
->block_group
->reserved
> 0);
4414 WARN_ON(btrfs_block_group_used(&rc
->block_group
->item
) > 0);
4417 btrfs_dec_block_group_ro(extent_root
, rc
->block_group
);
4418 iput(rc
->data_inode
);
4419 btrfs_put_block_group(rc
->block_group
);
4424 static noinline_for_stack
int mark_garbage_root(struct btrfs_root
*root
)
4426 struct btrfs_fs_info
*fs_info
= root
->fs_info
;
4427 struct btrfs_trans_handle
*trans
;
4430 trans
= btrfs_start_transaction(fs_info
->tree_root
, 0);
4432 return PTR_ERR(trans
);
4434 memset(&root
->root_item
.drop_progress
, 0,
4435 sizeof(root
->root_item
.drop_progress
));
4436 root
->root_item
.drop_level
= 0;
4437 btrfs_set_root_refs(&root
->root_item
, 0);
4438 ret
= btrfs_update_root(trans
, fs_info
->tree_root
,
4439 &root
->root_key
, &root
->root_item
);
4441 err
= btrfs_end_transaction(trans
, fs_info
->tree_root
);
4448 * recover relocation interrupted by system crash.
4450 * this function resumes merging reloc trees with corresponding fs trees.
4451 * this is important for keeping the sharing of tree blocks
4453 int btrfs_recover_relocation(struct btrfs_root
*root
)
4455 struct btrfs_fs_info
*fs_info
= root
->fs_info
;
4456 LIST_HEAD(reloc_roots
);
4457 struct btrfs_key key
;
4458 struct btrfs_root
*fs_root
;
4459 struct btrfs_root
*reloc_root
;
4460 struct btrfs_path
*path
;
4461 struct extent_buffer
*leaf
;
4462 struct reloc_control
*rc
= NULL
;
4463 struct btrfs_trans_handle
*trans
;
4467 path
= btrfs_alloc_path();
4470 path
->reada
= READA_BACK
;
4472 key
.objectid
= BTRFS_TREE_RELOC_OBJECTID
;
4473 key
.type
= BTRFS_ROOT_ITEM_KEY
;
4474 key
.offset
= (u64
)-1;
4477 ret
= btrfs_search_slot(NULL
, fs_info
->tree_root
, &key
,
4484 if (path
->slots
[0] == 0)
4488 leaf
= path
->nodes
[0];
4489 btrfs_item_key_to_cpu(leaf
, &key
, path
->slots
[0]);
4490 btrfs_release_path(path
);
4492 if (key
.objectid
!= BTRFS_TREE_RELOC_OBJECTID
||
4493 key
.type
!= BTRFS_ROOT_ITEM_KEY
)
4496 reloc_root
= btrfs_read_fs_root(root
, &key
);
4497 if (IS_ERR(reloc_root
)) {
4498 err
= PTR_ERR(reloc_root
);
4502 list_add(&reloc_root
->root_list
, &reloc_roots
);
4504 if (btrfs_root_refs(&reloc_root
->root_item
) > 0) {
4505 fs_root
= read_fs_root(fs_info
,
4506 reloc_root
->root_key
.offset
);
4507 if (IS_ERR(fs_root
)) {
4508 ret
= PTR_ERR(fs_root
);
4509 if (ret
!= -ENOENT
) {
4513 ret
= mark_garbage_root(reloc_root
);
4521 if (key
.offset
== 0)
4526 btrfs_release_path(path
);
4528 if (list_empty(&reloc_roots
))
4531 rc
= alloc_reloc_control(fs_info
);
4537 rc
->extent_root
= fs_info
->extent_root
;
4539 set_reloc_control(rc
);
4541 trans
= btrfs_join_transaction(rc
->extent_root
);
4542 if (IS_ERR(trans
)) {
4543 unset_reloc_control(rc
);
4544 err
= PTR_ERR(trans
);
4548 rc
->merge_reloc_tree
= 1;
4550 while (!list_empty(&reloc_roots
)) {
4551 reloc_root
= list_entry(reloc_roots
.next
,
4552 struct btrfs_root
, root_list
);
4553 list_del(&reloc_root
->root_list
);
4555 if (btrfs_root_refs(&reloc_root
->root_item
) == 0) {
4556 list_add_tail(&reloc_root
->root_list
,
4561 fs_root
= read_fs_root(fs_info
, reloc_root
->root_key
.offset
);
4562 if (IS_ERR(fs_root
)) {
4563 err
= PTR_ERR(fs_root
);
4567 err
= __add_reloc_root(reloc_root
);
4568 BUG_ON(err
< 0); /* -ENOMEM or logic error */
4569 fs_root
->reloc_root
= reloc_root
;
4572 err
= btrfs_commit_transaction(trans
, rc
->extent_root
);
4576 merge_reloc_roots(rc
);
4578 unset_reloc_control(rc
);
4580 trans
= btrfs_join_transaction(rc
->extent_root
);
4581 if (IS_ERR(trans
)) {
4582 err
= PTR_ERR(trans
);
4585 err
= btrfs_commit_transaction(trans
, rc
->extent_root
);
4589 if (!list_empty(&reloc_roots
))
4590 free_reloc_roots(&reloc_roots
);
4592 btrfs_free_path(path
);
4595 /* cleanup orphan inode in data relocation tree */
4596 fs_root
= read_fs_root(fs_info
, BTRFS_DATA_RELOC_TREE_OBJECTID
);
4597 if (IS_ERR(fs_root
))
4598 err
= PTR_ERR(fs_root
);
4600 err
= btrfs_orphan_cleanup(fs_root
);
4606 * helper to add ordered checksum for data relocation.
4608 * cloning checksum properly handles the nodatasum extents.
4609 * it also saves CPU time to re-calculate the checksum.
4611 int btrfs_reloc_clone_csums(struct inode
*inode
, u64 file_pos
, u64 len
)
4613 struct btrfs_fs_info
*fs_info
= btrfs_sb(inode
->i_sb
);
4614 struct btrfs_ordered_sum
*sums
;
4615 struct btrfs_ordered_extent
*ordered
;
4621 ordered
= btrfs_lookup_ordered_extent(inode
, file_pos
);
4622 BUG_ON(ordered
->file_offset
!= file_pos
|| ordered
->len
!= len
);
4624 disk_bytenr
= file_pos
+ BTRFS_I(inode
)->index_cnt
;
4625 ret
= btrfs_lookup_csums_range(fs_info
->csum_root
, disk_bytenr
,
4626 disk_bytenr
+ len
- 1, &list
, 0);
4630 while (!list_empty(&list
)) {
4631 sums
= list_entry(list
.next
, struct btrfs_ordered_sum
, list
);
4632 list_del_init(&sums
->list
);
4635 * We need to offset the new_bytenr based on where the csum is.
4636 * We need to do this because we will read in entire prealloc
4637 * extents but we may have written to say the middle of the
4638 * prealloc extent, so we need to make sure the csum goes with
4639 * the right disk offset.
4641 * We can do this because the data reloc inode refers strictly
4642 * to the on disk bytes, so we don't have to worry about
4643 * disk_len vs real len like with real inodes since it's all
4646 new_bytenr
= ordered
->start
+ (sums
->bytenr
- disk_bytenr
);
4647 sums
->bytenr
= new_bytenr
;
4649 btrfs_add_ordered_sum(inode
, ordered
, sums
);
4652 btrfs_put_ordered_extent(ordered
);
4656 int btrfs_reloc_cow_block(struct btrfs_trans_handle
*trans
,
4657 struct btrfs_root
*root
, struct extent_buffer
*buf
,
4658 struct extent_buffer
*cow
)
4660 struct btrfs_fs_info
*fs_info
= root
->fs_info
;
4661 struct reloc_control
*rc
;
4662 struct backref_node
*node
;
4667 rc
= fs_info
->reloc_ctl
;
4671 BUG_ON(rc
->stage
== UPDATE_DATA_PTRS
&&
4672 root
->root_key
.objectid
== BTRFS_DATA_RELOC_TREE_OBJECTID
);
4674 if (root
->root_key
.objectid
== BTRFS_TREE_RELOC_OBJECTID
) {
4675 if (buf
== root
->node
)
4676 __update_reloc_root(root
, cow
->start
);
4679 level
= btrfs_header_level(buf
);
4680 if (btrfs_header_generation(buf
) <=
4681 btrfs_root_last_snapshot(&root
->root_item
))
4684 if (root
->root_key
.objectid
== BTRFS_TREE_RELOC_OBJECTID
&&
4685 rc
->create_reloc_tree
) {
4686 WARN_ON(!first_cow
&& level
== 0);
4688 node
= rc
->backref_cache
.path
[level
];
4689 BUG_ON(node
->bytenr
!= buf
->start
&&
4690 node
->new_bytenr
!= buf
->start
);
4692 drop_node_buffer(node
);
4693 extent_buffer_get(cow
);
4695 node
->new_bytenr
= cow
->start
;
4697 if (!node
->pending
) {
4698 list_move_tail(&node
->list
,
4699 &rc
->backref_cache
.pending
[level
]);
4704 __mark_block_processed(rc
, node
);
4706 if (first_cow
&& level
> 0)
4707 rc
->nodes_relocated
+= buf
->len
;
4710 if (level
== 0 && first_cow
&& rc
->stage
== UPDATE_DATA_PTRS
)
4711 ret
= replace_file_extents(trans
, rc
, root
, cow
);
4716 * called before creating snapshot. it calculates metadata reservation
4717 * required for relocating tree blocks in the snapshot
4719 void btrfs_reloc_pre_snapshot(struct btrfs_pending_snapshot
*pending
,
4720 u64
*bytes_to_reserve
)
4722 struct btrfs_root
*root
;
4723 struct reloc_control
*rc
;
4725 root
= pending
->root
;
4726 if (!root
->reloc_root
)
4729 rc
= root
->fs_info
->reloc_ctl
;
4730 if (!rc
->merge_reloc_tree
)
4733 root
= root
->reloc_root
;
4734 BUG_ON(btrfs_root_refs(&root
->root_item
) == 0);
4736 * relocation is in the stage of merging trees. the space
4737 * used by merging a reloc tree is twice the size of
4738 * relocated tree nodes in the worst case. half for cowing
4739 * the reloc tree, half for cowing the fs tree. the space
4740 * used by cowing the reloc tree will be freed after the
4741 * tree is dropped. if we create snapshot, cowing the fs
4742 * tree may use more space than it frees. so we need
4743 * reserve extra space.
4745 *bytes_to_reserve
+= rc
->nodes_relocated
;
4749 * called after snapshot is created. migrate block reservation
4750 * and create reloc root for the newly created snapshot
4752 int btrfs_reloc_post_snapshot(struct btrfs_trans_handle
*trans
,
4753 struct btrfs_pending_snapshot
*pending
)
4755 struct btrfs_root
*root
= pending
->root
;
4756 struct btrfs_root
*reloc_root
;
4757 struct btrfs_root
*new_root
;
4758 struct reloc_control
*rc
;
4761 if (!root
->reloc_root
)
4764 rc
= root
->fs_info
->reloc_ctl
;
4765 rc
->merging_rsv_size
+= rc
->nodes_relocated
;
4767 if (rc
->merge_reloc_tree
) {
4768 ret
= btrfs_block_rsv_migrate(&pending
->block_rsv
,
4770 rc
->nodes_relocated
, 1);
4775 new_root
= pending
->snap
;
4776 reloc_root
= create_reloc_root(trans
, root
->reloc_root
,
4777 new_root
->root_key
.objectid
);
4778 if (IS_ERR(reloc_root
))
4779 return PTR_ERR(reloc_root
);
4781 ret
= __add_reloc_root(reloc_root
);
4783 new_root
->reloc_root
= reloc_root
;
4785 if (rc
->create_reloc_tree
)
4786 ret
= clone_backref_node(trans
, rc
, root
, reloc_root
);