#include <kernel/syscalls.h>
#include <kernel/alloc.h>
#include <kernel/vfs.h>
#include <kernel/auth.h>

#define FCHOWNAT_KNOWN_FLAGS (AT_EMPTY_PATH | AT_SYMLINK_NOFOLLOW)

syscallret_t syscall_fchownat(context_t *, int fd, char *upath, uid_t owner, gid_t group, int flags) {
	syscallret_t ret = {
		.ret = -1
	};

	vnode_t *node = NULL;
	vnode_t *dirnode = NULL;
	file_t *file = NULL;
	char *path = NULL;
	size_t pathlen;
	ret.errno = usercopy_strlen(upath, &pathlen);
	if (ret.errno)
		return ret;

	if (pathlen == 0 && (flags & AT_EMPTY_PATH)) {
		// chown is done on the file pointed to by fd
		file_t *file = fd_get(fd);
		if (file == NULL) {
			ret.errno = EBADF;
			goto cleanup;
		}

		node = file->vnode;
		VOP_HOLD(node);
		VOP_LOCK(node);
		fd_release(file);
	} else {
		// chown is done on the file pointed to by the path relative to fd
		path = alloc(pathlen + 1);
		if (path == NULL) {
			ret.errno = ENOMEM;
			goto cleanup;
		}

		ret.errno = usercopy_fromuser(path, upath, pathlen);
		if (ret.errno)
			goto cleanup;

		ret.errno = dirfd_enter(path, fd, &file, &dirnode);
		if (ret.errno)
			goto cleanup;

		ret.errno = vfs_lookup(&node, dirnode, path, NULL, (flags & AT_SYMLINK_NOFOLLOW) ? VFS_LOOKUP_NOLINK : 0);
		if (ret.errno)
			goto cleanup;
	}

	cred_t *cred = &current_thread()->proc->cred;
	ret.errno = auth_filesystem_check(cred, AUTH_ACTIONS_FILESYSTEM_SETATTR, node, NULL);
	if (ret.errno) {
		VOP_UNLOCK(node);
		goto cleanup;
	}

	if ((group != -1 && !CRED_IS_ESU(cred) && group != cred->gid && group != cred->egid && group != cred->sgid) || (owner != -1 && !CRED_IS_ESU(cred))) {
		ret.errno = EPERM;
		VOP_UNLOCK(node);
		goto cleanup;
	}

	vattr_t attr;
	ret.errno = VOP_GETATTR(node, &attr, cred);
	if (ret.errno) {
		VOP_UNLOCK(node);
		goto cleanup;
	}

	attr.uid = owner;
	attr.gid = group;

	int attrs = 0;
	if (owner != -1)
		attrs |= V_ATTR_UID;
	if (group != -1)
		attrs |= V_ATTR_GID;
	if (attrs) {
		attr.mode &= ~(V_ATTR_MODE_SGID | V_ATTR_MODE_SUID);
		attrs |= V_ATTR_MODE;
	}

	ret.errno = VOP_SETATTR(node, &attr, attrs, cred);
	ret.ret = ret.errno ? -1 : 0;

	VOP_UNLOCK(node);

	cleanup:
	if (node)
		VOP_RELEASE(node);

	if (dirnode)
		dirfd_leave(dirnode, file);

	if (path)
		free(path);

	return ret;
}