xref: /openbmc/linux/fs/f2fs/dir.c (revision 5b448065)
1 // SPDX-License-Identifier: GPL-2.0
2 /*
3  * fs/f2fs/dir.c
4  *
5  * Copyright (c) 2012 Samsung Electronics Co., Ltd.
6  *             http://www.samsung.com/
7  */
8 #include <asm/unaligned.h>
9 #include <linux/fs.h>
10 #include <linux/f2fs_fs.h>
11 #include <linux/sched/signal.h>
12 #include <linux/unicode.h>
13 #include "f2fs.h"
14 #include "node.h"
15 #include "acl.h"
16 #include "xattr.h"
17 #include <trace/events/f2fs.h>
18 
19 static unsigned long dir_blocks(struct inode *inode)
20 {
21 	return ((unsigned long long) (i_size_read(inode) + PAGE_SIZE - 1))
22 							>> PAGE_SHIFT;
23 }
24 
25 static unsigned int dir_buckets(unsigned int level, int dir_level)
26 {
27 	if (level + dir_level < MAX_DIR_HASH_DEPTH / 2)
28 		return 1 << (level + dir_level);
29 	else
30 		return MAX_DIR_BUCKETS;
31 }
32 
33 static unsigned int bucket_blocks(unsigned int level)
34 {
35 	if (level < MAX_DIR_HASH_DEPTH / 2)
36 		return 2;
37 	else
38 		return 4;
39 }
40 
41 static unsigned char f2fs_filetype_table[F2FS_FT_MAX] = {
42 	[F2FS_FT_UNKNOWN]	= DT_UNKNOWN,
43 	[F2FS_FT_REG_FILE]	= DT_REG,
44 	[F2FS_FT_DIR]		= DT_DIR,
45 	[F2FS_FT_CHRDEV]	= DT_CHR,
46 	[F2FS_FT_BLKDEV]	= DT_BLK,
47 	[F2FS_FT_FIFO]		= DT_FIFO,
48 	[F2FS_FT_SOCK]		= DT_SOCK,
49 	[F2FS_FT_SYMLINK]	= DT_LNK,
50 };
51 
52 static unsigned char f2fs_type_by_mode[S_IFMT >> S_SHIFT] = {
53 	[S_IFREG >> S_SHIFT]	= F2FS_FT_REG_FILE,
54 	[S_IFDIR >> S_SHIFT]	= F2FS_FT_DIR,
55 	[S_IFCHR >> S_SHIFT]	= F2FS_FT_CHRDEV,
56 	[S_IFBLK >> S_SHIFT]	= F2FS_FT_BLKDEV,
57 	[S_IFIFO >> S_SHIFT]	= F2FS_FT_FIFO,
58 	[S_IFSOCK >> S_SHIFT]	= F2FS_FT_SOCK,
59 	[S_IFLNK >> S_SHIFT]	= F2FS_FT_SYMLINK,
60 };
61 
62 static void set_de_type(struct f2fs_dir_entry *de, umode_t mode)
63 {
64 	de->file_type = f2fs_type_by_mode[(mode & S_IFMT) >> S_SHIFT];
65 }
66 
67 unsigned char f2fs_get_de_type(struct f2fs_dir_entry *de)
68 {
69 	if (de->file_type < F2FS_FT_MAX)
70 		return f2fs_filetype_table[de->file_type];
71 	return DT_UNKNOWN;
72 }
73 
74 /* If @dir is casefolded, initialize @fname->cf_name from @fname->usr_fname. */
75 int f2fs_init_casefolded_name(const struct inode *dir,
76 			      struct f2fs_filename *fname)
77 {
78 #ifdef CONFIG_UNICODE
79 	struct super_block *sb = dir->i_sb;
80 	struct f2fs_sb_info *sbi = F2FS_SB(sb);
81 
82 	if (IS_CASEFOLDED(dir)) {
83 		fname->cf_name.name = f2fs_kmalloc(sbi, F2FS_NAME_LEN,
84 						   GFP_NOFS);
85 		if (!fname->cf_name.name)
86 			return -ENOMEM;
87 		fname->cf_name.len = utf8_casefold(sb->s_encoding,
88 						   fname->usr_fname,
89 						   fname->cf_name.name,
90 						   F2FS_NAME_LEN);
91 		if ((int)fname->cf_name.len <= 0) {
92 			kfree(fname->cf_name.name);
93 			fname->cf_name.name = NULL;
94 			if (sb_has_strict_encoding(sb))
95 				return -EINVAL;
96 			/* fall back to treating name as opaque byte sequence */
97 		}
98 	}
99 #endif
100 	return 0;
101 }
102 
103 static int __f2fs_setup_filename(const struct inode *dir,
104 				 const struct fscrypt_name *crypt_name,
105 				 struct f2fs_filename *fname)
106 {
107 	int err;
108 
109 	memset(fname, 0, sizeof(*fname));
110 
111 	fname->usr_fname = crypt_name->usr_fname;
112 	fname->disk_name = crypt_name->disk_name;
113 #ifdef CONFIG_FS_ENCRYPTION
114 	fname->crypto_buf = crypt_name->crypto_buf;
115 #endif
116 	if (crypt_name->is_nokey_name) {
117 		/* hash was decoded from the no-key name */
118 		fname->hash = cpu_to_le32(crypt_name->hash);
119 	} else {
120 		err = f2fs_init_casefolded_name(dir, fname);
121 		if (err) {
122 			f2fs_free_filename(fname);
123 			return err;
124 		}
125 		f2fs_hash_filename(dir, fname);
126 	}
127 	return 0;
128 }
129 
130 /*
131  * Prepare to search for @iname in @dir.  This is similar to
132  * fscrypt_setup_filename(), but this also handles computing the casefolded name
133  * and the f2fs dirhash if needed, then packing all the information about this
134  * filename up into a 'struct f2fs_filename'.
135  */
136 int f2fs_setup_filename(struct inode *dir, const struct qstr *iname,
137 			int lookup, struct f2fs_filename *fname)
138 {
139 	struct fscrypt_name crypt_name;
140 	int err;
141 
142 	err = fscrypt_setup_filename(dir, iname, lookup, &crypt_name);
143 	if (err)
144 		return err;
145 
146 	return __f2fs_setup_filename(dir, &crypt_name, fname);
147 }
148 
149 /*
150  * Prepare to look up @dentry in @dir.  This is similar to
151  * fscrypt_prepare_lookup(), but this also handles computing the casefolded name
152  * and the f2fs dirhash if needed, then packing all the information about this
153  * filename up into a 'struct f2fs_filename'.
154  */
155 int f2fs_prepare_lookup(struct inode *dir, struct dentry *dentry,
156 			struct f2fs_filename *fname)
157 {
158 	struct fscrypt_name crypt_name;
159 	int err;
160 
161 	err = fscrypt_prepare_lookup(dir, dentry, &crypt_name);
162 	if (err)
163 		return err;
164 
165 	return __f2fs_setup_filename(dir, &crypt_name, fname);
166 }
167 
168 void f2fs_free_filename(struct f2fs_filename *fname)
169 {
170 #ifdef CONFIG_FS_ENCRYPTION
171 	kfree(fname->crypto_buf.name);
172 	fname->crypto_buf.name = NULL;
173 #endif
174 #ifdef CONFIG_UNICODE
175 	kfree(fname->cf_name.name);
176 	fname->cf_name.name = NULL;
177 #endif
178 }
179 
180 static unsigned long dir_block_index(unsigned int level,
181 				int dir_level, unsigned int idx)
182 {
183 	unsigned long i;
184 	unsigned long bidx = 0;
185 
186 	for (i = 0; i < level; i++)
187 		bidx += dir_buckets(i, dir_level) * bucket_blocks(i);
188 	bidx += idx * bucket_blocks(level);
189 	return bidx;
190 }
191 
192 static struct f2fs_dir_entry *find_in_block(struct inode *dir,
193 				struct page *dentry_page,
194 				const struct f2fs_filename *fname,
195 				int *max_slots)
196 {
197 	struct f2fs_dentry_block *dentry_blk;
198 	struct f2fs_dentry_ptr d;
199 
200 	dentry_blk = (struct f2fs_dentry_block *)page_address(dentry_page);
201 
202 	make_dentry_ptr_block(dir, &d, dentry_blk);
203 	return f2fs_find_target_dentry(&d, fname, max_slots);
204 }
205 
206 #ifdef CONFIG_UNICODE
207 /*
208  * Test whether a case-insensitive directory entry matches the filename
209  * being searched for.
210  *
211  * Returns 1 for a match, 0 for no match, and -errno on an error.
212  */
213 static int f2fs_match_ci_name(const struct inode *dir, const struct qstr *name,
214 			       const u8 *de_name, u32 de_name_len)
215 {
216 	const struct super_block *sb = dir->i_sb;
217 	const struct unicode_map *um = sb->s_encoding;
218 	struct fscrypt_str decrypted_name = FSTR_INIT(NULL, de_name_len);
219 	struct qstr entry = QSTR_INIT(de_name, de_name_len);
220 	int res;
221 
222 	if (IS_ENCRYPTED(dir)) {
223 		const struct fscrypt_str encrypted_name =
224 			FSTR_INIT((u8 *)de_name, de_name_len);
225 
226 		if (WARN_ON_ONCE(!fscrypt_has_encryption_key(dir)))
227 			return -EINVAL;
228 
229 		decrypted_name.name = kmalloc(de_name_len, GFP_KERNEL);
230 		if (!decrypted_name.name)
231 			return -ENOMEM;
232 		res = fscrypt_fname_disk_to_usr(dir, 0, 0, &encrypted_name,
233 						&decrypted_name);
234 		if (res < 0)
235 			goto out;
236 		entry.name = decrypted_name.name;
237 		entry.len = decrypted_name.len;
238 	}
239 
240 	res = utf8_strncasecmp_folded(um, name, &entry);
241 	/*
242 	 * In strict mode, ignore invalid names.  In non-strict mode,
243 	 * fall back to treating them as opaque byte sequences.
244 	 */
245 	if (res < 0 && !sb_has_strict_encoding(sb)) {
246 		res = name->len == entry.len &&
247 				memcmp(name->name, entry.name, name->len) == 0;
248 	} else {
249 		/* utf8_strncasecmp_folded returns 0 on match */
250 		res = (res == 0);
251 	}
252 out:
253 	kfree(decrypted_name.name);
254 	return res;
255 }
256 #endif /* CONFIG_UNICODE */
257 
258 static inline int f2fs_match_name(const struct inode *dir,
259 				   const struct f2fs_filename *fname,
260 				   const u8 *de_name, u32 de_name_len)
261 {
262 	struct fscrypt_name f;
263 
264 #ifdef CONFIG_UNICODE
265 	if (fname->cf_name.name) {
266 		struct qstr cf = FSTR_TO_QSTR(&fname->cf_name);
267 
268 		return f2fs_match_ci_name(dir, &cf, de_name, de_name_len);
269 	}
270 #endif
271 	f.usr_fname = fname->usr_fname;
272 	f.disk_name = fname->disk_name;
273 #ifdef CONFIG_FS_ENCRYPTION
274 	f.crypto_buf = fname->crypto_buf;
275 #endif
276 	return fscrypt_match_name(&f, de_name, de_name_len);
277 }
278 
279 struct f2fs_dir_entry *f2fs_find_target_dentry(const struct f2fs_dentry_ptr *d,
280 			const struct f2fs_filename *fname, int *max_slots)
281 {
282 	struct f2fs_dir_entry *de;
283 	unsigned long bit_pos = 0;
284 	int max_len = 0;
285 	int res = 0;
286 
287 	if (max_slots)
288 		*max_slots = 0;
289 	while (bit_pos < d->max) {
290 		if (!test_bit_le(bit_pos, d->bitmap)) {
291 			bit_pos++;
292 			max_len++;
293 			continue;
294 		}
295 
296 		de = &d->dentry[bit_pos];
297 
298 		if (unlikely(!de->name_len)) {
299 			bit_pos++;
300 			continue;
301 		}
302 
303 		if (de->hash_code == fname->hash) {
304 			res = f2fs_match_name(d->inode, fname,
305 					      d->filename[bit_pos],
306 					      le16_to_cpu(de->name_len));
307 			if (res < 0)
308 				return ERR_PTR(res);
309 			if (res)
310 				goto found;
311 		}
312 
313 		if (max_slots && max_len > *max_slots)
314 			*max_slots = max_len;
315 		max_len = 0;
316 
317 		bit_pos += GET_DENTRY_SLOTS(le16_to_cpu(de->name_len));
318 	}
319 
320 	de = NULL;
321 found:
322 	if (max_slots && max_len > *max_slots)
323 		*max_slots = max_len;
324 	return de;
325 }
326 
327 static struct f2fs_dir_entry *find_in_level(struct inode *dir,
328 					unsigned int level,
329 					const struct f2fs_filename *fname,
330 					struct page **res_page)
331 {
332 	int s = GET_DENTRY_SLOTS(fname->disk_name.len);
333 	unsigned int nbucket, nblock;
334 	unsigned int bidx, end_block;
335 	struct page *dentry_page;
336 	struct f2fs_dir_entry *de = NULL;
337 	bool room = false;
338 	int max_slots;
339 
340 	nbucket = dir_buckets(level, F2FS_I(dir)->i_dir_level);
341 	nblock = bucket_blocks(level);
342 
343 	bidx = dir_block_index(level, F2FS_I(dir)->i_dir_level,
344 			       le32_to_cpu(fname->hash) % nbucket);
345 	end_block = bidx + nblock;
346 
347 	for (; bidx < end_block; bidx++) {
348 		/* no need to allocate new dentry pages to all the indices */
349 		dentry_page = f2fs_find_data_page(dir, bidx);
350 		if (IS_ERR(dentry_page)) {
351 			if (PTR_ERR(dentry_page) == -ENOENT) {
352 				room = true;
353 				continue;
354 			} else {
355 				*res_page = dentry_page;
356 				break;
357 			}
358 		}
359 
360 		de = find_in_block(dir, dentry_page, fname, &max_slots);
361 		if (IS_ERR(de)) {
362 			*res_page = ERR_CAST(de);
363 			de = NULL;
364 			break;
365 		} else if (de) {
366 			*res_page = dentry_page;
367 			break;
368 		}
369 
370 		if (max_slots >= s)
371 			room = true;
372 		f2fs_put_page(dentry_page, 0);
373 	}
374 
375 	if (!de && room && F2FS_I(dir)->chash != fname->hash) {
376 		F2FS_I(dir)->chash = fname->hash;
377 		F2FS_I(dir)->clevel = level;
378 	}
379 
380 	return de;
381 }
382 
383 struct f2fs_dir_entry *__f2fs_find_entry(struct inode *dir,
384 					 const struct f2fs_filename *fname,
385 					 struct page **res_page)
386 {
387 	unsigned long npages = dir_blocks(dir);
388 	struct f2fs_dir_entry *de = NULL;
389 	unsigned int max_depth;
390 	unsigned int level;
391 
392 	*res_page = NULL;
393 
394 	if (f2fs_has_inline_dentry(dir)) {
395 		de = f2fs_find_in_inline_dir(dir, fname, res_page);
396 		goto out;
397 	}
398 
399 	if (npages == 0)
400 		goto out;
401 
402 	max_depth = F2FS_I(dir)->i_current_depth;
403 	if (unlikely(max_depth > MAX_DIR_HASH_DEPTH)) {
404 		f2fs_warn(F2FS_I_SB(dir), "Corrupted max_depth of %lu: %u",
405 			  dir->i_ino, max_depth);
406 		max_depth = MAX_DIR_HASH_DEPTH;
407 		f2fs_i_depth_write(dir, max_depth);
408 	}
409 
410 	for (level = 0; level < max_depth; level++) {
411 		de = find_in_level(dir, level, fname, res_page);
412 		if (de || IS_ERR(*res_page))
413 			break;
414 	}
415 out:
416 	/* This is to increase the speed of f2fs_create */
417 	if (!de)
418 		F2FS_I(dir)->task = current;
419 	return de;
420 }
421 
422 /*
423  * Find an entry in the specified directory with the wanted name.
424  * It returns the page where the entry was found (as a parameter - res_page),
425  * and the entry itself. Page is returned mapped and unlocked.
426  * Entry is guaranteed to be valid.
427  */
428 struct f2fs_dir_entry *f2fs_find_entry(struct inode *dir,
429 			const struct qstr *child, struct page **res_page)
430 {
431 	struct f2fs_dir_entry *de = NULL;
432 	struct f2fs_filename fname;
433 	int err;
434 
435 	err = f2fs_setup_filename(dir, child, 1, &fname);
436 	if (err) {
437 		if (err == -ENOENT)
438 			*res_page = NULL;
439 		else
440 			*res_page = ERR_PTR(err);
441 		return NULL;
442 	}
443 
444 	de = __f2fs_find_entry(dir, &fname, res_page);
445 
446 	f2fs_free_filename(&fname);
447 	return de;
448 }
449 
450 struct f2fs_dir_entry *f2fs_parent_dir(struct inode *dir, struct page **p)
451 {
452 	return f2fs_find_entry(dir, &dotdot_name, p);
453 }
454 
455 ino_t f2fs_inode_by_name(struct inode *dir, const struct qstr *qstr,
456 							struct page **page)
457 {
458 	ino_t res = 0;
459 	struct f2fs_dir_entry *de;
460 
461 	de = f2fs_find_entry(dir, qstr, page);
462 	if (de) {
463 		res = le32_to_cpu(de->ino);
464 		f2fs_put_page(*page, 0);
465 	}
466 
467 	return res;
468 }
469 
470 void f2fs_set_link(struct inode *dir, struct f2fs_dir_entry *de,
471 		struct page *page, struct inode *inode)
472 {
473 	enum page_type type = f2fs_has_inline_dentry(dir) ? NODE : DATA;
474 
475 	lock_page(page);
476 	f2fs_wait_on_page_writeback(page, type, true, true);
477 	de->ino = cpu_to_le32(inode->i_ino);
478 	set_de_type(de, inode->i_mode);
479 	set_page_dirty(page);
480 
481 	dir->i_mtime = dir->i_ctime = current_time(dir);
482 	f2fs_mark_inode_dirty_sync(dir, false);
483 	f2fs_put_page(page, 1);
484 }
485 
486 static void init_dent_inode(struct inode *dir, struct inode *inode,
487 			    const struct f2fs_filename *fname,
488 			    struct page *ipage)
489 {
490 	struct f2fs_inode *ri;
491 
492 	if (!fname) /* tmpfile case? */
493 		return;
494 
495 	f2fs_wait_on_page_writeback(ipage, NODE, true, true);
496 
497 	/* copy name info. to this inode page */
498 	ri = F2FS_INODE(ipage);
499 	ri->i_namelen = cpu_to_le32(fname->disk_name.len);
500 	memcpy(ri->i_name, fname->disk_name.name, fname->disk_name.len);
501 	if (IS_ENCRYPTED(dir)) {
502 		file_set_enc_name(inode);
503 		/*
504 		 * Roll-forward recovery doesn't have encryption keys available,
505 		 * so it can't compute the dirhash for encrypted+casefolded
506 		 * filenames.  Append it to i_name if possible.  Else, disable
507 		 * roll-forward recovery of the dentry (i.e., make fsync'ing the
508 		 * file force a checkpoint) by setting LOST_PINO.
509 		 */
510 		if (IS_CASEFOLDED(dir)) {
511 			if (fname->disk_name.len + sizeof(f2fs_hash_t) <=
512 			    F2FS_NAME_LEN)
513 				put_unaligned(fname->hash, (f2fs_hash_t *)
514 					&ri->i_name[fname->disk_name.len]);
515 			else
516 				file_lost_pino(inode);
517 		}
518 	}
519 	set_page_dirty(ipage);
520 }
521 
522 void f2fs_do_make_empty_dir(struct inode *inode, struct inode *parent,
523 					struct f2fs_dentry_ptr *d)
524 {
525 	struct fscrypt_str dot = FSTR_INIT(".", 1);
526 	struct fscrypt_str dotdot = FSTR_INIT("..", 2);
527 
528 	/* update dirent of "." */
529 	f2fs_update_dentry(inode->i_ino, inode->i_mode, d, &dot, 0, 0);
530 
531 	/* update dirent of ".." */
532 	f2fs_update_dentry(parent->i_ino, parent->i_mode, d, &dotdot, 0, 1);
533 }
534 
535 static int make_empty_dir(struct inode *inode,
536 		struct inode *parent, struct page *page)
537 {
538 	struct page *dentry_page;
539 	struct f2fs_dentry_block *dentry_blk;
540 	struct f2fs_dentry_ptr d;
541 
542 	if (f2fs_has_inline_dentry(inode))
543 		return f2fs_make_empty_inline_dir(inode, parent, page);
544 
545 	dentry_page = f2fs_get_new_data_page(inode, page, 0, true);
546 	if (IS_ERR(dentry_page))
547 		return PTR_ERR(dentry_page);
548 
549 	dentry_blk = page_address(dentry_page);
550 
551 	make_dentry_ptr_block(NULL, &d, dentry_blk);
552 	f2fs_do_make_empty_dir(inode, parent, &d);
553 
554 	set_page_dirty(dentry_page);
555 	f2fs_put_page(dentry_page, 1);
556 	return 0;
557 }
558 
559 struct page *f2fs_init_inode_metadata(struct inode *inode, struct inode *dir,
560 			const struct f2fs_filename *fname, struct page *dpage)
561 {
562 	struct page *page;
563 	int err;
564 
565 	if (is_inode_flag_set(inode, FI_NEW_INODE)) {
566 		page = f2fs_new_inode_page(inode);
567 		if (IS_ERR(page))
568 			return page;
569 
570 		if (S_ISDIR(inode->i_mode)) {
571 			/* in order to handle error case */
572 			get_page(page);
573 			err = make_empty_dir(inode, dir, page);
574 			if (err) {
575 				lock_page(page);
576 				goto put_error;
577 			}
578 			put_page(page);
579 		}
580 
581 		err = f2fs_init_acl(inode, dir, page, dpage);
582 		if (err)
583 			goto put_error;
584 
585 		err = f2fs_init_security(inode, dir,
586 					 fname ? fname->usr_fname : NULL, page);
587 		if (err)
588 			goto put_error;
589 
590 		if (IS_ENCRYPTED(inode)) {
591 			err = fscrypt_set_context(inode, page);
592 			if (err)
593 				goto put_error;
594 		}
595 	} else {
596 		page = f2fs_get_node_page(F2FS_I_SB(dir), inode->i_ino);
597 		if (IS_ERR(page))
598 			return page;
599 	}
600 
601 	init_dent_inode(dir, inode, fname, page);
602 
603 	/*
604 	 * This file should be checkpointed during fsync.
605 	 * We lost i_pino from now on.
606 	 */
607 	if (is_inode_flag_set(inode, FI_INC_LINK)) {
608 		if (!S_ISDIR(inode->i_mode))
609 			file_lost_pino(inode);
610 		/*
611 		 * If link the tmpfile to alias through linkat path,
612 		 * we should remove this inode from orphan list.
613 		 */
614 		if (inode->i_nlink == 0)
615 			f2fs_remove_orphan_inode(F2FS_I_SB(dir), inode->i_ino);
616 		f2fs_i_links_write(inode, true);
617 	}
618 	return page;
619 
620 put_error:
621 	clear_nlink(inode);
622 	f2fs_update_inode(inode, page);
623 	f2fs_put_page(page, 1);
624 	return ERR_PTR(err);
625 }
626 
627 void f2fs_update_parent_metadata(struct inode *dir, struct inode *inode,
628 						unsigned int current_depth)
629 {
630 	if (inode && is_inode_flag_set(inode, FI_NEW_INODE)) {
631 		if (S_ISDIR(inode->i_mode))
632 			f2fs_i_links_write(dir, true);
633 		clear_inode_flag(inode, FI_NEW_INODE);
634 	}
635 	dir->i_mtime = dir->i_ctime = current_time(dir);
636 	f2fs_mark_inode_dirty_sync(dir, false);
637 
638 	if (F2FS_I(dir)->i_current_depth != current_depth)
639 		f2fs_i_depth_write(dir, current_depth);
640 
641 	if (inode && is_inode_flag_set(inode, FI_INC_LINK))
642 		clear_inode_flag(inode, FI_INC_LINK);
643 }
644 
645 int f2fs_room_for_filename(const void *bitmap, int slots, int max_slots)
646 {
647 	int bit_start = 0;
648 	int zero_start, zero_end;
649 next:
650 	zero_start = find_next_zero_bit_le(bitmap, max_slots, bit_start);
651 	if (zero_start >= max_slots)
652 		return max_slots;
653 
654 	zero_end = find_next_bit_le(bitmap, max_slots, zero_start);
655 	if (zero_end - zero_start >= slots)
656 		return zero_start;
657 
658 	bit_start = zero_end + 1;
659 
660 	if (zero_end + 1 >= max_slots)
661 		return max_slots;
662 	goto next;
663 }
664 
665 bool f2fs_has_enough_room(struct inode *dir, struct page *ipage,
666 			  const struct f2fs_filename *fname)
667 {
668 	struct f2fs_dentry_ptr d;
669 	unsigned int bit_pos;
670 	int slots = GET_DENTRY_SLOTS(fname->disk_name.len);
671 
672 	make_dentry_ptr_inline(dir, &d, inline_data_addr(dir, ipage));
673 
674 	bit_pos = f2fs_room_for_filename(d.bitmap, slots, d.max);
675 
676 	return bit_pos < d.max;
677 }
678 
679 void f2fs_update_dentry(nid_t ino, umode_t mode, struct f2fs_dentry_ptr *d,
680 			const struct fscrypt_str *name, f2fs_hash_t name_hash,
681 			unsigned int bit_pos)
682 {
683 	struct f2fs_dir_entry *de;
684 	int slots = GET_DENTRY_SLOTS(name->len);
685 	int i;
686 
687 	de = &d->dentry[bit_pos];
688 	de->hash_code = name_hash;
689 	de->name_len = cpu_to_le16(name->len);
690 	memcpy(d->filename[bit_pos], name->name, name->len);
691 	de->ino = cpu_to_le32(ino);
692 	set_de_type(de, mode);
693 	for (i = 0; i < slots; i++) {
694 		__set_bit_le(bit_pos + i, (void *)d->bitmap);
695 		/* avoid wrong garbage data for readdir */
696 		if (i)
697 			(de + i)->name_len = 0;
698 	}
699 }
700 
701 int f2fs_add_regular_entry(struct inode *dir, const struct f2fs_filename *fname,
702 			   struct inode *inode, nid_t ino, umode_t mode)
703 {
704 	unsigned int bit_pos;
705 	unsigned int level;
706 	unsigned int current_depth;
707 	unsigned long bidx, block;
708 	unsigned int nbucket, nblock;
709 	struct page *dentry_page = NULL;
710 	struct f2fs_dentry_block *dentry_blk = NULL;
711 	struct f2fs_dentry_ptr d;
712 	struct page *page = NULL;
713 	int slots, err = 0;
714 
715 	level = 0;
716 	slots = GET_DENTRY_SLOTS(fname->disk_name.len);
717 
718 	current_depth = F2FS_I(dir)->i_current_depth;
719 	if (F2FS_I(dir)->chash == fname->hash) {
720 		level = F2FS_I(dir)->clevel;
721 		F2FS_I(dir)->chash = 0;
722 	}
723 
724 start:
725 	if (time_to_inject(F2FS_I_SB(dir), FAULT_DIR_DEPTH)) {
726 		f2fs_show_injection_info(F2FS_I_SB(dir), FAULT_DIR_DEPTH);
727 		return -ENOSPC;
728 	}
729 
730 	if (unlikely(current_depth == MAX_DIR_HASH_DEPTH))
731 		return -ENOSPC;
732 
733 	/* Increase the depth, if required */
734 	if (level == current_depth)
735 		++current_depth;
736 
737 	nbucket = dir_buckets(level, F2FS_I(dir)->i_dir_level);
738 	nblock = bucket_blocks(level);
739 
740 	bidx = dir_block_index(level, F2FS_I(dir)->i_dir_level,
741 				(le32_to_cpu(fname->hash) % nbucket));
742 
743 	for (block = bidx; block <= (bidx + nblock - 1); block++) {
744 		dentry_page = f2fs_get_new_data_page(dir, NULL, block, true);
745 		if (IS_ERR(dentry_page))
746 			return PTR_ERR(dentry_page);
747 
748 		dentry_blk = page_address(dentry_page);
749 		bit_pos = f2fs_room_for_filename(&dentry_blk->dentry_bitmap,
750 						slots, NR_DENTRY_IN_BLOCK);
751 		if (bit_pos < NR_DENTRY_IN_BLOCK)
752 			goto add_dentry;
753 
754 		f2fs_put_page(dentry_page, 1);
755 	}
756 
757 	/* Move to next level to find the empty slot for new dentry */
758 	++level;
759 	goto start;
760 add_dentry:
761 	f2fs_wait_on_page_writeback(dentry_page, DATA, true, true);
762 
763 	if (inode) {
764 		down_write(&F2FS_I(inode)->i_sem);
765 		page = f2fs_init_inode_metadata(inode, dir, fname, NULL);
766 		if (IS_ERR(page)) {
767 			err = PTR_ERR(page);
768 			goto fail;
769 		}
770 	}
771 
772 	make_dentry_ptr_block(NULL, &d, dentry_blk);
773 	f2fs_update_dentry(ino, mode, &d, &fname->disk_name, fname->hash,
774 			   bit_pos);
775 
776 	set_page_dirty(dentry_page);
777 
778 	if (inode) {
779 		f2fs_i_pino_write(inode, dir->i_ino);
780 
781 		/* synchronize inode page's data from inode cache */
782 		if (is_inode_flag_set(inode, FI_NEW_INODE))
783 			f2fs_update_inode(inode, page);
784 
785 		f2fs_put_page(page, 1);
786 	}
787 
788 	f2fs_update_parent_metadata(dir, inode, current_depth);
789 fail:
790 	if (inode)
791 		up_write(&F2FS_I(inode)->i_sem);
792 
793 	f2fs_put_page(dentry_page, 1);
794 
795 	return err;
796 }
797 
798 int f2fs_add_dentry(struct inode *dir, const struct f2fs_filename *fname,
799 		    struct inode *inode, nid_t ino, umode_t mode)
800 {
801 	int err = -EAGAIN;
802 
803 	if (f2fs_has_inline_dentry(dir))
804 		err = f2fs_add_inline_entry(dir, fname, inode, ino, mode);
805 	if (err == -EAGAIN)
806 		err = f2fs_add_regular_entry(dir, fname, inode, ino, mode);
807 
808 	f2fs_update_time(F2FS_I_SB(dir), REQ_TIME);
809 	return err;
810 }
811 
812 /*
813  * Caller should grab and release a rwsem by calling f2fs_lock_op() and
814  * f2fs_unlock_op().
815  */
816 int f2fs_do_add_link(struct inode *dir, const struct qstr *name,
817 				struct inode *inode, nid_t ino, umode_t mode)
818 {
819 	struct f2fs_filename fname;
820 	struct page *page = NULL;
821 	struct f2fs_dir_entry *de = NULL;
822 	int err;
823 
824 	err = f2fs_setup_filename(dir, name, 0, &fname);
825 	if (err)
826 		return err;
827 
828 	/*
829 	 * An immature stackable filesystem shows a race condition between lookup
830 	 * and create. If we have same task when doing lookup and create, it's
831 	 * definitely fine as expected by VFS normally. Otherwise, let's just
832 	 * verify on-disk dentry one more time, which guarantees filesystem
833 	 * consistency more.
834 	 */
835 	if (current != F2FS_I(dir)->task) {
836 		de = __f2fs_find_entry(dir, &fname, &page);
837 		F2FS_I(dir)->task = NULL;
838 	}
839 	if (de) {
840 		f2fs_put_page(page, 0);
841 		err = -EEXIST;
842 	} else if (IS_ERR(page)) {
843 		err = PTR_ERR(page);
844 	} else {
845 		err = f2fs_add_dentry(dir, &fname, inode, ino, mode);
846 	}
847 	f2fs_free_filename(&fname);
848 	return err;
849 }
850 
851 int f2fs_do_tmpfile(struct inode *inode, struct inode *dir)
852 {
853 	struct page *page;
854 	int err = 0;
855 
856 	down_write(&F2FS_I(inode)->i_sem);
857 	page = f2fs_init_inode_metadata(inode, dir, NULL, NULL);
858 	if (IS_ERR(page)) {
859 		err = PTR_ERR(page);
860 		goto fail;
861 	}
862 	f2fs_put_page(page, 1);
863 
864 	clear_inode_flag(inode, FI_NEW_INODE);
865 	f2fs_update_time(F2FS_I_SB(inode), REQ_TIME);
866 fail:
867 	up_write(&F2FS_I(inode)->i_sem);
868 	return err;
869 }
870 
871 void f2fs_drop_nlink(struct inode *dir, struct inode *inode)
872 {
873 	struct f2fs_sb_info *sbi = F2FS_I_SB(dir);
874 
875 	down_write(&F2FS_I(inode)->i_sem);
876 
877 	if (S_ISDIR(inode->i_mode))
878 		f2fs_i_links_write(dir, false);
879 	inode->i_ctime = current_time(inode);
880 
881 	f2fs_i_links_write(inode, false);
882 	if (S_ISDIR(inode->i_mode)) {
883 		f2fs_i_links_write(inode, false);
884 		f2fs_i_size_write(inode, 0);
885 	}
886 	up_write(&F2FS_I(inode)->i_sem);
887 
888 	if (inode->i_nlink == 0)
889 		f2fs_add_orphan_inode(inode);
890 	else
891 		f2fs_release_orphan_inode(sbi);
892 }
893 
894 /*
895  * It only removes the dentry from the dentry page, corresponding name
896  * entry in name page does not need to be touched during deletion.
897  */
898 void f2fs_delete_entry(struct f2fs_dir_entry *dentry, struct page *page,
899 					struct inode *dir, struct inode *inode)
900 {
901 	struct	f2fs_dentry_block *dentry_blk;
902 	unsigned int bit_pos;
903 	int slots = GET_DENTRY_SLOTS(le16_to_cpu(dentry->name_len));
904 	int i;
905 
906 	f2fs_update_time(F2FS_I_SB(dir), REQ_TIME);
907 
908 	if (F2FS_OPTION(F2FS_I_SB(dir)).fsync_mode == FSYNC_MODE_STRICT)
909 		f2fs_add_ino_entry(F2FS_I_SB(dir), dir->i_ino, TRANS_DIR_INO);
910 
911 	if (f2fs_has_inline_dentry(dir))
912 		return f2fs_delete_inline_entry(dentry, page, dir, inode);
913 
914 	lock_page(page);
915 	f2fs_wait_on_page_writeback(page, DATA, true, true);
916 
917 	dentry_blk = page_address(page);
918 	bit_pos = dentry - dentry_blk->dentry;
919 	for (i = 0; i < slots; i++)
920 		__clear_bit_le(bit_pos + i, &dentry_blk->dentry_bitmap);
921 
922 	/* Let's check and deallocate this dentry page */
923 	bit_pos = find_next_bit_le(&dentry_blk->dentry_bitmap,
924 			NR_DENTRY_IN_BLOCK,
925 			0);
926 	set_page_dirty(page);
927 
928 	if (bit_pos == NR_DENTRY_IN_BLOCK &&
929 		!f2fs_truncate_hole(dir, page->index, page->index + 1)) {
930 		f2fs_clear_page_cache_dirty_tag(page);
931 		clear_page_dirty_for_io(page);
932 		f2fs_clear_page_private(page);
933 		ClearPageUptodate(page);
934 		clear_cold_data(page);
935 		inode_dec_dirty_pages(dir);
936 		f2fs_remove_dirty_inode(dir);
937 	}
938 	f2fs_put_page(page, 1);
939 
940 	dir->i_ctime = dir->i_mtime = current_time(dir);
941 	f2fs_mark_inode_dirty_sync(dir, false);
942 
943 	if (inode)
944 		f2fs_drop_nlink(dir, inode);
945 }
946 
947 bool f2fs_empty_dir(struct inode *dir)
948 {
949 	unsigned long bidx;
950 	struct page *dentry_page;
951 	unsigned int bit_pos;
952 	struct f2fs_dentry_block *dentry_blk;
953 	unsigned long nblock = dir_blocks(dir);
954 
955 	if (f2fs_has_inline_dentry(dir))
956 		return f2fs_empty_inline_dir(dir);
957 
958 	for (bidx = 0; bidx < nblock; bidx++) {
959 		dentry_page = f2fs_get_lock_data_page(dir, bidx, false);
960 		if (IS_ERR(dentry_page)) {
961 			if (PTR_ERR(dentry_page) == -ENOENT)
962 				continue;
963 			else
964 				return false;
965 		}
966 
967 		dentry_blk = page_address(dentry_page);
968 		if (bidx == 0)
969 			bit_pos = 2;
970 		else
971 			bit_pos = 0;
972 		bit_pos = find_next_bit_le(&dentry_blk->dentry_bitmap,
973 						NR_DENTRY_IN_BLOCK,
974 						bit_pos);
975 
976 		f2fs_put_page(dentry_page, 1);
977 
978 		if (bit_pos < NR_DENTRY_IN_BLOCK)
979 			return false;
980 	}
981 	return true;
982 }
983 
984 int f2fs_fill_dentries(struct dir_context *ctx, struct f2fs_dentry_ptr *d,
985 			unsigned int start_pos, struct fscrypt_str *fstr)
986 {
987 	unsigned char d_type = DT_UNKNOWN;
988 	unsigned int bit_pos;
989 	struct f2fs_dir_entry *de = NULL;
990 	struct fscrypt_str de_name = FSTR_INIT(NULL, 0);
991 	struct f2fs_sb_info *sbi = F2FS_I_SB(d->inode);
992 	struct blk_plug plug;
993 	bool readdir_ra = sbi->readdir_ra == 1;
994 	int err = 0;
995 
996 	bit_pos = ((unsigned long)ctx->pos % d->max);
997 
998 	if (readdir_ra)
999 		blk_start_plug(&plug);
1000 
1001 	while (bit_pos < d->max) {
1002 		bit_pos = find_next_bit_le(d->bitmap, d->max, bit_pos);
1003 		if (bit_pos >= d->max)
1004 			break;
1005 
1006 		de = &d->dentry[bit_pos];
1007 		if (de->name_len == 0) {
1008 			bit_pos++;
1009 			ctx->pos = start_pos + bit_pos;
1010 			printk_ratelimited(
1011 				"%sF2FS-fs (%s): invalid namelen(0), ino:%u, run fsck to fix.",
1012 				KERN_WARNING, sbi->sb->s_id,
1013 				le32_to_cpu(de->ino));
1014 			set_sbi_flag(sbi, SBI_NEED_FSCK);
1015 			continue;
1016 		}
1017 
1018 		d_type = f2fs_get_de_type(de);
1019 
1020 		de_name.name = d->filename[bit_pos];
1021 		de_name.len = le16_to_cpu(de->name_len);
1022 
1023 		/* check memory boundary before moving forward */
1024 		bit_pos += GET_DENTRY_SLOTS(le16_to_cpu(de->name_len));
1025 		if (unlikely(bit_pos > d->max ||
1026 				le16_to_cpu(de->name_len) > F2FS_NAME_LEN)) {
1027 			f2fs_warn(sbi, "%s: corrupted namelen=%d, run fsck to fix.",
1028 				  __func__, le16_to_cpu(de->name_len));
1029 			set_sbi_flag(sbi, SBI_NEED_FSCK);
1030 			err = -EFSCORRUPTED;
1031 			goto out;
1032 		}
1033 
1034 		if (IS_ENCRYPTED(d->inode)) {
1035 			int save_len = fstr->len;
1036 
1037 			err = fscrypt_fname_disk_to_usr(d->inode,
1038 						(u32)le32_to_cpu(de->hash_code),
1039 						0, &de_name, fstr);
1040 			if (err)
1041 				goto out;
1042 
1043 			de_name = *fstr;
1044 			fstr->len = save_len;
1045 		}
1046 
1047 		if (!dir_emit(ctx, de_name.name, de_name.len,
1048 					le32_to_cpu(de->ino), d_type)) {
1049 			err = 1;
1050 			goto out;
1051 		}
1052 
1053 		if (readdir_ra)
1054 			f2fs_ra_node_page(sbi, le32_to_cpu(de->ino));
1055 
1056 		ctx->pos = start_pos + bit_pos;
1057 	}
1058 out:
1059 	if (readdir_ra)
1060 		blk_finish_plug(&plug);
1061 	return err;
1062 }
1063 
1064 static int f2fs_readdir(struct file *file, struct dir_context *ctx)
1065 {
1066 	struct inode *inode = file_inode(file);
1067 	unsigned long npages = dir_blocks(inode);
1068 	struct f2fs_dentry_block *dentry_blk = NULL;
1069 	struct page *dentry_page = NULL;
1070 	struct file_ra_state *ra = &file->f_ra;
1071 	loff_t start_pos = ctx->pos;
1072 	unsigned int n = ((unsigned long)ctx->pos / NR_DENTRY_IN_BLOCK);
1073 	struct f2fs_dentry_ptr d;
1074 	struct fscrypt_str fstr = FSTR_INIT(NULL, 0);
1075 	int err = 0;
1076 
1077 	if (IS_ENCRYPTED(inode)) {
1078 		err = fscrypt_prepare_readdir(inode);
1079 		if (err)
1080 			goto out;
1081 
1082 		err = fscrypt_fname_alloc_buffer(F2FS_NAME_LEN, &fstr);
1083 		if (err < 0)
1084 			goto out;
1085 	}
1086 
1087 	if (f2fs_has_inline_dentry(inode)) {
1088 		err = f2fs_read_inline_dir(file, ctx, &fstr);
1089 		goto out_free;
1090 	}
1091 
1092 	for (; n < npages; n++, ctx->pos = n * NR_DENTRY_IN_BLOCK) {
1093 
1094 		/* allow readdir() to be interrupted */
1095 		if (fatal_signal_pending(current)) {
1096 			err = -ERESTARTSYS;
1097 			goto out_free;
1098 		}
1099 		cond_resched();
1100 
1101 		/* readahead for multi pages of dir */
1102 		if (npages - n > 1 && !ra_has_index(ra, n))
1103 			page_cache_sync_readahead(inode->i_mapping, ra, file, n,
1104 				min(npages - n, (pgoff_t)MAX_DIR_RA_PAGES));
1105 
1106 		dentry_page = f2fs_find_data_page(inode, n);
1107 		if (IS_ERR(dentry_page)) {
1108 			err = PTR_ERR(dentry_page);
1109 			if (err == -ENOENT) {
1110 				err = 0;
1111 				continue;
1112 			} else {
1113 				goto out_free;
1114 			}
1115 		}
1116 
1117 		dentry_blk = page_address(dentry_page);
1118 
1119 		make_dentry_ptr_block(inode, &d, dentry_blk);
1120 
1121 		err = f2fs_fill_dentries(ctx, &d,
1122 				n * NR_DENTRY_IN_BLOCK, &fstr);
1123 		if (err) {
1124 			f2fs_put_page(dentry_page, 0);
1125 			break;
1126 		}
1127 
1128 		f2fs_put_page(dentry_page, 0);
1129 	}
1130 out_free:
1131 	fscrypt_fname_free_buffer(&fstr);
1132 out:
1133 	trace_f2fs_readdir(inode, start_pos, ctx->pos, err);
1134 	return err < 0 ? err : 0;
1135 }
1136 
1137 const struct file_operations f2fs_dir_operations = {
1138 	.llseek		= generic_file_llseek,
1139 	.read		= generic_read_dir,
1140 	.iterate_shared	= f2fs_readdir,
1141 	.fsync		= f2fs_sync_file,
1142 	.unlocked_ioctl	= f2fs_ioctl,
1143 #ifdef CONFIG_COMPAT
1144 	.compat_ioctl   = f2fs_compat_ioctl,
1145 #endif
1146 };
1147