// SPDX-License-Identifier: GPL-2.0
/*
 * KUnit test of ext4 inode that verify the seconds part of [a/c/m]
 * timestamps in ext4 inode structs are decoded correctly.
 */

#include <kunit/test.h>
#include <linux/kernel.h>
#include <linux/time64.h>

#include "ext4.h"

/*
 * For constructing the nonnegative timestamp lower bound value.
 * binary: 00000000 00000000 00000000 00000000
 */
#define LOWER_MSB_0 0L
/*
 * For constructing the nonnegative timestamp upper bound value.
 * binary: 01111111 11111111 11111111 11111111
 *
 */
#define UPPER_MSB_0 0x7fffffffL
/*
 * For constructing the negative timestamp lower bound value.
 * binary: 10000000 00000000 00000000 00000000
 */
#define LOWER_MSB_1 (-(UPPER_MSB_0) - 1L)  /* avoid overflow */
/*
 * For constructing the negative timestamp upper bound value.
 * binary: 11111111 11111111 11111111 11111111
 */
#define UPPER_MSB_1 (-1L)
/*
 * Upper bound for nanoseconds value supported by the encoding.
 * binary: 00111111 11111111 11111111 11111111
 */
#define MAX_NANOSECONDS ((1L << 30) - 1)

#define CASE_NAME_FORMAT "%s: msb:%x lower_bound:%x extra_bits: %x"

#define LOWER_BOUND_NEG_NO_EXTRA_BITS_CASE\
        "1901-12-13 Lower bound of 32bit < 0 timestamp, no extra bits"
#define UPPER_BOUND_NEG_NO_EXTRA_BITS_CASE\
        "1969-12-31 Upper bound of 32bit < 0 timestamp, no extra bits"
#define LOWER_BOUND_NONNEG_NO_EXTRA_BITS_CASE\
        "1970-01-01 Lower bound of 32bit >=0 timestamp, no extra bits"
#define UPPER_BOUND_NONNEG_NO_EXTRA_BITS_CASE\
        "2038-01-19 Upper bound of 32bit >=0 timestamp, no extra bits"
#define LOWER_BOUND_NEG_LO_1_CASE\
        "2038-01-19 Lower bound of 32bit <0 timestamp, lo extra sec bit on"
#define UPPER_BOUND_NEG_LO_1_CASE\
        "2106-02-07 Upper bound of 32bit <0 timestamp, lo extra sec bit on"
#define LOWER_BOUND_NONNEG_LO_1_CASE\
        "2106-02-07 Lower bound of 32bit >=0 timestamp, lo extra sec bit on"
#define UPPER_BOUND_NONNEG_LO_1_CASE\
        "2174-02-25 Upper bound of 32bit >=0 timestamp, lo extra sec bit on"
#define LOWER_BOUND_NEG_HI_1_CASE\
        "2174-02-25 Lower bound of 32bit <0 timestamp, hi extra sec bit on"
#define UPPER_BOUND_NEG_HI_1_CASE\
        "2242-03-16 Upper bound of 32bit <0 timestamp, hi extra sec bit on"
#define LOWER_BOUND_NONNEG_HI_1_CASE\
        "2242-03-16 Lower bound of 32bit >=0 timestamp, hi extra sec bit on"
#define UPPER_BOUND_NONNEG_HI_1_CASE\
        "2310-04-04 Upper bound of 32bit >=0 timestamp, hi extra sec bit on"
#define UPPER_BOUND_NONNEG_HI_1_NS_1_CASE\
        "2310-04-04 Upper bound of 32bit>=0 timestamp, hi extra sec bit 1. 1 ns"
#define LOWER_BOUND_NONNEG_HI_1_NS_MAX_CASE\
        "2378-04-22 Lower bound of 32bit>= timestamp. Extra sec bits 1. Max ns"
#define LOWER_BOUND_NONNEG_EXTRA_BITS_1_CASE\
        "2378-04-22 Lower bound of 32bit >=0 timestamp. All extra sec bits on"
#define UPPER_BOUND_NONNEG_EXTRA_BITS_1_CASE\
        "2446-05-10 Upper bound of 32bit >=0 timestamp. All extra sec bits on"

struct timestamp_expectation {
        const char *test_case_name;
        struct timespec64 expected;
        u32 extra_bits;
        bool msb_set;
        bool lower_bound;
};

