xref: /openbmc/linux/fs/btrfs/backref.c (revision 155725c9c051a343be5e555bf943da827e6cf721)
1a542ad1bSJan Schmidt /*
2a542ad1bSJan Schmidt  * Copyright (C) 2011 STRATO.  All rights reserved.
3a542ad1bSJan Schmidt  *
4a542ad1bSJan Schmidt  * This program is free software; you can redistribute it and/or
5a542ad1bSJan Schmidt  * modify it under the terms of the GNU General Public
6a542ad1bSJan Schmidt  * License v2 as published by the Free Software Foundation.
7a542ad1bSJan Schmidt  *
8a542ad1bSJan Schmidt  * This program is distributed in the hope that it will be useful,
9a542ad1bSJan Schmidt  * but WITHOUT ANY WARRANTY; without even the implied warranty of
10a542ad1bSJan Schmidt  * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU
11a542ad1bSJan Schmidt  * General Public License for more details.
12a542ad1bSJan Schmidt  *
13a542ad1bSJan Schmidt  * You should have received a copy of the GNU General Public
14a542ad1bSJan Schmidt  * License along with this program; if not, write to the
15a542ad1bSJan Schmidt  * Free Software Foundation, Inc., 59 Temple Place - Suite 330,
16a542ad1bSJan Schmidt  * Boston, MA 021110-1307, USA.
17a542ad1bSJan Schmidt  */
18a542ad1bSJan Schmidt 
19a542ad1bSJan Schmidt #include "ctree.h"
20a542ad1bSJan Schmidt #include "disk-io.h"
21a542ad1bSJan Schmidt #include "backref.h"
228da6d581SJan Schmidt #include "ulist.h"
238da6d581SJan Schmidt #include "transaction.h"
248da6d581SJan Schmidt #include "delayed-ref.h"
25b916a59aSJan Schmidt #include "locking.h"
26a542ad1bSJan Schmidt 
27976b1908SJan Schmidt struct extent_inode_elem {
28976b1908SJan Schmidt 	u64 inum;
29976b1908SJan Schmidt 	u64 offset;
30976b1908SJan Schmidt 	struct extent_inode_elem *next;
31976b1908SJan Schmidt };
32976b1908SJan Schmidt 
33976b1908SJan Schmidt static int check_extent_in_eb(struct btrfs_key *key, struct extent_buffer *eb,
34976b1908SJan Schmidt 				struct btrfs_file_extent_item *fi,
35976b1908SJan Schmidt 				u64 extent_item_pos,
36976b1908SJan Schmidt 				struct extent_inode_elem **eie)
37976b1908SJan Schmidt {
38976b1908SJan Schmidt 	u64 data_offset;
39976b1908SJan Schmidt 	u64 data_len;
40976b1908SJan Schmidt 	struct extent_inode_elem *e;
41976b1908SJan Schmidt 
42976b1908SJan Schmidt 	data_offset = btrfs_file_extent_offset(eb, fi);
43976b1908SJan Schmidt 	data_len = btrfs_file_extent_num_bytes(eb, fi);
44976b1908SJan Schmidt 
45976b1908SJan Schmidt 	if (extent_item_pos < data_offset ||
46976b1908SJan Schmidt 	    extent_item_pos >= data_offset + data_len)
47976b1908SJan Schmidt 		return 1;
48976b1908SJan Schmidt 
49976b1908SJan Schmidt 	e = kmalloc(sizeof(*e), GFP_NOFS);
50976b1908SJan Schmidt 	if (!e)
51976b1908SJan Schmidt 		return -ENOMEM;
52976b1908SJan Schmidt 
53976b1908SJan Schmidt 	e->next = *eie;
54976b1908SJan Schmidt 	e->inum = key->objectid;
55976b1908SJan Schmidt 	e->offset = key->offset + (extent_item_pos - data_offset);
56976b1908SJan Schmidt 	*eie = e;
57976b1908SJan Schmidt 
58976b1908SJan Schmidt 	return 0;
59976b1908SJan Schmidt }
60976b1908SJan Schmidt 
61976b1908SJan Schmidt static int find_extent_in_eb(struct extent_buffer *eb, u64 wanted_disk_byte,
62976b1908SJan Schmidt 				u64 extent_item_pos,
63976b1908SJan Schmidt 				struct extent_inode_elem **eie)
64976b1908SJan Schmidt {
65976b1908SJan Schmidt 	u64 disk_byte;
66976b1908SJan Schmidt 	struct btrfs_key key;
67976b1908SJan Schmidt 	struct btrfs_file_extent_item *fi;
68976b1908SJan Schmidt 	int slot;
69976b1908SJan Schmidt 	int nritems;
70976b1908SJan Schmidt 	int extent_type;
71976b1908SJan Schmidt 	int ret;
72976b1908SJan Schmidt 
73976b1908SJan Schmidt 	/*
74976b1908SJan Schmidt 	 * from the shared data ref, we only have the leaf but we need
75976b1908SJan Schmidt 	 * the key. thus, we must look into all items and see that we
76976b1908SJan Schmidt 	 * find one (some) with a reference to our extent item.
77976b1908SJan Schmidt 	 */
78976b1908SJan Schmidt 	nritems = btrfs_header_nritems(eb);
79976b1908SJan Schmidt 	for (slot = 0; slot < nritems; ++slot) {
80976b1908SJan Schmidt 		btrfs_item_key_to_cpu(eb, &key, slot);
81976b1908SJan Schmidt 		if (key.type != BTRFS_EXTENT_DATA_KEY)
82976b1908SJan Schmidt 			continue;
83976b1908SJan Schmidt 		fi = btrfs_item_ptr(eb, slot, struct btrfs_file_extent_item);
84976b1908SJan Schmidt 		extent_type = btrfs_file_extent_type(eb, fi);
85976b1908SJan Schmidt 		if (extent_type == BTRFS_FILE_EXTENT_INLINE)
86976b1908SJan Schmidt 			continue;
87976b1908SJan Schmidt 		/* don't skip BTRFS_FILE_EXTENT_PREALLOC, we can handle that */
88976b1908SJan Schmidt 		disk_byte = btrfs_file_extent_disk_bytenr(eb, fi);
89976b1908SJan Schmidt 		if (disk_byte != wanted_disk_byte)
90976b1908SJan Schmidt 			continue;
91976b1908SJan Schmidt 
92976b1908SJan Schmidt 		ret = check_extent_in_eb(&key, eb, fi, extent_item_pos, eie);
93976b1908SJan Schmidt 		if (ret < 0)
94976b1908SJan Schmidt 			return ret;
95976b1908SJan Schmidt 	}
96976b1908SJan Schmidt 
97976b1908SJan Schmidt 	return 0;
98976b1908SJan Schmidt }
99976b1908SJan Schmidt 
1008da6d581SJan Schmidt /*
1018da6d581SJan Schmidt  * this structure records all encountered refs on the way up to the root
1028da6d581SJan Schmidt  */
1038da6d581SJan Schmidt struct __prelim_ref {
1048da6d581SJan Schmidt 	struct list_head list;
1058da6d581SJan Schmidt 	u64 root_id;
106d5c88b73SJan Schmidt 	struct btrfs_key key_for_search;
1078da6d581SJan Schmidt 	int level;
1088da6d581SJan Schmidt 	int count;
1093301958bSJan Schmidt 	struct extent_inode_elem *inode_list;
1108da6d581SJan Schmidt 	u64 parent;
1118da6d581SJan Schmidt 	u64 wanted_disk_byte;
1128da6d581SJan Schmidt };
1138da6d581SJan Schmidt 
114d5c88b73SJan Schmidt /*
115d5c88b73SJan Schmidt  * the rules for all callers of this function are:
116d5c88b73SJan Schmidt  * - obtaining the parent is the goal
117d5c88b73SJan Schmidt  * - if you add a key, you must know that it is a correct key
118d5c88b73SJan Schmidt  * - if you cannot add the parent or a correct key, then we will look into the
119d5c88b73SJan Schmidt  *   block later to set a correct key
120d5c88b73SJan Schmidt  *
121d5c88b73SJan Schmidt  * delayed refs
122d5c88b73SJan Schmidt  * ============
123d5c88b73SJan Schmidt  *        backref type | shared | indirect | shared | indirect
124d5c88b73SJan Schmidt  * information         |   tree |     tree |   data |     data
125d5c88b73SJan Schmidt  * --------------------+--------+----------+--------+----------
126d5c88b73SJan Schmidt  *      parent logical |    y   |     -    |    -   |     -
127d5c88b73SJan Schmidt  *      key to resolve |    -   |     y    |    y   |     y
128d5c88b73SJan Schmidt  *  tree block logical |    -   |     -    |    -   |     -
129d5c88b73SJan Schmidt  *  root for resolving |    y   |     y    |    y   |     y
130d5c88b73SJan Schmidt  *
131d5c88b73SJan Schmidt  * - column 1:       we've the parent -> done
132d5c88b73SJan Schmidt  * - column 2, 3, 4: we use the key to find the parent
133d5c88b73SJan Schmidt  *
134d5c88b73SJan Schmidt  * on disk refs (inline or keyed)
135d5c88b73SJan Schmidt  * ==============================
136d5c88b73SJan Schmidt  *        backref type | shared | indirect | shared | indirect
137d5c88b73SJan Schmidt  * information         |   tree |     tree |   data |     data
138d5c88b73SJan Schmidt  * --------------------+--------+----------+--------+----------
139d5c88b73SJan Schmidt  *      parent logical |    y   |     -    |    y   |     -
140d5c88b73SJan Schmidt  *      key to resolve |    -   |     -    |    -   |     y
141d5c88b73SJan Schmidt  *  tree block logical |    y   |     y    |    y   |     y
142d5c88b73SJan Schmidt  *  root for resolving |    -   |     y    |    y   |     y
143d5c88b73SJan Schmidt  *
144d5c88b73SJan Schmidt  * - column 1, 3: we've the parent -> done
145d5c88b73SJan Schmidt  * - column 2:    we take the first key from the block to find the parent
146d5c88b73SJan Schmidt  *                (see __add_missing_keys)
147d5c88b73SJan Schmidt  * - column 4:    we use the key to find the parent
148d5c88b73SJan Schmidt  *
149d5c88b73SJan Schmidt  * additional information that's available but not required to find the parent
150d5c88b73SJan Schmidt  * block might help in merging entries to gain some speed.
151d5c88b73SJan Schmidt  */
152d5c88b73SJan Schmidt 
1538da6d581SJan Schmidt static int __add_prelim_ref(struct list_head *head, u64 root_id,
154d5c88b73SJan Schmidt 			    struct btrfs_key *key, int level,
155d5c88b73SJan Schmidt 			    u64 parent, u64 wanted_disk_byte, int count)
1568da6d581SJan Schmidt {
1578da6d581SJan Schmidt 	struct __prelim_ref *ref;
1588da6d581SJan Schmidt 
1598da6d581SJan Schmidt 	/* in case we're adding delayed refs, we're holding the refs spinlock */
1608da6d581SJan Schmidt 	ref = kmalloc(sizeof(*ref), GFP_ATOMIC);
1618da6d581SJan Schmidt 	if (!ref)
1628da6d581SJan Schmidt 		return -ENOMEM;
1638da6d581SJan Schmidt 
1648da6d581SJan Schmidt 	ref->root_id = root_id;
1658da6d581SJan Schmidt 	if (key)
166d5c88b73SJan Schmidt 		ref->key_for_search = *key;
1678da6d581SJan Schmidt 	else
168d5c88b73SJan Schmidt 		memset(&ref->key_for_search, 0, sizeof(ref->key_for_search));
1698da6d581SJan Schmidt 
1703301958bSJan Schmidt 	ref->inode_list = NULL;
1718da6d581SJan Schmidt 	ref->level = level;
1728da6d581SJan Schmidt 	ref->count = count;
1738da6d581SJan Schmidt 	ref->parent = parent;
1748da6d581SJan Schmidt 	ref->wanted_disk_byte = wanted_disk_byte;
1758da6d581SJan Schmidt 	list_add_tail(&ref->list, head);
1768da6d581SJan Schmidt 
1778da6d581SJan Schmidt 	return 0;
1788da6d581SJan Schmidt }
1798da6d581SJan Schmidt 
1808da6d581SJan Schmidt static int add_all_parents(struct btrfs_root *root, struct btrfs_path *path,
181976b1908SJan Schmidt 				struct ulist *parents, int level,
18269bca40dSAlexander Block 				struct btrfs_key *key_for_search, u64 time_seq,
1833d7806ecSJan Schmidt 				u64 wanted_disk_byte,
184976b1908SJan Schmidt 				const u64 *extent_item_pos)
1858da6d581SJan Schmidt {
18669bca40dSAlexander Block 	int ret = 0;
18769bca40dSAlexander Block 	int slot;
18869bca40dSAlexander Block 	struct extent_buffer *eb;
18969bca40dSAlexander Block 	struct btrfs_key key;
1908da6d581SJan Schmidt 	struct btrfs_file_extent_item *fi;
1913301958bSJan Schmidt 	struct extent_inode_elem *eie = NULL;
1928da6d581SJan Schmidt 	u64 disk_byte;
1938da6d581SJan Schmidt 
19469bca40dSAlexander Block 	if (level != 0) {
19569bca40dSAlexander Block 		eb = path->nodes[level];
19669bca40dSAlexander Block 		ret = ulist_add(parents, eb->start, 0, GFP_NOFS);
1973301958bSJan Schmidt 		if (ret < 0)
1983301958bSJan Schmidt 			return ret;
1998da6d581SJan Schmidt 		return 0;
20069bca40dSAlexander Block 	}
2018da6d581SJan Schmidt 
2028da6d581SJan Schmidt 	/*
20369bca40dSAlexander Block 	 * We normally enter this function with the path already pointing to
20469bca40dSAlexander Block 	 * the first item to check. But sometimes, we may enter it with
20569bca40dSAlexander Block 	 * slot==nritems. In that case, go to the next leaf before we continue.
2068da6d581SJan Schmidt 	 */
20769bca40dSAlexander Block 	if (path->slots[0] >= btrfs_header_nritems(path->nodes[0]))
2083d7806ecSJan Schmidt 		ret = btrfs_next_old_leaf(root, path, time_seq);
2098da6d581SJan Schmidt 
21069bca40dSAlexander Block 	while (!ret) {
2118da6d581SJan Schmidt 		eb = path->nodes[0];
21269bca40dSAlexander Block 		slot = path->slots[0];
21369bca40dSAlexander Block 
21469bca40dSAlexander Block 		btrfs_item_key_to_cpu(eb, &key, slot);
21569bca40dSAlexander Block 
21669bca40dSAlexander Block 		if (key.objectid != key_for_search->objectid ||
21769bca40dSAlexander Block 		    key.type != BTRFS_EXTENT_DATA_KEY)
21869bca40dSAlexander Block 			break;
21969bca40dSAlexander Block 
22069bca40dSAlexander Block 		fi = btrfs_item_ptr(eb, slot, struct btrfs_file_extent_item);
2218da6d581SJan Schmidt 		disk_byte = btrfs_file_extent_disk_bytenr(eb, fi);
22269bca40dSAlexander Block 
22369bca40dSAlexander Block 		if (disk_byte == wanted_disk_byte) {
22469bca40dSAlexander Block 			eie = NULL;
22569bca40dSAlexander Block 			if (extent_item_pos) {
22669bca40dSAlexander Block 				ret = check_extent_in_eb(&key, eb, fi,
22769bca40dSAlexander Block 						*extent_item_pos,
22869bca40dSAlexander Block 						&eie);
22969bca40dSAlexander Block 				if (ret < 0)
23069bca40dSAlexander Block 					break;
2318da6d581SJan Schmidt 			}
23269bca40dSAlexander Block 			if (!ret) {
23369bca40dSAlexander Block 				ret = ulist_add(parents, eb->start,
23469bca40dSAlexander Block 						(unsigned long)eie, GFP_NOFS);
23569bca40dSAlexander Block 				if (ret < 0)
23669bca40dSAlexander Block 					break;
23769bca40dSAlexander Block 				if (!extent_item_pos) {
23869bca40dSAlexander Block 					ret = btrfs_next_old_leaf(root, path,
23969bca40dSAlexander Block 							time_seq);
24069bca40dSAlexander Block 					continue;
24169bca40dSAlexander Block 				}
24269bca40dSAlexander Block 			}
24369bca40dSAlexander Block 		}
24469bca40dSAlexander Block 		ret = btrfs_next_old_item(root, path, time_seq);
2458da6d581SJan Schmidt 	}
2468da6d581SJan Schmidt 
24769bca40dSAlexander Block 	if (ret > 0)
24869bca40dSAlexander Block 		ret = 0;
24969bca40dSAlexander Block 	return ret;
2508da6d581SJan Schmidt }
2518da6d581SJan Schmidt 
2528da6d581SJan Schmidt /*
2538da6d581SJan Schmidt  * resolve an indirect backref in the form (root_id, key, level)
2548da6d581SJan Schmidt  * to a logical address
2558da6d581SJan Schmidt  */
2568da6d581SJan Schmidt static int __resolve_indirect_ref(struct btrfs_fs_info *fs_info,
2577a3ae2f8SJan Schmidt 					int search_commit_root,
2588445f61cSJan Schmidt 					u64 time_seq,
2598da6d581SJan Schmidt 					struct __prelim_ref *ref,
260976b1908SJan Schmidt 					struct ulist *parents,
261976b1908SJan Schmidt 					const u64 *extent_item_pos)
2628da6d581SJan Schmidt {
2638da6d581SJan Schmidt 	struct btrfs_path *path;
2648da6d581SJan Schmidt 	struct btrfs_root *root;
2658da6d581SJan Schmidt 	struct btrfs_key root_key;
2668da6d581SJan Schmidt 	struct extent_buffer *eb;
2678da6d581SJan Schmidt 	int ret = 0;
2688da6d581SJan Schmidt 	int root_level;
2698da6d581SJan Schmidt 	int level = ref->level;
2708da6d581SJan Schmidt 
2718da6d581SJan Schmidt 	path = btrfs_alloc_path();
2728da6d581SJan Schmidt 	if (!path)
2738da6d581SJan Schmidt 		return -ENOMEM;
2747a3ae2f8SJan Schmidt 	path->search_commit_root = !!search_commit_root;
2758da6d581SJan Schmidt 
2768da6d581SJan Schmidt 	root_key.objectid = ref->root_id;
2778da6d581SJan Schmidt 	root_key.type = BTRFS_ROOT_ITEM_KEY;
2788da6d581SJan Schmidt 	root_key.offset = (u64)-1;
2798da6d581SJan Schmidt 	root = btrfs_read_fs_root_no_name(fs_info, &root_key);
2808da6d581SJan Schmidt 	if (IS_ERR(root)) {
2818da6d581SJan Schmidt 		ret = PTR_ERR(root);
2828da6d581SJan Schmidt 		goto out;
2838da6d581SJan Schmidt 	}
2848da6d581SJan Schmidt 
2858da6d581SJan Schmidt 	rcu_read_lock();
2868da6d581SJan Schmidt 	root_level = btrfs_header_level(root->node);
2878da6d581SJan Schmidt 	rcu_read_unlock();
2888da6d581SJan Schmidt 
2898da6d581SJan Schmidt 	if (root_level + 1 == level)
2908da6d581SJan Schmidt 		goto out;
2918da6d581SJan Schmidt 
2928da6d581SJan Schmidt 	path->lowest_level = level;
2938445f61cSJan Schmidt 	ret = btrfs_search_old_slot(root, &ref->key_for_search, path, time_seq);
2948da6d581SJan Schmidt 	pr_debug("search slot in root %llu (level %d, ref count %d) returned "
2958da6d581SJan Schmidt 		 "%d for key (%llu %u %llu)\n",
2968da6d581SJan Schmidt 		 (unsigned long long)ref->root_id, level, ref->count, ret,
297d5c88b73SJan Schmidt 		 (unsigned long long)ref->key_for_search.objectid,
298d5c88b73SJan Schmidt 		 ref->key_for_search.type,
299d5c88b73SJan Schmidt 		 (unsigned long long)ref->key_for_search.offset);
3008da6d581SJan Schmidt 	if (ret < 0)
3018da6d581SJan Schmidt 		goto out;
3028da6d581SJan Schmidt 
3038da6d581SJan Schmidt 	eb = path->nodes[level];
3049345457fSJan Schmidt 	while (!eb) {
3059345457fSJan Schmidt 		if (!level) {
3068da6d581SJan Schmidt 			WARN_ON(1);
3078da6d581SJan Schmidt 			ret = 1;
3088da6d581SJan Schmidt 			goto out;
3098da6d581SJan Schmidt 		}
3109345457fSJan Schmidt 		level--;
3119345457fSJan Schmidt 		eb = path->nodes[level];
3129345457fSJan Schmidt 	}
3138da6d581SJan Schmidt 
31469bca40dSAlexander Block 	ret = add_all_parents(root, path, parents, level, &ref->key_for_search,
31569bca40dSAlexander Block 				time_seq, ref->wanted_disk_byte,
31669bca40dSAlexander Block 				extent_item_pos);
3178da6d581SJan Schmidt out:
3188da6d581SJan Schmidt 	btrfs_free_path(path);
3198da6d581SJan Schmidt 	return ret;
3208da6d581SJan Schmidt }
3218da6d581SJan Schmidt 
3228da6d581SJan Schmidt /*
3238da6d581SJan Schmidt  * resolve all indirect backrefs from the list
3248da6d581SJan Schmidt  */
3258da6d581SJan Schmidt static int __resolve_indirect_refs(struct btrfs_fs_info *fs_info,
3268445f61cSJan Schmidt 				   int search_commit_root, u64 time_seq,
327976b1908SJan Schmidt 				   struct list_head *head,
328976b1908SJan Schmidt 				   const u64 *extent_item_pos)
3298da6d581SJan Schmidt {
3308da6d581SJan Schmidt 	int err;
3318da6d581SJan Schmidt 	int ret = 0;
3328da6d581SJan Schmidt 	struct __prelim_ref *ref;
3338da6d581SJan Schmidt 	struct __prelim_ref *ref_safe;
3348da6d581SJan Schmidt 	struct __prelim_ref *new_ref;
3358da6d581SJan Schmidt 	struct ulist *parents;
3368da6d581SJan Schmidt 	struct ulist_node *node;
337cd1b413cSJan Schmidt 	struct ulist_iterator uiter;
3388da6d581SJan Schmidt 
3398da6d581SJan Schmidt 	parents = ulist_alloc(GFP_NOFS);
3408da6d581SJan Schmidt 	if (!parents)
3418da6d581SJan Schmidt 		return -ENOMEM;
3428da6d581SJan Schmidt 
3438da6d581SJan Schmidt 	/*
3448da6d581SJan Schmidt 	 * _safe allows us to insert directly after the current item without
3458da6d581SJan Schmidt 	 * iterating over the newly inserted items.
3468da6d581SJan Schmidt 	 * we're also allowed to re-assign ref during iteration.
3478da6d581SJan Schmidt 	 */
3488da6d581SJan Schmidt 	list_for_each_entry_safe(ref, ref_safe, head, list) {
3498da6d581SJan Schmidt 		if (ref->parent)	/* already direct */
3508da6d581SJan Schmidt 			continue;
3518da6d581SJan Schmidt 		if (ref->count == 0)
3528da6d581SJan Schmidt 			continue;
3537a3ae2f8SJan Schmidt 		err = __resolve_indirect_ref(fs_info, search_commit_root,
3548445f61cSJan Schmidt 					     time_seq, ref, parents,
3558445f61cSJan Schmidt 					     extent_item_pos);
3568da6d581SJan Schmidt 		if (err) {
3578da6d581SJan Schmidt 			if (ret == 0)
3588da6d581SJan Schmidt 				ret = err;
3598da6d581SJan Schmidt 			continue;
3608da6d581SJan Schmidt 		}
3618da6d581SJan Schmidt 
3628da6d581SJan Schmidt 		/* we put the first parent into the ref at hand */
363cd1b413cSJan Schmidt 		ULIST_ITER_INIT(&uiter);
364cd1b413cSJan Schmidt 		node = ulist_next(parents, &uiter);
3658da6d581SJan Schmidt 		ref->parent = node ? node->val : 0;
3663301958bSJan Schmidt 		ref->inode_list =
3673301958bSJan Schmidt 			node ? (struct extent_inode_elem *)node->aux : 0;
3688da6d581SJan Schmidt 
3698da6d581SJan Schmidt 		/* additional parents require new refs being added here */
370cd1b413cSJan Schmidt 		while ((node = ulist_next(parents, &uiter))) {
3718da6d581SJan Schmidt 			new_ref = kmalloc(sizeof(*new_ref), GFP_NOFS);
3728da6d581SJan Schmidt 			if (!new_ref) {
3738da6d581SJan Schmidt 				ret = -ENOMEM;
3748da6d581SJan Schmidt 				break;
3758da6d581SJan Schmidt 			}
3768da6d581SJan Schmidt 			memcpy(new_ref, ref, sizeof(*ref));
3778da6d581SJan Schmidt 			new_ref->parent = node->val;
3783301958bSJan Schmidt 			new_ref->inode_list =
3793301958bSJan Schmidt 					(struct extent_inode_elem *)node->aux;
3808da6d581SJan Schmidt 			list_add(&new_ref->list, &ref->list);
3818da6d581SJan Schmidt 		}
3828da6d581SJan Schmidt 		ulist_reinit(parents);
3838da6d581SJan Schmidt 	}
3848da6d581SJan Schmidt 
3858da6d581SJan Schmidt 	ulist_free(parents);
3868da6d581SJan Schmidt 	return ret;
3878da6d581SJan Schmidt }
3888da6d581SJan Schmidt 
389d5c88b73SJan Schmidt static inline int ref_for_same_block(struct __prelim_ref *ref1,
390d5c88b73SJan Schmidt 				     struct __prelim_ref *ref2)
391d5c88b73SJan Schmidt {
392d5c88b73SJan Schmidt 	if (ref1->level != ref2->level)
393d5c88b73SJan Schmidt 		return 0;
394d5c88b73SJan Schmidt 	if (ref1->root_id != ref2->root_id)
395d5c88b73SJan Schmidt 		return 0;
396d5c88b73SJan Schmidt 	if (ref1->key_for_search.type != ref2->key_for_search.type)
397d5c88b73SJan Schmidt 		return 0;
398d5c88b73SJan Schmidt 	if (ref1->key_for_search.objectid != ref2->key_for_search.objectid)
399d5c88b73SJan Schmidt 		return 0;
400d5c88b73SJan Schmidt 	if (ref1->key_for_search.offset != ref2->key_for_search.offset)
401d5c88b73SJan Schmidt 		return 0;
402d5c88b73SJan Schmidt 	if (ref1->parent != ref2->parent)
403d5c88b73SJan Schmidt 		return 0;
404d5c88b73SJan Schmidt 
405d5c88b73SJan Schmidt 	return 1;
406d5c88b73SJan Schmidt }
407d5c88b73SJan Schmidt 
408d5c88b73SJan Schmidt /*
409d5c88b73SJan Schmidt  * read tree blocks and add keys where required.
410d5c88b73SJan Schmidt  */
411d5c88b73SJan Schmidt static int __add_missing_keys(struct btrfs_fs_info *fs_info,
412d5c88b73SJan Schmidt 			      struct list_head *head)
413d5c88b73SJan Schmidt {
414d5c88b73SJan Schmidt 	struct list_head *pos;
415d5c88b73SJan Schmidt 	struct extent_buffer *eb;
416d5c88b73SJan Schmidt 
417d5c88b73SJan Schmidt 	list_for_each(pos, head) {
418d5c88b73SJan Schmidt 		struct __prelim_ref *ref;
419d5c88b73SJan Schmidt 		ref = list_entry(pos, struct __prelim_ref, list);
420d5c88b73SJan Schmidt 
421d5c88b73SJan Schmidt 		if (ref->parent)
422d5c88b73SJan Schmidt 			continue;
423d5c88b73SJan Schmidt 		if (ref->key_for_search.type)
424d5c88b73SJan Schmidt 			continue;
425d5c88b73SJan Schmidt 		BUG_ON(!ref->wanted_disk_byte);
426d5c88b73SJan Schmidt 		eb = read_tree_block(fs_info->tree_root, ref->wanted_disk_byte,
427d5c88b73SJan Schmidt 				     fs_info->tree_root->leafsize, 0);
428d5c88b73SJan Schmidt 		BUG_ON(!eb);
429d5c88b73SJan Schmidt 		btrfs_tree_read_lock(eb);
430d5c88b73SJan Schmidt 		if (btrfs_header_level(eb) == 0)
431d5c88b73SJan Schmidt 			btrfs_item_key_to_cpu(eb, &ref->key_for_search, 0);
432d5c88b73SJan Schmidt 		else
433d5c88b73SJan Schmidt 			btrfs_node_key_to_cpu(eb, &ref->key_for_search, 0);
434d5c88b73SJan Schmidt 		btrfs_tree_read_unlock(eb);
435d5c88b73SJan Schmidt 		free_extent_buffer(eb);
436d5c88b73SJan Schmidt 	}
437d5c88b73SJan Schmidt 	return 0;
438d5c88b73SJan Schmidt }
439d5c88b73SJan Schmidt 
4408da6d581SJan Schmidt /*
4418da6d581SJan Schmidt  * merge two lists of backrefs and adjust counts accordingly
4428da6d581SJan Schmidt  *
4438da6d581SJan Schmidt  * mode = 1: merge identical keys, if key is set
444d5c88b73SJan Schmidt  *    FIXME: if we add more keys in __add_prelim_ref, we can merge more here.
445d5c88b73SJan Schmidt  *           additionally, we could even add a key range for the blocks we
446d5c88b73SJan Schmidt  *           looked into to merge even more (-> replace unresolved refs by those
447d5c88b73SJan Schmidt  *           having a parent).
4488da6d581SJan Schmidt  * mode = 2: merge identical parents
4498da6d581SJan Schmidt  */
4508da6d581SJan Schmidt static int __merge_refs(struct list_head *head, int mode)
4518da6d581SJan Schmidt {
4528da6d581SJan Schmidt 	struct list_head *pos1;
4538da6d581SJan Schmidt 
4548da6d581SJan Schmidt 	list_for_each(pos1, head) {
4558da6d581SJan Schmidt 		struct list_head *n2;
4568da6d581SJan Schmidt 		struct list_head *pos2;
4578da6d581SJan Schmidt 		struct __prelim_ref *ref1;
4588da6d581SJan Schmidt 
4598da6d581SJan Schmidt 		ref1 = list_entry(pos1, struct __prelim_ref, list);
4608da6d581SJan Schmidt 
4618da6d581SJan Schmidt 		for (pos2 = pos1->next, n2 = pos2->next; pos2 != head;
4628da6d581SJan Schmidt 		     pos2 = n2, n2 = pos2->next) {
4638da6d581SJan Schmidt 			struct __prelim_ref *ref2;
464d5c88b73SJan Schmidt 			struct __prelim_ref *xchg;
4658da6d581SJan Schmidt 
4668da6d581SJan Schmidt 			ref2 = list_entry(pos2, struct __prelim_ref, list);
4678da6d581SJan Schmidt 
4688da6d581SJan Schmidt 			if (mode == 1) {
469d5c88b73SJan Schmidt 				if (!ref_for_same_block(ref1, ref2))
4708da6d581SJan Schmidt 					continue;
471d5c88b73SJan Schmidt 				if (!ref1->parent && ref2->parent) {
472d5c88b73SJan Schmidt 					xchg = ref1;
473d5c88b73SJan Schmidt 					ref1 = ref2;
474d5c88b73SJan Schmidt 					ref2 = xchg;
475d5c88b73SJan Schmidt 				}
4768da6d581SJan Schmidt 				ref1->count += ref2->count;
4778da6d581SJan Schmidt 			} else {
4788da6d581SJan Schmidt 				if (ref1->parent != ref2->parent)
4798da6d581SJan Schmidt 					continue;
4808da6d581SJan Schmidt 				ref1->count += ref2->count;
4818da6d581SJan Schmidt 			}
4828da6d581SJan Schmidt 			list_del(&ref2->list);
4838da6d581SJan Schmidt 			kfree(ref2);
4848da6d581SJan Schmidt 		}
4858da6d581SJan Schmidt 
4868da6d581SJan Schmidt 	}
4878da6d581SJan Schmidt 	return 0;
4888da6d581SJan Schmidt }
4898da6d581SJan Schmidt 
4908da6d581SJan Schmidt /*
4918da6d581SJan Schmidt  * add all currently queued delayed refs from this head whose seq nr is
4928da6d581SJan Schmidt  * smaller or equal that seq to the list
4938da6d581SJan Schmidt  */
4948da6d581SJan Schmidt static int __add_delayed_refs(struct btrfs_delayed_ref_head *head, u64 seq,
4958da6d581SJan Schmidt 			      struct list_head *prefs)
4968da6d581SJan Schmidt {
4978da6d581SJan Schmidt 	struct btrfs_delayed_extent_op *extent_op = head->extent_op;
4988da6d581SJan Schmidt 	struct rb_node *n = &head->node.rb_node;
499d5c88b73SJan Schmidt 	struct btrfs_key key;
500d5c88b73SJan Schmidt 	struct btrfs_key op_key = {0};
5018da6d581SJan Schmidt 	int sgn;
502b1375d64SJan Schmidt 	int ret = 0;
5038da6d581SJan Schmidt 
5048da6d581SJan Schmidt 	if (extent_op && extent_op->update_key)
505d5c88b73SJan Schmidt 		btrfs_disk_key_to_cpu(&op_key, &extent_op->key);
5068da6d581SJan Schmidt 
5078da6d581SJan Schmidt 	while ((n = rb_prev(n))) {
5088da6d581SJan Schmidt 		struct btrfs_delayed_ref_node *node;
5098da6d581SJan Schmidt 		node = rb_entry(n, struct btrfs_delayed_ref_node,
5108da6d581SJan Schmidt 				rb_node);
5118da6d581SJan Schmidt 		if (node->bytenr != head->node.bytenr)
5128da6d581SJan Schmidt 			break;
5138da6d581SJan Schmidt 		WARN_ON(node->is_head);
5148da6d581SJan Schmidt 
5158da6d581SJan Schmidt 		if (node->seq > seq)
5168da6d581SJan Schmidt 			continue;
5178da6d581SJan Schmidt 
5188da6d581SJan Schmidt 		switch (node->action) {
5198da6d581SJan Schmidt 		case BTRFS_ADD_DELAYED_EXTENT:
5208da6d581SJan Schmidt 		case BTRFS_UPDATE_DELAYED_HEAD:
5218da6d581SJan Schmidt 			WARN_ON(1);
5228da6d581SJan Schmidt 			continue;
5238da6d581SJan Schmidt 		case BTRFS_ADD_DELAYED_REF:
5248da6d581SJan Schmidt 			sgn = 1;
5258da6d581SJan Schmidt 			break;
5268da6d581SJan Schmidt 		case BTRFS_DROP_DELAYED_REF:
5278da6d581SJan Schmidt 			sgn = -1;
5288da6d581SJan Schmidt 			break;
5298da6d581SJan Schmidt 		default:
5308da6d581SJan Schmidt 			BUG_ON(1);
5318da6d581SJan Schmidt 		}
5328da6d581SJan Schmidt 		switch (node->type) {
5338da6d581SJan Schmidt 		case BTRFS_TREE_BLOCK_REF_KEY: {
5348da6d581SJan Schmidt 			struct btrfs_delayed_tree_ref *ref;
5358da6d581SJan Schmidt 
5368da6d581SJan Schmidt 			ref = btrfs_delayed_node_to_tree_ref(node);
537d5c88b73SJan Schmidt 			ret = __add_prelim_ref(prefs, ref->root, &op_key,
5388da6d581SJan Schmidt 					       ref->level + 1, 0, node->bytenr,
5398da6d581SJan Schmidt 					       node->ref_mod * sgn);
5408da6d581SJan Schmidt 			break;
5418da6d581SJan Schmidt 		}
5428da6d581SJan Schmidt 		case BTRFS_SHARED_BLOCK_REF_KEY: {
5438da6d581SJan Schmidt 			struct btrfs_delayed_tree_ref *ref;
5448da6d581SJan Schmidt 
5458da6d581SJan Schmidt 			ref = btrfs_delayed_node_to_tree_ref(node);
546d5c88b73SJan Schmidt 			ret = __add_prelim_ref(prefs, ref->root, NULL,
5478da6d581SJan Schmidt 					       ref->level + 1, ref->parent,
5488da6d581SJan Schmidt 					       node->bytenr,
5498da6d581SJan Schmidt 					       node->ref_mod * sgn);
5508da6d581SJan Schmidt 			break;
5518da6d581SJan Schmidt 		}
5528da6d581SJan Schmidt 		case BTRFS_EXTENT_DATA_REF_KEY: {
5538da6d581SJan Schmidt 			struct btrfs_delayed_data_ref *ref;
5548da6d581SJan Schmidt 			ref = btrfs_delayed_node_to_data_ref(node);
5558da6d581SJan Schmidt 
5568da6d581SJan Schmidt 			key.objectid = ref->objectid;
5578da6d581SJan Schmidt 			key.type = BTRFS_EXTENT_DATA_KEY;
5588da6d581SJan Schmidt 			key.offset = ref->offset;
5598da6d581SJan Schmidt 			ret = __add_prelim_ref(prefs, ref->root, &key, 0, 0,
5608da6d581SJan Schmidt 					       node->bytenr,
5618da6d581SJan Schmidt 					       node->ref_mod * sgn);
5628da6d581SJan Schmidt 			break;
5638da6d581SJan Schmidt 		}
5648da6d581SJan Schmidt 		case BTRFS_SHARED_DATA_REF_KEY: {
5658da6d581SJan Schmidt 			struct btrfs_delayed_data_ref *ref;
5668da6d581SJan Schmidt 
5678da6d581SJan Schmidt 			ref = btrfs_delayed_node_to_data_ref(node);
5688da6d581SJan Schmidt 
5698da6d581SJan Schmidt 			key.objectid = ref->objectid;
5708da6d581SJan Schmidt 			key.type = BTRFS_EXTENT_DATA_KEY;
5718da6d581SJan Schmidt 			key.offset = ref->offset;
5728da6d581SJan Schmidt 			ret = __add_prelim_ref(prefs, ref->root, &key, 0,
5738da6d581SJan Schmidt 					       ref->parent, node->bytenr,
5748da6d581SJan Schmidt 					       node->ref_mod * sgn);
5758da6d581SJan Schmidt 			break;
5768da6d581SJan Schmidt 		}
5778da6d581SJan Schmidt 		default:
5788da6d581SJan Schmidt 			WARN_ON(1);
5798da6d581SJan Schmidt 		}
5808da6d581SJan Schmidt 		BUG_ON(ret);
5818da6d581SJan Schmidt 	}
5828da6d581SJan Schmidt 
5838da6d581SJan Schmidt 	return 0;
5848da6d581SJan Schmidt }
5858da6d581SJan Schmidt 
5868da6d581SJan Schmidt /*
5878da6d581SJan Schmidt  * add all inline backrefs for bytenr to the list
5888da6d581SJan Schmidt  */
5898da6d581SJan Schmidt static int __add_inline_refs(struct btrfs_fs_info *fs_info,
5908da6d581SJan Schmidt 			     struct btrfs_path *path, u64 bytenr,
591d5c88b73SJan Schmidt 			     int *info_level, struct list_head *prefs)
5928da6d581SJan Schmidt {
593b1375d64SJan Schmidt 	int ret = 0;
5948da6d581SJan Schmidt 	int slot;
5958da6d581SJan Schmidt 	struct extent_buffer *leaf;
5968da6d581SJan Schmidt 	struct btrfs_key key;
5978da6d581SJan Schmidt 	unsigned long ptr;
5988da6d581SJan Schmidt 	unsigned long end;
5998da6d581SJan Schmidt 	struct btrfs_extent_item *ei;
6008da6d581SJan Schmidt 	u64 flags;
6018da6d581SJan Schmidt 	u64 item_size;
6028da6d581SJan Schmidt 
6038da6d581SJan Schmidt 	/*
6048da6d581SJan Schmidt 	 * enumerate all inline refs
6058da6d581SJan Schmidt 	 */
6068da6d581SJan Schmidt 	leaf = path->nodes[0];
607dadcaf78SJan Schmidt 	slot = path->slots[0];
6088da6d581SJan Schmidt 
6098da6d581SJan Schmidt 	item_size = btrfs_item_size_nr(leaf, slot);
6108da6d581SJan Schmidt 	BUG_ON(item_size < sizeof(*ei));
6118da6d581SJan Schmidt 
6128da6d581SJan Schmidt 	ei = btrfs_item_ptr(leaf, slot, struct btrfs_extent_item);
6138da6d581SJan Schmidt 	flags = btrfs_extent_flags(leaf, ei);
6148da6d581SJan Schmidt 
6158da6d581SJan Schmidt 	ptr = (unsigned long)(ei + 1);
6168da6d581SJan Schmidt 	end = (unsigned long)ei + item_size;
6178da6d581SJan Schmidt 
6188da6d581SJan Schmidt 	if (flags & BTRFS_EXTENT_FLAG_TREE_BLOCK) {
6198da6d581SJan Schmidt 		struct btrfs_tree_block_info *info;
6208da6d581SJan Schmidt 
6218da6d581SJan Schmidt 		info = (struct btrfs_tree_block_info *)ptr;
6228da6d581SJan Schmidt 		*info_level = btrfs_tree_block_level(leaf, info);
6238da6d581SJan Schmidt 		ptr += sizeof(struct btrfs_tree_block_info);
6248da6d581SJan Schmidt 		BUG_ON(ptr > end);
6258da6d581SJan Schmidt 	} else {
6268da6d581SJan Schmidt 		BUG_ON(!(flags & BTRFS_EXTENT_FLAG_DATA));
6278da6d581SJan Schmidt 	}
6288da6d581SJan Schmidt 
6298da6d581SJan Schmidt 	while (ptr < end) {
6308da6d581SJan Schmidt 		struct btrfs_extent_inline_ref *iref;
6318da6d581SJan Schmidt 		u64 offset;
6328da6d581SJan Schmidt 		int type;
6338da6d581SJan Schmidt 
6348da6d581SJan Schmidt 		iref = (struct btrfs_extent_inline_ref *)ptr;
6358da6d581SJan Schmidt 		type = btrfs_extent_inline_ref_type(leaf, iref);
6368da6d581SJan Schmidt 		offset = btrfs_extent_inline_ref_offset(leaf, iref);
6378da6d581SJan Schmidt 
6388da6d581SJan Schmidt 		switch (type) {
6398da6d581SJan Schmidt 		case BTRFS_SHARED_BLOCK_REF_KEY:
640d5c88b73SJan Schmidt 			ret = __add_prelim_ref(prefs, 0, NULL,
6418da6d581SJan Schmidt 						*info_level + 1, offset,
6428da6d581SJan Schmidt 						bytenr, 1);
6438da6d581SJan Schmidt 			break;
6448da6d581SJan Schmidt 		case BTRFS_SHARED_DATA_REF_KEY: {
6458da6d581SJan Schmidt 			struct btrfs_shared_data_ref *sdref;
6468da6d581SJan Schmidt 			int count;
6478da6d581SJan Schmidt 
6488da6d581SJan Schmidt 			sdref = (struct btrfs_shared_data_ref *)(iref + 1);
6498da6d581SJan Schmidt 			count = btrfs_shared_data_ref_count(leaf, sdref);
6508da6d581SJan Schmidt 			ret = __add_prelim_ref(prefs, 0, NULL, 0, offset,
6518da6d581SJan Schmidt 					       bytenr, count);
6528da6d581SJan Schmidt 			break;
6538da6d581SJan Schmidt 		}
6548da6d581SJan Schmidt 		case BTRFS_TREE_BLOCK_REF_KEY:
655d5c88b73SJan Schmidt 			ret = __add_prelim_ref(prefs, offset, NULL,
656d5c88b73SJan Schmidt 					       *info_level + 1, 0,
657d5c88b73SJan Schmidt 					       bytenr, 1);
6588da6d581SJan Schmidt 			break;
6598da6d581SJan Schmidt 		case BTRFS_EXTENT_DATA_REF_KEY: {
6608da6d581SJan Schmidt 			struct btrfs_extent_data_ref *dref;
6618da6d581SJan Schmidt 			int count;
6628da6d581SJan Schmidt 			u64 root;
6638da6d581SJan Schmidt 
6648da6d581SJan Schmidt 			dref = (struct btrfs_extent_data_ref *)(&iref->offset);
6658da6d581SJan Schmidt 			count = btrfs_extent_data_ref_count(leaf, dref);
6668da6d581SJan Schmidt 			key.objectid = btrfs_extent_data_ref_objectid(leaf,
6678da6d581SJan Schmidt 								      dref);
6688da6d581SJan Schmidt 			key.type = BTRFS_EXTENT_DATA_KEY;
6698da6d581SJan Schmidt 			key.offset = btrfs_extent_data_ref_offset(leaf, dref);
6708da6d581SJan Schmidt 			root = btrfs_extent_data_ref_root(leaf, dref);
671d5c88b73SJan Schmidt 			ret = __add_prelim_ref(prefs, root, &key, 0, 0,
672d5c88b73SJan Schmidt 					       bytenr, count);
6738da6d581SJan Schmidt 			break;
6748da6d581SJan Schmidt 		}
6758da6d581SJan Schmidt 		default:
6768da6d581SJan Schmidt 			WARN_ON(1);
6778da6d581SJan Schmidt 		}
6788da6d581SJan Schmidt 		BUG_ON(ret);
6798da6d581SJan Schmidt 		ptr += btrfs_extent_inline_ref_size(type);
6808da6d581SJan Schmidt 	}
6818da6d581SJan Schmidt 
6828da6d581SJan Schmidt 	return 0;
6838da6d581SJan Schmidt }
6848da6d581SJan Schmidt 
6858da6d581SJan Schmidt /*
6868da6d581SJan Schmidt  * add all non-inline backrefs for bytenr to the list
6878da6d581SJan Schmidt  */
6888da6d581SJan Schmidt static int __add_keyed_refs(struct btrfs_fs_info *fs_info,
6898da6d581SJan Schmidt 			    struct btrfs_path *path, u64 bytenr,
690d5c88b73SJan Schmidt 			    int info_level, struct list_head *prefs)
6918da6d581SJan Schmidt {
6928da6d581SJan Schmidt 	struct btrfs_root *extent_root = fs_info->extent_root;
6938da6d581SJan Schmidt 	int ret;
6948da6d581SJan Schmidt 	int slot;
6958da6d581SJan Schmidt 	struct extent_buffer *leaf;
6968da6d581SJan Schmidt 	struct btrfs_key key;
6978da6d581SJan Schmidt 
6988da6d581SJan Schmidt 	while (1) {
6998da6d581SJan Schmidt 		ret = btrfs_next_item(extent_root, path);
7008da6d581SJan Schmidt 		if (ret < 0)
7018da6d581SJan Schmidt 			break;
7028da6d581SJan Schmidt 		if (ret) {
7038da6d581SJan Schmidt 			ret = 0;
7048da6d581SJan Schmidt 			break;
7058da6d581SJan Schmidt 		}
7068da6d581SJan Schmidt 
7078da6d581SJan Schmidt 		slot = path->slots[0];
7088da6d581SJan Schmidt 		leaf = path->nodes[0];
7098da6d581SJan Schmidt 		btrfs_item_key_to_cpu(leaf, &key, slot);
7108da6d581SJan Schmidt 
7118da6d581SJan Schmidt 		if (key.objectid != bytenr)
7128da6d581SJan Schmidt 			break;
7138da6d581SJan Schmidt 		if (key.type < BTRFS_TREE_BLOCK_REF_KEY)
7148da6d581SJan Schmidt 			continue;
7158da6d581SJan Schmidt 		if (key.type > BTRFS_SHARED_DATA_REF_KEY)
7168da6d581SJan Schmidt 			break;
7178da6d581SJan Schmidt 
7188da6d581SJan Schmidt 		switch (key.type) {
7198da6d581SJan Schmidt 		case BTRFS_SHARED_BLOCK_REF_KEY:
720d5c88b73SJan Schmidt 			ret = __add_prelim_ref(prefs, 0, NULL,
7218da6d581SJan Schmidt 						info_level + 1, key.offset,
7228da6d581SJan Schmidt 						bytenr, 1);
7238da6d581SJan Schmidt 			break;
7248da6d581SJan Schmidt 		case BTRFS_SHARED_DATA_REF_KEY: {
7258da6d581SJan Schmidt 			struct btrfs_shared_data_ref *sdref;
7268da6d581SJan Schmidt 			int count;
7278da6d581SJan Schmidt 
7288da6d581SJan Schmidt 			sdref = btrfs_item_ptr(leaf, slot,
7298da6d581SJan Schmidt 					      struct btrfs_shared_data_ref);
7308da6d581SJan Schmidt 			count = btrfs_shared_data_ref_count(leaf, sdref);
7318da6d581SJan Schmidt 			ret = __add_prelim_ref(prefs, 0, NULL, 0, key.offset,
7328da6d581SJan Schmidt 						bytenr, count);
7338da6d581SJan Schmidt 			break;
7348da6d581SJan Schmidt 		}
7358da6d581SJan Schmidt 		case BTRFS_TREE_BLOCK_REF_KEY:
736d5c88b73SJan Schmidt 			ret = __add_prelim_ref(prefs, key.offset, NULL,
737d5c88b73SJan Schmidt 					       info_level + 1, 0,
738d5c88b73SJan Schmidt 					       bytenr, 1);
7398da6d581SJan Schmidt 			break;
7408da6d581SJan Schmidt 		case BTRFS_EXTENT_DATA_REF_KEY: {
7418da6d581SJan Schmidt 			struct btrfs_extent_data_ref *dref;
7428da6d581SJan Schmidt 			int count;
7438da6d581SJan Schmidt 			u64 root;
7448da6d581SJan Schmidt 
7458da6d581SJan Schmidt 			dref = btrfs_item_ptr(leaf, slot,
7468da6d581SJan Schmidt 					      struct btrfs_extent_data_ref);
7478da6d581SJan Schmidt 			count = btrfs_extent_data_ref_count(leaf, dref);
7488da6d581SJan Schmidt 			key.objectid = btrfs_extent_data_ref_objectid(leaf,
7498da6d581SJan Schmidt 								      dref);
7508da6d581SJan Schmidt 			key.type = BTRFS_EXTENT_DATA_KEY;
7518da6d581SJan Schmidt 			key.offset = btrfs_extent_data_ref_offset(leaf, dref);
7528da6d581SJan Schmidt 			root = btrfs_extent_data_ref_root(leaf, dref);
7538da6d581SJan Schmidt 			ret = __add_prelim_ref(prefs, root, &key, 0, 0,
7548da6d581SJan Schmidt 					       bytenr, count);
7558da6d581SJan Schmidt 			break;
7568da6d581SJan Schmidt 		}
7578da6d581SJan Schmidt 		default:
7588da6d581SJan Schmidt 			WARN_ON(1);
7598da6d581SJan Schmidt 		}
7608da6d581SJan Schmidt 		BUG_ON(ret);
7618da6d581SJan Schmidt 	}
7628da6d581SJan Schmidt 
7638da6d581SJan Schmidt 	return ret;
7648da6d581SJan Schmidt }
7658da6d581SJan Schmidt 
7668da6d581SJan Schmidt /*
7678da6d581SJan Schmidt  * this adds all existing backrefs (inline backrefs, backrefs and delayed
7688da6d581SJan Schmidt  * refs) for the given bytenr to the refs list, merges duplicates and resolves
7698da6d581SJan Schmidt  * indirect refs to their parent bytenr.
7708da6d581SJan Schmidt  * When roots are found, they're added to the roots list
7718da6d581SJan Schmidt  *
7728da6d581SJan Schmidt  * FIXME some caching might speed things up
7738da6d581SJan Schmidt  */
7748da6d581SJan Schmidt static int find_parent_nodes(struct btrfs_trans_handle *trans,
7758da6d581SJan Schmidt 			     struct btrfs_fs_info *fs_info, u64 bytenr,
7768445f61cSJan Schmidt 			     u64 delayed_ref_seq, u64 time_seq,
7778445f61cSJan Schmidt 			     struct ulist *refs, struct ulist *roots,
778976b1908SJan Schmidt 			     const u64 *extent_item_pos)
7798da6d581SJan Schmidt {
7808da6d581SJan Schmidt 	struct btrfs_key key;
7818da6d581SJan Schmidt 	struct btrfs_path *path;
7828da6d581SJan Schmidt 	struct btrfs_delayed_ref_root *delayed_refs = NULL;
783d3b01064SLi Zefan 	struct btrfs_delayed_ref_head *head;
7848da6d581SJan Schmidt 	int info_level = 0;
7858da6d581SJan Schmidt 	int ret;
7867a3ae2f8SJan Schmidt 	int search_commit_root = (trans == BTRFS_BACKREF_SEARCH_COMMIT_ROOT);
7878da6d581SJan Schmidt 	struct list_head prefs_delayed;
7888da6d581SJan Schmidt 	struct list_head prefs;
7898da6d581SJan Schmidt 	struct __prelim_ref *ref;
7908da6d581SJan Schmidt 
7918da6d581SJan Schmidt 	INIT_LIST_HEAD(&prefs);
7928da6d581SJan Schmidt 	INIT_LIST_HEAD(&prefs_delayed);
7938da6d581SJan Schmidt 
7948da6d581SJan Schmidt 	key.objectid = bytenr;
7958da6d581SJan Schmidt 	key.type = BTRFS_EXTENT_ITEM_KEY;
7968da6d581SJan Schmidt 	key.offset = (u64)-1;
7978da6d581SJan Schmidt 
7988da6d581SJan Schmidt 	path = btrfs_alloc_path();
7998da6d581SJan Schmidt 	if (!path)
8008da6d581SJan Schmidt 		return -ENOMEM;
8017a3ae2f8SJan Schmidt 	path->search_commit_root = !!search_commit_root;
8028da6d581SJan Schmidt 
8038da6d581SJan Schmidt 	/*
8048da6d581SJan Schmidt 	 * grab both a lock on the path and a lock on the delayed ref head.
8058da6d581SJan Schmidt 	 * We need both to get a consistent picture of how the refs look
8068da6d581SJan Schmidt 	 * at a specified point in time
8078da6d581SJan Schmidt 	 */
8088da6d581SJan Schmidt again:
809d3b01064SLi Zefan 	head = NULL;
810d3b01064SLi Zefan 
8118da6d581SJan Schmidt 	ret = btrfs_search_slot(trans, fs_info->extent_root, &key, path, 0, 0);
8128da6d581SJan Schmidt 	if (ret < 0)
8138da6d581SJan Schmidt 		goto out;
8148da6d581SJan Schmidt 	BUG_ON(ret == 0);
8158da6d581SJan Schmidt 
8167a3ae2f8SJan Schmidt 	if (trans != BTRFS_BACKREF_SEARCH_COMMIT_ROOT) {
8178da6d581SJan Schmidt 		/*
8187a3ae2f8SJan Schmidt 		 * look if there are updates for this ref queued and lock the
8197a3ae2f8SJan Schmidt 		 * head
8208da6d581SJan Schmidt 		 */
8218da6d581SJan Schmidt 		delayed_refs = &trans->transaction->delayed_refs;
8228da6d581SJan Schmidt 		spin_lock(&delayed_refs->lock);
8238da6d581SJan Schmidt 		head = btrfs_find_delayed_ref_head(trans, bytenr);
8248da6d581SJan Schmidt 		if (head) {
8258da6d581SJan Schmidt 			if (!mutex_trylock(&head->mutex)) {
8268da6d581SJan Schmidt 				atomic_inc(&head->node.refs);
8278da6d581SJan Schmidt 				spin_unlock(&delayed_refs->lock);
8288da6d581SJan Schmidt 
8298da6d581SJan Schmidt 				btrfs_release_path(path);
8308da6d581SJan Schmidt 
8318da6d581SJan Schmidt 				/*
8328da6d581SJan Schmidt 				 * Mutex was contended, block until it's
8338da6d581SJan Schmidt 				 * released and try again
8348da6d581SJan Schmidt 				 */
8358da6d581SJan Schmidt 				mutex_lock(&head->mutex);
8368da6d581SJan Schmidt 				mutex_unlock(&head->mutex);
8378da6d581SJan Schmidt 				btrfs_put_delayed_ref(&head->node);
8388da6d581SJan Schmidt 				goto again;
8398da6d581SJan Schmidt 			}
8408445f61cSJan Schmidt 			ret = __add_delayed_refs(head, delayed_ref_seq,
8418445f61cSJan Schmidt 						 &prefs_delayed);
842*155725c9SJan Schmidt 			mutex_unlock(&head->mutex);
843d3b01064SLi Zefan 			if (ret) {
844d3b01064SLi Zefan 				spin_unlock(&delayed_refs->lock);
8458da6d581SJan Schmidt 				goto out;
8468da6d581SJan Schmidt 			}
847d3b01064SLi Zefan 		}
8488da6d581SJan Schmidt 		spin_unlock(&delayed_refs->lock);
8497a3ae2f8SJan Schmidt 	}
8508da6d581SJan Schmidt 
8518da6d581SJan Schmidt 	if (path->slots[0]) {
8528da6d581SJan Schmidt 		struct extent_buffer *leaf;
8538da6d581SJan Schmidt 		int slot;
8548da6d581SJan Schmidt 
855dadcaf78SJan Schmidt 		path->slots[0]--;
8568da6d581SJan Schmidt 		leaf = path->nodes[0];
857dadcaf78SJan Schmidt 		slot = path->slots[0];
8588da6d581SJan Schmidt 		btrfs_item_key_to_cpu(leaf, &key, slot);
8598da6d581SJan Schmidt 		if (key.objectid == bytenr &&
8608da6d581SJan Schmidt 		    key.type == BTRFS_EXTENT_ITEM_KEY) {
8618da6d581SJan Schmidt 			ret = __add_inline_refs(fs_info, path, bytenr,
862d5c88b73SJan Schmidt 						&info_level, &prefs);
8638da6d581SJan Schmidt 			if (ret)
8648da6d581SJan Schmidt 				goto out;
865d5c88b73SJan Schmidt 			ret = __add_keyed_refs(fs_info, path, bytenr,
8668da6d581SJan Schmidt 					       info_level, &prefs);
8678da6d581SJan Schmidt 			if (ret)
8688da6d581SJan Schmidt 				goto out;
8698da6d581SJan Schmidt 		}
8708da6d581SJan Schmidt 	}
8718da6d581SJan Schmidt 	btrfs_release_path(path);
8728da6d581SJan Schmidt 
8738da6d581SJan Schmidt 	list_splice_init(&prefs_delayed, &prefs);
8748da6d581SJan Schmidt 
875d5c88b73SJan Schmidt 	ret = __add_missing_keys(fs_info, &prefs);
876d5c88b73SJan Schmidt 	if (ret)
877d5c88b73SJan Schmidt 		goto out;
878d5c88b73SJan Schmidt 
8798da6d581SJan Schmidt 	ret = __merge_refs(&prefs, 1);
8808da6d581SJan Schmidt 	if (ret)
8818da6d581SJan Schmidt 		goto out;
8828da6d581SJan Schmidt 
8838445f61cSJan Schmidt 	ret = __resolve_indirect_refs(fs_info, search_commit_root, time_seq,
8848445f61cSJan Schmidt 				      &prefs, extent_item_pos);
8858da6d581SJan Schmidt 	if (ret)
8868da6d581SJan Schmidt 		goto out;
8878da6d581SJan Schmidt 
8888da6d581SJan Schmidt 	ret = __merge_refs(&prefs, 2);
8898da6d581SJan Schmidt 	if (ret)
8908da6d581SJan Schmidt 		goto out;
8918da6d581SJan Schmidt 
8928da6d581SJan Schmidt 	while (!list_empty(&prefs)) {
8938da6d581SJan Schmidt 		ref = list_first_entry(&prefs, struct __prelim_ref, list);
8948da6d581SJan Schmidt 		list_del(&ref->list);
8958da6d581SJan Schmidt 		if (ref->count < 0)
8968da6d581SJan Schmidt 			WARN_ON(1);
8978da6d581SJan Schmidt 		if (ref->count && ref->root_id && ref->parent == 0) {
8988da6d581SJan Schmidt 			/* no parent == root of tree */
8998da6d581SJan Schmidt 			ret = ulist_add(roots, ref->root_id, 0, GFP_NOFS);
9008da6d581SJan Schmidt 			BUG_ON(ret < 0);
9018da6d581SJan Schmidt 		}
9028da6d581SJan Schmidt 		if (ref->count && ref->parent) {
903976b1908SJan Schmidt 			struct extent_inode_elem *eie = NULL;
9043301958bSJan Schmidt 			if (extent_item_pos && !ref->inode_list) {
905976b1908SJan Schmidt 				u32 bsz;
906976b1908SJan Schmidt 				struct extent_buffer *eb;
907976b1908SJan Schmidt 				bsz = btrfs_level_size(fs_info->extent_root,
908976b1908SJan Schmidt 							info_level);
909976b1908SJan Schmidt 				eb = read_tree_block(fs_info->extent_root,
910976b1908SJan Schmidt 							   ref->parent, bsz, 0);
911976b1908SJan Schmidt 				BUG_ON(!eb);
912976b1908SJan Schmidt 				ret = find_extent_in_eb(eb, bytenr,
913976b1908SJan Schmidt 							*extent_item_pos, &eie);
9143301958bSJan Schmidt 				ref->inode_list = eie;
915976b1908SJan Schmidt 				free_extent_buffer(eb);
916976b1908SJan Schmidt 			}
9173301958bSJan Schmidt 			ret = ulist_add_merge(refs, ref->parent,
9183301958bSJan Schmidt 					      (unsigned long)ref->inode_list,
9193301958bSJan Schmidt 					      (unsigned long *)&eie, GFP_NOFS);
9203301958bSJan Schmidt 			if (!ret && extent_item_pos) {
9213301958bSJan Schmidt 				/*
9223301958bSJan Schmidt 				 * we've recorded that parent, so we must extend
9233301958bSJan Schmidt 				 * its inode list here
9243301958bSJan Schmidt 				 */
9253301958bSJan Schmidt 				BUG_ON(!eie);
9263301958bSJan Schmidt 				while (eie->next)
9273301958bSJan Schmidt 					eie = eie->next;
9283301958bSJan Schmidt 				eie->next = ref->inode_list;
9293301958bSJan Schmidt 			}
9308da6d581SJan Schmidt 			BUG_ON(ret < 0);
9318da6d581SJan Schmidt 		}
9328da6d581SJan Schmidt 		kfree(ref);
9338da6d581SJan Schmidt 	}
9348da6d581SJan Schmidt 
9358da6d581SJan Schmidt out:
9368da6d581SJan Schmidt 	btrfs_free_path(path);
9378da6d581SJan Schmidt 	while (!list_empty(&prefs)) {
9388da6d581SJan Schmidt 		ref = list_first_entry(&prefs, struct __prelim_ref, list);
9398da6d581SJan Schmidt 		list_del(&ref->list);
9408da6d581SJan Schmidt 		kfree(ref);
9418da6d581SJan Schmidt 	}
9428da6d581SJan Schmidt 	while (!list_empty(&prefs_delayed)) {
9438da6d581SJan Schmidt 		ref = list_first_entry(&prefs_delayed, struct __prelim_ref,
9448da6d581SJan Schmidt 				       list);
9458da6d581SJan Schmidt 		list_del(&ref->list);
9468da6d581SJan Schmidt 		kfree(ref);
9478da6d581SJan Schmidt 	}
9488da6d581SJan Schmidt 
9498da6d581SJan Schmidt 	return ret;
9508da6d581SJan Schmidt }
9518da6d581SJan Schmidt 
952976b1908SJan Schmidt static void free_leaf_list(struct ulist *blocks)
953976b1908SJan Schmidt {
954976b1908SJan Schmidt 	struct ulist_node *node = NULL;
955976b1908SJan Schmidt 	struct extent_inode_elem *eie;
956976b1908SJan Schmidt 	struct extent_inode_elem *eie_next;
957976b1908SJan Schmidt 	struct ulist_iterator uiter;
958976b1908SJan Schmidt 
959976b1908SJan Schmidt 	ULIST_ITER_INIT(&uiter);
960976b1908SJan Schmidt 	while ((node = ulist_next(blocks, &uiter))) {
961976b1908SJan Schmidt 		if (!node->aux)
962976b1908SJan Schmidt 			continue;
963976b1908SJan Schmidt 		eie = (struct extent_inode_elem *)node->aux;
964976b1908SJan Schmidt 		for (; eie; eie = eie_next) {
965976b1908SJan Schmidt 			eie_next = eie->next;
966976b1908SJan Schmidt 			kfree(eie);
967976b1908SJan Schmidt 		}
968976b1908SJan Schmidt 		node->aux = 0;
969976b1908SJan Schmidt 	}
970976b1908SJan Schmidt 
971976b1908SJan Schmidt 	ulist_free(blocks);
972976b1908SJan Schmidt }
973976b1908SJan Schmidt 
9748da6d581SJan Schmidt /*
9758da6d581SJan Schmidt  * Finds all leafs with a reference to the specified combination of bytenr and
9768da6d581SJan Schmidt  * offset. key_list_head will point to a list of corresponding keys (caller must
9778da6d581SJan Schmidt  * free each list element). The leafs will be stored in the leafs ulist, which
9788da6d581SJan Schmidt  * must be freed with ulist_free.
9798da6d581SJan Schmidt  *
9808da6d581SJan Schmidt  * returns 0 on success, <0 on error
9818da6d581SJan Schmidt  */
9828da6d581SJan Schmidt static int btrfs_find_all_leafs(struct btrfs_trans_handle *trans,
9838da6d581SJan Schmidt 				struct btrfs_fs_info *fs_info, u64 bytenr,
9848445f61cSJan Schmidt 				u64 delayed_ref_seq, u64 time_seq,
9858445f61cSJan Schmidt 				struct ulist **leafs,
986976b1908SJan Schmidt 				const u64 *extent_item_pos)
9878da6d581SJan Schmidt {
9888da6d581SJan Schmidt 	struct ulist *tmp;
9898da6d581SJan Schmidt 	int ret;
9908da6d581SJan Schmidt 
9918da6d581SJan Schmidt 	tmp = ulist_alloc(GFP_NOFS);
9928da6d581SJan Schmidt 	if (!tmp)
9938da6d581SJan Schmidt 		return -ENOMEM;
9948da6d581SJan Schmidt 	*leafs = ulist_alloc(GFP_NOFS);
9958da6d581SJan Schmidt 	if (!*leafs) {
9968da6d581SJan Schmidt 		ulist_free(tmp);
9978da6d581SJan Schmidt 		return -ENOMEM;
9988da6d581SJan Schmidt 	}
9998da6d581SJan Schmidt 
10008445f61cSJan Schmidt 	ret = find_parent_nodes(trans, fs_info, bytenr, delayed_ref_seq,
10018445f61cSJan Schmidt 				time_seq, *leafs, tmp, extent_item_pos);
10028da6d581SJan Schmidt 	ulist_free(tmp);
10038da6d581SJan Schmidt 
10048da6d581SJan Schmidt 	if (ret < 0 && ret != -ENOENT) {
1005976b1908SJan Schmidt 		free_leaf_list(*leafs);
10068da6d581SJan Schmidt 		return ret;
10078da6d581SJan Schmidt 	}
10088da6d581SJan Schmidt 
10098da6d581SJan Schmidt 	return 0;
10108da6d581SJan Schmidt }
10118da6d581SJan Schmidt 
10128da6d581SJan Schmidt /*
10138da6d581SJan Schmidt  * walk all backrefs for a given extent to find all roots that reference this
10148da6d581SJan Schmidt  * extent. Walking a backref means finding all extents that reference this
10158da6d581SJan Schmidt  * extent and in turn walk the backrefs of those, too. Naturally this is a
10168da6d581SJan Schmidt  * recursive process, but here it is implemented in an iterative fashion: We
10178da6d581SJan Schmidt  * find all referencing extents for the extent in question and put them on a
10188da6d581SJan Schmidt  * list. In turn, we find all referencing extents for those, further appending
10198da6d581SJan Schmidt  * to the list. The way we iterate the list allows adding more elements after
10208da6d581SJan Schmidt  * the current while iterating. The process stops when we reach the end of the
10218da6d581SJan Schmidt  * list. Found roots are added to the roots list.
10228da6d581SJan Schmidt  *
10238da6d581SJan Schmidt  * returns 0 on success, < 0 on error.
10248da6d581SJan Schmidt  */
10258da6d581SJan Schmidt int btrfs_find_all_roots(struct btrfs_trans_handle *trans,
10268da6d581SJan Schmidt 				struct btrfs_fs_info *fs_info, u64 bytenr,
10278445f61cSJan Schmidt 				u64 delayed_ref_seq, u64 time_seq,
10288445f61cSJan Schmidt 				struct ulist **roots)
10298da6d581SJan Schmidt {
10308da6d581SJan Schmidt 	struct ulist *tmp;
10318da6d581SJan Schmidt 	struct ulist_node *node = NULL;
1032cd1b413cSJan Schmidt 	struct ulist_iterator uiter;
10338da6d581SJan Schmidt 	int ret;
10348da6d581SJan Schmidt 
10358da6d581SJan Schmidt 	tmp = ulist_alloc(GFP_NOFS);
10368da6d581SJan Schmidt 	if (!tmp)
10378da6d581SJan Schmidt 		return -ENOMEM;
10388da6d581SJan Schmidt 	*roots = ulist_alloc(GFP_NOFS);
10398da6d581SJan Schmidt 	if (!*roots) {
10408da6d581SJan Schmidt 		ulist_free(tmp);
10418da6d581SJan Schmidt 		return -ENOMEM;
10428da6d581SJan Schmidt 	}
10438da6d581SJan Schmidt 
1044cd1b413cSJan Schmidt 	ULIST_ITER_INIT(&uiter);
10458da6d581SJan Schmidt 	while (1) {
10468445f61cSJan Schmidt 		ret = find_parent_nodes(trans, fs_info, bytenr, delayed_ref_seq,
10478445f61cSJan Schmidt 					time_seq, tmp, *roots, NULL);
10488da6d581SJan Schmidt 		if (ret < 0 && ret != -ENOENT) {
10498da6d581SJan Schmidt 			ulist_free(tmp);
10508da6d581SJan Schmidt 			ulist_free(*roots);
10518da6d581SJan Schmidt 			return ret;
10528da6d581SJan Schmidt 		}
1053cd1b413cSJan Schmidt 		node = ulist_next(tmp, &uiter);
10548da6d581SJan Schmidt 		if (!node)
10558da6d581SJan Schmidt 			break;
10568da6d581SJan Schmidt 		bytenr = node->val;
10578da6d581SJan Schmidt 	}
10588da6d581SJan Schmidt 
10598da6d581SJan Schmidt 	ulist_free(tmp);
10608da6d581SJan Schmidt 	return 0;
10618da6d581SJan Schmidt }
10628da6d581SJan Schmidt 
10638da6d581SJan Schmidt 
1064a542ad1bSJan Schmidt static int __inode_info(u64 inum, u64 ioff, u8 key_type,
1065a542ad1bSJan Schmidt 			struct btrfs_root *fs_root, struct btrfs_path *path,
1066a542ad1bSJan Schmidt 			struct btrfs_key *found_key)
1067a542ad1bSJan Schmidt {
1068a542ad1bSJan Schmidt 	int ret;
1069a542ad1bSJan Schmidt 	struct btrfs_key key;
1070a542ad1bSJan Schmidt 	struct extent_buffer *eb;
1071a542ad1bSJan Schmidt 
1072a542ad1bSJan Schmidt 	key.type = key_type;
1073a542ad1bSJan Schmidt 	key.objectid = inum;
1074a542ad1bSJan Schmidt 	key.offset = ioff;
1075a542ad1bSJan Schmidt 
1076a542ad1bSJan Schmidt 	ret = btrfs_search_slot(NULL, fs_root, &key, path, 0, 0);
1077a542ad1bSJan Schmidt 	if (ret < 0)
1078a542ad1bSJan Schmidt 		return ret;
1079a542ad1bSJan Schmidt 
1080a542ad1bSJan Schmidt 	eb = path->nodes[0];
1081a542ad1bSJan Schmidt 	if (ret && path->slots[0] >= btrfs_header_nritems(eb)) {
1082a542ad1bSJan Schmidt 		ret = btrfs_next_leaf(fs_root, path);
1083a542ad1bSJan Schmidt 		if (ret)
1084a542ad1bSJan Schmidt 			return ret;
1085a542ad1bSJan Schmidt 		eb = path->nodes[0];
1086a542ad1bSJan Schmidt 	}
1087a542ad1bSJan Schmidt 
1088a542ad1bSJan Schmidt 	btrfs_item_key_to_cpu(eb, found_key, path->slots[0]);
1089a542ad1bSJan Schmidt 	if (found_key->type != key.type || found_key->objectid != key.objectid)
1090a542ad1bSJan Schmidt 		return 1;
1091a542ad1bSJan Schmidt 
1092a542ad1bSJan Schmidt 	return 0;
1093a542ad1bSJan Schmidt }
1094a542ad1bSJan Schmidt 
1095a542ad1bSJan Schmidt /*
1096a542ad1bSJan Schmidt  * this makes the path point to (inum INODE_ITEM ioff)
1097a542ad1bSJan Schmidt  */
1098a542ad1bSJan Schmidt int inode_item_info(u64 inum, u64 ioff, struct btrfs_root *fs_root,
1099a542ad1bSJan Schmidt 			struct btrfs_path *path)
1100a542ad1bSJan Schmidt {
1101a542ad1bSJan Schmidt 	struct btrfs_key key;
1102a542ad1bSJan Schmidt 	return __inode_info(inum, ioff, BTRFS_INODE_ITEM_KEY, fs_root, path,
1103a542ad1bSJan Schmidt 				&key);
1104a542ad1bSJan Schmidt }
1105a542ad1bSJan Schmidt 
1106a542ad1bSJan Schmidt static int inode_ref_info(u64 inum, u64 ioff, struct btrfs_root *fs_root,
1107a542ad1bSJan Schmidt 				struct btrfs_path *path,
1108a542ad1bSJan Schmidt 				struct btrfs_key *found_key)
1109a542ad1bSJan Schmidt {
1110a542ad1bSJan Schmidt 	return __inode_info(inum, ioff, BTRFS_INODE_REF_KEY, fs_root, path,
1111a542ad1bSJan Schmidt 				found_key);
1112a542ad1bSJan Schmidt }
1113a542ad1bSJan Schmidt 
1114a542ad1bSJan Schmidt /*
1115a542ad1bSJan Schmidt  * this iterates to turn a btrfs_inode_ref into a full filesystem path. elements
1116a542ad1bSJan Schmidt  * of the path are separated by '/' and the path is guaranteed to be
1117a542ad1bSJan Schmidt  * 0-terminated. the path is only given within the current file system.
1118a542ad1bSJan Schmidt  * Therefore, it never starts with a '/'. the caller is responsible to provide
1119a542ad1bSJan Schmidt  * "size" bytes in "dest". the dest buffer will be filled backwards. finally,
1120a542ad1bSJan Schmidt  * the start point of the resulting string is returned. this pointer is within
1121a542ad1bSJan Schmidt  * dest, normally.
1122a542ad1bSJan Schmidt  * in case the path buffer would overflow, the pointer is decremented further
1123a542ad1bSJan Schmidt  * as if output was written to the buffer, though no more output is actually
1124a542ad1bSJan Schmidt  * generated. that way, the caller can determine how much space would be
1125a542ad1bSJan Schmidt  * required for the path to fit into the buffer. in that case, the returned
1126a542ad1bSJan Schmidt  * value will be smaller than dest. callers must check this!
1127a542ad1bSJan Schmidt  */
1128a542ad1bSJan Schmidt static char *iref_to_path(struct btrfs_root *fs_root, struct btrfs_path *path,
1129a542ad1bSJan Schmidt 				struct btrfs_inode_ref *iref,
1130a542ad1bSJan Schmidt 				struct extent_buffer *eb_in, u64 parent,
1131a542ad1bSJan Schmidt 				char *dest, u32 size)
1132a542ad1bSJan Schmidt {
1133a542ad1bSJan Schmidt 	u32 len;
1134a542ad1bSJan Schmidt 	int slot;
1135a542ad1bSJan Schmidt 	u64 next_inum;
1136a542ad1bSJan Schmidt 	int ret;
1137a542ad1bSJan Schmidt 	s64 bytes_left = size - 1;
1138a542ad1bSJan Schmidt 	struct extent_buffer *eb = eb_in;
1139a542ad1bSJan Schmidt 	struct btrfs_key found_key;
1140b916a59aSJan Schmidt 	int leave_spinning = path->leave_spinning;
1141a542ad1bSJan Schmidt 
1142a542ad1bSJan Schmidt 	if (bytes_left >= 0)
1143a542ad1bSJan Schmidt 		dest[bytes_left] = '\0';
1144a542ad1bSJan Schmidt 
1145b916a59aSJan Schmidt 	path->leave_spinning = 1;
1146a542ad1bSJan Schmidt 	while (1) {
1147a542ad1bSJan Schmidt 		len = btrfs_inode_ref_name_len(eb, iref);
1148a542ad1bSJan Schmidt 		bytes_left -= len;
1149a542ad1bSJan Schmidt 		if (bytes_left >= 0)
1150a542ad1bSJan Schmidt 			read_extent_buffer(eb, dest + bytes_left,
1151a542ad1bSJan Schmidt 						(unsigned long)(iref + 1), len);
1152b916a59aSJan Schmidt 		if (eb != eb_in) {
1153b916a59aSJan Schmidt 			btrfs_tree_read_unlock_blocking(eb);
1154a542ad1bSJan Schmidt 			free_extent_buffer(eb);
1155b916a59aSJan Schmidt 		}
1156a542ad1bSJan Schmidt 		ret = inode_ref_info(parent, 0, fs_root, path, &found_key);
11578f24b496SJan Schmidt 		if (ret > 0)
11588f24b496SJan Schmidt 			ret = -ENOENT;
1159a542ad1bSJan Schmidt 		if (ret)
1160a542ad1bSJan Schmidt 			break;
1161a542ad1bSJan Schmidt 		next_inum = found_key.offset;
1162a542ad1bSJan Schmidt 
1163a542ad1bSJan Schmidt 		/* regular exit ahead */
1164a542ad1bSJan Schmidt 		if (parent == next_inum)
1165a542ad1bSJan Schmidt 			break;
1166a542ad1bSJan Schmidt 
1167a542ad1bSJan Schmidt 		slot = path->slots[0];
1168a542ad1bSJan Schmidt 		eb = path->nodes[0];
1169a542ad1bSJan Schmidt 		/* make sure we can use eb after releasing the path */
1170b916a59aSJan Schmidt 		if (eb != eb_in) {
1171a542ad1bSJan Schmidt 			atomic_inc(&eb->refs);
1172b916a59aSJan Schmidt 			btrfs_tree_read_lock(eb);
1173b916a59aSJan Schmidt 			btrfs_set_lock_blocking_rw(eb, BTRFS_READ_LOCK);
1174b916a59aSJan Schmidt 		}
1175a542ad1bSJan Schmidt 		btrfs_release_path(path);
1176a542ad1bSJan Schmidt 
1177a542ad1bSJan Schmidt 		iref = btrfs_item_ptr(eb, slot, struct btrfs_inode_ref);
1178a542ad1bSJan Schmidt 		parent = next_inum;
1179a542ad1bSJan Schmidt 		--bytes_left;
1180a542ad1bSJan Schmidt 		if (bytes_left >= 0)
1181a542ad1bSJan Schmidt 			dest[bytes_left] = '/';
1182a542ad1bSJan Schmidt 	}
1183a542ad1bSJan Schmidt 
1184a542ad1bSJan Schmidt 	btrfs_release_path(path);
1185b916a59aSJan Schmidt 	path->leave_spinning = leave_spinning;
1186a542ad1bSJan Schmidt 
1187a542ad1bSJan Schmidt 	if (ret)
1188a542ad1bSJan Schmidt 		return ERR_PTR(ret);
1189a542ad1bSJan Schmidt 
1190a542ad1bSJan Schmidt 	return dest + bytes_left;
1191a542ad1bSJan Schmidt }
1192a542ad1bSJan Schmidt 
1193a542ad1bSJan Schmidt /*
1194a542ad1bSJan Schmidt  * this makes the path point to (logical EXTENT_ITEM *)
1195a542ad1bSJan Schmidt  * returns BTRFS_EXTENT_FLAG_DATA for data, BTRFS_EXTENT_FLAG_TREE_BLOCK for
1196a542ad1bSJan Schmidt  * tree blocks and <0 on error.
1197a542ad1bSJan Schmidt  */
1198a542ad1bSJan Schmidt int extent_from_logical(struct btrfs_fs_info *fs_info, u64 logical,
1199a542ad1bSJan Schmidt 			struct btrfs_path *path, struct btrfs_key *found_key)
1200a542ad1bSJan Schmidt {
1201a542ad1bSJan Schmidt 	int ret;
1202a542ad1bSJan Schmidt 	u64 flags;
1203a542ad1bSJan Schmidt 	u32 item_size;
1204a542ad1bSJan Schmidt 	struct extent_buffer *eb;
1205a542ad1bSJan Schmidt 	struct btrfs_extent_item *ei;
1206a542ad1bSJan Schmidt 	struct btrfs_key key;
1207a542ad1bSJan Schmidt 
1208a542ad1bSJan Schmidt 	key.type = BTRFS_EXTENT_ITEM_KEY;
1209a542ad1bSJan Schmidt 	key.objectid = logical;
1210a542ad1bSJan Schmidt 	key.offset = (u64)-1;
1211a542ad1bSJan Schmidt 
1212a542ad1bSJan Schmidt 	ret = btrfs_search_slot(NULL, fs_info->extent_root, &key, path, 0, 0);
1213a542ad1bSJan Schmidt 	if (ret < 0)
1214a542ad1bSJan Schmidt 		return ret;
1215a542ad1bSJan Schmidt 	ret = btrfs_previous_item(fs_info->extent_root, path,
1216a542ad1bSJan Schmidt 					0, BTRFS_EXTENT_ITEM_KEY);
1217a542ad1bSJan Schmidt 	if (ret < 0)
1218a542ad1bSJan Schmidt 		return ret;
1219a542ad1bSJan Schmidt 
1220a542ad1bSJan Schmidt 	btrfs_item_key_to_cpu(path->nodes[0], found_key, path->slots[0]);
1221a542ad1bSJan Schmidt 	if (found_key->type != BTRFS_EXTENT_ITEM_KEY ||
1222a542ad1bSJan Schmidt 	    found_key->objectid > logical ||
12234692cf58SJan Schmidt 	    found_key->objectid + found_key->offset <= logical) {
12244692cf58SJan Schmidt 		pr_debug("logical %llu is not within any extent\n",
12254692cf58SJan Schmidt 			 (unsigned long long)logical);
1226a542ad1bSJan Schmidt 		return -ENOENT;
12274692cf58SJan Schmidt 	}
1228a542ad1bSJan Schmidt 
1229a542ad1bSJan Schmidt 	eb = path->nodes[0];
1230a542ad1bSJan Schmidt 	item_size = btrfs_item_size_nr(eb, path->slots[0]);
1231a542ad1bSJan Schmidt 	BUG_ON(item_size < sizeof(*ei));
1232a542ad1bSJan Schmidt 
1233a542ad1bSJan Schmidt 	ei = btrfs_item_ptr(eb, path->slots[0], struct btrfs_extent_item);
1234a542ad1bSJan Schmidt 	flags = btrfs_extent_flags(eb, ei);
1235a542ad1bSJan Schmidt 
12364692cf58SJan Schmidt 	pr_debug("logical %llu is at position %llu within the extent (%llu "
12374692cf58SJan Schmidt 		 "EXTENT_ITEM %llu) flags %#llx size %u\n",
12384692cf58SJan Schmidt 		 (unsigned long long)logical,
12394692cf58SJan Schmidt 		 (unsigned long long)(logical - found_key->objectid),
12404692cf58SJan Schmidt 		 (unsigned long long)found_key->objectid,
12414692cf58SJan Schmidt 		 (unsigned long long)found_key->offset,
12424692cf58SJan Schmidt 		 (unsigned long long)flags, item_size);
1243a542ad1bSJan Schmidt 	if (flags & BTRFS_EXTENT_FLAG_TREE_BLOCK)
1244a542ad1bSJan Schmidt 		return BTRFS_EXTENT_FLAG_TREE_BLOCK;
1245a542ad1bSJan Schmidt 	if (flags & BTRFS_EXTENT_FLAG_DATA)
1246a542ad1bSJan Schmidt 		return BTRFS_EXTENT_FLAG_DATA;
1247a542ad1bSJan Schmidt 
1248a542ad1bSJan Schmidt 	return -EIO;
1249a542ad1bSJan Schmidt }
1250a542ad1bSJan Schmidt 
1251a542ad1bSJan Schmidt /*
1252a542ad1bSJan Schmidt  * helper function to iterate extent inline refs. ptr must point to a 0 value
1253a542ad1bSJan Schmidt  * for the first call and may be modified. it is used to track state.
1254a542ad1bSJan Schmidt  * if more refs exist, 0 is returned and the next call to
1255a542ad1bSJan Schmidt  * __get_extent_inline_ref must pass the modified ptr parameter to get the
1256a542ad1bSJan Schmidt  * next ref. after the last ref was processed, 1 is returned.
1257a542ad1bSJan Schmidt  * returns <0 on error
1258a542ad1bSJan Schmidt  */
1259a542ad1bSJan Schmidt static int __get_extent_inline_ref(unsigned long *ptr, struct extent_buffer *eb,
1260a542ad1bSJan Schmidt 				struct btrfs_extent_item *ei, u32 item_size,
1261a542ad1bSJan Schmidt 				struct btrfs_extent_inline_ref **out_eiref,
1262a542ad1bSJan Schmidt 				int *out_type)
1263a542ad1bSJan Schmidt {
1264a542ad1bSJan Schmidt 	unsigned long end;
1265a542ad1bSJan Schmidt 	u64 flags;
1266a542ad1bSJan Schmidt 	struct btrfs_tree_block_info *info;
1267a542ad1bSJan Schmidt 
1268a542ad1bSJan Schmidt 	if (!*ptr) {
1269a542ad1bSJan Schmidt 		/* first call */
1270a542ad1bSJan Schmidt 		flags = btrfs_extent_flags(eb, ei);
1271a542ad1bSJan Schmidt 		if (flags & BTRFS_EXTENT_FLAG_TREE_BLOCK) {
1272a542ad1bSJan Schmidt 			info = (struct btrfs_tree_block_info *)(ei + 1);
1273a542ad1bSJan Schmidt 			*out_eiref =
1274a542ad1bSJan Schmidt 				(struct btrfs_extent_inline_ref *)(info + 1);
1275a542ad1bSJan Schmidt 		} else {
1276a542ad1bSJan Schmidt 			*out_eiref = (struct btrfs_extent_inline_ref *)(ei + 1);
1277a542ad1bSJan Schmidt 		}
1278a542ad1bSJan Schmidt 		*ptr = (unsigned long)*out_eiref;
1279a542ad1bSJan Schmidt 		if ((void *)*ptr >= (void *)ei + item_size)
1280a542ad1bSJan Schmidt 			return -ENOENT;
1281a542ad1bSJan Schmidt 	}
1282a542ad1bSJan Schmidt 
1283a542ad1bSJan Schmidt 	end = (unsigned long)ei + item_size;
1284a542ad1bSJan Schmidt 	*out_eiref = (struct btrfs_extent_inline_ref *)*ptr;
1285a542ad1bSJan Schmidt 	*out_type = btrfs_extent_inline_ref_type(eb, *out_eiref);
1286a542ad1bSJan Schmidt 
1287a542ad1bSJan Schmidt 	*ptr += btrfs_extent_inline_ref_size(*out_type);
1288a542ad1bSJan Schmidt 	WARN_ON(*ptr > end);
1289a542ad1bSJan Schmidt 	if (*ptr == end)
1290a542ad1bSJan Schmidt 		return 1; /* last */
1291a542ad1bSJan Schmidt 
1292a542ad1bSJan Schmidt 	return 0;
1293a542ad1bSJan Schmidt }
1294a542ad1bSJan Schmidt 
1295a542ad1bSJan Schmidt /*
1296a542ad1bSJan Schmidt  * reads the tree block backref for an extent. tree level and root are returned
1297a542ad1bSJan Schmidt  * through out_level and out_root. ptr must point to a 0 value for the first
1298a542ad1bSJan Schmidt  * call and may be modified (see __get_extent_inline_ref comment).
1299a542ad1bSJan Schmidt  * returns 0 if data was provided, 1 if there was no more data to provide or
1300a542ad1bSJan Schmidt  * <0 on error.
1301a542ad1bSJan Schmidt  */
1302a542ad1bSJan Schmidt int tree_backref_for_extent(unsigned long *ptr, struct extent_buffer *eb,
1303a542ad1bSJan Schmidt 				struct btrfs_extent_item *ei, u32 item_size,
1304a542ad1bSJan Schmidt 				u64 *out_root, u8 *out_level)
1305a542ad1bSJan Schmidt {
1306a542ad1bSJan Schmidt 	int ret;
1307a542ad1bSJan Schmidt 	int type;
1308a542ad1bSJan Schmidt 	struct btrfs_tree_block_info *info;
1309a542ad1bSJan Schmidt 	struct btrfs_extent_inline_ref *eiref;
1310a542ad1bSJan Schmidt 
1311a542ad1bSJan Schmidt 	if (*ptr == (unsigned long)-1)
1312a542ad1bSJan Schmidt 		return 1;
1313a542ad1bSJan Schmidt 
1314a542ad1bSJan Schmidt 	while (1) {
1315a542ad1bSJan Schmidt 		ret = __get_extent_inline_ref(ptr, eb, ei, item_size,
1316a542ad1bSJan Schmidt 						&eiref, &type);
1317a542ad1bSJan Schmidt 		if (ret < 0)
1318a542ad1bSJan Schmidt 			return ret;
1319a542ad1bSJan Schmidt 
1320a542ad1bSJan Schmidt 		if (type == BTRFS_TREE_BLOCK_REF_KEY ||
1321a542ad1bSJan Schmidt 		    type == BTRFS_SHARED_BLOCK_REF_KEY)
1322a542ad1bSJan Schmidt 			break;
1323a542ad1bSJan Schmidt 
1324a542ad1bSJan Schmidt 		if (ret == 1)
1325a542ad1bSJan Schmidt 			return 1;
1326a542ad1bSJan Schmidt 	}
1327a542ad1bSJan Schmidt 
1328a542ad1bSJan Schmidt 	/* we can treat both ref types equally here */
1329a542ad1bSJan Schmidt 	info = (struct btrfs_tree_block_info *)(ei + 1);
1330a542ad1bSJan Schmidt 	*out_root = btrfs_extent_inline_ref_offset(eb, eiref);
1331a542ad1bSJan Schmidt 	*out_level = btrfs_tree_block_level(eb, info);
1332a542ad1bSJan Schmidt 
1333a542ad1bSJan Schmidt 	if (ret == 1)
1334a542ad1bSJan Schmidt 		*ptr = (unsigned long)-1;
1335a542ad1bSJan Schmidt 
1336a542ad1bSJan Schmidt 	return 0;
1337a542ad1bSJan Schmidt }
1338a542ad1bSJan Schmidt 
1339976b1908SJan Schmidt static int iterate_leaf_refs(struct extent_inode_elem *inode_list,
1340976b1908SJan Schmidt 				u64 root, u64 extent_item_objectid,
13414692cf58SJan Schmidt 				iterate_extent_inodes_t *iterate, void *ctx)
1342a542ad1bSJan Schmidt {
1343976b1908SJan Schmidt 	struct extent_inode_elem *eie;
13444692cf58SJan Schmidt 	int ret = 0;
1345a542ad1bSJan Schmidt 
1346976b1908SJan Schmidt 	for (eie = inode_list; eie; eie = eie->next) {
13474692cf58SJan Schmidt 		pr_debug("ref for %llu resolved, key (%llu EXTEND_DATA %llu), "
1348976b1908SJan Schmidt 			 "root %llu\n", extent_item_objectid,
1349976b1908SJan Schmidt 			 eie->inum, eie->offset, root);
1350976b1908SJan Schmidt 		ret = iterate(eie->inum, eie->offset, root, ctx);
13514692cf58SJan Schmidt 		if (ret) {
1352976b1908SJan Schmidt 			pr_debug("stopping iteration for %llu due to ret=%d\n",
1353976b1908SJan Schmidt 				 extent_item_objectid, ret);
1354a542ad1bSJan Schmidt 			break;
1355a542ad1bSJan Schmidt 		}
1356a542ad1bSJan Schmidt 	}
1357a542ad1bSJan Schmidt 
1358a542ad1bSJan Schmidt 	return ret;
1359a542ad1bSJan Schmidt }
1360a542ad1bSJan Schmidt 
1361a542ad1bSJan Schmidt /*
1362a542ad1bSJan Schmidt  * calls iterate() for every inode that references the extent identified by
13634692cf58SJan Schmidt  * the given parameters.
1364a542ad1bSJan Schmidt  * when the iterator function returns a non-zero value, iteration stops.
1365a542ad1bSJan Schmidt  */
1366a542ad1bSJan Schmidt int iterate_extent_inodes(struct btrfs_fs_info *fs_info,
13674692cf58SJan Schmidt 				u64 extent_item_objectid, u64 extent_item_pos,
13687a3ae2f8SJan Schmidt 				int search_commit_root,
1369a542ad1bSJan Schmidt 				iterate_extent_inodes_t *iterate, void *ctx)
1370a542ad1bSJan Schmidt {
1371a542ad1bSJan Schmidt 	int ret;
1372a542ad1bSJan Schmidt 	struct list_head data_refs = LIST_HEAD_INIT(data_refs);
1373a542ad1bSJan Schmidt 	struct list_head shared_refs = LIST_HEAD_INIT(shared_refs);
13744692cf58SJan Schmidt 	struct btrfs_trans_handle *trans;
13757a3ae2f8SJan Schmidt 	struct ulist *refs = NULL;
13767a3ae2f8SJan Schmidt 	struct ulist *roots = NULL;
13774692cf58SJan Schmidt 	struct ulist_node *ref_node = NULL;
13784692cf58SJan Schmidt 	struct ulist_node *root_node = NULL;
13798445f61cSJan Schmidt 	struct seq_list seq_elem = {};
13808445f61cSJan Schmidt 	struct seq_list tree_mod_seq_elem = {};
1381cd1b413cSJan Schmidt 	struct ulist_iterator ref_uiter;
1382cd1b413cSJan Schmidt 	struct ulist_iterator root_uiter;
13837a3ae2f8SJan Schmidt 	struct btrfs_delayed_ref_root *delayed_refs = NULL;
1384a542ad1bSJan Schmidt 
13854692cf58SJan Schmidt 	pr_debug("resolving all inodes for extent %llu\n",
13864692cf58SJan Schmidt 			extent_item_objectid);
13874692cf58SJan Schmidt 
13887a3ae2f8SJan Schmidt 	if (search_commit_root) {
13897a3ae2f8SJan Schmidt 		trans = BTRFS_BACKREF_SEARCH_COMMIT_ROOT;
13907a3ae2f8SJan Schmidt 	} else {
13917a3ae2f8SJan Schmidt 		trans = btrfs_join_transaction(fs_info->extent_root);
13927a3ae2f8SJan Schmidt 		if (IS_ERR(trans))
13937a3ae2f8SJan Schmidt 			return PTR_ERR(trans);
13947a3ae2f8SJan Schmidt 
13954692cf58SJan Schmidt 		delayed_refs = &trans->transaction->delayed_refs;
13964692cf58SJan Schmidt 		spin_lock(&delayed_refs->lock);
13974692cf58SJan Schmidt 		btrfs_get_delayed_seq(delayed_refs, &seq_elem);
13984692cf58SJan Schmidt 		spin_unlock(&delayed_refs->lock);
13998445f61cSJan Schmidt 		btrfs_get_tree_mod_seq(fs_info, &tree_mod_seq_elem);
14007a3ae2f8SJan Schmidt 	}
14014692cf58SJan Schmidt 
14024692cf58SJan Schmidt 	ret = btrfs_find_all_leafs(trans, fs_info, extent_item_objectid,
14038445f61cSJan Schmidt 				   seq_elem.seq, tree_mod_seq_elem.seq, &refs,
14048445f61cSJan Schmidt 				   &extent_item_pos);
14054692cf58SJan Schmidt 	if (ret)
14064692cf58SJan Schmidt 		goto out;
14074692cf58SJan Schmidt 
1408cd1b413cSJan Schmidt 	ULIST_ITER_INIT(&ref_uiter);
1409cd1b413cSJan Schmidt 	while (!ret && (ref_node = ulist_next(refs, &ref_uiter))) {
1410976b1908SJan Schmidt 		ret = btrfs_find_all_roots(trans, fs_info, ref_node->val,
14118445f61cSJan Schmidt 						seq_elem.seq,
14128445f61cSJan Schmidt 						tree_mod_seq_elem.seq, &roots);
14134692cf58SJan Schmidt 		if (ret)
1414a542ad1bSJan Schmidt 			break;
1415cd1b413cSJan Schmidt 		ULIST_ITER_INIT(&root_uiter);
1416cd1b413cSJan Schmidt 		while (!ret && (root_node = ulist_next(roots, &root_uiter))) {
1417976b1908SJan Schmidt 			pr_debug("root %llu references leaf %llu, data list "
1418976b1908SJan Schmidt 				 "%#lx\n", root_node->val, ref_node->val,
1419976b1908SJan Schmidt 				 ref_node->aux);
1420976b1908SJan Schmidt 			ret = iterate_leaf_refs(
1421976b1908SJan Schmidt 				(struct extent_inode_elem *)ref_node->aux,
1422976b1908SJan Schmidt 				root_node->val, extent_item_objectid,
1423a542ad1bSJan Schmidt 				iterate, ctx);
14244692cf58SJan Schmidt 		}
1425976b1908SJan Schmidt 		ulist_free(roots);
1426976b1908SJan Schmidt 		roots = NULL;
1427a542ad1bSJan Schmidt 	}
1428a542ad1bSJan Schmidt 
1429976b1908SJan Schmidt 	free_leaf_list(refs);
14304692cf58SJan Schmidt 	ulist_free(roots);
14314692cf58SJan Schmidt out:
14327a3ae2f8SJan Schmidt 	if (!search_commit_root) {
14338445f61cSJan Schmidt 		btrfs_put_tree_mod_seq(fs_info, &tree_mod_seq_elem);
14344692cf58SJan Schmidt 		btrfs_put_delayed_seq(delayed_refs, &seq_elem);
14354692cf58SJan Schmidt 		btrfs_end_transaction(trans, fs_info->extent_root);
14367a3ae2f8SJan Schmidt 	}
14377a3ae2f8SJan Schmidt 
1438a542ad1bSJan Schmidt 	return ret;
1439a542ad1bSJan Schmidt }
1440a542ad1bSJan Schmidt 
1441a542ad1bSJan Schmidt int iterate_inodes_from_logical(u64 logical, struct btrfs_fs_info *fs_info,
1442a542ad1bSJan Schmidt 				struct btrfs_path *path,
1443a542ad1bSJan Schmidt 				iterate_extent_inodes_t *iterate, void *ctx)
1444a542ad1bSJan Schmidt {
1445a542ad1bSJan Schmidt 	int ret;
14464692cf58SJan Schmidt 	u64 extent_item_pos;
1447a542ad1bSJan Schmidt 	struct btrfs_key found_key;
14487a3ae2f8SJan Schmidt 	int search_commit_root = path->search_commit_root;
1449a542ad1bSJan Schmidt 
1450a542ad1bSJan Schmidt 	ret = extent_from_logical(fs_info, logical, path,
1451a542ad1bSJan Schmidt 					&found_key);
14524692cf58SJan Schmidt 	btrfs_release_path(path);
1453a542ad1bSJan Schmidt 	if (ret & BTRFS_EXTENT_FLAG_TREE_BLOCK)
1454a542ad1bSJan Schmidt 		ret = -EINVAL;
1455a542ad1bSJan Schmidt 	if (ret < 0)
1456a542ad1bSJan Schmidt 		return ret;
1457a542ad1bSJan Schmidt 
14584692cf58SJan Schmidt 	extent_item_pos = logical - found_key.objectid;
14597a3ae2f8SJan Schmidt 	ret = iterate_extent_inodes(fs_info, found_key.objectid,
14607a3ae2f8SJan Schmidt 					extent_item_pos, search_commit_root,
14617a3ae2f8SJan Schmidt 					iterate, ctx);
1462a542ad1bSJan Schmidt 
1463a542ad1bSJan Schmidt 	return ret;
1464a542ad1bSJan Schmidt }
1465a542ad1bSJan Schmidt 
1466a542ad1bSJan Schmidt static int iterate_irefs(u64 inum, struct btrfs_root *fs_root,
1467a542ad1bSJan Schmidt 				struct btrfs_path *path,
1468a542ad1bSJan Schmidt 				iterate_irefs_t *iterate, void *ctx)
1469a542ad1bSJan Schmidt {
1470aefc1eb1SJan Schmidt 	int ret = 0;
1471a542ad1bSJan Schmidt 	int slot;
1472a542ad1bSJan Schmidt 	u32 cur;
1473a542ad1bSJan Schmidt 	u32 len;
1474a542ad1bSJan Schmidt 	u32 name_len;
1475a542ad1bSJan Schmidt 	u64 parent = 0;
1476a542ad1bSJan Schmidt 	int found = 0;
1477a542ad1bSJan Schmidt 	struct extent_buffer *eb;
1478a542ad1bSJan Schmidt 	struct btrfs_item *item;
1479a542ad1bSJan Schmidt 	struct btrfs_inode_ref *iref;
1480a542ad1bSJan Schmidt 	struct btrfs_key found_key;
1481a542ad1bSJan Schmidt 
1482aefc1eb1SJan Schmidt 	while (!ret) {
1483b916a59aSJan Schmidt 		path->leave_spinning = 1;
1484a542ad1bSJan Schmidt 		ret = inode_ref_info(inum, parent ? parent+1 : 0, fs_root, path,
1485a542ad1bSJan Schmidt 					&found_key);
1486a542ad1bSJan Schmidt 		if (ret < 0)
1487a542ad1bSJan Schmidt 			break;
1488a542ad1bSJan Schmidt 		if (ret) {
1489a542ad1bSJan Schmidt 			ret = found ? 0 : -ENOENT;
1490a542ad1bSJan Schmidt 			break;
1491a542ad1bSJan Schmidt 		}
1492a542ad1bSJan Schmidt 		++found;
1493a542ad1bSJan Schmidt 
1494a542ad1bSJan Schmidt 		parent = found_key.offset;
1495a542ad1bSJan Schmidt 		slot = path->slots[0];
1496a542ad1bSJan Schmidt 		eb = path->nodes[0];
1497a542ad1bSJan Schmidt 		/* make sure we can use eb after releasing the path */
1498a542ad1bSJan Schmidt 		atomic_inc(&eb->refs);
1499b916a59aSJan Schmidt 		btrfs_tree_read_lock(eb);
1500b916a59aSJan Schmidt 		btrfs_set_lock_blocking_rw(eb, BTRFS_READ_LOCK);
1501a542ad1bSJan Schmidt 		btrfs_release_path(path);
1502a542ad1bSJan Schmidt 
1503a542ad1bSJan Schmidt 		item = btrfs_item_nr(eb, slot);
1504a542ad1bSJan Schmidt 		iref = btrfs_item_ptr(eb, slot, struct btrfs_inode_ref);
1505a542ad1bSJan Schmidt 
1506a542ad1bSJan Schmidt 		for (cur = 0; cur < btrfs_item_size(eb, item); cur += len) {
1507a542ad1bSJan Schmidt 			name_len = btrfs_inode_ref_name_len(eb, iref);
1508a542ad1bSJan Schmidt 			/* path must be released before calling iterate()! */
15094692cf58SJan Schmidt 			pr_debug("following ref at offset %u for inode %llu in "
15104692cf58SJan Schmidt 				 "tree %llu\n", cur,
15114692cf58SJan Schmidt 				 (unsigned long long)found_key.objectid,
15124692cf58SJan Schmidt 				 (unsigned long long)fs_root->objectid);
1513a542ad1bSJan Schmidt 			ret = iterate(parent, iref, eb, ctx);
1514aefc1eb1SJan Schmidt 			if (ret)
1515a542ad1bSJan Schmidt 				break;
1516a542ad1bSJan Schmidt 			len = sizeof(*iref) + name_len;
1517a542ad1bSJan Schmidt 			iref = (struct btrfs_inode_ref *)((char *)iref + len);
1518a542ad1bSJan Schmidt 		}
1519b916a59aSJan Schmidt 		btrfs_tree_read_unlock_blocking(eb);
1520a542ad1bSJan Schmidt 		free_extent_buffer(eb);
1521a542ad1bSJan Schmidt 	}
1522a542ad1bSJan Schmidt 
1523a542ad1bSJan Schmidt 	btrfs_release_path(path);
1524a542ad1bSJan Schmidt 
1525a542ad1bSJan Schmidt 	return ret;
1526a542ad1bSJan Schmidt }
1527a542ad1bSJan Schmidt 
1528a542ad1bSJan Schmidt /*
1529a542ad1bSJan Schmidt  * returns 0 if the path could be dumped (probably truncated)
1530a542ad1bSJan Schmidt  * returns <0 in case of an error
1531a542ad1bSJan Schmidt  */
1532a542ad1bSJan Schmidt static int inode_to_path(u64 inum, struct btrfs_inode_ref *iref,
1533a542ad1bSJan Schmidt 				struct extent_buffer *eb, void *ctx)
1534a542ad1bSJan Schmidt {
1535a542ad1bSJan Schmidt 	struct inode_fs_paths *ipath = ctx;
1536a542ad1bSJan Schmidt 	char *fspath;
1537a542ad1bSJan Schmidt 	char *fspath_min;
1538a542ad1bSJan Schmidt 	int i = ipath->fspath->elem_cnt;
1539a542ad1bSJan Schmidt 	const int s_ptr = sizeof(char *);
1540a542ad1bSJan Schmidt 	u32 bytes_left;
1541a542ad1bSJan Schmidt 
1542a542ad1bSJan Schmidt 	bytes_left = ipath->fspath->bytes_left > s_ptr ?
1543a542ad1bSJan Schmidt 					ipath->fspath->bytes_left - s_ptr : 0;
1544a542ad1bSJan Schmidt 
1545740c3d22SChris Mason 	fspath_min = (char *)ipath->fspath->val + (i + 1) * s_ptr;
1546a542ad1bSJan Schmidt 	fspath = iref_to_path(ipath->fs_root, ipath->btrfs_path, iref, eb,
1547a542ad1bSJan Schmidt 				inum, fspath_min, bytes_left);
1548a542ad1bSJan Schmidt 	if (IS_ERR(fspath))
1549a542ad1bSJan Schmidt 		return PTR_ERR(fspath);
1550a542ad1bSJan Schmidt 
1551a542ad1bSJan Schmidt 	if (fspath > fspath_min) {
15524692cf58SJan Schmidt 		pr_debug("path resolved: %s\n", fspath);
1553745c4d8eSJeff Mahoney 		ipath->fspath->val[i] = (u64)(unsigned long)fspath;
1554a542ad1bSJan Schmidt 		++ipath->fspath->elem_cnt;
1555a542ad1bSJan Schmidt 		ipath->fspath->bytes_left = fspath - fspath_min;
1556a542ad1bSJan Schmidt 	} else {
15574692cf58SJan Schmidt 		pr_debug("missed path, not enough space. missing bytes: %lu, "
15584692cf58SJan Schmidt 			 "constructed so far: %s\n",
15594692cf58SJan Schmidt 			 (unsigned long)(fspath_min - fspath), fspath_min);
1560a542ad1bSJan Schmidt 		++ipath->fspath->elem_missed;
1561a542ad1bSJan Schmidt 		ipath->fspath->bytes_missing += fspath_min - fspath;
1562a542ad1bSJan Schmidt 		ipath->fspath->bytes_left = 0;
1563a542ad1bSJan Schmidt 	}
1564a542ad1bSJan Schmidt 
1565a542ad1bSJan Schmidt 	return 0;
1566a542ad1bSJan Schmidt }
1567a542ad1bSJan Schmidt 
1568a542ad1bSJan Schmidt /*
1569a542ad1bSJan Schmidt  * this dumps all file system paths to the inode into the ipath struct, provided
1570a542ad1bSJan Schmidt  * is has been created large enough. each path is zero-terminated and accessed
1571740c3d22SChris Mason  * from ipath->fspath->val[i].
1572a542ad1bSJan Schmidt  * when it returns, there are ipath->fspath->elem_cnt number of paths available
1573740c3d22SChris Mason  * in ipath->fspath->val[]. when the allocated space wasn't sufficient, the
1574a542ad1bSJan Schmidt  * number of missed paths in recored in ipath->fspath->elem_missed, otherwise,
1575a542ad1bSJan Schmidt  * it's zero. ipath->fspath->bytes_missing holds the number of bytes that would
1576a542ad1bSJan Schmidt  * have been needed to return all paths.
1577a542ad1bSJan Schmidt  */
1578a542ad1bSJan Schmidt int paths_from_inode(u64 inum, struct inode_fs_paths *ipath)
1579a542ad1bSJan Schmidt {
1580a542ad1bSJan Schmidt 	return iterate_irefs(inum, ipath->fs_root, ipath->btrfs_path,
1581a542ad1bSJan Schmidt 				inode_to_path, ipath);
1582a542ad1bSJan Schmidt }
1583a542ad1bSJan Schmidt 
1584a542ad1bSJan Schmidt struct btrfs_data_container *init_data_container(u32 total_bytes)
1585a542ad1bSJan Schmidt {
1586a542ad1bSJan Schmidt 	struct btrfs_data_container *data;
1587a542ad1bSJan Schmidt 	size_t alloc_bytes;
1588a542ad1bSJan Schmidt 
1589a542ad1bSJan Schmidt 	alloc_bytes = max_t(size_t, total_bytes, sizeof(*data));
1590a542ad1bSJan Schmidt 	data = kmalloc(alloc_bytes, GFP_NOFS);
1591a542ad1bSJan Schmidt 	if (!data)
1592a542ad1bSJan Schmidt 		return ERR_PTR(-ENOMEM);
1593a542ad1bSJan Schmidt 
1594a542ad1bSJan Schmidt 	if (total_bytes >= sizeof(*data)) {
1595a542ad1bSJan Schmidt 		data->bytes_left = total_bytes - sizeof(*data);
1596a542ad1bSJan Schmidt 		data->bytes_missing = 0;
1597a542ad1bSJan Schmidt 	} else {
1598a542ad1bSJan Schmidt 		data->bytes_missing = sizeof(*data) - total_bytes;
1599a542ad1bSJan Schmidt 		data->bytes_left = 0;
1600a542ad1bSJan Schmidt 	}
1601a542ad1bSJan Schmidt 
1602a542ad1bSJan Schmidt 	data->elem_cnt = 0;
1603a542ad1bSJan Schmidt 	data->elem_missed = 0;
1604a542ad1bSJan Schmidt 
1605a542ad1bSJan Schmidt 	return data;
1606a542ad1bSJan Schmidt }
1607a542ad1bSJan Schmidt 
1608a542ad1bSJan Schmidt /*
1609a542ad1bSJan Schmidt  * allocates space to return multiple file system paths for an inode.
1610a542ad1bSJan Schmidt  * total_bytes to allocate are passed, note that space usable for actual path
1611a542ad1bSJan Schmidt  * information will be total_bytes - sizeof(struct inode_fs_paths).
1612a542ad1bSJan Schmidt  * the returned pointer must be freed with free_ipath() in the end.
1613a542ad1bSJan Schmidt  */
1614a542ad1bSJan Schmidt struct inode_fs_paths *init_ipath(s32 total_bytes, struct btrfs_root *fs_root,
1615a542ad1bSJan Schmidt 					struct btrfs_path *path)
1616a542ad1bSJan Schmidt {
1617a542ad1bSJan Schmidt 	struct inode_fs_paths *ifp;
1618a542ad1bSJan Schmidt 	struct btrfs_data_container *fspath;
1619a542ad1bSJan Schmidt 
1620a542ad1bSJan Schmidt 	fspath = init_data_container(total_bytes);
1621a542ad1bSJan Schmidt 	if (IS_ERR(fspath))
1622a542ad1bSJan Schmidt 		return (void *)fspath;
1623a542ad1bSJan Schmidt 
1624a542ad1bSJan Schmidt 	ifp = kmalloc(sizeof(*ifp), GFP_NOFS);
1625a542ad1bSJan Schmidt 	if (!ifp) {
1626a542ad1bSJan Schmidt 		kfree(fspath);
1627a542ad1bSJan Schmidt 		return ERR_PTR(-ENOMEM);
1628a542ad1bSJan Schmidt 	}
1629a542ad1bSJan Schmidt 
1630a542ad1bSJan Schmidt 	ifp->btrfs_path = path;
1631a542ad1bSJan Schmidt 	ifp->fspath = fspath;
1632a542ad1bSJan Schmidt 	ifp->fs_root = fs_root;
1633a542ad1bSJan Schmidt 
1634a542ad1bSJan Schmidt 	return ifp;
1635a542ad1bSJan Schmidt }
1636a542ad1bSJan Schmidt 
1637a542ad1bSJan Schmidt void free_ipath(struct inode_fs_paths *ipath)
1638a542ad1bSJan Schmidt {
16394735fb28SJesper Juhl 	if (!ipath)
16404735fb28SJesper Juhl 		return;
16415eb56d25SIlya Dryomov 	kfree(ipath->fspath);
1642a542ad1bSJan Schmidt 	kfree(ipath);
1643a542ad1bSJan Schmidt }
1644