1 /*
2  * fs/f2fs/dir.c
3  *
4  * Copyright (c) 2012 Samsung Electronics Co., Ltd.
5  *             http://www.samsung.com/
6  *
7  * This program is free software; you can redistribute it and/or modify
8  * it under the terms of the GNU General Public License version 2 as
9  * published by the Free Software Foundation.
10  */
11 #include <linux/fs.h>
12 #include <linux/f2fs_fs.h>
13 #include <linux/sched/signal.h>
14 #include "f2fs.h"
15 #include "node.h"
16 #include "acl.h"
17 #include "xattr.h"
18 #include <trace/events/f2fs.h>
19 
dir_blocks(struct inode * inode)20 static unsigned long dir_blocks(struct inode *inode)
21 {
22 	return ((unsigned long long) (i_size_read(inode) + PAGE_SIZE - 1))
23 							>> PAGE_SHIFT;
24 }
25 
dir_buckets(unsigned int level,int dir_level)26 static unsigned int dir_buckets(unsigned int level, int dir_level)
27 {
28 	if (level + dir_level < MAX_DIR_HASH_DEPTH / 2)
29 		return 1 << (level + dir_level);
30 	else
31 		return MAX_DIR_BUCKETS;
32 }
33 
bucket_blocks(unsigned int level)34 static unsigned int bucket_blocks(unsigned int level)
35 {
36 	if (level < MAX_DIR_HASH_DEPTH / 2)
37 		return 2;
38 	else
39 		return 4;
40 }
41 
42 static unsigned char f2fs_filetype_table[F2FS_FT_MAX] = {
43 	[F2FS_FT_UNKNOWN]	= DT_UNKNOWN,
44 	[F2FS_FT_REG_FILE]	= DT_REG,
45 	[F2FS_FT_DIR]		= DT_DIR,
46 	[F2FS_FT_CHRDEV]	= DT_CHR,
47 	[F2FS_FT_BLKDEV]	= DT_BLK,
48 	[F2FS_FT_FIFO]		= DT_FIFO,
49 	[F2FS_FT_SOCK]		= DT_SOCK,
50 	[F2FS_FT_SYMLINK]	= DT_LNK,
51 };
52 
53 static unsigned char f2fs_type_by_mode[S_IFMT >> S_SHIFT] = {
54 	[S_IFREG >> S_SHIFT]	= F2FS_FT_REG_FILE,
55 	[S_IFDIR >> S_SHIFT]	= F2FS_FT_DIR,
56 	[S_IFCHR >> S_SHIFT]	= F2FS_FT_CHRDEV,
57 	[S_IFBLK >> S_SHIFT]	= F2FS_FT_BLKDEV,
58 	[S_IFIFO >> S_SHIFT]	= F2FS_FT_FIFO,
59 	[S_IFSOCK >> S_SHIFT]	= F2FS_FT_SOCK,
60 	[S_IFLNK >> S_SHIFT]	= F2FS_FT_SYMLINK,
61 };
62 
set_de_type(struct f2fs_dir_entry * de,umode_t mode)63 static void set_de_type(struct f2fs_dir_entry *de, umode_t mode)
64 {
65 	de->file_type = f2fs_type_by_mode[(mode & S_IFMT) >> S_SHIFT];
66 }
67 
f2fs_get_de_type(struct f2fs_dir_entry * de)68 unsigned char f2fs_get_de_type(struct f2fs_dir_entry *de)
69 {
70 	if (de->file_type < F2FS_FT_MAX)
71 		return f2fs_filetype_table[de->file_type];
72 	return DT_UNKNOWN;
73 }
74 
dir_block_index(unsigned int level,int dir_level,unsigned int idx)75 static unsigned long dir_block_index(unsigned int level,
76 				int dir_level, unsigned int idx)
77 {
78 	unsigned long i;
79 	unsigned long bidx = 0;
80 
81 	for (i = 0; i < level; i++)
82 		bidx += dir_buckets(i, dir_level) * bucket_blocks(i);
83 	bidx += idx * bucket_blocks(level);
84 	return bidx;
85 }
86 
find_in_block(struct page * dentry_page,struct fscrypt_name * fname,f2fs_hash_t namehash,int * max_slots,struct page ** res_page)87 static struct f2fs_dir_entry *find_in_block(struct page *dentry_page,
88 				struct fscrypt_name *fname,
89 				f2fs_hash_t namehash,
90 				int *max_slots,
91 				struct page **res_page)
92 {
93 	struct f2fs_dentry_block *dentry_blk;
94 	struct f2fs_dir_entry *de;
95 	struct f2fs_dentry_ptr d;
96 
97 	dentry_blk = (struct f2fs_dentry_block *)page_address(dentry_page);
98 
99 	make_dentry_ptr_block(NULL, &d, dentry_blk);
100 	de = f2fs_find_target_dentry(fname, namehash, max_slots, &d);
101 	if (de)
102 		*res_page = dentry_page;
103 
104 	return de;
105 }
106 
f2fs_find_target_dentry(struct fscrypt_name * fname,f2fs_hash_t namehash,int * max_slots,struct f2fs_dentry_ptr * d)107 struct f2fs_dir_entry *f2fs_find_target_dentry(struct fscrypt_name *fname,
108 			f2fs_hash_t namehash, int *max_slots,
109 			struct f2fs_dentry_ptr *d)
110 {
111 	struct f2fs_dir_entry *de;
112 	unsigned long bit_pos = 0;
113 	int max_len = 0;
114 
115 	if (max_slots)
116 		*max_slots = 0;
117 	while (bit_pos < d->max) {
118 		if (!test_bit_le(bit_pos, d->bitmap)) {
119 			bit_pos++;
120 			max_len++;
121 			continue;
122 		}
123 
124 		de = &d->dentry[bit_pos];
125 
126 		if (unlikely(!de->name_len)) {
127 			bit_pos++;
128 			continue;
129 		}
130 
131 		if (de->hash_code == namehash &&
132 		    fscrypt_match_name(fname, d->filename[bit_pos],
133 				       le16_to_cpu(de->name_len)))
134 			goto found;
135 
136 		if (max_slots && max_len > *max_slots)
137 			*max_slots = max_len;
138 		max_len = 0;
139 
140 		bit_pos += GET_DENTRY_SLOTS(le16_to_cpu(de->name_len));
141 	}
142 
143 	de = NULL;
144 found:
145 	if (max_slots && max_len > *max_slots)
146 		*max_slots = max_len;
147 	return de;
148 }
149 
find_in_level(struct inode * dir,unsigned int level,struct fscrypt_name * fname,struct page ** res_page)150 static struct f2fs_dir_entry *find_in_level(struct inode *dir,
151 					unsigned int level,
152 					struct fscrypt_name *fname,
153 					struct page **res_page)
154 {
155 	struct qstr name = FSTR_TO_QSTR(&fname->disk_name);
156 	int s = GET_DENTRY_SLOTS(name.len);
157 	unsigned int nbucket, nblock;
158 	unsigned int bidx, end_block;
159 	struct page *dentry_page;
160 	struct f2fs_dir_entry *de = NULL;
161 	bool room = false;
162 	int max_slots;
163 	f2fs_hash_t namehash = f2fs_dentry_hash(&name, fname);
164 
165 	nbucket = dir_buckets(level, F2FS_I(dir)->i_dir_level);
166 	nblock = bucket_blocks(level);
167 
168 	bidx = dir_block_index(level, F2FS_I(dir)->i_dir_level,
169 					le32_to_cpu(namehash) % nbucket);
170 	end_block = bidx + nblock;
171 
172 	for (; bidx < end_block; bidx++) {
173 		/* no need to allocate new dentry pages to all the indices */
174 		dentry_page = f2fs_find_data_page(dir, bidx);
175 		if (IS_ERR(dentry_page)) {
176 			if (PTR_ERR(dentry_page) == -ENOENT) {
177 				room = true;
178 				continue;
179 			} else {
180 				*res_page = dentry_page;
181 				break;
182 			}
183 		}
184 
185 		de = find_in_block(dentry_page, fname, namehash, &max_slots,
186 								res_page);
187 		if (de)
188 			break;
189 
190 		if (max_slots >= s)
191 			room = true;
192 		f2fs_put_page(dentry_page, 0);
193 	}
194 
195 	if (!de && room && F2FS_I(dir)->chash != namehash) {
196 		F2FS_I(dir)->chash = namehash;
197 		F2FS_I(dir)->clevel = level;
198 	}
199 
200 	return de;
201 }
202 
__f2fs_find_entry(struct inode * dir,struct fscrypt_name * fname,struct page ** res_page)203 struct f2fs_dir_entry *__f2fs_find_entry(struct inode *dir,
204 			struct fscrypt_name *fname, struct page **res_page)
205 {
206 	unsigned long npages = dir_blocks(dir);
207 	struct f2fs_dir_entry *de = NULL;
208 	unsigned int max_depth;
209 	unsigned int level;
210 
211 	*res_page = NULL;
212 
213 	if (f2fs_has_inline_dentry(dir)) {
214 		de = f2fs_find_in_inline_dir(dir, fname, res_page);
215 		goto out;
216 	}
217 
218 	if (npages == 0)
219 		goto out;
220 
221 	max_depth = F2FS_I(dir)->i_current_depth;
222 	if (unlikely(max_depth > MAX_DIR_HASH_DEPTH)) {
223 		f2fs_msg(F2FS_I_SB(dir)->sb, KERN_WARNING,
224 				"Corrupted max_depth of %lu: %u",
225 				dir->i_ino, max_depth);
226 		max_depth = MAX_DIR_HASH_DEPTH;
227 		f2fs_i_depth_write(dir, max_depth);
228 	}
229 
230 	for (level = 0; level < max_depth; level++) {
231 		de = find_in_level(dir, level, fname, res_page);
232 		if (de || IS_ERR(*res_page))
233 			break;
234 	}
235 out:
236 	/* This is to increase the speed of f2fs_create */
237 	if (!de)
238 		F2FS_I(dir)->task = current;
239 	return de;
240 }
241 
242 /*
243  * Find an entry in the specified directory with the wanted name.
244  * It returns the page where the entry was found (as a parameter - res_page),
245  * and the entry itself. Page is returned mapped and unlocked.
246  * Entry is guaranteed to be valid.
247  */
f2fs_find_entry(struct inode * dir,const struct qstr * child,struct page ** res_page)248 struct f2fs_dir_entry *f2fs_find_entry(struct inode *dir,
249 			const struct qstr *child, struct page **res_page)
250 {
251 	struct f2fs_dir_entry *de = NULL;
252 	struct fscrypt_name fname;
253 	int err;
254 
255 	err = fscrypt_setup_filename(dir, child, 1, &fname);
256 	if (err) {
257 		if (err == -ENOENT)
258 			*res_page = NULL;
259 		else
260 			*res_page = ERR_PTR(err);
261 		return NULL;
262 	}
263 
264 	de = __f2fs_find_entry(dir, &fname, res_page);
265 
266 	fscrypt_free_filename(&fname);
267 	return de;
268 }
269 
f2fs_parent_dir(struct inode * dir,struct page ** p)270 struct f2fs_dir_entry *f2fs_parent_dir(struct inode *dir, struct page **p)
271 {
272 	struct qstr dotdot = QSTR_INIT("..", 2);
273 
274 	return f2fs_find_entry(dir, &dotdot, p);
275 }
276 
f2fs_inode_by_name(struct inode * dir,const struct qstr * qstr,struct page ** page)277 ino_t f2fs_inode_by_name(struct inode *dir, const struct qstr *qstr,
278 							struct page **page)
279 {
280 	ino_t res = 0;
281 	struct f2fs_dir_entry *de;
282 
283 	de = f2fs_find_entry(dir, qstr, page);
284 	if (de) {
285 		res = le32_to_cpu(de->ino);
286 		f2fs_put_page(*page, 0);
287 	}
288 
289 	return res;
290 }
291 
f2fs_set_link(struct inode * dir,struct f2fs_dir_entry * de,struct page * page,struct inode * inode)292 void f2fs_set_link(struct inode *dir, struct f2fs_dir_entry *de,
293 		struct page *page, struct inode *inode)
294 {
295 	enum page_type type = f2fs_has_inline_dentry(dir) ? NODE : DATA;
296 	lock_page(page);
297 	f2fs_wait_on_page_writeback(page, type, true);
298 	de->ino = cpu_to_le32(inode->i_ino);
299 	set_de_type(de, inode->i_mode);
300 	set_page_dirty(page);
301 
302 	dir->i_mtime = dir->i_ctime = current_time(dir);
303 	f2fs_mark_inode_dirty_sync(dir, false);
304 	f2fs_put_page(page, 1);
305 }
306 
init_dent_inode(const struct qstr * name,struct page * ipage)307 static void init_dent_inode(const struct qstr *name, struct page *ipage)
308 {
309 	struct f2fs_inode *ri;
310 
311 	f2fs_wait_on_page_writeback(ipage, NODE, true);
312 
313 	/* copy name info. to this inode page */
314 	ri = F2FS_INODE(ipage);
315 	ri->i_namelen = cpu_to_le32(name->len);
316 	memcpy(ri->i_name, name->name, name->len);
317 	set_page_dirty(ipage);
318 }
319 
f2fs_do_make_empty_dir(struct inode * inode,struct inode * parent,struct f2fs_dentry_ptr * d)320 void f2fs_do_make_empty_dir(struct inode *inode, struct inode *parent,
321 					struct f2fs_dentry_ptr *d)
322 {
323 	struct qstr dot = QSTR_INIT(".", 1);
324 	struct qstr dotdot = QSTR_INIT("..", 2);
325 
326 	/* update dirent of "." */
327 	f2fs_update_dentry(inode->i_ino, inode->i_mode, d, &dot, 0, 0);
328 
329 	/* update dirent of ".." */
330 	f2fs_update_dentry(parent->i_ino, parent->i_mode, d, &dotdot, 0, 1);
331 }
332 
make_empty_dir(struct inode * inode,struct inode * parent,struct page * page)333 static int make_empty_dir(struct inode *inode,
334 		struct inode *parent, struct page *page)
335 {
336 	struct page *dentry_page;
337 	struct f2fs_dentry_block *dentry_blk;
338 	struct f2fs_dentry_ptr d;
339 
340 	if (f2fs_has_inline_dentry(inode))
341 		return f2fs_make_empty_inline_dir(inode, parent, page);
342 
343 	dentry_page = f2fs_get_new_data_page(inode, page, 0, true);
344 	if (IS_ERR(dentry_page))
345 		return PTR_ERR(dentry_page);
346 
347 	dentry_blk = page_address(dentry_page);
348 
349 	make_dentry_ptr_block(NULL, &d, dentry_blk);
350 	f2fs_do_make_empty_dir(inode, parent, &d);
351 
352 	set_page_dirty(dentry_page);
353 	f2fs_put_page(dentry_page, 1);
354 	return 0;
355 }
356 
f2fs_init_inode_metadata(struct inode * inode,struct inode * dir,const struct qstr * new_name,const struct qstr * orig_name,struct page * dpage)357 struct page *f2fs_init_inode_metadata(struct inode *inode, struct inode *dir,
358 			const struct qstr *new_name, const struct qstr *orig_name,
359 			struct page *dpage)
360 {
361 	struct page *page;
362 	int dummy_encrypt = DUMMY_ENCRYPTION_ENABLED(F2FS_I_SB(dir));
363 	int err;
364 
365 	if (is_inode_flag_set(inode, FI_NEW_INODE)) {
366 		page = f2fs_new_inode_page(inode);
367 		if (IS_ERR(page))
368 			return page;
369 
370 		if (S_ISDIR(inode->i_mode)) {
371 			/* in order to handle error case */
372 			get_page(page);
373 			err = make_empty_dir(inode, dir, page);
374 			if (err) {
375 				lock_page(page);
376 				goto put_error;
377 			}
378 			put_page(page);
379 		}
380 
381 		err = f2fs_init_acl(inode, dir, page, dpage);
382 		if (err)
383 			goto put_error;
384 
385 		err = f2fs_init_security(inode, dir, orig_name, page);
386 		if (err)
387 			goto put_error;
388 
389 		if ((f2fs_encrypted_inode(dir) || dummy_encrypt) &&
390 					f2fs_may_encrypt(inode)) {
391 			err = fscrypt_inherit_context(dir, inode, page, false);
392 			if (err)
393 				goto put_error;
394 		}
395 	} else {
396 		page = f2fs_get_node_page(F2FS_I_SB(dir), inode->i_ino);
397 		if (IS_ERR(page))
398 			return page;
399 	}
400 
401 	if (new_name) {
402 		init_dent_inode(new_name, page);
403 		if (f2fs_encrypted_inode(dir))
404 			file_set_enc_name(inode);
405 	}
406 
407 	/*
408 	 * This file should be checkpointed during fsync.
409 	 * We lost i_pino from now on.
410 	 */
411 	if (is_inode_flag_set(inode, FI_INC_LINK)) {
412 		if (!S_ISDIR(inode->i_mode))
413 			file_lost_pino(inode);
414 		/*
415 		 * If link the tmpfile to alias through linkat path,
416 		 * we should remove this inode from orphan list.
417 		 */
418 		if (inode->i_nlink == 0)
419 			f2fs_remove_orphan_inode(F2FS_I_SB(dir), inode->i_ino);
420 		f2fs_i_links_write(inode, true);
421 	}
422 	return page;
423 
424 put_error:
425 	clear_nlink(inode);
426 	f2fs_update_inode(inode, page);
427 	f2fs_put_page(page, 1);
428 	return ERR_PTR(err);
429 }
430 
f2fs_update_parent_metadata(struct inode * dir,struct inode * inode,unsigned int current_depth)431 void f2fs_update_parent_metadata(struct inode *dir, struct inode *inode,
432 						unsigned int current_depth)
433 {
434 	if (inode && is_inode_flag_set(inode, FI_NEW_INODE)) {
435 		if (S_ISDIR(inode->i_mode))
436 			f2fs_i_links_write(dir, true);
437 		clear_inode_flag(inode, FI_NEW_INODE);
438 	}
439 	dir->i_mtime = dir->i_ctime = current_time(dir);
440 	f2fs_mark_inode_dirty_sync(dir, false);
441 
442 	if (F2FS_I(dir)->i_current_depth != current_depth)
443 		f2fs_i_depth_write(dir, current_depth);
444 
445 	if (inode && is_inode_flag_set(inode, FI_INC_LINK))
446 		clear_inode_flag(inode, FI_INC_LINK);
447 }
448 
f2fs_room_for_filename(const void * bitmap,int slots,int max_slots)449 int f2fs_room_for_filename(const void *bitmap, int slots, int max_slots)
450 {
451 	int bit_start = 0;
452 	int zero_start, zero_end;
453 next:
454 	zero_start = find_next_zero_bit_le(bitmap, max_slots, bit_start);
455 	if (zero_start >= max_slots)
456 		return max_slots;
457 
458 	zero_end = find_next_bit_le(bitmap, max_slots, zero_start);
459 	if (zero_end - zero_start >= slots)
460 		return zero_start;
461 
462 	bit_start = zero_end + 1;
463 
464 	if (zero_end + 1 >= max_slots)
465 		return max_slots;
466 	goto next;
467 }
468 
f2fs_update_dentry(nid_t ino,umode_t mode,struct f2fs_dentry_ptr * d,const struct qstr * name,f2fs_hash_t name_hash,unsigned int bit_pos)469 void f2fs_update_dentry(nid_t ino, umode_t mode, struct f2fs_dentry_ptr *d,
470 				const struct qstr *name, f2fs_hash_t name_hash,
471 				unsigned int bit_pos)
472 {
473 	struct f2fs_dir_entry *de;
474 	int slots = GET_DENTRY_SLOTS(name->len);
475 	int i;
476 
477 	de = &d->dentry[bit_pos];
478 	de->hash_code = name_hash;
479 	de->name_len = cpu_to_le16(name->len);
480 	memcpy(d->filename[bit_pos], name->name, name->len);
481 	de->ino = cpu_to_le32(ino);
482 	set_de_type(de, mode);
483 	for (i = 0; i < slots; i++) {
484 		__set_bit_le(bit_pos + i, (void *)d->bitmap);
485 		/* avoid wrong garbage data for readdir */
486 		if (i)
487 			(de + i)->name_len = 0;
488 	}
489 }
490 
f2fs_add_regular_entry(struct inode * dir,const struct qstr * new_name,const struct qstr * orig_name,struct inode * inode,nid_t ino,umode_t mode)491 int f2fs_add_regular_entry(struct inode *dir, const struct qstr *new_name,
492 				const struct qstr *orig_name,
493 				struct inode *inode, nid_t ino, umode_t mode)
494 {
495 	unsigned int bit_pos;
496 	unsigned int level;
497 	unsigned int current_depth;
498 	unsigned long bidx, block;
499 	f2fs_hash_t dentry_hash;
500 	unsigned int nbucket, nblock;
501 	struct page *dentry_page = NULL;
502 	struct f2fs_dentry_block *dentry_blk = NULL;
503 	struct f2fs_dentry_ptr d;
504 	struct page *page = NULL;
505 	int slots, err = 0;
506 
507 	level = 0;
508 	slots = GET_DENTRY_SLOTS(new_name->len);
509 	dentry_hash = f2fs_dentry_hash(new_name, NULL);
510 
511 	current_depth = F2FS_I(dir)->i_current_depth;
512 	if (F2FS_I(dir)->chash == dentry_hash) {
513 		level = F2FS_I(dir)->clevel;
514 		F2FS_I(dir)->chash = 0;
515 	}
516 
517 start:
518 	if (time_to_inject(F2FS_I_SB(dir), FAULT_DIR_DEPTH)) {
519 		f2fs_show_injection_info(FAULT_DIR_DEPTH);
520 		return -ENOSPC;
521 	}
522 
523 	if (unlikely(current_depth == MAX_DIR_HASH_DEPTH))
524 		return -ENOSPC;
525 
526 	/* Increase the depth, if required */
527 	if (level == current_depth)
528 		++current_depth;
529 
530 	nbucket = dir_buckets(level, F2FS_I(dir)->i_dir_level);
531 	nblock = bucket_blocks(level);
532 
533 	bidx = dir_block_index(level, F2FS_I(dir)->i_dir_level,
534 				(le32_to_cpu(dentry_hash) % nbucket));
535 
536 	for (block = bidx; block <= (bidx + nblock - 1); block++) {
537 		dentry_page = f2fs_get_new_data_page(dir, NULL, block, true);
538 		if (IS_ERR(dentry_page))
539 			return PTR_ERR(dentry_page);
540 
541 		dentry_blk = page_address(dentry_page);
542 		bit_pos = f2fs_room_for_filename(&dentry_blk->dentry_bitmap,
543 						slots, NR_DENTRY_IN_BLOCK);
544 		if (bit_pos < NR_DENTRY_IN_BLOCK)
545 			goto add_dentry;
546 
547 		f2fs_put_page(dentry_page, 1);
548 	}
549 
550 	/* Move to next level to find the empty slot for new dentry */
551 	++level;
552 	goto start;
553 add_dentry:
554 	f2fs_wait_on_page_writeback(dentry_page, DATA, true);
555 
556 	if (inode) {
557 		down_write(&F2FS_I(inode)->i_sem);
558 		page = f2fs_init_inode_metadata(inode, dir, new_name,
559 						orig_name, NULL);
560 		if (IS_ERR(page)) {
561 			err = PTR_ERR(page);
562 			goto fail;
563 		}
564 	}
565 
566 	make_dentry_ptr_block(NULL, &d, dentry_blk);
567 	f2fs_update_dentry(ino, mode, &d, new_name, dentry_hash, bit_pos);
568 
569 	set_page_dirty(dentry_page);
570 
571 	if (inode) {
572 		f2fs_i_pino_write(inode, dir->i_ino);
573 
574 		/* synchronize inode page's data from inode cache */
575 		if (is_inode_flag_set(inode, FI_NEW_INODE))
576 			f2fs_update_inode(inode, page);
577 
578 		f2fs_put_page(page, 1);
579 	}
580 
581 	f2fs_update_parent_metadata(dir, inode, current_depth);
582 fail:
583 	if (inode)
584 		up_write(&F2FS_I(inode)->i_sem);
585 
586 	f2fs_put_page(dentry_page, 1);
587 
588 	return err;
589 }
590 
f2fs_add_dentry(struct inode * dir,struct fscrypt_name * fname,struct inode * inode,nid_t ino,umode_t mode)591 int f2fs_add_dentry(struct inode *dir, struct fscrypt_name *fname,
592 				struct inode *inode, nid_t ino, umode_t mode)
593 {
594 	struct qstr new_name;
595 	int err = -EAGAIN;
596 
597 	new_name.name = fname_name(fname);
598 	new_name.len = fname_len(fname);
599 
600 	if (f2fs_has_inline_dentry(dir))
601 		err = f2fs_add_inline_entry(dir, &new_name, fname->usr_fname,
602 							inode, ino, mode);
603 	if (err == -EAGAIN)
604 		err = f2fs_add_regular_entry(dir, &new_name, fname->usr_fname,
605 							inode, ino, mode);
606 
607 	f2fs_update_time(F2FS_I_SB(dir), REQ_TIME);
608 	return err;
609 }
610 
611 /*
612  * Caller should grab and release a rwsem by calling f2fs_lock_op() and
613  * f2fs_unlock_op().
614  */
f2fs_do_add_link(struct inode * dir,const struct qstr * name,struct inode * inode,nid_t ino,umode_t mode)615 int f2fs_do_add_link(struct inode *dir, const struct qstr *name,
616 				struct inode *inode, nid_t ino, umode_t mode)
617 {
618 	struct fscrypt_name fname;
619 	struct page *page = NULL;
620 	struct f2fs_dir_entry *de = NULL;
621 	int err;
622 
623 	err = fscrypt_setup_filename(dir, name, 0, &fname);
624 	if (err)
625 		return err;
626 
627 	/*
628 	 * An immature stakable filesystem shows a race condition between lookup
629 	 * and create. If we have same task when doing lookup and create, it's
630 	 * definitely fine as expected by VFS normally. Otherwise, let's just
631 	 * verify on-disk dentry one more time, which guarantees filesystem
632 	 * consistency more.
633 	 */
634 	if (current != F2FS_I(dir)->task) {
635 		de = __f2fs_find_entry(dir, &fname, &page);
636 		F2FS_I(dir)->task = NULL;
637 	}
638 	if (de) {
639 		f2fs_put_page(page, 0);
640 		err = -EEXIST;
641 	} else if (IS_ERR(page)) {
642 		err = PTR_ERR(page);
643 	} else {
644 		err = f2fs_add_dentry(dir, &fname, inode, ino, mode);
645 	}
646 	fscrypt_free_filename(&fname);
647 	return err;
648 }
649 
f2fs_do_tmpfile(struct inode * inode,struct inode * dir)650 int f2fs_do_tmpfile(struct inode *inode, struct inode *dir)
651 {
652 	struct page *page;
653 	int err = 0;
654 
655 	down_write(&F2FS_I(inode)->i_sem);
656 	page = f2fs_init_inode_metadata(inode, dir, NULL, NULL, NULL);
657 	if (IS_ERR(page)) {
658 		err = PTR_ERR(page);
659 		goto fail;
660 	}
661 	f2fs_put_page(page, 1);
662 
663 	clear_inode_flag(inode, FI_NEW_INODE);
664 fail:
665 	up_write(&F2FS_I(inode)->i_sem);
666 	f2fs_update_time(F2FS_I_SB(inode), REQ_TIME);
667 	return err;
668 }
669 
f2fs_drop_nlink(struct inode * dir,struct inode * inode)670 void f2fs_drop_nlink(struct inode *dir, struct inode *inode)
671 {
672 	struct f2fs_sb_info *sbi = F2FS_I_SB(dir);
673 
674 	down_write(&F2FS_I(inode)->i_sem);
675 
676 	if (S_ISDIR(inode->i_mode))
677 		f2fs_i_links_write(dir, false);
678 	inode->i_ctime = current_time(inode);
679 
680 	f2fs_i_links_write(inode, false);
681 	if (S_ISDIR(inode->i_mode)) {
682 		f2fs_i_links_write(inode, false);
683 		f2fs_i_size_write(inode, 0);
684 	}
685 	up_write(&F2FS_I(inode)->i_sem);
686 
687 	if (inode->i_nlink == 0)
688 		f2fs_add_orphan_inode(inode);
689 	else
690 		f2fs_release_orphan_inode(sbi);
691 }
692 
693 /*
694  * It only removes the dentry from the dentry page, corresponding name
695  * entry in name page does not need to be touched during deletion.
696  */
f2fs_delete_entry(struct f2fs_dir_entry * dentry,struct page * page,struct inode * dir,struct inode * inode)697 void f2fs_delete_entry(struct f2fs_dir_entry *dentry, struct page *page,
698 					struct inode *dir, struct inode *inode)
699 {
700 	struct	f2fs_dentry_block *dentry_blk;
701 	unsigned int bit_pos;
702 	int slots = GET_DENTRY_SLOTS(le16_to_cpu(dentry->name_len));
703 	int i;
704 
705 	f2fs_update_time(F2FS_I_SB(dir), REQ_TIME);
706 
707 	if (F2FS_OPTION(F2FS_I_SB(dir)).fsync_mode == FSYNC_MODE_STRICT)
708 		f2fs_add_ino_entry(F2FS_I_SB(dir), dir->i_ino, TRANS_DIR_INO);
709 
710 	if (f2fs_has_inline_dentry(dir))
711 		return f2fs_delete_inline_entry(dentry, page, dir, inode);
712 
713 	lock_page(page);
714 	f2fs_wait_on_page_writeback(page, DATA, true);
715 
716 	dentry_blk = page_address(page);
717 	bit_pos = dentry - dentry_blk->dentry;
718 	for (i = 0; i < slots; i++)
719 		__clear_bit_le(bit_pos + i, &dentry_blk->dentry_bitmap);
720 
721 	/* Let's check and deallocate this dentry page */
722 	bit_pos = find_next_bit_le(&dentry_blk->dentry_bitmap,
723 			NR_DENTRY_IN_BLOCK,
724 			0);
725 	set_page_dirty(page);
726 
727 	dir->i_ctime = dir->i_mtime = current_time(dir);
728 	f2fs_mark_inode_dirty_sync(dir, false);
729 
730 	if (inode)
731 		f2fs_drop_nlink(dir, inode);
732 
733 	if (bit_pos == NR_DENTRY_IN_BLOCK &&
734 		!f2fs_truncate_hole(dir, page->index, page->index + 1)) {
735 		f2fs_clear_radix_tree_dirty_tag(page);
736 		clear_page_dirty_for_io(page);
737 		ClearPagePrivate(page);
738 		ClearPageUptodate(page);
739 		clear_cold_data(page);
740 		inode_dec_dirty_pages(dir);
741 		f2fs_remove_dirty_inode(dir);
742 	}
743 	f2fs_put_page(page, 1);
744 }
745 
f2fs_empty_dir(struct inode * dir)746 bool f2fs_empty_dir(struct inode *dir)
747 {
748 	unsigned long bidx;
749 	struct page *dentry_page;
750 	unsigned int bit_pos;
751 	struct f2fs_dentry_block *dentry_blk;
752 	unsigned long nblock = dir_blocks(dir);
753 
754 	if (f2fs_has_inline_dentry(dir))
755 		return f2fs_empty_inline_dir(dir);
756 
757 	for (bidx = 0; bidx < nblock; bidx++) {
758 		dentry_page = f2fs_get_lock_data_page(dir, bidx, false);
759 		if (IS_ERR(dentry_page)) {
760 			if (PTR_ERR(dentry_page) == -ENOENT)
761 				continue;
762 			else
763 				return false;
764 		}
765 
766 		dentry_blk = page_address(dentry_page);
767 		if (bidx == 0)
768 			bit_pos = 2;
769 		else
770 			bit_pos = 0;
771 		bit_pos = find_next_bit_le(&dentry_blk->dentry_bitmap,
772 						NR_DENTRY_IN_BLOCK,
773 						bit_pos);
774 
775 		f2fs_put_page(dentry_page, 1);
776 
777 		if (bit_pos < NR_DENTRY_IN_BLOCK)
778 			return false;
779 	}
780 	return true;
781 }
782 
f2fs_fill_dentries(struct dir_context * ctx,struct f2fs_dentry_ptr * d,unsigned int start_pos,struct fscrypt_str * fstr)783 int f2fs_fill_dentries(struct dir_context *ctx, struct f2fs_dentry_ptr *d,
784 			unsigned int start_pos, struct fscrypt_str *fstr)
785 {
786 	unsigned char d_type = DT_UNKNOWN;
787 	unsigned int bit_pos;
788 	struct f2fs_dir_entry *de = NULL;
789 	struct fscrypt_str de_name = FSTR_INIT(NULL, 0);
790 	struct f2fs_sb_info *sbi = F2FS_I_SB(d->inode);
791 	int err = 0;
792 
793 	bit_pos = ((unsigned long)ctx->pos % d->max);
794 
795 	while (bit_pos < d->max) {
796 		bit_pos = find_next_bit_le(d->bitmap, d->max, bit_pos);
797 		if (bit_pos >= d->max)
798 			break;
799 
800 		de = &d->dentry[bit_pos];
801 		if (de->name_len == 0) {
802 			bit_pos++;
803 			ctx->pos = start_pos + bit_pos;
804 			continue;
805 		}
806 
807 		d_type = f2fs_get_de_type(de);
808 
809 		de_name.name = d->filename[bit_pos];
810 		de_name.len = le16_to_cpu(de->name_len);
811 
812 		/* check memory boundary before moving forward */
813 		bit_pos += GET_DENTRY_SLOTS(le16_to_cpu(de->name_len));
814 		if (unlikely(bit_pos > d->max ||
815 				le16_to_cpu(de->name_len) > F2FS_NAME_LEN)) {
816 			f2fs_msg(sbi->sb, KERN_WARNING,
817 				"%s: corrupted namelen=%d, run fsck to fix.",
818 				__func__, le16_to_cpu(de->name_len));
819 			set_sbi_flag(sbi, SBI_NEED_FSCK);
820 			err = -EINVAL;
821 			goto out;
822 		}
823 
824 		if (f2fs_encrypted_inode(d->inode)) {
825 			int save_len = fstr->len;
826 
827 			err = fscrypt_fname_disk_to_usr(d->inode,
828 						(u32)de->hash_code, 0,
829 						&de_name, fstr);
830 			if (err)
831 				return err;
832 
833 			de_name = *fstr;
834 			fstr->len = save_len;
835 		}
836 
837 		if (!dir_emit(ctx, de_name.name, de_name.len,
838 					le32_to_cpu(de->ino), d_type))
839 			return 1;
840 
841 		if (sbi->readdir_ra == 1)
842 			f2fs_ra_node_page(sbi, le32_to_cpu(de->ino));
843 
844 		ctx->pos = start_pos + bit_pos;
845 	}
846 out:
847 	return err;
848 }
849 
f2fs_readdir(struct file * file,struct dir_context * ctx)850 static int f2fs_readdir(struct file *file, struct dir_context *ctx)
851 {
852 	struct inode *inode = file_inode(file);
853 	unsigned long npages = dir_blocks(inode);
854 	struct f2fs_dentry_block *dentry_blk = NULL;
855 	struct page *dentry_page = NULL;
856 	struct file_ra_state *ra = &file->f_ra;
857 	loff_t start_pos = ctx->pos;
858 	unsigned int n = ((unsigned long)ctx->pos / NR_DENTRY_IN_BLOCK);
859 	struct f2fs_dentry_ptr d;
860 	struct fscrypt_str fstr = FSTR_INIT(NULL, 0);
861 	int err = 0;
862 
863 	if (f2fs_encrypted_inode(inode)) {
864 		err = fscrypt_get_encryption_info(inode);
865 		if (err && err != -ENOKEY)
866 			goto out;
867 
868 		err = fscrypt_fname_alloc_buffer(inode, F2FS_NAME_LEN, &fstr);
869 		if (err < 0)
870 			goto out;
871 	}
872 
873 	if (f2fs_has_inline_dentry(inode)) {
874 		err = f2fs_read_inline_dir(file, ctx, &fstr);
875 		goto out_free;
876 	}
877 
878 	for (; n < npages; n++, ctx->pos = n * NR_DENTRY_IN_BLOCK) {
879 
880 		/* allow readdir() to be interrupted */
881 		if (fatal_signal_pending(current)) {
882 			err = -ERESTARTSYS;
883 			goto out_free;
884 		}
885 		cond_resched();
886 
887 		/* readahead for multi pages of dir */
888 		if (npages - n > 1 && !ra_has_index(ra, n))
889 			page_cache_sync_readahead(inode->i_mapping, ra, file, n,
890 				min(npages - n, (pgoff_t)MAX_DIR_RA_PAGES));
891 
892 		dentry_page = f2fs_get_lock_data_page(inode, n, false);
893 		if (IS_ERR(dentry_page)) {
894 			err = PTR_ERR(dentry_page);
895 			if (err == -ENOENT) {
896 				err = 0;
897 				continue;
898 			} else {
899 				goto out_free;
900 			}
901 		}
902 
903 		dentry_blk = page_address(dentry_page);
904 
905 		make_dentry_ptr_block(inode, &d, dentry_blk);
906 
907 		err = f2fs_fill_dentries(ctx, &d,
908 				n * NR_DENTRY_IN_BLOCK, &fstr);
909 		if (err) {
910 			f2fs_put_page(dentry_page, 1);
911 			break;
912 		}
913 
914 		f2fs_put_page(dentry_page, 1);
915 	}
916 out_free:
917 	fscrypt_fname_free_buffer(&fstr);
918 out:
919 	trace_f2fs_readdir(inode, start_pos, ctx->pos, err);
920 	return err < 0 ? err : 0;
921 }
922 
f2fs_dir_open(struct inode * inode,struct file * filp)923 static int f2fs_dir_open(struct inode *inode, struct file *filp)
924 {
925 	if (f2fs_encrypted_inode(inode))
926 		return fscrypt_get_encryption_info(inode) ? -EACCES : 0;
927 	return 0;
928 }
929 
930 const struct file_operations f2fs_dir_operations = {
931 	.llseek		= generic_file_llseek,
932 	.read		= generic_read_dir,
933 	.iterate_shared	= f2fs_readdir,
934 	.fsync		= f2fs_sync_file,
935 	.open		= f2fs_dir_open,
936 	.unlocked_ioctl	= f2fs_ioctl,
937 #ifdef CONFIG_COMPAT
938 	.compat_ioctl   = f2fs_compat_ioctl,
939 #endif
940 };
941