// SPDX-License-Identifier: GPL-2.0
#define _GNU_SOURCE

#include <errno.h>
#include <fcntl.h>
#include <limits.h>
#include <sched.h>
#include <stdio.h>
#include <unistd.h>
#include <sys/fsuid.h>
#include <sys/stat.h>
#include <sys/syscall.h>

#include <linux/mount.h>
#include <linux/types.h>

#include "kselftest_harness.h"
#include "wrappers.h"
#include "utils.h"

/*
 * The test mount maps caller-visible ids [0, MAP_RANGE) onto the on-disk range
 * [MAP_HOST, MAP_HOST + MAP_RANGE).  An id outside [0, MAP_RANGE) therefore has
 * no mapping in the mount and is not representable in the filesystem.
 */
#define MAP_HOST  10000
#define MAP_RANGE 10000
#define UNMAPPED  50000

#ifndef MOUNT_ATTR_IDMAP
#define MOUNT_ATTR_IDMAP 0x00100000
#endif

#ifndef __NR_mount_setattr
#define __NR_mount_setattr 442
#endif

static inline int sys_mount_setattr(int dfd, const char *path,
                                    unsigned int flags,
                                    struct mount_attr *attr, size_t size)
{
        return syscall(__NR_mount_setattr, dfd, path, flags, attr, size);
}

/*
 * Clone @path into a detached mount idmapped so that caller-visible ids
 * [0, MAP_RANGE) map onto the on-disk ids [MAP_HOST, MAP_HOST + MAP_RANGE).
 * Returns the mount fd, or -1 if idmapped mounts are not available.
 */
static int idmapped_clone(const char *path)
{
        struct mount_attr attr = {
                .attr_set = MOUNT_ATTR_IDMAP,
        };
        int fd_tree, userns_fd, ret;

        fd_tree = sys_open_tree(AT_FDCWD, path,
                                OPEN_TREE_CLONE | OPEN_TREE_CLOEXEC);
        if (fd_tree < 0)
                return -1;

        userns_fd = get_userns_fd(MAP_HOST, 0, MAP_RANGE);
        if (userns_fd < 0) {
                close(fd_tree);
                return -1;
        }

        attr.userns_fd = userns_fd;
        ret = sys_mount_setattr(fd_tree, "", AT_EMPTY_PATH, &attr, sizeof(attr));
        close(userns_fd);
        if (ret) {
                close(fd_tree);
                return -1;
        }

        return fd_tree;
}

FIXTURE(idmapped_tmpfile) {
        char dir[64];   /* non-idmapped path to the layer directory */
};

FIXTURE_SETUP(idmapped_tmpfile)
{
        /* Private mount namespace so test mounts need no cleanup. */
        ASSERT_EQ(unshare(CLONE_NEWNS), 0);
        ASSERT_EQ(sys_mount(NULL, "/", NULL, MS_SLAVE | MS_REC, NULL), 0);
        ASSERT_EQ(sys_mount("tmpfs", "/tmp", "tmpfs", 0, NULL), 0);

        snprintf(self->dir, sizeof(self->dir), "/tmp/d");
        ASSERT_EQ(mkdir(self->dir, 0777), 0);
        /* World-writable so an unmapped caller still passes permission(). */
        ASSERT_EQ(chmod(self->dir, 0777), 0);
}

FIXTURE_TEARDOWN(idmapped_tmpfile)
{
}

/*
 * A caller whose fsuid/fsgid have no mapping in the idmapped mount must not be
 * able to create an O_TMPFILE.  Without the check in vfs_tmpfile() the inode
 * would be created owned by (uid_t)-1 and could then be linked into the
 * namespace.
 */
TEST_F(idmapped_tmpfile, unmapped_caller_is_refused)
{
        int mfd, fd;

        mfd = idmapped_clone(self->dir);
        if (mfd < 0)
                SKIP(return, "idmapped mounts not supported");

        /* Become a caller outside the mount's [0, MAP_RANGE) range. */
        setfsgid(UNMAPPED);
        setfsuid(UNMAPPED);
        ASSERT_EQ(setfsuid(-1), UNMAPPED);

        fd = openat(mfd, ".", O_TMPFILE | O_WRONLY, 0644);
        ASSERT_LT(fd, 0);
        EXPECT_EQ(errno, EOVERFLOW);
        if (fd >= 0)
                close(fd);

        EXPECT_EQ(close(mfd), 0);
}

/*
 * A mapped caller can create an O_TMPFILE and link it into the namespace; the
 * ownership round-trips through the mount idmap.  This is what makes refusing
 * the unmapped case above necessary in the first place.
 */
TEST_F(idmapped_tmpfile, mapped_caller_creates_and_links)
{
        char path[PATH_MAX];
        struct stat st;
        int mfd, fd;

        mfd = idmapped_clone(self->dir);
        if (mfd < 0)
                SKIP(return, "idmapped mounts not supported");

        /* Caller is uid/gid 0, which maps to MAP_HOST through the mount. */
        fd = openat(mfd, ".", O_TMPFILE | O_RDWR, 0600);
        ASSERT_GE(fd, 0);

        ASSERT_EQ(fstat(fd, &st), 0);
        EXPECT_EQ(st.st_uid, 0);
        EXPECT_EQ(st.st_gid, 0);

        /* The tmpfile is linkable: splice it into the directory. */
        ASSERT_EQ(linkat(fd, "", mfd, "linked", AT_EMPTY_PATH), 0);
        EXPECT_EQ(close(fd), 0);

        ASSERT_EQ(fstatat(mfd, "linked", &st, 0), 0);
        EXPECT_EQ(st.st_uid, 0);
        EXPECT_EQ(st.st_gid, 0);

        /* On the underlying, non-idmapped tmpfs it is stored as MAP_HOST. */
        snprintf(path, sizeof(path), "%s/linked", self->dir);
        ASSERT_EQ(stat(path, &st), 0);
        EXPECT_EQ(st.st_uid, MAP_HOST);
        EXPECT_EQ(st.st_gid, MAP_HOST);

        EXPECT_EQ(close(mfd), 0);
}

TEST_HARNESS_MAIN