// SPDX-License-Identifier: GPL-2.0-or-later
/*
 * KUnit tests for commoncap.c security functions
 *
 * Tests for security-critical functions in the capability subsystem,
 * particularly namespace-related capability checks.
 */

#include <kunit/test.h>
#include <linux/user_namespace.h>
#include <linux/uidgid.h>
#include <linux/cred.h>
#include <linux/mnt_idmapping.h>
#include <linux/module.h>
#include <linux/slab.h>
#include <linux/refcount.h>

#ifdef CONFIG_SECURITY_COMMONCAP_KUNIT_TEST

/* Functions are static in commoncap.c, but we can call them since we're
 * included in the same compilation unit when tests are enabled.
 */

/**
 * test_vfsuid_root_in_currentns_init_ns - Test vfsuid_root_in_currentns with init ns
 *
 * Verifies that UID 0 in the init namespace correctly owns the current
 * namespace when running in init_user_ns.
 *
 * @test: KUnit test context
 */
static void test_vfsuid_root_in_currentns_init_ns(struct kunit *test)
{
	vfsuid_t vfsuid;
	kuid_t kuid;

	/* Create UID 0 in init namespace */
	kuid = KUIDT_INIT(0);
	vfsuid = VFSUIDT_INIT(kuid);

	/* In init namespace, UID 0 should own current namespace */
	KUNIT_EXPECT_TRUE(test, vfsuid_root_in_currentns(vfsuid));
}

/**
 * test_vfsuid_root_in_currentns_invalid - Test vfsuid_root_in_currentns with invalid vfsuid
 *
 * Verifies that an invalid vfsuid correctly returns false.
 *
 * @test: KUnit test context
 */
static void test_vfsuid_root_in_currentns_invalid(struct kunit *test)
{
	vfsuid_t invalid_vfsuid;

	/* Use the predefined invalid vfsuid */
	invalid_vfsuid = INVALID_VFSUID;

	/* Invalid vfsuid should return false */
	KUNIT_EXPECT_FALSE(test, vfsuid_root_in_currentns(invalid_vfsuid));
}

/**
 * test_vfsuid_root_in_currentns_nonzero - Test vfsuid_root_in_currentns with non-zero UID
 *
 * Verifies that a non-zero UID correctly returns false.
 *
 * @test: KUnit test context
 */
static void test_vfsuid_root_in_currentns_nonzero(struct kunit *test)
{
	vfsuid_t vfsuid;
	kuid_t kuid;

	/* Create a non-zero UID */
	kuid = KUIDT_INIT(1000);
	vfsuid = VFSUIDT_INIT(kuid);

	/* Non-zero UID should return false */
	KUNIT_EXPECT_FALSE(test, vfsuid_root_in_currentns(vfsuid));
}

/**
 * test_kuid_root_in_ns_init_ns_uid0 - Test kuid_root_in_ns with init namespace and UID 0
 *
 * Verifies that kuid_root_in_ns correctly identifies UID 0 in init namespace.
 * This tests the core namespace traversal logic. In init namespace, UID 0
 * maps to itself, so it should own the namespace.
 *
 * @test: KUnit test context
 */
static void test_kuid_root_in_ns_init_ns_uid0(struct kunit *test)
{
	kuid_t kuid;
	struct user_namespace *init_ns;

	kuid = KUIDT_INIT(0);
	init_ns = &init_user_ns;

	/* UID 0 should own init namespace */
	KUNIT_EXPECT_TRUE(test, kuid_root_in_ns(kuid, init_ns));
}

/**
 * test_kuid_root_in_ns_init_ns_nonzero - Test kuid_root_in_ns with init namespace and non-zero UID
 *
 * Verifies that kuid_root_in_ns correctly rejects non-zero UIDs in init namespace.
 * Only UID 0 should own a namespace.
 *
 * @test: KUnit test context
 */
static void test_kuid_root_in_ns_init_ns_nonzero(struct kunit *test)
{
	kuid_t kuid;
	struct user_namespace *init_ns;

	kuid = KUIDT_INIT(1000);
	init_ns = &init_user_ns;

	/* Non-zero UID should not own namespace */
	KUNIT_EXPECT_FALSE(test, kuid_root_in_ns(kuid, init_ns));
}

