// SPDX-License-Identifier: GPL-2.0-or-later
/*
 * Author: Aleksa Sarai <cyphar@cyphar.com>
 * Copyright (C) 2025 SUSE LLC.
 */

#include <assert.h>
#include <errno.h>
#include <sched.h>
#include <stdio.h>
#include <stdlib.h>
#include <string.h>
#include <unistd.h>
#include <sys/mount.h>

#include "../kselftest_harness.h"

#define ASSERT_ERRNO(expected, _t, seen)				\
	__EXPECT(expected, #expected,					\
		({__typeof__(seen) _tmp_seen = (seen);			\
		  _tmp_seen >= 0 ? _tmp_seen : -errno; }), #seen, _t, 1)

#define ASSERT_ERRNO_EQ(expected, seen) \
	ASSERT_ERRNO(expected, ==, seen)

#define ASSERT_SUCCESS(seen) \
	ASSERT_ERRNO(0, <=, seen)

FIXTURE(ns)
{
	int host_mntns;
};

FIXTURE_SETUP(ns)
{
	/* Stash the old mntns. */
	self->host_mntns = open("/proc/self/ns/mnt", O_RDONLY|O_CLOEXEC);
	ASSERT_SUCCESS(self->host_mntns);

	/* Create a new mount namespace and make it private. */
	ASSERT_SUCCESS(unshare(CLONE_NEWNS));
	ASSERT_SUCCESS(mount(NULL, "/", NULL, MS_PRIVATE|MS_REC, NULL));
}

FIXTURE_TEARDOWN(ns)
{
	ASSERT_SUCCESS(setns(self->host_mntns, CLONE_NEWNS));
	ASSERT_SUCCESS(close(self->host_mntns));
}

TEST_F(ns, fscontext_log_enodata)
{
	int fsfd = fsopen("tmpfs", FSOPEN_CLOEXEC);
	ASSERT_SUCCESS(fsfd);

	/* A brand new fscontext has no log entries. */
	char buf[128] = {};
	for (int i = 0; i < 16; i++)
		ASSERT_ERRNO_EQ(-ENODATA, read(fsfd, buf, sizeof(buf)));

	ASSERT_SUCCESS(close(fsfd));
}

TEST_F(ns, fscontext_log_errorfc)
{
	int fsfd = fsopen("tmpfs", FSOPEN_CLOEXEC);
	ASSERT_SUCCESS(fsfd);

	ASSERT_ERRNO_EQ(-EINVAL, fsconfig(fsfd, FSCONFIG_SET_STRING, "invalid-arg", "123", 0));

	char buf[128] = {};
	ASSERT_SUCCESS(read(fsfd, buf, sizeof(buf)));
	EXPECT_STREQ("e tmpfs: Unknown parameter 'invalid-arg'\n", buf);

	/* The message has been consumed. */
	ASSERT_ERRNO_EQ(-ENODATA, read(fsfd, buf, sizeof(buf)));
	ASSERT_SUCCESS(close(fsfd));
}

TEST_F(ns, fscontext_log_errorfc_after_fsmount)
{
	int fsfd = fsopen("tmpfs", FSOPEN_CLOEXEC);
	ASSERT_SUCCESS(fsfd);

	ASSERT_ERRNO_EQ(-EINVAL, fsconfig(fsfd, FSCONFIG_SET_STRING, "invalid-arg", "123", 0));

	ASSERT_SUCCESS(fsconfig(fsfd, FSCONFIG_CMD_CREATE, NULL, NULL, 0));
	int mfd = fsmount(fsfd, FSMOUNT_CLOEXEC, MOUNT_ATTR_NOEXEC | MOUNT_ATTR_NOSUID);
	ASSERT_SUCCESS(mfd);
	ASSERT_SUCCESS(move_mount(mfd, "", AT_FDCWD, "/tmp", MOVE_MOUNT_F_EMPTY_PATH));

	/*
	 * The fscontext log should still contain data even after
	 * FSCONFIG_CMD_CREATE and fsmount().
	 */
	char buf[128] = {};
	ASSERT_SUCCESS(read(fsfd, buf, sizeof(buf)));
	EXPECT_STREQ("e tmpfs: Unknown parameter 'invalid-arg'\n", buf);

	/* The message has been consumed. */
	ASSERT_ERRNO_EQ(-ENODATA, read(fsfd, buf, sizeof(buf)));
	ASSERT_SUCCESS(close(fsfd));
}

TEST_F(ns, fscontext_log_emsgsize)
{
	int fsfd = fsopen("tmpfs", FSOPEN_CLOEXEC);
	ASSERT_SUCCESS(fsfd);

	ASSERT_ERRNO_EQ(-EINVAL, fsconfig(fsfd, FSCONFIG_SET_STRING, "invalid-arg", "123", 0));

	char buf[128] = {};
	/*
	 * Attempting to read a message with too small a buffer should not
	 * result in the message getting consumed.
	 */
	ASSERT_ERRNO_EQ(-EMSGSIZE, read(fsfd, buf, 0));
	ASSERT_ERRNO_EQ(-EMSGSIZE, read(fsfd, buf, 1));
	for (int i = 0; i < 16; i++)
		ASSERT_ERRNO_EQ(-EMSGSIZE, read(fsfd, buf, 16));

	ASSERT_SUCCESS(read(fsfd, buf, sizeof(buf)));
	EXPECT_STREQ("e tmpfs: Unknown parameter 'invalid-arg'\n", buf);

	/* The message has been consumed. */
	ASSERT_ERRNO_EQ(-ENODATA, read(fsfd, buf, sizeof(buf)));
	ASSERT_SUCCESS(close(fsfd));
}

TEST_HARNESS_MAIN