xref: /openbmc/linux/fs/overlayfs/readdir.c (revision a16be368)
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 int 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 0;
183 
184 	p = ovl_cache_entry_new(rdd, name, len, ino, d_type);
185 	if (p == NULL) {
186 		rdd->err = -ENOMEM;
187 		return -ENOMEM;
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 0;
195 }
196 
197 static int 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;
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 dentry *dentry)
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(d_inode(dentry)) == cache)
246 			ovl_set_dir_cache(d_inode(dentry), NULL);
247 
248 		ovl_cache_free(&cache->entries);
249 		kfree(cache);
250 	}
251 }
252 
253 static int 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(struct dentry *dir, struct ovl_readdir_data *rdd)
268 {
269 	int err;
270 	struct ovl_cache_entry *p;
271 	struct dentry *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_len(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(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->dentry, rdd);
316 
317 	fput(realfile);
318 
319 	return err;
320 }
321 
322 /*
323  * Can we iterate real dir directly?
324  *
325  * Non-merge dir may contain whiteouts from a time it was a merge upper, before
326  * lower dir was removed under it and possibly before it was rotated from upper
327  * to lower layer.
328  */
329 static bool ovl_dir_is_real(struct dentry *dir)
330 {
331 	return !ovl_test_flag(OVL_WHITEOUTS, d_inode(dir));
332 }
333 
334 static void ovl_dir_reset(struct file *file)
335 {
336 	struct ovl_dir_file *od = file->private_data;
337 	struct ovl_dir_cache *cache = od->cache;
338 	struct dentry *dentry = file->f_path.dentry;
339 	bool is_real;
340 
341 	if (cache && ovl_dentry_version_get(dentry) != cache->version) {
342 		ovl_cache_put(od, dentry);
343 		od->cache = NULL;
344 		od->cursor = NULL;
345 	}
346 	is_real = ovl_dir_is_real(dentry);
347 	if (od->is_real != is_real) {
348 		/* is_real can only become false when dir is copied up */
349 		if (WARN_ON(is_real))
350 			return;
351 		od->is_real = false;
352 	}
353 }
354 
355 static int ovl_dir_read_merged(struct dentry *dentry, struct list_head *list,
356 	struct rb_root *root)
357 {
358 	int err;
359 	struct path realpath;
360 	struct ovl_readdir_data rdd = {
361 		.ctx.actor = ovl_fill_merge,
362 		.dentry = dentry,
363 		.list = list,
364 		.root = root,
365 		.is_lowest = false,
366 	};
367 	int idx, next;
368 
369 	for (idx = 0; idx != -1; idx = next) {
370 		next = ovl_path_next(idx, dentry, &realpath);
371 		rdd.is_upper = ovl_dentry_upper(dentry) == realpath.dentry;
372 
373 		if (next != -1) {
374 			err = ovl_dir_read(&realpath, &rdd);
375 			if (err)
376 				break;
377 		} else {
378 			/*
379 			 * Insert lowest layer entries before upper ones, this
380 			 * allows offsets to be reasonably constant
381 			 */
382 			list_add(&rdd.middle, rdd.list);
383 			rdd.is_lowest = true;
384 			err = ovl_dir_read(&realpath, &rdd);
385 			list_del(&rdd.middle);
386 		}
387 	}
388 	return err;
389 }
390 
391 static void ovl_seek_cursor(struct ovl_dir_file *od, loff_t pos)
392 {
393 	struct list_head *p;
394 	loff_t off = 0;
395 
396 	list_for_each(p, &od->cache->entries) {
397 		if (off >= pos)
398 			break;
399 		off++;
400 	}
401 	/* Cursor is safe since the cache is stable */
402 	od->cursor = p;
403 }
404 
405 static struct ovl_dir_cache *ovl_cache_get(struct dentry *dentry)
406 {
407 	int res;
408 	struct ovl_dir_cache *cache;
409 
410 	cache = ovl_dir_cache(d_inode(dentry));
411 	if (cache && ovl_dentry_version_get(dentry) == cache->version) {
412 		WARN_ON(!cache->refcount);
413 		cache->refcount++;
414 		return cache;
415 	}
416 	ovl_set_dir_cache(d_inode(dentry), NULL);
417 
418 	cache = kzalloc(sizeof(struct ovl_dir_cache), GFP_KERNEL);
419 	if (!cache)
420 		return ERR_PTR(-ENOMEM);
421 
422 	cache->refcount = 1;
423 	INIT_LIST_HEAD(&cache->entries);
424 	cache->root = RB_ROOT;
425 
426 	res = ovl_dir_read_merged(dentry, &cache->entries, &cache->root);
427 	if (res) {
428 		ovl_cache_free(&cache->entries);
429 		kfree(cache);
430 		return ERR_PTR(res);
431 	}
432 
433 	cache->version = ovl_dentry_version_get(dentry);
434 	ovl_set_dir_cache(d_inode(dentry), cache);
435 
436 	return cache;
437 }
438 
439 /* Map inode number to lower fs unique range */
440 static u64 ovl_remap_lower_ino(u64 ino, int xinobits, int fsid,
441 			       const char *name, int namelen, bool warn)
442 {
443 	unsigned int xinoshift = 64 - xinobits;
444 
445 	if (unlikely(ino >> xinoshift)) {
446 		if (warn) {
447 			pr_warn_ratelimited("d_ino too big (%.*s, ino=%llu, xinobits=%d)\n",
448 					    namelen, name, ino, xinobits);
449 		}
450 		return ino;
451 	}
452 
453 	/*
454 	 * The lowest xinobit is reserved for mapping the non-peresistent inode
455 	 * numbers range, but this range is only exposed via st_ino, not here.
456 	 */
457 	return ino | ((u64)fsid) << (xinoshift + 1);
458 }
459 
460 /*
461  * Set d_ino for upper entries. Non-upper entries should always report
462  * the uppermost real inode ino and should not call this function.
463  *
464  * When not all layer are on same fs, report real ino also for upper.
465  *
466  * When all layers are on the same fs, and upper has a reference to
467  * copy up origin, call vfs_getattr() on the overlay entry to make
468  * sure that d_ino will be consistent with st_ino from stat(2).
469  */
470 static int ovl_cache_update_ino(struct path *path, struct ovl_cache_entry *p)
471 
472 {
473 	struct dentry *dir = path->dentry;
474 	struct dentry *this = NULL;
475 	enum ovl_path_type type;
476 	u64 ino = p->real_ino;
477 	int xinobits = ovl_xino_bits(dir->d_sb);
478 	int err = 0;
479 
480 	if (!ovl_same_dev(dir->d_sb))
481 		goto out;
482 
483 	if (p->name[0] == '.') {
484 		if (p->len == 1) {
485 			this = dget(dir);
486 			goto get;
487 		}
488 		if (p->len == 2 && p->name[1] == '.') {
489 			/* we shall not be moved */
490 			this = dget(dir->d_parent);
491 			goto get;
492 		}
493 	}
494 	this = lookup_one_len(p->name, dir, p->len);
495 	if (IS_ERR_OR_NULL(this) || !this->d_inode) {
496 		if (IS_ERR(this)) {
497 			err = PTR_ERR(this);
498 			this = NULL;
499 			goto fail;
500 		}
501 		goto out;
502 	}
503 
504 get:
505 	type = ovl_path_type(this);
506 	if (OVL_TYPE_ORIGIN(type)) {
507 		struct kstat stat;
508 		struct path statpath = *path;
509 
510 		statpath.dentry = this;
511 		err = vfs_getattr(&statpath, &stat, STATX_INO, 0);
512 		if (err)
513 			goto fail;
514 
515 		/*
516 		 * Directory inode is always on overlay st_dev.
517 		 * Non-dir with ovl_same_dev() could be on pseudo st_dev in case
518 		 * of xino bits overflow.
519 		 */
520 		WARN_ON_ONCE(S_ISDIR(stat.mode) &&
521 			     dir->d_sb->s_dev != stat.dev);
522 		ino = stat.ino;
523 	} else if (xinobits && !OVL_TYPE_UPPER(type)) {
524 		ino = ovl_remap_lower_ino(ino, xinobits,
525 					  ovl_layer_lower(this)->fsid,
526 					  p->name, p->len,
527 					  ovl_xino_warn(dir->d_sb));
528 	}
529 
530 out:
531 	p->ino = ino;
532 	dput(this);
533 	return err;
534 
535 fail:
536 	pr_warn_ratelimited("failed to look up (%s) for ino (%i)\n",
537 			    p->name, err);
538 	goto out;
539 }
540 
541 static int ovl_fill_plain(struct dir_context *ctx, const char *name,
542 			  int namelen, loff_t offset, u64 ino,
543 			  unsigned int d_type)
544 {
545 	struct ovl_cache_entry *p;
546 	struct ovl_readdir_data *rdd =
547 		container_of(ctx, struct ovl_readdir_data, ctx);
548 
549 	rdd->count++;
550 	p = ovl_cache_entry_new(rdd, name, namelen, ino, d_type);
551 	if (p == NULL) {
552 		rdd->err = -ENOMEM;
553 		return -ENOMEM;
554 	}
555 	list_add_tail(&p->l_node, rdd->list);
556 
557 	return 0;
558 }
559 
560 static int ovl_dir_read_impure(struct path *path,  struct list_head *list,
561 			       struct rb_root *root)
562 {
563 	int err;
564 	struct path realpath;
565 	struct ovl_cache_entry *p, *n;
566 	struct ovl_readdir_data rdd = {
567 		.ctx.actor = ovl_fill_plain,
568 		.list = list,
569 		.root = root,
570 	};
571 
572 	INIT_LIST_HEAD(list);
573 	*root = RB_ROOT;
574 	ovl_path_upper(path->dentry, &realpath);
575 
576 	err = ovl_dir_read(&realpath, &rdd);
577 	if (err)
578 		return err;
579 
580 	list_for_each_entry_safe(p, n, list, l_node) {
581 		if (strcmp(p->name, ".") != 0 &&
582 		    strcmp(p->name, "..") != 0) {
583 			err = ovl_cache_update_ino(path, p);
584 			if (err)
585 				return err;
586 		}
587 		if (p->ino == p->real_ino) {
588 			list_del(&p->l_node);
589 			kfree(p);
590 		} else {
591 			struct rb_node **newp = &root->rb_node;
592 			struct rb_node *parent = NULL;
593 
594 			if (WARN_ON(ovl_cache_entry_find_link(p->name, p->len,
595 							      &newp, &parent)))
596 				return -EIO;
597 
598 			rb_link_node(&p->node, parent, newp);
599 			rb_insert_color(&p->node, root);
600 		}
601 	}
602 	return 0;
603 }
604 
605 static struct ovl_dir_cache *ovl_cache_get_impure(struct path *path)
606 {
607 	int res;
608 	struct dentry *dentry = path->dentry;
609 	struct ovl_dir_cache *cache;
610 
611 	cache = ovl_dir_cache(d_inode(dentry));
612 	if (cache && ovl_dentry_version_get(dentry) == cache->version)
613 		return cache;
614 
615 	/* Impure cache is not refcounted, free it here */
616 	ovl_dir_cache_free(d_inode(dentry));
617 	ovl_set_dir_cache(d_inode(dentry), NULL);
618 
619 	cache = kzalloc(sizeof(struct ovl_dir_cache), GFP_KERNEL);
620 	if (!cache)
621 		return ERR_PTR(-ENOMEM);
622 
623 	res = ovl_dir_read_impure(path, &cache->entries, &cache->root);
624 	if (res) {
625 		ovl_cache_free(&cache->entries);
626 		kfree(cache);
627 		return ERR_PTR(res);
628 	}
629 	if (list_empty(&cache->entries)) {
630 		/*
631 		 * A good opportunity to get rid of an unneeded "impure" flag.
632 		 * Removing the "impure" xattr is best effort.
633 		 */
634 		if (!ovl_want_write(dentry)) {
635 			ovl_do_removexattr(ovl_dentry_upper(dentry),
636 					   OVL_XATTR_IMPURE);
637 			ovl_drop_write(dentry);
638 		}
639 		ovl_clear_flag(OVL_IMPURE, d_inode(dentry));
640 		kfree(cache);
641 		return NULL;
642 	}
643 
644 	cache->version = ovl_dentry_version_get(dentry);
645 	ovl_set_dir_cache(d_inode(dentry), cache);
646 
647 	return cache;
648 }
649 
650 struct ovl_readdir_translate {
651 	struct dir_context *orig_ctx;
652 	struct ovl_dir_cache *cache;
653 	struct dir_context ctx;
654 	u64 parent_ino;
655 	int fsid;
656 	int xinobits;
657 	bool xinowarn;
658 };
659 
660 static int ovl_fill_real(struct dir_context *ctx, const char *name,
661 			   int namelen, loff_t offset, u64 ino,
662 			   unsigned int d_type)
663 {
664 	struct ovl_readdir_translate *rdt =
665 		container_of(ctx, struct ovl_readdir_translate, ctx);
666 	struct dir_context *orig_ctx = rdt->orig_ctx;
667 
668 	if (rdt->parent_ino && strcmp(name, "..") == 0) {
669 		ino = rdt->parent_ino;
670 	} else if (rdt->cache) {
671 		struct ovl_cache_entry *p;
672 
673 		p = ovl_cache_entry_find(&rdt->cache->root, name, namelen);
674 		if (p)
675 			ino = p->ino;
676 	} else if (rdt->xinobits) {
677 		ino = ovl_remap_lower_ino(ino, rdt->xinobits, rdt->fsid,
678 					  name, namelen, rdt->xinowarn);
679 	}
680 
681 	return orig_ctx->actor(orig_ctx, name, namelen, offset, ino, d_type);
682 }
683 
684 static bool ovl_is_impure_dir(struct file *file)
685 {
686 	struct ovl_dir_file *od = file->private_data;
687 	struct inode *dir = d_inode(file->f_path.dentry);
688 
689 	/*
690 	 * Only upper dir can be impure, but if we are in the middle of
691 	 * iterating a lower real dir, dir could be copied up and marked
692 	 * impure. We only want the impure cache if we started iterating
693 	 * a real upper dir to begin with.
694 	 */
695 	return od->is_upper && ovl_test_flag(OVL_IMPURE, dir);
696 
697 }
698 
699 static int ovl_iterate_real(struct file *file, struct dir_context *ctx)
700 {
701 	int err;
702 	struct ovl_dir_file *od = file->private_data;
703 	struct dentry *dir = file->f_path.dentry;
704 	const struct ovl_layer *lower_layer = ovl_layer_lower(dir);
705 	struct ovl_readdir_translate rdt = {
706 		.ctx.actor = ovl_fill_real,
707 		.orig_ctx = ctx,
708 		.xinobits = ovl_xino_bits(dir->d_sb),
709 		.xinowarn = ovl_xino_warn(dir->d_sb),
710 	};
711 
712 	if (rdt.xinobits && lower_layer)
713 		rdt.fsid = lower_layer->fsid;
714 
715 	if (OVL_TYPE_MERGE(ovl_path_type(dir->d_parent))) {
716 		struct kstat stat;
717 		struct path statpath = file->f_path;
718 
719 		statpath.dentry = dir->d_parent;
720 		err = vfs_getattr(&statpath, &stat, STATX_INO, 0);
721 		if (err)
722 			return err;
723 
724 		WARN_ON_ONCE(dir->d_sb->s_dev != stat.dev);
725 		rdt.parent_ino = stat.ino;
726 	}
727 
728 	if (ovl_is_impure_dir(file)) {
729 		rdt.cache = ovl_cache_get_impure(&file->f_path);
730 		if (IS_ERR(rdt.cache))
731 			return PTR_ERR(rdt.cache);
732 	}
733 
734 	err = iterate_dir(od->realfile, &rdt.ctx);
735 	ctx->pos = rdt.ctx.pos;
736 
737 	return err;
738 }
739 
740 
741 static int ovl_iterate(struct file *file, struct dir_context *ctx)
742 {
743 	struct ovl_dir_file *od = file->private_data;
744 	struct dentry *dentry = file->f_path.dentry;
745 	struct ovl_cache_entry *p;
746 	const struct cred *old_cred;
747 	int err;
748 
749 	old_cred = ovl_override_creds(dentry->d_sb);
750 	if (!ctx->pos)
751 		ovl_dir_reset(file);
752 
753 	if (od->is_real) {
754 		/*
755 		 * If parent is merge, then need to adjust d_ino for '..', if
756 		 * dir is impure then need to adjust d_ino for copied up
757 		 * entries.
758 		 */
759 		if (ovl_xino_bits(dentry->d_sb) ||
760 		    (ovl_same_fs(dentry->d_sb) &&
761 		     (ovl_is_impure_dir(file) ||
762 		      OVL_TYPE_MERGE(ovl_path_type(dentry->d_parent))))) {
763 			err = ovl_iterate_real(file, ctx);
764 		} else {
765 			err = iterate_dir(od->realfile, ctx);
766 		}
767 		goto out;
768 	}
769 
770 	if (!od->cache) {
771 		struct ovl_dir_cache *cache;
772 
773 		cache = ovl_cache_get(dentry);
774 		err = PTR_ERR(cache);
775 		if (IS_ERR(cache))
776 			goto out;
777 
778 		od->cache = cache;
779 		ovl_seek_cursor(od, ctx->pos);
780 	}
781 
782 	while (od->cursor != &od->cache->entries) {
783 		p = list_entry(od->cursor, struct ovl_cache_entry, l_node);
784 		if (!p->is_whiteout) {
785 			if (!p->ino) {
786 				err = ovl_cache_update_ino(&file->f_path, p);
787 				if (err)
788 					goto out;
789 			}
790 			if (!dir_emit(ctx, p->name, p->len, p->ino, p->type))
791 				break;
792 		}
793 		od->cursor = p->l_node.next;
794 		ctx->pos++;
795 	}
796 	err = 0;
797 out:
798 	revert_creds(old_cred);
799 	return err;
800 }
801 
802 static loff_t ovl_dir_llseek(struct file *file, loff_t offset, int origin)
803 {
804 	loff_t res;
805 	struct ovl_dir_file *od = file->private_data;
806 
807 	inode_lock(file_inode(file));
808 	if (!file->f_pos)
809 		ovl_dir_reset(file);
810 
811 	if (od->is_real) {
812 		res = vfs_llseek(od->realfile, offset, origin);
813 		file->f_pos = od->realfile->f_pos;
814 	} else {
815 		res = -EINVAL;
816 
817 		switch (origin) {
818 		case SEEK_CUR:
819 			offset += file->f_pos;
820 			break;
821 		case SEEK_SET:
822 			break;
823 		default:
824 			goto out_unlock;
825 		}
826 		if (offset < 0)
827 			goto out_unlock;
828 
829 		if (offset != file->f_pos) {
830 			file->f_pos = offset;
831 			if (od->cache)
832 				ovl_seek_cursor(od, offset);
833 		}
834 		res = offset;
835 	}
836 out_unlock:
837 	inode_unlock(file_inode(file));
838 
839 	return res;
840 }
841 
842 static struct file *ovl_dir_open_realfile(struct file *file,
843 					  struct path *realpath)
844 {
845 	struct file *res;
846 	const struct cred *old_cred;
847 
848 	old_cred = ovl_override_creds(file_inode(file)->i_sb);
849 	res = ovl_path_open(realpath, O_RDONLY | (file->f_flags & O_LARGEFILE));
850 	revert_creds(old_cred);
851 
852 	return res;
853 }
854 
855 static int ovl_dir_fsync(struct file *file, loff_t start, loff_t end,
856 			 int datasync)
857 {
858 	struct ovl_dir_file *od = file->private_data;
859 	struct dentry *dentry = file->f_path.dentry;
860 	struct file *realfile = od->realfile;
861 
862 	/* Nothing to sync for lower */
863 	if (!OVL_TYPE_UPPER(ovl_path_type(dentry)))
864 		return 0;
865 
866 	/*
867 	 * Need to check if we started out being a lower dir, but got copied up
868 	 */
869 	if (!od->is_upper) {
870 		struct inode *inode = file_inode(file);
871 
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 
879 			inode_lock(inode);
880 			if (!od->upperfile) {
881 				if (IS_ERR(realfile)) {
882 					inode_unlock(inode);
883 					return PTR_ERR(realfile);
884 				}
885 				smp_store_release(&od->upperfile, realfile);
886 			} else {
887 				/* somebody has beaten us to it */
888 				if (!IS_ERR(realfile))
889 					fput(realfile);
890 				realfile = od->upperfile;
891 			}
892 			inode_unlock(inode);
893 		}
894 	}
895 
896 	return vfs_fsync_range(realfile, start, end, datasync);
897 }
898 
899 static int ovl_dir_release(struct inode *inode, struct file *file)
900 {
901 	struct ovl_dir_file *od = file->private_data;
902 
903 	if (od->cache) {
904 		inode_lock(inode);
905 		ovl_cache_put(od, file->f_path.dentry);
906 		inode_unlock(inode);
907 	}
908 	fput(od->realfile);
909 	if (od->upperfile)
910 		fput(od->upperfile);
911 	kfree(od);
912 
913 	return 0;
914 }
915 
916 static int ovl_dir_open(struct inode *inode, struct file *file)
917 {
918 	struct path realpath;
919 	struct file *realfile;
920 	struct ovl_dir_file *od;
921 	enum ovl_path_type type;
922 
923 	od = kzalloc(sizeof(struct ovl_dir_file), GFP_KERNEL);
924 	if (!od)
925 		return -ENOMEM;
926 
927 	type = ovl_path_real(file->f_path.dentry, &realpath);
928 	realfile = ovl_dir_open_realfile(file, &realpath);
929 	if (IS_ERR(realfile)) {
930 		kfree(od);
931 		return PTR_ERR(realfile);
932 	}
933 	od->realfile = realfile;
934 	od->is_real = ovl_dir_is_real(file->f_path.dentry);
935 	od->is_upper = OVL_TYPE_UPPER(type);
936 	file->private_data = od;
937 
938 	return 0;
939 }
940 
941 const struct file_operations ovl_dir_operations = {
942 	.read		= generic_read_dir,
943 	.open		= ovl_dir_open,
944 	.iterate	= ovl_iterate,
945 	.llseek		= ovl_dir_llseek,
946 	.fsync		= ovl_dir_fsync,
947 	.release	= ovl_dir_release,
948 };
949 
950 int ovl_check_empty_dir(struct dentry *dentry, struct list_head *list)
951 {
952 	int err;
953 	struct ovl_cache_entry *p, *n;
954 	struct rb_root root = RB_ROOT;
955 	const struct cred *old_cred;
956 
957 	old_cred = ovl_override_creds(dentry->d_sb);
958 	err = ovl_dir_read_merged(dentry, list, &root);
959 	revert_creds(old_cred);
960 	if (err)
961 		return err;
962 
963 	err = 0;
964 
965 	list_for_each_entry_safe(p, n, list, l_node) {
966 		/*
967 		 * Select whiteouts in upperdir, they should
968 		 * be cleared when deleting this directory.
969 		 */
970 		if (p->is_whiteout) {
971 			if (p->is_upper)
972 				continue;
973 			goto del_entry;
974 		}
975 
976 		if (p->name[0] == '.') {
977 			if (p->len == 1)
978 				goto del_entry;
979 			if (p->len == 2 && p->name[1] == '.')
980 				goto del_entry;
981 		}
982 		err = -ENOTEMPTY;
983 		break;
984 
985 del_entry:
986 		list_del(&p->l_node);
987 		kfree(p);
988 	}
989 
990 	return err;
991 }
992 
993 void ovl_cleanup_whiteouts(struct dentry *upper, struct list_head *list)
994 {
995 	struct ovl_cache_entry *p;
996 
997 	inode_lock_nested(upper->d_inode, I_MUTEX_CHILD);
998 	list_for_each_entry(p, list, l_node) {
999 		struct dentry *dentry;
1000 
1001 		if (WARN_ON(!p->is_whiteout || !p->is_upper))
1002 			continue;
1003 
1004 		dentry = lookup_one_len(p->name, upper, p->len);
1005 		if (IS_ERR(dentry)) {
1006 			pr_err("lookup '%s/%.*s' failed (%i)\n",
1007 			       upper->d_name.name, p->len, p->name,
1008 			       (int) PTR_ERR(dentry));
1009 			continue;
1010 		}
1011 		if (dentry->d_inode)
1012 			ovl_cleanup(upper->d_inode, dentry);
1013 		dput(dentry);
1014 	}
1015 	inode_unlock(upper->d_inode);
1016 }
1017 
1018 static int ovl_check_d_type(struct dir_context *ctx, const char *name,
1019 			  int namelen, loff_t offset, u64 ino,
1020 			  unsigned int d_type)
1021 {
1022 	struct ovl_readdir_data *rdd =
1023 		container_of(ctx, struct ovl_readdir_data, ctx);
1024 
1025 	/* Even if d_type is not supported, DT_DIR is returned for . and .. */
1026 	if (!strncmp(name, ".", namelen) || !strncmp(name, "..", namelen))
1027 		return 0;
1028 
1029 	if (d_type != DT_UNKNOWN)
1030 		rdd->d_type_supported = true;
1031 
1032 	return 0;
1033 }
1034 
1035 /*
1036  * Returns 1 if d_type is supported, 0 not supported/unknown. Negative values
1037  * if error is encountered.
1038  */
1039 int ovl_check_d_type_supported(struct path *realpath)
1040 {
1041 	int err;
1042 	struct ovl_readdir_data rdd = {
1043 		.ctx.actor = ovl_check_d_type,
1044 		.d_type_supported = false,
1045 	};
1046 
1047 	err = ovl_dir_read(realpath, &rdd);
1048 	if (err)
1049 		return err;
1050 
1051 	return rdd.d_type_supported;
1052 }
1053 
1054 static void ovl_workdir_cleanup_recurse(struct path *path, int level)
1055 {
1056 	int err;
1057 	struct inode *dir = path->dentry->d_inode;
1058 	LIST_HEAD(list);
1059 	struct rb_root root = RB_ROOT;
1060 	struct ovl_cache_entry *p;
1061 	struct ovl_readdir_data rdd = {
1062 		.ctx.actor = ovl_fill_merge,
1063 		.dentry = NULL,
1064 		.list = &list,
1065 		.root = &root,
1066 		.is_lowest = false,
1067 	};
1068 
1069 	err = ovl_dir_read(path, &rdd);
1070 	if (err)
1071 		goto out;
1072 
1073 	inode_lock_nested(dir, I_MUTEX_PARENT);
1074 	list_for_each_entry(p, &list, l_node) {
1075 		struct dentry *dentry;
1076 
1077 		if (p->name[0] == '.') {
1078 			if (p->len == 1)
1079 				continue;
1080 			if (p->len == 2 && p->name[1] == '.')
1081 				continue;
1082 		}
1083 		dentry = lookup_one_len(p->name, path->dentry, p->len);
1084 		if (IS_ERR(dentry))
1085 			continue;
1086 		if (dentry->d_inode)
1087 			ovl_workdir_cleanup(dir, path->mnt, dentry, level);
1088 		dput(dentry);
1089 	}
1090 	inode_unlock(dir);
1091 out:
1092 	ovl_cache_free(&list);
1093 }
1094 
1095 int ovl_workdir_cleanup(struct inode *dir, struct vfsmount *mnt,
1096 			 struct dentry *dentry, int level)
1097 {
1098 	int err;
1099 
1100 	if (!d_is_dir(dentry) || level > 1) {
1101 		return ovl_cleanup(dir, dentry);
1102 	}
1103 
1104 	err = ovl_do_rmdir(dir, dentry);
1105 	if (err) {
1106 		struct path path = { .mnt = mnt, .dentry = dentry };
1107 
1108 		inode_unlock(dir);
1109 		ovl_workdir_cleanup_recurse(&path, level + 1);
1110 		inode_lock_nested(dir, I_MUTEX_PARENT);
1111 		err = ovl_cleanup(dir, dentry);
1112 	}
1113 
1114 	return err;
1115 }
1116 
1117 int ovl_indexdir_cleanup(struct ovl_fs *ofs)
1118 {
1119 	int err;
1120 	struct dentry *indexdir = ofs->indexdir;
1121 	struct dentry *index = NULL;
1122 	struct inode *dir = indexdir->d_inode;
1123 	struct path path = { .mnt = ovl_upper_mnt(ofs), .dentry = indexdir };
1124 	LIST_HEAD(list);
1125 	struct rb_root root = RB_ROOT;
1126 	struct ovl_cache_entry *p;
1127 	struct ovl_readdir_data rdd = {
1128 		.ctx.actor = ovl_fill_merge,
1129 		.dentry = NULL,
1130 		.list = &list,
1131 		.root = &root,
1132 		.is_lowest = false,
1133 	};
1134 
1135 	err = ovl_dir_read(&path, &rdd);
1136 	if (err)
1137 		goto out;
1138 
1139 	inode_lock_nested(dir, I_MUTEX_PARENT);
1140 	list_for_each_entry(p, &list, l_node) {
1141 		if (p->name[0] == '.') {
1142 			if (p->len == 1)
1143 				continue;
1144 			if (p->len == 2 && p->name[1] == '.')
1145 				continue;
1146 		}
1147 		index = lookup_one_len(p->name, indexdir, p->len);
1148 		if (IS_ERR(index)) {
1149 			err = PTR_ERR(index);
1150 			index = NULL;
1151 			break;
1152 		}
1153 		/* Cleanup leftover from index create/cleanup attempt */
1154 		if (index->d_name.name[0] == '#') {
1155 			err = ovl_workdir_cleanup(dir, path.mnt, index, 1);
1156 			if (err)
1157 				break;
1158 			goto next;
1159 		}
1160 		err = ovl_verify_index(ofs, index);
1161 		if (!err) {
1162 			goto next;
1163 		} else if (err == -ESTALE) {
1164 			/* Cleanup stale index entries */
1165 			err = ovl_cleanup(dir, index);
1166 		} else if (err != -ENOENT) {
1167 			/*
1168 			 * Abort mount to avoid corrupting the index if
1169 			 * an incompatible index entry was found or on out
1170 			 * of memory.
1171 			 */
1172 			break;
1173 		} else if (ofs->config.nfs_export) {
1174 			/*
1175 			 * Whiteout orphan index to block future open by
1176 			 * handle after overlay nlink dropped to zero.
1177 			 */
1178 			err = ovl_cleanup_and_whiteout(ofs, dir, index);
1179 		} else {
1180 			/* Cleanup orphan index entries */
1181 			err = ovl_cleanup(dir, index);
1182 		}
1183 
1184 		if (err)
1185 			break;
1186 
1187 next:
1188 		dput(index);
1189 		index = NULL;
1190 	}
1191 	dput(index);
1192 	inode_unlock(dir);
1193 out:
1194 	ovl_cache_free(&list);
1195 	if (err)
1196 		pr_err("failed index dir cleanup (%i)\n", err);
1197 	return err;
1198 }
1199