// SPDX-License-Identifier: GPL-2.0
/*
 * Copyright (C) 2023 Rivos Inc.
 */

#include <linux/module.h>
#include <linux/kernel.h>
#include <linux/init.h>
#include <kunit/test.h>

MODULE_LICENSE("GPL");
MODULE_DESCRIPTION("Test module linking");

extern int test_set32(void);
extern int test_set16(void);
extern int test_set8(void);
extern int test_set6(void);
extern long test_sub64(void);
extern int test_sub32(void);
extern int test_sub16(void);
extern int test_sub8(void);
extern int test_sub6(void);

#ifdef CONFIG_AS_HAS_ULEB128
extern int test_uleb_basic(void);
extern int test_uleb_large(void);
#endif

#define CHECK_EQ(lhs, rhs) KUNIT_ASSERT_EQ(test, lhs, rhs)

void run_test_set(struct kunit *test);
void run_test_sub(struct kunit *test);
void run_test_uleb(struct kunit *test);

void run_test_set(struct kunit *test)
{
	int val32 = test_set32();
	int val16 = test_set16();
	int val8 = test_set8();
	int val6 = test_set6();

	CHECK_EQ(val32, 0);
	CHECK_EQ(val16, 0);
	CHECK_EQ(val8, 0);
	CHECK_EQ(val6, 0);
}

void run_test_sub(struct kunit *test)
{
	int val64 = test_sub64();
	int val32 = test_sub32();
	int val16 = test_sub16();
	int val8 = test_sub8();
	int val6 = test_sub6();

	CHECK_EQ(val64, 0);
	CHECK_EQ(val32, 0);
	CHECK_EQ(val16, 0);
	CHECK_EQ(val8, 0);
	CHECK_EQ(val6, 0);
}

#ifdef CONFIG_AS_HAS_ULEB128
void run_test_uleb(struct kunit *test)
{
	int val_uleb = test_uleb_basic();
	int val_uleb2 = test_uleb_large();

	CHECK_EQ(val_uleb, 0);
	CHECK_EQ(val_uleb2, 0);
}
#endif

static struct kunit_case __refdata riscv_module_linking_test_cases[] = {
	KUNIT_CASE(run_test_set),
	KUNIT_CASE(run_test_sub),
#ifdef CONFIG_AS_HAS_ULEB128
	KUNIT_CASE(run_test_uleb),
#endif
	{}
};

static struct kunit_suite riscv_module_linking_test_suite = {
	.name = "riscv_checksum",
	.test_cases = riscv_module_linking_test_cases,
};

kunit_test_suites(&riscv_module_linking_test_suite);