33 static int insert_normal_tree_ref(struct btrfs_root *root, u64 bytenr,  in insert_normal_tree_ref()  argument
59 	ret = btrfs_insert_empty_item(&trans, root, path, &ins, size);  in insert_normal_tree_ref()
86 static int add_tree_ref(struct btrfs_root *root, u64 bytenr, u64 num_bytes,  in add_tree_ref()  argument
109 	ret = btrfs_search_slot(&trans, root, &key, path, 0, 1);  in add_tree_ref()
131 	ret = btrfs_insert_empty_item(&trans, root, path, &key, 0);  in add_tree_ref()
138 static int remove_extent_item(struct btrfs_root *root, u64 bytenr,  in remove_extent_item()  argument
159 	ret = btrfs_search_slot(&trans, root, &key, path, -1, 1);  in remove_extent_item()
165 	btrfs_del_item(&trans, root, path);  in remove_extent_item()
170 static int remove_extent_ref(struct btrfs_root *root, u64 bytenr,  in remove_extent_ref()  argument
193 	ret = btrfs_search_slot(&trans, root, &key, path, 0, 1);  in remove_extent_ref()
215 	ret = btrfs_search_slot(&trans, root, &key, path, -1, 1);  in remove_extent_ref()
221 	btrfs_del_item(&trans, root, path);  in remove_extent_ref()
226 static int test_no_shared_qgroup(struct btrfs_root *root)  in test_no_shared_qgroup()  argument
229 	struct btrfs_fs_info *fs_info = root->fs_info;  in test_no_shared_qgroup()
248 	ret = insert_normal_tree_ref(root, 4096, 4096, 0, 5);  in test_no_shared_qgroup()
263 	ret = remove_extent_item(root, 4096, 4096);  in test_no_shared_qgroup()
293 static int test_multiple_refs(struct btrfs_root *root)  in test_multiple_refs()  argument
296 	struct btrfs_fs_info *fs_info = root->fs_info;  in test_multiple_refs()
310 	ret = insert_normal_tree_ref(root, 4096, 4096, 0, 5);  in test_multiple_refs()
332 	ret = add_tree_ref(root, 4096, 4096, 0, 256);  in test_multiple_refs()
359 	ret = remove_extent_ref(root, 4096, 4096, 0, 256);  in test_multiple_refs()
391 	struct btrfs_root *root;  in btrfs_test_qgroups()  local
395 	root = btrfs_alloc_dummy_root();  in btrfs_test_qgroups()
396 	if (IS_ERR(root)) {  in btrfs_test_qgroups()
398 		return PTR_ERR(root);  in btrfs_test_qgroups()
401 	root->fs_info = btrfs_alloc_dummy_fs_info();  in btrfs_test_qgroups()
402 	if (!root->fs_info) {  in btrfs_test_qgroups()
408 	root->fs_info->extent_root = root;  in btrfs_test_qgroups()
414 	root->fs_info->tree_root = root;  in btrfs_test_qgroups()
415 	root->fs_info->quota_root = root;  in btrfs_test_qgroups()
416 	root->fs_info->quota_enabled = 1;  in btrfs_test_qgroups()
422 	root->node = alloc_test_extent_buffer(root->fs_info, 4096);  in btrfs_test_qgroups()
423 	if (!root->node) {  in btrfs_test_qgroups()
428 	btrfs_set_header_level(root->node, 0);  in btrfs_test_qgroups()
429 	btrfs_set_header_nritems(root->node, 0);  in btrfs_test_qgroups()
430 	root->alloc_bytenr += 8192;  in btrfs_test_qgroups()
440 	root->fs_info->fs_root = tmp_root;  in btrfs_test_qgroups()
441 	ret = btrfs_insert_fs_root(root->fs_info, tmp_root);  in btrfs_test_qgroups()
455 	ret = btrfs_insert_fs_root(root->fs_info, tmp_root);  in btrfs_test_qgroups()
462 	ret = test_no_shared_qgroup(root);  in btrfs_test_qgroups()
465 	ret = test_multiple_refs(root);  in btrfs_test_qgroups()
467 	btrfs_free_dummy_root(root);  in btrfs_test_qgroups()