xref: /openbmc/linux/fs/kernfs/dir.c (revision dfeb0750)
1b8441ed2STejun Heo /*
2b8441ed2STejun Heo  * fs/kernfs/dir.c - kernfs directory implementation
3b8441ed2STejun Heo  *
4b8441ed2STejun Heo  * Copyright (c) 2001-3 Patrick Mochel
5b8441ed2STejun Heo  * Copyright (c) 2007 SUSE Linux Products GmbH
6b8441ed2STejun Heo  * Copyright (c) 2007, 2013 Tejun Heo <tj@kernel.org>
7b8441ed2STejun Heo  *
8b8441ed2STejun Heo  * This file is released under the GPLv2.
9b8441ed2STejun Heo  */
10fd7b9f7bSTejun Heo 
11abd54f02STejun Heo #include <linux/sched.h>
12fd7b9f7bSTejun Heo #include <linux/fs.h>
13fd7b9f7bSTejun Heo #include <linux/namei.h>
14fd7b9f7bSTejun Heo #include <linux/idr.h>
15fd7b9f7bSTejun Heo #include <linux/slab.h>
16fd7b9f7bSTejun Heo #include <linux/security.h>
17fd7b9f7bSTejun Heo #include <linux/hash.h>
18fd7b9f7bSTejun Heo 
19fd7b9f7bSTejun Heo #include "kernfs-internal.h"
20fd7b9f7bSTejun Heo 
21a797bfc3STejun Heo DEFINE_MUTEX(kernfs_mutex);
223eef34adSTejun Heo static DEFINE_SPINLOCK(kernfs_rename_lock);	/* kn->parent and ->name */
233eef34adSTejun Heo static char kernfs_pr_cont_buf[PATH_MAX];	/* protected by rename_lock */
24fd7b9f7bSTejun Heo 
25adc5e8b5STejun Heo #define rb_to_kn(X) rb_entry((X), struct kernfs_node, rb)
26fd7b9f7bSTejun Heo 
2781c173cbSTejun Heo static bool kernfs_active(struct kernfs_node *kn)
2881c173cbSTejun Heo {
2981c173cbSTejun Heo 	lockdep_assert_held(&kernfs_mutex);
3081c173cbSTejun Heo 	return atomic_read(&kn->active) >= 0;
3181c173cbSTejun Heo }
3281c173cbSTejun Heo 
33182fd64bSTejun Heo static bool kernfs_lockdep(struct kernfs_node *kn)
34182fd64bSTejun Heo {
35182fd64bSTejun Heo #ifdef CONFIG_DEBUG_LOCK_ALLOC
36182fd64bSTejun Heo 	return kn->flags & KERNFS_LOCKDEP;
37182fd64bSTejun Heo #else
38182fd64bSTejun Heo 	return false;
39182fd64bSTejun Heo #endif
40182fd64bSTejun Heo }
41182fd64bSTejun Heo 
423eef34adSTejun Heo static int kernfs_name_locked(struct kernfs_node *kn, char *buf, size_t buflen)
433eef34adSTejun Heo {
443eef34adSTejun Heo 	return strlcpy(buf, kn->parent ? kn->name : "/", buflen);
453eef34adSTejun Heo }
463eef34adSTejun Heo 
473eef34adSTejun Heo static char * __must_check kernfs_path_locked(struct kernfs_node *kn, char *buf,
483eef34adSTejun Heo 					      size_t buflen)
493eef34adSTejun Heo {
503eef34adSTejun Heo 	char *p = buf + buflen;
513eef34adSTejun Heo 	int len;
523eef34adSTejun Heo 
533eef34adSTejun Heo 	*--p = '\0';
543eef34adSTejun Heo 
553eef34adSTejun Heo 	do {
563eef34adSTejun Heo 		len = strlen(kn->name);
573eef34adSTejun Heo 		if (p - buf < len + 1) {
583eef34adSTejun Heo 			buf[0] = '\0';
593eef34adSTejun Heo 			p = NULL;
603eef34adSTejun Heo 			break;
613eef34adSTejun Heo 		}
623eef34adSTejun Heo 		p -= len;
633eef34adSTejun Heo 		memcpy(p, kn->name, len);
643eef34adSTejun Heo 		*--p = '/';
653eef34adSTejun Heo 		kn = kn->parent;
663eef34adSTejun Heo 	} while (kn && kn->parent);
673eef34adSTejun Heo 
683eef34adSTejun Heo 	return p;
693eef34adSTejun Heo }
703eef34adSTejun Heo 
713eef34adSTejun Heo /**
723eef34adSTejun Heo  * kernfs_name - obtain the name of a given node
733eef34adSTejun Heo  * @kn: kernfs_node of interest
743eef34adSTejun Heo  * @buf: buffer to copy @kn's name into
753eef34adSTejun Heo  * @buflen: size of @buf
763eef34adSTejun Heo  *
773eef34adSTejun Heo  * Copies the name of @kn into @buf of @buflen bytes.  The behavior is
783eef34adSTejun Heo  * similar to strlcpy().  It returns the length of @kn's name and if @buf
793eef34adSTejun Heo  * isn't long enough, it's filled upto @buflen-1 and nul terminated.
803eef34adSTejun Heo  *
813eef34adSTejun Heo  * This function can be called from any context.
823eef34adSTejun Heo  */
833eef34adSTejun Heo int kernfs_name(struct kernfs_node *kn, char *buf, size_t buflen)
843eef34adSTejun Heo {
853eef34adSTejun Heo 	unsigned long flags;
863eef34adSTejun Heo 	int ret;
873eef34adSTejun Heo 
883eef34adSTejun Heo 	spin_lock_irqsave(&kernfs_rename_lock, flags);
893eef34adSTejun Heo 	ret = kernfs_name_locked(kn, buf, buflen);
903eef34adSTejun Heo 	spin_unlock_irqrestore(&kernfs_rename_lock, flags);
913eef34adSTejun Heo 	return ret;
923eef34adSTejun Heo }
933eef34adSTejun Heo 
943eef34adSTejun Heo /**
953eef34adSTejun Heo  * kernfs_path - build full path of a given node
963eef34adSTejun Heo  * @kn: kernfs_node of interest
973eef34adSTejun Heo  * @buf: buffer to copy @kn's name into
983eef34adSTejun Heo  * @buflen: size of @buf
993eef34adSTejun Heo  *
1003eef34adSTejun Heo  * Builds and returns the full path of @kn in @buf of @buflen bytes.  The
1013eef34adSTejun Heo  * path is built from the end of @buf so the returned pointer usually
1023eef34adSTejun Heo  * doesn't match @buf.  If @buf isn't long enough, @buf is nul terminated
1033eef34adSTejun Heo  * and %NULL is returned.
1043eef34adSTejun Heo  */
1053eef34adSTejun Heo char *kernfs_path(struct kernfs_node *kn, char *buf, size_t buflen)
1063eef34adSTejun Heo {
1073eef34adSTejun Heo 	unsigned long flags;
1083eef34adSTejun Heo 	char *p;
1093eef34adSTejun Heo 
1103eef34adSTejun Heo 	spin_lock_irqsave(&kernfs_rename_lock, flags);
1113eef34adSTejun Heo 	p = kernfs_path_locked(kn, buf, buflen);
1123eef34adSTejun Heo 	spin_unlock_irqrestore(&kernfs_rename_lock, flags);
1133eef34adSTejun Heo 	return p;
1143eef34adSTejun Heo }
115e61734c5STejun Heo EXPORT_SYMBOL_GPL(kernfs_path);
1163eef34adSTejun Heo 
1173eef34adSTejun Heo /**
1183eef34adSTejun Heo  * pr_cont_kernfs_name - pr_cont name of a kernfs_node
1193eef34adSTejun Heo  * @kn: kernfs_node of interest
1203eef34adSTejun Heo  *
1213eef34adSTejun Heo  * This function can be called from any context.
1223eef34adSTejun Heo  */
1233eef34adSTejun Heo void pr_cont_kernfs_name(struct kernfs_node *kn)
1243eef34adSTejun Heo {
1253eef34adSTejun Heo 	unsigned long flags;
1263eef34adSTejun Heo 
1273eef34adSTejun Heo 	spin_lock_irqsave(&kernfs_rename_lock, flags);
1283eef34adSTejun Heo 
1293eef34adSTejun Heo 	kernfs_name_locked(kn, kernfs_pr_cont_buf, sizeof(kernfs_pr_cont_buf));
1303eef34adSTejun Heo 	pr_cont("%s", kernfs_pr_cont_buf);
1313eef34adSTejun Heo 
1323eef34adSTejun Heo 	spin_unlock_irqrestore(&kernfs_rename_lock, flags);
1333eef34adSTejun Heo }
1343eef34adSTejun Heo 
1353eef34adSTejun Heo /**
1363eef34adSTejun Heo  * pr_cont_kernfs_path - pr_cont path of a kernfs_node
1373eef34adSTejun Heo  * @kn: kernfs_node of interest
1383eef34adSTejun Heo  *
1393eef34adSTejun Heo  * This function can be called from any context.
1403eef34adSTejun Heo  */
1413eef34adSTejun Heo void pr_cont_kernfs_path(struct kernfs_node *kn)
1423eef34adSTejun Heo {
1433eef34adSTejun Heo 	unsigned long flags;
1443eef34adSTejun Heo 	char *p;
1453eef34adSTejun Heo 
1463eef34adSTejun Heo 	spin_lock_irqsave(&kernfs_rename_lock, flags);
1473eef34adSTejun Heo 
1483eef34adSTejun Heo 	p = kernfs_path_locked(kn, kernfs_pr_cont_buf,
1493eef34adSTejun Heo 			       sizeof(kernfs_pr_cont_buf));
1503eef34adSTejun Heo 	if (p)
1513eef34adSTejun Heo 		pr_cont("%s", p);
1523eef34adSTejun Heo 	else
1533eef34adSTejun Heo 		pr_cont("<name too long>");
1543eef34adSTejun Heo 
1553eef34adSTejun Heo 	spin_unlock_irqrestore(&kernfs_rename_lock, flags);
1563eef34adSTejun Heo }
1573eef34adSTejun Heo 
1583eef34adSTejun Heo /**
1593eef34adSTejun Heo  * kernfs_get_parent - determine the parent node and pin it
1603eef34adSTejun Heo  * @kn: kernfs_node of interest
1613eef34adSTejun Heo  *
1623eef34adSTejun Heo  * Determines @kn's parent, pins and returns it.  This function can be
1633eef34adSTejun Heo  * called from any context.
1643eef34adSTejun Heo  */
1653eef34adSTejun Heo struct kernfs_node *kernfs_get_parent(struct kernfs_node *kn)
1663eef34adSTejun Heo {
1673eef34adSTejun Heo 	struct kernfs_node *parent;
1683eef34adSTejun Heo 	unsigned long flags;
1693eef34adSTejun Heo 
1703eef34adSTejun Heo 	spin_lock_irqsave(&kernfs_rename_lock, flags);
1713eef34adSTejun Heo 	parent = kn->parent;
1723eef34adSTejun Heo 	kernfs_get(parent);
1733eef34adSTejun Heo 	spin_unlock_irqrestore(&kernfs_rename_lock, flags);
1743eef34adSTejun Heo 
1753eef34adSTejun Heo 	return parent;
1763eef34adSTejun Heo }
1773eef34adSTejun Heo 
178fd7b9f7bSTejun Heo /**
179c637b8acSTejun Heo  *	kernfs_name_hash
180fd7b9f7bSTejun Heo  *	@name: Null terminated string to hash
181fd7b9f7bSTejun Heo  *	@ns:   Namespace tag to hash
182fd7b9f7bSTejun Heo  *
183fd7b9f7bSTejun Heo  *	Returns 31 bit hash of ns + name (so it fits in an off_t )
184fd7b9f7bSTejun Heo  */
185c637b8acSTejun Heo static unsigned int kernfs_name_hash(const char *name, const void *ns)
186fd7b9f7bSTejun Heo {
187fd7b9f7bSTejun Heo 	unsigned long hash = init_name_hash();
188fd7b9f7bSTejun Heo 	unsigned int len = strlen(name);
189fd7b9f7bSTejun Heo 	while (len--)
190fd7b9f7bSTejun Heo 		hash = partial_name_hash(*name++, hash);
191fd7b9f7bSTejun Heo 	hash = (end_name_hash(hash) ^ hash_ptr((void *)ns, 31));
192fd7b9f7bSTejun Heo 	hash &= 0x7fffffffU;
193fd7b9f7bSTejun Heo 	/* Reserve hash numbers 0, 1 and INT_MAX for magic directory entries */
19488391d49SRichard Cochran 	if (hash < 2)
195fd7b9f7bSTejun Heo 		hash += 2;
196fd7b9f7bSTejun Heo 	if (hash >= INT_MAX)
197fd7b9f7bSTejun Heo 		hash = INT_MAX - 1;
198fd7b9f7bSTejun Heo 	return hash;
199fd7b9f7bSTejun Heo }
200fd7b9f7bSTejun Heo 
201c637b8acSTejun Heo static int kernfs_name_compare(unsigned int hash, const char *name,
202324a56e1STejun Heo 			       const void *ns, const struct kernfs_node *kn)
203fd7b9f7bSTejun Heo {
20472392ed0SRasmus Villemoes 	if (hash < kn->hash)
20572392ed0SRasmus Villemoes 		return -1;
20672392ed0SRasmus Villemoes 	if (hash > kn->hash)
20772392ed0SRasmus Villemoes 		return 1;
20872392ed0SRasmus Villemoes 	if (ns < kn->ns)
20972392ed0SRasmus Villemoes 		return -1;
21072392ed0SRasmus Villemoes 	if (ns > kn->ns)
21172392ed0SRasmus Villemoes 		return 1;
212adc5e8b5STejun Heo 	return strcmp(name, kn->name);
213fd7b9f7bSTejun Heo }
214fd7b9f7bSTejun Heo 
215c637b8acSTejun Heo static int kernfs_sd_compare(const struct kernfs_node *left,
216324a56e1STejun Heo 			     const struct kernfs_node *right)
217fd7b9f7bSTejun Heo {
218c637b8acSTejun Heo 	return kernfs_name_compare(left->hash, left->name, left->ns, right);
219fd7b9f7bSTejun Heo }
220fd7b9f7bSTejun Heo 
221fd7b9f7bSTejun Heo /**
222c637b8acSTejun Heo  *	kernfs_link_sibling - link kernfs_node into sibling rbtree
223324a56e1STejun Heo  *	@kn: kernfs_node of interest
224fd7b9f7bSTejun Heo  *
225324a56e1STejun Heo  *	Link @kn into its sibling rbtree which starts from
226adc5e8b5STejun Heo  *	@kn->parent->dir.children.
227fd7b9f7bSTejun Heo  *
228fd7b9f7bSTejun Heo  *	Locking:
229a797bfc3STejun Heo  *	mutex_lock(kernfs_mutex)
230fd7b9f7bSTejun Heo  *
231fd7b9f7bSTejun Heo  *	RETURNS:
232fd7b9f7bSTejun Heo  *	0 on susccess -EEXIST on failure.
233fd7b9f7bSTejun Heo  */
234c637b8acSTejun Heo static int kernfs_link_sibling(struct kernfs_node *kn)
235fd7b9f7bSTejun Heo {
236adc5e8b5STejun Heo 	struct rb_node **node = &kn->parent->dir.children.rb_node;
237fd7b9f7bSTejun Heo 	struct rb_node *parent = NULL;
238fd7b9f7bSTejun Heo 
239fd7b9f7bSTejun Heo 	while (*node) {
240324a56e1STejun Heo 		struct kernfs_node *pos;
241fd7b9f7bSTejun Heo 		int result;
242fd7b9f7bSTejun Heo 
243324a56e1STejun Heo 		pos = rb_to_kn(*node);
244fd7b9f7bSTejun Heo 		parent = *node;
245c637b8acSTejun Heo 		result = kernfs_sd_compare(kn, pos);
246fd7b9f7bSTejun Heo 		if (result < 0)
247adc5e8b5STejun Heo 			node = &pos->rb.rb_left;
248fd7b9f7bSTejun Heo 		else if (result > 0)
249adc5e8b5STejun Heo 			node = &pos->rb.rb_right;
250fd7b9f7bSTejun Heo 		else
251fd7b9f7bSTejun Heo 			return -EEXIST;
252fd7b9f7bSTejun Heo 	}
253c1befb88SJianyu Zhan 
254fd7b9f7bSTejun Heo 	/* add new node and rebalance the tree */
255adc5e8b5STejun Heo 	rb_link_node(&kn->rb, parent, node);
256adc5e8b5STejun Heo 	rb_insert_color(&kn->rb, &kn->parent->dir.children);
257c1befb88SJianyu Zhan 
258c1befb88SJianyu Zhan 	/* successfully added, account subdir number */
259c1befb88SJianyu Zhan 	if (kernfs_type(kn) == KERNFS_DIR)
260c1befb88SJianyu Zhan 		kn->parent->dir.subdirs++;
261c1befb88SJianyu Zhan 
262fd7b9f7bSTejun Heo 	return 0;
263fd7b9f7bSTejun Heo }
264fd7b9f7bSTejun Heo 
265fd7b9f7bSTejun Heo /**
266c637b8acSTejun Heo  *	kernfs_unlink_sibling - unlink kernfs_node from sibling rbtree
267324a56e1STejun Heo  *	@kn: kernfs_node of interest
268fd7b9f7bSTejun Heo  *
26935beab06STejun Heo  *	Try to unlink @kn from its sibling rbtree which starts from
27035beab06STejun Heo  *	kn->parent->dir.children.  Returns %true if @kn was actually
27135beab06STejun Heo  *	removed, %false if @kn wasn't on the rbtree.
272fd7b9f7bSTejun Heo  *
273fd7b9f7bSTejun Heo  *	Locking:
274a797bfc3STejun Heo  *	mutex_lock(kernfs_mutex)
275fd7b9f7bSTejun Heo  */
27635beab06STejun Heo static bool kernfs_unlink_sibling(struct kernfs_node *kn)
277fd7b9f7bSTejun Heo {
27835beab06STejun Heo 	if (RB_EMPTY_NODE(&kn->rb))
27935beab06STejun Heo 		return false;
28035beab06STejun Heo 
281df23fc39STejun Heo 	if (kernfs_type(kn) == KERNFS_DIR)
282adc5e8b5STejun Heo 		kn->parent->dir.subdirs--;
283fd7b9f7bSTejun Heo 
284adc5e8b5STejun Heo 	rb_erase(&kn->rb, &kn->parent->dir.children);
28535beab06STejun Heo 	RB_CLEAR_NODE(&kn->rb);
28635beab06STejun Heo 	return true;
287fd7b9f7bSTejun Heo }
288fd7b9f7bSTejun Heo 
289fd7b9f7bSTejun Heo /**
290c637b8acSTejun Heo  *	kernfs_get_active - get an active reference to kernfs_node
291324a56e1STejun Heo  *	@kn: kernfs_node to get an active reference to
292fd7b9f7bSTejun Heo  *
293324a56e1STejun Heo  *	Get an active reference of @kn.  This function is noop if @kn
294fd7b9f7bSTejun Heo  *	is NULL.
295fd7b9f7bSTejun Heo  *
296fd7b9f7bSTejun Heo  *	RETURNS:
297324a56e1STejun Heo  *	Pointer to @kn on success, NULL on failure.
298fd7b9f7bSTejun Heo  */
299c637b8acSTejun Heo struct kernfs_node *kernfs_get_active(struct kernfs_node *kn)
300fd7b9f7bSTejun Heo {
301324a56e1STejun Heo 	if (unlikely(!kn))
302fd7b9f7bSTejun Heo 		return NULL;
303fd7b9f7bSTejun Heo 
304f4b3e631SGreg Kroah-Hartman 	if (!atomic_inc_unless_negative(&kn->active))
305f4b3e631SGreg Kroah-Hartman 		return NULL;
306f4b3e631SGreg Kroah-Hartman 
307182fd64bSTejun Heo 	if (kernfs_lockdep(kn))
308324a56e1STejun Heo 		rwsem_acquire_read(&kn->dep_map, 0, 1, _RET_IP_);
309324a56e1STejun Heo 	return kn;
310fd7b9f7bSTejun Heo }
311fd7b9f7bSTejun Heo 
312fd7b9f7bSTejun Heo /**
313c637b8acSTejun Heo  *	kernfs_put_active - put an active reference to kernfs_node
314324a56e1STejun Heo  *	@kn: kernfs_node to put an active reference to
315fd7b9f7bSTejun Heo  *
316324a56e1STejun Heo  *	Put an active reference to @kn.  This function is noop if @kn
317fd7b9f7bSTejun Heo  *	is NULL.
318fd7b9f7bSTejun Heo  */
319c637b8acSTejun Heo void kernfs_put_active(struct kernfs_node *kn)
320fd7b9f7bSTejun Heo {
321abd54f02STejun Heo 	struct kernfs_root *root = kernfs_root(kn);
322fd7b9f7bSTejun Heo 	int v;
323fd7b9f7bSTejun Heo 
324324a56e1STejun Heo 	if (unlikely(!kn))
325fd7b9f7bSTejun Heo 		return;
326fd7b9f7bSTejun Heo 
327182fd64bSTejun Heo 	if (kernfs_lockdep(kn))
328324a56e1STejun Heo 		rwsem_release(&kn->dep_map, 1, _RET_IP_);
329adc5e8b5STejun Heo 	v = atomic_dec_return(&kn->active);
330df23fc39STejun Heo 	if (likely(v != KN_DEACTIVATED_BIAS))
331fd7b9f7bSTejun Heo 		return;
332fd7b9f7bSTejun Heo 
333abd54f02STejun Heo 	wake_up_all(&root->deactivate_waitq);
334fd7b9f7bSTejun Heo }
335fd7b9f7bSTejun Heo 
336fd7b9f7bSTejun Heo /**
33781c173cbSTejun Heo  * kernfs_drain - drain kernfs_node
33881c173cbSTejun Heo  * @kn: kernfs_node to drain
339fd7b9f7bSTejun Heo  *
34081c173cbSTejun Heo  * Drain existing usages and nuke all existing mmaps of @kn.  Mutiple
34181c173cbSTejun Heo  * removers may invoke this function concurrently on @kn and all will
34281c173cbSTejun Heo  * return after draining is complete.
343fd7b9f7bSTejun Heo  */
34481c173cbSTejun Heo static void kernfs_drain(struct kernfs_node *kn)
34535beab06STejun Heo 	__releases(&kernfs_mutex) __acquires(&kernfs_mutex)
346fd7b9f7bSTejun Heo {
347abd54f02STejun Heo 	struct kernfs_root *root = kernfs_root(kn);
348fd7b9f7bSTejun Heo 
34935beab06STejun Heo 	lockdep_assert_held(&kernfs_mutex);
35081c173cbSTejun Heo 	WARN_ON_ONCE(kernfs_active(kn));
351abd54f02STejun Heo 
35235beab06STejun Heo 	mutex_unlock(&kernfs_mutex);
353abd54f02STejun Heo 
354182fd64bSTejun Heo 	if (kernfs_lockdep(kn)) {
35535beab06STejun Heo 		rwsem_acquire(&kn->dep_map, 0, 0, _RET_IP_);
35635beab06STejun Heo 		if (atomic_read(&kn->active) != KN_DEACTIVATED_BIAS)
35735beab06STejun Heo 			lock_contended(&kn->dep_map, _RET_IP_);
35835beab06STejun Heo 	}
35935beab06STejun Heo 
36035beab06STejun Heo 	/* but everyone should wait for draining */
361abd54f02STejun Heo 	wait_event(root->deactivate_waitq,
362abd54f02STejun Heo 		   atomic_read(&kn->active) == KN_DEACTIVATED_BIAS);
363fd7b9f7bSTejun Heo 
364182fd64bSTejun Heo 	if (kernfs_lockdep(kn)) {
365324a56e1STejun Heo 		lock_acquired(&kn->dep_map, _RET_IP_);
366324a56e1STejun Heo 		rwsem_release(&kn->dep_map, 1, _RET_IP_);
367fd7b9f7bSTejun Heo 	}
36835beab06STejun Heo 
369ccf02aafSTejun Heo 	kernfs_unmap_bin_file(kn);
370ccf02aafSTejun Heo 
37135beab06STejun Heo 	mutex_lock(&kernfs_mutex);
372a6607930STejun Heo }
373fd7b9f7bSTejun Heo 
374fd7b9f7bSTejun Heo /**
375324a56e1STejun Heo  * kernfs_get - get a reference count on a kernfs_node
376324a56e1STejun Heo  * @kn: the target kernfs_node
377fd7b9f7bSTejun Heo  */
378324a56e1STejun Heo void kernfs_get(struct kernfs_node *kn)
379fd7b9f7bSTejun Heo {
380324a56e1STejun Heo 	if (kn) {
381adc5e8b5STejun Heo 		WARN_ON(!atomic_read(&kn->count));
382adc5e8b5STejun Heo 		atomic_inc(&kn->count);
383fd7b9f7bSTejun Heo 	}
384fd7b9f7bSTejun Heo }
385fd7b9f7bSTejun Heo EXPORT_SYMBOL_GPL(kernfs_get);
386fd7b9f7bSTejun Heo 
387fd7b9f7bSTejun Heo /**
388324a56e1STejun Heo  * kernfs_put - put a reference count on a kernfs_node
389324a56e1STejun Heo  * @kn: the target kernfs_node
390fd7b9f7bSTejun Heo  *
391324a56e1STejun Heo  * Put a reference count of @kn and destroy it if it reached zero.
392fd7b9f7bSTejun Heo  */
393324a56e1STejun Heo void kernfs_put(struct kernfs_node *kn)
394fd7b9f7bSTejun Heo {
395324a56e1STejun Heo 	struct kernfs_node *parent;
396ba7443bcSTejun Heo 	struct kernfs_root *root;
397fd7b9f7bSTejun Heo 
398adc5e8b5STejun Heo 	if (!kn || !atomic_dec_and_test(&kn->count))
399fd7b9f7bSTejun Heo 		return;
400324a56e1STejun Heo 	root = kernfs_root(kn);
401fd7b9f7bSTejun Heo  repeat:
40281c173cbSTejun Heo 	/*
40381c173cbSTejun Heo 	 * Moving/renaming is always done while holding reference.
404adc5e8b5STejun Heo 	 * kn->parent won't change beneath us.
405fd7b9f7bSTejun Heo 	 */
406adc5e8b5STejun Heo 	parent = kn->parent;
407fd7b9f7bSTejun Heo 
40881c173cbSTejun Heo 	WARN_ONCE(atomic_read(&kn->active) != KN_DEACTIVATED_BIAS,
40981c173cbSTejun Heo 		  "kernfs_put: %s/%s: released with incorrect active_ref %d\n",
41081c173cbSTejun Heo 		  parent ? parent->name : "", kn->name, atomic_read(&kn->active));
411fd7b9f7bSTejun Heo 
412df23fc39STejun Heo 	if (kernfs_type(kn) == KERNFS_LINK)
413adc5e8b5STejun Heo 		kernfs_put(kn->symlink.target_kn);
414dfeb0750STejun Heo 
41575287a67SAndrzej Hajda 	kfree_const(kn->name);
416dfeb0750STejun Heo 
417adc5e8b5STejun Heo 	if (kn->iattr) {
418adc5e8b5STejun Heo 		if (kn->iattr->ia_secdata)
419adc5e8b5STejun Heo 			security_release_secctx(kn->iattr->ia_secdata,
420adc5e8b5STejun Heo 						kn->iattr->ia_secdata_len);
421adc5e8b5STejun Heo 		simple_xattrs_free(&kn->iattr->xattrs);
4222322392bSTejun Heo 	}
423adc5e8b5STejun Heo 	kfree(kn->iattr);
424adc5e8b5STejun Heo 	ida_simple_remove(&root->ino_ida, kn->ino);
425a797bfc3STejun Heo 	kmem_cache_free(kernfs_node_cache, kn);
426fd7b9f7bSTejun Heo 
427324a56e1STejun Heo 	kn = parent;
428324a56e1STejun Heo 	if (kn) {
429adc5e8b5STejun Heo 		if (atomic_dec_and_test(&kn->count))
430fd7b9f7bSTejun Heo 			goto repeat;
431ba7443bcSTejun Heo 	} else {
432324a56e1STejun Heo 		/* just released the root kn, free @root too */
433bc755553STejun Heo 		ida_destroy(&root->ino_ida);
434ba7443bcSTejun Heo 		kfree(root);
435ba7443bcSTejun Heo 	}
436fd7b9f7bSTejun Heo }
437fd7b9f7bSTejun Heo EXPORT_SYMBOL_GPL(kernfs_put);
438fd7b9f7bSTejun Heo 
439c637b8acSTejun Heo static int kernfs_dop_revalidate(struct dentry *dentry, unsigned int flags)
440fd7b9f7bSTejun Heo {
441324a56e1STejun Heo 	struct kernfs_node *kn;
442fd7b9f7bSTejun Heo 
443fd7b9f7bSTejun Heo 	if (flags & LOOKUP_RCU)
444fd7b9f7bSTejun Heo 		return -ECHILD;
445fd7b9f7bSTejun Heo 
44619bbb926STejun Heo 	/* Always perform fresh lookup for negatives */
44719bbb926STejun Heo 	if (!dentry->d_inode)
44819bbb926STejun Heo 		goto out_bad_unlocked;
44919bbb926STejun Heo 
450324a56e1STejun Heo 	kn = dentry->d_fsdata;
451a797bfc3STejun Heo 	mutex_lock(&kernfs_mutex);
452fd7b9f7bSTejun Heo 
45381c173cbSTejun Heo 	/* The kernfs node has been deactivated */
45481c173cbSTejun Heo 	if (!kernfs_active(kn))
455fd7b9f7bSTejun Heo 		goto out_bad;
456fd7b9f7bSTejun Heo 
457c637b8acSTejun Heo 	/* The kernfs node has been moved? */
458adc5e8b5STejun Heo 	if (dentry->d_parent->d_fsdata != kn->parent)
459fd7b9f7bSTejun Heo 		goto out_bad;
460fd7b9f7bSTejun Heo 
461c637b8acSTejun Heo 	/* The kernfs node has been renamed */
462adc5e8b5STejun Heo 	if (strcmp(dentry->d_name.name, kn->name) != 0)
463fd7b9f7bSTejun Heo 		goto out_bad;
464fd7b9f7bSTejun Heo 
465c637b8acSTejun Heo 	/* The kernfs node has been moved to a different namespace */
466adc5e8b5STejun Heo 	if (kn->parent && kernfs_ns_enabled(kn->parent) &&
467c525aaddSTejun Heo 	    kernfs_info(dentry->d_sb)->ns != kn->ns)
468fd7b9f7bSTejun Heo 		goto out_bad;
469fd7b9f7bSTejun Heo 
470a797bfc3STejun Heo 	mutex_unlock(&kernfs_mutex);
471fd7b9f7bSTejun Heo 	return 1;
472fd7b9f7bSTejun Heo out_bad:
473a797bfc3STejun Heo 	mutex_unlock(&kernfs_mutex);
47419bbb926STejun Heo out_bad_unlocked:
475fd7b9f7bSTejun Heo 	return 0;
476fd7b9f7bSTejun Heo }
477fd7b9f7bSTejun Heo 
478c637b8acSTejun Heo static void kernfs_dop_release(struct dentry *dentry)
479fd7b9f7bSTejun Heo {
480fd7b9f7bSTejun Heo 	kernfs_put(dentry->d_fsdata);
481fd7b9f7bSTejun Heo }
482fd7b9f7bSTejun Heo 
483a797bfc3STejun Heo const struct dentry_operations kernfs_dops = {
484c637b8acSTejun Heo 	.d_revalidate	= kernfs_dop_revalidate,
485c637b8acSTejun Heo 	.d_release	= kernfs_dop_release,
486fd7b9f7bSTejun Heo };
487fd7b9f7bSTejun Heo 
4880c23b225STejun Heo /**
4890c23b225STejun Heo  * kernfs_node_from_dentry - determine kernfs_node associated with a dentry
4900c23b225STejun Heo  * @dentry: the dentry in question
4910c23b225STejun Heo  *
4920c23b225STejun Heo  * Return the kernfs_node associated with @dentry.  If @dentry is not a
4930c23b225STejun Heo  * kernfs one, %NULL is returned.
4940c23b225STejun Heo  *
4950c23b225STejun Heo  * While the returned kernfs_node will stay accessible as long as @dentry
4960c23b225STejun Heo  * is accessible, the returned node can be in any state and the caller is
4970c23b225STejun Heo  * fully responsible for determining what's accessible.
4980c23b225STejun Heo  */
4990c23b225STejun Heo struct kernfs_node *kernfs_node_from_dentry(struct dentry *dentry)
5000c23b225STejun Heo {
501f41c5934SLi Zefan 	if (dentry->d_sb->s_op == &kernfs_sops)
5020c23b225STejun Heo 		return dentry->d_fsdata;
5030c23b225STejun Heo 	return NULL;
5040c23b225STejun Heo }
5050c23b225STejun Heo 
506db4aad20STejun Heo static struct kernfs_node *__kernfs_new_node(struct kernfs_root *root,
507db4aad20STejun Heo 					     const char *name, umode_t mode,
508db4aad20STejun Heo 					     unsigned flags)
509fd7b9f7bSTejun Heo {
510324a56e1STejun Heo 	struct kernfs_node *kn;
511bc755553STejun Heo 	int ret;
512fd7b9f7bSTejun Heo 
513dfeb0750STejun Heo 	name = kstrdup_const(name, GFP_KERNEL);
514fd7b9f7bSTejun Heo 	if (!name)
515fd7b9f7bSTejun Heo 		return NULL;
516fd7b9f7bSTejun Heo 
517a797bfc3STejun Heo 	kn = kmem_cache_zalloc(kernfs_node_cache, GFP_KERNEL);
518324a56e1STejun Heo 	if (!kn)
519fd7b9f7bSTejun Heo 		goto err_out1;
520fd7b9f7bSTejun Heo 
521bc755553STejun Heo 	ret = ida_simple_get(&root->ino_ida, 1, 0, GFP_KERNEL);
522bc755553STejun Heo 	if (ret < 0)
523fd7b9f7bSTejun Heo 		goto err_out2;
524adc5e8b5STejun Heo 	kn->ino = ret;
525fd7b9f7bSTejun Heo 
526adc5e8b5STejun Heo 	atomic_set(&kn->count, 1);
52781c173cbSTejun Heo 	atomic_set(&kn->active, KN_DEACTIVATED_BIAS);
52835beab06STejun Heo 	RB_CLEAR_NODE(&kn->rb);
529fd7b9f7bSTejun Heo 
530adc5e8b5STejun Heo 	kn->name = name;
531adc5e8b5STejun Heo 	kn->mode = mode;
53281c173cbSTejun Heo 	kn->flags = flags;
533fd7b9f7bSTejun Heo 
534324a56e1STejun Heo 	return kn;
535fd7b9f7bSTejun Heo 
536fd7b9f7bSTejun Heo  err_out2:
537a797bfc3STejun Heo 	kmem_cache_free(kernfs_node_cache, kn);
538fd7b9f7bSTejun Heo  err_out1:
539dfeb0750STejun Heo 	kfree_const(name);
540fd7b9f7bSTejun Heo 	return NULL;
541fd7b9f7bSTejun Heo }
542fd7b9f7bSTejun Heo 
543db4aad20STejun Heo struct kernfs_node *kernfs_new_node(struct kernfs_node *parent,
544db4aad20STejun Heo 				    const char *name, umode_t mode,
545db4aad20STejun Heo 				    unsigned flags)
546db4aad20STejun Heo {
547db4aad20STejun Heo 	struct kernfs_node *kn;
548db4aad20STejun Heo 
549db4aad20STejun Heo 	kn = __kernfs_new_node(kernfs_root(parent), name, mode, flags);
550db4aad20STejun Heo 	if (kn) {
551db4aad20STejun Heo 		kernfs_get(parent);
552db4aad20STejun Heo 		kn->parent = parent;
553db4aad20STejun Heo 	}
554db4aad20STejun Heo 	return kn;
555db4aad20STejun Heo }
556db4aad20STejun Heo 
557fd7b9f7bSTejun Heo /**
558c637b8acSTejun Heo  *	kernfs_add_one - add kernfs_node to parent without warning
559324a56e1STejun Heo  *	@kn: kernfs_node to be added
560fd7b9f7bSTejun Heo  *
561db4aad20STejun Heo  *	The caller must already have initialized @kn->parent.  This
562db4aad20STejun Heo  *	function increments nlink of the parent's inode if @kn is a
563db4aad20STejun Heo  *	directory and link into the children list of the parent.
564fd7b9f7bSTejun Heo  *
565fd7b9f7bSTejun Heo  *	RETURNS:
566fd7b9f7bSTejun Heo  *	0 on success, -EEXIST if entry with the given name already
567fd7b9f7bSTejun Heo  *	exists.
568fd7b9f7bSTejun Heo  */
569988cd7afSTejun Heo int kernfs_add_one(struct kernfs_node *kn)
570fd7b9f7bSTejun Heo {
571db4aad20STejun Heo 	struct kernfs_node *parent = kn->parent;
572c525aaddSTejun Heo 	struct kernfs_iattrs *ps_iattr;
573988cd7afSTejun Heo 	bool has_ns;
574fd7b9f7bSTejun Heo 	int ret;
575fd7b9f7bSTejun Heo 
576988cd7afSTejun Heo 	mutex_lock(&kernfs_mutex);
577988cd7afSTejun Heo 
578988cd7afSTejun Heo 	ret = -EINVAL;
579988cd7afSTejun Heo 	has_ns = kernfs_ns_enabled(parent);
580988cd7afSTejun Heo 	if (WARN(has_ns != (bool)kn->ns, KERN_WARNING "kernfs: ns %s in '%s' for '%s'\n",
581988cd7afSTejun Heo 		 has_ns ? "required" : "invalid", parent->name, kn->name))
582988cd7afSTejun Heo 		goto out_unlock;
583fd7b9f7bSTejun Heo 
584df23fc39STejun Heo 	if (kernfs_type(parent) != KERNFS_DIR)
585988cd7afSTejun Heo 		goto out_unlock;
586fd7b9f7bSTejun Heo 
587988cd7afSTejun Heo 	ret = -ENOENT;
588d35258efSTejun Heo 	if ((parent->flags & KERNFS_ACTIVATED) && !kernfs_active(parent))
589988cd7afSTejun Heo 		goto out_unlock;
590798c75a0SGreg Kroah-Hartman 
591c637b8acSTejun Heo 	kn->hash = kernfs_name_hash(kn->name, kn->ns);
592fd7b9f7bSTejun Heo 
593c637b8acSTejun Heo 	ret = kernfs_link_sibling(kn);
594fd7b9f7bSTejun Heo 	if (ret)
595988cd7afSTejun Heo 		goto out_unlock;
596fd7b9f7bSTejun Heo 
597fd7b9f7bSTejun Heo 	/* Update timestamps on the parent */
598adc5e8b5STejun Heo 	ps_iattr = parent->iattr;
599fd7b9f7bSTejun Heo 	if (ps_iattr) {
600fd7b9f7bSTejun Heo 		struct iattr *ps_iattrs = &ps_iattr->ia_iattr;
601fd7b9f7bSTejun Heo 		ps_iattrs->ia_ctime = ps_iattrs->ia_mtime = CURRENT_TIME;
602fd7b9f7bSTejun Heo 	}
603fd7b9f7bSTejun Heo 
604d35258efSTejun Heo 	mutex_unlock(&kernfs_mutex);
605d35258efSTejun Heo 
606d35258efSTejun Heo 	/*
607d35258efSTejun Heo 	 * Activate the new node unless CREATE_DEACTIVATED is requested.
608d35258efSTejun Heo 	 * If not activated here, the kernfs user is responsible for
609d35258efSTejun Heo 	 * activating the node with kernfs_activate().  A node which hasn't
610d35258efSTejun Heo 	 * been activated is not visible to userland and its removal won't
611d35258efSTejun Heo 	 * trigger deactivation.
612d35258efSTejun Heo 	 */
613d35258efSTejun Heo 	if (!(kernfs_root(kn)->flags & KERNFS_ROOT_CREATE_DEACTIVATED))
614d35258efSTejun Heo 		kernfs_activate(kn);
615d35258efSTejun Heo 	return 0;
616d35258efSTejun Heo 
617988cd7afSTejun Heo out_unlock:
618a797bfc3STejun Heo 	mutex_unlock(&kernfs_mutex);
619988cd7afSTejun Heo 	return ret;
620fd7b9f7bSTejun Heo }
621fd7b9f7bSTejun Heo 
622fd7b9f7bSTejun Heo /**
623324a56e1STejun Heo  * kernfs_find_ns - find kernfs_node with the given name
624324a56e1STejun Heo  * @parent: kernfs_node to search under
625fd7b9f7bSTejun Heo  * @name: name to look for
626fd7b9f7bSTejun Heo  * @ns: the namespace tag to use
627fd7b9f7bSTejun Heo  *
628324a56e1STejun Heo  * Look for kernfs_node with name @name under @parent.  Returns pointer to
629324a56e1STejun Heo  * the found kernfs_node on success, %NULL on failure.
630fd7b9f7bSTejun Heo  */
631324a56e1STejun Heo static struct kernfs_node *kernfs_find_ns(struct kernfs_node *parent,
632fd7b9f7bSTejun Heo 					  const unsigned char *name,
633fd7b9f7bSTejun Heo 					  const void *ns)
634fd7b9f7bSTejun Heo {
635adc5e8b5STejun Heo 	struct rb_node *node = parent->dir.children.rb_node;
636ac9bba03STejun Heo 	bool has_ns = kernfs_ns_enabled(parent);
637fd7b9f7bSTejun Heo 	unsigned int hash;
638fd7b9f7bSTejun Heo 
639a797bfc3STejun Heo 	lockdep_assert_held(&kernfs_mutex);
640fd7b9f7bSTejun Heo 
641fd7b9f7bSTejun Heo 	if (has_ns != (bool)ns) {
642c637b8acSTejun Heo 		WARN(1, KERN_WARNING "kernfs: ns %s in '%s' for '%s'\n",
643adc5e8b5STejun Heo 		     has_ns ? "required" : "invalid", parent->name, name);
644fd7b9f7bSTejun Heo 		return NULL;
645fd7b9f7bSTejun Heo 	}
646fd7b9f7bSTejun Heo 
647c637b8acSTejun Heo 	hash = kernfs_name_hash(name, ns);
648fd7b9f7bSTejun Heo 	while (node) {
649324a56e1STejun Heo 		struct kernfs_node *kn;
650fd7b9f7bSTejun Heo 		int result;
651fd7b9f7bSTejun Heo 
652324a56e1STejun Heo 		kn = rb_to_kn(node);
653c637b8acSTejun Heo 		result = kernfs_name_compare(hash, name, ns, kn);
654fd7b9f7bSTejun Heo 		if (result < 0)
655fd7b9f7bSTejun Heo 			node = node->rb_left;
656fd7b9f7bSTejun Heo 		else if (result > 0)
657fd7b9f7bSTejun Heo 			node = node->rb_right;
658fd7b9f7bSTejun Heo 		else
659324a56e1STejun Heo 			return kn;
660fd7b9f7bSTejun Heo 	}
661fd7b9f7bSTejun Heo 	return NULL;
662fd7b9f7bSTejun Heo }
663fd7b9f7bSTejun Heo 
664fd7b9f7bSTejun Heo /**
665324a56e1STejun Heo  * kernfs_find_and_get_ns - find and get kernfs_node with the given name
666324a56e1STejun Heo  * @parent: kernfs_node to search under
667fd7b9f7bSTejun Heo  * @name: name to look for
668fd7b9f7bSTejun Heo  * @ns: the namespace tag to use
669fd7b9f7bSTejun Heo  *
670324a56e1STejun Heo  * Look for kernfs_node with name @name under @parent and get a reference
671fd7b9f7bSTejun Heo  * if found.  This function may sleep and returns pointer to the found
672324a56e1STejun Heo  * kernfs_node on success, %NULL on failure.
673fd7b9f7bSTejun Heo  */
674324a56e1STejun Heo struct kernfs_node *kernfs_find_and_get_ns(struct kernfs_node *parent,
675fd7b9f7bSTejun Heo 					   const char *name, const void *ns)
676fd7b9f7bSTejun Heo {
677324a56e1STejun Heo 	struct kernfs_node *kn;
678fd7b9f7bSTejun Heo 
679a797bfc3STejun Heo 	mutex_lock(&kernfs_mutex);
680324a56e1STejun Heo 	kn = kernfs_find_ns(parent, name, ns);
681324a56e1STejun Heo 	kernfs_get(kn);
682a797bfc3STejun Heo 	mutex_unlock(&kernfs_mutex);
683fd7b9f7bSTejun Heo 
684324a56e1STejun Heo 	return kn;
685fd7b9f7bSTejun Heo }
686fd7b9f7bSTejun Heo EXPORT_SYMBOL_GPL(kernfs_find_and_get_ns);
687fd7b9f7bSTejun Heo 
688fd7b9f7bSTejun Heo /**
689ba7443bcSTejun Heo  * kernfs_create_root - create a new kernfs hierarchy
69090c07c89STejun Heo  * @scops: optional syscall operations for the hierarchy
691d35258efSTejun Heo  * @flags: KERNFS_ROOT_* flags
692ba7443bcSTejun Heo  * @priv: opaque data associated with the new directory
693ba7443bcSTejun Heo  *
694ba7443bcSTejun Heo  * Returns the root of the new hierarchy on success, ERR_PTR() value on
695ba7443bcSTejun Heo  * failure.
696ba7443bcSTejun Heo  */
69790c07c89STejun Heo struct kernfs_root *kernfs_create_root(struct kernfs_syscall_ops *scops,
698d35258efSTejun Heo 				       unsigned int flags, void *priv)
699ba7443bcSTejun Heo {
700ba7443bcSTejun Heo 	struct kernfs_root *root;
701324a56e1STejun Heo 	struct kernfs_node *kn;
702ba7443bcSTejun Heo 
703ba7443bcSTejun Heo 	root = kzalloc(sizeof(*root), GFP_KERNEL);
704ba7443bcSTejun Heo 	if (!root)
705ba7443bcSTejun Heo 		return ERR_PTR(-ENOMEM);
706ba7443bcSTejun Heo 
707bc755553STejun Heo 	ida_init(&root->ino_ida);
7087d568a83STejun Heo 	INIT_LIST_HEAD(&root->supers);
709bc755553STejun Heo 
710db4aad20STejun Heo 	kn = __kernfs_new_node(root, "", S_IFDIR | S_IRUGO | S_IXUGO,
711db4aad20STejun Heo 			       KERNFS_DIR);
712324a56e1STejun Heo 	if (!kn) {
713bc755553STejun Heo 		ida_destroy(&root->ino_ida);
714ba7443bcSTejun Heo 		kfree(root);
715ba7443bcSTejun Heo 		return ERR_PTR(-ENOMEM);
716ba7443bcSTejun Heo 	}
717ba7443bcSTejun Heo 
718324a56e1STejun Heo 	kn->priv = priv;
719adc5e8b5STejun Heo 	kn->dir.root = root;
720ba7443bcSTejun Heo 
72190c07c89STejun Heo 	root->syscall_ops = scops;
722d35258efSTejun Heo 	root->flags = flags;
723324a56e1STejun Heo 	root->kn = kn;
724abd54f02STejun Heo 	init_waitqueue_head(&root->deactivate_waitq);
725ba7443bcSTejun Heo 
726d35258efSTejun Heo 	if (!(root->flags & KERNFS_ROOT_CREATE_DEACTIVATED))
727d35258efSTejun Heo 		kernfs_activate(kn);
728d35258efSTejun Heo 
729ba7443bcSTejun Heo 	return root;
730ba7443bcSTejun Heo }
731ba7443bcSTejun Heo 
732ba7443bcSTejun Heo /**
733ba7443bcSTejun Heo  * kernfs_destroy_root - destroy a kernfs hierarchy
734ba7443bcSTejun Heo  * @root: root of the hierarchy to destroy
735ba7443bcSTejun Heo  *
736ba7443bcSTejun Heo  * Destroy the hierarchy anchored at @root by removing all existing
737ba7443bcSTejun Heo  * directories and destroying @root.
738ba7443bcSTejun Heo  */
739ba7443bcSTejun Heo void kernfs_destroy_root(struct kernfs_root *root)
740ba7443bcSTejun Heo {
741324a56e1STejun Heo 	kernfs_remove(root->kn);	/* will also free @root */
742ba7443bcSTejun Heo }
743ba7443bcSTejun Heo 
744ba7443bcSTejun Heo /**
745fd7b9f7bSTejun Heo  * kernfs_create_dir_ns - create a directory
746fd7b9f7bSTejun Heo  * @parent: parent in which to create a new directory
747fd7b9f7bSTejun Heo  * @name: name of the new directory
748bb8b9d09STejun Heo  * @mode: mode of the new directory
749fd7b9f7bSTejun Heo  * @priv: opaque data associated with the new directory
750fd7b9f7bSTejun Heo  * @ns: optional namespace tag of the directory
751fd7b9f7bSTejun Heo  *
752fd7b9f7bSTejun Heo  * Returns the created node on success, ERR_PTR() value on failure.
753fd7b9f7bSTejun Heo  */
754324a56e1STejun Heo struct kernfs_node *kernfs_create_dir_ns(struct kernfs_node *parent,
755bb8b9d09STejun Heo 					 const char *name, umode_t mode,
756bb8b9d09STejun Heo 					 void *priv, const void *ns)
757fd7b9f7bSTejun Heo {
758324a56e1STejun Heo 	struct kernfs_node *kn;
759fd7b9f7bSTejun Heo 	int rc;
760fd7b9f7bSTejun Heo 
761fd7b9f7bSTejun Heo 	/* allocate */
762db4aad20STejun Heo 	kn = kernfs_new_node(parent, name, mode | S_IFDIR, KERNFS_DIR);
763324a56e1STejun Heo 	if (!kn)
764fd7b9f7bSTejun Heo 		return ERR_PTR(-ENOMEM);
765fd7b9f7bSTejun Heo 
766adc5e8b5STejun Heo 	kn->dir.root = parent->dir.root;
767adc5e8b5STejun Heo 	kn->ns = ns;
768324a56e1STejun Heo 	kn->priv = priv;
769fd7b9f7bSTejun Heo 
770fd7b9f7bSTejun Heo 	/* link in */
771988cd7afSTejun Heo 	rc = kernfs_add_one(kn);
772fd7b9f7bSTejun Heo 	if (!rc)
773324a56e1STejun Heo 		return kn;
774fd7b9f7bSTejun Heo 
775324a56e1STejun Heo 	kernfs_put(kn);
776fd7b9f7bSTejun Heo 	return ERR_PTR(rc);
777fd7b9f7bSTejun Heo }
778fd7b9f7bSTejun Heo 
779c637b8acSTejun Heo static struct dentry *kernfs_iop_lookup(struct inode *dir,
780c637b8acSTejun Heo 					struct dentry *dentry,
781fd7b9f7bSTejun Heo 					unsigned int flags)
782fd7b9f7bSTejun Heo {
78319bbb926STejun Heo 	struct dentry *ret;
784324a56e1STejun Heo 	struct kernfs_node *parent = dentry->d_parent->d_fsdata;
785324a56e1STejun Heo 	struct kernfs_node *kn;
786fd7b9f7bSTejun Heo 	struct inode *inode;
787fd7b9f7bSTejun Heo 	const void *ns = NULL;
788fd7b9f7bSTejun Heo 
789a797bfc3STejun Heo 	mutex_lock(&kernfs_mutex);
790fd7b9f7bSTejun Heo 
791324a56e1STejun Heo 	if (kernfs_ns_enabled(parent))
792c525aaddSTejun Heo 		ns = kernfs_info(dir->i_sb)->ns;
793fd7b9f7bSTejun Heo 
794324a56e1STejun Heo 	kn = kernfs_find_ns(parent, dentry->d_name.name, ns);
795fd7b9f7bSTejun Heo 
796fd7b9f7bSTejun Heo 	/* no such entry */
797b9c9dad0STejun Heo 	if (!kn || !kernfs_active(kn)) {
79819bbb926STejun Heo 		ret = NULL;
799fd7b9f7bSTejun Heo 		goto out_unlock;
800fd7b9f7bSTejun Heo 	}
801324a56e1STejun Heo 	kernfs_get(kn);
802324a56e1STejun Heo 	dentry->d_fsdata = kn;
803fd7b9f7bSTejun Heo 
804fd7b9f7bSTejun Heo 	/* attach dentry and inode */
805c637b8acSTejun Heo 	inode = kernfs_get_inode(dir->i_sb, kn);
806fd7b9f7bSTejun Heo 	if (!inode) {
807fd7b9f7bSTejun Heo 		ret = ERR_PTR(-ENOMEM);
808fd7b9f7bSTejun Heo 		goto out_unlock;
809fd7b9f7bSTejun Heo 	}
810fd7b9f7bSTejun Heo 
811fd7b9f7bSTejun Heo 	/* instantiate and hash dentry */
81241d28bcaSAl Viro 	ret = d_splice_alias(inode, dentry);
813fd7b9f7bSTejun Heo  out_unlock:
814a797bfc3STejun Heo 	mutex_unlock(&kernfs_mutex);
815fd7b9f7bSTejun Heo 	return ret;
816fd7b9f7bSTejun Heo }
817fd7b9f7bSTejun Heo 
81880b9bbefSTejun Heo static int kernfs_iop_mkdir(struct inode *dir, struct dentry *dentry,
81980b9bbefSTejun Heo 			    umode_t mode)
82080b9bbefSTejun Heo {
82180b9bbefSTejun Heo 	struct kernfs_node *parent = dir->i_private;
82290c07c89STejun Heo 	struct kernfs_syscall_ops *scops = kernfs_root(parent)->syscall_ops;
82307c7530dSTejun Heo 	int ret;
82480b9bbefSTejun Heo 
82590c07c89STejun Heo 	if (!scops || !scops->mkdir)
82680b9bbefSTejun Heo 		return -EPERM;
82780b9bbefSTejun Heo 
82807c7530dSTejun Heo 	if (!kernfs_get_active(parent))
82907c7530dSTejun Heo 		return -ENODEV;
83007c7530dSTejun Heo 
83190c07c89STejun Heo 	ret = scops->mkdir(parent, dentry->d_name.name, mode);
83207c7530dSTejun Heo 
83307c7530dSTejun Heo 	kernfs_put_active(parent);
83407c7530dSTejun Heo 	return ret;
83580b9bbefSTejun Heo }
83680b9bbefSTejun Heo 
83780b9bbefSTejun Heo static int kernfs_iop_rmdir(struct inode *dir, struct dentry *dentry)
83880b9bbefSTejun Heo {
83980b9bbefSTejun Heo 	struct kernfs_node *kn  = dentry->d_fsdata;
84090c07c89STejun Heo 	struct kernfs_syscall_ops *scops = kernfs_root(kn)->syscall_ops;
84107c7530dSTejun Heo 	int ret;
84280b9bbefSTejun Heo 
84390c07c89STejun Heo 	if (!scops || !scops->rmdir)
84480b9bbefSTejun Heo 		return -EPERM;
84580b9bbefSTejun Heo 
84607c7530dSTejun Heo 	if (!kernfs_get_active(kn))
84707c7530dSTejun Heo 		return -ENODEV;
84807c7530dSTejun Heo 
84990c07c89STejun Heo 	ret = scops->rmdir(kn);
85007c7530dSTejun Heo 
85107c7530dSTejun Heo 	kernfs_put_active(kn);
85207c7530dSTejun Heo 	return ret;
85380b9bbefSTejun Heo }
85480b9bbefSTejun Heo 
85580b9bbefSTejun Heo static int kernfs_iop_rename(struct inode *old_dir, struct dentry *old_dentry,
85680b9bbefSTejun Heo 			     struct inode *new_dir, struct dentry *new_dentry)
85780b9bbefSTejun Heo {
85880b9bbefSTejun Heo 	struct kernfs_node *kn  = old_dentry->d_fsdata;
85980b9bbefSTejun Heo 	struct kernfs_node *new_parent = new_dir->i_private;
86090c07c89STejun Heo 	struct kernfs_syscall_ops *scops = kernfs_root(kn)->syscall_ops;
86107c7530dSTejun Heo 	int ret;
86280b9bbefSTejun Heo 
86390c07c89STejun Heo 	if (!scops || !scops->rename)
86480b9bbefSTejun Heo 		return -EPERM;
86580b9bbefSTejun Heo 
86607c7530dSTejun Heo 	if (!kernfs_get_active(kn))
86707c7530dSTejun Heo 		return -ENODEV;
86807c7530dSTejun Heo 
86907c7530dSTejun Heo 	if (!kernfs_get_active(new_parent)) {
87007c7530dSTejun Heo 		kernfs_put_active(kn);
87107c7530dSTejun Heo 		return -ENODEV;
87207c7530dSTejun Heo 	}
87307c7530dSTejun Heo 
87490c07c89STejun Heo 	ret = scops->rename(kn, new_parent, new_dentry->d_name.name);
87507c7530dSTejun Heo 
87607c7530dSTejun Heo 	kernfs_put_active(new_parent);
87707c7530dSTejun Heo 	kernfs_put_active(kn);
87807c7530dSTejun Heo 	return ret;
87980b9bbefSTejun Heo }
88080b9bbefSTejun Heo 
881a797bfc3STejun Heo const struct inode_operations kernfs_dir_iops = {
882c637b8acSTejun Heo 	.lookup		= kernfs_iop_lookup,
883c637b8acSTejun Heo 	.permission	= kernfs_iop_permission,
884c637b8acSTejun Heo 	.setattr	= kernfs_iop_setattr,
885c637b8acSTejun Heo 	.getattr	= kernfs_iop_getattr,
886c637b8acSTejun Heo 	.setxattr	= kernfs_iop_setxattr,
887c637b8acSTejun Heo 	.removexattr	= kernfs_iop_removexattr,
888c637b8acSTejun Heo 	.getxattr	= kernfs_iop_getxattr,
889c637b8acSTejun Heo 	.listxattr	= kernfs_iop_listxattr,
89080b9bbefSTejun Heo 
89180b9bbefSTejun Heo 	.mkdir		= kernfs_iop_mkdir,
89280b9bbefSTejun Heo 	.rmdir		= kernfs_iop_rmdir,
89380b9bbefSTejun Heo 	.rename		= kernfs_iop_rename,
894fd7b9f7bSTejun Heo };
895fd7b9f7bSTejun Heo 
896c637b8acSTejun Heo static struct kernfs_node *kernfs_leftmost_descendant(struct kernfs_node *pos)
897fd7b9f7bSTejun Heo {
898324a56e1STejun Heo 	struct kernfs_node *last;
899fd7b9f7bSTejun Heo 
900fd7b9f7bSTejun Heo 	while (true) {
901fd7b9f7bSTejun Heo 		struct rb_node *rbn;
902fd7b9f7bSTejun Heo 
903fd7b9f7bSTejun Heo 		last = pos;
904fd7b9f7bSTejun Heo 
905df23fc39STejun Heo 		if (kernfs_type(pos) != KERNFS_DIR)
906fd7b9f7bSTejun Heo 			break;
907fd7b9f7bSTejun Heo 
908adc5e8b5STejun Heo 		rbn = rb_first(&pos->dir.children);
909fd7b9f7bSTejun Heo 		if (!rbn)
910fd7b9f7bSTejun Heo 			break;
911fd7b9f7bSTejun Heo 
912324a56e1STejun Heo 		pos = rb_to_kn(rbn);
913fd7b9f7bSTejun Heo 	}
914fd7b9f7bSTejun Heo 
915fd7b9f7bSTejun Heo 	return last;
916fd7b9f7bSTejun Heo }
917fd7b9f7bSTejun Heo 
918fd7b9f7bSTejun Heo /**
919c637b8acSTejun Heo  * kernfs_next_descendant_post - find the next descendant for post-order walk
920fd7b9f7bSTejun Heo  * @pos: the current position (%NULL to initiate traversal)
921324a56e1STejun Heo  * @root: kernfs_node whose descendants to walk
922fd7b9f7bSTejun Heo  *
923fd7b9f7bSTejun Heo  * Find the next descendant to visit for post-order traversal of @root's
924fd7b9f7bSTejun Heo  * descendants.  @root is included in the iteration and the last node to be
925fd7b9f7bSTejun Heo  * visited.
926fd7b9f7bSTejun Heo  */
927c637b8acSTejun Heo static struct kernfs_node *kernfs_next_descendant_post(struct kernfs_node *pos,
928324a56e1STejun Heo 						       struct kernfs_node *root)
929fd7b9f7bSTejun Heo {
930fd7b9f7bSTejun Heo 	struct rb_node *rbn;
931fd7b9f7bSTejun Heo 
932a797bfc3STejun Heo 	lockdep_assert_held(&kernfs_mutex);
933fd7b9f7bSTejun Heo 
934fd7b9f7bSTejun Heo 	/* if first iteration, visit leftmost descendant which may be root */
935fd7b9f7bSTejun Heo 	if (!pos)
936c637b8acSTejun Heo 		return kernfs_leftmost_descendant(root);
937fd7b9f7bSTejun Heo 
938fd7b9f7bSTejun Heo 	/* if we visited @root, we're done */
939fd7b9f7bSTejun Heo 	if (pos == root)
940fd7b9f7bSTejun Heo 		return NULL;
941fd7b9f7bSTejun Heo 
942fd7b9f7bSTejun Heo 	/* if there's an unvisited sibling, visit its leftmost descendant */
943adc5e8b5STejun Heo 	rbn = rb_next(&pos->rb);
944fd7b9f7bSTejun Heo 	if (rbn)
945c637b8acSTejun Heo 		return kernfs_leftmost_descendant(rb_to_kn(rbn));
946fd7b9f7bSTejun Heo 
947fd7b9f7bSTejun Heo 	/* no sibling left, visit parent */
948adc5e8b5STejun Heo 	return pos->parent;
949fd7b9f7bSTejun Heo }
950fd7b9f7bSTejun Heo 
951d35258efSTejun Heo /**
952d35258efSTejun Heo  * kernfs_activate - activate a node which started deactivated
953d35258efSTejun Heo  * @kn: kernfs_node whose subtree is to be activated
954d35258efSTejun Heo  *
955d35258efSTejun Heo  * If the root has KERNFS_ROOT_CREATE_DEACTIVATED set, a newly created node
956d35258efSTejun Heo  * needs to be explicitly activated.  A node which hasn't been activated
957d35258efSTejun Heo  * isn't visible to userland and deactivation is skipped during its
958d35258efSTejun Heo  * removal.  This is useful to construct atomic init sequences where
959d35258efSTejun Heo  * creation of multiple nodes should either succeed or fail atomically.
960d35258efSTejun Heo  *
961d35258efSTejun Heo  * The caller is responsible for ensuring that this function is not called
962d35258efSTejun Heo  * after kernfs_remove*() is invoked on @kn.
963d35258efSTejun Heo  */
964d35258efSTejun Heo void kernfs_activate(struct kernfs_node *kn)
965d35258efSTejun Heo {
966d35258efSTejun Heo 	struct kernfs_node *pos;
967d35258efSTejun Heo 
968d35258efSTejun Heo 	mutex_lock(&kernfs_mutex);
969d35258efSTejun Heo 
970d35258efSTejun Heo 	pos = NULL;
971d35258efSTejun Heo 	while ((pos = kernfs_next_descendant_post(pos, kn))) {
972d35258efSTejun Heo 		if (!pos || (pos->flags & KERNFS_ACTIVATED))
973d35258efSTejun Heo 			continue;
974d35258efSTejun Heo 
975d35258efSTejun Heo 		WARN_ON_ONCE(pos->parent && RB_EMPTY_NODE(&pos->rb));
976d35258efSTejun Heo 		WARN_ON_ONCE(atomic_read(&pos->active) != KN_DEACTIVATED_BIAS);
977d35258efSTejun Heo 
978d35258efSTejun Heo 		atomic_sub(KN_DEACTIVATED_BIAS, &pos->active);
979d35258efSTejun Heo 		pos->flags |= KERNFS_ACTIVATED;
980d35258efSTejun Heo 	}
981d35258efSTejun Heo 
982d35258efSTejun Heo 	mutex_unlock(&kernfs_mutex);
983d35258efSTejun Heo }
984d35258efSTejun Heo 
985988cd7afSTejun Heo static void __kernfs_remove(struct kernfs_node *kn)
986fd7b9f7bSTejun Heo {
98735beab06STejun Heo 	struct kernfs_node *pos;
98835beab06STejun Heo 
98935beab06STejun Heo 	lockdep_assert_held(&kernfs_mutex);
990fd7b9f7bSTejun Heo 
9916b0afc2aSTejun Heo 	/*
9926b0afc2aSTejun Heo 	 * Short-circuit if non-root @kn has already finished removal.
9936b0afc2aSTejun Heo 	 * This is for kernfs_remove_self() which plays with active ref
9946b0afc2aSTejun Heo 	 * after removal.
9956b0afc2aSTejun Heo 	 */
9966b0afc2aSTejun Heo 	if (!kn || (kn->parent && RB_EMPTY_NODE(&kn->rb)))
997ce9b499cSGreg Kroah-Hartman 		return;
998ce9b499cSGreg Kroah-Hartman 
999c637b8acSTejun Heo 	pr_debug("kernfs %s: removing\n", kn->name);
1000fd7b9f7bSTejun Heo 
100181c173cbSTejun Heo 	/* prevent any new usage under @kn by deactivating all nodes */
100235beab06STejun Heo 	pos = NULL;
100335beab06STejun Heo 	while ((pos = kernfs_next_descendant_post(pos, kn)))
100481c173cbSTejun Heo 		if (kernfs_active(pos))
100581c173cbSTejun Heo 			atomic_add(KN_DEACTIVATED_BIAS, &pos->active);
100635beab06STejun Heo 
100735beab06STejun Heo 	/* deactivate and unlink the subtree node-by-node */
1008fd7b9f7bSTejun Heo 	do {
100935beab06STejun Heo 		pos = kernfs_leftmost_descendant(kn);
101035beab06STejun Heo 
101135beab06STejun Heo 		/*
101281c173cbSTejun Heo 		 * kernfs_drain() drops kernfs_mutex temporarily and @pos's
101381c173cbSTejun Heo 		 * base ref could have been put by someone else by the time
101481c173cbSTejun Heo 		 * the function returns.  Make sure it doesn't go away
101581c173cbSTejun Heo 		 * underneath us.
101635beab06STejun Heo 		 */
101735beab06STejun Heo 		kernfs_get(pos);
101835beab06STejun Heo 
1019d35258efSTejun Heo 		/*
1020d35258efSTejun Heo 		 * Drain iff @kn was activated.  This avoids draining and
1021d35258efSTejun Heo 		 * its lockdep annotations for nodes which have never been
1022d35258efSTejun Heo 		 * activated and allows embedding kernfs_remove() in create
1023d35258efSTejun Heo 		 * error paths without worrying about draining.
1024d35258efSTejun Heo 		 */
1025d35258efSTejun Heo 		if (kn->flags & KERNFS_ACTIVATED)
102681c173cbSTejun Heo 			kernfs_drain(pos);
1027d35258efSTejun Heo 		else
1028d35258efSTejun Heo 			WARN_ON_ONCE(atomic_read(&kn->active) != KN_DEACTIVATED_BIAS);
102935beab06STejun Heo 
103035beab06STejun Heo 		/*
103135beab06STejun Heo 		 * kernfs_unlink_sibling() succeeds once per node.  Use it
103235beab06STejun Heo 		 * to decide who's responsible for cleanups.
103335beab06STejun Heo 		 */
103435beab06STejun Heo 		if (!pos->parent || kernfs_unlink_sibling(pos)) {
103535beab06STejun Heo 			struct kernfs_iattrs *ps_iattr =
103635beab06STejun Heo 				pos->parent ? pos->parent->iattr : NULL;
103735beab06STejun Heo 
103835beab06STejun Heo 			/* update timestamps on the parent */
103935beab06STejun Heo 			if (ps_iattr) {
104035beab06STejun Heo 				ps_iattr->ia_iattr.ia_ctime = CURRENT_TIME;
104135beab06STejun Heo 				ps_iattr->ia_iattr.ia_mtime = CURRENT_TIME;
104235beab06STejun Heo 			}
104335beab06STejun Heo 
1044988cd7afSTejun Heo 			kernfs_put(pos);
104535beab06STejun Heo 		}
104635beab06STejun Heo 
104735beab06STejun Heo 		kernfs_put(pos);
104835beab06STejun Heo 	} while (pos != kn);
1049fd7b9f7bSTejun Heo }
1050fd7b9f7bSTejun Heo 
1051fd7b9f7bSTejun Heo /**
1052324a56e1STejun Heo  * kernfs_remove - remove a kernfs_node recursively
1053324a56e1STejun Heo  * @kn: the kernfs_node to remove
1054fd7b9f7bSTejun Heo  *
1055324a56e1STejun Heo  * Remove @kn along with all its subdirectories and files.
1056fd7b9f7bSTejun Heo  */
1057324a56e1STejun Heo void kernfs_remove(struct kernfs_node *kn)
1058fd7b9f7bSTejun Heo {
1059988cd7afSTejun Heo 	mutex_lock(&kernfs_mutex);
1060988cd7afSTejun Heo 	__kernfs_remove(kn);
1061988cd7afSTejun Heo 	mutex_unlock(&kernfs_mutex);
1062fd7b9f7bSTejun Heo }
1063fd7b9f7bSTejun Heo 
1064fd7b9f7bSTejun Heo /**
10656b0afc2aSTejun Heo  * kernfs_break_active_protection - break out of active protection
10666b0afc2aSTejun Heo  * @kn: the self kernfs_node
10676b0afc2aSTejun Heo  *
10686b0afc2aSTejun Heo  * The caller must be running off of a kernfs operation which is invoked
10696b0afc2aSTejun Heo  * with an active reference - e.g. one of kernfs_ops.  Each invocation of
10706b0afc2aSTejun Heo  * this function must also be matched with an invocation of
10716b0afc2aSTejun Heo  * kernfs_unbreak_active_protection().
10726b0afc2aSTejun Heo  *
10736b0afc2aSTejun Heo  * This function releases the active reference of @kn the caller is
10746b0afc2aSTejun Heo  * holding.  Once this function is called, @kn may be removed at any point
10756b0afc2aSTejun Heo  * and the caller is solely responsible for ensuring that the objects it
10766b0afc2aSTejun Heo  * dereferences are accessible.
10776b0afc2aSTejun Heo  */
10786b0afc2aSTejun Heo void kernfs_break_active_protection(struct kernfs_node *kn)
10796b0afc2aSTejun Heo {
10806b0afc2aSTejun Heo 	/*
10816b0afc2aSTejun Heo 	 * Take out ourself out of the active ref dependency chain.  If
10826b0afc2aSTejun Heo 	 * we're called without an active ref, lockdep will complain.
10836b0afc2aSTejun Heo 	 */
10846b0afc2aSTejun Heo 	kernfs_put_active(kn);
10856b0afc2aSTejun Heo }
10866b0afc2aSTejun Heo 
10876b0afc2aSTejun Heo /**
10886b0afc2aSTejun Heo  * kernfs_unbreak_active_protection - undo kernfs_break_active_protection()
10896b0afc2aSTejun Heo  * @kn: the self kernfs_node
10906b0afc2aSTejun Heo  *
10916b0afc2aSTejun Heo  * If kernfs_break_active_protection() was called, this function must be
10926b0afc2aSTejun Heo  * invoked before finishing the kernfs operation.  Note that while this
10936b0afc2aSTejun Heo  * function restores the active reference, it doesn't and can't actually
10946b0afc2aSTejun Heo  * restore the active protection - @kn may already or be in the process of
10956b0afc2aSTejun Heo  * being removed.  Once kernfs_break_active_protection() is invoked, that
10966b0afc2aSTejun Heo  * protection is irreversibly gone for the kernfs operation instance.
10976b0afc2aSTejun Heo  *
10986b0afc2aSTejun Heo  * While this function may be called at any point after
10996b0afc2aSTejun Heo  * kernfs_break_active_protection() is invoked, its most useful location
11006b0afc2aSTejun Heo  * would be right before the enclosing kernfs operation returns.
11016b0afc2aSTejun Heo  */
11026b0afc2aSTejun Heo void kernfs_unbreak_active_protection(struct kernfs_node *kn)
11036b0afc2aSTejun Heo {
11046b0afc2aSTejun Heo 	/*
11056b0afc2aSTejun Heo 	 * @kn->active could be in any state; however, the increment we do
11066b0afc2aSTejun Heo 	 * here will be undone as soon as the enclosing kernfs operation
11076b0afc2aSTejun Heo 	 * finishes and this temporary bump can't break anything.  If @kn
11086b0afc2aSTejun Heo 	 * is alive, nothing changes.  If @kn is being deactivated, the
11096b0afc2aSTejun Heo 	 * soon-to-follow put will either finish deactivation or restore
11106b0afc2aSTejun Heo 	 * deactivated state.  If @kn is already removed, the temporary
11116b0afc2aSTejun Heo 	 * bump is guaranteed to be gone before @kn is released.
11126b0afc2aSTejun Heo 	 */
11136b0afc2aSTejun Heo 	atomic_inc(&kn->active);
11146b0afc2aSTejun Heo 	if (kernfs_lockdep(kn))
11156b0afc2aSTejun Heo 		rwsem_acquire(&kn->dep_map, 0, 1, _RET_IP_);
11166b0afc2aSTejun Heo }
11176b0afc2aSTejun Heo 
11186b0afc2aSTejun Heo /**
11196b0afc2aSTejun Heo  * kernfs_remove_self - remove a kernfs_node from its own method
11206b0afc2aSTejun Heo  * @kn: the self kernfs_node to remove
11216b0afc2aSTejun Heo  *
11226b0afc2aSTejun Heo  * The caller must be running off of a kernfs operation which is invoked
11236b0afc2aSTejun Heo  * with an active reference - e.g. one of kernfs_ops.  This can be used to
11246b0afc2aSTejun Heo  * implement a file operation which deletes itself.
11256b0afc2aSTejun Heo  *
11266b0afc2aSTejun Heo  * For example, the "delete" file for a sysfs device directory can be
11276b0afc2aSTejun Heo  * implemented by invoking kernfs_remove_self() on the "delete" file
11286b0afc2aSTejun Heo  * itself.  This function breaks the circular dependency of trying to
11296b0afc2aSTejun Heo  * deactivate self while holding an active ref itself.  It isn't necessary
11306b0afc2aSTejun Heo  * to modify the usual removal path to use kernfs_remove_self().  The
11316b0afc2aSTejun Heo  * "delete" implementation can simply invoke kernfs_remove_self() on self
11326b0afc2aSTejun Heo  * before proceeding with the usual removal path.  kernfs will ignore later
11336b0afc2aSTejun Heo  * kernfs_remove() on self.
11346b0afc2aSTejun Heo  *
11356b0afc2aSTejun Heo  * kernfs_remove_self() can be called multiple times concurrently on the
11366b0afc2aSTejun Heo  * same kernfs_node.  Only the first one actually performs removal and
11376b0afc2aSTejun Heo  * returns %true.  All others will wait until the kernfs operation which
11386b0afc2aSTejun Heo  * won self-removal finishes and return %false.  Note that the losers wait
11396b0afc2aSTejun Heo  * for the completion of not only the winning kernfs_remove_self() but also
11406b0afc2aSTejun Heo  * the whole kernfs_ops which won the arbitration.  This can be used to
11416b0afc2aSTejun Heo  * guarantee, for example, all concurrent writes to a "delete" file to
11426b0afc2aSTejun Heo  * finish only after the whole operation is complete.
11436b0afc2aSTejun Heo  */
11446b0afc2aSTejun Heo bool kernfs_remove_self(struct kernfs_node *kn)
11456b0afc2aSTejun Heo {
11466b0afc2aSTejun Heo 	bool ret;
11476b0afc2aSTejun Heo 
11486b0afc2aSTejun Heo 	mutex_lock(&kernfs_mutex);
11496b0afc2aSTejun Heo 	kernfs_break_active_protection(kn);
11506b0afc2aSTejun Heo 
11516b0afc2aSTejun Heo 	/*
11526b0afc2aSTejun Heo 	 * SUICIDAL is used to arbitrate among competing invocations.  Only
11536b0afc2aSTejun Heo 	 * the first one will actually perform removal.  When the removal
11546b0afc2aSTejun Heo 	 * is complete, SUICIDED is set and the active ref is restored
11556b0afc2aSTejun Heo 	 * while holding kernfs_mutex.  The ones which lost arbitration
11566b0afc2aSTejun Heo 	 * waits for SUICDED && drained which can happen only after the
11576b0afc2aSTejun Heo 	 * enclosing kernfs operation which executed the winning instance
11586b0afc2aSTejun Heo 	 * of kernfs_remove_self() finished.
11596b0afc2aSTejun Heo 	 */
11606b0afc2aSTejun Heo 	if (!(kn->flags & KERNFS_SUICIDAL)) {
11616b0afc2aSTejun Heo 		kn->flags |= KERNFS_SUICIDAL;
11626b0afc2aSTejun Heo 		__kernfs_remove(kn);
11636b0afc2aSTejun Heo 		kn->flags |= KERNFS_SUICIDED;
11646b0afc2aSTejun Heo 		ret = true;
11656b0afc2aSTejun Heo 	} else {
11666b0afc2aSTejun Heo 		wait_queue_head_t *waitq = &kernfs_root(kn)->deactivate_waitq;
11676b0afc2aSTejun Heo 		DEFINE_WAIT(wait);
11686b0afc2aSTejun Heo 
11696b0afc2aSTejun Heo 		while (true) {
11706b0afc2aSTejun Heo 			prepare_to_wait(waitq, &wait, TASK_UNINTERRUPTIBLE);
11716b0afc2aSTejun Heo 
11726b0afc2aSTejun Heo 			if ((kn->flags & KERNFS_SUICIDED) &&
11736b0afc2aSTejun Heo 			    atomic_read(&kn->active) == KN_DEACTIVATED_BIAS)
11746b0afc2aSTejun Heo 				break;
11756b0afc2aSTejun Heo 
11766b0afc2aSTejun Heo 			mutex_unlock(&kernfs_mutex);
11776b0afc2aSTejun Heo 			schedule();
11786b0afc2aSTejun Heo 			mutex_lock(&kernfs_mutex);
11796b0afc2aSTejun Heo 		}
11806b0afc2aSTejun Heo 		finish_wait(waitq, &wait);
11816b0afc2aSTejun Heo 		WARN_ON_ONCE(!RB_EMPTY_NODE(&kn->rb));
11826b0afc2aSTejun Heo 		ret = false;
11836b0afc2aSTejun Heo 	}
11846b0afc2aSTejun Heo 
11856b0afc2aSTejun Heo 	/*
11866b0afc2aSTejun Heo 	 * This must be done while holding kernfs_mutex; otherwise, waiting
11876b0afc2aSTejun Heo 	 * for SUICIDED && deactivated could finish prematurely.
11886b0afc2aSTejun Heo 	 */
11896b0afc2aSTejun Heo 	kernfs_unbreak_active_protection(kn);
11906b0afc2aSTejun Heo 
11916b0afc2aSTejun Heo 	mutex_unlock(&kernfs_mutex);
11926b0afc2aSTejun Heo 	return ret;
11936b0afc2aSTejun Heo }
11946b0afc2aSTejun Heo 
11956b0afc2aSTejun Heo /**
1196324a56e1STejun Heo  * kernfs_remove_by_name_ns - find a kernfs_node by name and remove it
1197324a56e1STejun Heo  * @parent: parent of the target
1198324a56e1STejun Heo  * @name: name of the kernfs_node to remove
1199324a56e1STejun Heo  * @ns: namespace tag of the kernfs_node to remove
1200fd7b9f7bSTejun Heo  *
1201324a56e1STejun Heo  * Look for the kernfs_node with @name and @ns under @parent and remove it.
1202324a56e1STejun Heo  * Returns 0 on success, -ENOENT if such entry doesn't exist.
1203fd7b9f7bSTejun Heo  */
1204324a56e1STejun Heo int kernfs_remove_by_name_ns(struct kernfs_node *parent, const char *name,
1205fd7b9f7bSTejun Heo 			     const void *ns)
1206fd7b9f7bSTejun Heo {
1207324a56e1STejun Heo 	struct kernfs_node *kn;
1208fd7b9f7bSTejun Heo 
1209324a56e1STejun Heo 	if (!parent) {
1210c637b8acSTejun Heo 		WARN(1, KERN_WARNING "kernfs: can not remove '%s', no directory\n",
1211fd7b9f7bSTejun Heo 			name);
1212fd7b9f7bSTejun Heo 		return -ENOENT;
1213fd7b9f7bSTejun Heo 	}
1214fd7b9f7bSTejun Heo 
1215988cd7afSTejun Heo 	mutex_lock(&kernfs_mutex);
1216fd7b9f7bSTejun Heo 
1217324a56e1STejun Heo 	kn = kernfs_find_ns(parent, name, ns);
1218324a56e1STejun Heo 	if (kn)
1219988cd7afSTejun Heo 		__kernfs_remove(kn);
1220fd7b9f7bSTejun Heo 
1221988cd7afSTejun Heo 	mutex_unlock(&kernfs_mutex);
1222fd7b9f7bSTejun Heo 
1223324a56e1STejun Heo 	if (kn)
1224fd7b9f7bSTejun Heo 		return 0;
1225fd7b9f7bSTejun Heo 	else
1226fd7b9f7bSTejun Heo 		return -ENOENT;
1227fd7b9f7bSTejun Heo }
1228fd7b9f7bSTejun Heo 
1229fd7b9f7bSTejun Heo /**
1230fd7b9f7bSTejun Heo  * kernfs_rename_ns - move and rename a kernfs_node
1231324a56e1STejun Heo  * @kn: target node
1232fd7b9f7bSTejun Heo  * @new_parent: new parent to put @sd under
1233fd7b9f7bSTejun Heo  * @new_name: new name
1234fd7b9f7bSTejun Heo  * @new_ns: new namespace tag
1235fd7b9f7bSTejun Heo  */
1236324a56e1STejun Heo int kernfs_rename_ns(struct kernfs_node *kn, struct kernfs_node *new_parent,
1237fd7b9f7bSTejun Heo 		     const char *new_name, const void *new_ns)
1238fd7b9f7bSTejun Heo {
12393eef34adSTejun Heo 	struct kernfs_node *old_parent;
12403eef34adSTejun Heo 	const char *old_name = NULL;
1241fd7b9f7bSTejun Heo 	int error;
1242fd7b9f7bSTejun Heo 
12433eef34adSTejun Heo 	/* can't move or rename root */
12443eef34adSTejun Heo 	if (!kn->parent)
12453eef34adSTejun Heo 		return -EINVAL;
12463eef34adSTejun Heo 
1247ae34372eSTejun Heo 	mutex_lock(&kernfs_mutex);
1248d0ae3d43STejun Heo 
1249798c75a0SGreg Kroah-Hartman 	error = -ENOENT;
125081c173cbSTejun Heo 	if (!kernfs_active(kn) || !kernfs_active(new_parent))
1251798c75a0SGreg Kroah-Hartman 		goto out;
1252798c75a0SGreg Kroah-Hartman 
1253fd7b9f7bSTejun Heo 	error = 0;
1254adc5e8b5STejun Heo 	if ((kn->parent == new_parent) && (kn->ns == new_ns) &&
1255adc5e8b5STejun Heo 	    (strcmp(kn->name, new_name) == 0))
1256798c75a0SGreg Kroah-Hartman 		goto out;	/* nothing to rename */
1257fd7b9f7bSTejun Heo 
1258fd7b9f7bSTejun Heo 	error = -EEXIST;
1259fd7b9f7bSTejun Heo 	if (kernfs_find_ns(new_parent, new_name, new_ns))
1260798c75a0SGreg Kroah-Hartman 		goto out;
1261fd7b9f7bSTejun Heo 
1262324a56e1STejun Heo 	/* rename kernfs_node */
1263adc5e8b5STejun Heo 	if (strcmp(kn->name, new_name) != 0) {
1264fd7b9f7bSTejun Heo 		error = -ENOMEM;
126575287a67SAndrzej Hajda 		new_name = kstrdup_const(new_name, GFP_KERNEL);
1266fd7b9f7bSTejun Heo 		if (!new_name)
1267798c75a0SGreg Kroah-Hartman 			goto out;
12683eef34adSTejun Heo 	} else {
12693eef34adSTejun Heo 		new_name = NULL;
1270fd7b9f7bSTejun Heo 	}
1271fd7b9f7bSTejun Heo 
1272fd7b9f7bSTejun Heo 	/*
1273fd7b9f7bSTejun Heo 	 * Move to the appropriate place in the appropriate directories rbtree.
1274fd7b9f7bSTejun Heo 	 */
1275c637b8acSTejun Heo 	kernfs_unlink_sibling(kn);
1276fd7b9f7bSTejun Heo 	kernfs_get(new_parent);
12773eef34adSTejun Heo 
12783eef34adSTejun Heo 	/* rename_lock protects ->parent and ->name accessors */
12793eef34adSTejun Heo 	spin_lock_irq(&kernfs_rename_lock);
12803eef34adSTejun Heo 
12813eef34adSTejun Heo 	old_parent = kn->parent;
1282adc5e8b5STejun Heo 	kn->parent = new_parent;
12833eef34adSTejun Heo 
12843eef34adSTejun Heo 	kn->ns = new_ns;
12853eef34adSTejun Heo 	if (new_name) {
12863eef34adSTejun Heo 		old_name = kn->name;
12873eef34adSTejun Heo 		kn->name = new_name;
12883eef34adSTejun Heo 	}
12893eef34adSTejun Heo 
12903eef34adSTejun Heo 	spin_unlock_irq(&kernfs_rename_lock);
12913eef34adSTejun Heo 
12929561a896STejun Heo 	kn->hash = kernfs_name_hash(kn->name, kn->ns);
1293c637b8acSTejun Heo 	kernfs_link_sibling(kn);
1294fd7b9f7bSTejun Heo 
12953eef34adSTejun Heo 	kernfs_put(old_parent);
129675287a67SAndrzej Hajda 	kfree_const(old_name);
12973eef34adSTejun Heo 
1298fd7b9f7bSTejun Heo 	error = 0;
1299ae34372eSTejun Heo  out:
1300798c75a0SGreg Kroah-Hartman 	mutex_unlock(&kernfs_mutex);
1301fd7b9f7bSTejun Heo 	return error;
1302fd7b9f7bSTejun Heo }
1303fd7b9f7bSTejun Heo 
1304fd7b9f7bSTejun Heo /* Relationship between s_mode and the DT_xxx types */
1305324a56e1STejun Heo static inline unsigned char dt_type(struct kernfs_node *kn)
1306fd7b9f7bSTejun Heo {
1307adc5e8b5STejun Heo 	return (kn->mode >> 12) & 15;
1308fd7b9f7bSTejun Heo }
1309fd7b9f7bSTejun Heo 
1310c637b8acSTejun Heo static int kernfs_dir_fop_release(struct inode *inode, struct file *filp)
1311fd7b9f7bSTejun Heo {
1312fd7b9f7bSTejun Heo 	kernfs_put(filp->private_data);
1313fd7b9f7bSTejun Heo 	return 0;
1314fd7b9f7bSTejun Heo }
1315fd7b9f7bSTejun Heo 
1316c637b8acSTejun Heo static struct kernfs_node *kernfs_dir_pos(const void *ns,
1317324a56e1STejun Heo 	struct kernfs_node *parent, loff_t hash, struct kernfs_node *pos)
1318fd7b9f7bSTejun Heo {
1319fd7b9f7bSTejun Heo 	if (pos) {
132081c173cbSTejun Heo 		int valid = kernfs_active(pos) &&
1321798c75a0SGreg Kroah-Hartman 			pos->parent == parent && hash == pos->hash;
1322fd7b9f7bSTejun Heo 		kernfs_put(pos);
1323fd7b9f7bSTejun Heo 		if (!valid)
1324fd7b9f7bSTejun Heo 			pos = NULL;
1325fd7b9f7bSTejun Heo 	}
1326fd7b9f7bSTejun Heo 	if (!pos && (hash > 1) && (hash < INT_MAX)) {
1327adc5e8b5STejun Heo 		struct rb_node *node = parent->dir.children.rb_node;
1328fd7b9f7bSTejun Heo 		while (node) {
1329324a56e1STejun Heo 			pos = rb_to_kn(node);
1330fd7b9f7bSTejun Heo 
1331adc5e8b5STejun Heo 			if (hash < pos->hash)
1332fd7b9f7bSTejun Heo 				node = node->rb_left;
1333adc5e8b5STejun Heo 			else if (hash > pos->hash)
1334fd7b9f7bSTejun Heo 				node = node->rb_right;
1335fd7b9f7bSTejun Heo 			else
1336fd7b9f7bSTejun Heo 				break;
1337fd7b9f7bSTejun Heo 		}
1338fd7b9f7bSTejun Heo 	}
1339b9c9dad0STejun Heo 	/* Skip over entries which are dying/dead or in the wrong namespace */
1340b9c9dad0STejun Heo 	while (pos && (!kernfs_active(pos) || pos->ns != ns)) {
1341adc5e8b5STejun Heo 		struct rb_node *node = rb_next(&pos->rb);
1342fd7b9f7bSTejun Heo 		if (!node)
1343fd7b9f7bSTejun Heo 			pos = NULL;
1344fd7b9f7bSTejun Heo 		else
1345324a56e1STejun Heo 			pos = rb_to_kn(node);
1346fd7b9f7bSTejun Heo 	}
1347fd7b9f7bSTejun Heo 	return pos;
1348fd7b9f7bSTejun Heo }
1349fd7b9f7bSTejun Heo 
1350c637b8acSTejun Heo static struct kernfs_node *kernfs_dir_next_pos(const void *ns,
1351324a56e1STejun Heo 	struct kernfs_node *parent, ino_t ino, struct kernfs_node *pos)
1352fd7b9f7bSTejun Heo {
1353c637b8acSTejun Heo 	pos = kernfs_dir_pos(ns, parent, ino, pos);
1354b9c9dad0STejun Heo 	if (pos) {
1355fd7b9f7bSTejun Heo 		do {
1356adc5e8b5STejun Heo 			struct rb_node *node = rb_next(&pos->rb);
1357fd7b9f7bSTejun Heo 			if (!node)
1358fd7b9f7bSTejun Heo 				pos = NULL;
1359fd7b9f7bSTejun Heo 			else
1360324a56e1STejun Heo 				pos = rb_to_kn(node);
1361b9c9dad0STejun Heo 		} while (pos && (!kernfs_active(pos) || pos->ns != ns));
1362b9c9dad0STejun Heo 	}
1363fd7b9f7bSTejun Heo 	return pos;
1364fd7b9f7bSTejun Heo }
1365fd7b9f7bSTejun Heo 
1366c637b8acSTejun Heo static int kernfs_fop_readdir(struct file *file, struct dir_context *ctx)
1367fd7b9f7bSTejun Heo {
1368fd7b9f7bSTejun Heo 	struct dentry *dentry = file->f_path.dentry;
1369324a56e1STejun Heo 	struct kernfs_node *parent = dentry->d_fsdata;
1370324a56e1STejun Heo 	struct kernfs_node *pos = file->private_data;
1371fd7b9f7bSTejun Heo 	const void *ns = NULL;
1372fd7b9f7bSTejun Heo 
1373fd7b9f7bSTejun Heo 	if (!dir_emit_dots(file, ctx))
1374fd7b9f7bSTejun Heo 		return 0;
1375a797bfc3STejun Heo 	mutex_lock(&kernfs_mutex);
1376fd7b9f7bSTejun Heo 
1377324a56e1STejun Heo 	if (kernfs_ns_enabled(parent))
1378c525aaddSTejun Heo 		ns = kernfs_info(dentry->d_sb)->ns;
1379fd7b9f7bSTejun Heo 
1380c637b8acSTejun Heo 	for (pos = kernfs_dir_pos(ns, parent, ctx->pos, pos);
1381fd7b9f7bSTejun Heo 	     pos;
1382c637b8acSTejun Heo 	     pos = kernfs_dir_next_pos(ns, parent, ctx->pos, pos)) {
1383adc5e8b5STejun Heo 		const char *name = pos->name;
1384fd7b9f7bSTejun Heo 		unsigned int type = dt_type(pos);
1385fd7b9f7bSTejun Heo 		int len = strlen(name);
1386adc5e8b5STejun Heo 		ino_t ino = pos->ino;
1387fd7b9f7bSTejun Heo 
1388adc5e8b5STejun Heo 		ctx->pos = pos->hash;
1389fd7b9f7bSTejun Heo 		file->private_data = pos;
1390fd7b9f7bSTejun Heo 		kernfs_get(pos);
1391fd7b9f7bSTejun Heo 
1392a797bfc3STejun Heo 		mutex_unlock(&kernfs_mutex);
1393fd7b9f7bSTejun Heo 		if (!dir_emit(ctx, name, len, ino, type))
1394fd7b9f7bSTejun Heo 			return 0;
1395a797bfc3STejun Heo 		mutex_lock(&kernfs_mutex);
1396fd7b9f7bSTejun Heo 	}
1397a797bfc3STejun Heo 	mutex_unlock(&kernfs_mutex);
1398fd7b9f7bSTejun Heo 	file->private_data = NULL;
1399fd7b9f7bSTejun Heo 	ctx->pos = INT_MAX;
1400fd7b9f7bSTejun Heo 	return 0;
1401fd7b9f7bSTejun Heo }
1402fd7b9f7bSTejun Heo 
1403c637b8acSTejun Heo static loff_t kernfs_dir_fop_llseek(struct file *file, loff_t offset,
1404c637b8acSTejun Heo 				    int whence)
1405fd7b9f7bSTejun Heo {
1406fd7b9f7bSTejun Heo 	struct inode *inode = file_inode(file);
1407fd7b9f7bSTejun Heo 	loff_t ret;
1408fd7b9f7bSTejun Heo 
1409fd7b9f7bSTejun Heo 	mutex_lock(&inode->i_mutex);
1410fd7b9f7bSTejun Heo 	ret = generic_file_llseek(file, offset, whence);
1411fd7b9f7bSTejun Heo 	mutex_unlock(&inode->i_mutex);
1412fd7b9f7bSTejun Heo 
1413fd7b9f7bSTejun Heo 	return ret;
1414fd7b9f7bSTejun Heo }
1415fd7b9f7bSTejun Heo 
1416a797bfc3STejun Heo const struct file_operations kernfs_dir_fops = {
1417fd7b9f7bSTejun Heo 	.read		= generic_read_dir,
1418c637b8acSTejun Heo 	.iterate	= kernfs_fop_readdir,
1419c637b8acSTejun Heo 	.release	= kernfs_dir_fop_release,
1420c637b8acSTejun Heo 	.llseek		= kernfs_dir_fop_llseek,
1421fd7b9f7bSTejun Heo };
1422