4d012d46cec1688cfa58b90c1b352a52b68c6e87
[platform/upstream/btrfs-progs.git] / dir-item.c
1 /*
2  * Copyright (C) 2007 Oracle.  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 "ctree.h"
20 #include "disk-io.h"
21 #include "hash.h"
22 #include "transaction.h"
23
24 static struct btrfs_dir_item *btrfs_match_dir_item_name(struct btrfs_root *root,
25                               struct btrfs_path *path,
26                               const char *name, int name_len);
27
28 static struct btrfs_dir_item *insert_with_overflow(struct btrfs_trans_handle
29                                                    *trans,
30                                                    struct btrfs_root *root,
31                                                    struct btrfs_path *path,
32                                                    struct btrfs_key *cpu_key,
33                                                    u32 data_size,
34                                                    const char *name,
35                                                    int name_len)
36 {
37         int ret;
38         char *ptr;
39         struct btrfs_item *item;
40         struct extent_buffer *leaf;
41
42         ret = btrfs_insert_empty_item(trans, root, path, cpu_key, data_size);
43         if (ret == -EEXIST) {
44                 struct btrfs_dir_item *di;
45                 di = btrfs_match_dir_item_name(root, path, name, name_len);
46                 if (di)
47                         return ERR_PTR(-EEXIST);
48                 ret = btrfs_extend_item(trans, root, path, data_size);
49                 WARN_ON(ret > 0);
50         }
51         if (ret < 0)
52                 return ERR_PTR(ret);
53         WARN_ON(ret > 0);
54         leaf = path->nodes[0];
55         item = btrfs_item_nr(leaf, path->slots[0]);
56         ptr = btrfs_item_ptr(leaf, path->slots[0], char);
57         BUG_ON(data_size > btrfs_item_size(leaf, item));
58         ptr += btrfs_item_size(leaf, item) - data_size;
59         return (struct btrfs_dir_item *)ptr;
60 }
61
62 int btrfs_insert_xattr_item(struct btrfs_trans_handle *trans,
63                             struct btrfs_root *root, const char *name,
64                             u16 name_len, const void *data, u16 data_len,
65                             u64 dir)
66 {
67         int ret = 0;
68         struct btrfs_path *path;
69         struct btrfs_dir_item *dir_item;
70         unsigned long name_ptr, data_ptr;
71         struct btrfs_key key, location;
72         struct btrfs_disk_key disk_key;
73         struct extent_buffer *leaf;
74         u32 data_size;
75
76         key.objectid = dir;
77         btrfs_set_key_type(&key, BTRFS_XATTR_ITEM_KEY);
78         key.offset = btrfs_name_hash(name, name_len);
79         path = btrfs_alloc_path();
80         if (!path)
81                 return -ENOMEM;
82
83         data_size = sizeof(*dir_item) + name_len + data_len;
84         dir_item = insert_with_overflow(trans, root, path, &key, data_size,
85                                         name, name_len);
86         /*
87          * FIXME: at some point we should handle xattr's that are larger than
88          * what we can fit in our leaf.  We set location to NULL b/c we arent
89          * pointing at anything else, that will change if we store the xattr
90          * data in a separate inode.
91          */
92         BUG_ON(IS_ERR(dir_item));
93         memset(&location, 0, sizeof(location));
94
95         leaf = path->nodes[0];
96         btrfs_cpu_key_to_disk(&disk_key, &location);
97         btrfs_set_dir_item_key(leaf, dir_item, &disk_key);
98         btrfs_set_dir_type(leaf, dir_item, BTRFS_FT_XATTR);
99         btrfs_set_dir_name_len(leaf, dir_item, name_len);
100         btrfs_set_dir_data_len(leaf, dir_item, data_len);
101         name_ptr = (unsigned long)(dir_item + 1);
102         data_ptr = (unsigned long)((char *)name_ptr + name_len);
103
104         write_extent_buffer(leaf, name, name_ptr, name_len);
105         write_extent_buffer(leaf, data, data_ptr, data_len);
106         btrfs_mark_buffer_dirty(path->nodes[0]);
107
108         btrfs_free_path(path);
109         return ret;
110 }
111
112 int btrfs_insert_dir_item(struct btrfs_trans_handle *trans, struct btrfs_root
113                           *root, const char *name, int name_len, u64 dir,
114                           struct btrfs_key *location, u8 type, u64 index)
115 {
116         int ret = 0;
117         int ret2 = 0;
118         struct btrfs_path *path;
119         struct btrfs_dir_item *dir_item;
120         struct extent_buffer *leaf;
121         unsigned long name_ptr;
122         struct btrfs_key key;
123         struct btrfs_disk_key disk_key;
124         u32 data_size;
125
126         key.objectid = dir;
127         btrfs_set_key_type(&key, BTRFS_DIR_ITEM_KEY);
128         key.offset = btrfs_name_hash(name, name_len);
129         path = btrfs_alloc_path();
130         if (!path)
131                 return -ENOMEM;
132         data_size = sizeof(*dir_item) + name_len;
133         dir_item = insert_with_overflow(trans, root, path, &key, data_size,
134                                         name, name_len);
135         if (IS_ERR(dir_item)) {
136                 ret = PTR_ERR(dir_item);
137                 goto out;
138         }
139
140         leaf = path->nodes[0];
141         btrfs_cpu_key_to_disk(&disk_key, location);
142         btrfs_set_dir_item_key(leaf, dir_item, &disk_key);
143         btrfs_set_dir_type(leaf, dir_item, type);
144         btrfs_set_dir_data_len(leaf, dir_item, 0);
145         btrfs_set_dir_name_len(leaf, dir_item, name_len);
146         name_ptr = (unsigned long)(dir_item + 1);
147
148         write_extent_buffer(leaf, name, name_ptr, name_len);
149         btrfs_mark_buffer_dirty(leaf);
150
151         /* FIXME, use some real flag for selecting the extra index */
152         if (root == root->fs_info->tree_root) {
153                 ret = 0;
154                 goto out;
155         }
156         btrfs_release_path(path);
157
158         btrfs_set_key_type(&key, BTRFS_DIR_INDEX_KEY);
159         key.offset = index;
160         dir_item = insert_with_overflow(trans, root, path, &key, data_size,
161                                         name, name_len);
162         if (IS_ERR(dir_item)) {
163                 ret2 = PTR_ERR(dir_item);
164                 goto out;
165         }
166         leaf = path->nodes[0];
167         btrfs_cpu_key_to_disk(&disk_key, location);
168         btrfs_set_dir_item_key(leaf, dir_item, &disk_key);
169         btrfs_set_dir_type(leaf, dir_item, type);
170         btrfs_set_dir_data_len(leaf, dir_item, 0);
171         btrfs_set_dir_name_len(leaf, dir_item, name_len);
172         name_ptr = (unsigned long)(dir_item + 1);
173         write_extent_buffer(leaf, name, name_ptr, name_len);
174         btrfs_mark_buffer_dirty(leaf);
175 out:
176         btrfs_free_path(path);
177         if (ret)
178                 return ret;
179         if (ret2)
180                 return ret2;
181         return 0;
182 }
183
184 struct btrfs_dir_item *btrfs_lookup_dir_item(struct btrfs_trans_handle *trans,
185                                              struct btrfs_root *root,
186                                              struct btrfs_path *path, u64 dir,
187                                              const char *name, int name_len,
188                                              int mod)
189 {
190         int ret;
191         struct btrfs_key key;
192         int ins_len = mod < 0 ? -1 : 0;
193         int cow = mod != 0;
194         struct btrfs_key found_key;
195         struct extent_buffer *leaf;
196
197         key.objectid = dir;
198         btrfs_set_key_type(&key, BTRFS_DIR_ITEM_KEY);
199
200         key.offset = btrfs_name_hash(name, name_len);
201
202         ret = btrfs_search_slot(trans, root, &key, path, ins_len, cow);
203         if (ret < 0)
204                 return ERR_PTR(ret);
205         if (ret > 0) {
206                 if (path->slots[0] == 0)
207                         return NULL;
208                 path->slots[0]--;
209         }
210
211         leaf = path->nodes[0];
212         btrfs_item_key_to_cpu(leaf, &found_key, path->slots[0]);
213
214         if (found_key.objectid != dir ||
215             btrfs_key_type(&found_key) != BTRFS_DIR_ITEM_KEY ||
216             found_key.offset != key.offset)
217                 return NULL;
218
219         return btrfs_match_dir_item_name(root, path, name, name_len);
220 }
221
222 static struct btrfs_dir_item *btrfs_match_dir_item_name(struct btrfs_root *root,
223                               struct btrfs_path *path,
224                               const char *name, int name_len)
225 {
226         struct btrfs_dir_item *dir_item;
227         unsigned long name_ptr;
228         u32 total_len;
229         u32 cur = 0;
230         u32 this_len;
231         struct extent_buffer *leaf;
232
233         leaf = path->nodes[0];
234         dir_item = btrfs_item_ptr(leaf, path->slots[0], struct btrfs_dir_item);
235         total_len = btrfs_item_size_nr(leaf, path->slots[0]);
236         while(cur < total_len) {
237                 this_len = sizeof(*dir_item) +
238                         btrfs_dir_name_len(leaf, dir_item) +
239                         btrfs_dir_data_len(leaf, dir_item);
240                 name_ptr = (unsigned long)(dir_item + 1);
241
242                 if (btrfs_dir_name_len(leaf, dir_item) == name_len &&
243                     memcmp_extent_buffer(leaf, name, name_ptr, name_len) == 0)
244                         return dir_item;
245
246                 cur += this_len;
247                 dir_item = (struct btrfs_dir_item *)((char *)dir_item +
248                                                      this_len);
249         }
250         return NULL;
251 }