test nvm sector flag invalid magic guard

F/2579

Author: danielinux

tools/unit-tests/unit-nvm.c

@@ -330,6 +330,38 @@ START_TEST(test_partition_magic_write_stops_on_flash_write_error)
 }
 END_TEST
 
+START_TEST(test_get_update_sector_flag_rejects_invalid_magic)
+{
+    int ret;
+    uint8_t st = 0xA5;
+    uint32_t *magic;
+
+    ret = mmap_file("/tmp/wolfboot-unit-invalid-magic.bin", (void *)MOCK_ADDRESS,
+            WOLFBOOT_PARTITION_SIZE, NULL);
+    ck_assert(ret >= 0);
+#ifdef FLAGS_HOME
+    ret = mmap_file("/tmp/wolfboot-unit-invalid-magic-int.bin",
+            (void *)MOCK_ADDRESS_BOOT, WOLFBOOT_PARTITION_SIZE, NULL);
+    ck_assert(ret >= 0);
+#endif
+    ret = mmap_file("/tmp/wolfboot-unit-invalid-magic-swap.bin",
+            (void *)MOCK_ADDRESS_SWAP, WOLFBOOT_SECTOR_SIZE, NULL);
+    ck_assert(ret >= 0);
+
+    hal_flash_unlock();
+    wolfBoot_erase_partition(PART_UPDATE);
+
+    magic = get_partition_magic(PART_UPDATE);
+    *magic = 0xFFFFFFFF;
+
+    ret = wolfBoot_get_update_sector_flag(0, &st);
+    ck_assert_int_eq(ret, -1);
+    ck_assert_uint_eq(st, 0xA5);
+
+    hal_flash_lock();
+}
+END_TEST
+
 
 Suite *wolfboot_suite(void)
 {
@@ -341,6 +373,8 @@ Suite *wolfboot_suite(void)
     tcase_add_test(nvm_select_fresh_sector, test_nvm_select_fresh_sector);
     tcase_add_test(nvm_select_fresh_sector,
             test_partition_magic_write_stops_on_flash_write_error);
+    tcase_add_test(nvm_select_fresh_sector,
+            test_get_update_sector_flag_rejects_invalid_magic);
     suite_add_tcase(s, nvm_select_fresh_sector);
 
     return s;