// SPDX-License-Identifier: GPL-2.0+
#include <errno.h>
#include <fcntl.h>
#include <stdio.h>
#include <string.h>
#include <unistd.h>
#include <sys/socket.h>
#include <sys/stat.h>
#include <sys/sysmacros.h>
#include <sys/types.h>

#include "kselftest_harness.h"

/* Remove a file, ignoring the result if it didn't exist. */
void rm(struct __test_metadata *_metadata, const char *pathname,
        int is_dir)
{
        int rc;

        if (is_dir)
                rc = rmdir(pathname);
        else
                rc = unlink(pathname);

        if (rc < 0) {
                ASSERT_EQ(errno, ENOENT) {
                        TH_LOG("Not ENOENT: %s", pathname);
                }
        } else {
                ASSERT_EQ(rc, 0) {
                        TH_LOG("Failed to remove: %s", pathname);
                }
        }
}

FIXTURE(file) {
        char *pathname;
        int is_dir;
};

FIXTURE_VARIANT(file)
{
        const char *name;
        int expected;
        int is_dir;
        void (*setup)(struct __test_metadata *_metadata,
                      FIXTURE_DATA(file) *self,
                      const FIXTURE_VARIANT(file) *variant);
        int major, minor, mode; /* for mknod() */
};

void setup_link(struct __test_metadata *_metadata,
                FIXTURE_DATA(file) *self,
                const FIXTURE_VARIANT(file) *variant)
{
        const char * const paths[] = {
                "/bin/true",
                "/usr/bin/true",
        };
        int i;

        for (i = 0; i < ARRAY_SIZE(paths); i++) {
                if (access(paths[i], X_OK) == 0) {
                        ASSERT_EQ(symlink(paths[i], self->pathname), 0);
                        return;
                }
        }
        ASSERT_EQ(1, 0) {
                TH_LOG("Could not find viable 'true' binary");
        }
}

FIXTURE_VARIANT_ADD(file, S_IFLNK)
{
        .name = "S_IFLNK",
        .expected = ELOOP,
        .setup = setup_link,
};

void setup_dir(struct __test_metadata *_metadata,
               FIXTURE_DATA(file) *self,
               const FIXTURE_VARIANT(file) *variant)
{
        ASSERT_EQ(mkdir(self->pathname, 0755), 0);
}

FIXTURE_VARIANT_ADD(file, S_IFDIR)
{
        .name = "S_IFDIR",
        .is_dir = 1,
        .expected = EACCES,
        .setup = setup_dir,
};

void setup_node(struct __test_metadata *_metadata,
                FIXTURE_DATA(file) *self,
                const FIXTURE_VARIANT(file) *variant)
{
        dev_t dev;
        int rc;

        dev = makedev(variant->major, variant->minor);
        rc = mknod(self->pathname, 0755 | variant->mode, dev);
        ASSERT_EQ(rc, 0) {
                if (errno == EPERM)
                        SKIP(return, "Please run as root; cannot mknod(%s)",
                                variant->name);
        }
}

FIXTURE_VARIANT_ADD(file, S_IFBLK)
{
        .name = "S_IFBLK",
        .expected = EACCES,
        .setup = setup_node,
        /* /dev/loop0 */
        .major = 7,
        .minor = 0,
        .mode = S_IFBLK,
};

FIXTURE_VARIANT_ADD(file, S_IFCHR)
{
        .name = "S_IFCHR",
        .expected = EACCES,
        .setup = setup_node,
        /* /dev/zero */
        .major = 1,
        .minor = 5,
        .mode = S_IFCHR,
};

void setup_fifo(struct __test_metadata *_metadata,
                FIXTURE_DATA(file) *self,
                const FIXTURE_VARIANT(file) *variant)
{
        ASSERT_EQ(mkfifo(self->pathname, 0755), 0);
}

FIXTURE_VARIANT_ADD(file, S_IFIFO)
{
        .name = "S_IFIFO",
        .expected = EACCES,
        .setup = setup_fifo,
};

FIXTURE_SETUP(file)
{
        ASSERT_GT(asprintf(&self->pathname, "%s.test", variant->name), 6);
        self->is_dir = variant->is_dir;

        rm(_metadata, self->pathname, variant->is_dir);
        variant->setup(_metadata, self, variant);
}

FIXTURE_TEARDOWN(file)
{
        rm(_metadata, self->pathname, self->is_dir);
}

TEST_F(file, exec_errno)
{
        char * const argv[2] = { (char * const)self->pathname, NULL };

        EXPECT_LT(execv(argv[0], argv), 0);
        EXPECT_EQ(errno, variant->expected);
}

/* S_IFSOCK */
FIXTURE(sock)
{
        int fd;
};

FIXTURE_SETUP(sock)
{
        self->fd = socket(AF_INET, SOCK_STREAM, 0);
        ASSERT_GE(self->fd, 0);
}

FIXTURE_TEARDOWN(sock)
{
        if (self->fd >= 0)
                ASSERT_EQ(close(self->fd), 0);
}

TEST_F(sock, exec_errno)
{
        char * const argv[2] = { " magic socket ", NULL };
        char * const envp[1] = { NULL };

        EXPECT_LT(fexecve(self->fd, argv, envp), 0);
        EXPECT_EQ(errno, EACCES);
}

TEST_HARNESS_MAIN