1 files changed, 257 insertions, 0 deletions
diff --git a/tests/lib/spd_cache-test.c b/tests/lib/spd_cache-test.c
new file mode 100644
index 0000000000..4502ce6d23
--- /dev/null
+++ b/tests/lib/spd_cache-test.c
@@ -0,0 +1,257 @@
+/* SPDX-License-Identifier: GPL-2.0-only */
+
+#include <crc_byte.h>
+#include <spd_bin.h>
+#include <spd_cache.h>
+#include <stdlib.h>
+#include <string.h>
+#include <tests/test.h>
+#include <tests/lib/spd_cache_data.h>
+
+struct region_device flash_rdev_rw;
+static char *flash_buffer = NULL;
+static size_t flash_buffer_size = 0;
+
+static int setup_spd_cache(void **state)
+{
+	flash_buffer_size = SC_SPD_TOTAL_LEN + SC_CRC_LEN;
+	flash_buffer = malloc(flash_buffer_size);
+
+	if (flash_buffer == NULL) {
+		flash_buffer_size = 0;
+		return -1;
+	}
+
+	rdev_chain_mem_rw(&flash_rdev_rw, flash_buffer, flash_buffer_size);
+	return 0;
+}
+
+static int setup_spd_cache_test(void **state)
+{
+	memset(flash_buffer, 0xff, flash_buffer_size);
+	return 0;
+}
+
+static int teardown_spd_cache(void **state)
+{
+	rdev_chain_mem_rw(&flash_rdev_rw, NULL, 0);
+	free(flash_buffer);
+	flash_buffer = NULL;
+	flash_buffer_size = 0;
+	return 0;
+}
+
+
+int __wrap_fmap_locate_area_as_rdev(const char *name, struct region_device *area)
+{
+	return rdev_chain(area, &flash_rdev_rw, 0, flash_buffer_size);
+}
+
+/* This test verifies if load_spd_cache() correctly loads spd_cache pointer and size
+   from provided region_device. Memory region device is returned by
+   __wrap_fmap_locate_area_as_rdev() */
+static void test_load_spd_cache(void **state)
+{
+	uint8_t *spd_cache;
+	size_t spd_cache_sz;
+
+	assert_int_equal(CB_SUCCESS, load_spd_cache(&spd_cache, &spd_cache_sz));
+	assert_ptr_equal(flash_buffer, spd_cache);
+	assert_int_equal(SC_SPD_TOTAL_LEN + SC_CRC_LEN, spd_cache_sz);
+}
+
+static void calc_spd_cache_crc(uint8_t *spd_cache)
+{
+	*(uint16_t *)(spd_cache + SC_CRC_OFFSET) =
+			CRC(spd_cache, SC_SPD_TOTAL_LEN, crc16_byte);
+}
+
+__attribute__((unused))
+static void fill_spd_cache_ddr3(uint8_t *spd_cache, size_t spd_cache_sz)
+{
+	assert_true(spd_cache_sz >= (spd_data_ddr3_1_sz + sizeof(uint16_t)));
+
+	memcpy(spd_cache, spd_data_ddr3_1, spd_data_ddr3_1_sz);
+	memset(spd_cache + spd_data_ddr3_1_sz, 0, spd_cache_sz - spd_data_ddr3_1_sz);
+	calc_spd_cache_crc(spd_cache);
+}
+
+__attribute__((unused))
+static void fill_spd_cache_ddr4(uint8_t *spd_cache, size_t spd_cache_sz)
+{
+	assert_true(spd_cache_sz >=
+			(spd_data_ddr4_1_sz + spd_data_ddr4_2_sz + sizeof(uint16_t)));
+
+	memcpy(spd_cache, spd_data_ddr4_1, spd_data_ddr4_1_sz);
+	memcpy(spd_cache + spd_data_ddr4_1_sz, spd_data_ddr4_2, spd_data_ddr4_2_sz);
+	memset(spd_cache + spd_data_ddr4_1_sz + spd_data_ddr4_2_sz, 0,
+			spd_cache_sz - (spd_data_ddr4_1_sz + spd_data_ddr4_2_sz));
+	calc_spd_cache_crc(spd_cache);
+}
+
+static void test_spd_fill_from_cache(void **state)
+{
+	struct spd_block blk;
+	uint8_t *spd_cache;
+	size_t spd_cache_sz;
+	assert_int_equal(CB_SUCCESS, load_spd_cache(&spd_cache, &spd_cache_sz));
+
+	/* Empty spd cache */
+	assert_int_equal(CB_ERR, spd_fill_from_cache(spd_cache, &blk));
+
+#if __TEST_SPD_CACHE_DDR == 3
+	fill_spd_cache_ddr3(spd_cache, spd_cache_sz);
+#elif __TEST_SPD_CACHE_DDR == 4
+	fill_spd_cache_ddr4(spd_cache, spd_cache_sz);
+#endif
+	assert_int_equal(CB_SUCCESS, spd_fill_from_cache(spd_cache, &blk));
+}
+
+
+static void test_spd_cache_is_valid(void **state)
+{
+	uint8_t *spd_cache;
+	size_t spd_cache_sz;
+	assert_int_equal(CB_SUCCESS, load_spd_cache(&spd_cache, &spd_cache_sz));
+
+	/* Empty, incorrect SPD */
+	assert_false(spd_cache_is_valid(spd_cache, spd_cache_sz));
+
+#if __TEST_SPD_CACHE_DDR == 3
+	fill_spd_cache_ddr3(spd_cache, spd_cache_sz);
+#elif __TEST_SPD_CACHE_DDR == 4
+	fill_spd_cache_ddr4(spd_cache, spd_cache_sz);
+#endif
+	assert_true(spd_cache_is_valid(spd_cache, spd_cache_sz));
+}
+
+
+/* Used for setting `sn` parameter value */
+static u32 get_spd_sn_ret_sn[SC_SPD_NUMS] = { 0 };
+static size_t get_spd_sn_ret_sn_idx = 0;
+/* Implementation for testing purposes.  */
+enum cb_err get_spd_sn(u8 addr, u32 *sn)
+{
+	*sn = get_spd_sn_ret_sn[get_spd_sn_ret_sn_idx];
+	get_spd_sn_ret_sn_idx = (get_spd_sn_ret_sn_idx + 1) % ARRAY_SIZE(get_spd_sn_ret_sn);
+
+	return mock_type(enum cb_err);
+}
+
+static void get_sn_from_spd_cache(uint8_t *spd_cache, u32 arr[])
+{
+	for (int i = 0; i < SC_SPD_NUMS; ++i)
+		arr[i] = *(u32 *)(spd_cache + SC_SPD_OFFSET(i) + DDR4_SPD_SN_OFF);
+}
+
+/* check_if_dimm_changed() has is used only with DDR4, so there tests are not used for DDR3 */
+__attribute__((unused))
+static void test_check_if_dimm_changed_not_changed(void **state)
+{
+	uint8_t *spd_cache;
+	size_t spd_cache_sz;
+	struct spd_block blk;
+
+	assert_int_equal(CB_SUCCESS, load_spd_cache(&spd_cache, &spd_cache_sz));
+	fill_spd_cache_ddr4(spd_cache, spd_cache_sz);
+	assert_int_equal(CB_SUCCESS, spd_fill_from_cache(spd_cache, &blk));
+
+	get_sn_from_spd_cache(spd_cache, get_spd_sn_ret_sn);
+	get_spd_sn_ret_sn_idx = 0;
+	will_return_count(get_spd_sn, CB_SUCCESS, SC_SPD_NUMS);
+	assert_false(check_if_dimm_changed(spd_cache, &blk));
+}
+
+__attribute__((unused))
+static void test_check_if_dimm_changed_sn_error(void **state)
+{
+	uint8_t *spd_cache;
+	size_t spd_cache_sz;
+	struct spd_block blk;
+
+	assert_int_equal(CB_SUCCESS, load_spd_cache(&spd_cache, &spd_cache_sz));
+	fill_spd_cache_ddr4(spd_cache, spd_cache_sz);
+	assert_int_equal(CB_SUCCESS, spd_fill_from_cache(spd_cache, &blk));
+
+	/* Simulate error */
+	will_return_count(get_spd_sn, CB_ERR, 1);
+	assert_true(check_if_dimm_changed(spd_cache, &blk));
+}
+
+__attribute__((unused))
+static void test_check_if_dimm_changed_sodimm_lost(void **state)
+{
+	uint8_t *spd_cache;
+	size_t spd_cache_sz;
+	struct spd_block blk;
+
+	assert_int_equal(CB_SUCCESS, load_spd_cache(&spd_cache, &spd_cache_sz));
+	fill_spd_cache_ddr4(spd_cache, spd_cache_sz);
+	assert_int_equal(CB_SUCCESS, spd_fill_from_cache(spd_cache, &blk));
+	get_sn_from_spd_cache(spd_cache, get_spd_sn_ret_sn);
+	memset(spd_cache + spd_data_ddr4_1_sz, 0xff, spd_data_ddr4_2_sz);
+
+	get_spd_sn_ret_sn_idx = 0;
+	will_return_always(get_spd_sn, CB_SUCCESS);
+	assert_true(check_if_dimm_changed(spd_cache, &blk));
+}
+
+__attribute__((unused))
+static void test_check_if_dimm_changed_new_sodimm(void **state)
+{
+	uint8_t *spd_cache;
+	size_t spd_cache_sz;
+	struct spd_block blk;
+
+	assert_int_equal(CB_SUCCESS, load_spd_cache(&spd_cache, &spd_cache_sz));
+	fill_spd_cache_ddr4(spd_cache, spd_cache_sz);
+	assert_int_equal(CB_SUCCESS, spd_fill_from_cache(spd_cache, &blk));
+	get_sn_from_spd_cache(spd_cache, get_spd_sn_ret_sn);
+	memcpy(spd_cache + spd_data_ddr4_1_sz + spd_data_ddr4_2_sz,
+			spd_data_ddr4_2, spd_data_ddr4_2_sz);
+
+	get_spd_sn_ret_sn_idx = 0;
+	will_return_always(get_spd_sn, CB_SUCCESS);
+	assert_true(check_if_dimm_changed(spd_cache, &blk));
+}
+
+__attribute__((unused))
+static void test_check_if_dimm_changed_sn_changed(void **state)
+{
+	uint8_t *spd_cache;
+	size_t spd_cache_sz;
+	struct spd_block blk;
+
+	assert_int_equal(CB_SUCCESS, load_spd_cache(&spd_cache, &spd_cache_sz));
+	fill_spd_cache_ddr4(spd_cache, spd_cache_sz);
+	assert_int_equal(CB_SUCCESS, spd_fill_from_cache(spd_cache, &blk));
+	get_sn_from_spd_cache(spd_cache, get_spd_sn_ret_sn);
+	*(u32 *)(spd_cache + SC_SPD_OFFSET(0) + DDR4_SPD_SN_OFF) = 0x43211234;
+
+	get_spd_sn_ret_sn_idx = 0;
+	will_return_always(get_spd_sn, CB_SUCCESS);
+	assert_true(check_if_dimm_changed(spd_cache, &blk));
+}
+
+int main(void)
+{
+	const struct CMUnitTest tests[] = {
+		cmocka_unit_test_setup(test_load_spd_cache, setup_spd_cache_test),
+		cmocka_unit_test_setup(test_spd_fill_from_cache, setup_spd_cache_test),
+		cmocka_unit_test_setup(test_spd_cache_is_valid, setup_spd_cache_test),
+#if __TEST_SPD_CACHE_DDR == 4
+		cmocka_unit_test_setup(test_check_if_dimm_changed_not_changed,
+					setup_spd_cache_test),
+		cmocka_unit_test_setup(test_check_if_dimm_changed_sn_error,
+					setup_spd_cache_test),
+		cmocka_unit_test_setup(test_check_if_dimm_changed_sodimm_lost,
+					setup_spd_cache_test),
+		cmocka_unit_test_setup(test_check_if_dimm_changed_new_sodimm,
+					setup_spd_cache_test),
+		cmocka_unit_test_setup(test_check_if_dimm_changed_sn_changed,
+					setup_spd_cache_test),
+#endif
+	};
+
+	return cmocka_run_group_tests(tests, setup_spd_cache, teardown_spd_cache);
+}