blob: 9626252ee6b47d2b391f3383cfa9b3bb80e4110c [file] [log] [blame]
Josef Bacik294e30f2013-10-09 12:00:56 -04001/*
2 * Copyright (C) 2013 Fusion IO. All rights reserved.
3 *
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.
7 *
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.
12 *
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.
17 */
18
19#include <linux/fs.h>
20#include <linux/mount.h>
21#include <linux/magic.h>
22#include "btrfs-tests.h"
23#include "../ctree.h"
Josef Bacikfaa2dbf2014-05-07 17:06:09 -040024#include "../volumes.h"
25#include "../disk-io.h"
26#include "../qgroup.h"
Josef Bacik294e30f2013-10-09 12:00:56 -040027
28static struct vfsmount *test_mnt = NULL;
29
Josef Bacikaaedb552013-10-11 14:44:09 -040030static const struct super_operations btrfs_test_super_ops = {
31 .alloc_inode = btrfs_alloc_inode,
32 .destroy_inode = btrfs_test_destroy_inode,
33};
34
Josef Bacik294e30f2013-10-09 12:00:56 -040035static struct dentry *btrfs_test_mount(struct file_system_type *fs_type,
36 int flags, const char *dev_name,
37 void *data)
38{
Josef Bacikaaedb552013-10-11 14:44:09 -040039 return mount_pseudo(fs_type, "btrfs_test:", &btrfs_test_super_ops,
40 NULL, BTRFS_TEST_MAGIC);
Josef Bacik294e30f2013-10-09 12:00:56 -040041}
42
43static struct file_system_type test_type = {
44 .name = "btrfs_test_fs",
45 .mount = btrfs_test_mount,
46 .kill_sb = kill_anon_super,
47};
48
49struct inode *btrfs_new_test_inode(void)
50{
51 return new_inode(test_mnt->mnt_sb);
52}
53
54int btrfs_init_test_fs(void)
55{
56 int ret;
57
58 ret = register_filesystem(&test_type);
59 if (ret) {
60 printk(KERN_ERR "btrfs: cannot register test file system\n");
61 return ret;
62 }
63
64 test_mnt = kern_mount(&test_type);
65 if (IS_ERR(test_mnt)) {
66 printk(KERN_ERR "btrfs: cannot mount test file system\n");
67 unregister_filesystem(&test_type);
68 return ret;
69 }
70 return 0;
71}
72
73void btrfs_destroy_test_fs(void)
74{
75 kern_unmount(test_mnt);
76 unregister_filesystem(&test_type);
77}
Josef Bacikfaa2dbf2014-05-07 17:06:09 -040078
79struct btrfs_fs_info *btrfs_alloc_dummy_fs_info(void)
80{
81 struct btrfs_fs_info *fs_info = kzalloc(sizeof(struct btrfs_fs_info),
82 GFP_NOFS);
83
84 if (!fs_info)
85 return fs_info;
86 fs_info->fs_devices = kzalloc(sizeof(struct btrfs_fs_devices),
87 GFP_NOFS);
88 if (!fs_info->fs_devices) {
89 kfree(fs_info);
90 return NULL;
91 }
92 fs_info->super_copy = kzalloc(sizeof(struct btrfs_super_block),
93 GFP_NOFS);
94 if (!fs_info->super_copy) {
95 kfree(fs_info->fs_devices);
96 kfree(fs_info);
97 return NULL;
98 }
99
100 if (init_srcu_struct(&fs_info->subvol_srcu)) {
101 kfree(fs_info->fs_devices);
102 kfree(fs_info->super_copy);
103 kfree(fs_info);
104 return NULL;
105 }
106
107 spin_lock_init(&fs_info->buffer_lock);
108 spin_lock_init(&fs_info->qgroup_lock);
109 spin_lock_init(&fs_info->qgroup_op_lock);
110 spin_lock_init(&fs_info->super_lock);
111 spin_lock_init(&fs_info->fs_roots_radix_lock);
112 spin_lock_init(&fs_info->tree_mod_seq_lock);
113 mutex_init(&fs_info->qgroup_ioctl_lock);
114 mutex_init(&fs_info->qgroup_rescan_lock);
115 rwlock_init(&fs_info->tree_mod_log_lock);
116 fs_info->running_transaction = NULL;
117 fs_info->qgroup_tree = RB_ROOT;
118 fs_info->qgroup_ulist = NULL;
119 atomic64_set(&fs_info->tree_mod_seq, 0);
120 INIT_LIST_HEAD(&fs_info->dirty_qgroups);
121 INIT_LIST_HEAD(&fs_info->dead_roots);
122 INIT_LIST_HEAD(&fs_info->tree_mod_seq_list);
123 INIT_RADIX_TREE(&fs_info->buffer_radix, GFP_ATOMIC);
124 INIT_RADIX_TREE(&fs_info->fs_roots_radix, GFP_ATOMIC);
125 return fs_info;
126}
127
128static void btrfs_free_dummy_fs_info(struct btrfs_fs_info *fs_info)
129{
130 struct radix_tree_iter iter;
131 void **slot;
132
133 spin_lock(&fs_info->buffer_lock);
134restart:
135 radix_tree_for_each_slot(slot, &fs_info->buffer_radix, &iter, 0) {
136 struct extent_buffer *eb;
137
Sasha Levinf1e3c282014-06-11 12:00:25 -0400138 eb = radix_tree_deref_slot_protected(slot, &fs_info->buffer_lock);
Josef Bacikfaa2dbf2014-05-07 17:06:09 -0400139 if (!eb)
140 continue;
141 /* Shouldn't happen but that kind of thinking creates CVE's */
142 if (radix_tree_exception(eb)) {
143 if (radix_tree_deref_retry(eb))
144 goto restart;
145 continue;
146 }
147 spin_unlock(&fs_info->buffer_lock);
148 free_extent_buffer_stale(eb);
149 spin_lock(&fs_info->buffer_lock);
150 }
151 spin_unlock(&fs_info->buffer_lock);
152
153 btrfs_free_qgroup_config(fs_info);
154 btrfs_free_fs_roots(fs_info);
155 cleanup_srcu_struct(&fs_info->subvol_srcu);
156 kfree(fs_info->super_copy);
157 kfree(fs_info->fs_devices);
158 kfree(fs_info);
159}
160
161void btrfs_free_dummy_root(struct btrfs_root *root)
162{
163 if (!root)
164 return;
165 if (root->node)
166 free_extent_buffer(root->node);
167 if (root->fs_info)
168 btrfs_free_dummy_fs_info(root->fs_info);
169 kfree(root);
170}
171