// SPDX-License-Identifier: GPL-2.0
/*
 * Copyright (C) 2020-2024 Microsoft Corporation. All rights reserved.
 */

#include <linux/slab.h>
#include <linux/types.h>
#include <linux/list.h>
#include <kunit/test.h>
#include "policy.h"
struct policy_case {
        const char *const policy;
        int errno;
        const char *const desc;
};

static const struct policy_case policy_cases[] = {
        {
                "policy_name=allowall policy_version=0.0.0\n"
                "DEFAULT action=ALLOW",
                0,
                "basic",
        },
        {
                "policy_name=trailing_comment policy_version=152.0.0 #This is comment\n"
                "DEFAULT action=ALLOW",
                0,
                "trailing comment",
        },
        {
                "policy_name=allowallnewline policy_version=0.2.0\n"
                "DEFAULT action=ALLOW\n"
                "\n",
                0,
                "trailing newline",
        },
        {
                "policy_name=carriagereturnlinefeed policy_version=0.0.1\n"
                "DEFAULT action=ALLOW\n"
                "\r\n",
                0,
                "clrf newline",
        },
        {
                "policy_name=whitespace policy_version=0.0.0\n"
                "DEFAULT\taction=ALLOW\n"
                "     \t     DEFAULT \t    op=EXECUTE      action=DENY\n"
                "op=EXECUTE boot_verified=TRUE action=ALLOW\n"
                "# this is a\tcomment\t\t\t\t\n"
                "DEFAULT \t op=KMODULE\t\t\t  action=DENY\r\n"
                "op=KMODULE boot_verified=TRUE action=ALLOW\n",
                0,
                "various whitespaces and nested default",
        },
        {
                "policy_name=boot_verified policy_version=-1236.0.0\n"
                "DEFAULT\taction=ALLOW\n",
                -EINVAL,
                "negative version",
        },
        {
                "policy_name=$@!*&^%%\\:;{}() policy_version=0.0.0\n"
                "DEFAULT action=ALLOW",
                0,
                "special characters",
        },
        {
                "policy_name=test policy_version=999999.0.0\n"
                "DEFAULT action=ALLOW",
                -ERANGE,
                "overflow version",
        },
        {
                "policy_name=test policy_version=255.0\n"
                "DEFAULT action=ALLOW",
                -EBADMSG,
                "incomplete version",
        },
        {
                "policy_name=test policy_version=111.0.0.0\n"
                "DEFAULT action=ALLOW",
                -EBADMSG,
                "extra version",
        },
        {
                "",
                -EBADMSG,
                "0-length policy",
        },
        {
                "policy_name=test\0policy_version=0.0.0\n"
                "DEFAULT action=ALLOW",
                -EBADMSG,
                "random null in header",
        },
        {
                "policy_name=test policy_version=0.0.0\n"
                "\0DEFAULT action=ALLOW",
                -EBADMSG,
                "incomplete policy from NULL",
        },
        {
                "policy_name=test policy_version=0.0.0\n"
                "DEFAULT action=DENY\n\0"
                "op=EXECUTE dmverity_signature=TRUE action=ALLOW\n",
                0,
                "NULL truncates policy",
        },
        {
                "policy_name=test policy_version=0.0.0\n"
                "DEFAULT action=ALLOW\n"
                "op=EXECUTE dmverity_signature=abc action=ALLOW",
                -EBADMSG,
                "invalid property type",
        },
        {
                "DEFAULT action=ALLOW",
                -EBADMSG,
                "missing policy header",
        },
        {
                "policy_name=test policy_version=0.0.0\n",
                -EBADMSG,
                "missing default definition",
        },
        {
                "policy_name=test policy_version=0.0.0\n"
                "DEFAULT action=ALLOW\n"
                "dmverity_signature=TRUE op=EXECUTE action=ALLOW",
                -EBADMSG,
                "invalid rule ordering"
        },
        {
                "policy_name=test policy_version=0.0.0\n"
                "DEFAULT action=ALLOW\n"
                "action=ALLOW op=EXECUTE dmverity_signature=TRUE",
                -EBADMSG,
                "invalid rule ordering (2)",
        },
        {
                "policy_name=test policy_version=0.0\n"
                "DEFAULT action=ALLOW\n"
                "op=EXECUTE dmverity_signature=TRUE action=ALLOW",
                -EBADMSG,
                "invalid version",
        },
        {
                "policy_name=test policy_version=0.0.0\n"
                "DEFAULT action=ALLOW\n"
                "op=UNKNOWN dmverity_signature=TRUE action=ALLOW",
                -EBADMSG,
                "unknown operation",
        },
        {
                "policy_name=asdvpolicy_version=0.0.0\n"
                "DEFAULT action=ALLOW\n",
                -EBADMSG,
                "missing space after policy name",
        },
        {
                "policy_name=test\xFF\xEF policy_version=0.0.0\n"
                "DEFAULT action=ALLOW\n"
                "op=EXECUTE dmverity_signature=TRUE action=ALLOW",
                0,
                "expanded ascii",
        },
        {
                "policy_name=test\xFF\xEF policy_version=0.0.0\n"
                "DEFAULT action=ALLOW\n"
                "op=EXECUTE dmverity_roothash=GOOD_DOG action=ALLOW",
                -EBADMSG,
                "invalid property value (2)",
        },
        {
                "policy_name=test policy_version=0.0.0\n"
                "policy_name=test policy_version=0.1.0\n"
                "DEFAULT action=ALLOW",
                -EBADMSG,
                "double header"
        },
        {
                "policy_name=test policy_version=0.0.0\n"
                "DEFAULT action=ALLOW\n"
                "DEFAULT action=ALLOW\n",
                -EBADMSG,
                "double default"
        },
        {
                "policy_name=test policy_version=0.0.0\n"
                "DEFAULT action=ALLOW\n"
                "DEFAULT op=EXECUTE action=DENY\n"
                "DEFAULT op=EXECUTE action=ALLOW\n",
                -EBADMSG,
                "double operation default"
        },
        {
                "policy_name=test policy_version=0.0.0\n"
                "DEFAULT action=ALLOW\n"
                "DEFAULT op=EXECUTE action=DEN\n",
                -EBADMSG,
                "invalid action value"
        },
        {
                "policy_name=test policy_version=0.0.0\n"
                "DEFAULT action=ALLOW\n"
                "DEFAULT op=EXECUTE action\n",
                -EBADMSG,
                "invalid action value (2)"
        },
        {
                "policy_name=test policy_version=0.0.0\n"
                "DEFAULT action=ALLOW\n"
                "UNKNOWN value=true\n",
                -EBADMSG,
                "unrecognized statement"
        },
        {
                "policy_name=test policy_version=0.0.0\n"
                "DEFAULT action=ALLOW\n"
                "op=EXECUTE dmverity_roothash=1c0d7ee1f8343b7fbe418378e8eb22c061d7dec7 action=DENY\n",
                -EBADMSG,
                "old-style digest"
        },
        {
                "policy_name=test policy_version=0.0.0\n"
                "DEFAULT action=ALLOW\n"
                "op=EXECUTE fsverity_digest=1c0d7ee1f8343b7fbe418378e8eb22c061d7dec7 action=DENY\n",
                -EBADMSG,
                "old-style digest"
        }
};

