From be5eca9126a048d7187d5c96de46d27df3b1359c Mon Sep 17 00:00:00 2001 From: Daisuke Nojiri Date: Tue, 6 Oct 2015 17:03:08 -0700 Subject: Add VbExGetLocalizationCount VbExGetLocalizationCount is a callback function which is supposed to return the number of screen locales supported by VbExDisplayScreen. After this change, we still try to get the number of locales from GBB first but when it fails, VbExGetLocalizationCount is called. The error code from VbGbbReadBmpHeader will be masked, similarly to the error from VbDislayScreenFromGBB. BUG=chromium:502066 BRANCH=tot TEST=Tested on Samus. make runtests Change-Id: I04ef8bf1ea02b1aaa05e65673b57bcea1932d8b0 Signed-off-by: Daisuke Nojiri Reviewed-on: https://chromium-review.googlesource.com/304376 Reviewed-by: Aaron Durbin --- firmware/include/vboot_api.h | 8 ++++++++ firmware/lib/vboot_display.c | 14 +++++++++++--- tests/vboot_display_tests.c | 4 ++-- 3 files changed, 21 insertions(+), 5 deletions(-) diff --git a/firmware/include/vboot_api.h b/firmware/include/vboot_api.h index bc90b19e..18ee4cc1 100644 --- a/firmware/include/vboot_api.h +++ b/firmware/include/vboot_api.h @@ -1078,4 +1078,12 @@ VbError_t VbLockDevice(void); * @return 0 if no override, 1-15 for override priority. */ uint8_t VbExOverrideGptEntryPriority(const GptEntry *e); + +/** + * Return number of locales supported + * + * @param count Pointer to the number of locales. + * @return VBERROR_... error, VBERROR_SUCCESS on success. + */ +VbError_t VbExGetLocalizationCount(uint32_t *count); #endif /* VBOOT_REFERENCE_VBOOT_API_H_ */ diff --git a/firmware/lib/vboot_display.c b/firmware/lib/vboot_display.c index 1a127e2d..f8417b9f 100644 --- a/firmware/lib/vboot_display.c +++ b/firmware/lib/vboot_display.c @@ -20,6 +20,11 @@ static uint32_t disp_current_screen = VB_SCREEN_BLANK; static uint32_t disp_width = 0, disp_height = 0; +__attribute__((weak)) +VbError_t VbExGetLocalizationCount(uint32_t *count) { + return VBERROR_UNKNOWN; +} + VbError_t VbGetLocalizationCount(VbCommonParams *cparams, uint32_t *count) { BmpBlockHeader hdr; @@ -28,12 +33,15 @@ VbError_t VbGetLocalizationCount(VbCommonParams *cparams, uint32_t *count) /* Default to 0 on error */ *count = 0; + /* First try to get the count from GBB */ ret = VbGbbReadBmpHeader(cparams, &hdr); - if (ret) + if (ret == VBERROR_SUCCESS) { + *count = hdr.number_of_localizations; return ret; + } - *count = hdr.number_of_localizations; - return VBERROR_SUCCESS; + /* If GBB is broken or missing, fallback to the callback */ + return VbExGetLocalizationCount(count); } /* diff --git a/tests/vboot_display_tests.c b/tests/vboot_display_tests.c index 63f68903..6e36f647 100644 --- a/tests/vboot_display_tests.c +++ b/tests/vboot_display_tests.c @@ -149,14 +149,14 @@ static void LocalizationTest(void) ResetMocks(); cparams.gbb->bmpfv_size = 0; TEST_EQ(VbGetLocalizationCount(&cparams, &count), - VBERROR_INVALID_GBB, "VbGetLocalizationCount bad gbb"); + VBERROR_UNKNOWN, "VbGetLocalizationCount bad gbb"); TEST_EQ(count, 0, " count"); VbApiKernelFree(&cparams); ResetMocks(); bhdr->signature[0] ^= 0x5a; TEST_EQ(VbGetLocalizationCount(&cparams, &count), - VBERROR_INVALID_BMPFV, "VbGetLocalizationCount bad bmpfv"); + VBERROR_UNKNOWN, "VbGetLocalizationCount bad bmpfv"); VbApiKernelFree(&cparams); ResetMocks(); -- cgit v1.2.1