vboot/ui: Implement firmware log screen

Implement firmware log screen which can be accessed from advanced
options menu.

The screen displays a snapshot for the firmware log using the same
layout of debug info 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
     select "advanced options",
     and navigate to firmware log screen

Cq-Depend: chromium:2328247
Signed-off-by: Hsuan Ting Chen <roccochen@chromium.org>
Change-Id: I3cb5800d71925aa20ca4d5636172885e23fd0099
Reviewed-on: https://chromium-review.googlesource.com/c/chromiumos/platform/vboot_reference/+/2328242
Reviewed-by: Yu-Ping Wu <yupingso@chromium.org>

4 files changed, 122 insertions, 9 deletions

diff --git a/firmware/2lib/2stub.c b/firmware/2lib/2stub.c
index c4b9bdf3..0d6d5b46 100644
--- a/firmware/2lib/2stub.c
+++ b/firmware/2lib/2stub.c
@@ -100,6 +100,12 @@ const char *vb2ex_get_debug_info(struct vb2_context *ctx)
 }
 
 __attribute__((weak))
+const char *vb2ex_get_firmware_log(void)
+{
+	return NULL;
+}
+
+__attribute__((weak))
 uint32_t vb2ex_prepare_log_screen(const char *str)
 {
 	return 1;
diff --git a/firmware/2lib/2ui_screens.c b/firmware/2lib/2ui_screens.c
index 4af80950..653c279a 100644
--- a/firmware/2lib/2ui_screens.c
+++ b/firmware/2lib/2ui_screens.c
@@ -240,6 +240,10 @@ static const struct vb2_menu_item advanced_options_items[] = {
 		.text = "Debug info",
 		.target = VB2_SCREEN_DEBUG_INFO,
 	},
+	{
+		.text = "Firmware log",
+		.target = VB2_SCREEN_FIRMWARE_LOG,
+	},
 	BACK_ITEM,
 	POWER_OFF_ITEM,
 };
@@ -308,6 +312,62 @@ static const struct vb2_screen_info debug_info_screen = {
 };
 
 /******************************************************************************/
+/* VB2_SCREEN_FIRMWARE_LOG */
+
+#define FIRMWARE_LOG_ITEM_PAGE_UP 1
+#define FIRMWARE_LOG_ITEM_PAGE_DOWN 2
+#define FIRMWARE_LOG_ITEM_BACK 3
+
+static vb2_error_t firmware_log_init(struct vb2_ui_context *ui)
+{
+	const char *log_string = vb2ex_get_firmware_log();
+	ui->state->page_count = vb2ex_prepare_log_screen(log_string);
+	if (ui->state->page_count == 0) {
+		ui->error_code = VB2_UI_ERROR_FIRMWARE_LOG;
+		return vb2_ui_screen_back(ui);
+	}
+	return log_page_init(ui,
+			     FIRMWARE_LOG_ITEM_PAGE_UP,
+			     FIRMWARE_LOG_ITEM_PAGE_DOWN,
+			     FIRMWARE_LOG_ITEM_BACK);
+}
+
+static vb2_error_t firmware_log_page_prev_action(struct vb2_ui_context *ui)
+{
+	return log_page_prev(ui,
+			     FIRMWARE_LOG_ITEM_PAGE_UP,
+			     FIRMWARE_LOG_ITEM_PAGE_DOWN);
+}
+
+static vb2_error_t firmware_log_page_next_action(struct vb2_ui_context *ui)
+{
+	return log_page_next(ui,
+			     FIRMWARE_LOG_ITEM_PAGE_UP,
+			     FIRMWARE_LOG_ITEM_PAGE_DOWN);
+}
+
+static const struct vb2_menu_item firmware_log_items[] = {
+	LANGUAGE_SELECT_ITEM,
+	[FIRMWARE_LOG_ITEM_PAGE_UP] = {
+		.text = "Page up",
+		.action = firmware_log_page_prev_action,
+	},
+	[FIRMWARE_LOG_ITEM_PAGE_DOWN] = {
+		.text = "Page down",
+		.action = firmware_log_page_next_action,
+	},
+	[FIRMWARE_LOG_ITEM_BACK] = BACK_ITEM,
+	POWER_OFF_ITEM,
+};
+
+static const struct vb2_screen_info firmware_log_screen = {
+	.id = VB2_SCREEN_FIRMWARE_LOG,
+	.name = "Firmware log",
+	.init = firmware_log_init,
+	.menu = MENU_ITEMS(firmware_log_items),
+};
+
+/******************************************************************************/
 /* VB2_SCREEN_RECOVERY_SELECT */
 
 #define RECOVERY_SELECT_ITEM_PHONE 1
