From 4417f398d9eaee16e9c08e5b231244b01ea745b1 Mon Sep 17 00:00:00 2001 From: Hsuan Ting Chen Date: Thu, 30 Jul 2020 13:52:34 +0800 Subject: vboot/ui: Add reinit screen hook Re-initialize the log screen when going back from another screen. BRANCH=puff BUG=b:146399181, b:146105976 TEST=CC=x86_64-pc-linux-gnu-clang; make clean && make runtests TEST=USE="menu_ui" emerge-puff depthcharge when the firmware log screen is implemented, select "advanced options", navigate to firmware log screen, press to debug info, and select back to firmware log screen Signed-off-by: Hsuan Ting Chen Change-Id: Ie44805e3bfb14a4a8b660a18a123a184a4c5ea45 Reviewed-on: https://chromium-review.googlesource.com/c/chromiumos/platform/vboot_reference/+/2329224 Reviewed-by: Yu-Ping Wu (cherry picked from commit 2cb872fa3056158c35f82a9412e667b69ccf5a49) Reviewed-on: https://chromium-review.googlesource.com/c/chromiumos/platform/vboot_reference/+/2332200 Reviewed-by: Joel Kitching --- firmware/2lib/2ui.c | 4 ++++ firmware/2lib/2ui_screens.c | 14 ++++++++++++++ firmware/2lib/include/2ui.h | 11 ++++++++++- 3 files changed, 28 insertions(+), 1 deletion(-) diff --git a/firmware/2lib/2ui.c b/firmware/2lib/2ui.c index 1bdc2739..99cedb0c 100644 --- a/firmware/2lib/2ui.c +++ b/firmware/2lib/2ui.c @@ -214,6 +214,8 @@ vb2_error_t vb2_ui_screen_back(struct vb2_ui_context *ui) tmp = ui->state->prev; free(ui->state); ui->state = tmp; + if (ui->state->screen->reinit) + return ui->state->screen->reinit(ui); } else { VB2_DEBUG("ERROR: No previous screen; ignoring\n"); } @@ -259,6 +261,8 @@ vb2_error_t vb2_ui_screen_change(struct vb2_ui_context *ui, enum vb2_screen id) ui->state = cur_state->prev; free(cur_state); } + if (ui->state->screen->reinit) + return ui->state->screen->reinit(ui); } else { /* Allocate the requested screen on top of the stack. */ cur_state = malloc(sizeof(*ui->state)); diff --git a/firmware/2lib/2ui_screens.c b/firmware/2lib/2ui_screens.c index e8f59164..a2cd60ff 100644 --- a/firmware/2lib/2ui_screens.c +++ b/firmware/2lib/2ui_screens.c @@ -266,12 +266,25 @@ static vb2_error_t debug_info_init(struct vb2_ui_context *ui) ui->error_code = VB2_UI_ERROR_DEBUG_LOG; return vb2_ui_screen_back(ui); } + return log_page_init(ui, DEBUG_INFO_ITEM_PAGE_UP, DEBUG_INFO_ITEM_PAGE_DOWN, DEBUG_INFO_ITEM_BACK); } +static vb2_error_t debug_info_reinit(struct vb2_ui_context *ui) +{ + const char *log_string = vb2ex_get_debug_info(ui->ctx); + ui->state->page_count = vb2ex_prepare_log_screen(log_string); + if (ui->state->page_count == 0) { + ui->error_code = VB2_UI_ERROR_DEBUG_LOG; + return vb2_ui_screen_back(ui); + } + + return VB2_REQUEST_UI_CONTINUE; +} + static vb2_error_t debug_info_page_prev_action(struct vb2_ui_context *ui) { return log_page_prev(ui, @@ -304,6 +317,7 @@ static const struct vb2_screen_info debug_info_screen = { .id = VB2_SCREEN_DEBUG_INFO, .name = "Debug info", .init = debug_info_init, + .reinit = debug_info_reinit, .menu = MENU_ITEMS(debug_info_items), }; diff --git a/firmware/2lib/include/2ui.h b/firmware/2lib/include/2ui.h index 5f2434c2..867b0c81 100644 --- a/firmware/2lib/include/2ui.h +++ b/firmware/2lib/include/2ui.h @@ -39,8 +39,17 @@ struct vb2_screen_info { enum vb2_screen id; /* Screen name for printing to console only */ const char *name; - /* Init function runs once when changing to the screen. */ + /* + * Init function runs once when changing to the screen which is not in + * the history stack. + */ vb2_error_t (*init)(struct vb2_ui_context *ui); + /* + * Re-init function runs once when changing to the screen which is + * already in the history stack, for example, when going back to the + * screen. Exactly one of init() and reinit() will be called. + */ + vb2_error_t (*reinit)(struct vb2_ui_context *ui); /* Action function runs repeatedly while on the screen. */ vb2_error_t (*action)(struct vb2_ui_context *ui); /* Menu items. */ -- cgit v1.2.1