/**
 * create_test_user_ns_with_mapping - Create a mock user namespace with UID mapping
 *
 * Creates a minimal user namespace structure for testing where uid 0 in the
 * namespace maps to a specific kuid in the parent namespace.
 *
 * @test: KUnit test context
 * @parent_ns: Parent namespace (typically init_user_ns)
 * @mapped_kuid: The kuid that uid 0 in this namespace maps to in parent
 *
 * Returns: Pointer to allocated namespace, or NULL on failure
 */
static struct user_namespace *create_test_user_ns_with_mapping(struct kunit *test,
								 struct user_namespace *parent_ns,
								 kuid_t mapped_kuid)
{
	struct user_namespace *ns;
	struct uid_gid_extent extent;

	/* Allocate a test namespace - use kzalloc to zero all fields */
	ns = kunit_kzalloc(test, sizeof(*ns), GFP_KERNEL);
	if (!ns)
		return NULL;

	/* Initialize basic namespace structure fields */
	ns->parent = parent_ns;
	ns->level = parent_ns ? parent_ns->level + 1 : 0;
	ns->owner = mapped_kuid;
	ns->group = KGIDT_INIT(0);

	/* Initialize ns_common structure */
	refcount_set(&ns->ns.__ns_ref, 1);
	ns->ns.inum = 0; /* Mock inum */

	/* Set up uid mapping: uid 0 in this namespace maps to mapped_kuid in parent
	 * Format: first (uid in ns) : lower_first (kuid in parent) : count
	 * So: uid 0 in ns -> kuid mapped_kuid in parent
	 * This means from_kuid(ns, mapped_kuid) returns 0
	 */
	extent.first = 0;                              /* uid 0 in this namespace */
	extent.lower_first = __kuid_val(mapped_kuid);  /* maps to this kuid in parent */
	extent.count = 1;

	ns->uid_map.extent[0] = extent;
	ns->uid_map.nr_extents = 1;

	/* Set up gid mapping: gid 0 maps to gid 0 in parent (simplified) */
	extent.first = 0;
	extent.lower_first = 0;
	extent.count = 1;

	ns->gid_map.extent[0] = extent;
	ns->gid_map.nr_extents = 1;

	return ns;
}

/**
 * test_kuid_root_in_ns_with_mapping - Test kuid_root_in_ns with namespace where uid 0
 *				       maps to different kuid
 *
 * Creates a user namespace where uid 0 maps to kuid 1000 in the parent namespace.
 * Verifies that kuid_root_in_ns correctly identifies kuid 1000 as owning the namespace.
 *
 * Note: kuid_root_in_ns walks up the namespace hierarchy, so it checks the current
 * namespace first, then parent, then parent's parent, etc. So:
 * - kuid 1000 owns test_ns because from_kuid(test_ns, 1000) == 0
 * - kuid 0 also owns test_ns because from_kuid(init_user_ns, 0) == 0
 *   (checked in parent)
 *
 * This tests the actual functionality as requested: creating namespaces with
 * different values for the namespace's uid 0.
 *
 * @test: KUnit test context
 */
static void test_kuid_root_in_ns_with_mapping(struct kunit *test)
{
	struct user_namespace *test_ns;
	struct user_namespace *parent_ns;
	kuid_t mapped_kuid, other_kuid;

	parent_ns = &init_user_ns;
	mapped_kuid = KUIDT_INIT(1000);
	other_kuid = KUIDT_INIT(2000);

	test_ns = create_test_user_ns_with_mapping(test, parent_ns, mapped_kuid);
	KUNIT_ASSERT_NOT_ERR_OR_NULL(test, test_ns);

	/* kuid 1000 should own test_ns because it maps to uid 0 in test_ns */
	KUNIT_EXPECT_TRUE(test, kuid_root_in_ns(mapped_kuid, test_ns));

	/* kuid 0 should also own test_ns (checked via parent init_user_ns) */
	KUNIT_EXPECT_TRUE(test, kuid_root_in_ns(KUIDT_INIT(0), test_ns));

	/* Other kuids should not own test_ns */
	KUNIT_EXPECT_FALSE(test, kuid_root_in_ns(other_kuid, test_ns));
	KUNIT_EXPECT_FALSE(test, kuid_root_in_ns(KUIDT_INIT(500), test_ns));
}

