tests: Add lib/coreboot_table-test test case

Signed-off-by: Jakub Czapiga <jacz@semihalf.com>
Change-Id: I4d7d9ca43316bf514bb5ebd8909f441bcfa20eba
Reviewed-on: https://review.coreboot.org/c/coreboot/+/52159
Tested-by: build bot (Jenkins) <no-reply@coreboot.org>
Reviewed-by: Paul Fagerburg <pfagerburg@chromium.org>
diff --git a/tests/lib/Makefile.inc b/tests/lib/Makefile.inc
index ee2bf5a..4943e06 100644
--- a/tests/lib/Makefile.inc
+++ b/tests/lib/Makefile.inc
@@ -26,6 +26,7 @@
 tests-y += uuid-test
 tests-y += bootmem-test
 tests-y += dimm_info_util-test
+tests-y += coreboot_table-test
 
 string-test-srcs += tests/lib/string-test.c
 string-test-srcs += src/lib/string.c
@@ -129,3 +130,12 @@
 dimm_info_util-test-srcs += src/lib/dimm_info_util.c
 dimm_info_util-test-srcs += tests/stubs/console.c
 
+coreboot_table-test-srcs += tests/lib/coreboot_table-test.c
+coreboot_table-test-srcs += tests/stubs/console.c
+coreboot_table-test-srcs += src/lib/compute_ip_checksum.c
+coreboot_table-test-srcs += src/lib/coreboot_table.c
+coreboot_table-test-srcs += src/lib/imd_cbmem.c
+coreboot_table-test-srcs += src/lib/imd.c
+coreboot_table-test-cflags += -I src -I 3rdparty/vboot/firmware/include
+coreboot_table-test-cflags += -I tests/include/tests/lib/fmap
+coreboot_table-test-mocks += cbmem_top_chipset
diff --git a/tests/lib/coreboot_table-test.c b/tests/lib/coreboot_table-test.c
new file mode 100644
index 0000000..cd82ee5
--- /dev/null
+++ b/tests/lib/coreboot_table-test.c
@@ -0,0 +1,497 @@
+/* SPDX-License-Identifier: GPL-2.0-only */
+
+#include <tests/test.h>
+#include <boardid.h>
+#include <boot/coreboot_tables.h>
+#include <boot/tables.h>
+#include <cbfs.h>
+#include <cbmem.h>
+#include <commonlib/helpers.h>
+#include <commonlib/region.h>
+#include <fmap_config.h>
+#include <fw_config.h>
+#include <stdbool.h>
+#include <version.h>
+
+
+/* Copy of lb_table_init() implementation for testing purposes */
+static struct lb_header *lb_table_init(unsigned long addr)
+{
+	struct lb_header *header;
+
+	/* 16 byte align the address */
+	addr += 15;
+	addr &= ~15;
+
+	header = (void *)addr;
+	header->signature[0] = 'L';
+	header->signature[1] = 'B';
+	header->signature[2] = 'I';
+	header->signature[3] = 'O';
+	header->header_bytes = sizeof(*header);
+	header->header_checksum = 0;
+	header->table_bytes = 0;
+	header->table_checksum = 0;
+	header->table_entries = 0;
+	return header;
+}
+
+static struct lb_record *lb_first_record(struct lb_header *header)
+{
+	struct lb_record *rec;
+	rec = (void *)(((char *)header) + sizeof(*header));
+	return rec;
+}
+
+#define LB_RECORD_FOR_EACH(record_ptr, index, header)					\
+	for (index = 0, record_ptr = lb_first_record(header);				\
+		index < header->table_entries;						\
+		record_ptr = (struct lb_record *)((uintptr_t)record_ptr			\
+							+ record_ptr->size), index++)
+
+static void test_lb_add_gpios(void **state)
+{
+	struct lb_gpio gpios[] = {
+		{-1, ACTIVE_HIGH, 1, "lid"},
+		{-1, ACTIVE_HIGH, 0, "power"},
+		{-1, ACTIVE_HIGH, 1, "oprom"},
+		{-1, ACTIVE_HIGH, 0, "EC in RW"},
+	};
+	const size_t gpios_buf_size = sizeof(struct lb_gpios) + sizeof(struct lb_gpio) * 32;
+	uint8_t gpios_buf[gpios_buf_size];
+	struct lb_gpios *gpios_table = (struct lb_gpios *)gpios_buf;
+	gpios_table->count = 0;
+	gpios_table->size = 0;
+	gpios_table->tag = LB_TAG_GPIO;
+
+	/* Add GPIOs an check if they have been added to the table.
+	   GPIOs are added in the same order to the end of the table. */
+	lb_add_gpios(gpios_table, gpios, ARRAY_SIZE(gpios));
+	assert_int_equal(ARRAY_SIZE(gpios), gpios_table->count);
+	assert_int_equal(sizeof(gpios), gpios_table->size);
+	assert_memory_equal(&gpios_table->gpios[0], gpios, sizeof(gpios));
+
+	/* Add subset of gpios and check if they have been added correctly. */
+	lb_add_gpios(gpios_table, &gpios[1], 2);
+	assert_int_equal(ARRAY_SIZE(gpios) + 2, gpios_table->count);
+	assert_int_equal(sizeof(gpios) + 2 * sizeof(gpios[0]), gpios_table->size);
+	assert_memory_equal(&gpios_table->gpios[0], gpios, sizeof(gpios));
+	assert_memory_equal(&gpios_table->gpios[ARRAY_SIZE(gpios)], &gpios[1],
+				2 * sizeof(gpios[0]));
+}
+
+uint8_t tables_buffer[sizeof(struct lb_header) + 10 * KiB];
+static int setup_test_header(void **state)
+{
+	*state = lb_table_init((uintptr_t)tables_buffer);
+
+	return 0;
+}
+
+static void test_lb_new_record(void **state)
+{
+	struct lb_header *header = *state;
+	const size_t entries = 10;
+	int i;
+	size_t entries_offset;
+	size_t accumulated_size = 0;
+	struct lb_record *curr;
+
+
+	assert_int_equal(0, header->table_entries);
+	assert_int_equal(0, header->table_bytes);
+
+	lb_new_record(header);
+	assert_int_equal(1, header->table_entries);
+	assert_int_equal(0, header->table_bytes);
+
+	/* Create few entries with varying sizes (but at least of sizeof(struct lb_record))
+	   Accumulate and check size of table after each lb_new_record() call. */
+	entries_offset = header->table_entries;
+	accumulated_size = sizeof(struct lb_record);
+	for (i = 0; i < entries; ++i) {
+		curr = lb_new_record(header);
+		curr->size = sizeof(struct lb_record) + ((i + 2) * 7) % 32;
+
+		assert_int_equal(entries_offset + (i + 1), header->table_entries);
+		assert_int_equal(accumulated_size, header->table_bytes);
+		accumulated_size += curr->size;
+	}
+}
+
+static void test_lb_add_serial(void **state)
+{
+	struct lb_header *header = *state;
+	struct lb_serial serial;
+
+	serial.type = LB_SERIAL_TYPE_MEMORY_MAPPED;
+	serial.baseaddr = 0xFEDC6000;
+	serial.baud = 115200;
+	serial.regwidth = 1;
+	serial.input_hertz = 115200 * 16;
+	serial.uart_pci_addr = 0x0;
+	lb_add_serial(&serial, header);
+
+	assert_int_equal(1, header->table_entries);
+	/* Table bytes and checksum should be zero, because it is updated with size of previous
+	   record or when table is closed. No previous record is present. */
+	assert_int_equal(0, header->table_bytes);
+	assert_int_equal(0, header->table_checksum);
+}
+
+static void test_lb_add_console(void **state)
+{
+	struct lb_header *header = *state;
+
+	lb_add_console(LB_TAG_CONSOLE_SERIAL8250MEM, header);
+	assert_int_equal(1, header->table_entries);
+	/* Table bytes and checksum should be zero, because it is updated with size of previous
+	   record or when table is closed. No previous record is present. */
+	assert_int_equal(0, header->table_bytes);
+	assert_int_equal(0, header->table_checksum);
+}
+
+static void test_multiple_entries(void **state)
+{
+	struct lb_header *header = *state;
+
+	/* Add two entries */
+	lb_add_console(LB_TAG_CONSOLE_SERIAL8250, header);
+	lb_add_console(LB_TAG_CONSOLE_SERIAL8250MEM, header);
+
+	assert_int_equal(2, header->table_entries);
+	assert_int_equal(sizeof(struct lb_console), header->table_bytes);
+}
+
+static void test_write_coreboot_forwarding_table(void **state)
+{
+	struct lb_header *header = *state;
+	uint8_t forwarding_table_buffer[sizeof(struct lb_header)
+					+ 2 * sizeof(struct lb_forward)];
+	struct lb_header *forward_header =
+			(struct lb_header *)ALIGN_UP((uintptr_t)forwarding_table_buffer, 16);
+	size_t forwarding_table_size =
+			write_coreboot_forwarding_table((uintptr_t)forwarding_table_buffer,
+							(uintptr_t)header);
+	size_t expected_forwarding_table_size = ALIGN_UP((uintptr_t)forwarding_table_buffer, 16)
+						+ sizeof(struct lb_header)
+						+ sizeof(struct lb_forward)
+						- (uintptr_t)forwarding_table_buffer;
+	assert_int_equal(expected_forwarding_table_size, forwarding_table_size);
+
+	assert_int_equal(1, forward_header->table_entries);
+	assert_int_equal(sizeof(struct lb_forward), forward_header->table_bytes);
+	assert_ptr_equal(header,
+			((struct lb_forward *)lb_first_record(forward_header))->forward);
+}
+
+/* Mocks for write_tables() */
+const char mainboard_vendor[] = CONFIG_MAINBOARD_VENDOR;
+const char mainboard_part_number[] = CONFIG_MAINBOARD_PART_NUMBER;
+
+const char coreboot_version[] = "4.13";
+const char coreboot_extra_version[] = "abcdef";
+const char coreboot_build[] = "Coreboot build info";
+const unsigned int coreboot_version_timestamp = 1617191902U;
+const unsigned int coreboot_major_revision = 4;
+const unsigned int coreboot_minor_revision = 13;
+
+const char coreboot_compile_time[] = "13:58:22";
+const char coreboot_dmi_date[] = "03/31/2021";
+
+const struct bcd_date coreboot_build_date = {
+	.century = 0x20,
+	.year = 0x20,
+	.month = 0x03,
+	.day = 0x31,
+	.weekday = 0x2,
+};
+
+const unsigned int asl_revision = 0x20200925;
+
+void arch_write_tables(uintptr_t coreboot_table)
+{
+}
+
+struct resource mock_bootmem_ranges[] = {
+	{ .base = 0x1000, .size = 0x2000, .flags = LB_MEM_RAM },
+	{ .base = 0x0000, .size = 0x4000, .flags = LB_MEM_RAM },
+};
+
+void bootmem_write_memory_table(struct lb_memory *mem)
+{
+	struct lb_memory_range *lb_r = &mem->map[0];
+	int i;
+
+	/* Insert entries for testing */
+	for (i = 0; i < ARRAY_SIZE(mock_bootmem_ranges); ++i) {
+		struct resource *res = &mock_bootmem_ranges[i];
+		lb_r->start = pack_lb64(res->base);
+		lb_r->size = pack_lb64(res->size);
+		lb_r->type = res->flags;
+		lb_r++;
+		mem->size += sizeof(struct lb_memory_range);
+	}
+}
+
+void uart_fill_lb(void *data)
+{
+	struct lb_serial serial;
+	serial.type = LB_SERIAL_TYPE_MEMORY_MAPPED;
+	serial.baseaddr = 0xFEDC6000;
+	serial.baud = 115200;
+	serial.regwidth = 1;
+	serial.input_hertz = 115200 * 16;
+	serial.uart_pci_addr = 0x0;
+	lb_add_serial(&serial, data);
+
+	lb_add_console(LB_TAG_CONSOLE_SERIAL8250MEM, data);
+}
+
+struct cbfs_boot_device cbfs_boot_dev = {
+	.rdev = REGION_DEV_INIT(NULL, 0, 0x1000),
+	.mcache = (void *)0x1000,
+	.mcache_size = 0x1000,
+};
+
+const struct cbfs_boot_device *cbfs_get_boot_device(bool force_ro)
+{
+	return &cbfs_boot_dev;
+}
+
+void cbmem_run_init_hooks(int is_recovery)
+{
+}
+
+extern uintptr_t _cbmem_top_ptr;
+void *__wrap_cbmem_top_chipset(void)
+{
+	return (void *)_cbmem_top_ptr;
+}
+
+#define CBMEM_SIZE (64 * KiB)
+
+static int teardown_write_tables_test(void **state)
+{
+	free(*state);
+	_cbmem_top_ptr = 0;
+	return 0;
+}
+
+static int setup_write_tables_test(void **state)
+{
+	/* Allocate more data to have space for alignment */
+	void *top_ptr = malloc(CBMEM_SIZE + DYN_CBMEM_ALIGN_SIZE);
+	int32_t *mmc_status = NULL;
+
+	if (!top_ptr)
+		return -1;
+
+	*state = top_ptr;
+
+	_cbmem_top_ptr = ALIGN_UP((uintptr_t)top_ptr + CBMEM_SIZE, DYN_CBMEM_ALIGN_SIZE);
+
+	cbmem_initialize_empty();
+
+	mmc_status = cbmem_add(CBMEM_ID_MMC_STATUS, sizeof(int32_t));
+
+	if (mmc_status == NULL) {
+		teardown_write_tables_test(state);
+		return -1;
+	}
+
+	*mmc_status = 0x4433AADD;
+
+	return 0;
+}
+
+const struct region_device *boot_device_ro(void)
+{
+	return &cbfs_boot_dev.rdev;
+}
+
+uint64_t get_fmap_flash_offset(void)
+{
+	return FMAP_OFFSET;
+}
+
+uint32_t freq_khz = 5000 * 1000;
+void lb_arch_add_records(struct lb_header *header)
+{
+	struct lb_tsc_info *tsc_info;
+
+	tsc_info = (void *)lb_new_record(header);
+	tsc_info->tag = LB_TAG_TSC_INFO;
+	tsc_info->size = sizeof(*tsc_info);
+	tsc_info->freq_khz = freq_khz;
+}
+
+static void test_write_tables(void **state)
+{
+	void *cbtable_start;
+	struct lb_header *header;
+	struct lb_record *record;
+	int32_t *mmc_status = cbmem_find(CBMEM_ID_MMC_STATUS);
+	size_t i = 0;
+
+	/* Expect function to store cbtable entry in cbmem */
+	cbtable_start = write_tables();
+	assert_ptr_equal(cbtable_start, cbmem_find(CBMEM_ID_CBTABLE));
+
+	/* Expect correct lb_header at cbtable_start address */
+	header = (struct lb_header *)cbtable_start;
+	assert_non_null(header);
+	assert_memory_equal("LBIO", header, 4);
+	assert_int_equal(sizeof(*header), header->header_bytes);
+	/* At least one entry should be present. */
+	assert_int_not_equal(0, header->table_entries);
+
+	LB_RECORD_FOR_EACH(record, i, header) {
+		switch (record->tag) {
+		case LB_TAG_MEMORY:
+			/* Should be the same as in bootmem_write_memory_table() */
+			assert_int_equal(sizeof(struct lb_memory)
+					+ ARRAY_SIZE(mock_bootmem_ranges)
+						* sizeof(struct lb_memory_range),
+					record->size);
+
+			const struct lb_memory *memory = (struct lb_memory *)record;
+			const struct lb_memory_range *range;
+			const struct resource *res;
+			struct lb_uint64 value;
+
+			for (int i = 0; i < ARRAY_SIZE(mock_bootmem_ranges); ++i) {
+				res = &mock_bootmem_ranges[i];
+				range = &memory->map[i];
+
+				value = pack_lb64(res->base);
+				assert_memory_equal(&value, &range->start,
+							sizeof(struct lb_uint64));
+				value = pack_lb64(res->size);
+				assert_memory_equal(&value, &range->size,
+						sizeof(struct lb_uint64));
+				assert_int_equal(range->type, res->flags);
+			}
+			break;
+		case LB_TAG_MAINBOARD:
+			/* Mainboard record contains its header followed
+			   by two null-terminated strings */
+			assert_int_equal(ALIGN_UP(sizeof(struct lb_mainboard) +
+					ARRAY_SIZE(mainboard_vendor) +
+					ARRAY_SIZE(mainboard_part_number), 8), record->size);
+			break;
+		case LB_TAG_VERSION:
+			assert_int_equal(ALIGN_UP(sizeof(struct lb_string)
+					+ ARRAY_SIZE(coreboot_version), 8), record->size);
+			break;
+		case LB_TAG_EXTRA_VERSION:
+			assert_int_equal(ALIGN_UP(sizeof(struct lb_string)
+					+ ARRAY_SIZE(coreboot_extra_version), 8),
+					record->size);
+			break;
+		case LB_TAG_BUILD:
+			assert_int_equal(ALIGN_UP(sizeof(struct lb_string)
+					+ ARRAY_SIZE(coreboot_build), 8),
+					record->size);
+			break;
+		case LB_TAG_COMPILE_TIME:
+			assert_int_equal(ALIGN_UP(sizeof(struct lb_string)
+					+ ARRAY_SIZE(coreboot_compile_time), 8),
+					record->size);
+			break;
+		case LB_TAG_SERIAL:
+			assert_int_equal(sizeof(struct lb_serial), record->size);
+
+			/* This struct have the same values as created in uart_fill_lb() */
+			const struct lb_serial *serial = (struct lb_serial *)record;
+			assert_int_equal(LB_SERIAL_TYPE_MEMORY_MAPPED, serial->type);
+			assert_int_equal(0xFEDC6000, serial->baseaddr);
+			assert_int_equal(115200, serial->baud);
+			assert_int_equal(1, serial->regwidth);
+			assert_int_equal(115200 * 16, serial->input_hertz);
+			assert_int_equal(0x0, serial->uart_pci_addr);
+			break;
+		case LB_TAG_CONSOLE:
+			assert_int_equal(sizeof(struct lb_console), record->size);
+
+			/* This struct have the same values as created in uart_fill_lb() */
+			const struct lb_console *console = (struct lb_console *)record;
+			assert_int_equal(LB_TAG_CONSOLE_SERIAL8250MEM, console->type);
+			break;
+		case LB_TAG_VERSION_TIMESTAMP:
+			assert_int_equal(sizeof(struct lb_timestamp), record->size);
+
+			const struct lb_timestamp *timestamp = (struct lb_timestamp *)record;
+			assert_int_equal(coreboot_version_timestamp, timestamp->timestamp);
+			break;
+		case LB_TAG_BOOT_MEDIA_PARAMS:
+			assert_int_equal(sizeof(struct lb_boot_media_params), record->size);
+
+			const struct lb_boot_media_params *bmp =
+					(struct lb_boot_media_params *)record;
+			const struct cbfs_boot_device *cbd = cbfs_get_boot_device(false);
+			const struct region_device *boot_dev = boot_device_ro();
+			assert_int_equal(region_device_offset(&cbd->rdev), bmp->cbfs_offset);
+			assert_int_equal(region_device_sz(&cbd->rdev), bmp->cbfs_size);
+			assert_int_equal(region_device_sz(boot_dev), bmp->boot_media_size);
+			assert_int_equal(get_fmap_flash_offset(), bmp->fmap_offset);
+
+			break;
+		case LB_TAG_CBMEM_ENTRY:
+			assert_int_equal(sizeof(struct lb_cbmem_entry), record->size);
+
+			const struct lb_cbmem_entry *cbmem_entry =
+					(struct lb_cbmem_entry *)record;
+			const LargestIntegralType expected_tags[] = {
+				CBMEM_ID_CBTABLE,
+				CBMEM_ID_MMC_STATUS
+			};
+			assert_in_set(cbmem_entry->id,
+					expected_tags, ARRAY_SIZE(expected_tags));
+			break;
+		case LB_TAG_TSC_INFO:
+			assert_int_equal(sizeof(struct lb_tsc_info), record->size);
+
+			const struct lb_tsc_info *tsc_info = (struct lb_tsc_info *)record;
+			assert_int_equal(freq_khz, tsc_info->freq_khz);
+			break;
+		case LB_TAG_MMC_INFO:
+			assert_int_equal(sizeof(struct lb_mmc_info), record->size);
+
+			const struct lb_mmc_info *mmc_info = (struct lb_mmc_info *)record;
+			assert_int_equal(*mmc_status, mmc_info->early_cmd1_status);
+			break;
+		case LB_TAG_BOARD_CONFIG:
+			assert_int_equal(sizeof(struct lb_board_config), record->size);
+
+			const struct lb_board_config *board_config =
+					(struct lb_board_config *)record;
+			const struct lb_uint64 expected_fw_version = pack_lb64(fw_config_get());
+			assert_memory_equal(&expected_fw_version,
+					&board_config->fw_config, sizeof(struct lb_uint64));
+			assert_int_equal(board_id(), board_config->board_id);
+			assert_int_equal(ram_code(), board_config->ram_code);
+			assert_int_equal(sku_id(), board_config->sku_id);
+			break;
+		default:
+			fail_msg("Unexpected tag found in record. Tag ID: 0x%x", record->tag);
+		}
+	}
+}
+
+int main(void)
+{
+	const struct CMUnitTest tests[] = {
+		cmocka_unit_test(test_lb_add_gpios),
+		cmocka_unit_test_setup(test_lb_new_record, setup_test_header),
+		cmocka_unit_test_setup(test_lb_add_serial, setup_test_header),
+		cmocka_unit_test_setup(test_lb_add_console, setup_test_header),
+		cmocka_unit_test_setup(test_multiple_entries, setup_test_header),
+		cmocka_unit_test_setup(test_write_coreboot_forwarding_table, setup_test_header),
+		cmocka_unit_test_setup_teardown(test_write_tables,
+						setup_write_tables_test,
+						teardown_write_tables_test),
+	};
+
+	return cmocka_run_group_tests(tests, NULL, NULL);
+}