@@ -783,6 +843,7 @@ static const struct vb2_screen_info *screens[] = {
 	&recovery_broken_screen,
 	&advanced_options_screen,
 	&debug_info_screen,
+	&firmware_log_screen,
 	&recovery_select_screen,
 	&recovery_invalid_screen,
 	&recovery_to_dev_screen,
diff --git a/firmware/2lib/include/2api.h b/firmware/2lib/include/2api.h
index 8243e852..f546f12c 100644
--- a/firmware/2lib/include/2api.h
+++ b/firmware/2lib/include/2api.h
@@ -1213,6 +1213,8 @@ enum vb2_screen {
 	VB2_SCREEN_LANGUAGE_SELECT		= 0x130,
 	/* Debug info */
 	VB2_SCREEN_DEBUG_INFO			= 0x140,
+	/* Firmware log */
+	VB2_SCREEN_FIRMWARE_LOG			= 0x150,
 	/* First recovery screen to select recovering from disk or phone */
 	VB2_SCREEN_RECOVERY_SELECT		= 0x200,
 	/* Invalid recovery media inserted */
@@ -1243,6 +1245,8 @@ enum vb2_ui_error {
 	VB2_UI_ERROR_DEV_MODE_ALREADY_ENABLED,
 	/* Debug info screen initialization failed */
 	VB2_UI_ERROR_DEBUG_LOG,
+	/* Firmware log screen initialization failed */
+	VB2_UI_ERROR_FIRMWARE_LOG,
 };
 
 /**
@@ -1330,6 +1334,18 @@ const char *vb2ex_get_debug_info(struct vb2_context *ctx);
 char *vb2api_get_debug_info(struct vb2_context *ctx);
 
 /**
+ * Get the full firmware log string.
+ *
+ * Return a pointer to the full firmware log string which is guaranteed to be
+ * null-terminated.  The function implementation should snapshot the full
+ * firmware log when it is called the first time.  Subsequent calls should
+ * return the same pointer.
+ *
+ * @return The pointer to the full firmware log string.  NULL on error.
+ */
+const char *vb2ex_get_firmware_log(void);
+
+/**
  * Specify the string to be used for an upcoming log screen display.
  *
  * Before a vb2ex_display_ui() call is made for a screen which displays logs,
diff --git a/tests/vb2_ui_tests.c b/tests/vb2_ui_tests.c
index d4e0ae57..00afdb3b 100644
--- a/tests/vb2_ui_tests.c
+++ b/tests/vb2_ui_tests.c
@@ -1247,7 +1247,11 @@ static void developer_screen_tests(void)
 	add_mock_keypress(VB_KEY_ESC);
 	add_mock_keypress(VB_KEY_DOWN);
 	add_mock_keypress(VB_KEY_ENTER);
-	/* #3: Back */
+	/* #3: Firmware log */
+	add_mock_keypress(VB_KEY_ESC);
+	add_mock_keypress(VB_KEY_DOWN);
+	add_mock_keypress(VB_KEY_ENTER);
+	/* #4: Back */
 	add_mock_keypress(VB_KEY_ESC);
 	add_mock_keypress(VB_KEY_DOWN);
 	add_mock_keypress(VB_KEY_ENTER);
@@ -1272,11 +1276,17 @@ static void developer_screen_tests(void)
 		     MOCK_IGNORE, 2, 0x2, MOCK_IGNORE);
 	DISPLAYED_EQ("#2: debug info", VB2_SCREEN_DEBUG_INFO,
 		     MOCK_IGNORE, MOCK_IGNORE, MOCK_IGNORE, MOCK_IGNORE);
-	/* #3: Back */
+	/* #3: Firmware log */
 	DISPLAYED_PASS();
 	DISPLAYED_EQ("advanced options", VB2_SCREEN_ADVANCED_OPTIONS,
 		     MOCK_IGNORE, 3, 0x2, MOCK_IGNORE);
-	DISPLAYED_EQ("#3: back", VB2_SCREEN_DEVELOPER_MODE,
+	DISPLAYED_EQ("#3: firmware log", VB2_SCREEN_FIRMWARE_LOG,
+		     MOCK_IGNORE, MOCK_IGNORE, MOCK_IGNORE, MOCK_IGNORE);
+	/* #4: Back */
+	DISPLAYED_PASS();
+	DISPLAYED_EQ("advanced options", VB2_SCREEN_ADVANCED_OPTIONS,
+		     MOCK_IGNORE, 4, 0x2, MOCK_IGNORE);
+	DISPLAYED_EQ("#4: back", VB2_SCREEN_DEVELOPER_MODE,
 		     MOCK_IGNORE, MOCK_IGNORE, MOCK_IGNORE, MOCK_IGNORE);
 	/* End of menu */
 	DISPLAYED_EQ("end of menu", VB2_SCREEN_ADVANCED_OPTIONS,
@@ -1330,7 +1340,11 @@ static void broken_recovery_screen_tests(void)
 	add_mock_keypress(VB_KEY_ESC);
 	add_mock_keypress(VB_KEY_DOWN);
 	add_mock_keypress(VB_KEY_ENTER);
-	/* #3: Back */
+	/* #3: Firmware log */
+	add_mock_keypress(VB_KEY_ESC);
+	add_mock_keypress(VB_KEY_DOWN);
+	add_mock_keypress(VB_KEY_ENTER);
+	/* #4: Back */
 	add_mock_keypress(VB_KEY_ESC);
 	add_mock_keypress(VB_KEY_DOWN);
 	add_mock_keypress(VB_KEY_ENTER);
@@ -1353,11 +1367,17 @@ static void broken_recovery_screen_tests(void)
 		     MOCK_IGNORE, 2, 0x2, MOCK_IGNORE);
 	DISPLAYED_EQ("#2: debug info", VB2_SCREEN_DEBUG_INFO,
 		     MOCK_IGNORE, MOCK_IGNORE, MOCK_IGNORE, MOCK_IGNORE);
-	/* #3: Back */
+	/* #3: Firmware log */
 	DISPLAYED_PASS();
 	DISPLAYED_EQ("advanced options", VB2_SCREEN_ADVANCED_OPTIONS,
 		     MOCK_IGNORE, 3, 0x2, MOCK_IGNORE);
-	DISPLAYED_EQ("#3: back", VB2_SCREEN_RECOVERY_BROKEN,
+	DISPLAYED_EQ("#3: firmware log", VB2_SCREEN_FIRMWARE_LOG,
+		     MOCK_IGNORE, MOCK_IGNORE, MOCK_IGNORE, MOCK_IGNORE);
+	/* #4: Back */
+	DISPLAYED_PASS();
+	DISPLAYED_EQ("advanced options", VB2_SCREEN_ADVANCED_OPTIONS,
+		     MOCK_IGNORE, 4, 0x2, MOCK_IGNORE);
+	DISPLAYED_EQ("#4: back", VB2_SCREEN_RECOVERY_BROKEN,
 		     MOCK_IGNORE, MOCK_IGNORE, MOCK_IGNORE, MOCK_IGNORE);
 	/* End of menu */
 	DISPLAYED_EQ("end of menu", VB2_SCREEN_ADVANCED_OPTIONS,
@@ -1439,7 +1459,11 @@ static void manual_recovery_screen_tests(void)
 	add_mock_keypress(VB_KEY_ESC);
 	add_mock_keypress(VB_KEY_DOWN);
 	add_mock_keypress(VB_KEY_ENTER);
-	/* #3: Back */
+	/* #3: Firmware log */
+	add_mock_keypress(VB_KEY_ESC);
+	add_mock_keypress(VB_KEY_DOWN);
+	add_mock_keypress(VB_KEY_ENTER);
+	/* #4: Back */
 	add_mock_keypress(VB_KEY_ESC);
 	add_mock_keypress(VB_KEY_DOWN);
 	add_mock_keypress(VB_KEY_ENTER);
@@ -1470,11 +1494,17 @@ static void manual_recovery_screen_tests(void)
 		     MOCK_IGNORE, 2, 0x0, MOCK_IGNORE);
 	DISPLAYED_EQ("#2: debug info", VB2_SCREEN_DEBUG_INFO,
 		     MOCK_IGNORE, MOCK_IGNORE, MOCK_IGNORE, MOCK_IGNORE);
-	/* #3: Back */
+	/* #3: Firmware log */
 	DISPLAYED_PASS();
 	DISPLAYED_EQ("advanced options", VB2_SCREEN_ADVANCED_OPTIONS,
 		     MOCK_IGNORE, 3, 0x0, MOCK_IGNORE);
-	DISPLAYED_EQ("#3: back", VB2_SCREEN_RECOVERY_SELECT,
+	DISPLAYED_EQ("#3: firmware log", VB2_SCREEN_FIRMWARE_LOG,
+		     MOCK_IGNORE, MOCK_IGNORE, MOCK_IGNORE, MOCK_IGNORE);
+	/* #4: Back */
+	DISPLAYED_PASS();
+	DISPLAYED_EQ("advanced options", VB2_SCREEN_ADVANCED_OPTIONS,
+		     MOCK_IGNORE, 4, 0x0, MOCK_IGNORE);
+	DISPLAYED_EQ("#4: back", VB2_SCREEN_RECOVERY_SELECT,
 		     MOCK_IGNORE, MOCK_IGNORE, MOCK_IGNORE, MOCK_IGNORE);
 	/* End of menu */
 	DISPLAYED_PASS();