diff options
-rw-r--r-- | firmware/lib/vboot_ui_menu.c | 5 | ||||
-rw-r--r-- | tests/vboot_detach_menu_tests.c | 5 |
2 files changed, 7 insertions, 3 deletions
diff --git a/firmware/lib/vboot_ui_menu.c b/firmware/lib/vboot_ui_menu.c index 628b6b89..3c202c28 100644 --- a/firmware/lib/vboot_ui_menu.c +++ b/firmware/lib/vboot_ui_menu.c @@ -713,9 +713,8 @@ static vb2_error_t vb2_init_menus(struct vb2_context *ctx) int i; /* Initialize language menu with the correct amount of entries. */ - VbExGetLocalizationCount(&count); - if (!count) - count = 1; /* Always need at least one language entry. */ + if (VB2_SUCCESS != VbExGetLocalizationCount(&count) || count == 0) + count = 1; /* Fall back to 1 language entry on failure */ items = malloc(count * sizeof(struct vb2_menu_item)); if (!items) diff --git a/tests/vboot_detach_menu_tests.c b/tests/vboot_detach_menu_tests.c index 613c8ef5..1a0c6175 100644 --- a/tests/vboot_detach_menu_tests.c +++ b/tests/vboot_detach_menu_tests.c @@ -134,6 +134,11 @@ struct vb2_gbb_header *vb2_get_gbb(struct vb2_context *c) return &gbb; } +vb2_error_t VbExGetLocalizationCount(uint32_t *count) { + *count = 1; + return VB2_SUCCESS; +} + uint32_t VbExGetAltFwIdxMask() { vbexaltfwmask_called++; |