static const struct timestamp_expectation test_data[] = {
        {
                .test_case_name = LOWER_BOUND_NEG_NO_EXTRA_BITS_CASE,
                .msb_set = true,
                .lower_bound = true,
                .extra_bits = 0,
                .expected = {.tv_sec = -0x80000000LL, .tv_nsec = 0L},
        },

        {
                .test_case_name = UPPER_BOUND_NEG_NO_EXTRA_BITS_CASE,
                .msb_set = true,
                .lower_bound = false,
                .extra_bits = 0,
                .expected = {.tv_sec = -1LL, .tv_nsec = 0L},
        },

        {
                .test_case_name = LOWER_BOUND_NONNEG_NO_EXTRA_BITS_CASE,
                .msb_set = false,
                .lower_bound = true,
                .extra_bits = 0,
                .expected = {0LL, 0L},
        },

        {
                .test_case_name = UPPER_BOUND_NONNEG_NO_EXTRA_BITS_CASE,
                .msb_set = false,
                .lower_bound = false,
                .extra_bits = 0,
                .expected = {.tv_sec = 0x7fffffffLL, .tv_nsec = 0L},
        },

        {
                .test_case_name = LOWER_BOUND_NEG_LO_1_CASE,
                .msb_set = true,
                .lower_bound = true,
                .extra_bits = 1,
                .expected = {.tv_sec = 0x80000000LL, .tv_nsec = 0L},
        },

        {
                .test_case_name = UPPER_BOUND_NEG_LO_1_CASE,
                .msb_set = true,
                .lower_bound = false,
                .extra_bits = 1,
                .expected = {.tv_sec = 0xffffffffLL, .tv_nsec = 0L},
        },

        {
                .test_case_name = LOWER_BOUND_NONNEG_LO_1_CASE,
                .msb_set = false,
                .lower_bound = true,
                .extra_bits = 1,
                .expected = {.tv_sec = 0x100000000LL, .tv_nsec = 0L},
        },

        {
                .test_case_name = UPPER_BOUND_NONNEG_LO_1_CASE,
                .msb_set = false,
                .lower_bound = false,
                .extra_bits = 1,
                .expected = {.tv_sec = 0x17fffffffLL, .tv_nsec = 0L},
        },

        {
                .test_case_name = LOWER_BOUND_NEG_HI_1_CASE,
                .msb_set = true,
                .lower_bound = true,
                .extra_bits =  2,
                .expected = {.tv_sec = 0x180000000LL, .tv_nsec = 0L},
        },

        {
                .test_case_name = UPPER_BOUND_NEG_HI_1_CASE,
                .msb_set = true,
                .lower_bound = false,
                .extra_bits = 2,
                .expected = {.tv_sec = 0x1ffffffffLL, .tv_nsec = 0L},
        },

        {
                .test_case_name = LOWER_BOUND_NONNEG_HI_1_CASE,
                .msb_set = false,
                .lower_bound = true,
                .extra_bits = 2,
                .expected = {.tv_sec = 0x200000000LL, .tv_nsec = 0L},
        },

        {
                .test_case_name = UPPER_BOUND_NONNEG_HI_1_CASE,
                .msb_set = false,
                .lower_bound = false,
                .extra_bits = 2,
                .expected = {.tv_sec = 0x27fffffffLL, .tv_nsec = 0L},
        },

        {
                .test_case_name = UPPER_BOUND_NONNEG_HI_1_NS_1_CASE,
                .msb_set = false,
                .lower_bound = false,
                .extra_bits = 6,
                .expected = {.tv_sec = 0x27fffffffLL, .tv_nsec = 1L},
        },

        {
                .test_case_name = LOWER_BOUND_NONNEG_HI_1_NS_MAX_CASE,
                .msb_set = false,
                .lower_bound = true,
                .extra_bits = 0xFFFFFFFF,
                .expected = {.tv_sec = 0x300000000LL,
                             .tv_nsec = MAX_NANOSECONDS},
        },

        {
                .test_case_name = LOWER_BOUND_NONNEG_EXTRA_BITS_1_CASE,
                .msb_set = false,
                .lower_bound = true,
                .extra_bits = 3,
                .expected = {.tv_sec = 0x300000000LL, .tv_nsec = 0L},
        },

        {
                .test_case_name = UPPER_BOUND_NONNEG_EXTRA_BITS_1_CASE,
                .msb_set = false,
                .lower_bound = false,
                .extra_bits = 3,
                .expected = {.tv_sec = 0x37fffffffLL, .tv_nsec = 0L},
        }
};

static void timestamp_expectation_to_desc(const struct timestamp_expectation *t,
                                          char *desc)
{
        strscpy(desc, t->test_case_name, KUNIT_PARAM_DESC_SIZE);
}

KUNIT_ARRAY_PARAM(ext4_inode, test_data, timestamp_expectation_to_desc);

static time64_t get_32bit_time(const struct timestamp_expectation * const test)
{
        if (test->msb_set) {
                if (test->lower_bound)
                        return LOWER_MSB_1;

                return UPPER_MSB_1;
        }

        if (test->lower_bound)
                return LOWER_MSB_0;
        return UPPER_MSB_0;
}


/*
 *  Test data is derived from the table in the Inode Timestamps section of
 *  Documentation/filesystems/ext4/inodes.rst.
 */
static void inode_test_xtimestamp_decoding(struct kunit *test)
{
        struct timespec64 timestamp;

        struct timestamp_expectation *test_param =
                        (struct timestamp_expectation *)(test->param_value);

        timestamp = ext4_decode_extra_time(
                                cpu_to_le32(get_32bit_time(test_param)),
                                cpu_to_le32(test_param->extra_bits));

        KUNIT_EXPECT_EQ_MSG(test,
                            test_param->expected.tv_sec,
                            timestamp.tv_sec,
                            CASE_NAME_FORMAT,
                            test_param->test_case_name,
                            test_param->msb_set,
                            test_param->lower_bound,
                            test_param->extra_bits);
        KUNIT_EXPECT_EQ_MSG(test,
                            test_param->expected.tv_nsec,
                            timestamp.tv_nsec,
                            CASE_NAME_FORMAT,
                            test_param->test_case_name,
                            test_param->msb_set,
                            test_param->lower_bound,
                            test_param->extra_bits);
}

static struct kunit_case ext4_inode_test_cases[] = {
        KUNIT_CASE_PARAM(inode_test_xtimestamp_decoding, ext4_inode_gen_params),
        {}
};

static struct kunit_suite ext4_inode_test_suite = {
        .name = "ext4_inode_test",
        .test_cases = ext4_inode_test_cases,
};

kunit_test_suites(&ext4_inode_test_suite);

MODULE_DESCRIPTION("KUnit test of ext4 inode timestamp decoding");
MODULE_LICENSE("GPL v2");