static void pol_to_desc(const struct policy_case *c, char *desc)
{
        strscpy(desc, c->desc, KUNIT_PARAM_DESC_SIZE);
}

KUNIT_ARRAY_PARAM(ipe_policies, policy_cases, pol_to_desc);

/**
 * ipe_parser_unsigned_test - Test the parser by passing unsigned policies.
 * @test: Supplies a pointer to a kunit structure.
 *
 * This is called by the kunit harness. This test does not check the correctness
 * of the policy, but ensures that errors are handled correctly.
 */
static void ipe_parser_unsigned_test(struct kunit *test)
{
        const struct policy_case *p = test->param_value;
        struct ipe_policy *pol;

        pol = ipe_new_policy(p->policy, strlen(p->policy), NULL, 0);

        if (p->errno) {
                KUNIT_EXPECT_EQ(test, PTR_ERR(pol), p->errno);
                return;
        }

        KUNIT_ASSERT_NOT_ERR_OR_NULL(test, pol);
        KUNIT_EXPECT_NOT_ERR_OR_NULL(test, pol->parsed);
        KUNIT_EXPECT_STREQ(test, pol->text, p->policy);
        KUNIT_EXPECT_PTR_EQ(test, NULL, pol->pkcs7);
        KUNIT_EXPECT_EQ(test, 0, pol->pkcs7len);

        ipe_free_policy(pol);
}

/**
 * ipe_parser_widestring_test - Ensure parser fail on a wide string policy.
 * @test: Supplies a pointer to a kunit structure.
 *
 * This is called by the kunit harness.
 */
static void ipe_parser_widestring_test(struct kunit *test)
{
        const unsigned short policy[] = L"policy_name=Test policy_version=0.0.0\n"
                                        L"DEFAULT action=ALLOW";
        struct ipe_policy *pol = NULL;

        pol = ipe_new_policy((const char *)policy, (ARRAY_SIZE(policy) - 1) * 2, NULL, 0);
        KUNIT_EXPECT_TRUE(test, IS_ERR_OR_NULL(pol));

        ipe_free_policy(pol);
}

static struct kunit_case ipe_parser_test_cases[] = {
        KUNIT_CASE_PARAM(ipe_parser_unsigned_test, ipe_policies_gen_params),
        KUNIT_CASE(ipe_parser_widestring_test),
        { }
};

static struct kunit_suite ipe_parser_test_suite = {
        .name = "ipe-parser",
        .test_cases = ipe_parser_test_cases,
};

kunit_test_suite(ipe_parser_test_suite);