xref: /kernel/linux/linux-6.6/fs/exfat/dir.c (revision 62306a36)
1// SPDX-License-Identifier: GPL-2.0-or-later
2/*
3 * Copyright (C) 2012-2013 Samsung Electronics Co., Ltd.
4 */
5
6#include <linux/slab.h>
7#include <linux/compat.h>
8#include <linux/bio.h>
9#include <linux/buffer_head.h>
10
11#include "exfat_raw.h"
12#include "exfat_fs.h"
13
14static int exfat_extract_uni_name(struct exfat_dentry *ep,
15		unsigned short *uniname)
16{
17	int i, len = 0;
18
19	for (i = 0; i < EXFAT_FILE_NAME_LEN; i++) {
20		*uniname = le16_to_cpu(ep->dentry.name.unicode_0_14[i]);
21		if (*uniname == 0x0)
22			return len;
23		uniname++;
24		len++;
25	}
26
27	*uniname = 0x0;
28	return len;
29
30}
31
32static int exfat_get_uniname_from_ext_entry(struct super_block *sb,
33		struct exfat_chain *p_dir, int entry, unsigned short *uniname)
34{
35	int i, err;
36	struct exfat_entry_set_cache es;
37	unsigned int uni_len = 0, len;
38
39	err = exfat_get_dentry_set(&es, sb, p_dir, entry, ES_ALL_ENTRIES);
40	if (err)
41		return err;
42
43	/*
44	 * First entry  : file entry
45	 * Second entry : stream-extension entry
46	 * Third entry  : first file-name entry
47	 * So, the index of first file-name dentry should start from 2.
48	 */
49	for (i = ES_IDX_FIRST_FILENAME; i < es.num_entries; i++) {
50		struct exfat_dentry *ep = exfat_get_dentry_cached(&es, i);
51
52		/* end of name entry */
53		if (exfat_get_entry_type(ep) != TYPE_EXTEND)
54			break;
55
56		len = exfat_extract_uni_name(ep, uniname);
57		uni_len += len;
58		if (len != EXFAT_FILE_NAME_LEN || uni_len >= MAX_NAME_LENGTH)
59			break;
60		uniname += EXFAT_FILE_NAME_LEN;
61	}
62
63	exfat_put_dentry_set(&es, false);
64	return 0;
65}
66
67/* read a directory entry from the opened directory */
68static int exfat_readdir(struct inode *inode, loff_t *cpos, struct exfat_dir_entry *dir_entry)
69{
70	int i, dentries_per_clu, num_ext, err;
71	unsigned int type, clu_offset, max_dentries;
72	struct exfat_chain dir, clu;
73	struct exfat_uni_name uni_name;
74	struct exfat_dentry *ep;
75	struct super_block *sb = inode->i_sb;
76	struct exfat_sb_info *sbi = EXFAT_SB(sb);
77	struct exfat_inode_info *ei = EXFAT_I(inode);
78	unsigned int dentry = EXFAT_B_TO_DEN(*cpos) & 0xFFFFFFFF;
79	struct buffer_head *bh;
80
81	/* check if the given file ID is opened */
82	if (ei->type != TYPE_DIR)
83		return -EPERM;
84
85	if (ei->entry == -1)
86		exfat_chain_set(&dir, sbi->root_dir, 0, ALLOC_FAT_CHAIN);
87	else
88		exfat_chain_set(&dir, ei->start_clu,
89			EXFAT_B_TO_CLU(i_size_read(inode), sbi), ei->flags);
90
91	dentries_per_clu = sbi->dentries_per_clu;
92	max_dentries = (unsigned int)min_t(u64, MAX_EXFAT_DENTRIES,
93				(u64)EXFAT_CLU_TO_DEN(sbi->num_clusters, sbi));
94
95	clu_offset = EXFAT_DEN_TO_CLU(dentry, sbi);
96	exfat_chain_dup(&clu, &dir);
97
98	if (clu.flags == ALLOC_NO_FAT_CHAIN) {
99		clu.dir += clu_offset;
100		clu.size -= clu_offset;
101	} else {
102		/* hint_information */
103		if (clu_offset > 0 && ei->hint_bmap.off != EXFAT_EOF_CLUSTER &&
104		    ei->hint_bmap.off > 0 && clu_offset >= ei->hint_bmap.off) {
105			clu_offset -= ei->hint_bmap.off;
106			clu.dir = ei->hint_bmap.clu;
107		}
108
109		while (clu_offset > 0 && clu.dir != EXFAT_EOF_CLUSTER) {
110			if (exfat_get_next_cluster(sb, &(clu.dir)))
111				return -EIO;
112
113			clu_offset--;
114		}
115	}
116
117	while (clu.dir != EXFAT_EOF_CLUSTER && dentry < max_dentries) {
118		i = dentry & (dentries_per_clu - 1);
119
120		for ( ; i < dentries_per_clu; i++, dentry++) {
121			ep = exfat_get_dentry(sb, &clu, i, &bh);
122			if (!ep)
123				return -EIO;
124
125			type = exfat_get_entry_type(ep);
126			if (type == TYPE_UNUSED) {
127				brelse(bh);
128				break;
129			}
130
131			if (type != TYPE_FILE && type != TYPE_DIR) {
132				brelse(bh);
133				continue;
134			}
135
136			num_ext = ep->dentry.file.num_ext;
137			dir_entry->attr = le16_to_cpu(ep->dentry.file.attr);
138			exfat_get_entry_time(sbi, &dir_entry->crtime,
139					ep->dentry.file.create_tz,
140					ep->dentry.file.create_time,
141					ep->dentry.file.create_date,
142					ep->dentry.file.create_time_cs);
143			exfat_get_entry_time(sbi, &dir_entry->mtime,
144					ep->dentry.file.modify_tz,
145					ep->dentry.file.modify_time,
146					ep->dentry.file.modify_date,
147					ep->dentry.file.modify_time_cs);
148			exfat_get_entry_time(sbi, &dir_entry->atime,
149					ep->dentry.file.access_tz,
150					ep->dentry.file.access_time,
151					ep->dentry.file.access_date,
152					0);
153
154			*uni_name.name = 0x0;
155			err = exfat_get_uniname_from_ext_entry(sb, &clu, i,
156				uni_name.name);
157			if (err) {
158				brelse(bh);
159				continue;
160			}
161			exfat_utf16_to_nls(sb, &uni_name,
162				dir_entry->namebuf.lfn,
163				dir_entry->namebuf.lfnbuf_len);
164			brelse(bh);
165
166			ep = exfat_get_dentry(sb, &clu, i + 1, &bh);
167			if (!ep)
168				return -EIO;
169			dir_entry->size =
170				le64_to_cpu(ep->dentry.stream.valid_size);
171			dir_entry->entry = dentry;
172			brelse(bh);
173
174			ei->hint_bmap.off = EXFAT_DEN_TO_CLU(dentry, sbi);
175			ei->hint_bmap.clu = clu.dir;
176
177			*cpos = EXFAT_DEN_TO_B(dentry + 1 + num_ext);
178			return 0;
179		}
180
181		if (clu.flags == ALLOC_NO_FAT_CHAIN) {
182			if (--clu.size > 0)
183				clu.dir++;
184			else
185				clu.dir = EXFAT_EOF_CLUSTER;
186		} else {
187			if (exfat_get_next_cluster(sb, &(clu.dir)))
188				return -EIO;
189		}
190	}
191
192	dir_entry->namebuf.lfn[0] = '\0';
193	*cpos = EXFAT_DEN_TO_B(dentry);
194	return 0;
195}
196
197static void exfat_init_namebuf(struct exfat_dentry_namebuf *nb)
198{
199	nb->lfn = NULL;
200	nb->lfnbuf_len = 0;
201}
202
203static int exfat_alloc_namebuf(struct exfat_dentry_namebuf *nb)
204{
205	nb->lfn = __getname();
206	if (!nb->lfn)
207		return -ENOMEM;
208	nb->lfnbuf_len = MAX_VFSNAME_BUF_SIZE;
209	return 0;
210}
211
212static void exfat_free_namebuf(struct exfat_dentry_namebuf *nb)
213{
214	if (!nb->lfn)
215		return;
216
217	__putname(nb->lfn);
218	exfat_init_namebuf(nb);
219}
220
221/*
222 * Before calling dir_emit*(), sbi->s_lock should be released
223 * because page fault can occur in dir_emit*().
224 */
225#define ITER_POS_FILLED_DOTS    (2)
226static int exfat_iterate(struct file *file, struct dir_context *ctx)
227{
228	struct inode *inode = file_inode(file);
229	struct super_block *sb = inode->i_sb;
230	struct inode *tmp;
231	struct exfat_dir_entry de;
232	struct exfat_dentry_namebuf *nb = &(de.namebuf);
233	struct exfat_inode_info *ei = EXFAT_I(inode);
234	unsigned long inum;
235	loff_t cpos, i_pos;
236	int err = 0, fake_offset = 0;
237
238	exfat_init_namebuf(nb);
239
240	cpos = ctx->pos;
241	if (!dir_emit_dots(file, ctx))
242		goto out;
243
244	if (ctx->pos == ITER_POS_FILLED_DOTS) {
245		cpos = 0;
246		fake_offset = 1;
247	}
248
249	cpos = round_up(cpos, DENTRY_SIZE);
250
251	/* name buffer should be allocated before use */
252	err = exfat_alloc_namebuf(nb);
253	if (err)
254		goto out;
255get_new:
256	mutex_lock(&EXFAT_SB(sb)->s_lock);
257
258	if (ei->flags == ALLOC_NO_FAT_CHAIN && cpos >= i_size_read(inode))
259		goto end_of_dir;
260
261	err = exfat_readdir(inode, &cpos, &de);
262	if (err) {
263		/*
264		 * At least we tried to read a sector.
265		 * Move cpos to next sector position (should be aligned).
266		 */
267		if (err == -EIO) {
268			cpos += 1 << (sb->s_blocksize_bits);
269			cpos &= ~(sb->s_blocksize - 1);
270		}
271
272		err = -EIO;
273		goto end_of_dir;
274	}
275
276	if (!nb->lfn[0])
277		goto end_of_dir;
278
279	i_pos = ((loff_t)ei->start_clu << 32) |	(de.entry & 0xffffffff);
280	tmp = exfat_iget(sb, i_pos);
281	if (tmp) {
282		inum = tmp->i_ino;
283		iput(tmp);
284	} else {
285		inum = iunique(sb, EXFAT_ROOT_INO);
286	}
287
288	mutex_unlock(&EXFAT_SB(sb)->s_lock);
289	if (!dir_emit(ctx, nb->lfn, strlen(nb->lfn), inum,
290			(de.attr & ATTR_SUBDIR) ? DT_DIR : DT_REG))
291		goto out;
292	ctx->pos = cpos;
293	goto get_new;
294
295end_of_dir:
296	if (!cpos && fake_offset)
297		cpos = ITER_POS_FILLED_DOTS;
298	ctx->pos = cpos;
299	mutex_unlock(&EXFAT_SB(sb)->s_lock);
300out:
301	/*
302	 * To improve performance, free namebuf after unlock sb_lock.
303	 * If namebuf is not allocated, this function do nothing
304	 */
305	exfat_free_namebuf(nb);
306	return err;
307}
308
309WRAP_DIR_ITER(exfat_iterate) // FIXME!
310const struct file_operations exfat_dir_operations = {
311	.llseek		= generic_file_llseek,
312	.read		= generic_read_dir,
313	.iterate_shared	= shared_exfat_iterate,
314	.unlocked_ioctl = exfat_ioctl,
315#ifdef CONFIG_COMPAT
316	.compat_ioctl = exfat_compat_ioctl,
317#endif
318	.fsync		= exfat_file_fsync,
319};
320
321int exfat_alloc_new_dir(struct inode *inode, struct exfat_chain *clu)
322{
323	int ret;
324
325	exfat_chain_set(clu, EXFAT_EOF_CLUSTER, 0, ALLOC_NO_FAT_CHAIN);
326
327	ret = exfat_alloc_cluster(inode, 1, clu, IS_DIRSYNC(inode));
328	if (ret)
329		return ret;
330
331	return exfat_zeroed_cluster(inode, clu->dir);
332}
333
334int exfat_calc_num_entries(struct exfat_uni_name *p_uniname)
335{
336	int len;
337
338	len = p_uniname->name_len;
339	if (len == 0)
340		return -EINVAL;
341
342	/* 1 file entry + 1 stream entry + name entries */
343	return ES_ENTRY_NUM(len);
344}
345
346unsigned int exfat_get_entry_type(struct exfat_dentry *ep)
347{
348	if (ep->type == EXFAT_UNUSED)
349		return TYPE_UNUSED;
350	if (IS_EXFAT_DELETED(ep->type))
351		return TYPE_DELETED;
352	if (ep->type == EXFAT_INVAL)
353		return TYPE_INVALID;
354	if (IS_EXFAT_CRITICAL_PRI(ep->type)) {
355		if (ep->type == EXFAT_BITMAP)
356			return TYPE_BITMAP;
357		if (ep->type == EXFAT_UPCASE)
358			return TYPE_UPCASE;
359		if (ep->type == EXFAT_VOLUME)
360			return TYPE_VOLUME;
361		if (ep->type == EXFAT_FILE) {
362			if (le16_to_cpu(ep->dentry.file.attr) & ATTR_SUBDIR)
363				return TYPE_DIR;
364			return TYPE_FILE;
365		}
366		return TYPE_CRITICAL_PRI;
367	}
368	if (IS_EXFAT_BENIGN_PRI(ep->type)) {
369		if (ep->type == EXFAT_GUID)
370			return TYPE_GUID;
371		if (ep->type == EXFAT_PADDING)
372			return TYPE_PADDING;
373		if (ep->type == EXFAT_ACLTAB)
374			return TYPE_ACLTAB;
375		return TYPE_BENIGN_PRI;
376	}
377	if (IS_EXFAT_CRITICAL_SEC(ep->type)) {
378		if (ep->type == EXFAT_STREAM)
379			return TYPE_STREAM;
380		if (ep->type == EXFAT_NAME)
381			return TYPE_EXTEND;
382		if (ep->type == EXFAT_ACL)
383			return TYPE_ACL;
384		return TYPE_CRITICAL_SEC;
385	}
386
387	if (ep->type == EXFAT_VENDOR_EXT)
388		return TYPE_VENDOR_EXT;
389	if (ep->type == EXFAT_VENDOR_ALLOC)
390		return TYPE_VENDOR_ALLOC;
391
392	return TYPE_BENIGN_SEC;
393}
394
395static void exfat_set_entry_type(struct exfat_dentry *ep, unsigned int type)
396{
397	if (type == TYPE_UNUSED) {
398		ep->type = EXFAT_UNUSED;
399	} else if (type == TYPE_DELETED) {
400		ep->type &= EXFAT_DELETE;
401	} else if (type == TYPE_STREAM) {
402		ep->type = EXFAT_STREAM;
403	} else if (type == TYPE_EXTEND) {
404		ep->type = EXFAT_NAME;
405	} else if (type == TYPE_BITMAP) {
406		ep->type = EXFAT_BITMAP;
407	} else if (type == TYPE_UPCASE) {
408		ep->type = EXFAT_UPCASE;
409	} else if (type == TYPE_VOLUME) {
410		ep->type = EXFAT_VOLUME;
411	} else if (type == TYPE_DIR) {
412		ep->type = EXFAT_FILE;
413		ep->dentry.file.attr = cpu_to_le16(ATTR_SUBDIR);
414	} else if (type == TYPE_FILE) {
415		ep->type = EXFAT_FILE;
416		ep->dentry.file.attr = cpu_to_le16(ATTR_ARCHIVE);
417	}
418}
419
420static void exfat_init_stream_entry(struct exfat_dentry *ep,
421		unsigned char flags, unsigned int start_clu,
422		unsigned long long size)
423{
424	exfat_set_entry_type(ep, TYPE_STREAM);
425	ep->dentry.stream.flags = flags;
426	ep->dentry.stream.start_clu = cpu_to_le32(start_clu);
427	ep->dentry.stream.valid_size = cpu_to_le64(size);
428	ep->dentry.stream.size = cpu_to_le64(size);
429}
430
431static void exfat_init_name_entry(struct exfat_dentry *ep,
432		unsigned short *uniname)
433{
434	int i;
435
436	exfat_set_entry_type(ep, TYPE_EXTEND);
437	ep->dentry.name.flags = 0x0;
438
439	for (i = 0; i < EXFAT_FILE_NAME_LEN; i++) {
440		if (*uniname != 0x0) {
441			ep->dentry.name.unicode_0_14[i] = cpu_to_le16(*uniname);
442			uniname++;
443		} else {
444			ep->dentry.name.unicode_0_14[i] = 0x0;
445		}
446	}
447}
448
449int exfat_init_dir_entry(struct inode *inode, struct exfat_chain *p_dir,
450		int entry, unsigned int type, unsigned int start_clu,
451		unsigned long long size)
452{
453	struct super_block *sb = inode->i_sb;
454	struct exfat_sb_info *sbi = EXFAT_SB(sb);
455	struct timespec64 ts = current_time(inode);
456	struct exfat_dentry *ep;
457	struct buffer_head *bh;
458
459	/*
460	 * We cannot use exfat_get_dentry_set here because file ep is not
461	 * initialized yet.
462	 */
463	ep = exfat_get_dentry(sb, p_dir, entry, &bh);
464	if (!ep)
465		return -EIO;
466
467	exfat_set_entry_type(ep, type);
468	exfat_set_entry_time(sbi, &ts,
469			&ep->dentry.file.create_tz,
470			&ep->dentry.file.create_time,
471			&ep->dentry.file.create_date,
472			&ep->dentry.file.create_time_cs);
473	exfat_set_entry_time(sbi, &ts,
474			&ep->dentry.file.modify_tz,
475			&ep->dentry.file.modify_time,
476			&ep->dentry.file.modify_date,
477			&ep->dentry.file.modify_time_cs);
478	exfat_set_entry_time(sbi, &ts,
479			&ep->dentry.file.access_tz,
480			&ep->dentry.file.access_time,
481			&ep->dentry.file.access_date,
482			NULL);
483
484	exfat_update_bh(bh, IS_DIRSYNC(inode));
485	brelse(bh);
486
487	ep = exfat_get_dentry(sb, p_dir, entry + 1, &bh);
488	if (!ep)
489		return -EIO;
490
491	exfat_init_stream_entry(ep,
492		(type == TYPE_FILE) ? ALLOC_FAT_CHAIN : ALLOC_NO_FAT_CHAIN,
493		start_clu, size);
494	exfat_update_bh(bh, IS_DIRSYNC(inode));
495	brelse(bh);
496
497	return 0;
498}
499
500int exfat_update_dir_chksum(struct inode *inode, struct exfat_chain *p_dir,
501		int entry)
502{
503	struct super_block *sb = inode->i_sb;
504	int ret = 0;
505	int i, num_entries;
506	u16 chksum;
507	struct exfat_dentry *ep, *fep;
508	struct buffer_head *fbh, *bh;
509
510	fep = exfat_get_dentry(sb, p_dir, entry, &fbh);
511	if (!fep)
512		return -EIO;
513
514	num_entries = fep->dentry.file.num_ext + 1;
515	chksum = exfat_calc_chksum16(fep, DENTRY_SIZE, 0, CS_DIR_ENTRY);
516
517	for (i = 1; i < num_entries; i++) {
518		ep = exfat_get_dentry(sb, p_dir, entry + i, &bh);
519		if (!ep) {
520			ret = -EIO;
521			goto release_fbh;
522		}
523		chksum = exfat_calc_chksum16(ep, DENTRY_SIZE, chksum,
524				CS_DEFAULT);
525		brelse(bh);
526	}
527
528	fep->dentry.file.checksum = cpu_to_le16(chksum);
529	exfat_update_bh(fbh, IS_DIRSYNC(inode));
530release_fbh:
531	brelse(fbh);
532	return ret;
533}
534
535static void exfat_free_benign_secondary_clusters(struct inode *inode,
536		struct exfat_dentry *ep)
537{
538	struct super_block *sb = inode->i_sb;
539	struct exfat_chain dir;
540	unsigned int start_clu =
541		le32_to_cpu(ep->dentry.generic_secondary.start_clu);
542	u64 size = le64_to_cpu(ep->dentry.generic_secondary.size);
543	unsigned char flags = ep->dentry.generic_secondary.flags;
544
545	if (!(flags & ALLOC_POSSIBLE) || !start_clu || !size)
546		return;
547
548	exfat_chain_set(&dir, start_clu,
549			EXFAT_B_TO_CLU_ROUND_UP(size, EXFAT_SB(sb)),
550			flags);
551	exfat_free_cluster(inode, &dir);
552}
553
554int exfat_init_ext_entry(struct inode *inode, struct exfat_chain *p_dir,
555		int entry, int num_entries, struct exfat_uni_name *p_uniname)
556{
557	struct super_block *sb = inode->i_sb;
558	int i;
559	unsigned short *uniname = p_uniname->name;
560	struct exfat_dentry *ep;
561	struct buffer_head *bh;
562	int sync = IS_DIRSYNC(inode);
563
564	ep = exfat_get_dentry(sb, p_dir, entry, &bh);
565	if (!ep)
566		return -EIO;
567
568	ep->dentry.file.num_ext = (unsigned char)(num_entries - 1);
569	exfat_update_bh(bh, sync);
570	brelse(bh);
571
572	ep = exfat_get_dentry(sb, p_dir, entry + 1, &bh);
573	if (!ep)
574		return -EIO;
575
576	ep->dentry.stream.name_len = p_uniname->name_len;
577	ep->dentry.stream.name_hash = cpu_to_le16(p_uniname->name_hash);
578	exfat_update_bh(bh, sync);
579	brelse(bh);
580
581	for (i = EXFAT_FIRST_CLUSTER; i < num_entries; i++) {
582		ep = exfat_get_dentry(sb, p_dir, entry + i, &bh);
583		if (!ep)
584			return -EIO;
585
586		if (exfat_get_entry_type(ep) & TYPE_BENIGN_SEC)
587			exfat_free_benign_secondary_clusters(inode, ep);
588
589		exfat_init_name_entry(ep, uniname);
590		exfat_update_bh(bh, sync);
591		brelse(bh);
592		uniname += EXFAT_FILE_NAME_LEN;
593	}
594
595	exfat_update_dir_chksum(inode, p_dir, entry);
596	return 0;
597}
598
599int exfat_remove_entries(struct inode *inode, struct exfat_chain *p_dir,
600		int entry, int order, int num_entries)
601{
602	struct super_block *sb = inode->i_sb;
603	int i;
604	struct exfat_dentry *ep;
605	struct buffer_head *bh;
606
607	for (i = order; i < num_entries; i++) {
608		ep = exfat_get_dentry(sb, p_dir, entry + i, &bh);
609		if (!ep)
610			return -EIO;
611
612		if (exfat_get_entry_type(ep) & TYPE_BENIGN_SEC)
613			exfat_free_benign_secondary_clusters(inode, ep);
614
615		exfat_set_entry_type(ep, TYPE_DELETED);
616		exfat_update_bh(bh, IS_DIRSYNC(inode));
617		brelse(bh);
618	}
619
620	return 0;
621}
622
623void exfat_update_dir_chksum_with_entry_set(struct exfat_entry_set_cache *es)
624{
625	int chksum_type = CS_DIR_ENTRY, i;
626	unsigned short chksum = 0;
627	struct exfat_dentry *ep;
628
629	for (i = ES_IDX_FILE; i < es->num_entries; i++) {
630		ep = exfat_get_dentry_cached(es, i);
631		chksum = exfat_calc_chksum16(ep, DENTRY_SIZE, chksum,
632					     chksum_type);
633		chksum_type = CS_DEFAULT;
634	}
635	ep = exfat_get_dentry_cached(es, ES_IDX_FILE);
636	ep->dentry.file.checksum = cpu_to_le16(chksum);
637	es->modified = true;
638}
639
640int exfat_put_dentry_set(struct exfat_entry_set_cache *es, int sync)
641{
642	int i, err = 0;
643
644	if (es->modified)
645		err = exfat_update_bhs(es->bh, es->num_bh, sync);
646
647	for (i = 0; i < es->num_bh; i++)
648		if (err)
649			bforget(es->bh[i]);
650		else
651			brelse(es->bh[i]);
652
653	if (IS_DYNAMIC_ES(es))
654		kfree(es->bh);
655
656	return err;
657}
658
659static int exfat_walk_fat_chain(struct super_block *sb,
660		struct exfat_chain *p_dir, unsigned int byte_offset,
661		unsigned int *clu)
662{
663	struct exfat_sb_info *sbi = EXFAT_SB(sb);
664	unsigned int clu_offset;
665	unsigned int cur_clu;
666
667	clu_offset = EXFAT_B_TO_CLU(byte_offset, sbi);
668	cur_clu = p_dir->dir;
669
670	if (p_dir->flags == ALLOC_NO_FAT_CHAIN) {
671		cur_clu += clu_offset;
672	} else {
673		while (clu_offset > 0) {
674			if (exfat_get_next_cluster(sb, &cur_clu))
675				return -EIO;
676			if (cur_clu == EXFAT_EOF_CLUSTER) {
677				exfat_fs_error(sb,
678					"invalid dentry access beyond EOF (clu : %u, eidx : %d)",
679					p_dir->dir,
680					EXFAT_B_TO_DEN(byte_offset));
681				return -EIO;
682			}
683			clu_offset--;
684		}
685	}
686
687	*clu = cur_clu;
688	return 0;
689}
690
691static int exfat_find_location(struct super_block *sb, struct exfat_chain *p_dir,
692			       int entry, sector_t *sector, int *offset)
693{
694	int ret;
695	unsigned int off, clu = 0;
696	struct exfat_sb_info *sbi = EXFAT_SB(sb);
697
698	off = EXFAT_DEN_TO_B(entry);
699
700	ret = exfat_walk_fat_chain(sb, p_dir, off, &clu);
701	if (ret)
702		return ret;
703
704	/* byte offset in cluster */
705	off = EXFAT_CLU_OFFSET(off, sbi);
706
707	/* byte offset in sector    */
708	*offset = EXFAT_BLK_OFFSET(off, sb);
709
710	/* sector offset in cluster */
711	*sector = EXFAT_B_TO_BLK(off, sb);
712	*sector += exfat_cluster_to_sector(sbi, clu);
713	return 0;
714}
715
716#define EXFAT_MAX_RA_SIZE     (128*1024)
717static int exfat_dir_readahead(struct super_block *sb, sector_t sec)
718{
719	struct exfat_sb_info *sbi = EXFAT_SB(sb);
720	struct buffer_head *bh;
721	unsigned int max_ra_count = EXFAT_MAX_RA_SIZE >> sb->s_blocksize_bits;
722	unsigned int page_ra_count = PAGE_SIZE >> sb->s_blocksize_bits;
723	unsigned int adj_ra_count = max(sbi->sect_per_clus, page_ra_count);
724	unsigned int ra_count = min(adj_ra_count, max_ra_count);
725
726	/* Read-ahead is not required */
727	if (sbi->sect_per_clus == 1)
728		return 0;
729
730	if (sec < sbi->data_start_sector) {
731		exfat_err(sb, "requested sector is invalid(sect:%llu, root:%llu)",
732			  (unsigned long long)sec, sbi->data_start_sector);
733		return -EIO;
734	}
735
736	/* Not sector aligned with ra_count, resize ra_count to page size */
737	if ((sec - sbi->data_start_sector) & (ra_count - 1))
738		ra_count = page_ra_count;
739
740	bh = sb_find_get_block(sb, sec);
741	if (!bh || !buffer_uptodate(bh)) {
742		unsigned int i;
743
744		for (i = 0; i < ra_count; i++)
745			sb_breadahead(sb, (sector_t)(sec + i));
746	}
747	brelse(bh);
748	return 0;
749}
750
751struct exfat_dentry *exfat_get_dentry(struct super_block *sb,
752		struct exfat_chain *p_dir, int entry, struct buffer_head **bh)
753{
754	unsigned int dentries_per_page = EXFAT_B_TO_DEN(PAGE_SIZE);
755	int off;
756	sector_t sec;
757
758	if (p_dir->dir == DIR_DELETED) {
759		exfat_err(sb, "abnormal access to deleted dentry");
760		return NULL;
761	}
762
763	if (exfat_find_location(sb, p_dir, entry, &sec, &off))
764		return NULL;
765
766	if (p_dir->dir != EXFAT_FREE_CLUSTER &&
767			!(entry & (dentries_per_page - 1)))
768		exfat_dir_readahead(sb, sec);
769
770	*bh = sb_bread(sb, sec);
771	if (!*bh)
772		return NULL;
773
774	return (struct exfat_dentry *)((*bh)->b_data + off);
775}
776
777enum exfat_validate_dentry_mode {
778	ES_MODE_STARTED,
779	ES_MODE_GET_FILE_ENTRY,
780	ES_MODE_GET_STRM_ENTRY,
781	ES_MODE_GET_NAME_ENTRY,
782	ES_MODE_GET_CRITICAL_SEC_ENTRY,
783	ES_MODE_GET_BENIGN_SEC_ENTRY,
784};
785
786static bool exfat_validate_entry(unsigned int type,
787		enum exfat_validate_dentry_mode *mode)
788{
789	if (type == TYPE_UNUSED || type == TYPE_DELETED)
790		return false;
791
792	switch (*mode) {
793	case ES_MODE_STARTED:
794		if  (type != TYPE_FILE && type != TYPE_DIR)
795			return false;
796		*mode = ES_MODE_GET_FILE_ENTRY;
797		break;
798	case ES_MODE_GET_FILE_ENTRY:
799		if (type != TYPE_STREAM)
800			return false;
801		*mode = ES_MODE_GET_STRM_ENTRY;
802		break;
803	case ES_MODE_GET_STRM_ENTRY:
804		if (type != TYPE_EXTEND)
805			return false;
806		*mode = ES_MODE_GET_NAME_ENTRY;
807		break;
808	case ES_MODE_GET_NAME_ENTRY:
809		if (type & TYPE_BENIGN_SEC)
810			*mode = ES_MODE_GET_BENIGN_SEC_ENTRY;
811		else if (type != TYPE_EXTEND)
812			return false;
813		break;
814	case ES_MODE_GET_BENIGN_SEC_ENTRY:
815		/* Assume unreconized benign secondary entry */
816		if (!(type & TYPE_BENIGN_SEC))
817			return false;
818		break;
819	default:
820		return false;
821	}
822
823	return true;
824}
825
826struct exfat_dentry *exfat_get_dentry_cached(
827	struct exfat_entry_set_cache *es, int num)
828{
829	int off = es->start_off + num * DENTRY_SIZE;
830	struct buffer_head *bh = es->bh[EXFAT_B_TO_BLK(off, es->sb)];
831	char *p = bh->b_data + EXFAT_BLK_OFFSET(off, es->sb);
832
833	return (struct exfat_dentry *)p;
834}
835
836/*
837 * Returns a set of dentries for a file or dir.
838 *
839 * Note It provides a direct pointer to bh->data via exfat_get_dentry_cached().
840 * User should call exfat_get_dentry_set() after setting 'modified' to apply
841 * changes made in this entry set to the real device.
842 *
843 * in:
844 *   sb+p_dir+entry: indicates a file/dir
845 *   type:  specifies how many dentries should be included.
846 * return:
847 *   pointer of entry set on success,
848 *   NULL on failure.
849 */
850int exfat_get_dentry_set(struct exfat_entry_set_cache *es,
851		struct super_block *sb, struct exfat_chain *p_dir, int entry,
852		unsigned int type)
853{
854	int ret, i, num_bh;
855	unsigned int off;
856	sector_t sec;
857	struct exfat_sb_info *sbi = EXFAT_SB(sb);
858	struct exfat_dentry *ep;
859	int num_entries;
860	enum exfat_validate_dentry_mode mode = ES_MODE_STARTED;
861	struct buffer_head *bh;
862
863	if (p_dir->dir == DIR_DELETED) {
864		exfat_err(sb, "access to deleted dentry");
865		return -EIO;
866	}
867
868	ret = exfat_find_location(sb, p_dir, entry, &sec, &off);
869	if (ret)
870		return ret;
871
872	memset(es, 0, sizeof(*es));
873	es->sb = sb;
874	es->modified = false;
875	es->start_off = off;
876	es->bh = es->__bh;
877
878	bh = sb_bread(sb, sec);
879	if (!bh)
880		return -EIO;
881	es->bh[es->num_bh++] = bh;
882
883	ep = exfat_get_dentry_cached(es, ES_IDX_FILE);
884	if (!exfat_validate_entry(exfat_get_entry_type(ep), &mode))
885		goto put_es;
886
887	num_entries = type == ES_ALL_ENTRIES ?
888		ep->dentry.file.num_ext + 1 : type;
889	es->num_entries = num_entries;
890
891	num_bh = EXFAT_B_TO_BLK_ROUND_UP(off + num_entries * DENTRY_SIZE, sb);
892	if (num_bh > ARRAY_SIZE(es->__bh)) {
893		es->bh = kmalloc_array(num_bh, sizeof(*es->bh), GFP_KERNEL);
894		if (!es->bh) {
895			brelse(bh);
896			return -ENOMEM;
897		}
898		es->bh[0] = bh;
899	}
900
901	for (i = 1; i < num_bh; i++) {
902		/* get the next sector */
903		if (exfat_is_last_sector_in_cluster(sbi, sec)) {
904			unsigned int clu = exfat_sector_to_cluster(sbi, sec);
905
906			if (p_dir->flags == ALLOC_NO_FAT_CHAIN)
907				clu++;
908			else if (exfat_get_next_cluster(sb, &clu))
909				goto put_es;
910			sec = exfat_cluster_to_sector(sbi, clu);
911		} else {
912			sec++;
913		}
914
915		bh = sb_bread(sb, sec);
916		if (!bh)
917			goto put_es;
918		es->bh[es->num_bh++] = bh;
919	}
920
921	/* validate cached dentries */
922	for (i = ES_IDX_STREAM; i < num_entries; i++) {
923		ep = exfat_get_dentry_cached(es, i);
924		if (!exfat_validate_entry(exfat_get_entry_type(ep), &mode))
925			goto put_es;
926	}
927	return 0;
928
929put_es:
930	exfat_put_dentry_set(es, false);
931	return -EIO;
932}
933
934static inline void exfat_reset_empty_hint(struct exfat_hint_femp *hint_femp)
935{
936	hint_femp->eidx = EXFAT_HINT_NONE;
937	hint_femp->count = 0;
938}
939
940static inline void exfat_set_empty_hint(struct exfat_inode_info *ei,
941		struct exfat_hint_femp *candi_empty, struct exfat_chain *clu,
942		int dentry, int num_entries, int entry_type)
943{
944	if (ei->hint_femp.eidx == EXFAT_HINT_NONE ||
945	    ei->hint_femp.eidx > dentry) {
946		int total_entries = EXFAT_B_TO_DEN(i_size_read(&ei->vfs_inode));
947
948		if (candi_empty->count == 0) {
949			candi_empty->cur = *clu;
950			candi_empty->eidx = dentry;
951		}
952
953		if (entry_type == TYPE_UNUSED)
954			candi_empty->count += total_entries - dentry;
955		else
956			candi_empty->count++;
957
958		if (candi_empty->count == num_entries ||
959		    candi_empty->count + candi_empty->eidx == total_entries)
960			ei->hint_femp = *candi_empty;
961	}
962}
963
964enum {
965	DIRENT_STEP_FILE,
966	DIRENT_STEP_STRM,
967	DIRENT_STEP_NAME,
968	DIRENT_STEP_SECD,
969};
970
971/*
972 * @ei:         inode info of parent directory
973 * @p_dir:      directory structure of parent directory
974 * @num_entries:entry size of p_uniname
975 * @hint_opt:   If p_uniname is found, filled with optimized dir/entry
976 *              for traversing cluster chain.
977 * @return:
978 *   >= 0:      file directory entry position where the name exists
979 *   -ENOENT:   entry with the name does not exist
980 *   -EIO:      I/O error
981 */
982int exfat_find_dir_entry(struct super_block *sb, struct exfat_inode_info *ei,
983		struct exfat_chain *p_dir, struct exfat_uni_name *p_uniname,
984		struct exfat_hint *hint_opt)
985{
986	int i, rewind = 0, dentry = 0, end_eidx = 0, num_ext = 0, len;
987	int order, step, name_len = 0;
988	int dentries_per_clu;
989	unsigned int entry_type;
990	unsigned short *uniname = NULL;
991	struct exfat_chain clu;
992	struct exfat_hint *hint_stat = &ei->hint_stat;
993	struct exfat_hint_femp candi_empty;
994	struct exfat_sb_info *sbi = EXFAT_SB(sb);
995	int num_entries = exfat_calc_num_entries(p_uniname);
996
997	if (num_entries < 0)
998		return num_entries;
999
1000	dentries_per_clu = sbi->dentries_per_clu;
1001
1002	exfat_chain_dup(&clu, p_dir);
1003
1004	if (hint_stat->eidx) {
1005		clu.dir = hint_stat->clu;
1006		dentry = hint_stat->eidx;
1007		end_eidx = dentry;
1008	}
1009
1010	exfat_reset_empty_hint(&ei->hint_femp);
1011
1012rewind:
1013	order = 0;
1014	step = DIRENT_STEP_FILE;
1015	exfat_reset_empty_hint(&candi_empty);
1016
1017	while (clu.dir != EXFAT_EOF_CLUSTER) {
1018		i = dentry & (dentries_per_clu - 1);
1019		for (; i < dentries_per_clu; i++, dentry++) {
1020			struct exfat_dentry *ep;
1021			struct buffer_head *bh;
1022
1023			if (rewind && dentry == end_eidx)
1024				goto not_found;
1025
1026			ep = exfat_get_dentry(sb, &clu, i, &bh);
1027			if (!ep)
1028				return -EIO;
1029
1030			entry_type = exfat_get_entry_type(ep);
1031
1032			if (entry_type == TYPE_UNUSED ||
1033			    entry_type == TYPE_DELETED) {
1034				step = DIRENT_STEP_FILE;
1035
1036				exfat_set_empty_hint(ei, &candi_empty, &clu,
1037						dentry, num_entries,
1038						entry_type);
1039
1040				brelse(bh);
1041				if (entry_type == TYPE_UNUSED)
1042					goto not_found;
1043				continue;
1044			}
1045
1046			exfat_reset_empty_hint(&candi_empty);
1047
1048			if (entry_type == TYPE_FILE || entry_type == TYPE_DIR) {
1049				step = DIRENT_STEP_FILE;
1050				hint_opt->clu = clu.dir;
1051				hint_opt->eidx = i;
1052				num_ext = ep->dentry.file.num_ext;
1053				step = DIRENT_STEP_STRM;
1054				brelse(bh);
1055				continue;
1056			}
1057
1058			if (entry_type == TYPE_STREAM) {
1059				u16 name_hash;
1060
1061				if (step != DIRENT_STEP_STRM) {
1062					step = DIRENT_STEP_FILE;
1063					brelse(bh);
1064					continue;
1065				}
1066				step = DIRENT_STEP_FILE;
1067				name_hash = le16_to_cpu(
1068						ep->dentry.stream.name_hash);
1069				if (p_uniname->name_hash == name_hash &&
1070				    p_uniname->name_len ==
1071						ep->dentry.stream.name_len) {
1072					step = DIRENT_STEP_NAME;
1073					order = 1;
1074					name_len = 0;
1075				}
1076				brelse(bh);
1077				continue;
1078			}
1079
1080			brelse(bh);
1081			if (entry_type == TYPE_EXTEND) {
1082				unsigned short entry_uniname[16], unichar;
1083
1084				if (step != DIRENT_STEP_NAME ||
1085				    name_len >= MAX_NAME_LENGTH) {
1086					step = DIRENT_STEP_FILE;
1087					continue;
1088				}
1089
1090				if (++order == 2)
1091					uniname = p_uniname->name;
1092				else
1093					uniname += EXFAT_FILE_NAME_LEN;
1094
1095				len = exfat_extract_uni_name(ep, entry_uniname);
1096				name_len += len;
1097
1098				unichar = *(uniname+len);
1099				*(uniname+len) = 0x0;
1100
1101				if (exfat_uniname_ncmp(sb, uniname,
1102					entry_uniname, len)) {
1103					step = DIRENT_STEP_FILE;
1104				} else if (p_uniname->name_len == name_len) {
1105					if (order == num_ext)
1106						goto found;
1107					step = DIRENT_STEP_SECD;
1108				}
1109
1110				*(uniname+len) = unichar;
1111				continue;
1112			}
1113
1114			if (entry_type &
1115					(TYPE_CRITICAL_SEC | TYPE_BENIGN_SEC)) {
1116				if (step == DIRENT_STEP_SECD) {
1117					if (++order == num_ext)
1118						goto found;
1119					continue;
1120				}
1121			}
1122			step = DIRENT_STEP_FILE;
1123		}
1124
1125		if (clu.flags == ALLOC_NO_FAT_CHAIN) {
1126			if (--clu.size > 0)
1127				clu.dir++;
1128			else
1129				clu.dir = EXFAT_EOF_CLUSTER;
1130		} else {
1131			if (exfat_get_next_cluster(sb, &clu.dir))
1132				return -EIO;
1133		}
1134	}
1135
1136not_found:
1137	/*
1138	 * We started at not 0 index,so we should try to find target
1139	 * from 0 index to the index we started at.
1140	 */
1141	if (!rewind && end_eidx) {
1142		rewind = 1;
1143		dentry = 0;
1144		clu.dir = p_dir->dir;
1145		goto rewind;
1146	}
1147
1148	/*
1149	 * set the EXFAT_EOF_CLUSTER flag to avoid search
1150	 * from the beginning again when allocated a new cluster
1151	 */
1152	if (ei->hint_femp.eidx == EXFAT_HINT_NONE) {
1153		ei->hint_femp.cur.dir = EXFAT_EOF_CLUSTER;
1154		ei->hint_femp.eidx = p_dir->size * dentries_per_clu;
1155		ei->hint_femp.count = 0;
1156	}
1157
1158	/* initialized hint_stat */
1159	hint_stat->clu = p_dir->dir;
1160	hint_stat->eidx = 0;
1161	return -ENOENT;
1162
1163found:
1164	/* next dentry we'll find is out of this cluster */
1165	if (!((dentry + 1) & (dentries_per_clu - 1))) {
1166		int ret = 0;
1167
1168		if (clu.flags == ALLOC_NO_FAT_CHAIN) {
1169			if (--clu.size > 0)
1170				clu.dir++;
1171			else
1172				clu.dir = EXFAT_EOF_CLUSTER;
1173		} else {
1174			ret = exfat_get_next_cluster(sb, &clu.dir);
1175		}
1176
1177		if (ret || clu.dir == EXFAT_EOF_CLUSTER) {
1178			/* just initialized hint_stat */
1179			hint_stat->clu = p_dir->dir;
1180			hint_stat->eidx = 0;
1181			return (dentry - num_ext);
1182		}
1183	}
1184
1185	hint_stat->clu = clu.dir;
1186	hint_stat->eidx = dentry + 1;
1187	return dentry - num_ext;
1188}
1189
1190int exfat_count_ext_entries(struct super_block *sb, struct exfat_chain *p_dir,
1191		int entry, struct exfat_dentry *ep)
1192{
1193	int i, count = 0;
1194	unsigned int type;
1195	struct exfat_dentry *ext_ep;
1196	struct buffer_head *bh;
1197
1198	for (i = 0, entry++; i < ep->dentry.file.num_ext; i++, entry++) {
1199		ext_ep = exfat_get_dentry(sb, p_dir, entry, &bh);
1200		if (!ext_ep)
1201			return -EIO;
1202
1203		type = exfat_get_entry_type(ext_ep);
1204		brelse(bh);
1205		if (type & TYPE_CRITICAL_SEC || type & TYPE_BENIGN_SEC)
1206			count++;
1207	}
1208	return count;
1209}
1210
1211int exfat_count_dir_entries(struct super_block *sb, struct exfat_chain *p_dir)
1212{
1213	int i, count = 0;
1214	int dentries_per_clu;
1215	unsigned int entry_type;
1216	struct exfat_chain clu;
1217	struct exfat_dentry *ep;
1218	struct exfat_sb_info *sbi = EXFAT_SB(sb);
1219	struct buffer_head *bh;
1220
1221	dentries_per_clu = sbi->dentries_per_clu;
1222
1223	exfat_chain_dup(&clu, p_dir);
1224
1225	while (clu.dir != EXFAT_EOF_CLUSTER) {
1226		for (i = 0; i < dentries_per_clu; i++) {
1227			ep = exfat_get_dentry(sb, &clu, i, &bh);
1228			if (!ep)
1229				return -EIO;
1230			entry_type = exfat_get_entry_type(ep);
1231			brelse(bh);
1232
1233			if (entry_type == TYPE_UNUSED)
1234				return count;
1235			if (entry_type != TYPE_DIR)
1236				continue;
1237			count++;
1238		}
1239
1240		if (clu.flags == ALLOC_NO_FAT_CHAIN) {
1241			if (--clu.size > 0)
1242				clu.dir++;
1243			else
1244				clu.dir = EXFAT_EOF_CLUSTER;
1245		} else {
1246			if (exfat_get_next_cluster(sb, &(clu.dir)))
1247				return -EIO;
1248		}
1249	}
1250
1251	return count;
1252}
1253