xref: /openbmc/linux/fs/overlayfs/readdir.c (revision 2fa5ebe3)
1 // SPDX-License-Identifier: GPL-2.0-only
2 /*
3  *
4  * Copyright (C) 2011 Novell Inc.
5  */
6 
7 #include <linux/fs.h>
8 #include <linux/slab.h>
9 #include <linux/namei.h>
10 #include <linux/file.h>
11 #include <linux/xattr.h>
12 #include <linux/rbtree.h>
13 #include <linux/security.h>
14 #include <linux/cred.h>
15 #include <linux/ratelimit.h>
16 #include "overlayfs.h"
17 
18 struct ovl_cache_entry {
19 	unsigned int len;
20 	unsigned int type;
21 	u64 real_ino;
22 	u64 ino;
23 	struct list_head l_node;
24 	struct rb_node node;
25 	struct ovl_cache_entry *next_maybe_whiteout;
26 	bool is_upper;
27 	bool is_whiteout;
28 	char name[];
29 };
30 
31 struct ovl_dir_cache {
32 	long refcount;
33 	u64 version;
34 	struct list_head entries;
35 	struct rb_root root;
36 };
37 
38 struct ovl_readdir_data {
39 	struct dir_context ctx;
40 	struct dentry *dentry;
41 	bool is_lowest;
42 	struct rb_root *root;
43 	struct list_head *list;
44 	struct list_head middle;
45 	struct ovl_cache_entry *first_maybe_whiteout;
46 	int count;
47 	int err;
48 	bool is_upper;
49 	bool d_type_supported;
50 };
51 
52 struct ovl_dir_file {
53 	bool is_real;
54 	bool is_upper;
55 	struct ovl_dir_cache *cache;
56 	struct list_head *cursor;
57 	struct file *realfile;
58 	struct file *upperfile;
59 };
60 
61 static struct ovl_cache_entry *ovl_cache_entry_from_node(struct rb_node *n)
62 {
63 	return rb_entry(n, struct ovl_cache_entry, node);
64 }
65 
66 static bool ovl_cache_entry_find_link(const char *name, int len,
67 				      struct rb_node ***link,
68 				      struct rb_node **parent)
69 {
70 	bool found = false;
71 	struct rb_node **newp = *link;
72 
73 	while (!found && *newp) {
74 		int cmp;
75 		struct ovl_cache_entry *tmp;
76 
77 		*parent = *newp;
78 		tmp = ovl_cache_entry_from_node(*newp);
79 		cmp = strncmp(name, tmp->name, len);
80 		if (cmp > 0)
81 			newp = &tmp->node.rb_right;
82 		else if (cmp < 0 || len < tmp->len)
83 			newp = &tmp->node.rb_left;
84 		else
85 			found = true;
86 	}
87 	*link = newp;
88 
89 	return found;
90 }
91 
92 static struct ovl_cache_entry *ovl_cache_entry_find(struct rb_root *root,
93 						    const char *name, int len)
94 {
95 	struct rb_node *node = root->rb_node;
96 	int cmp;
97 
98 	while (node) {
99 		struct ovl_cache_entry *p = ovl_cache_entry_from_node(node);
100 
101 		cmp = strncmp(name, p->name, len);
102 		if (cmp > 0)
103 			node = p->node.rb_right;
104 		else if (cmp < 0 || len < p->len)
105 			node = p->node.rb_left;
106 		else
107 			return p;
108 	}
109 
110 	return NULL;
111 }
112 
113 static bool ovl_calc_d_ino(struct ovl_readdir_data *rdd,
114 			   struct ovl_cache_entry *p)
115 {
116 	/* Don't care if not doing ovl_iter() */
117 	if (!rdd->dentry)
118 		return false;
119 
120 	/* Always recalc d_ino when remapping lower inode numbers */
121 	if (ovl_xino_bits(rdd->dentry->d_sb))
122 		return true;
123 
124 	/* Always recalc d_ino for parent */
125 	if (strcmp(p->name, "..") == 0)
126 		return true;
127 
128 	/* If this is lower, then native d_ino will do */
129 	if (!rdd->is_upper)
130 		return false;
131 
132 	/*
133 	 * Recalc d_ino for '.' and for all entries if dir is impure (contains
134 	 * copied up entries)
135 	 */
136 	if ((p->name[0] == '.' && p->len == 1) ||
137 	    ovl_test_flag(OVL_IMPURE, d_inode(rdd->dentry)))
138 		return true;
139 
140 	return false;
141 }
142 
143 static struct ovl_cache_entry *ovl_cache_entry_new(struct ovl_readdir_data *rdd,
144 						   const char *name, int len,
145 						   u64 ino, unsigned int d_type)
146 {
147 	struct ovl_cache_entry *p;
148 	size_t size = offsetof(struct ovl_cache_entry, name[len + 1]);
149 
150 	p = kmalloc(size, GFP_KERNEL);
151 	if (!p)
152 		return NULL;
153 
154 	memcpy(p->name, name, len);
155 	p->name[len] = '\0';
156 	p->len = len;
157 	p->type = d_type;
158 	p->real_ino = ino;
159 	p->ino = ino;
160 	/* Defer setting d_ino for upper entry to ovl_iterate() */
161 	if (ovl_calc_d_ino(rdd, p))
162 		p->ino = 0;
163 	p->is_upper = rdd->is_upper;
164 	p->is_whiteout = false;
165 
166 	if (d_type == DT_CHR) {
167 		p->next_maybe_whiteout = rdd->first_maybe_whiteout;
168 		rdd->first_maybe_whiteout = p;
169 	}
170 	return p;
171 }
172 
173 static bool ovl_cache_entry_add_rb(struct ovl_readdir_data *rdd,
174 				  const char *name, int len, u64 ino,
175 				  unsigned int d_type)
176 {
177 	struct rb_node **newp = &rdd->root->rb_node;
178 	struct rb_node *parent = NULL;
179 	struct ovl_cache_entry *p;
180 
181 	if (ovl_cache_entry_find_link(name, len, &newp, &parent))
182 		return true;
183 
184 	p = ovl_cache_entry_new(rdd, name, len, ino, d_type);
185 	if (p == NULL) {
186 		rdd->err = -ENOMEM;
187 		return false;
188 	}
189 
190 	list_add_tail(&p->l_node, rdd->list);
191 	rb_link_node(&p->node, parent, newp);
192 	rb_insert_color(&p->node, rdd->root);
193 
194 	return true;
195 }
196 
197 static bool ovl_fill_lowest(struct ovl_readdir_data *rdd,
198 			   const char *name, int namelen,
199 			   loff_t offset, u64 ino, unsigned int d_type)
200 {
201 	struct ovl_cache_entry *p;
202 
203 	p = ovl_cache_entry_find(rdd->root, name, namelen);
204 	if (p) {
205 		list_move_tail(&p->l_node, &rdd->middle);
206 	} else {
207 		p = ovl_cache_entry_new(rdd, name, namelen, ino, d_type);
208 		if (p == NULL)
209 			rdd->err = -ENOMEM;
210 		else
211 			list_add_tail(&p->l_node, &rdd->middle);
212 	}
213 
214 	return rdd->err == 0;
215 }
216 
217 void ovl_cache_free(struct list_head *list)
218 {
219 	struct ovl_cache_entry *p;
220 	struct ovl_cache_entry *n;
221 
222 	list_for_each_entry_safe(p, n, list, l_node)
223 		kfree(p);
224 
225 	INIT_LIST_HEAD(list);
226 }
227 
228 void ovl_dir_cache_free(struct inode *inode)
229 {
230 	struct ovl_dir_cache *cache = ovl_dir_cache(inode);
231 
232 	if (cache) {
233 		ovl_cache_free(&cache->entries);
234 		kfree(cache);
235 	}
236 }
237 
238 static void ovl_cache_put(struct ovl_dir_file *od, struct inode *inode)
239 {
240 	struct ovl_dir_cache *cache = od->cache;
241 
242 	WARN_ON(cache->refcount <= 0);
243 	cache->refcount--;
244 	if (!cache->refcount) {
245 		if (ovl_dir_cache(inode) == cache)
246 			ovl_set_dir_cache(inode, NULL);
247 
248 		ovl_cache_free(&cache->entries);
249 		kfree(cache);
250 	}
251 }
252 
253 static bool ovl_fill_merge(struct dir_context *ctx, const char *name,
254 			  int namelen, loff_t offset, u64 ino,
255 			  unsigned int d_type)
256 {
257 	struct ovl_readdir_data *rdd =
258 		container_of(ctx, struct ovl_readdir_data, ctx);
259 
260 	rdd->count++;
261 	if (!rdd->is_lowest)
262 		return ovl_cache_entry_add_rb(rdd, name, namelen, ino, d_type);
263 	else
264 		return ovl_fill_lowest(rdd, name, namelen, offset, ino, d_type);
265 }
266 
267 static int ovl_check_whiteouts(const struct path *path, struct ovl_readdir_data *rdd)
268 {
269 	int err;
270 	struct ovl_cache_entry *p;
271 	struct dentry *dentry, *dir = path->dentry;
272 	const struct cred *old_cred;
273 
274 	old_cred = ovl_override_creds(rdd->dentry->d_sb);
275 
276 	err = down_write_killable(&dir->d_inode->i_rwsem);
277 	if (!err) {
278 		while (rdd->first_maybe_whiteout) {
279 			p = rdd->first_maybe_whiteout;
280 			rdd->first_maybe_whiteout = p->next_maybe_whiteout;
281 			dentry = lookup_one(mnt_idmap(path->mnt), p->name, dir, p->len);
282 			if (!IS_ERR(dentry)) {
283 				p->is_whiteout = ovl_is_whiteout(dentry);
284 				dput(dentry);
285 			}
286 		}
287 		inode_unlock(dir->d_inode);
288 	}
289 	revert_creds(old_cred);
290 
291 	return err;
292 }
293 
294 static inline int ovl_dir_read(const struct path *realpath,
295 			       struct ovl_readdir_data *rdd)
296 {
297 	struct file *realfile;
298 	int err;
299 
300 	realfile = ovl_path_open(realpath, O_RDONLY | O_LARGEFILE);
301 	if (IS_ERR(realfile))
302 		return PTR_ERR(realfile);
303 
304 	rdd->first_maybe_whiteout = NULL;
305 	rdd->ctx.pos = 0;
306 	do {
307 		rdd->count = 0;
308 		rdd->err = 0;
309 		err = iterate_dir(realfile, &rdd->ctx);
310 		if (err >= 0)
311 			err = rdd->err;
312 	} while (!err && rdd->count);
313 
314 	if (!err && rdd->first_maybe_whiteout && rdd->dentry)
315 		err = ovl_check_whiteouts(realpath, rdd);
316 
317 	fput(realfile);
318 
319 	return err;
320 }
321 
322 static void ovl_dir_reset(struct file *file)
323 {
324 	struct ovl_dir_file *od = file->private_data;
325 	struct ovl_dir_cache *cache = od->cache;
326 	struct inode *inode = file_inode(file);
327 	bool is_real;
328 
329 	if (cache && ovl_inode_version_get(inode) != cache->version) {
330 		ovl_cache_put(od, inode);
331 		od->cache = NULL;
332 		od->cursor = NULL;
333 	}
334 	is_real = ovl_dir_is_real(inode);
335 	if (od->is_real != is_real) {
336 		/* is_real can only become false when dir is copied up */
337 		if (WARN_ON(is_real))
338 			return;
339 		od->is_real = false;
340 	}
341 }
342 
343 static int ovl_dir_read_merged(struct dentry *dentry, struct list_head *list,
344 	struct rb_root *root)
345 {
346 	int err;
347 	struct path realpath;
348 	struct ovl_readdir_data rdd = {
349 		.ctx.actor = ovl_fill_merge,
350 		.dentry = dentry,
351 		.list = list,
352 		.root = root,
353 		.is_lowest = false,
354 	};
355 	int idx, next;
356 
357 	for (idx = 0; idx != -1; idx = next) {
358 		next = ovl_path_next(idx, dentry, &realpath);
359 		rdd.is_upper = ovl_dentry_upper(dentry) == realpath.dentry;
360 
361 		if (next != -1) {
362 			err = ovl_dir_read(&realpath, &rdd);
363 			if (err)
364 				break;
365 		} else {
366 			/*
367 			 * Insert lowest layer entries before upper ones, this
368 			 * allows offsets to be reasonably constant
369 			 */
370 			list_add(&rdd.middle, rdd.list);
371 			rdd.is_lowest = true;
372 			err = ovl_dir_read(&realpath, &rdd);
373 			list_del(&rdd.middle);
374 		}
375 	}
376 	return err;
377 }
378 
379 static void ovl_seek_cursor(struct ovl_dir_file *od, loff_t pos)
380 {
381 	struct list_head *p;
382 	loff_t off = 0;
383 
384 	list_for_each(p, &od->cache->entries) {
385 		if (off >= pos)
386 			break;
387 		off++;
388 	}
389 	/* Cursor is safe since the cache is stable */
390 	od->cursor = p;
391 }
392 
393 static struct ovl_dir_cache *ovl_cache_get(struct dentry *dentry)
394 {
395 	int res;
396 	struct ovl_dir_cache *cache;
397 	struct inode *inode = d_inode(dentry);
398 
399 	cache = ovl_dir_cache(inode);
400 	if (cache && ovl_inode_version_get(inode) == cache->version) {
401 		WARN_ON(!cache->refcount);
402 		cache->refcount++;
403 		return cache;
404 	}
405 	ovl_set_dir_cache(d_inode(dentry), NULL);
406 
407 	cache = kzalloc(sizeof(struct ovl_dir_cache), GFP_KERNEL);
408 	if (!cache)
409 		return ERR_PTR(-ENOMEM);
410 
411 	cache->refcount = 1;
412 	INIT_LIST_HEAD(&cache->entries);
413 	cache->root = RB_ROOT;
414 
415 	res = ovl_dir_read_merged(dentry, &cache->entries, &cache->root);
416 	if (res) {
417 		ovl_cache_free(&cache->entries);
418 		kfree(cache);
419 		return ERR_PTR(res);
420 	}
421 
422 	cache->version = ovl_inode_version_get(inode);
423 	ovl_set_dir_cache(inode, cache);
424 
425 	return cache;
426 }
427 
428 /* Map inode number to lower fs unique range */
429 static u64 ovl_remap_lower_ino(u64 ino, int xinobits, int fsid,
430 			       const char *name, int namelen, bool warn)
431 {
432 	unsigned int xinoshift = 64 - xinobits;
433 
434 	if (unlikely(ino >> xinoshift)) {
435 		if (warn) {
436 			pr_warn_ratelimited("d_ino too big (%.*s, ino=%llu, xinobits=%d)\n",
437 					    namelen, name, ino, xinobits);
438 		}
439 		return ino;
440 	}
441 
442 	/*
443 	 * The lowest xinobit is reserved for mapping the non-peresistent inode
444 	 * numbers range, but this range is only exposed via st_ino, not here.
445 	 */
446 	return ino | ((u64)fsid) << (xinoshift + 1);
447 }
448 
449 /*
450  * Set d_ino for upper entries. Non-upper entries should always report
451  * the uppermost real inode ino and should not call this function.
452  *
453  * When not all layer are on same fs, report real ino also for upper.
454  *
455  * When all layers are on the same fs, and upper has a reference to
456  * copy up origin, call vfs_getattr() on the overlay entry to make
457  * sure that d_ino will be consistent with st_ino from stat(2).
458  */
459 static int ovl_cache_update_ino(const struct path *path, struct ovl_cache_entry *p)
460 
461 {
462 	struct dentry *dir = path->dentry;
463 	struct dentry *this = NULL;
464 	enum ovl_path_type type;
465 	u64 ino = p->real_ino;
466 	int xinobits = ovl_xino_bits(dir->d_sb);
467 	int err = 0;
468 
469 	if (!ovl_same_dev(dir->d_sb))
470 		goto out;
471 
472 	if (p->name[0] == '.') {
473 		if (p->len == 1) {
474 			this = dget(dir);
475 			goto get;
476 		}
477 		if (p->len == 2 && p->name[1] == '.') {
478 			/* we shall not be moved */
479 			this = dget(dir->d_parent);
480 			goto get;
481 		}
482 	}
483 	this = lookup_one(mnt_idmap(path->mnt), p->name, dir, p->len);
484 	if (IS_ERR_OR_NULL(this) || !this->d_inode) {
485 		/* Mark a stale entry */
486 		p->is_whiteout = true;
487 		if (IS_ERR(this)) {
488 			err = PTR_ERR(this);
489 			this = NULL;
490 			goto fail;
491 		}
492 		goto out;
493 	}
494 
495 get:
496 	type = ovl_path_type(this);
497 	if (OVL_TYPE_ORIGIN(type)) {
498 		struct kstat stat;
499 		struct path statpath = *path;
500 
501 		statpath.dentry = this;
502 		err = vfs_getattr(&statpath, &stat, STATX_INO, 0);
503 		if (err)
504 			goto fail;
505 
506 		/*
507 		 * Directory inode is always on overlay st_dev.
508 		 * Non-dir with ovl_same_dev() could be on pseudo st_dev in case
509 		 * of xino bits overflow.
510 		 */
511 		WARN_ON_ONCE(S_ISDIR(stat.mode) &&
512 			     dir->d_sb->s_dev != stat.dev);
513 		ino = stat.ino;
514 	} else if (xinobits && !OVL_TYPE_UPPER(type)) {
515 		ino = ovl_remap_lower_ino(ino, xinobits,
516 					  ovl_layer_lower(this)->fsid,
517 					  p->name, p->len,
518 					  ovl_xino_warn(dir->d_sb));
519 	}
520 
521 out:
522 	p->ino = ino;
523 	dput(this);
524 	return err;
525 
526 fail:
527 	pr_warn_ratelimited("failed to look up (%s) for ino (%i)\n",
528 			    p->name, err);
529 	goto out;
530 }
531 
532 static bool ovl_fill_plain(struct dir_context *ctx, const char *name,
533 			  int namelen, loff_t offset, u64 ino,
534 			  unsigned int d_type)
535 {
536 	struct ovl_cache_entry *p;
537 	struct ovl_readdir_data *rdd =
538 		container_of(ctx, struct ovl_readdir_data, ctx);
539 
540 	rdd->count++;
541 	p = ovl_cache_entry_new(rdd, name, namelen, ino, d_type);
542 	if (p == NULL) {
543 		rdd->err = -ENOMEM;
544 		return false;
545 	}
546 	list_add_tail(&p->l_node, rdd->list);
547 
548 	return true;
549 }
550 
551 static int ovl_dir_read_impure(const struct path *path,  struct list_head *list,
552 			       struct rb_root *root)
553 {
554 	int err;
555 	struct path realpath;
556 	struct ovl_cache_entry *p, *n;
557 	struct ovl_readdir_data rdd = {
558 		.ctx.actor = ovl_fill_plain,
559 		.list = list,
560 		.root = root,
561 	};
562 
563 	INIT_LIST_HEAD(list);
564 	*root = RB_ROOT;
565 	ovl_path_upper(path->dentry, &realpath);
566 
567 	err = ovl_dir_read(&realpath, &rdd);
568 	if (err)
569 		return err;
570 
571 	list_for_each_entry_safe(p, n, list, l_node) {
572 		if (strcmp(p->name, ".") != 0 &&
573 		    strcmp(p->name, "..") != 0) {
574 			err = ovl_cache_update_ino(path, p);
575 			if (err)
576 				return err;
577 		}
578 		if (p->ino == p->real_ino) {
579 			list_del(&p->l_node);
580 			kfree(p);
581 		} else {
582 			struct rb_node **newp = &root->rb_node;
583 			struct rb_node *parent = NULL;
584 
585 			if (WARN_ON(ovl_cache_entry_find_link(p->name, p->len,
586 							      &newp, &parent)))
587 				return -EIO;
588 
589 			rb_link_node(&p->node, parent, newp);
590 			rb_insert_color(&p->node, root);
591 		}
592 	}
593 	return 0;
594 }
595 
596 static struct ovl_dir_cache *ovl_cache_get_impure(const struct path *path)
597 {
598 	int res;
599 	struct dentry *dentry = path->dentry;
600 	struct inode *inode = d_inode(dentry);
601 	struct ovl_fs *ofs = OVL_FS(dentry->d_sb);
602 	struct ovl_dir_cache *cache;
603 
604 	cache = ovl_dir_cache(inode);
605 	if (cache && ovl_inode_version_get(inode) == cache->version)
606 		return cache;
607 
608 	/* Impure cache is not refcounted, free it here */
609 	ovl_dir_cache_free(inode);
610 	ovl_set_dir_cache(inode, NULL);
611 
612 	cache = kzalloc(sizeof(struct ovl_dir_cache), GFP_KERNEL);
613 	if (!cache)
614 		return ERR_PTR(-ENOMEM);
615 
616 	res = ovl_dir_read_impure(path, &cache->entries, &cache->root);
617 	if (res) {
618 		ovl_cache_free(&cache->entries);
619 		kfree(cache);
620 		return ERR_PTR(res);
621 	}
622 	if (list_empty(&cache->entries)) {
623 		/*
624 		 * A good opportunity to get rid of an unneeded "impure" flag.
625 		 * Removing the "impure" xattr is best effort.
626 		 */
627 		if (!ovl_want_write(dentry)) {
628 			ovl_removexattr(ofs, ovl_dentry_upper(dentry),
629 					OVL_XATTR_IMPURE);
630 			ovl_drop_write(dentry);
631 		}
632 		ovl_clear_flag(OVL_IMPURE, inode);
633 		kfree(cache);
634 		return NULL;
635 	}
636 
637 	cache->version = ovl_inode_version_get(inode);
638 	ovl_set_dir_cache(inode, cache);
639 
640 	return cache;
641 }
642 
643 struct ovl_readdir_translate {
644 	struct dir_context *orig_ctx;
645 	struct ovl_dir_cache *cache;
646 	struct dir_context ctx;
647 	u64 parent_ino;
648 	int fsid;
649 	int xinobits;
650 	bool xinowarn;
651 };
652 
653 static bool ovl_fill_real(struct dir_context *ctx, const char *name,
654 			   int namelen, loff_t offset, u64 ino,
655 			   unsigned int d_type)
656 {
657 	struct ovl_readdir_translate *rdt =
658 		container_of(ctx, struct ovl_readdir_translate, ctx);
659 	struct dir_context *orig_ctx = rdt->orig_ctx;
660 
661 	if (rdt->parent_ino && strcmp(name, "..") == 0) {
662 		ino = rdt->parent_ino;
663 	} else if (rdt->cache) {
664 		struct ovl_cache_entry *p;
665 
666 		p = ovl_cache_entry_find(&rdt->cache->root, name, namelen);
667 		if (p)
668 			ino = p->ino;
669 	} else if (rdt->xinobits) {
670 		ino = ovl_remap_lower_ino(ino, rdt->xinobits, rdt->fsid,
671 					  name, namelen, rdt->xinowarn);
672 	}
673 
674 	return orig_ctx->actor(orig_ctx, name, namelen, offset, ino, d_type);
675 }
676 
677 static bool ovl_is_impure_dir(struct file *file)
678 {
679 	struct ovl_dir_file *od = file->private_data;
680 	struct inode *dir = file_inode(file);
681 
682 	/*
683 	 * Only upper dir can be impure, but if we are in the middle of
684 	 * iterating a lower real dir, dir could be copied up and marked
685 	 * impure. We only want the impure cache if we started iterating
686 	 * a real upper dir to begin with.
687 	 */
688 	return od->is_upper && ovl_test_flag(OVL_IMPURE, dir);
689 
690 }
691 
692 static int ovl_iterate_real(struct file *file, struct dir_context *ctx)
693 {
694 	int err;
695 	struct ovl_dir_file *od = file->private_data;
696 	struct dentry *dir = file->f_path.dentry;
697 	const struct ovl_layer *lower_layer = ovl_layer_lower(dir);
698 	struct ovl_readdir_translate rdt = {
699 		.ctx.actor = ovl_fill_real,
700 		.orig_ctx = ctx,
701 		.xinobits = ovl_xino_bits(dir->d_sb),
702 		.xinowarn = ovl_xino_warn(dir->d_sb),
703 	};
704 
705 	if (rdt.xinobits && lower_layer)
706 		rdt.fsid = lower_layer->fsid;
707 
708 	if (OVL_TYPE_MERGE(ovl_path_type(dir->d_parent))) {
709 		struct kstat stat;
710 		struct path statpath = file->f_path;
711 
712 		statpath.dentry = dir->d_parent;
713 		err = vfs_getattr(&statpath, &stat, STATX_INO, 0);
714 		if (err)
715 			return err;
716 
717 		WARN_ON_ONCE(dir->d_sb->s_dev != stat.dev);
718 		rdt.parent_ino = stat.ino;
719 	}
720 
721 	if (ovl_is_impure_dir(file)) {
722 		rdt.cache = ovl_cache_get_impure(&file->f_path);
723 		if (IS_ERR(rdt.cache))
724 			return PTR_ERR(rdt.cache);
725 	}
726 
727 	err = iterate_dir(od->realfile, &rdt.ctx);
728 	ctx->pos = rdt.ctx.pos;
729 
730 	return err;
731 }
732 
733 
734 static int ovl_iterate(struct file *file, struct dir_context *ctx)
735 {
736 	struct ovl_dir_file *od = file->private_data;
737 	struct dentry *dentry = file->f_path.dentry;
738 	struct ovl_cache_entry *p;
739 	const struct cred *old_cred;
740 	int err;
741 
742 	old_cred = ovl_override_creds(dentry->d_sb);
743 	if (!ctx->pos)
744 		ovl_dir_reset(file);
745 
746 	if (od->is_real) {
747 		/*
748 		 * If parent is merge, then need to adjust d_ino for '..', if
749 		 * dir is impure then need to adjust d_ino for copied up
750 		 * entries.
751 		 */
752 		if (ovl_xino_bits(dentry->d_sb) ||
753 		    (ovl_same_fs(dentry->d_sb) &&
754 		     (ovl_is_impure_dir(file) ||
755 		      OVL_TYPE_MERGE(ovl_path_type(dentry->d_parent))))) {
756 			err = ovl_iterate_real(file, ctx);
757 		} else {
758 			err = iterate_dir(od->realfile, ctx);
759 		}
760 		goto out;
761 	}
762 
763 	if (!od->cache) {
764 		struct ovl_dir_cache *cache;
765 
766 		cache = ovl_cache_get(dentry);
767 		err = PTR_ERR(cache);
768 		if (IS_ERR(cache))
769 			goto out;
770 
771 		od->cache = cache;
772 		ovl_seek_cursor(od, ctx->pos);
773 	}
774 
775 	while (od->cursor != &od->cache->entries) {
776 		p = list_entry(od->cursor, struct ovl_cache_entry, l_node);
777 		if (!p->is_whiteout) {
778 			if (!p->ino) {
779 				err = ovl_cache_update_ino(&file->f_path, p);
780 				if (err)
781 					goto out;
782 			}
783 		}
784 		/* ovl_cache_update_ino() sets is_whiteout on stale entry */
785 		if (!p->is_whiteout) {
786 			if (!dir_emit(ctx, p->name, p->len, p->ino, p->type))
787 				break;
788 		}
789 		od->cursor = p->l_node.next;
790 		ctx->pos++;
791 	}
792 	err = 0;
793 out:
794 	revert_creds(old_cred);
795 	return err;
796 }
797 
798 static loff_t ovl_dir_llseek(struct file *file, loff_t offset, int origin)
799 {
800 	loff_t res;
801 	struct ovl_dir_file *od = file->private_data;
802 
803 	inode_lock(file_inode(file));
804 	if (!file->f_pos)
805 		ovl_dir_reset(file);
806 
807 	if (od->is_real) {
808 		res = vfs_llseek(od->realfile, offset, origin);
809 		file->f_pos = od->realfile->f_pos;
810 	} else {
811 		res = -EINVAL;
812 
813 		switch (origin) {
814 		case SEEK_CUR:
815 			offset += file->f_pos;
816 			break;
817 		case SEEK_SET:
818 			break;
819 		default:
820 			goto out_unlock;
821 		}
822 		if (offset < 0)
823 			goto out_unlock;
824 
825 		if (offset != file->f_pos) {
826 			file->f_pos = offset;
827 			if (od->cache)
828 				ovl_seek_cursor(od, offset);
829 		}
830 		res = offset;
831 	}
832 out_unlock:
833 	inode_unlock(file_inode(file));
834 
835 	return res;
836 }
837 
838 static struct file *ovl_dir_open_realfile(const struct file *file,
839 					  const struct path *realpath)
840 {
841 	struct file *res;
842 	const struct cred *old_cred;
843 
844 	old_cred = ovl_override_creds(file_inode(file)->i_sb);
845 	res = ovl_path_open(realpath, O_RDONLY | (file->f_flags & O_LARGEFILE));
846 	revert_creds(old_cred);
847 
848 	return res;
849 }
850 
851 /*
852  * Like ovl_real_fdget(), returns upperfile if dir was copied up since open.
853  * Unlike ovl_real_fdget(), this caches upperfile in file->private_data.
854  *
855  * TODO: use same abstract type for file->private_data of dir and file so
856  * upperfile could also be cached for files as well.
857  */
858 struct file *ovl_dir_real_file(const struct file *file, bool want_upper)
859 {
860 
861 	struct ovl_dir_file *od = file->private_data;
862 	struct dentry *dentry = file->f_path.dentry;
863 	struct file *old, *realfile = od->realfile;
864 
865 	if (!OVL_TYPE_UPPER(ovl_path_type(dentry)))
866 		return want_upper ? NULL : realfile;
867 
868 	/*
869 	 * Need to check if we started out being a lower dir, but got copied up
870 	 */
871 	if (!od->is_upper) {
872 		realfile = READ_ONCE(od->upperfile);
873 		if (!realfile) {
874 			struct path upperpath;
875 
876 			ovl_path_upper(dentry, &upperpath);
877 			realfile = ovl_dir_open_realfile(file, &upperpath);
878 			if (IS_ERR(realfile))
879 				return realfile;
880 
881 			old = cmpxchg_release(&od->upperfile, NULL, realfile);
882 			if (old) {
883 				fput(realfile);
884 				realfile = old;
885 			}
886 		}
887 	}
888 
889 	return realfile;
890 }
891 
892 static int ovl_dir_fsync(struct file *file, loff_t start, loff_t end,
893 			 int datasync)
894 {
895 	struct file *realfile;
896 	int err;
897 
898 	err = ovl_sync_status(OVL_FS(file_inode(file)->i_sb));
899 	if (err <= 0)
900 		return err;
901 
902 	realfile = ovl_dir_real_file(file, true);
903 	err = PTR_ERR_OR_ZERO(realfile);
904 
905 	/* Nothing to sync for lower */
906 	if (!realfile || err)
907 		return err;
908 
909 	return vfs_fsync_range(realfile, start, end, datasync);
910 }
911 
912 static int ovl_dir_release(struct inode *inode, struct file *file)
913 {
914 	struct ovl_dir_file *od = file->private_data;
915 
916 	if (od->cache) {
917 		inode_lock(inode);
918 		ovl_cache_put(od, inode);
919 		inode_unlock(inode);
920 	}
921 	fput(od->realfile);
922 	if (od->upperfile)
923 		fput(od->upperfile);
924 	kfree(od);
925 
926 	return 0;
927 }
928 
929 static int ovl_dir_open(struct inode *inode, struct file *file)
930 {
931 	struct path realpath;
932 	struct file *realfile;
933 	struct ovl_dir_file *od;
934 	enum ovl_path_type type;
935 
936 	od = kzalloc(sizeof(struct ovl_dir_file), GFP_KERNEL);
937 	if (!od)
938 		return -ENOMEM;
939 
940 	type = ovl_path_real(file->f_path.dentry, &realpath);
941 	realfile = ovl_dir_open_realfile(file, &realpath);
942 	if (IS_ERR(realfile)) {
943 		kfree(od);
944 		return PTR_ERR(realfile);
945 	}
946 	od->realfile = realfile;
947 	od->is_real = ovl_dir_is_real(inode);
948 	od->is_upper = OVL_TYPE_UPPER(type);
949 	file->private_data = od;
950 
951 	return 0;
952 }
953 
954 const struct file_operations ovl_dir_operations = {
955 	.read		= generic_read_dir,
956 	.open		= ovl_dir_open,
957 	.iterate	= ovl_iterate,
958 	.llseek		= ovl_dir_llseek,
959 	.fsync		= ovl_dir_fsync,
960 	.release	= ovl_dir_release,
961 };
962 
963 int ovl_check_empty_dir(struct dentry *dentry, struct list_head *list)
964 {
965 	int err;
966 	struct ovl_cache_entry *p, *n;
967 	struct rb_root root = RB_ROOT;
968 	const struct cred *old_cred;
969 
970 	old_cred = ovl_override_creds(dentry->d_sb);
971 	err = ovl_dir_read_merged(dentry, list, &root);
972 	revert_creds(old_cred);
973 	if (err)
974 		return err;
975 
976 	err = 0;
977 
978 	list_for_each_entry_safe(p, n, list, l_node) {
979 		/*
980 		 * Select whiteouts in upperdir, they should
981 		 * be cleared when deleting this directory.
982 		 */
983 		if (p->is_whiteout) {
984 			if (p->is_upper)
985 				continue;
986 			goto del_entry;
987 		}
988 
989 		if (p->name[0] == '.') {
990 			if (p->len == 1)
991 				goto del_entry;
992 			if (p->len == 2 && p->name[1] == '.')
993 				goto del_entry;
994 		}
995 		err = -ENOTEMPTY;
996 		break;
997 
998 del_entry:
999 		list_del(&p->l_node);
1000 		kfree(p);
1001 	}
1002 
1003 	return err;
1004 }
1005 
1006 void ovl_cleanup_whiteouts(struct ovl_fs *ofs, struct dentry *upper,
1007 			   struct list_head *list)
1008 {
1009 	struct ovl_cache_entry *p;
1010 
1011 	inode_lock_nested(upper->d_inode, I_MUTEX_CHILD);
1012 	list_for_each_entry(p, list, l_node) {
1013 		struct dentry *dentry;
1014 
1015 		if (WARN_ON(!p->is_whiteout || !p->is_upper))
1016 			continue;
1017 
1018 		dentry = ovl_lookup_upper(ofs, p->name, upper, p->len);
1019 		if (IS_ERR(dentry)) {
1020 			pr_err("lookup '%s/%.*s' failed (%i)\n",
1021 			       upper->d_name.name, p->len, p->name,
1022 			       (int) PTR_ERR(dentry));
1023 			continue;
1024 		}
1025 		if (dentry->d_inode)
1026 			ovl_cleanup(ofs, upper->d_inode, dentry);
1027 		dput(dentry);
1028 	}
1029 	inode_unlock(upper->d_inode);
1030 }
1031 
1032 static bool ovl_check_d_type(struct dir_context *ctx, const char *name,
1033 			  int namelen, loff_t offset, u64 ino,
1034 			  unsigned int d_type)
1035 {
1036 	struct ovl_readdir_data *rdd =
1037 		container_of(ctx, struct ovl_readdir_data, ctx);
1038 
1039 	/* Even if d_type is not supported, DT_DIR is returned for . and .. */
1040 	if (!strncmp(name, ".", namelen) || !strncmp(name, "..", namelen))
1041 		return true;
1042 
1043 	if (d_type != DT_UNKNOWN)
1044 		rdd->d_type_supported = true;
1045 
1046 	return true;
1047 }
1048 
1049 /*
1050  * Returns 1 if d_type is supported, 0 not supported/unknown. Negative values
1051  * if error is encountered.
1052  */
1053 int ovl_check_d_type_supported(const struct path *realpath)
1054 {
1055 	int err;
1056 	struct ovl_readdir_data rdd = {
1057 		.ctx.actor = ovl_check_d_type,
1058 		.d_type_supported = false,
1059 	};
1060 
1061 	err = ovl_dir_read(realpath, &rdd);
1062 	if (err)
1063 		return err;
1064 
1065 	return rdd.d_type_supported;
1066 }
1067 
1068 #define OVL_INCOMPATDIR_NAME "incompat"
1069 
1070 static int ovl_workdir_cleanup_recurse(struct ovl_fs *ofs, const struct path *path,
1071 				       int level)
1072 {
1073 	int err;
1074 	struct inode *dir = path->dentry->d_inode;
1075 	LIST_HEAD(list);
1076 	struct ovl_cache_entry *p;
1077 	struct ovl_readdir_data rdd = {
1078 		.ctx.actor = ovl_fill_plain,
1079 		.list = &list,
1080 	};
1081 	bool incompat = false;
1082 
1083 	/*
1084 	 * The "work/incompat" directory is treated specially - if it is not
1085 	 * empty, instead of printing a generic error and mounting read-only,
1086 	 * we will error about incompat features and fail the mount.
1087 	 *
1088 	 * When called from ovl_indexdir_cleanup(), path->dentry->d_name.name
1089 	 * starts with '#'.
1090 	 */
1091 	if (level == 2 &&
1092 	    !strcmp(path->dentry->d_name.name, OVL_INCOMPATDIR_NAME))
1093 		incompat = true;
1094 
1095 	err = ovl_dir_read(path, &rdd);
1096 	if (err)
1097 		goto out;
1098 
1099 	inode_lock_nested(dir, I_MUTEX_PARENT);
1100 	list_for_each_entry(p, &list, l_node) {
1101 		struct dentry *dentry;
1102 
1103 		if (p->name[0] == '.') {
1104 			if (p->len == 1)
1105 				continue;
1106 			if (p->len == 2 && p->name[1] == '.')
1107 				continue;
1108 		} else if (incompat) {
1109 			pr_err("overlay with incompat feature '%s' cannot be mounted\n",
1110 				p->name);
1111 			err = -EINVAL;
1112 			break;
1113 		}
1114 		dentry = ovl_lookup_upper(ofs, p->name, path->dentry, p->len);
1115 		if (IS_ERR(dentry))
1116 			continue;
1117 		if (dentry->d_inode)
1118 			err = ovl_workdir_cleanup(ofs, dir, path->mnt, dentry, level);
1119 		dput(dentry);
1120 		if (err)
1121 			break;
1122 	}
1123 	inode_unlock(dir);
1124 out:
1125 	ovl_cache_free(&list);
1126 	return err;
1127 }
1128 
1129 int ovl_workdir_cleanup(struct ovl_fs *ofs, struct inode *dir,
1130 			struct vfsmount *mnt, struct dentry *dentry, int level)
1131 {
1132 	int err;
1133 
1134 	if (!d_is_dir(dentry) || level > 1) {
1135 		return ovl_cleanup(ofs, dir, dentry);
1136 	}
1137 
1138 	err = ovl_do_rmdir(ofs, dir, dentry);
1139 	if (err) {
1140 		struct path path = { .mnt = mnt, .dentry = dentry };
1141 
1142 		inode_unlock(dir);
1143 		err = ovl_workdir_cleanup_recurse(ofs, &path, level + 1);
1144 		inode_lock_nested(dir, I_MUTEX_PARENT);
1145 		if (!err)
1146 			err = ovl_cleanup(ofs, dir, dentry);
1147 	}
1148 
1149 	return err;
1150 }
1151 
1152 int ovl_indexdir_cleanup(struct ovl_fs *ofs)
1153 {
1154 	int err;
1155 	struct dentry *indexdir = ofs->indexdir;
1156 	struct dentry *index = NULL;
1157 	struct inode *dir = indexdir->d_inode;
1158 	struct path path = { .mnt = ovl_upper_mnt(ofs), .dentry = indexdir };
1159 	LIST_HEAD(list);
1160 	struct ovl_cache_entry *p;
1161 	struct ovl_readdir_data rdd = {
1162 		.ctx.actor = ovl_fill_plain,
1163 		.list = &list,
1164 	};
1165 
1166 	err = ovl_dir_read(&path, &rdd);
1167 	if (err)
1168 		goto out;
1169 
1170 	inode_lock_nested(dir, I_MUTEX_PARENT);
1171 	list_for_each_entry(p, &list, l_node) {
1172 		if (p->name[0] == '.') {
1173 			if (p->len == 1)
1174 				continue;
1175 			if (p->len == 2 && p->name[1] == '.')
1176 				continue;
1177 		}
1178 		index = ovl_lookup_upper(ofs, p->name, indexdir, p->len);
1179 		if (IS_ERR(index)) {
1180 			err = PTR_ERR(index);
1181 			index = NULL;
1182 			break;
1183 		}
1184 		/* Cleanup leftover from index create/cleanup attempt */
1185 		if (index->d_name.name[0] == '#') {
1186 			err = ovl_workdir_cleanup(ofs, dir, path.mnt, index, 1);
1187 			if (err)
1188 				break;
1189 			goto next;
1190 		}
1191 		err = ovl_verify_index(ofs, index);
1192 		if (!err) {
1193 			goto next;
1194 		} else if (err == -ESTALE) {
1195 			/* Cleanup stale index entries */
1196 			err = ovl_cleanup(ofs, dir, index);
1197 		} else if (err != -ENOENT) {
1198 			/*
1199 			 * Abort mount to avoid corrupting the index if
1200 			 * an incompatible index entry was found or on out
1201 			 * of memory.
1202 			 */
1203 			break;
1204 		} else if (ofs->config.nfs_export) {
1205 			/*
1206 			 * Whiteout orphan index to block future open by
1207 			 * handle after overlay nlink dropped to zero.
1208 			 */
1209 			err = ovl_cleanup_and_whiteout(ofs, dir, index);
1210 		} else {
1211 			/* Cleanup orphan index entries */
1212 			err = ovl_cleanup(ofs, dir, index);
1213 		}
1214 
1215 		if (err)
1216 			break;
1217 
1218 next:
1219 		dput(index);
1220 		index = NULL;
1221 	}
1222 	dput(index);
1223 	inode_unlock(dir);
1224 out:
1225 	ovl_cache_free(&list);
1226 	if (err)
1227 		pr_err("failed index dir cleanup (%i)\n", err);
1228 	return err;
1229 }
1230