162306a36Sopenharmony_ci// SPDX-License-Identifier: GPL-2.0-only
262306a36Sopenharmony_ci/*
362306a36Sopenharmony_ci * V9FS FID Management
462306a36Sopenharmony_ci *
562306a36Sopenharmony_ci *  Copyright (C) 2007 by Latchesar Ionkov <lucho@ionkov.net>
662306a36Sopenharmony_ci *  Copyright (C) 2005, 2006 by Eric Van Hensbergen <ericvh@gmail.com>
762306a36Sopenharmony_ci */
862306a36Sopenharmony_ci
962306a36Sopenharmony_ci#include <linux/module.h>
1062306a36Sopenharmony_ci#include <linux/errno.h>
1162306a36Sopenharmony_ci#include <linux/fs.h>
1262306a36Sopenharmony_ci#include <linux/slab.h>
1362306a36Sopenharmony_ci#include <linux/sched.h>
1462306a36Sopenharmony_ci#include <net/9p/9p.h>
1562306a36Sopenharmony_ci#include <net/9p/client.h>
1662306a36Sopenharmony_ci
1762306a36Sopenharmony_ci#include "v9fs.h"
1862306a36Sopenharmony_ci#include "v9fs_vfs.h"
1962306a36Sopenharmony_ci#include "fid.h"
2062306a36Sopenharmony_ci
2162306a36Sopenharmony_cistatic inline void __add_fid(struct dentry *dentry, struct p9_fid *fid)
2262306a36Sopenharmony_ci{
2362306a36Sopenharmony_ci	hlist_add_head(&fid->dlist, (struct hlist_head *)&dentry->d_fsdata);
2462306a36Sopenharmony_ci}
2562306a36Sopenharmony_ci
2662306a36Sopenharmony_ci
2762306a36Sopenharmony_ci/**
2862306a36Sopenharmony_ci * v9fs_fid_add - add a fid to a dentry
2962306a36Sopenharmony_ci * @dentry: dentry that the fid is being added to
3062306a36Sopenharmony_ci * @pfid: fid to add, NULLed out
3162306a36Sopenharmony_ci *
3262306a36Sopenharmony_ci */
3362306a36Sopenharmony_civoid v9fs_fid_add(struct dentry *dentry, struct p9_fid **pfid)
3462306a36Sopenharmony_ci{
3562306a36Sopenharmony_ci	struct p9_fid *fid = *pfid;
3662306a36Sopenharmony_ci
3762306a36Sopenharmony_ci	spin_lock(&dentry->d_lock);
3862306a36Sopenharmony_ci	__add_fid(dentry, fid);
3962306a36Sopenharmony_ci	spin_unlock(&dentry->d_lock);
4062306a36Sopenharmony_ci
4162306a36Sopenharmony_ci	*pfid = NULL;
4262306a36Sopenharmony_ci}
4362306a36Sopenharmony_ci
4462306a36Sopenharmony_cistatic bool v9fs_is_writeable(int mode)
4562306a36Sopenharmony_ci{
4662306a36Sopenharmony_ci	if (mode & (P9_OWRITE|P9_ORDWR))
4762306a36Sopenharmony_ci		return true;
4862306a36Sopenharmony_ci	else
4962306a36Sopenharmony_ci		return false;
5062306a36Sopenharmony_ci}
5162306a36Sopenharmony_ci
5262306a36Sopenharmony_ci/**
5362306a36Sopenharmony_ci * v9fs_fid_find_inode - search for an open fid off of the inode list
5462306a36Sopenharmony_ci * @inode: return a fid pointing to a specific inode
5562306a36Sopenharmony_ci * @want_writeable: only consider fids which are writeable
5662306a36Sopenharmony_ci * @uid: return a fid belonging to the specified user
5762306a36Sopenharmony_ci * @any: ignore uid as a selection criteria
5862306a36Sopenharmony_ci *
5962306a36Sopenharmony_ci */
6062306a36Sopenharmony_cistruct p9_fid *v9fs_fid_find_inode(struct inode *inode, bool want_writeable,
6162306a36Sopenharmony_ci	kuid_t uid, bool any)
6262306a36Sopenharmony_ci{
6362306a36Sopenharmony_ci	struct hlist_head *h;
6462306a36Sopenharmony_ci	struct p9_fid *fid, *ret = NULL;
6562306a36Sopenharmony_ci
6662306a36Sopenharmony_ci	p9_debug(P9_DEBUG_VFS, " inode: %p\n", inode);
6762306a36Sopenharmony_ci
6862306a36Sopenharmony_ci	spin_lock(&inode->i_lock);
6962306a36Sopenharmony_ci	h = (struct hlist_head *)&inode->i_private;
7062306a36Sopenharmony_ci	hlist_for_each_entry(fid, h, ilist) {
7162306a36Sopenharmony_ci		if (any || uid_eq(fid->uid, uid)) {
7262306a36Sopenharmony_ci			if (want_writeable && !v9fs_is_writeable(fid->mode)) {
7362306a36Sopenharmony_ci				p9_debug(P9_DEBUG_VFS, " mode: %x not writeable?\n",
7462306a36Sopenharmony_ci							fid->mode);
7562306a36Sopenharmony_ci				continue;
7662306a36Sopenharmony_ci			}
7762306a36Sopenharmony_ci			p9_fid_get(fid);
7862306a36Sopenharmony_ci			ret = fid;
7962306a36Sopenharmony_ci			break;
8062306a36Sopenharmony_ci		}
8162306a36Sopenharmony_ci	}
8262306a36Sopenharmony_ci	spin_unlock(&inode->i_lock);
8362306a36Sopenharmony_ci	return ret;
8462306a36Sopenharmony_ci}
8562306a36Sopenharmony_ci
8662306a36Sopenharmony_ci/**
8762306a36Sopenharmony_ci * v9fs_open_fid_add - add an open fid to an inode
8862306a36Sopenharmony_ci * @inode: inode that the fid is being added to
8962306a36Sopenharmony_ci * @pfid: fid to add, NULLed out
9062306a36Sopenharmony_ci *
9162306a36Sopenharmony_ci */
9262306a36Sopenharmony_ci
9362306a36Sopenharmony_civoid v9fs_open_fid_add(struct inode *inode, struct p9_fid **pfid)
9462306a36Sopenharmony_ci{
9562306a36Sopenharmony_ci	struct p9_fid *fid = *pfid;
9662306a36Sopenharmony_ci
9762306a36Sopenharmony_ci	spin_lock(&inode->i_lock);
9862306a36Sopenharmony_ci	hlist_add_head(&fid->ilist, (struct hlist_head *)&inode->i_private);
9962306a36Sopenharmony_ci	spin_unlock(&inode->i_lock);
10062306a36Sopenharmony_ci
10162306a36Sopenharmony_ci	*pfid = NULL;
10262306a36Sopenharmony_ci}
10362306a36Sopenharmony_ci
10462306a36Sopenharmony_ci
10562306a36Sopenharmony_ci/**
10662306a36Sopenharmony_ci * v9fs_fid_find - retrieve a fid that belongs to the specified uid
10762306a36Sopenharmony_ci * @dentry: dentry to look for fid in
10862306a36Sopenharmony_ci * @uid: return fid that belongs to the specified user
10962306a36Sopenharmony_ci * @any: if non-zero, return any fid associated with the dentry
11062306a36Sopenharmony_ci *
11162306a36Sopenharmony_ci */
11262306a36Sopenharmony_ci
11362306a36Sopenharmony_cistatic struct p9_fid *v9fs_fid_find(struct dentry *dentry, kuid_t uid, int any)
11462306a36Sopenharmony_ci{
11562306a36Sopenharmony_ci	struct p9_fid *fid, *ret;
11662306a36Sopenharmony_ci
11762306a36Sopenharmony_ci	p9_debug(P9_DEBUG_VFS, " dentry: %pd (%p) uid %d any %d\n",
11862306a36Sopenharmony_ci		 dentry, dentry, from_kuid(&init_user_ns, uid),
11962306a36Sopenharmony_ci		 any);
12062306a36Sopenharmony_ci	ret = NULL;
12162306a36Sopenharmony_ci	/* we'll recheck under lock if there's anything to look in */
12262306a36Sopenharmony_ci	if (dentry->d_fsdata) {
12362306a36Sopenharmony_ci		struct hlist_head *h = (struct hlist_head *)&dentry->d_fsdata;
12462306a36Sopenharmony_ci
12562306a36Sopenharmony_ci		spin_lock(&dentry->d_lock);
12662306a36Sopenharmony_ci		hlist_for_each_entry(fid, h, dlist) {
12762306a36Sopenharmony_ci			if (any || uid_eq(fid->uid, uid)) {
12862306a36Sopenharmony_ci				ret = fid;
12962306a36Sopenharmony_ci				p9_fid_get(ret);
13062306a36Sopenharmony_ci				break;
13162306a36Sopenharmony_ci			}
13262306a36Sopenharmony_ci		}
13362306a36Sopenharmony_ci		spin_unlock(&dentry->d_lock);
13462306a36Sopenharmony_ci	} else {
13562306a36Sopenharmony_ci		if (dentry->d_inode)
13662306a36Sopenharmony_ci			ret = v9fs_fid_find_inode(dentry->d_inode, false, uid, any);
13762306a36Sopenharmony_ci	}
13862306a36Sopenharmony_ci
13962306a36Sopenharmony_ci	return ret;
14062306a36Sopenharmony_ci}
14162306a36Sopenharmony_ci
14262306a36Sopenharmony_ci/*
14362306a36Sopenharmony_ci * We need to hold v9ses->rename_sem as long as we hold references
14462306a36Sopenharmony_ci * to returned path array. Array element contain pointers to
14562306a36Sopenharmony_ci * dentry names.
14662306a36Sopenharmony_ci */
14762306a36Sopenharmony_cistatic int build_path_from_dentry(struct v9fs_session_info *v9ses,
14862306a36Sopenharmony_ci				  struct dentry *dentry, const unsigned char ***names)
14962306a36Sopenharmony_ci{
15062306a36Sopenharmony_ci	int n = 0, i;
15162306a36Sopenharmony_ci	const unsigned char **wnames;
15262306a36Sopenharmony_ci	struct dentry *ds;
15362306a36Sopenharmony_ci
15462306a36Sopenharmony_ci	for (ds = dentry; !IS_ROOT(ds); ds = ds->d_parent)
15562306a36Sopenharmony_ci		n++;
15662306a36Sopenharmony_ci
15762306a36Sopenharmony_ci	wnames = kmalloc_array(n, sizeof(char *), GFP_KERNEL);
15862306a36Sopenharmony_ci	if (!wnames)
15962306a36Sopenharmony_ci		goto err_out;
16062306a36Sopenharmony_ci
16162306a36Sopenharmony_ci	for (ds = dentry, i = (n-1); i >= 0; i--, ds = ds->d_parent)
16262306a36Sopenharmony_ci		wnames[i] = ds->d_name.name;
16362306a36Sopenharmony_ci
16462306a36Sopenharmony_ci	*names = wnames;
16562306a36Sopenharmony_ci	return n;
16662306a36Sopenharmony_cierr_out:
16762306a36Sopenharmony_ci	return -ENOMEM;
16862306a36Sopenharmony_ci}
16962306a36Sopenharmony_ci
17062306a36Sopenharmony_cistatic struct p9_fid *v9fs_fid_lookup_with_uid(struct dentry *dentry,
17162306a36Sopenharmony_ci					       kuid_t uid, int any)
17262306a36Sopenharmony_ci{
17362306a36Sopenharmony_ci	struct dentry *ds;
17462306a36Sopenharmony_ci	const unsigned char **wnames, *uname;
17562306a36Sopenharmony_ci	int i, n, l, access;
17662306a36Sopenharmony_ci	struct v9fs_session_info *v9ses;
17762306a36Sopenharmony_ci	struct p9_fid *fid, *root_fid, *old_fid;
17862306a36Sopenharmony_ci
17962306a36Sopenharmony_ci	v9ses = v9fs_dentry2v9ses(dentry);
18062306a36Sopenharmony_ci	access = v9ses->flags & V9FS_ACCESS_MASK;
18162306a36Sopenharmony_ci	fid = v9fs_fid_find(dentry, uid, any);
18262306a36Sopenharmony_ci	if (fid)
18362306a36Sopenharmony_ci		return fid;
18462306a36Sopenharmony_ci	/*
18562306a36Sopenharmony_ci	 * we don't have a matching fid. To do a TWALK we need
18662306a36Sopenharmony_ci	 * parent fid. We need to prevent rename when we want to
18762306a36Sopenharmony_ci	 * look at the parent.
18862306a36Sopenharmony_ci	 */
18962306a36Sopenharmony_ci	down_read(&v9ses->rename_sem);
19062306a36Sopenharmony_ci	ds = dentry->d_parent;
19162306a36Sopenharmony_ci	fid = v9fs_fid_find(ds, uid, any);
19262306a36Sopenharmony_ci	if (fid) {
19362306a36Sopenharmony_ci		/* Found the parent fid do a lookup with that */
19462306a36Sopenharmony_ci		old_fid = fid;
19562306a36Sopenharmony_ci
19662306a36Sopenharmony_ci		fid = p9_client_walk(old_fid, 1, &dentry->d_name.name, 1);
19762306a36Sopenharmony_ci		p9_fid_put(old_fid);
19862306a36Sopenharmony_ci		goto fid_out;
19962306a36Sopenharmony_ci	}
20062306a36Sopenharmony_ci	up_read(&v9ses->rename_sem);
20162306a36Sopenharmony_ci
20262306a36Sopenharmony_ci	/* start from the root and try to do a lookup */
20362306a36Sopenharmony_ci	root_fid = v9fs_fid_find(dentry->d_sb->s_root, uid, any);
20462306a36Sopenharmony_ci	if (!root_fid) {
20562306a36Sopenharmony_ci		/* the user is not attached to the fs yet */
20662306a36Sopenharmony_ci		if (access == V9FS_ACCESS_SINGLE)
20762306a36Sopenharmony_ci			return ERR_PTR(-EPERM);
20862306a36Sopenharmony_ci
20962306a36Sopenharmony_ci		if (v9fs_proto_dotu(v9ses) || v9fs_proto_dotl(v9ses))
21062306a36Sopenharmony_ci			uname = NULL;
21162306a36Sopenharmony_ci		else
21262306a36Sopenharmony_ci			uname = v9ses->uname;
21362306a36Sopenharmony_ci
21462306a36Sopenharmony_ci		fid = p9_client_attach(v9ses->clnt, NULL, uname, uid,
21562306a36Sopenharmony_ci				       v9ses->aname);
21662306a36Sopenharmony_ci		if (IS_ERR(fid))
21762306a36Sopenharmony_ci			return fid;
21862306a36Sopenharmony_ci
21962306a36Sopenharmony_ci		root_fid = p9_fid_get(fid);
22062306a36Sopenharmony_ci		v9fs_fid_add(dentry->d_sb->s_root, &fid);
22162306a36Sopenharmony_ci	}
22262306a36Sopenharmony_ci	/* If we are root ourself just return that */
22362306a36Sopenharmony_ci	if (dentry->d_sb->s_root == dentry)
22462306a36Sopenharmony_ci		return root_fid;
22562306a36Sopenharmony_ci
22662306a36Sopenharmony_ci	/*
22762306a36Sopenharmony_ci	 * Do a multipath walk with attached root.
22862306a36Sopenharmony_ci	 * When walking parent we need to make sure we
22962306a36Sopenharmony_ci	 * don't have a parallel rename happening
23062306a36Sopenharmony_ci	 */
23162306a36Sopenharmony_ci	down_read(&v9ses->rename_sem);
23262306a36Sopenharmony_ci	n  = build_path_from_dentry(v9ses, dentry, &wnames);
23362306a36Sopenharmony_ci	if (n < 0) {
23462306a36Sopenharmony_ci		fid = ERR_PTR(n);
23562306a36Sopenharmony_ci		goto err_out;
23662306a36Sopenharmony_ci	}
23762306a36Sopenharmony_ci	fid = root_fid;
23862306a36Sopenharmony_ci	old_fid = root_fid;
23962306a36Sopenharmony_ci	i = 0;
24062306a36Sopenharmony_ci	while (i < n) {
24162306a36Sopenharmony_ci		l = min(n - i, P9_MAXWELEM);
24262306a36Sopenharmony_ci		/*
24362306a36Sopenharmony_ci		 * We need to hold rename lock when doing a multipath
24462306a36Sopenharmony_ci		 * walk to ensure none of the path components change
24562306a36Sopenharmony_ci		 */
24662306a36Sopenharmony_ci		fid = p9_client_walk(old_fid, l, &wnames[i],
24762306a36Sopenharmony_ci				     old_fid == root_fid /* clone */);
24862306a36Sopenharmony_ci		/* non-cloning walk will return the same fid */
24962306a36Sopenharmony_ci		if (fid != old_fid) {
25062306a36Sopenharmony_ci			p9_fid_put(old_fid);
25162306a36Sopenharmony_ci			old_fid = fid;
25262306a36Sopenharmony_ci		}
25362306a36Sopenharmony_ci		if (IS_ERR(fid)) {
25462306a36Sopenharmony_ci			kfree(wnames);
25562306a36Sopenharmony_ci			goto err_out;
25662306a36Sopenharmony_ci		}
25762306a36Sopenharmony_ci		i += l;
25862306a36Sopenharmony_ci	}
25962306a36Sopenharmony_ci	kfree(wnames);
26062306a36Sopenharmony_cifid_out:
26162306a36Sopenharmony_ci	if (!IS_ERR(fid)) {
26262306a36Sopenharmony_ci		spin_lock(&dentry->d_lock);
26362306a36Sopenharmony_ci		if (d_unhashed(dentry)) {
26462306a36Sopenharmony_ci			spin_unlock(&dentry->d_lock);
26562306a36Sopenharmony_ci			p9_fid_put(fid);
26662306a36Sopenharmony_ci			fid = ERR_PTR(-ENOENT);
26762306a36Sopenharmony_ci		} else {
26862306a36Sopenharmony_ci			__add_fid(dentry, fid);
26962306a36Sopenharmony_ci			p9_fid_get(fid);
27062306a36Sopenharmony_ci			spin_unlock(&dentry->d_lock);
27162306a36Sopenharmony_ci		}
27262306a36Sopenharmony_ci	}
27362306a36Sopenharmony_cierr_out:
27462306a36Sopenharmony_ci	up_read(&v9ses->rename_sem);
27562306a36Sopenharmony_ci	return fid;
27662306a36Sopenharmony_ci}
27762306a36Sopenharmony_ci
27862306a36Sopenharmony_ci/**
27962306a36Sopenharmony_ci * v9fs_fid_lookup - lookup for a fid, try to walk if not found
28062306a36Sopenharmony_ci * @dentry: dentry to look for fid in
28162306a36Sopenharmony_ci *
28262306a36Sopenharmony_ci * Look for a fid in the specified dentry for the current user.
28362306a36Sopenharmony_ci * If no fid is found, try to create one walking from a fid from the parent
28462306a36Sopenharmony_ci * dentry (if it has one), or the root dentry. If the user haven't accessed
28562306a36Sopenharmony_ci * the fs yet, attach now and walk from the root.
28662306a36Sopenharmony_ci */
28762306a36Sopenharmony_ci
28862306a36Sopenharmony_cistruct p9_fid *v9fs_fid_lookup(struct dentry *dentry)
28962306a36Sopenharmony_ci{
29062306a36Sopenharmony_ci	kuid_t uid;
29162306a36Sopenharmony_ci	int  any, access;
29262306a36Sopenharmony_ci	struct v9fs_session_info *v9ses;
29362306a36Sopenharmony_ci
29462306a36Sopenharmony_ci	v9ses = v9fs_dentry2v9ses(dentry);
29562306a36Sopenharmony_ci	access = v9ses->flags & V9FS_ACCESS_MASK;
29662306a36Sopenharmony_ci	switch (access) {
29762306a36Sopenharmony_ci	case V9FS_ACCESS_SINGLE:
29862306a36Sopenharmony_ci	case V9FS_ACCESS_USER:
29962306a36Sopenharmony_ci	case V9FS_ACCESS_CLIENT:
30062306a36Sopenharmony_ci		uid = current_fsuid();
30162306a36Sopenharmony_ci		any = 0;
30262306a36Sopenharmony_ci		break;
30362306a36Sopenharmony_ci
30462306a36Sopenharmony_ci	case V9FS_ACCESS_ANY:
30562306a36Sopenharmony_ci		uid = v9ses->uid;
30662306a36Sopenharmony_ci		any = 1;
30762306a36Sopenharmony_ci		break;
30862306a36Sopenharmony_ci
30962306a36Sopenharmony_ci	default:
31062306a36Sopenharmony_ci		uid = INVALID_UID;
31162306a36Sopenharmony_ci		any = 0;
31262306a36Sopenharmony_ci		break;
31362306a36Sopenharmony_ci	}
31462306a36Sopenharmony_ci	return v9fs_fid_lookup_with_uid(dentry, uid, any);
31562306a36Sopenharmony_ci}
31662306a36Sopenharmony_ci
317