/* SPDX-License-Identifier: GPL-2.0-only */ #include <stdlib.h> #include <string.h> #include <tests/test.h> #include <fmap.h> #include <commonlib/region.h> #include <tests/lib/fmap/fmap_data.h> #include <tests/lib/fmap/fmap_config.h> static struct region_device flash_rdev_rw; static struct region_device flash_rdev_ro; static char *flash_buffer = NULL; static size_t flash_buffer_size = 0; static void prepare_flash_buffer(void) { /* Prepare flash buffer with dummy data and FMAP */ flash_buffer = malloc(FMAP_SECTION_FLASH_SIZE); flash_buffer_size = FMAP_SECTION_FLASH_SIZE; /* Fill first part of buffer with dummy data */ for (int i = 0; i < FMAP_SECTION_FMAP_START; ++i) flash_buffer[i] = 'a' + i % ('z' - 'a'); /* Copy FMAP section into buffer */ memcpy(flash_buffer + FMAP_SECTION_FMAP_START, tests_fmap_bin, FMAP_SIZE); /* Fill rest of buffer with dummy data */ for (int i = FMAP_SECTION_FMAP_START + FMAP_SECTION_FMAP_SIZE; i < FMAP_SECTION_FLASH_SIZE; ++i) flash_buffer[i] = 'a' + i % ('z' - 'a'); } static int setup_fmap(void **state) { prepare_flash_buffer(); rdev_chain_mem_rw(&flash_rdev_rw, flash_buffer, FMAP_SECTION_FLASH_SIZE); rdev_chain_mem(&flash_rdev_ro, flash_buffer, FMAP_SECTION_FLASH_SIZE); return 0; } static int teardown_fmap(void **state) { rdev_chain_mem_rw(&flash_rdev_rw, NULL, 0); rdev_chain_mem(&flash_rdev_ro, NULL, 0); free(flash_buffer); flash_buffer = NULL; flash_buffer_size = 0; return 0; } void boot_device_init(void) { /* Setup in unit test setup function */ } const struct region_device *boot_device_ro(void) { return &flash_rdev_ro; } const struct region_device *boot_device_rw(void) { return &flash_rdev_rw; } static void test_fmap_locate_area_as_rdev(void **state) { const char buffer[] = "abcdefghijk0123456789"; struct region_device rdev; assert_int_not_equal(-1, fmap_locate_area_as_rdev("RO_VPD", &rdev)); assert_ptr_equal(flash_buffer + FMAP_SECTION_RO_VPD_START, rdev_mmap_full(&rdev)); assert_int_equal(FMAP_SECTION_RO_VPD_SIZE, region_device_sz(&rdev)); /* Check if locating area second time works */ assert_int_not_equal(-1, fmap_locate_area_as_rdev("RO_VPD", &rdev)); assert_ptr_equal(flash_buffer + FMAP_SECTION_RO_VPD_START, rdev_mmap_full(&rdev)); assert_int_equal(FMAP_SECTION_RO_VPD_SIZE, region_device_sz(&rdev)); assert_int_not_equal(-1, fmap_locate_area_as_rdev("RECOVERY_MRC_CACHE", &rdev)); assert_ptr_equal(flash_buffer + FMAP_SECTION_RECOVERY_MRC_CACHE_START, rdev_mmap_full(&rdev)); assert_int_equal(FMAP_SECTION_RECOVERY_MRC_CACHE_SIZE, region_device_sz(&rdev)); /* Expect error when writing to read-only area */ assert_int_equal(-1, rdev_writeat(&rdev, buffer, 0, sizeof(buffer))); /* Expect error when looking for incorrect area */ assert_int_equal(-1, fmap_locate_area_as_rdev("NONEXISTENT_AREA", &rdev)); assert_int_equal(-1, fmap_locate_area_as_rdev("", &rdev)); assert_int_equal(-1, fmap_locate_area_as_rdev(NULL, &rdev)); /* Function fmap_locate_area_as_rdev is not tested with NULL as region_device pointer as it is not allowed. */ } static void test_fmap_locate_area_as_rdev_rw(void **state) { struct region_device rdev; size_t ro_rw_section_size = FMAP_SECTION_MISC_RW_SIZE; char *buffer1 = malloc(ro_rw_section_size); char *buffer2 = malloc(ro_rw_section_size); char *dummy_data = malloc(ro_rw_section_size); /* Fill buffer with dummy data */ for (int i = 0; i < ro_rw_section_size; ++i) dummy_data[i] = '0' + i % ('9' - '0'); assert_int_not_equal(-1, fmap_locate_area_as_rdev_rw("RW_SECTION_A", &rdev)); assert_ptr_equal(flash_buffer + FMAP_SECTION_RW_SECTION_A_START, rdev_mmap_full(&rdev)); assert_int_equal(FMAP_SECTION_RW_SECTION_A_SIZE, region_device_sz(&rdev)); /* Check if locating area second time works */ assert_int_not_equal(-1, fmap_locate_area_as_rdev_rw("RW_SECTION_A", &rdev)); assert_ptr_equal(flash_buffer + FMAP_SECTION_RW_SECTION_A_START, rdev_mmap_full(&rdev)); assert_int_equal(FMAP_SECTION_RW_SECTION_A_SIZE, region_device_sz(&rdev)); assert_int_not_equal(-1, fmap_locate_area_as_rdev_rw("MISC_RW", &rdev)); assert_ptr_equal(flash_buffer + FMAP_SECTION_MISC_RW_START, rdev_mmap_full(&rdev)); assert_int_equal(FMAP_SECTION_MISC_RW_SIZE, region_device_sz(&rdev)); /* Expect error when looking for incorrect area */ assert_int_equal(-1, fmap_locate_area_as_rdev_rw("NONEXISTENT_AREA", &rdev)); assert_int_equal(-1, fmap_locate_area_as_rdev_rw("", &rdev)); /* Expect error when passing invalid references */ assert_int_equal(-1, fmap_locate_area_as_rdev_rw(NULL, &rdev)); /* Function fmap_locate_area_as_rdev_rw is not tested with NULL as region_device pointer as it is not allowed. */ /* Test if returned section region device is writable */ assert_int_not_equal(-1, fmap_locate_area_as_rdev_rw("MISC_RW", &rdev)); assert_int_equal(ro_rw_section_size, rdev_readat(&rdev, buffer1, 0, ro_rw_section_size)); assert_int_equal(ro_rw_section_size, rdev_writeat(&rdev, dummy_data, 0, ro_rw_section_size)); /* Check if written data is visible and correct after locating area as RO */ assert_int_not_equal(-1, fmap_locate_area_as_rdev("MISC_RW", &rdev)); assert_int_equal(ro_rw_section_size, rdev_readat(&rdev, buffer2, 0, ro_rw_section_size)); assert_memory_not_equal(buffer1, buffer2, ro_rw_section_size); assert_memory_equal(dummy_data, buffer2, ro_rw_section_size); free(buffer1); free(buffer2); free(dummy_data); } static void test_fmap_locate_area(void **state) { struct region ar; /* Try to locate named area */ assert_int_not_equal(-1, fmap_locate_area("COREBOOT", &ar)); assert_int_equal(FMAP_SECTION_COREBOOT_START, region_offset(&ar)); assert_int_equal(FMAP_SECTION_COREBOOT_SIZE, region_sz(&ar)); /* Check if locating area second time works */ assert_int_not_equal(-1, fmap_locate_area("COREBOOT", &ar)); assert_int_equal(FMAP_SECTION_COREBOOT_START, region_offset(&ar)); assert_int_equal(FMAP_SECTION_COREBOOT_SIZE, region_sz(&ar)); /* Look for another area */ assert_int_not_equal(-1, fmap_locate_area("GBB", &ar)); assert_int_equal(FMAP_SECTION_GBB_START, region_offset(&ar)); assert_int_equal(FMAP_SECTION_GBB_SIZE, region_sz(&ar)); /* Expect error when looking for incorrect area */ assert_int_equal(-1, fmap_locate_area("NONEXISTENT_AREA", &ar)); assert_int_equal(-1, fmap_locate_area("", &ar)); assert_int_equal(-1, fmap_locate_area(NULL, &ar)); /* Expect error when passing invalid region pointer */ assert_int_equal(-1, fmap_locate_area("SHARED_DATA", NULL)); } static void test_fmap_find_region_name(void **state) { (void)state; struct region ar; char found_area_name[FMAP_STRLEN] = ""; const char *area_name = "RW_PRESERVE"; /* Find area by name */ assert_int_not_equal(-1, fmap_locate_area(area_name, &ar)); /* Find name of previously located region */ assert_int_not_equal(-1, fmap_find_region_name(&ar, found_area_name)); assert_string_equal(area_name, found_area_name); /* Expect error when passing invalid buffer */ assert_int_equal(-1, fmap_find_region_name(&ar, NULL)); /* Expect error when passing invalid region pointer */ assert_int_equal(-1, fmap_find_region_name(NULL, found_area_name)); /* Try to find area outside of flash region */ ar.offset = FMAP_SECTION_FLASH_START + FMAP_SECTION_FLASH_SIZE + 0x100; ar.size = 0x1000; assert_int_equal(-1, fmap_find_region_name(&ar, found_area_name)); /* Try to find area with correct offset and incorrect size */ ar.offset = FMAP_SECTION_COREBOOT_START; ar.size = FMAP_SECTION_COREBOOT_SIZE / 4; assert_int_equal(-1, fmap_find_region_name(&ar, found_area_name)); /* Try to find area with correct size and incorrect offset */ ar.offset = FMAP_SECTION_GBB_START - 0x100; ar.size = FMAP_SECTION_GBB_START; assert_int_equal(-1, fmap_find_region_name(&ar, found_area_name)); /* Try to find area with correct offset overlapping with another area */ ar.offset = FMAP_SECTION_MISC_RW_START; ar.size = FMAP_SECTION_MISC_RW_START + 0x1000; assert_int_equal(-1, fmap_find_region_name(&ar, found_area_name)); } static void test_fmap_read_area(void **state) { const unsigned int section_size = FMAP_SECTION_RW_SECTION_A_SIZE; const unsigned int section_start = FMAP_SECTION_RW_SECTION_A_START; char *buffer = malloc(section_size); /* Find and read area data. Compare with memory device simulating flash. */ assert_int_equal(section_size, fmap_read_area("RW_SECTION_A", buffer, section_size)); assert_memory_equal(flash_buffer + section_start, buffer, section_size); /* Expect error when reading incorrect area */ assert_int_equal(-1, fmap_read_area("NONEXISTENT_SECTION", buffer, section_size)); assert_int_equal(-1, fmap_read_area("", buffer, section_size)); assert_int_equal(-1, fmap_read_area(NULL, buffer, section_size)); /* Function fmap_read_area is not tested with NULL as output buffer pointer as it is not allowed. */ free(buffer); } static void test_fmap_overwrite_area(void **state) { const char *section_name = "FW_MAIN_A"; const unsigned int section_size = FMAP_SECTION_FW_MAIN_A_SIZE; char *buffer1 = malloc(section_size); char *buffer2 = malloc(section_size); char *new_data = malloc(section_size / 2); char *zero_buffer = malloc(section_size / 2); memset(zero_buffer, 0, section_size / 2); memset(new_data, 0x42, section_size / 2); /* Save buffer for future comparisons */ assert_int_equal(section_size, fmap_read_area(section_name, buffer1, section_size)); /* Overwrite part of section. */ assert_int_equal(section_size / 2, fmap_overwrite_area(section_name, new_data, section_size / 2)); /* Read and check if memory has changed as expected */ assert_int_equal(section_size, fmap_read_area(section_name, buffer2, section_size)); assert_memory_not_equal(buffer1, buffer2, section_size); /* Check if requested section area was overwritten properly */ assert_memory_equal(buffer2, new_data, section_size / 2); /* Check if rest of section was zero-filled */ assert_memory_equal(buffer2 + (section_size / 2), zero_buffer, section_size / 2); /* Expect error when overwriting incorrect section */ assert_int_equal( -1, fmap_overwrite_area("NONEXISTENT_SECTION", new_data, section_size / 2)); assert_int_equal(-1, fmap_overwrite_area(NULL, new_data, section_size / 2)); /* Function fmap_overwrite_area is not tested with NULL as input buffer pointer as it is not allowed. */ free(buffer1); free(buffer2); free(new_data); free(zero_buffer); } int main(void) { const struct CMUnitTest tests[] = { cmocka_unit_test_setup_teardown(test_fmap_locate_area_as_rdev, setup_fmap, teardown_fmap), cmocka_unit_test_setup_teardown(test_fmap_locate_area_as_rdev_rw, setup_fmap, teardown_fmap), cmocka_unit_test_setup_teardown(test_fmap_locate_area, setup_fmap, teardown_fmap), cmocka_unit_test_setup_teardown(test_fmap_find_region_name, setup_fmap, teardown_fmap), cmocka_unit_test_setup_teardown(test_fmap_read_area, setup_fmap, teardown_fmap), cmocka_unit_test_setup_teardown(test_fmap_overwrite_area, setup_fmap, teardown_fmap), }; return cb_run_group_tests(tests, NULL, NULL); }