/**
 * test_kuid_root_in_ns_with_different_mappings - Test with multiple namespaces
 *
 * Creates multiple user namespaces with different UID mappings to verify
 * that kuid_root_in_ns correctly distinguishes between namespaces.
 *
 * Each namespace maps uid 0 to a different kuid, and we verify that each
 * kuid only owns its corresponding namespace (plus kuid 0 owns all via
 * init_user_ns parent).
 *
 * @test: KUnit test context
 */
static void test_kuid_root_in_ns_with_different_mappings(struct kunit *test)
{
	struct user_namespace *ns1, *ns2, *ns3;

	/* Create three independent namespaces, each mapping uid 0 to different kuids */
	ns1 = create_test_user_ns_with_mapping(test, &init_user_ns, KUIDT_INIT(1000));
	KUNIT_ASSERT_NOT_ERR_OR_NULL(test, ns1);

	ns2 = create_test_user_ns_with_mapping(test, &init_user_ns, KUIDT_INIT(2000));
	KUNIT_ASSERT_NOT_ERR_OR_NULL(test, ns2);

	ns3 = create_test_user_ns_with_mapping(test, &init_user_ns, KUIDT_INIT(3000));
	KUNIT_ASSERT_NOT_ERR_OR_NULL(test, ns3);

	/* Test ns1: kuid 1000 owns it, kuid 0 owns it (via parent), others do not */
	KUNIT_EXPECT_TRUE(test, kuid_root_in_ns(KUIDT_INIT(1000), ns1));
	KUNIT_EXPECT_TRUE(test, kuid_root_in_ns(KUIDT_INIT(0), ns1));
	KUNIT_EXPECT_FALSE(test, kuid_root_in_ns(KUIDT_INIT(2000), ns1));
	KUNIT_EXPECT_FALSE(test, kuid_root_in_ns(KUIDT_INIT(3000), ns1));

	/* Test ns2: kuid 2000 owns it, kuid 0 owns it (via parent), others do not */
	KUNIT_EXPECT_TRUE(test, kuid_root_in_ns(KUIDT_INIT(2000), ns2));
	KUNIT_EXPECT_TRUE(test, kuid_root_in_ns(KUIDT_INIT(0), ns2));
	KUNIT_EXPECT_FALSE(test, kuid_root_in_ns(KUIDT_INIT(1000), ns2));
	KUNIT_EXPECT_FALSE(test, kuid_root_in_ns(KUIDT_INIT(3000), ns2));

	/* Test ns3: kuid 3000 owns it, kuid 0 owns it (via parent), others do not */
	KUNIT_EXPECT_TRUE(test, kuid_root_in_ns(KUIDT_INIT(3000), ns3));
	KUNIT_EXPECT_TRUE(test, kuid_root_in_ns(KUIDT_INIT(0), ns3));
	KUNIT_EXPECT_FALSE(test, kuid_root_in_ns(KUIDT_INIT(1000), ns3));
	KUNIT_EXPECT_FALSE(test, kuid_root_in_ns(KUIDT_INIT(2000), ns3));
}

static struct kunit_case commoncap_test_cases[] = {
	KUNIT_CASE(test_vfsuid_root_in_currentns_init_ns),
	KUNIT_CASE(test_vfsuid_root_in_currentns_invalid),
	KUNIT_CASE(test_vfsuid_root_in_currentns_nonzero),
	KUNIT_CASE(test_kuid_root_in_ns_init_ns_uid0),
	KUNIT_CASE(test_kuid_root_in_ns_init_ns_nonzero),
	KUNIT_CASE(test_kuid_root_in_ns_with_mapping),
	KUNIT_CASE(test_kuid_root_in_ns_with_different_mappings),
	{}
};

static struct kunit_suite commoncap_test_suite = {
	.name = "commoncap",
	.test_cases = commoncap_test_cases,
};

kunit_test_suite(commoncap_test_suite);

MODULE_LICENSE("GPL");

#endif /* CONFIG_SECURITY_COMMONCAP_KUNIT_TEST */