diff options
Diffstat (limited to 'tests/vboot_legacy_menu_tests.c')
| -rw-r--r-- | tests/vboot_legacy_menu_tests.c | 2372 |
1 files changed, 2372 insertions, 0 deletions
diff --git a/tests/vboot_legacy_menu_tests.c b/tests/vboot_legacy_menu_tests.c new file mode 100644 index 00000000..dfcaaedd --- /dev/null +++ b/tests/vboot_legacy_menu_tests.c @@ -0,0 +1,2372 @@ +/* Copyright 2017 The Chromium OS Authors. All rights reserved. + * Use of this source code is governed by a BSD-style license that can be + * found in the LICENSE file. + * + * Tests legacy menu UI + */ + +#include "2common.h" +#include "2misc.h" +#include "2nvstorage.h" +#include "2secdata.h" +#include "2secdata_struct.h" +#include "host_common.h" +#include "load_kernel_fw.h" +#include "test_common.h" +#include "tss_constants.h" +#include "vboot_api.h" +#include "vboot_audio.h" +#include "vboot_display.h" +#include "vboot_kernel.h" +#include "vboot_struct.h" +#include "vboot_test.h" +#include "vboot_ui_legacy_menu_private.h" + +/* Mock data */ +static uint8_t shared_data[VB_SHARED_DATA_MIN_SIZE]; +static VbSharedDataHeader *shared = (VbSharedDataHeader *)shared_data; +static LoadKernelParams lkp; +static uint8_t workbuf[VB2_KERNEL_WORKBUF_RECOMMENDED_SIZE] + __attribute__((aligned(VB2_WORKBUF_ALIGN))); +static struct vb2_context *ctx; +static struct vb2_shared_data *sd; +static struct vb2_gbb_header gbb; +static struct vb2_secdata_fwmp *fwmp; + +static int shutdown_request_calls_left; +static int audio_looping_calls_left; +static vb2_error_t vbtlk_retval[5]; +static vb2_error_t vbtlk_last_retval; +static int vbtlk_retval_count; +static const vb2_error_t vbtlk_retval_fixed = 1002; +static int vbexlegacy_called; +static enum VbAltFwIndex_t altfw_num; +static int debug_info_displayed; +static int trust_ec; +static int virtdev_set; +static uint32_t virtdev_retval; +static uint32_t mock_keypress[64]; +static uint32_t mock_keyflags[64]; +static uint32_t mock_keypress_count; +static uint32_t mock_switches[8]; +static uint32_t mock_switches_count; +static int mock_switches_are_stuck; +static uint32_t screens_displayed[64]; +static uint32_t screens_count = 0; +static uint32_t beeps_played[64]; +static uint32_t beeps_count = 0; +static uint32_t mock_altfw_mask; +static int vbexaltfwmask_called; + +/* Reset mock data (for use before each test) */ +static void ResetMocks(void) +{ + memset(&shared_data, 0, sizeof(shared_data)); + + memset(&lkp, 0, sizeof(lkp)); + + TEST_SUCC(vb2api_init(workbuf, sizeof(workbuf), &ctx), + "vb2api_init failed"); + vb2_nv_init(ctx); + + sd = vb2_get_sd(ctx); + sd->vbsd = shared; + + /* CRC will be invalid after here, but nobody's checking */ + sd->status |= VB2_SD_STATUS_SECDATA_FWMP_INIT; + fwmp = (struct vb2_secdata_fwmp *)ctx->secdata_fwmp; + + memset(&gbb, 0, sizeof(gbb)); + + /* In recovery we have 50 keyscans per disk scan, this must be high. */ + shutdown_request_calls_left = 301; + audio_looping_calls_left = 60; + vbexlegacy_called = 0; + altfw_num = -100; + debug_info_displayed = 0; + trust_ec = 0; + virtdev_set = 0; + virtdev_retval = 0; + + vbtlk_last_retval = vbtlk_retval_fixed - VB_DISK_FLAG_FIXED; + memset(vbtlk_retval, 0, sizeof(vbtlk_retval)); + vbtlk_retval_count = 0; + + memset(screens_displayed, 0, sizeof(screens_displayed)); + screens_count = 0; + memset(beeps_played, 0, sizeof(beeps_played)); + beeps_count = 0; + + memset(mock_keypress, 0, sizeof(mock_keypress)); + memset(mock_keyflags, 0, sizeof(mock_keyflags)); + mock_keypress_count = 0; + + memset(mock_switches, 0, sizeof(mock_switches)); + mock_switches_count = 0; + mock_switches_are_stuck = 0; + + mock_altfw_mask = 3 << 1; /* This mask selects 1 and 2 */ + vbexaltfwmask_called = 0; +} + +static void ResetMocksForDeveloper(void) +{ + ResetMocks(); + sd->flags |= VB2_SD_FLAG_DEV_MODE_ENABLED; + shutdown_request_calls_left = -1; +} + +static void ResetMocksForManualRecovery(void) +{ + ResetMocks(); + sd->flags |= VB2_SD_FLAG_MANUAL_RECOVERY; + trust_ec = 1; +} + +/* Mock functions */ + +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++; + + return mock_altfw_mask; +} + +uint32_t VbExIsShutdownRequested(void) +{ + if (shutdown_request_calls_left == 0) + return 1; + else if (shutdown_request_calls_left > 0) + shutdown_request_calls_left--; + + return 0; +} + +uint32_t VbExKeyboardRead(void) +{ + return VbExKeyboardReadWithFlags(NULL); +} + +uint32_t VbExKeyboardReadWithFlags(uint32_t *key_flags) +{ + if (mock_keypress_count < ARRAY_SIZE(mock_keypress)) { + if (key_flags != NULL) + *key_flags = mock_keyflags[mock_keypress_count]; + return mock_keypress[mock_keypress_count++]; + } else + return 0; +} + +uint32_t VbExGetSwitches(uint32_t request_mask) +{ + if (mock_switches_are_stuck) + return mock_switches[0] & request_mask; + if (mock_switches_count < ARRAY_SIZE(mock_switches)) + return mock_switches[mock_switches_count++] & request_mask; + else + return 0; +} + +vb2_error_t VbExLegacy(enum VbAltFwIndex_t _altfw_num) +{ + vbexlegacy_called++; + altfw_num = _altfw_num; + + return 0; +} + +int vb2ex_ec_trusted(void) +{ + return trust_ec; +} + +int vb2_audio_looping(void) +{ + if (audio_looping_calls_left == 0) + return 0; + else if (audio_looping_calls_left > 0) + audio_looping_calls_left--; + + return 1; +} + +vb2_error_t VbTryLoadKernel(struct vb2_context *c, uint32_t get_info_flags) +{ + if (vbtlk_retval_count < ARRAY_SIZE(vbtlk_retval) && + vbtlk_retval[vbtlk_retval_count] != 0) + vbtlk_last_retval = vbtlk_retval[vbtlk_retval_count++]; + return vbtlk_last_retval + get_info_flags; +} + +vb2_error_t VbDisplayScreen(struct vb2_context *c, uint32_t screen, int force, + const VbScreenData *data) +{ + if (screens_count < ARRAY_SIZE(screens_displayed)) + screens_displayed[screens_count++] = screen; + printf("VbDisplayScreen: screens_displayed[%d] = %#x\n", + screens_count - 1, screen); + return VB2_SUCCESS; +} + +vb2_error_t VbDisplayMenu(struct vb2_context *c, uint32_t screen, int force, + uint32_t selected_index, uint32_t disabled_idx_mask) +{ + if (screens_count < ARRAY_SIZE(screens_displayed)) + screens_displayed[screens_count++] = screen; + else + printf("Ran out of screens_displayed entries!\n"); + printf("VbDisplayMenu: screens_displayed[%d] = %#x," + " selected_index = %u, disabled_idx_mask = %#x\n", + screens_count - 1, screen, + selected_index, disabled_idx_mask); + + return VB2_SUCCESS; +} + +vb2_error_t VbDisplayDebugInfo(struct vb2_context *c) +{ + debug_info_displayed = 1; + return VB2_SUCCESS; +} + +vb2_error_t VbExBeep(uint32_t msec, uint32_t frequency) +{ + if (beeps_count < ARRAY_SIZE(beeps_played)) + beeps_played[beeps_count++] = frequency; + printf("VbExBeep: beeps_played[%d] = %dHz for %dms\n", + beeps_count - 1, frequency, msec); + return VB2_SUCCESS; +} + +vb2_error_t vb2_enable_developer_mode(struct vb2_context *c) +{ + virtdev_set = 1; + return virtdev_retval; +} + +/* Tests */ + +/* + * VbBootNormal tests: Please see VbBootTest in vboot_legacy_clamshell_tests.c + */ + +static void VbBootDevTest(void) +{ + int i; + + printf("Testing VbBootDeveloperLegacyMenu()...\n"); + + /* Proceed after timeout */ + ResetMocksForDeveloper(); + TEST_EQ(VbBootDeveloperLegacyMenu(ctx), vbtlk_retval_fixed, "Timeout"); + TEST_EQ(screens_displayed[0], VB_SCREEN_DEVELOPER_WARNING_MENU, + " warning screen"); + TEST_EQ(screens_displayed[1], VB_SCREEN_BLANK, " final blank screen"); + TEST_EQ(screens_count, 2, " no extra screens"); + TEST_EQ(vb2_nv_get(ctx, VB2_NV_RECOVERY_REQUEST), 0, + " recovery reason"); + TEST_EQ(debug_info_displayed, 0, " no debug info"); + TEST_EQ(audio_looping_calls_left, 0, " used up audio"); + TEST_EQ(beeps_count, 0, " no beeps on normal boot"); + + /* Proceed to legacy after timeout if GBB flag set */ + ResetMocksForDeveloper(); + gbb.flags |= VB2_GBB_FLAG_DEFAULT_DEV_BOOT_LEGACY | + VB2_GBB_FLAG_FORCE_DEV_BOOT_LEGACY; + TEST_EQ(VbBootDeveloperLegacyMenu(ctx), vbtlk_retval_fixed, + "default legacy GBB"); + TEST_EQ(vbexlegacy_called, 1, " try legacy"); + TEST_EQ(altfw_num, 0, " check altfw_num"); + TEST_EQ(screens_displayed[0], VB_SCREEN_DEVELOPER_WARNING_MENU, + " warning screen"); + TEST_EQ(screens_displayed[1], VB_SCREEN_BLANK, " blank (error flash)"); + TEST_EQ(screens_displayed[2], VB_SCREEN_DEVELOPER_WARNING_MENU, + " warning screen"); + TEST_EQ(screens_displayed[3], VB_SCREEN_BLANK, " final blank screen"); + TEST_EQ(screens_count, 4, " no extra screens"); + TEST_EQ(audio_looping_calls_left, 0, " used up audio"); + TEST_EQ(beeps_count, 1, " error beep: legacy BIOS not found"); + TEST_EQ(beeps_played[0], 200, " low-frequency error beep"); + + /* Proceed to legacy after timeout if boot legacy and default boot + * legacy are set */ + ResetMocksForDeveloper(); + vb2_nv_set(ctx, VB2_NV_DEV_DEFAULT_BOOT, VB2_DEV_DEFAULT_BOOT_LEGACY); + vb2_nv_set(ctx, VB2_NV_DEV_BOOT_LEGACY, 1); + TEST_EQ(VbBootDeveloperLegacyMenu(ctx), vbtlk_retval_fixed, + "default legacy NV"); + TEST_EQ(vbexlegacy_called, 1, " try legacy"); + TEST_EQ(altfw_num, 0, " check altfw_num"); + TEST_EQ(screens_displayed[0], VB_SCREEN_DEVELOPER_WARNING_MENU, + " warning screen"); + TEST_EQ(screens_displayed[1], VB_SCREEN_BLANK, " blank (error flash)"); + TEST_EQ(screens_displayed[2], VB_SCREEN_DEVELOPER_WARNING_MENU, + " warning screen"); + TEST_EQ(screens_displayed[3], VB_SCREEN_BLANK, " final blank screen"); + TEST_EQ(screens_count, 4, " no extra screens"); + TEST_EQ(audio_looping_calls_left, 0, " used up audio"); + TEST_EQ(beeps_count, 1, " error beep: legacy BIOS not found"); + TEST_EQ(beeps_played[0], 200, " low-frequency error beep"); + + /* Proceed to legacy boot mode only if enabled */ + ResetMocksForDeveloper(); + vb2_nv_set(ctx, VB2_NV_DEV_DEFAULT_BOOT, VB2_DEV_DEFAULT_BOOT_LEGACY); + TEST_EQ(VbBootDeveloperLegacyMenu(ctx), vbtlk_retval_fixed, + "default legacy not enabled"); + TEST_EQ(vbexlegacy_called, 0, " not legacy"); + TEST_EQ(vb2_nv_get(ctx, VB2_NV_RECOVERY_REQUEST), 0, " no recovery"); + TEST_EQ(screens_displayed[0], VB_SCREEN_DEVELOPER_WARNING_MENU, + " warning screen"); + TEST_EQ(screens_displayed[1], VB_SCREEN_BLANK, " blank (error flash)"); + TEST_EQ(screens_displayed[2], VB_SCREEN_DEVELOPER_WARNING_MENU, + " warning screen"); + TEST_EQ(screens_displayed[3], VB_SCREEN_BLANK, " final blank screen"); + TEST_EQ(screens_count, 4, " no extra screens"); + TEST_EQ(audio_looping_calls_left, 0, " used up audio"); + TEST_EQ(beeps_count, 2, " error beeps: legacy boot not enabled"); + TEST_EQ(beeps_played[0], 400, " first error beep"); + TEST_EQ(beeps_played[1], 400, " second error beep"); + + /* Proceed to usb after timeout if boot usb and default boot + * usb are set */ + ResetMocksForDeveloper(); + vb2_nv_set(ctx, VB2_NV_DEV_DEFAULT_BOOT, VB2_DEV_DEFAULT_BOOT_USB); + vb2_nv_set(ctx, VB2_NV_DEV_BOOT_USB, 1); + vbtlk_retval[0] = VB2_SUCCESS - VB_DISK_FLAG_REMOVABLE; + TEST_EQ(VbBootDeveloperLegacyMenu(ctx), 0, "Ctrl+U USB"); + TEST_EQ(screens_displayed[0], VB_SCREEN_DEVELOPER_WARNING_MENU, + " warning screen"); + TEST_EQ(screens_displayed[1], VB_SCREEN_BLANK, " final blank screen"); + TEST_EQ(screens_count, 2, " no extra screens"); + TEST_EQ(audio_looping_calls_left, 0, " used up audio"); + TEST_EQ(beeps_count, 0, " no beeps on USB boot"); + + /* Proceed to usb boot mode only if enabled */ + ResetMocksForDeveloper(); + vb2_nv_set(ctx, VB2_NV_DEV_DEFAULT_BOOT, VB2_DEV_DEFAULT_BOOT_USB); + TEST_EQ(VbBootDeveloperLegacyMenu(ctx), vbtlk_retval_fixed, + "default USB not enabled"); + TEST_EQ(vb2_nv_get(ctx, VB2_NV_RECOVERY_REQUEST), 0, " no recovery"); + TEST_EQ(screens_displayed[0], VB_SCREEN_DEVELOPER_WARNING_MENU, + " warning screen"); + TEST_EQ(screens_displayed[1], VB_SCREEN_BLANK, " blank (error flash)"); + TEST_EQ(screens_displayed[2], VB_SCREEN_DEVELOPER_WARNING_MENU, + " warning screen"); + TEST_EQ(screens_displayed[3], VB_SCREEN_BLANK, " final blank screen"); + TEST_EQ(screens_count, 4, " no extra screens"); + TEST_EQ(audio_looping_calls_left, 0, " used up audio"); + TEST_EQ(beeps_count, 2, " error beeps: USB boot not enabled"); + TEST_EQ(beeps_played[0], 400, " first error beep"); + TEST_EQ(beeps_played[1], 400, " second error beep"); + + /* If no USB tries fixed disk */ + ResetMocksForDeveloper(); + vb2_nv_set(ctx, VB2_NV_DEV_BOOT_USB, 1); + vb2_nv_set(ctx, VB2_NV_DEV_DEFAULT_BOOT, VB2_DEV_DEFAULT_BOOT_USB); + TEST_EQ(VbBootDeveloperLegacyMenu(ctx), vbtlk_retval_fixed, + "default USB with no disk"); + TEST_EQ(vbexlegacy_called, 0, " not legacy"); + TEST_EQ(vb2_nv_get(ctx, VB2_NV_RECOVERY_REQUEST), 0, " no recovery"); + TEST_EQ(screens_displayed[0], VB_SCREEN_DEVELOPER_WARNING_MENU, + " warning screen"); + TEST_EQ(screens_displayed[1], VB_SCREEN_BLANK, " blank (error flash)"); + TEST_EQ(screens_displayed[2], VB_SCREEN_DEVELOPER_WARNING_MENU, + " warning screen"); + TEST_EQ(screens_displayed[3], VB_SCREEN_BLANK, " final blank screen"); + TEST_EQ(screens_count, 4, " no extra screens"); + TEST_EQ(audio_looping_calls_left, 0, " used up audio"); + TEST_EQ(beeps_count, 1, " error beep: USB not found"); + TEST_EQ(beeps_played[0], 200, " low-frequency error beep"); + + /* Shutdown requested in loop */ + ResetMocksForDeveloper(); + shutdown_request_calls_left = 2; + TEST_EQ(VbBootDeveloperLegacyMenu(ctx), VBERROR_SHUTDOWN_REQUESTED, + "Shutdown requested"); + TEST_EQ(vb2_nv_get(ctx, VB2_NV_RECOVERY_REQUEST), 0, " no recovery"); + TEST_EQ(debug_info_displayed, 0, " no debug info"); + TEST_EQ(screens_displayed[0], VB_SCREEN_DEVELOPER_WARNING_MENU, + " warning screen"); + TEST_EQ(screens_displayed[1], VB_SCREEN_BLANK, " final blank screen"); + TEST_EQ(screens_count, 2, " no extra screens"); + TEST_NEQ(audio_looping_calls_left, 0, " aborts audio"); + TEST_EQ(beeps_count, 0, " no beep on shutdown"); + + /* + * Pushing power should shut down the DUT because default + * selection is power off + */ + ResetMocksForDeveloper(); + mock_keypress[0] = VB_BUTTON_POWER_SHORT_PRESS; + TEST_EQ(VbBootDeveloperLegacyMenu(ctx), VBERROR_SHUTDOWN_REQUESTED, + "dev warning menu: default to power off"); + TEST_EQ(vb2_nv_get(ctx, VB2_NV_RECOVERY_REQUEST), 0, " no recovery"); + TEST_EQ(screens_displayed[0], VB_SCREEN_DEVELOPER_WARNING_MENU, + " warning screen"); + TEST_EQ(screens_displayed[1], VB_SCREEN_BLANK, " final blank screen"); + TEST_EQ(screens_count, 2, " no extra screens"); + TEST_NEQ(audio_looping_calls_left, 0, " aborts audio"); + TEST_EQ(beeps_count, 0, " no beep on shutdown"); + + /* Selecting Power Off in developer options. */ + ResetMocksForDeveloper(); + i = 0; + mock_keypress[i++] = VB_BUTTON_VOL_UP_SHORT_PRESS; // Enable OS Verif + mock_keypress[i++] = VB_BUTTON_VOL_UP_SHORT_PRESS; // Show Debug Info + mock_keypress[i++] = VB_BUTTON_VOL_UP_SHORT_PRESS; // Developer Options + mock_keypress[i++] = VB_BUTTON_POWER_SHORT_PRESS; + mock_keypress[i++] = VB_BUTTON_VOL_DOWN_SHORT_PRESS; // Cancel + mock_keypress[i++] = VB_BUTTON_VOL_DOWN_SHORT_PRESS; // Power Off + mock_keypress[i++] = VB_BUTTON_POWER_SHORT_PRESS; + TEST_EQ(VbBootDeveloperLegacyMenu(ctx), VBERROR_SHUTDOWN_REQUESTED, + "Power Off in DEVELOPER"); + TEST_EQ(debug_info_displayed, 0, " no debug info"); + TEST_EQ(vbexlegacy_called, 0, " not legacy"); + TEST_NEQ(audio_looping_calls_left, 0, " aborts audio"); + TEST_EQ(vb2_nv_get(ctx, VB2_NV_RECOVERY_REQUEST), 0, " no recovery"); + i = 0; + TEST_EQ(screens_displayed[i++], VB_SCREEN_DEVELOPER_WARNING_MENU, + " dev warning menu: power off"); + TEST_EQ(screens_displayed[i++], VB_SCREEN_DEVELOPER_WARNING_MENU, + " dev warning menu: enable root verification"); + TEST_EQ(screens_displayed[i++], VB_SCREEN_DEVELOPER_WARNING_MENU, + " dev warning menu: show debug info"); + TEST_EQ(screens_displayed[i++], VB_SCREEN_DEVELOPER_WARNING_MENU, + " dev warning menu: developer options"); + TEST_EQ(screens_displayed[i++], VB_SCREEN_DEVELOPER_MENU, + " dev menu: disk boot"); + TEST_EQ(screens_displayed[i++], VB_SCREEN_DEVELOPER_MENU, + " dev menu: cancel"); + TEST_EQ(screens_displayed[i++], VB_SCREEN_DEVELOPER_MENU, + " dev menu: power off"); + TEST_EQ(screens_displayed[i++], VB_SCREEN_BLANK," final blank screen"); + TEST_EQ(screens_count, i, " no extra screens"); + TEST_EQ(beeps_count, 0, " no beeps on USB boot"); + + /* Pressing ENTER is equivalent to power button. */ + ResetMocksForDeveloper(); + mock_keypress[0] = VB_KEY_ENTER; + TEST_EQ(VbBootDeveloperLegacyMenu(ctx), VBERROR_SHUTDOWN_REQUESTED, + "dev warning menu: ENTER is power button"); + TEST_EQ(screens_displayed[0], VB_SCREEN_DEVELOPER_WARNING_MENU, + " warning screen"); + TEST_EQ(screens_displayed[1], VB_SCREEN_BLANK, " final blank screen"); + TEST_EQ(screens_count, 2, " no extra screens"); + TEST_NEQ(audio_looping_calls_left, 0, " aborts audio"); + TEST_EQ(beeps_count, 0, " no beep on shutdown"); + + /* ENTER functionality is unaffected by GBB flag in detachable UI. */ + ResetMocksForDeveloper(); + gbb.flags |= VB2_GBB_FLAG_ENTER_TRIGGERS_TONORM; + mock_keypress[0] = VB_KEY_ENTER; + TEST_EQ(VbBootDeveloperLegacyMenu(ctx), VBERROR_SHUTDOWN_REQUESTED, + "dev warning menu: ENTER unaffected by GBB"); + TEST_EQ(screens_displayed[0], VB_SCREEN_DEVELOPER_WARNING_MENU, + " warning screen"); + TEST_EQ(screens_displayed[1], VB_SCREEN_BLANK, " final blank screen"); + TEST_EQ(screens_count, 2, " no extra screens"); + TEST_NEQ(audio_looping_calls_left, 0, " aborts audio"); + TEST_EQ(beeps_count, 0, " no beep on shutdown"); + + /* Pressing SPACE or VolUp+Down combo has no effect in dev mode */ + ResetMocksForDeveloper(); + mock_keypress[0] = ' '; + mock_keypress[1] = VB_BUTTON_VOL_UP_DOWN_COMBO_PRESS; + mock_keypress[2] = VB_BUTTON_POWER_SHORT_PRESS; // select Power Off + TEST_EQ(VbBootDeveloperLegacyMenu(ctx), VBERROR_SHUTDOWN_REQUESTED, + "SPACE or VolUp+Down have no effect"); + TEST_EQ(vb2_nv_get(ctx, VB2_NV_RECOVERY_REQUEST), 0, " no recovery"); + TEST_EQ(debug_info_displayed, 0, " no debug info"); + TEST_EQ(screens_displayed[0], VB_SCREEN_DEVELOPER_WARNING_MENU, + " warning screen"); + TEST_EQ(screens_displayed[1], VB_SCREEN_BLANK, " final blank screen"); + TEST_EQ(screens_count, 2, " no extra screens"); + TEST_NEQ(audio_looping_calls_left, 0, " aborts audio"); + TEST_EQ(beeps_count, 0, " no beep on shutdown"); + + /* Disable developer mode */ + ResetMocksForDeveloper(); + sd->flags |= VB2_SD_FLAG_DEV_MODE_ENABLED; + mock_keypress[0] = VB_BUTTON_VOL_UP_SHORT_PRESS; + mock_keypress[1] = VB_BUTTON_POWER_SHORT_PRESS; + mock_keypress[2] = VB_BUTTON_POWER_SHORT_PRESS; + TEST_EQ(VbBootDeveloperLegacyMenu(ctx), VBERROR_REBOOT_REQUIRED, + "disable developer mode"); + TEST_EQ(vb2_nv_get(ctx, VB2_NV_RECOVERY_REQUEST), 0, " no recovery"); + TEST_EQ(screens_displayed[0], VB_SCREEN_DEVELOPER_WARNING_MENU, + " warning screen"); + TEST_EQ(screens_displayed[1], VB_SCREEN_DEVELOPER_WARNING_MENU, + " warning screen"); + TEST_EQ(screens_displayed[2], VB_SCREEN_DEVELOPER_TO_NORM_MENU, + " tonorm screen"); + TEST_EQ(screens_displayed[3], VB_SCREEN_TO_NORM_CONFIRMED, + " confirm screen"); + TEST_EQ(screens_displayed[4], VB_SCREEN_BLANK, " final blank screen"); + TEST_EQ(screens_count, 5, " no extra screens"); + TEST_EQ(vb2_nv_get(ctx, VB2_NV_DISABLE_DEV_REQUEST), 1, + " disable dev request"); + TEST_EQ(debug_info_displayed, 0, " no debug info"); + TEST_NEQ(audio_looping_calls_left, 0, " aborts audio"); + TEST_EQ(beeps_count, 0, " no beep on reboot"); + + /* Tonorm ignored if GBB forces dev switch on */ + ResetMocksForDeveloper(); + sd->flags |= VB2_SD_FLAG_DEV_MODE_ENABLED; + gbb.flags |= VB2_GBB_FLAG_FORCE_DEV_SWITCH_ON; + mock_keypress[0] = VB_BUTTON_VOL_UP_SHORT_PRESS; + mock_keypress[1] = VB_BUTTON_POWER_SHORT_PRESS; + mock_keypress[2] = VB_BUTTON_POWER_SHORT_PRESS; + TEST_EQ(VbBootDeveloperLegacyMenu(ctx), vbtlk_retval_fixed, + "Can't tonorm gbb-dev"); + TEST_EQ(vb2_nv_get(ctx, VB2_NV_RECOVERY_REQUEST), 0, " no recovery"); + TEST_EQ(screens_displayed[0], VB_SCREEN_DEVELOPER_WARNING_MENU, + " warning screen: power off"); + TEST_EQ(screens_displayed[1], VB_SCREEN_DEVELOPER_WARNING_MENU, + " wanring screen: enable verification"); + TEST_EQ(screens_displayed[2], VB_SCREEN_DEVELOPER_TO_NORM_MENU, + " tonorm screen: confirm"); + TEST_EQ(screens_displayed[3], VB_SCREEN_BLANK, + " blank (error flash)"); + TEST_EQ(screens_displayed[4], VB_SCREEN_DEVELOPER_TO_NORM_MENU, + " tonorm screen: confirm"); + TEST_EQ(screens_displayed[5], VB_SCREEN_BLANK, " final blank screen"); + TEST_EQ(screens_count, 6, " no extra screens"); + TEST_EQ(audio_looping_calls_left, 0, " audio timeout"); + TEST_EQ(beeps_count, 2, " played error beeps"); + TEST_EQ(beeps_played[0], 400, " first beep"); + TEST_EQ(beeps_played[1], 400, " second beep"); + + /* Shutdown requested at tonorm screen */ + ResetMocksForDeveloper(); + sd->flags |= VB2_SD_FLAG_DEV_MODE_ENABLED; + mock_keypress[0] = VB_BUTTON_VOL_UP_SHORT_PRESS; + mock_keypress[1] = VB_BUTTON_POWER_SHORT_PRESS; + shutdown_request_calls_left = 2; + TEST_EQ(VbBootDeveloperLegacyMenu(ctx), VBERROR_SHUTDOWN_REQUESTED, + "Shutdown requested at tonorm"); + TEST_EQ(vb2_nv_get(ctx, VB2_NV_RECOVERY_REQUEST), 0, " no recovery"); + TEST_EQ(screens_displayed[0], VB_SCREEN_DEVELOPER_WARNING_MENU, + " developer warning screen: power off"); + TEST_EQ(screens_displayed[1], VB_SCREEN_DEVELOPER_WARNING_MENU, + " developer warning screen: enable root verification"); + TEST_EQ(screens_displayed[2], VB_SCREEN_DEVELOPER_TO_NORM_MENU, + " developer warning screen: power off"); + TEST_EQ(screens_displayed[3], VB_SCREEN_BLANK, " final blank screen"); + TEST_EQ(screens_count, 4, " no extra screens"); + TEST_EQ(beeps_count, 0, " no beeps on shutdown"); + + /* Ctrl+D dismisses warning */ + ResetMocksForDeveloper(); + mock_keypress[0] = VB_KEY_CTRL('D'); + TEST_EQ(VbBootDeveloperLegacyMenu(ctx), vbtlk_retval_fixed, "Ctrl+D"); + TEST_EQ(vb2_nv_get(ctx, VB2_NV_RECOVERY_REQUEST), 0, + " recovery reason"); + TEST_NEQ(audio_looping_calls_left, 0, " aborts audio"); + TEST_EQ(vbexlegacy_called, 0, " not legacy"); + TEST_EQ(debug_info_displayed, 0, " no debug info"); + TEST_EQ(screens_displayed[0], VB_SCREEN_DEVELOPER_WARNING_MENU, + " warning screen"); + TEST_EQ(screens_displayed[1], VB_SCREEN_BLANK, " final blank screen"); + TEST_EQ(screens_count, 2, " no extra screens"); + TEST_EQ(beeps_count, 0, " no beeps on normal boot"); + + /* Ctrl+D doesn't boot legacy even if GBB flag is set */ + ResetMocksForDeveloper(); + mock_keypress[0] = VB_KEY_CTRL('D'); + gbb.flags |= VB2_GBB_FLAG_DEFAULT_DEV_BOOT_LEGACY; + TEST_EQ(VbBootDeveloperLegacyMenu(ctx), vbtlk_retval_fixed, "Ctrl+D"); + TEST_EQ(vbexlegacy_called, 0, " not legacy"); + + /* Volume-down long press shortcut acts like Ctrl+D */ + ResetMocksForDeveloper(); + mock_keypress[0] = VB_BUTTON_VOL_DOWN_LONG_PRESS; + TEST_EQ(VbBootDeveloperLegacyMenu(ctx), vbtlk_retval_fixed, + "VolDown long press"); + TEST_EQ(vb2_nv_get(ctx, VB2_NV_RECOVERY_REQUEST), 0, + " recovery reason"); + TEST_NEQ(audio_looping_calls_left, 0, " aborts audio"); + TEST_EQ(vbexlegacy_called, 0, " not legacy"); + TEST_EQ(screens_displayed[0], VB_SCREEN_DEVELOPER_WARNING_MENU, + " warning screen"); + TEST_EQ(screens_displayed[1], VB_SCREEN_BLANK, " final blank screen"); + TEST_EQ(screens_count, 2, " no extra screens"); + TEST_EQ(beeps_count, 0, " no beeps on normal boot"); + + /* Ctrl+L tries legacy boot mode only if enabled */ + ResetMocksForDeveloper(); + mock_keypress[0] = VB_KEY_CTRL('L'); + TEST_EQ(VbBootDeveloperLegacyMenu(ctx), vbtlk_retval_fixed, + "Ctrl+L normal"); + TEST_EQ(vbexlegacy_called, 0, " not legacy"); + TEST_EQ(vb2_nv_get(ctx, VB2_NV_RECOVERY_REQUEST), 0, " no recovery"); + TEST_EQ(debug_info_displayed, 0, " no debug info"); + TEST_EQ(audio_looping_calls_left, 0, " audio timed out"); + TEST_EQ(screens_displayed[0], VB_SCREEN_DEVELOPER_WARNING_MENU, + " warning screen"); + TEST_EQ(screens_displayed[1], VB_SCREEN_BLANK, " blank (error flash)"); + TEST_EQ(screens_displayed[2], VB_SCREEN_DEVELOPER_WARNING_MENU, + " warning screen"); + TEST_EQ(screens_displayed[3], VB_SCREEN_BLANK, " final blank screen"); + TEST_EQ(screens_count, 4, " no extra screens"); + TEST_EQ(beeps_count, 2, " played error beeps"); + TEST_EQ(beeps_played[0], 400, " first beep"); + TEST_EQ(beeps_played[1], 400, " second beep"); + + /* Ctrl+L boots legacy if enabled by GBB flag */ + ResetMocksForDeveloper(); + gbb.flags |= VB2_GBB_FLAG_FORCE_DEV_BOOT_LEGACY; + mock_keypress[0] = VB_KEY_CTRL('L'); + mock_keypress[1] = VB_BUTTON_POWER_SHORT_PRESS; + TEST_EQ(VbBootDeveloperLegacyMenu(ctx), vbtlk_retval_fixed, + "Ctrl+L force legacy"); + TEST_EQ(vbexlegacy_called, 1, " try legacy"); + TEST_EQ(altfw_num, 1, " check altfw_num"); + TEST_EQ(screens_displayed[0], VB_SCREEN_DEVELOPER_WARNING_MENU, + " warning screen"); + TEST_EQ(screens_displayed[1], VB_SCREEN_ALT_FW_MENU, " altfw"); + TEST_EQ(screens_displayed[2], VB_SCREEN_BLANK, " blank (error flash)"); + TEST_EQ(screens_displayed[3], VB_SCREEN_ALT_FW_MENU, " altfw"); + TEST_EQ(screens_displayed[4], VB_SCREEN_BLANK, " blank (error flash)"); + TEST_EQ(screens_count, 5, " no extra screens"); + TEST_EQ(beeps_count, 1, " error beep: legacy BIOS not found"); + TEST_EQ(beeps_played[0], 200, " low-frequency error beep"); + + /* Ctrl+L boots legacy if enabled by NVRAM */ + ResetMocksForDeveloper(); + vb2_nv_set(ctx, VB2_NV_DEV_BOOT_LEGACY, 1); + mock_keypress[0] = VB_KEY_CTRL('L'); + mock_keypress[1] = VB_BUTTON_POWER_SHORT_PRESS; + TEST_EQ(VbBootDeveloperLegacyMenu(ctx), vbtlk_retval_fixed, + "Ctrl+L nv legacy"); + TEST_EQ(vbexlegacy_called, 1, " try legacy"); + TEST_EQ(vb2_nv_get(ctx, VB2_NV_RECOVERY_REQUEST), 0, " no recovery"); + TEST_EQ(screens_displayed[0], VB_SCREEN_DEVELOPER_WARNING_MENU, + " warning screen"); + TEST_EQ(screens_displayed[1], VB_SCREEN_ALT_FW_MENU, " altfw"); + TEST_EQ(screens_displayed[2], VB_SCREEN_BLANK, " blank (error flash)"); + TEST_EQ(screens_displayed[3], VB_SCREEN_ALT_FW_MENU, " altfw"); + TEST_EQ(screens_displayed[4], VB_SCREEN_BLANK, " blank (error flash)"); + TEST_EQ(screens_count, 5, " no extra screens"); + TEST_EQ(beeps_count, 1, " error beep: legacy BIOS not found"); + TEST_EQ(beeps_played[0], 200, " low-frequency error beep"); + + /* Ctrl+L boots legacy if enabled by FWMP */ + ResetMocksForDeveloper(); + fwmp->flags |= VB2_SECDATA_FWMP_DEV_ENABLE_LEGACY; + mock_keypress[0] = VB_KEY_CTRL('L'); + mock_keypress[1] = VB_BUTTON_POWER_SHORT_PRESS; + TEST_EQ(VbBootDeveloperLegacyMenu(ctx), vbtlk_retval_fixed, + "Ctrl+L fwmp legacy"); + TEST_EQ(vbexlegacy_called, 1, " fwmp legacy"); + TEST_EQ(altfw_num, 1, " check altfw_num"); + TEST_EQ(screens_displayed[0], VB_SCREEN_DEVELOPER_WARNING_MENU, + " warning screen"); + TEST_EQ(screens_displayed[1], VB_SCREEN_ALT_FW_MENU, " altfw"); + TEST_EQ(screens_displayed[2], VB_SCREEN_BLANK, " blank (error flash)"); + TEST_EQ(screens_displayed[3], VB_SCREEN_ALT_FW_MENU, " altfw"); + TEST_EQ(screens_displayed[4], VB_SCREEN_BLANK, " final blank screen"); + TEST_EQ(screens_count, 5, " no extra screens"); + TEST_EQ(beeps_count, 1, " error beep: legacy BIOS not found"); + TEST_EQ(beeps_played[0], 200, " low-frequency error beep"); + + /* TODO(sjg@chromium.org): Add a test for there being no bootloaders */ + + /* Ctrl+U boots USB only if enabled */ + ResetMocksForDeveloper(); + mock_keypress[0] = VB_KEY_CTRL('U'); + TEST_EQ(VbBootDeveloperLegacyMenu(ctx), vbtlk_retval_fixed, + "Ctrl+U not enabled"); + TEST_EQ(vbexlegacy_called, 0, " not legacy"); + TEST_EQ(debug_info_displayed, 0, " no debug info"); + TEST_EQ(vb2_nv_get(ctx, VB2_NV_RECOVERY_REQUEST), 0, " no recovery"); + TEST_EQ(audio_looping_calls_left, 0, " audio timed out"); + TEST_EQ(screens_displayed[0], VB_SCREEN_DEVELOPER_WARNING_MENU, + " warning screen"); + TEST_EQ(screens_displayed[1], VB_SCREEN_BLANK, " blank (error flash)"); + TEST_EQ(screens_displayed[2], VB_SCREEN_DEVELOPER_WARNING_MENU, + " warning screen"); + TEST_EQ(screens_displayed[3], VB_SCREEN_BLANK, " final blank screen"); + TEST_EQ(screens_count, 4, " no extra screens"); + TEST_EQ(beeps_count, 2, " played error beeps"); + TEST_EQ(beeps_played[0], 400, " first beep"); + TEST_EQ(beeps_played[1], 400, " second beep"); + + /* Ctrl+U enabled, with good USB boot */ + ResetMocksForDeveloper(); + vb2_nv_set(ctx, VB2_NV_DEV_BOOT_USB, 1); + mock_keypress[0] = VB_KEY_CTRL('U'); + vbtlk_retval[0] = VB2_SUCCESS - VB_DISK_FLAG_REMOVABLE; + TEST_EQ(VbBootDeveloperLegacyMenu(ctx), VB2_SUCCESS, "Ctrl+U USB"); + TEST_EQ(vbexlegacy_called, 0, " not legacy"); + TEST_EQ(debug_info_displayed, 0, " no debug info"); + TEST_EQ(vb2_nv_get(ctx, VB2_NV_RECOVERY_REQUEST), 0, " no recovery"); + TEST_NEQ(audio_looping_calls_left, 0, " aborts audio"); + TEST_EQ(screens_displayed[0], VB_SCREEN_DEVELOPER_WARNING_MENU, + " warning screen"); + TEST_EQ(screens_displayed[1], VB_SCREEN_BLANK, " final blank screen"); + TEST_EQ(screens_count, 2, " no extra screens"); + TEST_EQ(beeps_count, 0, " no beeps on USB boot"); + + /* Ctrl+U enabled, without valid USB */ + ResetMocksForDeveloper(); + vb2_nv_set(ctx, VB2_NV_DEV_BOOT_USB, 1); + mock_keypress[0] = VB_KEY_CTRL('U'); + TEST_EQ(VbBootDeveloperLegacyMenu(ctx), vbtlk_retval_fixed, + "Ctrl+U without valid USB"); + TEST_EQ(vbexlegacy_called, 0, " not legacy"); + TEST_EQ(vb2_nv_get(ctx, VB2_NV_RECOVERY_REQUEST), 0, " no recovery"); + TEST_EQ(debug_info_displayed, 0, " no debug info"); + TEST_EQ(screens_displayed[0], VB_SCREEN_DEVELOPER_WARNING_MENU, + " warning screen"); + TEST_EQ(screens_displayed[1], VB_SCREEN_BLANK, " blank (error flash)"); + TEST_EQ(screens_displayed[2], VB_SCREEN_DEVELOPER_WARNING_MENU, + " warning screen"); + TEST_EQ(screens_displayed[3], VB_SCREEN_BLANK, " final blank screen"); + TEST_EQ(screens_count, 4, " no extra screens"); + TEST_EQ(audio_looping_calls_left, 0, " used up audio"); + TEST_EQ(beeps_count, 1, " error beep: USB not found"); + TEST_EQ(beeps_played[0], 200, " low-frequency error beep"); + + /* Ctrl+U enabled via GBB */ + ResetMocksForDeveloper(); + gbb.flags |= VB2_GBB_FLAG_FORCE_DEV_BOOT_USB; + mock_keypress[0] = VB_KEY_CTRL('U'); + vbtlk_retval[0] = VB2_SUCCESS - VB_DISK_FLAG_REMOVABLE; + TEST_EQ(VbBootDeveloperLegacyMenu(ctx), VB2_SUCCESS, + "Ctrl+U force USB"); + TEST_NEQ(audio_looping_calls_left, 0, " aborts audio"); + TEST_EQ(screens_displayed[0], VB_SCREEN_DEVELOPER_WARNING_MENU, + " warning screen"); + TEST_EQ(screens_displayed[1], VB_SCREEN_BLANK, " final blank screen"); + TEST_EQ(screens_count, 2, " no extra screens"); + TEST_EQ(beeps_count, 0, " no beeps on USB boot"); + + /* Ctrl+U enabled via FWMP */ + ResetMocksForDeveloper(); + fwmp->flags |= VB2_SECDATA_FWMP_DEV_ENABLE_USB; + mock_keypress[0] = VB_KEY_CTRL('U'); + vbtlk_retval[0] = VB2_SUCCESS - VB_DISK_FLAG_REMOVABLE; + TEST_EQ(VbBootDeveloperLegacyMenu(ctx), VB2_SUCCESS, + "Ctrl+U force USB"); + TEST_NEQ(audio_looping_calls_left, 0, " aborts audio"); + TEST_EQ(screens_displayed[0], VB_SCREEN_DEVELOPER_WARNING_MENU, + " warning screen"); + TEST_EQ(screens_displayed[1], VB_SCREEN_BLANK, " final blank screen"); + TEST_EQ(screens_count, 2, " no extra screens"); + TEST_EQ(beeps_count, 0, " no beeps on USB boot"); + + /* If no valid USB, eventually times out and tries fixed disk */ + ResetMocksForDeveloper(); + vb2_nv_set(ctx, VB2_NV_DEV_BOOT_USB, 1); + mock_keypress[0] = VB_KEY_CTRL('U'); + TEST_EQ(VbBootDeveloperLegacyMenu(ctx), vbtlk_retval_fixed, + "Ctrl+U failed - no USB"); + TEST_EQ(vbexlegacy_called, 0, " not legacy"); + TEST_EQ(vb2_nv_get(ctx, VB2_NV_RECOVERY_REQUEST), 0, " no recovery"); + TEST_EQ(audio_looping_calls_left, 0, " used up audio"); + TEST_EQ(screens_displayed[0], VB_SCREEN_DEVELOPER_WARNING_MENU, + " warning screen"); + TEST_EQ(screens_displayed[1], VB_SCREEN_BLANK, " blank (error flash)"); + TEST_EQ(screens_displayed[2], VB_SCREEN_DEVELOPER_WARNING_MENU, + " warning screen"); + TEST_EQ(screens_displayed[3], VB_SCREEN_BLANK, " final blank screen"); + TEST_EQ(screens_count, 4, " no extra screens"); + TEST_EQ(beeps_count, 1, " error beep: USB not found"); + TEST_EQ(beeps_played[0], 200, " low-frequency error beep"); + + /* Now go to USB boot through menus */ + ResetMocksForDeveloper(); + i = 0; + mock_keypress[i++] = VB_BUTTON_VOL_UP_SHORT_PRESS; // Enable OS Verif + mock_keypress[i++] = VB_BUTTON_VOL_UP_SHORT_PRESS; // Show Debug Info + mock_keypress[i++] = VB_BUTTON_VOL_UP_SHORT_PRESS; // Developer Options + mock_keypress[i++] = VB_BUTTON_POWER_SHORT_PRESS; + mock_keypress[i++] = VB_BUTTON_VOL_UP_SHORT_PRESS; // Boot From USB + mock_keypress[i++] = VB_BUTTON_POWER_SHORT_PRESS; + vb2_nv_set(ctx, VB2_NV_DEV_BOOT_USB, 1); + vbtlk_retval[0] = VB2_SUCCESS - VB_DISK_FLAG_REMOVABLE; + TEST_EQ(VbBootDeveloperLegacyMenu(ctx), VB2_SUCCESS, + "Menu selected USB boot"); + TEST_EQ(debug_info_displayed, 0, " no debug info"); + TEST_EQ(vbexlegacy_called, 0, " not legacy"); + TEST_NEQ(audio_looping_calls_left, 0, " aborts audio"); + TEST_EQ(vb2_nv_get(ctx, VB2_NV_RECOVERY_REQUEST), 0, " no recovery"); + i = 0; + TEST_EQ(screens_displayed[i++], VB_SCREEN_DEVELOPER_WARNING_MENU, + " dev warning menu: power off"); + TEST_EQ(screens_displayed[i++], VB_SCREEN_DEVELOPER_WARNING_MENU, + " dev warning menu: enable root verification"); + TEST_EQ(screens_displayed[i++], VB_SCREEN_DEVELOPER_WARNING_MENU, + " dev warning menu: show debug info"); + TEST_EQ(screens_displayed[i++], VB_SCREEN_DEVELOPER_WARNING_MENU, + " dev warning menu: developer options"); + TEST_EQ(screens_displayed[i++], VB_SCREEN_DEVELOPER_MENU, + " dev menu: disk boot"); + TEST_EQ(screens_displayed[i++], VB_SCREEN_DEVELOPER_MENU, + " dev menu: USB boot"); + TEST_EQ(screens_displayed[i++], VB_SCREEN_BLANK, " final blank screen"); + TEST_EQ(screens_count, i, " no extra screens"); + TEST_EQ(beeps_count, 0, " no beeps on USB boot"); + + /* If default USB, the option is preselected */ + ResetMocksForDeveloper(); + vb2_nv_set(ctx, VB2_NV_DEV_DEFAULT_BOOT, VB2_DEV_DEFAULT_BOOT_USB); + i = 0; + mock_keypress[i++] = VB_BUTTON_VOL_UP_SHORT_PRESS; // Enable OS Verif + mock_keypress[i++] = VB_BUTTON_VOL_UP_SHORT_PRESS; // Show Debug Info + mock_keypress[i++] = VB_BUTTON_VOL_UP_SHORT_PRESS; // Developer Options + mock_keypress[i++] = VB_BUTTON_POWER_SHORT_PRESS; + mock_keypress[i++] = VB_BUTTON_POWER_SHORT_PRESS; + vb2_nv_set(ctx, VB2_NV_DEV_BOOT_USB, 1); + vbtlk_retval[0] = VB2_SUCCESS - VB_DISK_FLAG_REMOVABLE; + TEST_EQ(VbBootDeveloperLegacyMenu(ctx), VB2_SUCCESS, + "Menu selected USB default boot"); + TEST_EQ(vbexlegacy_called, 0, " not legacy"); + TEST_NEQ(audio_looping_calls_left, 0, " aborts audio"); + TEST_EQ(vb2_nv_get(ctx, VB2_NV_RECOVERY_REQUEST), 0, " no recovery"); + i = 0; + TEST_EQ(screens_displayed[i++], VB_SCREEN_DEVELOPER_WARNING_MENU, + " dev warning menu: power off"); + TEST_EQ(screens_displayed[i++], VB_SCREEN_DEVELOPER_WARNING_MENU, + " dev warning menu: enable root verification"); + TEST_EQ(screens_displayed[i++], VB_SCREEN_DEVELOPER_WARNING_MENU, + " dev warning menu: show debug info"); + TEST_EQ(screens_displayed[i++], VB_SCREEN_DEVELOPER_WARNING_MENU, + " dev warning menu: developer options"); + TEST_EQ(screens_displayed[i++], VB_SCREEN_DEVELOPER_MENU, + " dev menu: USB boot"); + TEST_EQ(screens_displayed[i++], VB_SCREEN_BLANK," final blank screen"); + TEST_EQ(screens_count, i, " no extra screens"); + TEST_EQ(beeps_count, 0, " no beeps on USB boot"); + + /* If invalid USB, we still timeout after selecting it in menu */ + ResetMocksForDeveloper(); + i = 0; + mock_keypress[i++] = VB_BUTTON_VOL_UP_SHORT_PRESS; // Enable OS Verif + mock_keypress[i++] = VB_BUTTON_VOL_UP_SHORT_PRESS; // Show Debug Info + mock_keypress[i++] = VB_BUTTON_VOL_UP_SHORT_PRESS; // Developer Options + mock_keypress[i++] = VB_BUTTON_POWER_SHORT_PRESS; + mock_keypress[i++] = VB_BUTTON_VOL_UP_SHORT_PRESS; // Boot From USB + mock_keypress[i++] = VB_BUTTON_POWER_SHORT_PRESS; + vb2_nv_set(ctx, VB2_NV_DEV_BOOT_USB, 1); + TEST_EQ(VbBootDeveloperLegacyMenu(ctx), vbtlk_retval_fixed, + "Menu selected invalid USB boot"); + TEST_EQ(vbexlegacy_called, 0, " not legacy"); + TEST_EQ(audio_looping_calls_left, 0, " used up audio"); + TEST_EQ(vb2_nv_get(ctx, VB2_NV_RECOVERY_REQUEST), 0, " no recovery"); + i = 0; + TEST_EQ(screens_displayed[i++], VB_SCREEN_DEVELOPER_WARNING_MENU, + " dev warning menu: power off"); + TEST_EQ(screens_displayed[i++], VB_SCREEN_DEVELOPER_WARNING_MENU, + " dev warning menu: enable root verification"); + TEST_EQ(screens_displayed[i++], VB_SCREEN_DEVELOPER_WARNING_MENU, + " dev warning menu: show debug info"); + TEST_EQ(screens_displayed[i++], VB_SCREEN_DEVELOPER_WARNING_MENU, + " dev warning menu: developer options"); + TEST_EQ(screens_displayed[i++], VB_SCREEN_DEVELOPER_MENU, + " dev menu: disk boot"); + TEST_EQ(screens_displayed[i++], VB_SCREEN_DEVELOPER_MENU, + " dev menu: USB boot"); + TEST_EQ(screens_displayed[i++], VB_SCREEN_BLANK, " blank (flash)"); + TEST_EQ(screens_displayed[i++], VB_SCREEN_DEVELOPER_MENU, + " dev menu: USB boot"); + TEST_EQ(screens_displayed[i++], VB_SCREEN_BLANK," final blank screen"); + TEST_EQ(screens_count, i, " no extra screens"); + TEST_EQ(beeps_count, 1, " error beep: USB not found"); + TEST_EQ(beeps_played[0], 200, " low-frequency error beep"); + + /* If USB is disabled, we error flash/beep from the menu option */ + ResetMocksForDeveloper(); + i = 0; + mock_keypress[i++] = VB_BUTTON_VOL_UP_SHORT_PRESS; // Enable OS Verif + mock_keypress[i++] = VB_BUTTON_VOL_UP_SHORT_PRESS; // Show Debug Info + mock_keypress[i++] = VB_BUTTON_VOL_UP_SHORT_PRESS; // Developer Options + mock_keypress[i++] = VB_BUTTON_POWER_SHORT_PRESS; + mock_keypress[i++] = VB_BUTTON_VOL_UP_SHORT_PRESS; // Boot From USB + mock_keypress[i++] = VB_BUTTON_POWER_SHORT_PRESS; + TEST_EQ(VbBootDeveloperLegacyMenu(ctx), vbtlk_retval_fixed, + "Menu selected disabled USB boot"); + TEST_EQ(vbexlegacy_called, 0, " not legacy"); + TEST_EQ(audio_looping_calls_left, 0, " used up audio"); + TEST_EQ(vb2_nv_get(ctx, VB2_NV_RECOVERY_REQUEST), 0, " no recovery"); + i = 0; + TEST_EQ(screens_displayed[i++], VB_SCREEN_DEVELOPER_WARNING_MENU, + " dev warning menu: power off"); + TEST_EQ(screens_displayed[i++], VB_SCREEN_DEVELOPER_WARNING_MENU, + " dev warning menu: enable root verification"); + TEST_EQ(screens_displayed[i++], VB_SCREEN_DEVELOPER_WARNING_MENU, + " dev warning menu: show debug info"); + TEST_EQ(screens_displayed[i++], VB_SCREEN_DEVELOPER_WARNING_MENU, + " dev warning menu: developer options"); + TEST_EQ(screens_displayed[i++], VB_SCREEN_DEVELOPER_MENU, + " dev menu: disk boot"); + TEST_EQ(screens_displayed[i++], VB_SCREEN_DEVELOPER_MENU, + " dev menu: USB boot"); + TEST_EQ(screens_displayed[i++], VB_SCREEN_BLANK, " blank (flash)"); + TEST_EQ(screens_displayed[i++], VB_SCREEN_DEVELOPER_MENU, + " dev menu: USB boot"); + TEST_EQ(screens_displayed[i++], VB_SCREEN_BLANK, " final blank screen"); + TEST_EQ(screens_count, i, " no extra screens"); + TEST_EQ(beeps_count, 2, " played error beeps"); + TEST_EQ(beeps_played[0], 400, " first beep"); + TEST_EQ(beeps_played[1], 400, " second beep"); + + /* Boot Legacy via menu and default */ + ResetMocksForDeveloper(); + i = 0; + mock_keypress[i++] = VB_BUTTON_VOL_UP_SHORT_PRESS; // Enable OS Verif + mock_keypress[i++] = VB_BUTTON_VOL_UP_SHORT_PRESS; // Show Debug Info + mock_keypress[i++] = VB_BUTTON_VOL_UP_SHORT_PRESS; // Developer Options + mock_keypress[i++] = VB_BUTTON_POWER_SHORT_PRESS; // Enter altfw menu + mock_keypress[i++] = VB_BUTTON_POWER_SHORT_PRESS; // Select first option + mock_keypress[i++] = VB_BUTTON_POWER_SHORT_PRESS; // and again + vb2_nv_set(ctx, VB2_NV_DEV_BOOT_LEGACY, 1); + vb2_nv_set(ctx, VB2_NV_DEV_DEFAULT_BOOT, VB2_DEV_DEFAULT_BOOT_LEGACY); + TEST_EQ(VbBootDeveloperLegacyMenu(ctx), vbtlk_retval_fixed, + "Menu selected legacy boot"); + TEST_EQ(debug_info_displayed, 0, " no debug info"); + TEST_EQ(vbexlegacy_called, 2, " tried legacy boot twice"); + TEST_EQ(altfw_num, 0, " check altfw_num"); + TEST_EQ(audio_looping_calls_left, 0, " audio timeout"); + TEST_EQ(vb2_nv_get(ctx, VB2_NV_RECOVERY_REQUEST), 0, " no recovery"); + i = 0; + TEST_EQ(screens_displayed[i++], VB_SCREEN_DEVELOPER_WARNING_MENU, + " dev warning menu: power off"); + TEST_EQ(screens_displayed[i++], VB_SCREEN_DEVELOPER_WARNING_MENU, + " dev warning menu: enable root verification"); + TEST_EQ(screens_displayed[i++], VB_SCREEN_DEVELOPER_WARNING_MENU, + " dev warning menu: show debug info"); + TEST_EQ(screens_displayed[i++], VB_SCREEN_DEVELOPER_WARNING_MENU, + " dev warning menu: developer options"); + TEST_EQ(screens_displayed[i++], VB_SCREEN_DEVELOPER_MENU, + " dev menu: legacy boot"); + TEST_EQ(screens_displayed[i++], VB_SCREEN_ALT_FW_MENU, " altfw"); + TEST_EQ(screens_displayed[i++], VB_SCREEN_BLANK, " blank (flash)"); + TEST_EQ(screens_displayed[i++], VB_SCREEN_ALT_FW_MENU, " altfw"); + TEST_EQ(screens_displayed[i++], VB_SCREEN_BLANK, " blank (flash)"); + TEST_EQ(screens_displayed[i++], VB_SCREEN_ALT_FW_MENU, " altfw"); + TEST_EQ(screens_displayed[i++], VB_SCREEN_BLANK, " blank (flash)"); + TEST_EQ(screens_count, i, " no extra screens"); + TEST_EQ(beeps_count, 2, " two error beeps: legacy BIOS not found"); + TEST_EQ(beeps_played[0], 200, " low-frequency error beep"); + TEST_EQ(beeps_played[1], 200, " low-frequency error beep"); + + /* Refuse to boot legacy via menu if not enabled */ + ResetMocksForDeveloper(); + i = 0; + mock_keypress[i++] = VB_BUTTON_VOL_UP_SHORT_PRESS; // Enable OS Verif + mock_keypress[i++] = VB_BUTTON_VOL_UP_SHORT_PRESS; // Show Debug Info + mock_keypress[i++] = VB_BUTTON_VOL_UP_SHORT_PRESS; // Developer Options + mock_keypress[i++] = VB_BUTTON_POWER_SHORT_PRESS; + mock_keypress[i++] = VB_BUTTON_VOL_UP_SHORT_PRESS; // Boot From USB + mock_keypress[i++] = VB_BUTTON_VOL_UP_SHORT_PRESS; // Boot Legacy BIOS + mock_keypress[i++] = VB_BUTTON_POWER_SHORT_PRESS; + TEST_EQ(VbBootDeveloperLegacyMenu(ctx), vbtlk_retval_fixed, + "Menu selected legacy boot when not enabled"); + TEST_EQ(debug_info_displayed, 0, " no debug info"); + TEST_EQ(vbexlegacy_called, 0, " did not attempt legacy boot"); + TEST_EQ(audio_looping_calls_left, 0, " audio timeout"); + TEST_EQ(vb2_nv_get(ctx, VB2_NV_RECOVERY_REQUEST), 0, " no recovery"); + i = 0; + TEST_EQ(screens_displayed[i++], VB_SCREEN_DEVELOPER_WARNING_MENU, + " dev warning menu: power off"); + TEST_EQ(screens_displayed[i++], VB_SCREEN_DEVELOPER_WARNING_MENU, + " dev warning menu: enable root verification"); + TEST_EQ(screens_displayed[i++], VB_SCREEN_DEVELOPER_WARNING_MENU, + " dev warning menu: show debug info"); + TEST_EQ(screens_displayed[i++], VB_SCREEN_DEVELOPER_WARNING_MENU, + " dev warning menu: developer options"); + TEST_EQ(screens_displayed[i++], VB_SCREEN_DEVELOPER_MENU, + " dev menu: disk boot"); + TEST_EQ(screens_displayed[i++], VB_SCREEN_DEVELOPER_MENU, + " dev menu: USB boot"); + TEST_EQ(screens_displayed[i++], VB_SCREEN_DEVELOPER_MENU, + " dev menu: legacy boot"); + TEST_EQ(screens_displayed[i++], VB_SCREEN_BLANK, " blank (flash)"); + TEST_EQ(screens_displayed[i++], VB_SCREEN_DEVELOPER_MENU, + " dev menu: legacy boot"); + TEST_EQ(screens_displayed[i++], VB_SCREEN_BLANK," final blank screen"); + TEST_EQ(screens_count, i, " no extra screens"); + TEST_EQ(beeps_count, 2, " played error beeps"); + TEST_EQ(beeps_played[0], 400, " first beep"); + TEST_EQ(beeps_played[1], 400, " second beep"); + + /* Use volume-up long press shortcut to boot USB */ + ResetMocksForDeveloper(); + vb2_nv_set(ctx, VB2_NV_DEV_BOOT_USB, 1); + mock_keypress[0] = VB_BUTTON_VOL_UP_LONG_PRESS; + vbtlk_retval[0] = VB2_SUCCESS - VB_DISK_FLAG_REMOVABLE; + TEST_EQ(VbBootDeveloperLegacyMenu(ctx), VB2_SUCCESS, "VolUp USB"); + TEST_EQ(vbexlegacy_called, 0, " not legacy"); + TEST_EQ(vb2_nv_get(ctx, VB2_NV_RECOVERY_REQUEST), 0, " no recovery"); + TEST_NEQ(audio_looping_calls_left, 0, " aborts audio"); + TEST_EQ(screens_displayed[0], VB_SCREEN_DEVELOPER_WARNING_MENU, + " warning screen"); + TEST_EQ(screens_displayed[1], VB_SCREEN_BLANK, " final blank screen"); + TEST_EQ(screens_count, 2, " no extra screens"); + TEST_EQ(beeps_count, 0, " no beeps on USB boot"); + + /* Can boot a valid USB image after failing to boot invalid image */ + ResetMocksForDeveloper(); + vb2_nv_set(ctx, VB2_NV_DEV_BOOT_USB, 1); + mock_keypress[0] = VB_BUTTON_VOL_UP_LONG_PRESS; + mock_keypress[1] = VB_BUTTON_VOL_UP_LONG_PRESS; + vbtlk_retval[0] = VB2_ERROR_MOCK - VB_DISK_FLAG_REMOVABLE; + vbtlk_retval[1] = VB2_SUCCESS - VB_DISK_FLAG_REMOVABLE; + TEST_EQ(VbBootDeveloperLegacyMenu(ctx), VB2_SUCCESS, + "VolUp USB valid after invalid"); + TEST_EQ(vbexlegacy_called, 0, " not legacy"); + TEST_EQ(vb2_nv_get(ctx, VB2_NV_RECOVERY_REQUEST), 0, " no recovery"); + TEST_NEQ(audio_looping_calls_left, 0, " aborts audio"); + TEST_EQ(screens_displayed[0], VB_SCREEN_DEVELOPER_WARNING_MENU, + " warning screen"); + TEST_EQ(screens_displayed[1], VB_SCREEN_BLANK, " blank (error flash)"); + TEST_EQ(screens_displayed[2], VB_SCREEN_DEVELOPER_WARNING_MENU, + " warning screen"); + TEST_EQ(screens_displayed[3], VB_SCREEN_BLANK, " final blank screen"); + TEST_EQ(screens_count, 4, " no extra screens"); + TEST_EQ(beeps_count, 1, " error beep: first USB invalid"); + TEST_EQ(beeps_played[0], 200, " low-frequency error beep"); + + /* Volume-up long press only works if USB is enabled */ + ResetMocksForDeveloper(); + mock_keypress[0] = VB_BUTTON_VOL_UP_LONG_PRESS; + TEST_EQ(VbBootDeveloperLegacyMenu(ctx), vbtlk_retval_fixed, + "VolUp not enabled"); + TEST_EQ(vbexlegacy_called, 0, " not legacy"); + TEST_EQ(vb2_nv_get(ctx, VB2_NV_RECOVERY_REQUEST), 0, " no recovery"); + TEST_EQ(audio_looping_calls_left, 0, " audio timed out"); + TEST_EQ(screens_displayed[0], VB_SCREEN_DEVELOPER_WARNING_MENU, + " warning screen"); + TEST_EQ(screens_displayed[1], VB_SCREEN_BLANK, " blank (error flash)"); + TEST_EQ(screens_displayed[2], VB_SCREEN_DEVELOPER_WARNING_MENU, + " warning screen"); + TEST_EQ(screens_displayed[3], VB_SCREEN_BLANK, " final blank screen"); + TEST_EQ(screens_count, 4, " no extra screens"); + TEST_EQ(beeps_count, 2, " played error beeps"); + TEST_EQ(beeps_played[0], 400, " first beep"); + TEST_EQ(beeps_played[1], 400, " second beep"); + + /* Volume-up long press without valid USB will still time out */ + ResetMocksForDeveloper(); + vb2_nv_set(ctx, VB2_NV_DEV_BOOT_USB, 1); + mock_keypress[0] = VB_BUTTON_VOL_UP_LONG_PRESS; + TEST_EQ(VbBootDeveloperLegacyMenu(ctx), vbtlk_retval_fixed, + "VolUp without valid USB"); + TEST_EQ(vbexlegacy_called, 0, " not legacy"); + TEST_EQ(vb2_nv_get(ctx, VB2_NV_RECOVERY_REQUEST), 0, " no recovery"); + TEST_EQ(screens_displayed[0], VB_SCREEN_DEVELOPER_WARNING_MENU, + " warning screen"); + TEST_EQ(screens_displayed[1], VB_SCREEN_BLANK, " blank (error flash)"); + TEST_EQ(screens_displayed[2], VB_SCREEN_DEVELOPER_WARNING_MENU, + " warning screen"); + TEST_EQ(screens_displayed[3], VB_SCREEN_BLANK, " final blank screen"); + TEST_EQ(screens_count, 4, " no extra screens"); + TEST_EQ(audio_looping_calls_left, 0, " used up audio"); + TEST_EQ(beeps_count, 1, " error beep: USB not found"); + TEST_EQ(beeps_played[0], 200, " low-frequency error beep"); + + /* Volume-up long press works from other menus, like LANGUAGE */ + ResetMocksForDeveloper(); + vb2_nv_set(ctx, VB2_NV_DEV_BOOT_USB, 1); + i = 0; + mock_keypress[i++] = VB_BUTTON_VOL_UP_SHORT_PRESS; // Enable OS Verif + mock_keypress[i++] = VB_BUTTON_VOL_UP_SHORT_PRESS; // Show Debug Info + mock_keypress[i++] = VB_BUTTON_VOL_UP_SHORT_PRESS; // Developer Options + mock_keypress[i++] = VB_BUTTON_POWER_SHORT_PRESS; + mock_keypress[i++] = VB_BUTTON_VOL_DOWN_SHORT_PRESS; // Cancel + mock_keypress[i++] = VB_BUTTON_VOL_DOWN_SHORT_PRESS; // Power Off + mock_keypress[i++] = VB_BUTTON_VOL_DOWN_SHORT_PRESS; // Language + mock_keypress[i++] = VB_BUTTON_POWER_SHORT_PRESS; + mock_keypress[i++] = VB_BUTTON_VOL_UP_LONG_PRESS; + vbtlk_retval[0] = VB2_SUCCESS - VB_DISK_FLAG_REMOVABLE; + TEST_EQ(VbBootDeveloperLegacyMenu(ctx), VB2_SUCCESS, + "VolUp USB from LANGUAGE"); + TEST_EQ(vbexlegacy_called, 0, " not legacy"); + TEST_EQ(vb2_nv_get(ctx, VB2_NV_RECOVERY_REQUEST), 0, " no recovery"); + TEST_NEQ(audio_looping_calls_left, 0, " aborts audio"); + i = 0; + TEST_EQ(screens_displayed[i++], VB_SCREEN_DEVELOPER_WARNING_MENU, + " dev warning menu: power off"); + TEST_EQ(screens_displayed[i++], VB_SCREEN_DEVELOPER_WARNING_MENU, + " dev warning menu: enable root verification"); + TEST_EQ(screens_displayed[i++], VB_SCREEN_DEVELOPER_WARNING_MENU, + " dev warning menu: show debug info"); + TEST_EQ(screens_displayed[i++], VB_SCREEN_DEVELOPER_WARNING_MENU, + " dev warning menu: developer options"); + TEST_EQ(screens_displayed[i++], VB_SCREEN_DEVELOPER_MENU, + " dev menu: disk boot"); + TEST_EQ(screens_displayed[i++], VB_SCREEN_DEVELOPER_MENU, + " dev menu: cancel"); + TEST_EQ(screens_displayed[i++], VB_SCREEN_DEVELOPER_MENU, + " dev menu: power off"); + TEST_EQ(screens_displayed[i++], VB_SCREEN_DEVELOPER_MENU, + " dev menu: language"); + TEST_EQ(screens_displayed[i++], VB_SCREEN_LANGUAGES_MENU, + " language menu"); + TEST_EQ(screens_displayed[i++], VB_SCREEN_BLANK," final blank screen"); + TEST_EQ(screens_count, i, " no extra screens"); + TEST_EQ(beeps_count, 0, " no beeps on USB boot"); + + /* Can disable developer mode through TONORM screen */ + ResetMocksForDeveloper(); + i = 0; + mock_keypress[i++] = VB_BUTTON_VOL_UP_SHORT_PRESS; // enable os verification + mock_keypress[i++] = VB_BUTTON_POWER_SHORT_PRESS; + mock_keypress[i++] = VB_BUTTON_POWER_SHORT_PRESS; // confirm is the default + TEST_EQ(VbBootDeveloperLegacyMenu(ctx), VBERROR_REBOOT_REQUIRED, + "TONORM via menu"); + TEST_EQ(vb2_nv_get(ctx, VB2_NV_DISABLE_DEV_REQUEST), 1, + " disable dev request"); + TEST_EQ(vb2_nv_get(ctx, VB2_NV_RECOVERY_REQUEST), 0, " no recovery"); + TEST_EQ(vbexlegacy_called, 0, " not legacy"); + TEST_EQ(debug_info_displayed, 0, " no debug info"); + TEST_NEQ(audio_looping_calls_left, 0, " aborts audio"); + i = 0; + TEST_EQ(screens_displayed[i++], VB_SCREEN_DEVELOPER_WARNING_MENU, + " dev warning: power off"); + TEST_EQ(screens_displayed[i++], VB_SCREEN_DEVELOPER_WARNING_MENU, + " dev warning: enable os verification"); + TEST_EQ(screens_displayed[i++], VB_SCREEN_DEVELOPER_TO_NORM_MENU, + " tonorm: confirm"); + TEST_EQ(screens_displayed[i++], VB_SCREEN_TO_NORM_CONFIRMED, + " confirm screen"); + TEST_EQ(screens_displayed[i++], VB_SCREEN_BLANK," final blank screen"); + TEST_EQ(screens_count, i, " no extra screens"); + TEST_EQ(beeps_count, 0, " no beeps on reboot"); + + /* If dev mode is disabled, goes to TONORM screen repeatedly */ + ResetMocksForDeveloper(); + fwmp->flags |= VB2_SECDATA_FWMP_DEV_DISABLE_BOOT; + audio_looping_calls_left = 1; /* Confirm audio doesn't tick down. */ + i = 0; + mock_keypress[i++] = VB_KEY_CTRL('D'); /* Just stays on TONORM and flashes */ + mock_keypress[i++] = VB_KEY_CTRL('U'); /* same */ + mock_keypress[i++] = VB_KEY_CTRL('L'); /* same */ + mock_keypress[i++] = VB_BUTTON_VOL_UP_LONG_PRESS; /* same */ + mock_keypress[i++] = VB_BUTTON_VOL_DOWN_LONG_PRESS; /* same */ + mock_keypress[i++] = VB_BUTTON_VOL_UP_DOWN_COMBO_PRESS; /* noop */ + mock_keypress[i++] = VB_KEY_ENTER; + TEST_EQ(VbBootDeveloperLegacyMenu(ctx), VBERROR_REBOOT_REQUIRED, + "FWMP dev disabled"); + TEST_EQ(vb2_nv_get(ctx, VB2_NV_DISABLE_DEV_REQUEST), 1, + " disable dev request"); + TEST_EQ(vb2_nv_get(ctx, VB2_NV_RECOVERY_REQUEST), 0, " no recovery"); + TEST_EQ(debug_info_displayed, 0, " no debug info"); + TEST_EQ(vbexlegacy_called, 0, " not legacy"); + TEST_NEQ(audio_looping_calls_left, 0, " aborts audio"); + i = 0; + TEST_EQ(screens_displayed[i++], VB_SCREEN_DEVELOPER_TO_NORM_MENU, + " tonorm screen"); + TEST_EQ(screens_displayed[i++], VB_SCREEN_BLANK, " blank (flash)"); + TEST_EQ(screens_displayed[i++], VB_SCREEN_DEVELOPER_TO_NORM_MENU, + " tonorm screen"); + TEST_EQ(screens_displayed[i++], VB_SCREEN_BLANK, " blank (flash)"); + TEST_EQ(screens_displayed[i++], VB_SCREEN_DEVELOPER_TO_NORM_MENU, + " tonorm screen"); + TEST_EQ(screens_displayed[i++], VB_SCREEN_BLANK, " blank (flash)"); + TEST_EQ(screens_displayed[i++], VB_SCREEN_DEVELOPER_TO_NORM_MENU, + " tonorm screen"); + TEST_EQ(screens_displayed[i++], VB_SCREEN_BLANK, " blank (flash)"); + TEST_EQ(screens_displayed[i++], VB_SCREEN_DEVELOPER_TO_NORM_MENU, + " tonorm screen"); + TEST_EQ(screens_displayed[i++], VB_SCREEN_BLANK, " blank (flash)"); + TEST_EQ(screens_displayed[i++], VB_SCREEN_DEVELOPER_TO_NORM_MENU, + " tonorm screen"); + TEST_EQ(screens_displayed[i++], VB_SCREEN_TO_NORM_CONFIRMED, + " confirm screen"); + TEST_EQ(screens_displayed[i++], VB_SCREEN_BLANK," final blank screen"); + TEST_EQ(screens_count, i, " no extra screens"); + TEST_EQ(beeps_count, 10, " 5 * 2 error beeps: booting not allowed"); + TEST_EQ(beeps_played[0], 400, " first error beep"); + TEST_EQ(beeps_played[1], 400, " second error beep"); + TEST_EQ(beeps_played[2], 400, " first error beep"); + TEST_EQ(beeps_played[3], 400, " second error beep"); + TEST_EQ(beeps_played[4], 400, " first error beep"); + TEST_EQ(beeps_played[5], 400, " second error beep"); + TEST_EQ(beeps_played[6], 400, " first error beep"); + TEST_EQ(beeps_played[7], 400, " second error beep"); + TEST_EQ(beeps_played[8], 400, " first error beep"); + TEST_EQ(beeps_played[9], 400, " second error beep"); + + /* Shutdown requested when dev disabled */ + ResetMocksForDeveloper(); + fwmp->flags |= VB2_SECDATA_FWMP_DEV_DISABLE_BOOT; + shutdown_request_calls_left = 1; + TEST_EQ(VbBootDeveloperLegacyMenu(ctx), VBERROR_SHUTDOWN_REQUESTED, + "Shutdown requested when dev disabled"); + TEST_EQ(vb2_nv_get(ctx, VB2_NV_RECOVERY_REQUEST), 0, " no recovery"); + TEST_EQ(vb2_nv_get(ctx, VB2_NV_DISABLE_DEV_REQUEST), 0, + " did not exit dev mode"); + TEST_EQ(screens_displayed[0], VB_SCREEN_DEVELOPER_TO_NORM_MENU, + " tonorm screen"); + TEST_EQ(screens_displayed[1], VB_SCREEN_BLANK, " final blank screen"); + TEST_EQ(screens_count, 2, " no extra screens"); + TEST_NEQ(audio_looping_calls_left, 0, " aborts audio"); + TEST_EQ(beeps_count, 0, " no beep on shutdown"); + + /* Explicit Power Off when dev disabled */ + ResetMocksForDeveloper(); + fwmp->flags |= VB2_SECDATA_FWMP_DEV_DISABLE_BOOT; + i = 0; + mock_keypress[i++] = VB_BUTTON_VOL_DOWN_SHORT_PRESS; // Power Off + mock_keypress[i++] = VB_BUTTON_POWER_SHORT_PRESS; + TEST_EQ(VbBootDeveloperLegacyMenu(ctx), VBERROR_SHUTDOWN_REQUESTED, + "Power Off when dev disabled"); + TEST_EQ(vb2_nv_get(ctx, VB2_NV_RECOVERY_REQUEST), 0, " no recovery"); + TEST_EQ(vb2_nv_get(ctx, VB2_NV_DISABLE_DEV_REQUEST), 0, + " did not exit dev mode"); + TEST_EQ(debug_info_displayed, 0, " no debug info"); + TEST_NEQ(audio_looping_calls_left, 0, " aborts audio"); + i = 0; + TEST_EQ(screens_displayed[i++], VB_SCREEN_DEVELOPER_TO_NORM_MENU, + " tonorm screen: confirm enabling OS verification"); + /* Cancel option is removed with dev_disable_boot! */ + TEST_EQ(screens_displayed[i++], VB_SCREEN_DEVELOPER_TO_NORM_MENU, + " tonorm screen: power off"); + TEST_EQ(screens_displayed[i++], VB_SCREEN_BLANK," final blank screen"); + TEST_EQ(screens_count, i, " no extra screens"); + TEST_EQ(beeps_count, 0, " no beeps for power off"); + + /* Show Debug Info displays debug info, then times out to boot */ + ResetMocksForDeveloper(); + i = 0; + mock_keypress[i++] = VB_BUTTON_VOL_UP_SHORT_PRESS; // Enable OS Verif + mock_keypress[i++] = VB_BUTTON_VOL_UP_SHORT_PRESS; // Show Debug Info + mock_keypress[i++] = VB_BUTTON_POWER_SHORT_PRESS; + TEST_EQ(VbBootDeveloperLegacyMenu(ctx), vbtlk_retval_fixed, + "Show Debug Info"); + TEST_EQ(debug_info_displayed, 1, " debug info displayed"); + TEST_EQ(vbexlegacy_called, 0, " not legacy"); + TEST_EQ(audio_looping_calls_left, 0, " audio timed out"); + TEST_EQ(vb2_nv_get(ctx, VB2_NV_RECOVERY_REQUEST), 0, " no recovery"); + i = 0; + TEST_EQ(screens_displayed[i++], VB_SCREEN_DEVELOPER_WARNING_MENU, + " dev warning menu: power off"); + TEST_EQ(screens_displayed[i++], VB_SCREEN_DEVELOPER_WARNING_MENU, + " dev warning menu: enable root verification"); + TEST_EQ(screens_displayed[i++], VB_SCREEN_DEVELOPER_WARNING_MENU, + " dev warning menu: show debug info"); + TEST_EQ(screens_displayed[i++], VB_SCREEN_BLANK," final blank screen"); + TEST_EQ(screens_count, i, " no extra screens"); + TEST_EQ(beeps_count, 0, " no beeps for debug info"); + + /* Pressing Tab displays debug info, then times out to boot */ + ResetMocksForDeveloper(); + i = 0; + mock_keypress[i++] = '\t'; + TEST_EQ(VbBootDeveloperLegacyMenu(ctx), vbtlk_retval_fixed, + "Show Debug Info (Tab shortcut)"); + TEST_EQ(debug_info_displayed, 1, " debug info displayed"); + TEST_EQ(vbexlegacy_called, 0, " not legacy"); + TEST_EQ(audio_looping_calls_left, 0, " audio timed out"); + TEST_EQ(vb2_nv_get(ctx, VB2_NV_RECOVERY_REQUEST), 0, " no recovery"); + i = 0; + TEST_EQ(screens_displayed[i++], VB_SCREEN_DEVELOPER_WARNING_MENU, + " dev warning menu: power off"); + TEST_EQ(screens_displayed[i++], VB_SCREEN_BLANK," final blank screen"); + TEST_EQ(screens_count, i, " no extra screens"); + TEST_EQ(beeps_count, 0, " no beeps for debug info"); + + + printf("...done.\n"); +} + +static void VbBootRecTest(void) +{ + int i; + + printf("Testing VbBootRecoveryLegacyMenu()...\n"); + + /* Shutdown requested in BROKEN */ + ResetMocks(); + TEST_EQ(VbBootRecoveryLegacyMenu(ctx), VBERROR_SHUTDOWN_REQUESTED, + "Shutdown requested in BROKEN"); + TEST_EQ(vb2_nv_get(ctx, VB2_NV_RECOVERY_REQUEST), 0, " no recovery"); + TEST_EQ(debug_info_displayed, 0, " no debug info"); + TEST_EQ(screens_displayed[0], VB_SCREEN_OS_BROKEN, + " broken screen"); + TEST_EQ(screens_displayed[1], VB_SCREEN_BLANK, " final blank screen"); + TEST_EQ(screens_count, 2, " no extra screens"); + TEST_EQ(beeps_count, 0, " no beep on shutdown"); + TEST_EQ(vbexaltfwmask_called, 0, " VbExGetAltFwIdxMask not called"); + + /* BROKEN screen with disks inserted */ + ResetMocks(); + vbtlk_retval[0] = VB2_SUCCESS - VB_DISK_FLAG_REMOVABLE; + vbtlk_retval[1] = VB2_SUCCESS - VB_DISK_FLAG_REMOVABLE; + vbtlk_retval[2] = VB2_SUCCESS - VB_DISK_FLAG_REMOVABLE; + vbtlk_retval[3] = VB2_ERROR_LK_NO_DISK_FOUND - + VB_DISK_FLAG_REMOVABLE; + TEST_EQ(VbBootRecoveryLegacyMenu(ctx), VBERROR_SHUTDOWN_REQUESTED, + "Shutdown requested in BROKEN with disks"); + TEST_EQ(vb2_nv_get(ctx, VB2_NV_RECOVERY_REQUEST), 0, " no recovery"); + TEST_EQ(debug_info_displayed, 0, " no debug info"); + TEST_EQ(screens_displayed[0], VB_SCREEN_OS_BROKEN, + " broken screen"); + TEST_EQ(screens_displayed[1], VB_SCREEN_BLANK, " final blank screen"); + TEST_EQ(screens_count, 2, " no extra screens"); + TEST_EQ(beeps_count, 0, " no beep on shutdown"); + + /* BROKEN screen with disks on second attempt */ + ResetMocks(); + vbtlk_retval[0] = VB2_ERROR_LK_NO_DISK_FOUND - + VB_DISK_FLAG_REMOVABLE; + vbtlk_retval[1] = VB2_SUCCESS - VB_DISK_FLAG_REMOVABLE; + TEST_EQ(VbBootRecoveryLegacyMenu(ctx), VBERROR_SHUTDOWN_REQUESTED, + "Shutdown requested in BROKEN with later disk"); + TEST_EQ(vb2_nv_get(ctx, VB2_NV_RECOVERY_REQUEST), 0, " no recovery"); + TEST_EQ(debug_info_displayed, 0, " no debug info"); + TEST_EQ(screens_displayed[0], VB_SCREEN_OS_BROKEN, + " broken screen"); + TEST_EQ(screens_displayed[1], VB_SCREEN_BLANK, " final blank screen"); + TEST_EQ(screens_count, 2, " no extra screens"); + TEST_EQ(beeps_count, 0, " no beep on shutdown"); + + /* BROKEN screen even if dev switch is on */ + ResetMocks(); + vbtlk_retval[0] = VB2_SUCCESS - VB_DISK_FLAG_REMOVABLE; + vbtlk_retval[1] = VB2_ERROR_LK_NO_DISK_FOUND - + VB_DISK_FLAG_REMOVABLE; + vbtlk_retval[2] = VB2_SUCCESS - VB_DISK_FLAG_REMOVABLE; + sd->flags |= VB2_SD_FLAG_DEV_MODE_ENABLED; + TEST_EQ(VbBootRecoveryLegacyMenu(ctx), VBERROR_SHUTDOWN_REQUESTED, + "Shutdown requested in BROKEN with dev switch"); + TEST_EQ(vb2_nv_get(ctx, VB2_NV_RECOVERY_REQUEST), 0, " no recovery"); + TEST_EQ(debug_info_displayed, 0, " no debug info"); + TEST_EQ(screens_displayed[0], VB_SCREEN_OS_BROKEN, + " broken screen"); + TEST_EQ(screens_displayed[1], VB_SCREEN_BLANK, " final blank screen"); + TEST_EQ(screens_count, 2, " no extra screens"); + TEST_EQ(beeps_count, 0, " no beep on shutdown"); + + /* go to INSERT if recovery button physically pressed and EC trusted */ + ResetMocksForManualRecovery(); + vbtlk_retval[0] = VB2_ERROR_LK_NO_DISK_FOUND - + VB_DISK_FLAG_REMOVABLE; + TEST_EQ(VbBootRecoveryLegacyMenu(ctx), VBERROR_SHUTDOWN_REQUESTED, + "Shutdown requested in INSERT with manual rec"); + TEST_EQ(vb2_nv_get(ctx, VB2_NV_RECOVERY_REQUEST), 0, " no recovery"); + TEST_EQ(debug_info_displayed, 0, " no debug info"); + TEST_EQ(screens_displayed[0], VB_SCREEN_RECOVERY_INSERT, + " insert screen"); + TEST_EQ(screens_displayed[1], VB_SCREEN_BLANK, " final blank screen"); + TEST_EQ(screens_count, 2, " no extra screens"); + TEST_EQ(beeps_count, 0, " no beep on shutdown"); + + /* go to INSERT if forced by GBB flag */ + ResetMocks(); + vbtlk_retval[0] = VB2_ERROR_LK_NO_DISK_FOUND - + VB_DISK_FLAG_REMOVABLE; + gbb.flags |= VB2_GBB_FLAG_FORCE_MANUAL_RECOVERY; + TEST_EQ(VbBootRecoveryLegacyMenu(ctx), VBERROR_SHUTDOWN_REQUESTED, + "Shutdown requested in INSERT forced by GBB flag"); + TEST_EQ(vb2_nv_get(ctx, VB2_NV_RECOVERY_REQUEST), 0, " no recovery"); + TEST_EQ(debug_info_displayed, 0, " no debug info"); + TEST_EQ(screens_displayed[0], VB_SCREEN_RECOVERY_INSERT, + " insert screen"); + TEST_EQ(screens_displayed[1], VB_SCREEN_BLANK, " final blank screen"); + TEST_EQ(screens_count, 2, " no extra screens"); + TEST_EQ(beeps_count, 0, " no beep on shutdown"); + + /* Stay at BROKEN if recovery button not physically pressed */ + ResetMocksForManualRecovery(); + vbtlk_retval[0] = VB2_ERROR_LK_NO_DISK_FOUND - + VB_DISK_FLAG_REMOVABLE; + sd->flags &= ~VB2_SD_FLAG_MANUAL_RECOVERY; + TEST_EQ(VbBootRecoveryLegacyMenu(ctx), VBERROR_SHUTDOWN_REQUESTED, + "Go to BROKEN if recovery not manually requested"); + TEST_EQ(vb2_nv_get(ctx, VB2_NV_RECOVERY_REQUEST), 0, " no recovery"); + TEST_EQ(debug_info_displayed, 0, " no debug info"); + TEST_EQ(screens_displayed[0], VB_SCREEN_OS_BROKEN, + " broken screen"); + TEST_EQ(screens_displayed[1], VB_SCREEN_BLANK, " final blank screen"); + TEST_EQ(screens_count, 2, " no extra screens"); + TEST_EQ(beeps_count, 0, " no beep on shutdown"); + + /* Stay at BROKEN if EC is untrusted */ + ResetMocksForManualRecovery(); + vbtlk_retval[0] = VB2_ERROR_LK_NO_DISK_FOUND - + VB_DISK_FLAG_REMOVABLE; + trust_ec = 0; + TEST_EQ(VbBootRecoveryLegacyMenu(ctx), VBERROR_SHUTDOWN_REQUESTED, + "Go to BROKEN if EC is not trusted"); + TEST_EQ(vb2_nv_get(ctx, VB2_NV_RECOVERY_REQUEST), 0, " no recovery"); + TEST_EQ(debug_info_displayed, 0, " no debug info"); + TEST_EQ(screens_displayed[0], VB_SCREEN_OS_BROKEN, + " broken screen"); + TEST_EQ(screens_displayed[1], VB_SCREEN_BLANK, " final blank screen"); + TEST_EQ(screens_count, 2, " no extra screens"); + TEST_EQ(beeps_count, 0, " no beep on shutdown"); + + /* INSERT boots without screens if we have a valid image on first try */ + ResetMocksForManualRecovery(); + vbtlk_retval[0] = VB2_SUCCESS - VB_DISK_FLAG_REMOVABLE; + vbtlk_retval[1] = VB2_ERROR_MOCK - VB_DISK_FLAG_REMOVABLE; + TEST_EQ(VbBootRecoveryLegacyMenu(ctx), VB2_SUCCESS, + "INSERT boots without screens if valid on first try"); + TEST_EQ(vb2_nv_get(ctx, VB2_NV_RECOVERY_REQUEST), 0, " no recovery"); + TEST_EQ(debug_info_displayed, 0, " no debug info"); + TEST_EQ(virtdev_set, 0, " virtual dev mode off"); + TEST_EQ(screens_displayed[0], VB_SCREEN_BLANK, " final blank screen"); + TEST_EQ(screens_count, 1, " no extra screens"); + TEST_EQ(beeps_count, 0, " no beep on shutdown"); + + /* INSERT boots eventually if we get a valid image later */ + ResetMocksForManualRecovery(); + vbtlk_retval[0] = VB2_ERROR_LK_NO_DISK_FOUND - + VB_DISK_FLAG_REMOVABLE; + vbtlk_retval[1] = VB2_ERROR_LK_NO_DISK_FOUND - + VB_DISK_FLAG_REMOVABLE; + vbtlk_retval[2] = VB2_ERROR_LK_NO_DISK_FOUND - + VB_DISK_FLAG_REMOVABLE; + vbtlk_retval[3] = VB2_ERROR_LK_NO_DISK_FOUND - + VB_DISK_FLAG_REMOVABLE; + vbtlk_retval[4] = VB2_SUCCESS - VB_DISK_FLAG_REMOVABLE; + TEST_EQ(VbBootRecoveryLegacyMenu(ctx), VB2_SUCCESS, + "INSERT boots after valid image appears"); + TEST_EQ(vb2_nv_get(ctx, VB2_NV_RECOVERY_REQUEST), 0, " no recovery"); + TEST_EQ(debug_info_displayed, 0, " no debug info"); + TEST_EQ(virtdev_set, 0, " virtual dev mode off"); + TEST_EQ(screens_displayed[0], VB_SCREEN_RECOVERY_INSERT, + " insert screen"); + TEST_EQ(screens_displayed[1], VB_SCREEN_BLANK, " final blank screen"); + TEST_EQ(screens_count, 2, " no extra screens"); + TEST_EQ(beeps_count, 0, " no beep on shutdown"); + + /* invalid image, then remove, then valid image */ + ResetMocksForManualRecovery(); + vbtlk_retval[0] = VB2_ERROR_MOCK - VB_DISK_FLAG_REMOVABLE; + vbtlk_retval[1] = VB2_ERROR_LK_NO_DISK_FOUND - + VB_DISK_FLAG_REMOVABLE; + vbtlk_retval[2] = VB2_ERROR_LK_NO_DISK_FOUND - + VB_DISK_FLAG_REMOVABLE; + vbtlk_retval[3] = VB2_ERROR_LK_NO_DISK_FOUND - + VB_DISK_FLAG_REMOVABLE; + vbtlk_retval[4] = VB2_SUCCESS - VB_DISK_FLAG_REMOVABLE; + TEST_EQ(VbBootRecoveryLegacyMenu(ctx), VB2_SUCCESS, + "INSERT boots after valid image appears"); + TEST_EQ(vb2_nv_get(ctx, VB2_NV_RECOVERY_REQUEST), 0, " no recovery"); + TEST_EQ(debug_info_displayed, 0, " no debug info"); + TEST_EQ(virtdev_set, 0, " virtual dev mode off"); + TEST_EQ(screens_displayed[0], VB_SCREEN_RECOVERY_NO_GOOD, + " nogood screen"); + TEST_EQ(screens_displayed[1], VB_SCREEN_RECOVERY_INSERT, + " insert screen"); + TEST_EQ(screens_displayed[2], VB_SCREEN_BLANK, " final blank screen"); + TEST_EQ(screens_count, 3, " no extra screens"); + TEST_EQ(beeps_count, 0, " no beep on shutdown"); + + /* Shortcuts that are always ignored in BROKEN for detachables. */ + ResetMocks(); + i = 0; + mock_keypress[i] = VB_KEY_FLAG_TRUSTED_KEYBOARD; + mock_keypress[i++] = VB_KEY_CTRL('D'); + mock_keypress[i] = VB_KEY_FLAG_TRUSTED_KEYBOARD; + mock_keypress[i++] = VB_KEY_CTRL('U'); + mock_keypress[i] = VB_KEY_FLAG_TRUSTED_KEYBOARD; + mock_keypress[i++] = VB_KEY_CTRL('L'); + mock_keypress[i] = VB_KEY_FLAG_TRUSTED_KEYBOARD; + mock_keypress[i++] = VB_BUTTON_VOL_UP_DOWN_COMBO_PRESS; + mock_keypress[i] = VB_KEY_FLAG_TRUSTED_KEYBOARD; + mock_keypress[i++] = VB_BUTTON_VOL_UP_LONG_PRESS; + mock_keypress[i] = VB_KEY_FLAG_TRUSTED_KEYBOARD; + mock_keypress[i++] = VB_BUTTON_VOL_DOWN_LONG_PRESS; + mock_keypress[i++] = VB_BUTTON_POWER_SHORT_PRESS; + TEST_EQ(VbBootRecoveryLegacyMenu(ctx), VBERROR_SHUTDOWN_REQUESTED, + "Shortcuts ignored in BROKEN"); + TEST_EQ(virtdev_set, 0, " virtual dev mode off"); + TEST_NEQ(shutdown_request_calls_left, 0, " powered down explicitly"); + TEST_EQ(vb2_nv_get(ctx, VB2_NV_RECOVERY_REQUEST), 0, " no recovery"); + TEST_EQ(debug_info_displayed, 0, " no debug info"); + TEST_EQ(screens_displayed[0], VB_SCREEN_OS_BROKEN, + " broken screen"); + TEST_EQ(screens_displayed[1], VB_SCREEN_BLANK, " final blank screen"); + TEST_EQ(screens_count, 2, " no extra screens"); + TEST_EQ(beeps_count, 0, " no beep from invalid keys"); + + /* Shortcuts that are always ignored in INSERT for detachables. */ + ResetMocksForManualRecovery(); + i = 0; + mock_keypress[i] = VB_KEY_FLAG_TRUSTED_KEYBOARD; + mock_keypress[i++] = VB_KEY_CTRL('D'); + mock_keypress[i] = VB_KEY_FLAG_TRUSTED_KEYBOARD; + mock_keypress[i++] = VB_KEY_CTRL('U'); + mock_keypress[i] = VB_KEY_FLAG_TRUSTED_KEYBOARD; + mock_keypress[i++] = VB_KEY_CTRL('L'); + mock_keypress[i] = VB_KEY_FLAG_TRUSTED_KEYBOARD; + mock_keypress[i++] = VB_BUTTON_VOL_UP_LONG_PRESS; + mock_keypress[i] = VB_KEY_FLAG_TRUSTED_KEYBOARD; + mock_keypress[i++] = VB_BUTTON_VOL_DOWN_LONG_PRESS; + mock_keypress[i++] = VB_BUTTON_POWER_SHORT_PRESS; + vbtlk_retval[0] = VB2_ERROR_LK_NO_DISK_FOUND - + VB_DISK_FLAG_REMOVABLE; + TEST_EQ(VbBootRecoveryLegacyMenu(ctx), VBERROR_SHUTDOWN_REQUESTED, + "Shortcuts ignored in INSERT"); + TEST_EQ(virtdev_set, 0, " virtual dev mode off"); + TEST_NEQ(shutdown_request_calls_left, 0, " powered down explicitly"); + TEST_EQ(vb2_nv_get(ctx, VB2_NV_RECOVERY_REQUEST), 0, " no recovery"); + TEST_EQ(debug_info_displayed, 0, " no debug info"); + TEST_EQ(screens_displayed[0], VB_SCREEN_RECOVERY_INSERT, + " insert screen"); + TEST_EQ(screens_displayed[1], VB_SCREEN_BLANK, " final blank screen"); + TEST_EQ(screens_count, 2, " no extra screens"); + TEST_EQ(beeps_count, 0, " no beep from invalid keys"); + + /* Power Off BROKEN through OPTIONS menu */ + ResetMocks(); + mock_keypress[0] = VB_BUTTON_VOL_UP_SHORT_PRESS; // enter options + mock_keypress[1] = VB_BUTTON_VOL_DOWN_SHORT_PRESS; // power off + mock_keypress[2] = VB_BUTTON_POWER_SHORT_PRESS; + TEST_EQ(VbBootRecoveryLegacyMenu(ctx), VBERROR_SHUTDOWN_REQUESTED, + "Power Off BROKEN through OPTIONS"); + TEST_EQ(vb2_nv_get(ctx, VB2_NV_RECOVERY_REQUEST), 0, " no recovery"); + TEST_EQ(debug_info_displayed, 0, " no debug info"); + TEST_NEQ(shutdown_request_calls_left, 0, " powered down explicitly"); + TEST_EQ(screens_displayed[0], VB_SCREEN_OS_BROKEN, + " broken screen"); + TEST_EQ(screens_displayed[1], VB_SCREEN_OPTIONS_MENU, + " options: cancel"); + TEST_EQ(screens_displayed[2], VB_SCREEN_OPTIONS_MENU, + " options: power off"); + TEST_EQ(screens_displayed[3], VB_SCREEN_BLANK, " final blank screen"); + TEST_EQ(screens_count, 4, " no extra screens"); + TEST_EQ(beeps_count, 0, " no beep from power off"); + + /* Power Off NOGOOD through OPTIONS menu */ + ResetMocksForManualRecovery(); + mock_keypress[0] = VB_BUTTON_VOL_UP_SHORT_PRESS; // enter options + mock_keypress[1] = VB_BUTTON_VOL_DOWN_SHORT_PRESS; // power off + mock_keypress[2] = VB_BUTTON_POWER_SHORT_PRESS; + TEST_EQ(VbBootRecoveryLegacyMenu(ctx), VBERROR_SHUTDOWN_REQUESTED, + "Power Off NOGOOD through OPTIONS"); + TEST_EQ(vb2_nv_get(ctx, VB2_NV_RECOVERY_REQUEST), 0, " no recovery"); + TEST_EQ(debug_info_displayed, 0, " no debug info"); + TEST_NEQ(shutdown_request_calls_left, 0, " powered down explicitly"); + TEST_EQ(screens_displayed[0], VB_SCREEN_RECOVERY_NO_GOOD, + " nogood screen"); + TEST_EQ(screens_displayed[1], VB_SCREEN_OPTIONS_MENU, + " options: cancel"); + TEST_EQ(screens_displayed[2], VB_SCREEN_OPTIONS_MENU, + " options: power off"); + TEST_EQ(screens_displayed[3], VB_SCREEN_BLANK, " final blank screen"); + TEST_EQ(screens_count, 4, " no extra screens"); + TEST_EQ(beeps_count, 0, " no beep from power off"); + + /* Power Off INSERT through TO_DEV menu */ + ResetMocksForManualRecovery(); + mock_keyflags[0] = VB_KEY_FLAG_TRUSTED_KEYBOARD; + mock_keypress[0] = VB_BUTTON_VOL_UP_DOWN_COMBO_PRESS; // enter to_dev + mock_keyflags[1] = VB_KEY_FLAG_TRUSTED_KEYBOARD; + mock_keypress[1] = VB_BUTTON_VOL_DOWN_SHORT_PRESS; // power off + mock_keypress[2] = VB_BUTTON_POWER_SHORT_PRESS; + vbtlk_retval[0] = VB2_ERROR_LK_NO_DISK_FOUND - + VB_DISK_FLAG_REMOVABLE; + TEST_EQ(VbBootRecoveryLegacyMenu(ctx), VBERROR_SHUTDOWN_REQUESTED, + "Power Off INSERT through TO_DEV"); + TEST_EQ(vb2_nv_get(ctx, VB2_NV_RECOVERY_REQUEST), 0, " no recovery"); + TEST_EQ(debug_info_displayed, 0, " no debug info"); + TEST_NEQ(shutdown_request_calls_left, 0, " powered down explicitly"); + TEST_EQ(screens_displayed[0], VB_SCREEN_RECOVERY_INSERT, + " insert screen"); + TEST_EQ(screens_displayed[1], VB_SCREEN_RECOVERY_TO_DEV_MENU, + " to_dev: cancel"); + TEST_EQ(screens_displayed[2], VB_SCREEN_RECOVERY_TO_DEV_MENU, + " to_dev: power off"); + TEST_EQ(screens_displayed[3], VB_SCREEN_BLANK, " final blank screen"); + TEST_EQ(screens_count, 4, " no extra screens"); + TEST_EQ(beeps_count, 0, " no beep from power off"); + + /* Show Debug Info from BROKEN through OPTIONS menu */ + ResetMocks(); + mock_keypress[0] = VB_BUTTON_VOL_UP_SHORT_PRESS; // enter options + mock_keypress[1] = VB_BUTTON_VOL_UP_SHORT_PRESS; // show debug info + mock_keypress[2] = VB_BUTTON_POWER_SHORT_PRESS; + TEST_EQ(VbBootRecoveryLegacyMenu(ctx), VBERROR_SHUTDOWN_REQUESTED, + "Show Debug info from BROKEN through OPTIONS"); + TEST_EQ(vb2_nv_get(ctx, VB2_NV_RECOVERY_REQUEST), 0, " no recovery"); + TEST_EQ(debug_info_displayed, 1, " no debug info"); + TEST_EQ(shutdown_request_calls_left, 0, " timed out"); + TEST_EQ(screens_displayed[0], VB_SCREEN_OS_BROKEN, + " broken screen"); + TEST_EQ(screens_displayed[1], VB_SCREEN_OPTIONS_MENU, + " options: cancel"); + TEST_EQ(screens_displayed[2], VB_SCREEN_OPTIONS_MENU, + " options: show debug info"); + TEST_EQ(screens_displayed[3], VB_SCREEN_BLANK, " final blank screen"); + TEST_EQ(screens_count, 4, " no extra screens"); + TEST_EQ(beeps_count, 0, " no beep from power off"); + + /* Show Debug Info on NOGOOD with Tab */ + ResetMocksForManualRecovery(); + mock_keypress[0] = '\t'; + TEST_EQ(VbBootRecoveryLegacyMenu(ctx), VBERROR_SHUTDOWN_REQUESTED, + "Show Debug info on NOGOOD with Tab"); + TEST_EQ(vb2_nv_get(ctx, VB2_NV_RECOVERY_REQUEST), 0, " no recovery"); + TEST_EQ(debug_info_displayed, 1, " no debug info"); + TEST_EQ(shutdown_request_calls_left, 0, " timed out"); + TEST_EQ(screens_displayed[0], VB_SCREEN_RECOVERY_NO_GOOD, + " nogood screen"); + TEST_EQ(screens_displayed[1], VB_SCREEN_BLANK, " final blank screen"); + TEST_EQ(screens_count, 2, " no extra screens"); + TEST_EQ(beeps_count, 0, " no beep from power off"); + + /* Navigate to confirm dev mode selection and then cancel */ + ResetMocksForManualRecovery(); + vbtlk_retval[0] = VB2_ERROR_LK_NO_DISK_FOUND - + VB_DISK_FLAG_REMOVABLE; + i = 0; + mock_keyflags[i] = VB_KEY_FLAG_TRUSTED_KEYBOARD; + mock_keypress[i++] = VB_BUTTON_VOL_UP_DOWN_COMBO_PRESS; // enter TO_DEV + mock_keyflags[i] = VB_KEY_FLAG_TRUSTED_KEYBOARD; + mock_keypress[i++] = VB_BUTTON_VOL_UP_SHORT_PRESS; // confirm disabling + mock_keyflags[i] = VB_KEY_FLAG_TRUSTED_KEYBOARD; + mock_keypress[i++] = VB_BUTTON_VOL_DOWN_SHORT_PRESS; // cancel + mock_keypress[i++] = VB_BUTTON_POWER_SHORT_PRESS; + mock_keypress[i++] = VB_BUTTON_POWER_SHORT_PRESS; // power off + TEST_EQ(VbBootRecoveryLegacyMenu(ctx), VBERROR_SHUTDOWN_REQUESTED, + "go to TO_DEV screen and cancel"); + TEST_NEQ(shutdown_request_calls_left, 0, " powered down explicitly"); + TEST_EQ(virtdev_set, 0, " virtual dev mode off"); + TEST_EQ(vb2_nv_get(ctx, VB2_NV_RECOVERY_REQUEST), 0, " no recovery"); + TEST_EQ(debug_info_displayed, 0, " no debug info"); + i = 0; + TEST_EQ(screens_displayed[i++], VB_SCREEN_RECOVERY_INSERT, + " insert screen"); + TEST_EQ(screens_displayed[i++], VB_SCREEN_RECOVERY_TO_DEV_MENU, + " recovery to_dev menu: cancel"); + TEST_EQ(screens_displayed[i++], VB_SCREEN_RECOVERY_TO_DEV_MENU, + " recovery to_dev menu: confirm disabling"); + TEST_EQ(screens_displayed[i++], VB_SCREEN_RECOVERY_TO_DEV_MENU, + " recovery to_dev menu: cancel"); + TEST_EQ(screens_displayed[i++], VB_SCREEN_RECOVERY_INSERT, + " back to insert screen"); + TEST_EQ(screens_displayed[i++], VB_SCREEN_BLANK," final blank screen"); + TEST_EQ(screens_count, i, " no extra screens"); + TEST_EQ(beeps_count, 0, " no beeps"); + + /* Navigate to confirm dev mode selection and then confirm */ + ResetMocksForManualRecovery(); + vbtlk_retval[0] = VB2_ERROR_LK_NO_DISK_FOUND - + VB_DISK_FLAG_REMOVABLE; + i = 0; + mock_keyflags[i] = VB_KEY_FLAG_TRUSTED_KEYBOARD; + mock_keypress[i++] = VB_BUTTON_VOL_UP_DOWN_COMBO_PRESS; // enter to_dev + mock_keyflags[i] = VB_KEY_FLAG_TRUSTED_KEYBOARD; + mock_keypress[i++] = VB_BUTTON_VOL_UP_SHORT_PRESS; + mock_keypress[i++] = VB_BUTTON_POWER_SHORT_PRESS; + TEST_EQ(VbBootRecoveryLegacyMenu(ctx), VBERROR_REBOOT_REQUIRED, + "go to TO_DEV screen and confirm"); + TEST_EQ(vb2_nv_get(ctx, VB2_NV_RECOVERY_REQUEST), 0, " no recovery"); + TEST_EQ(debug_info_displayed, 0, " no debug info"); + TEST_EQ(virtdev_set, 1, " virtual dev mode on"); + i = 0; + TEST_EQ(screens_displayed[i++], VB_SCREEN_RECOVERY_INSERT, + " insert screen"); + TEST_EQ(screens_displayed[i++], VB_SCREEN_RECOVERY_TO_DEV_MENU, + " recovery to_dev menu: cancel"); + TEST_EQ(screens_displayed[i++], VB_SCREEN_RECOVERY_TO_DEV_MENU, + " recovery to_dev menu: confirm disabling os verification"); + TEST_EQ(screens_displayed[i++], VB_SCREEN_BLANK," final blank screen"); + TEST_EQ(screens_count, i, " no extra screens"); + TEST_EQ(beeps_count, 0, " no beeps"); + + /* Untrusted keyboard cannot enter TO_DEV (must be malicious anyway) */ + ResetMocksForManualRecovery(); + i = 0; + mock_keypress[i++] = VB_BUTTON_VOL_UP_DOWN_COMBO_PRESS; // try to_dev + mock_keypress[i++] = VB_BUTTON_VOL_UP_SHORT_PRESS; // try confirm + mock_keypress[i++] = VB_BUTTON_POWER_SHORT_PRESS; + TEST_EQ(VbBootRecoveryLegacyMenu(ctx), VBERROR_SHUTDOWN_REQUESTED, + "Untrusted keyboard cannot enter TO_DEV"); + TEST_EQ(vb2_nv_get(ctx, VB2_NV_RECOVERY_REQUEST), 0, " no recovery"); + TEST_EQ(debug_info_displayed, 0, " no debug info"); + TEST_EQ(shutdown_request_calls_left, 0, " timed out"); + i = 0; + TEST_EQ(screens_displayed[i++], VB_SCREEN_RECOVERY_NO_GOOD, + " nogood screen"); + TEST_EQ(screens_displayed[i++], VB_SCREEN_OPTIONS_MENU, + " options menu"); + TEST_EQ(screens_displayed[i++], VB_SCREEN_RECOVERY_NO_GOOD, + " nogood screen"); + TEST_EQ(screens_displayed[i++], VB_SCREEN_BLANK," final blank screen"); + TEST_EQ(screens_count, i, " no extra screens"); + TEST_EQ(beeps_count, 0, " no beeps"); + + /* Untrusted keyboard cannot navigate in TO_DEV menu if already there */ + ResetMocksForManualRecovery(); + i = 0; + mock_keyflags[i] = VB_KEY_FLAG_TRUSTED_KEYBOARD; + mock_keypress[i++] = VB_BUTTON_VOL_UP_DOWN_COMBO_PRESS; // enter to_dev + mock_keypress[i++] = VB_BUTTON_VOL_UP_SHORT_PRESS; // try to confirm... + mock_keypress[i++] = VB_BUTTON_POWER_SHORT_PRESS; + TEST_EQ(VbBootRecoveryLegacyMenu(ctx), VBERROR_SHUTDOWN_REQUESTED, + "Untrusted keyboard cannot navigate in TO_DEV"); + TEST_EQ(vb2_nv_get(ctx, VB2_NV_RECOVERY_REQUEST), 0, " no recovery"); + TEST_EQ(debug_info_displayed, 0, " no debug info"); + TEST_EQ(shutdown_request_calls_left, 0, " timed out"); + i = 0; + TEST_EQ(screens_displayed[i++], VB_SCREEN_RECOVERY_NO_GOOD, + " nogood screen"); + TEST_EQ(screens_displayed[i++], VB_SCREEN_RECOVERY_TO_DEV_MENU, + " todev menu: cancel"); + TEST_EQ(screens_displayed[i++], VB_SCREEN_BLANK, " blank (flash)"); + TEST_EQ(screens_displayed[i++], VB_SCREEN_RECOVERY_TO_DEV_MENU, + " todev menu: cancel"); + TEST_EQ(screens_displayed[i++], VB_SCREEN_RECOVERY_NO_GOOD, + " nogood screen"); + TEST_EQ(screens_displayed[i++], VB_SCREEN_BLANK," final blank screen"); + TEST_EQ(screens_count, i, " no extra screens"); + TEST_EQ(beeps_count, 2, " played error beeps"); + TEST_EQ(beeps_played[0], 400, " first beep"); + TEST_EQ(beeps_played[1], 400, " second beep"); + + /* Handle TPM error in enabling dev mode */ + ResetMocksForManualRecovery(); + i = 0; + mock_keyflags[i] = VB_KEY_FLAG_TRUSTED_KEYBOARD; + mock_keypress[i++] = VB_BUTTON_VOL_UP_DOWN_COMBO_PRESS; // enter to_dev + mock_keyflags[i] = VB_KEY_FLAG_TRUSTED_KEYBOARD; + mock_keypress[i++] = VB_BUTTON_VOL_UP_SHORT_PRESS; // confirm enabling + mock_keypress[i++] = VB_BUTTON_POWER_SHORT_PRESS; + virtdev_retval = VB2_ERROR_MOCK; + TEST_EQ(VbBootRecoveryLegacyMenu(ctx), VBERROR_TPM_SET_BOOT_MODE_STATE, + "todev TPM failure"); + TEST_EQ(vb2_nv_get(ctx, VB2_NV_RECOVERY_REQUEST), 0, " no recovery"); + TEST_EQ(debug_info_displayed, 0, " no debug info"); + TEST_NEQ(shutdown_request_calls_left, 0, " aborted explicitly"); + TEST_EQ(virtdev_set, 1, " virtual dev mode on"); + i = 0; + TEST_EQ(screens_displayed[i++], VB_SCREEN_RECOVERY_NO_GOOD, + " nogood screen"); + TEST_EQ(screens_displayed[i++], VB_SCREEN_RECOVERY_TO_DEV_MENU, + " recovery to_dev menu: cancel"); + TEST_EQ(screens_displayed[i++], VB_SCREEN_RECOVERY_TO_DEV_MENU, + " recovery to_dev menu: confirm disabling os verification"); + TEST_EQ(screens_displayed[i++], VB_SCREEN_BLANK," final blank screen"); + TEST_EQ(screens_count, i, " no extra screens"); + TEST_EQ(beeps_count, 0, " no beeps"); + + /* Cannot enable dev mode if already enabled. */ + ResetMocksForManualRecovery(); + sd->flags |= VB2_SD_FLAG_DEV_MODE_ENABLED; + vbtlk_retval[0] = VB2_ERROR_LK_NO_DISK_FOUND - + VB_DISK_FLAG_REMOVABLE; + i = 0; + mock_keyflags[i] = VB_KEY_FLAG_TRUSTED_KEYBOARD; + mock_keypress[i++] = VB_BUTTON_VOL_UP_DOWN_COMBO_PRESS; // enter to_dev + TEST_EQ(VbBootRecoveryLegacyMenu(ctx), VBERROR_SHUTDOWN_REQUESTED, + "Ctrl+D ignored if already in dev mode"); + TEST_EQ(vb2_nv_get(ctx, VB2_NV_RECOVERY_REQUEST), 0, " no recovery"); + TEST_EQ(debug_info_displayed, 0, " no debug info"); + TEST_EQ(shutdown_request_calls_left, 0, " timed out"); + TEST_EQ(virtdev_set, 0, " virtual dev mode wasn't enabled again"); + i = 0; + TEST_EQ(screens_displayed[i++], VB_SCREEN_RECOVERY_INSERT, + " insert screen"); + TEST_EQ(screens_displayed[i++], VB_SCREEN_BLANK, " blank (flash)"); + TEST_EQ(screens_displayed[i++], VB_SCREEN_RECOVERY_INSERT, + " insert screen"); + TEST_EQ(screens_displayed[i++], VB_SCREEN_BLANK," final blank screen"); + TEST_EQ(screens_count, i, " no extra screens"); + TEST_EQ(beeps_count, 2, " played error beeps"); + TEST_EQ(beeps_played[0], 400, " first beep"); + TEST_EQ(beeps_played[1], 400, " second beep"); + + /* Removing invalid USB drops back to INSERT from TO_DEV menu. */ + ResetMocksForManualRecovery(); + mock_keyflags[0] = VB_KEY_FLAG_TRUSTED_KEYBOARD; + mock_keypress[0] = VB_BUTTON_VOL_UP_DOWN_COMBO_PRESS; // enter TO_DEV + /* asynchronous transition to INSERT before keypress[50] */ + mock_keypress[55] = VB_BUTTON_VOL_UP_SHORT_PRESS; // enter OPTIONS + mock_keypress[56] = VB_BUTTON_VOL_DOWN_SHORT_PRESS; // power off + mock_keypress[57] = VB_BUTTON_POWER_SHORT_PRESS; + vbtlk_retval[0] = VB2_ERROR_MOCK - VB_DISK_FLAG_REMOVABLE; + vbtlk_retval[1] = VB2_ERROR_LK_NO_DISK_FOUND - + VB_DISK_FLAG_REMOVABLE; + TEST_EQ(VbBootRecoveryLegacyMenu(ctx), VBERROR_SHUTDOWN_REQUESTED, + "Drop back to INSERT from TO_DEV when removing invalid USB"); + TEST_NEQ(shutdown_request_calls_left, 0, " powered down explicitly"); + TEST_EQ(virtdev_set, 0, " virtual dev mode off"); + TEST_EQ(vb2_nv_get(ctx, VB2_NV_RECOVERY_REQUEST), 0, " no recovery"); + TEST_EQ(debug_info_displayed, 0, " no debug info"); + TEST_EQ(screens_displayed[0], VB_SCREEN_RECOVERY_NO_GOOD, + " nogood screen"); + TEST_EQ(screens_displayed[1], VB_SCREEN_RECOVERY_TO_DEV_MENU, + " todev menu"); + TEST_EQ(screens_displayed[2], VB_SCREEN_RECOVERY_INSERT, + " insert screen"); + TEST_EQ(screens_displayed[3], VB_SCREEN_OPTIONS_MENU, + " options menu: cancel"); + TEST_EQ(screens_displayed[4], VB_SCREEN_OPTIONS_MENU, + " options menu: power off"); + TEST_EQ(screens_displayed[5], VB_SCREEN_BLANK, " final blank screen"); + TEST_EQ(screens_count, 6, " no extra screens"); + TEST_EQ(beeps_count, 0, " no beeps"); + + /* Plugging in invalid USB drops back to NOGOOD from LANGUAGE. */ + ResetMocksForManualRecovery(); + mock_keypress[0] = VB_BUTTON_VOL_UP_SHORT_PRESS; // enter OPTIONS + mock_keypress[1] = VB_BUTTON_VOL_DOWN_SHORT_PRESS; // power off + mock_keypress[2] = VB_BUTTON_VOL_DOWN_SHORT_PRESS; // language + mock_keypress[3] = VB_BUTTON_POWER_SHORT_PRESS; + vbtlk_retval[0] = VB2_ERROR_LK_NO_DISK_FOUND - + VB_DISK_FLAG_REMOVABLE; + vbtlk_retval[1] = VB2_ERROR_LK_NO_DISK_FOUND - + VB_DISK_FLAG_REMOVABLE; + vbtlk_retval[2] = VB2_ERROR_LK_NO_DISK_FOUND - + VB_DISK_FLAG_REMOVABLE; + vbtlk_retval[3] = VB2_ERROR_LK_NO_DISK_FOUND - + VB_DISK_FLAG_REMOVABLE; + vbtlk_retval[4] = VB2_ERROR_MOCK - VB_DISK_FLAG_REMOVABLE; + TEST_EQ(VbBootRecoveryLegacyMenu(ctx), VBERROR_SHUTDOWN_REQUESTED, + "Drop back to NOGOOD from LANGUAGE when inserting invalid USB"); + TEST_EQ(shutdown_request_calls_left, 0, " timed out"); + TEST_EQ(virtdev_set, 0, " virtual dev mode off"); + TEST_EQ(vb2_nv_get(ctx, VB2_NV_RECOVERY_REQUEST), 0, " no recovery"); + TEST_EQ(debug_info_displayed, 0, " no debug info"); + TEST_EQ(screens_displayed[0], VB_SCREEN_RECOVERY_INSERT, + " insert screen"); + TEST_EQ(screens_displayed[1], VB_SCREEN_OPTIONS_MENU, + " options menu: cancel"); + TEST_EQ(screens_displayed[2], VB_SCREEN_OPTIONS_MENU, + " options menu: power off"); + TEST_EQ(screens_displayed[3], VB_SCREEN_OPTIONS_MENU, + " options menu: language"); + TEST_EQ(screens_displayed[4], VB_SCREEN_LANGUAGES_MENU, + " languages menu"); + TEST_EQ(screens_displayed[5], VB_SCREEN_RECOVERY_NO_GOOD, + " nogood screen"); + TEST_EQ(screens_displayed[6], VB_SCREEN_BLANK, " final blank screen"); + TEST_EQ(screens_count, 7, " no extra screens"); + TEST_EQ(beeps_count, 0, " no beeps"); + + /* Plugging in valid USB boots straight from OPTIONS menu. */ + ResetMocksForManualRecovery(); + mock_keypress[0] = VB_BUTTON_VOL_UP_SHORT_PRESS; // enter OPTIONS + vbtlk_retval[0] = VB2_ERROR_LK_NO_DISK_FOUND - + VB_DISK_FLAG_REMOVABLE; + vbtlk_retval[1] = VB2_SUCCESS - VB_DISK_FLAG_REMOVABLE; + TEST_EQ(VbBootRecoveryLegacyMenu(ctx), VB2_SUCCESS, + "Boot by plugging in USB straight from OPTIONS menu"); + TEST_NEQ(shutdown_request_calls_left, 0, " booted explicitly"); + TEST_EQ(virtdev_set, 0, " virtual dev mode off"); + TEST_EQ(vb2_nv_get(ctx, VB2_NV_RECOVERY_REQUEST), 0, " no recovery"); + TEST_EQ(debug_info_displayed, 0, " no debug info"); + TEST_EQ(screens_displayed[0], VB_SCREEN_RECOVERY_INSERT, + " insert screen"); + TEST_EQ(screens_displayed[1], VB_SCREEN_OPTIONS_MENU, + " options menu: cancel"); + TEST_EQ(screens_displayed[2], VB_SCREEN_BLANK, " final blank screen"); + TEST_EQ(screens_count, 3, " no extra screens"); + TEST_EQ(beeps_count, 0, " no beeps"); + + printf("...done.\n"); +} + +static void VbTestLanguageMenu(void) +{ + int i; + + printf("Testing VbTestLanguageMenu()...\n"); + + /* Navigate to language menu from BROKEN */ + ResetMocks(); + i = 0; + mock_keypress[i++] = VB_BUTTON_VOL_UP_SHORT_PRESS; // enter OPTIONS + mock_keypress[i++] = VB_BUTTON_VOL_DOWN_SHORT_PRESS; // power off + mock_keypress[i++] = VB_BUTTON_VOL_DOWN_SHORT_PRESS; // languages + mock_keypress[i++] = VB_BUTTON_POWER_SHORT_PRESS; // enter languages + mock_keypress[i++] = VB_BUTTON_POWER_SHORT_PRESS; // select current lang + mock_keypress[i++] = VB_BUTTON_POWER_SHORT_PRESS; // cancel -> BROKEN + mock_keypress[i++] = VB_BUTTON_POWER_SHORT_PRESS; // power off + TEST_EQ(VbBootRecoveryLegacyMenu(ctx), VBERROR_SHUTDOWN_REQUESTED, + "go to language menu from BROKEN"); + TEST_EQ(debug_info_displayed, 0, " no debug info"); + TEST_NEQ(shutdown_request_calls_left, 0, " powered down explicitly"); + i = 0; + TEST_EQ(screens_displayed[i++], VB_SCREEN_OS_BROKEN, + " broken screen"); + TEST_EQ(screens_displayed[i++], VB_SCREEN_OPTIONS_MENU, + " options menu: cancel"); + TEST_EQ(screens_displayed[i++], VB_SCREEN_OPTIONS_MENU, + " options menu: power off"); + TEST_EQ(screens_displayed[i++], VB_SCREEN_OPTIONS_MENU, + " options menu: language"); + TEST_EQ(screens_displayed[i++], VB_SCREEN_LANGUAGES_MENU, + " language menu"); + TEST_EQ(screens_displayed[i++], VB_SCREEN_OPTIONS_MENU, + " options menu: cancel"); + TEST_EQ(screens_displayed[i++], VB_SCREEN_OS_BROKEN, + " broken screen"); + TEST_EQ(screens_displayed[i++], VB_SCREEN_BLANK," final blank screen"); + TEST_EQ(screens_count, i, " no extra screens"); + TEST_EQ(beeps_count, 0, " no beeps"); + + /* Navigate to all language menus from recovery */ + ResetMocksForManualRecovery(); + vbtlk_retval[0] = VB2_ERROR_LK_NO_DISK_FOUND - + VB_DISK_FLAG_REMOVABLE; + i = 0; + mock_keypress[i++] = VB_BUTTON_VOL_UP_SHORT_PRESS; // enter OPTIONS + mock_keypress[i++] = VB_BUTTON_VOL_DOWN_SHORT_PRESS; // power off + mock_keypress[i++] = VB_BUTTON_VOL_DOWN_SHORT_PRESS; // languages + mock_keypress[i++] = VB_BUTTON_POWER_SHORT_PRESS; + mock_keypress[i++] = VB_BUTTON_POWER_SHORT_PRESS; // select current lang + mock_keyflags[i] = VB_KEY_FLAG_TRUSTED_KEYBOARD; + mock_keypress[i++] = VB_BUTTON_VOL_UP_DOWN_COMBO_PRESS; // enter TO_DEV + mock_keyflags[i] = VB_KEY_FLAG_TRUSTED_KEYBOARD; + mock_keypress[i++] = VB_BUTTON_VOL_UP_SHORT_PRESS; // confirm disabling + mock_keyflags[i] = VB_KEY_FLAG_TRUSTED_KEYBOARD; + mock_keypress[i++] = VB_BUTTON_VOL_DOWN_SHORT_PRESS; // cancel + mock_keyflags[i] = VB_KEY_FLAG_TRUSTED_KEYBOARD; + mock_keypress[i++] = VB_BUTTON_VOL_DOWN_SHORT_PRESS; // power off + mock_keyflags[i] = VB_KEY_FLAG_TRUSTED_KEYBOARD; + mock_keypress[i++] = VB_BUTTON_VOL_DOWN_SHORT_PRESS; // language + mock_keypress[i++] = VB_BUTTON_POWER_SHORT_PRESS; + mock_keypress[i++] = VB_BUTTON_POWER_SHORT_PRESS; // select current lang + mock_keyflags[i] = VB_KEY_FLAG_TRUSTED_KEYBOARD; + mock_keypress[i++] = VB_BUTTON_VOL_DOWN_SHORT_PRESS; // power off + mock_keypress[i++] = VB_BUTTON_POWER_SHORT_PRESS; + TEST_EQ(VbBootRecoveryLegacyMenu(ctx), VBERROR_SHUTDOWN_REQUESTED, + "go to language menus from INSERT"); + TEST_EQ(debug_info_displayed, 0, " no debug info"); + TEST_NEQ(shutdown_request_calls_left, 0, " powered down explicitly"); + i = 0; + TEST_EQ(screens_displayed[i++], VB_SCREEN_RECOVERY_INSERT, + " insert screen"); + TEST_EQ(screens_displayed[i++], VB_SCREEN_OPTIONS_MENU, + " options menu: cancel"); + TEST_EQ(screens_displayed[i++], VB_SCREEN_OPTIONS_MENU, + " options menu: power off"); + TEST_EQ(screens_displayed[i++], VB_SCREEN_OPTIONS_MENU, + " options menu: language"); + TEST_EQ(screens_displayed[i++], VB_SCREEN_LANGUAGES_MENU, + " language menu"); + TEST_EQ(screens_displayed[i++], VB_SCREEN_OPTIONS_MENU, + " options menu: cancel"); + TEST_EQ(screens_displayed[i++], VB_SCREEN_RECOVERY_TO_DEV_MENU, + " to dev menu: cancel"); + TEST_EQ(screens_displayed[i++], VB_SCREEN_RECOVERY_TO_DEV_MENU, + " to dev menu: confirm disabling os verification"); + TEST_EQ(screens_displayed[i++], VB_SCREEN_RECOVERY_TO_DEV_MENU, + " to dev menu: cancel"); + TEST_EQ(screens_displayed[i++], VB_SCREEN_RECOVERY_TO_DEV_MENU, + " to dev menu: power_off"); + TEST_EQ(screens_displayed[i++], VB_SCREEN_RECOVERY_TO_DEV_MENU, + " to dev menu: language"); + TEST_EQ(screens_displayed[i++], VB_SCREEN_LANGUAGES_MENU, + " language menu"); + TEST_EQ(screens_displayed[i++], VB_SCREEN_RECOVERY_TO_DEV_MENU, + " to dev menu: cancel"); + TEST_EQ(screens_displayed[i++], VB_SCREEN_RECOVERY_TO_DEV_MENU, + " to dev menu: power off"); + TEST_EQ(screens_displayed[i++], VB_SCREEN_BLANK, " final blank screen"); + TEST_EQ(screens_count, i, " no extra screens"); + TEST_EQ(beeps_count, 0, " no beeps"); + + /* Navigate to all language menus from developer menu */ + ResetMocksForDeveloper(); + i = 0; + mock_keypress[i++] = VB_BUTTON_VOL_DOWN_SHORT_PRESS; // language + mock_keypress[i++] = VB_BUTTON_POWER_SHORT_PRESS; + mock_keypress[i++] = VB_BUTTON_POWER_SHORT_PRESS; // select current lang + mock_keypress[i++] = VB_BUTTON_VOL_UP_SHORT_PRESS; // enable OS verif + mock_keypress[i++] = VB_BUTTON_POWER_SHORT_PRESS; + mock_keypress[i++] = VB_BUTTON_VOL_DOWN_SHORT_PRESS; // cancel + mock_keypress[i++] = VB_BUTTON_VOL_DOWN_SHORT_PRESS; // power off + mock_keypress[i++] = VB_BUTTON_VOL_DOWN_SHORT_PRESS; // language + mock_keypress[i++] = VB_BUTTON_POWER_SHORT_PRESS; + mock_keypress[i++] = VB_BUTTON_POWER_SHORT_PRESS; // select current lang + mock_keypress[i++] = VB_BUTTON_VOL_DOWN_SHORT_PRESS; // cancel + mock_keypress[i++] = VB_BUTTON_POWER_SHORT_PRESS; // return to dev_warn + mock_keypress[i++] = VB_BUTTON_VOL_UP_SHORT_PRESS; // enable OS verif + mock_keypress[i++] = VB_BUTTON_VOL_UP_SHORT_PRESS; // show debug info + mock_keypress[i++] = VB_BUTTON_VOL_UP_SHORT_PRESS; // developer options + mock_keypress[i++] = VB_BUTTON_POWER_SHORT_PRESS; + mock_keypress[i++] = VB_BUTTON_VOL_DOWN_SHORT_PRESS; // cancel + mock_keypress[i++] = VB_BUTTON_VOL_DOWN_SHORT_PRESS; // power off + mock_keypress[i++] = VB_BUTTON_VOL_DOWN_SHORT_PRESS; // language + mock_keypress[i++] = VB_BUTTON_POWER_SHORT_PRESS; + mock_keypress[i++] = VB_BUTTON_POWER_SHORT_PRESS; // select current lang + mock_keypress[i++] = VB_BUTTON_VOL_DOWN_SHORT_PRESS; // cancel + mock_keypress[i++] = VB_BUTTON_VOL_DOWN_SHORT_PRESS; // power off + mock_keypress[i++] = VB_BUTTON_POWER_SHORT_PRESS; + TEST_EQ(VbBootDeveloperLegacyMenu(ctx), VBERROR_SHUTDOWN_REQUESTED, + " scroll through all language menus in developer options"); + TEST_EQ(debug_info_displayed, 0, " no debug info"); + TEST_NEQ(shutdown_request_calls_left, 0, " powered down explicitly"); + i = 0; + TEST_EQ(screens_displayed[i++], VB_SCREEN_DEVELOPER_WARNING_MENU, + " warning screen: power off"); + TEST_EQ(screens_displayed[i++], VB_SCREEN_DEVELOPER_WARNING_MENU, + " warning screen: language"); + TEST_EQ(screens_displayed[i++], VB_SCREEN_LANGUAGES_MENU, + " language menu: select current language"); + TEST_EQ(screens_displayed[i++], VB_SCREEN_DEVELOPER_WARNING_MENU, + " warning screen: cancel "); + TEST_EQ(screens_displayed[i++], VB_SCREEN_DEVELOPER_WARNING_MENU, + " warning screen: enable root verification"); + TEST_EQ(screens_displayed[i++], VB_SCREEN_DEVELOPER_TO_NORM_MENU, + " to norm screen: cancel"); + TEST_EQ(screens_displayed[i++], VB_SCREEN_DEVELOPER_TO_NORM_MENU, + " to norm screen: power off"); + TEST_EQ(screens_displayed[i++], VB_SCREEN_DEVELOPER_TO_NORM_MENU, + " to norm screen: language"); + TEST_EQ(screens_displayed[i++], VB_SCREEN_DEVELOPER_TO_NORM_MENU, + " to norm screen: language"); + TEST_EQ(screens_displayed[i++], VB_SCREEN_LANGUAGES_MENU, + " language menu: select current language"); + TEST_EQ(screens_displayed[i++], VB_SCREEN_DEVELOPER_TO_NORM_MENU, + " to norm screen: confirm enabling os verification"); + TEST_EQ(screens_displayed[i++], VB_SCREEN_DEVELOPER_TO_NORM_MENU, + " to norm screen: cancel"); + TEST_EQ(screens_displayed[i++], VB_SCREEN_DEVELOPER_WARNING_MENU, + " warning screen: enable root verification"); + TEST_EQ(screens_displayed[i++], VB_SCREEN_DEVELOPER_WARNING_MENU, + " warning screen: show debug info"); + TEST_EQ(screens_displayed[i++], VB_SCREEN_DEVELOPER_WARNING_MENU, + " warning screen: developer options"); + TEST_EQ(screens_displayed[i++], VB_SCREEN_DEVELOPER_WARNING_MENU, + " select developer options"); + TEST_EQ(screens_displayed[i++], VB_SCREEN_DEVELOPER_MENU, + " developer menu: boot developer image"); + TEST_EQ(screens_displayed[i++], VB_SCREEN_DEVELOPER_MENU, + " developer menu: cancel"); + TEST_EQ(screens_displayed[i++], VB_SCREEN_DEVELOPER_MENU, + " developer menu: power off"); + TEST_EQ(screens_displayed[i++], VB_SCREEN_DEVELOPER_MENU, + " developer menu: language"); + TEST_EQ(screens_displayed[i++], VB_SCREEN_LANGUAGES_MENU, + " language menu"); + TEST_EQ(screens_displayed[i++], VB_SCREEN_DEVELOPER_MENU, + " developer menu: boot from disk"); + TEST_EQ(screens_displayed[i++], VB_SCREEN_DEVELOPER_MENU, + " developer menu: cancel"); + TEST_EQ(screens_displayed[i++], VB_SCREEN_DEVELOPER_MENU, + " developer menu: power off"); + TEST_EQ(screens_displayed[i++], VB_SCREEN_BLANK," final blank screen"); + TEST_EQ(screens_count, i, " no extra screens"); + TEST_EQ(beeps_count, 0, " no beeps"); + + printf("...done.\n"); +} + +static void VbNavigationTest(void) +{ + int i; + + printf("Testing long navigation sequences..."); + + /* + * NOGOOD, OPTIONS, LANGUAGE, TODEV, LANGUAGE, TODEV, + * LANGUAGE, select, Cancel, OPTIONS, LANGUAGE + */ + ResetMocksForManualRecovery(); + i = 0; + mock_keypress[i++] = VB_BUTTON_VOL_UP_SHORT_PRESS; // enter OPTIONS + mock_keypress[i++] = VB_BUTTON_VOL_DOWN_SHORT_PRESS; // power off + mock_keypress[i++] = VB_BUTTON_VOL_DOWN_SHORT_PRESS; // language + mock_keypress[i++] = VB_BUTTON_VOL_DOWN_SHORT_PRESS; // (end of menu) + mock_keypress[i++] = VB_BUTTON_VOL_DOWN_SHORT_PRESS; // (end of menu) + mock_keypress[i++] = VB_BUTTON_POWER_SHORT_PRESS; + mock_keyflags[i] = VB_KEY_FLAG_TRUSTED_KEYBOARD; + mock_keypress[i++] = VB_BUTTON_VOL_UP_DOWN_COMBO_PRESS; // enter TO_DEV + mock_keyflags[i] = VB_KEY_FLAG_TRUSTED_KEYBOARD; + mock_keypress[i++] = VB_BUTTON_VOL_UP_SHORT_PRESS; // confirm enabling + mock_keyflags[i] = VB_KEY_FLAG_TRUSTED_KEYBOARD; + mock_keypress[i++] = VB_BUTTON_VOL_UP_SHORT_PRESS; // (end of menu) + mock_keyflags[i] = VB_KEY_FLAG_TRUSTED_KEYBOARD; + mock_keypress[i++] = VB_BUTTON_VOL_DOWN_SHORT_PRESS; // cancel + mock_keyflags[i] = VB_KEY_FLAG_TRUSTED_KEYBOARD; + mock_keypress[i++] = VB_BUTTON_VOL_DOWN_SHORT_PRESS; // power off + mock_keyflags[i] = VB_KEY_FLAG_TRUSTED_KEYBOARD; + mock_keypress[i++] = VB_BUTTON_VOL_DOWN_SHORT_PRESS; // language + mock_keypress[i++] = VB_BUTTON_POWER_SHORT_PRESS; + mock_keyflags[i] = VB_KEY_FLAG_TRUSTED_KEYBOARD; + mock_keypress[i++] = VB_BUTTON_VOL_UP_DOWN_COMBO_PRESS; // enter TO_DEV + mock_keyflags[i] = VB_KEY_FLAG_TRUSTED_KEYBOARD; + mock_keypress[i++] = VB_BUTTON_VOL_DOWN_SHORT_PRESS; // power off + mock_keyflags[i] = VB_KEY_FLAG_TRUSTED_KEYBOARD; + mock_keypress[i++] = VB_BUTTON_VOL_DOWN_SHORT_PRESS; // language + mock_keypress[i++] = VB_BUTTON_POWER_SHORT_PRESS; + mock_keypress[i++] = VB_BUTTON_POWER_SHORT_PRESS; // select current lang + mock_keyflags[i] = VB_KEY_FLAG_TRUSTED_KEYBOARD; + mock_keypress[i++] = VB_BUTTON_VOL_DOWN_SHORT_PRESS; // power off + mock_keyflags[i] = VB_KEY_FLAG_TRUSTED_KEYBOARD; + mock_keypress[i++] = VB_BUTTON_VOL_DOWN_SHORT_PRESS; // language + mock_keyflags[i] = VB_KEY_FLAG_TRUSTED_KEYBOARD; + mock_keypress[i++] = VB_BUTTON_VOL_DOWN_SHORT_PRESS; // (end of menu) + mock_keyflags[i] = VB_KEY_FLAG_TRUSTED_KEYBOARD; + mock_keypress[i++] = VB_BUTTON_VOL_DOWN_SHORT_PRESS; // (end of menu) + mock_keyflags[i] = VB_KEY_FLAG_TRUSTED_KEYBOARD; + mock_keypress[i++] = VB_BUTTON_VOL_UP_SHORT_PRESS; // power off + mock_keyflags[i] = VB_KEY_FLAG_TRUSTED_KEYBOARD; + mock_keypress[i++] = VB_BUTTON_VOL_UP_SHORT_PRESS; // cancel + mock_keypress[i++] = VB_BUTTON_POWER_SHORT_PRESS; + mock_keypress[i++] = VB_BUTTON_VOL_UP_SHORT_PRESS; // enter OPTIONS + mock_keypress[i++] = VB_BUTTON_VOL_UP_SHORT_PRESS; // show debug info + mock_keypress[i++] = VB_BUTTON_VOL_UP_SHORT_PRESS; // (end of menu) + mock_keypress[i++] = VB_BUTTON_POWER_SHORT_PRESS; + mock_keypress[i++] = VB_BUTTON_VOL_DOWN_SHORT_PRESS; // cancel + mock_keypress[i++] = VB_BUTTON_VOL_DOWN_SHORT_PRESS; // power off + mock_keypress[i++] = VB_BUTTON_VOL_DOWN_SHORT_PRESS; // language + mock_keypress[i++] = VB_BUTTON_POWER_SHORT_PRESS; + TEST_EQ(VbBootRecoveryLegacyMenu(ctx), VBERROR_SHUTDOWN_REQUESTED, + "recovery mode long navigation"); + TEST_EQ(debug_info_displayed, 1, " showed debug info"); + TEST_EQ(shutdown_request_calls_left, 0, " timed out"); + i = 0; + TEST_EQ(screens_displayed[i++], VB_SCREEN_RECOVERY_NO_GOOD, + " nogood screen"); + TEST_EQ(screens_displayed[i++], VB_SCREEN_OPTIONS_MENU, + " options: cancel"); + TEST_EQ(screens_displayed[i++], VB_SCREEN_OPTIONS_MENU, + " options: power off"); + TEST_EQ(screens_displayed[i++], VB_SCREEN_OPTIONS_MENU, + " options: language"); + TEST_EQ(screens_displayed[i++], VB_SCREEN_OPTIONS_MENU, + " options: language"); + TEST_EQ(screens_displayed[i++], VB_SCREEN_OPTIONS_MENU, + " options: language"); + TEST_EQ(screens_displayed[i++], VB_SCREEN_LANGUAGES_MENU, + " languages menu"); + TEST_EQ(screens_displayed[i++], VB_SCREEN_RECOVERY_TO_DEV_MENU, + " todev: cancel"); + TEST_EQ(screens_displayed[i++], VB_SCREEN_RECOVERY_TO_DEV_MENU, + " todev: confirm enabling"); + TEST_EQ(screens_displayed[i++], VB_SCREEN_RECOVERY_TO_DEV_MENU, + " todev: confirm enabling"); + TEST_EQ(screens_displayed[i++], VB_SCREEN_RECOVERY_TO_DEV_MENU, + " todev: cancel"); + TEST_EQ(screens_displayed[i++], VB_SCREEN_RECOVERY_TO_DEV_MENU, + " todev: power off"); + TEST_EQ(screens_displayed[i++], VB_SCREEN_RECOVERY_TO_DEV_MENU, + " todev: language"); + TEST_EQ(screens_displayed[i++], VB_SCREEN_LANGUAGES_MENU, + " languages menu"); + TEST_EQ(screens_displayed[i++], VB_SCREEN_RECOVERY_TO_DEV_MENU, + " todev: cancel"); + TEST_EQ(screens_displayed[i++], VB_SCREEN_RECOVERY_TO_DEV_MENU, + " todev: power off"); + TEST_EQ(screens_displayed[i++], VB_SCREEN_RECOVERY_TO_DEV_MENU, + " todev: language"); + TEST_EQ(screens_displayed[i++], VB_SCREEN_LANGUAGES_MENU, + " languages menu"); + TEST_EQ(screens_displayed[i++], VB_SCREEN_RECOVERY_TO_DEV_MENU, + " todev: cancel"); + TEST_EQ(screens_displayed[i++], VB_SCREEN_RECOVERY_TO_DEV_MENU, + " todev: power off"); + TEST_EQ(screens_displayed[i++], VB_SCREEN_RECOVERY_TO_DEV_MENU, + " todev: language"); + TEST_EQ(screens_displayed[i++], VB_SCREEN_RECOVERY_TO_DEV_MENU, + " todev: language"); + TEST_EQ(screens_displayed[i++], VB_SCREEN_RECOVERY_TO_DEV_MENU, + " todev: language"); + TEST_EQ(screens_displayed[i++], VB_SCREEN_RECOVERY_TO_DEV_MENU, + " todev: power off"); + TEST_EQ(screens_displayed[i++], VB_SCREEN_RECOVERY_TO_DEV_MENU, + " todev: cancel"); + TEST_EQ(screens_displayed[i++], VB_SCREEN_RECOVERY_NO_GOOD, + " nogood screen"); + TEST_EQ(screens_displayed[i++], VB_SCREEN_OPTIONS_MENU, + " options: cancel"); + TEST_EQ(screens_displayed[i++], VB_SCREEN_OPTIONS_MENU, + " options: show debug info"); + TEST_EQ(screens_displayed[i++], VB_SCREEN_OPTIONS_MENU, + " options: show debug info"); + TEST_EQ(screens_displayed[i++], VB_SCREEN_OPTIONS_MENU, + " options: cancel"); + TEST_EQ(screens_displayed[i++], VB_SCREEN_OPTIONS_MENU, + " options: power off"); + TEST_EQ(screens_displayed[i++], VB_SCREEN_OPTIONS_MENU, + " options: language"); + TEST_EQ(screens_displayed[i++], VB_SCREEN_LANGUAGES_MENU, + " languages menu"); + TEST_EQ(screens_displayed[i++], VB_SCREEN_BLANK," final blank screen"); + TEST_EQ(screens_count, i, " no extra screens"); + TEST_EQ(beeps_count, 0, " no beeps"); + + /* DEVELOPER, Cancel, Show Debug, TO_NORM, Cancel, Boot Legacy */ + ResetMocksForDeveloper(); + i = 0; + mock_keypress[i++] = VB_BUTTON_VOL_UP_SHORT_PRESS; // Enable OS verif + mock_keypress[i++] = VB_BUTTON_VOL_UP_SHORT_PRESS; // Show Debug Info + mock_keypress[i++] = VB_BUTTON_VOL_UP_SHORT_PRESS; // Developer Options + mock_keypress[i++] = VB_BUTTON_VOL_UP_SHORT_PRESS; // (end of menu) + mock_keypress[i++] = VB_BUTTON_VOL_UP_SHORT_PRESS; // (end of menu) + mock_keypress[i++] = VB_BUTTON_POWER_SHORT_PRESS; + mock_keypress[i++] = VB_BUTTON_VOL_UP_SHORT_PRESS; // Boot From USB + mock_keypress[i++] = VB_BUTTON_VOL_UP_SHORT_PRESS; // Boot Legacy + mock_keypress[i++] = VB_BUTTON_VOL_UP_SHORT_PRESS; // (end of menu) + mock_keypress[i++] = VB_BUTTON_VOL_DOWN_SHORT_PRESS; // Boot From USB + mock_keypress[i++] = VB_BUTTON_VOL_DOWN_SHORT_PRESS; // Boot From Disk + mock_keypress[i++] = VB_BUTTON_VOL_DOWN_SHORT_PRESS; // Cancel + mock_keypress[i++] = VB_BUTTON_POWER_SHORT_PRESS; + mock_keypress[i++] = VB_BUTTON_VOL_UP_SHORT_PRESS; // Enable OS verif + mock_keypress[i++] = VB_BUTTON_VOL_UP_SHORT_PRESS; // Show Debug Info + mock_keypress[i++] = VB_BUTTON_POWER_SHORT_PRESS; + mock_keypress[i++] = VB_BUTTON_VOL_DOWN_SHORT_PRESS; // Enable OS verif + mock_keypress[i++] = VB_BUTTON_POWER_SHORT_PRESS; + mock_keypress[i++] = VB_BUTTON_VOL_DOWN_SHORT_PRESS; // Cancel + mock_keypress[i++] = VB_BUTTON_VOL_DOWN_SHORT_PRESS; // Power Off + mock_keypress[i++] = VB_BUTTON_VOL_DOWN_SHORT_PRESS; // Language + mock_keypress[i++] = VB_BUTTON_VOL_DOWN_SHORT_PRESS; // (end of menu) + mock_keypress[i++] = VB_BUTTON_VOL_DOWN_SHORT_PRESS; // (end of menu) + mock_keypress[i++] = VB_BUTTON_VOL_UP_SHORT_PRESS; // Power Off + mock_keypress[i++] = VB_BUTTON_VOL_UP_SHORT_PRESS; // Cancel + mock_keypress[i++] = VB_BUTTON_POWER_SHORT_PRESS; + mock_keypress[i++] = VB_BUTTON_VOL_UP_SHORT_PRESS; // Enable OS verif + mock_keypress[i++] = VB_BUTTON_VOL_UP_SHORT_PRESS; // Show Debug Info + mock_keypress[i++] = VB_BUTTON_VOL_UP_SHORT_PRESS; // Developer Options + mock_keypress[i++] = VB_BUTTON_POWER_SHORT_PRESS; + mock_keypress[i++] = VB_BUTTON_VOL_DOWN_SHORT_PRESS; // Cancel + mock_keypress[i++] = VB_BUTTON_VOL_DOWN_SHORT_PRESS; // Power Off + mock_keypress[i++] = VB_BUTTON_VOL_DOWN_SHORT_PRESS; // Language + mock_keypress[i++] = VB_BUTTON_VOL_UP_SHORT_PRESS; // Power Off + mock_keypress[i++] = VB_BUTTON_VOL_UP_SHORT_PRESS; // Cancel + mock_keypress[i++] = VB_BUTTON_VOL_UP_SHORT_PRESS; // Boot From Disk + mock_keypress[i++] = VB_BUTTON_VOL_UP_SHORT_PRESS; // Boot From USB + mock_keypress[i++] = VB_BUTTON_VOL_UP_SHORT_PRESS; // Boot Legacy + mock_keypress[i++] = VB_BUTTON_VOL_UP_SHORT_PRESS; // (end of menu) + mock_keypress[i++] = VB_BUTTON_POWER_SHORT_PRESS; // Select altfw menu + mock_keypress[i++] = VB_BUTTON_VOL_DOWN_SHORT_PRESS; // Second altfw + mock_keypress[i++] = VB_BUTTON_VOL_DOWN_SHORT_PRESS; // Cancel + mock_keypress[i++] = VB_BUTTON_VOL_DOWN_SHORT_PRESS; // (end of menu) + mock_keypress[i++] = VB_BUTTON_POWER_SHORT_PRESS; // Back to dev options + mock_keypress[i++] = VB_BUTTON_VOL_UP_SHORT_PRESS; // Boot From Disk + mock_keypress[i++] = VB_BUTTON_VOL_UP_SHORT_PRESS; // Boot From USB + mock_keypress[i++] = VB_BUTTON_VOL_UP_SHORT_PRESS; // Boot Legacy + mock_keypress[i++] = VB_BUTTON_POWER_SHORT_PRESS; // Select altfw menu + mock_keypress[i++] = VB_BUTTON_VOL_DOWN_SHORT_PRESS; // Second altfw + mock_keypress[i++] = VB_BUTTON_POWER_SHORT_PRESS; // Select 2nd altfw + vb2_nv_set(ctx, VB2_NV_DEV_BOOT_LEGACY, 1); + TEST_EQ(VbBootDeveloperLegacyMenu(ctx), vbtlk_retval_fixed, + "developer mode long navigation"); + TEST_EQ(debug_info_displayed, 1, " showed debug info"); + TEST_EQ(vbexlegacy_called, 1, " tried legacy"); + TEST_EQ(altfw_num, 2, " check altfw_num"); + TEST_EQ(audio_looping_calls_left, 0, " audio timeout"); + TEST_EQ(vb2_nv_get(ctx, VB2_NV_RECOVERY_REQUEST), 0, " no recovery"); + i = 0; + TEST_EQ(screens_displayed[i++], VB_SCREEN_DEVELOPER_WARNING_MENU, + " dev warning menu: power off"); + TEST_EQ(screens_displayed[i++], VB_SCREEN_DEVELOPER_WARNING_MENU, + " dev warning menu: enable root verification"); + TEST_EQ(screens_displayed[i++], VB_SCREEN_DEVELOPER_WARNING_MENU, + " dev warning menu: show debug info"); + TEST_EQ(screens_displayed[i++], VB_SCREEN_DEVELOPER_WARNING_MENU, + " dev warning menu: developer options"); + TEST_EQ(screens_displayed[i++], VB_SCREEN_DEVELOPER_WARNING_MENU, + " dev warning menu: developer options"); + TEST_EQ(screens_displayed[i++], VB_SCREEN_DEVELOPER_WARNING_MENU, + " dev warning menu: developer options"); + TEST_EQ(screens_displayed[i++], VB_SCREEN_DEVELOPER_MENU, + " dev menu: disk boot"); + TEST_EQ(screens_displayed[i++], VB_SCREEN_DEVELOPER_MENU, + " dev menu: USB boot"); + TEST_EQ(screens_displayed[i++], VB_SCREEN_DEVELOPER_MENU, + " dev menu: Legacy boot"); + TEST_EQ(screens_displayed[i++], VB_SCREEN_DEVELOPER_MENU, + " dev menu: Legacy boot"); + TEST_EQ(screens_displayed[i++], VB_SCREEN_DEVELOPER_MENU, + " dev menu: USB boot"); + TEST_EQ(screens_displayed[i++], VB_SCREEN_DEVELOPER_MENU, + " dev menu: disk boot"); + TEST_EQ(screens_displayed[i++], VB_SCREEN_DEVELOPER_MENU, + " dev menu: cancel"); + TEST_EQ(screens_displayed[i++], VB_SCREEN_DEVELOPER_WARNING_MENU, + " dev warning menu: power off"); + TEST_EQ(screens_displayed[i++], VB_SCREEN_DEVELOPER_WARNING_MENU, + " dev warning menu: enable root verification"); + TEST_EQ(screens_displayed[i++], VB_SCREEN_DEVELOPER_WARNING_MENU, + " dev warning menu: show debug info"); + TEST_EQ(screens_displayed[i++], VB_SCREEN_DEVELOPER_WARNING_MENU, + " dev warning menu: enable root verification"); + TEST_EQ(screens_displayed[i++], VB_SCREEN_DEVELOPER_TO_NORM_MENU, + " to dev menu: confirm enabling"); + TEST_EQ(screens_displayed[i++], VB_SCREEN_DEVELOPER_TO_NORM_MENU, + " to dev menu: cancel"); + TEST_EQ(screens_displayed[i++], VB_SCREEN_DEVELOPER_TO_NORM_MENU, + " to dev menu: power off"); + TEST_EQ(screens_displayed[i++], VB_SCREEN_DEVELOPER_TO_NORM_MENU, + " to dev menu: language"); + TEST_EQ(screens_displayed[i++], VB_SCREEN_DEVELOPER_TO_NORM_MENU, + " to dev menu: language"); + TEST_EQ(screens_displayed[i++], VB_SCREEN_DEVELOPER_TO_NORM_MENU, + " to dev menu: language"); + TEST_EQ(screens_displayed[i++], VB_SCREEN_DEVELOPER_TO_NORM_MENU, + " to dev menu: power off"); + TEST_EQ(screens_displayed[i++], VB_SCREEN_DEVELOPER_TO_NORM_MENU, + " to dev menu: cancel"); + TEST_EQ(screens_displayed[i++], VB_SCREEN_DEVELOPER_WARNING_MENU, + " dev warning menu: power off"); + TEST_EQ(screens_displayed[i++], VB_SCREEN_DEVELOPER_WARNING_MENU, + " dev warning menu: enable root verification"); + TEST_EQ(screens_displayed[i++], VB_SCREEN_DEVELOPER_WARNING_MENU, + " dev warning menu: show debug info"); + TEST_EQ(screens_displayed[i++], VB_SCREEN_DEVELOPER_WARNING_MENU, + " dev warning menu: developer options"); + TEST_EQ(screens_displayed[i++], VB_SCREEN_DEVELOPER_MENU, + " dev menu: boot from disk"); + TEST_EQ(screens_displayed[i++], VB_SCREEN_DEVELOPER_MENU, + " dev menu: cancel"); + TEST_EQ(screens_displayed[i++], VB_SCREEN_DEVELOPER_MENU, + " dev menu: power off"); + TEST_EQ(screens_displayed[i++], VB_SCREEN_DEVELOPER_MENU, + " dev menu: language"); + TEST_EQ(screens_displayed[i++], VB_SCREEN_DEVELOPER_MENU, + " dev menu: power off"); + TEST_EQ(screens_displayed[i++], VB_SCREEN_DEVELOPER_MENU, + " dev menu: cancel"); + TEST_EQ(screens_displayed[i++], VB_SCREEN_DEVELOPER_MENU, + " dev menu: boot from disk"); + TEST_EQ(screens_displayed[i++], VB_SCREEN_DEVELOPER_MENU, + " dev menu: boot from USB"); + TEST_EQ(screens_displayed[i++], VB_SCREEN_DEVELOPER_MENU, + " dev menu: boot legacy"); + TEST_EQ(screens_displayed[i++], VB_SCREEN_DEVELOPER_MENU, + " dev menu: boot legacy"); + TEST_EQ(screens_displayed[i++], VB_SCREEN_ALT_FW_MENU, " altfw 1"); + TEST_EQ(screens_displayed[i++], VB_SCREEN_ALT_FW_MENU, " altfw 2"); + TEST_EQ(screens_displayed[i++], VB_SCREEN_ALT_FW_MENU, + " altfw cancel"); + TEST_EQ(screens_displayed[i++], VB_SCREEN_ALT_FW_MENU, + " altfw end"); + TEST_EQ(screens_displayed[i++], VB_SCREEN_DEVELOPER_MENU, + " dev menu: cancel"); + TEST_EQ(screens_displayed[i++], VB_SCREEN_DEVELOPER_MENU, + " dev menu: boot from disk"); + TEST_EQ(screens_displayed[i++], VB_SCREEN_DEVELOPER_MENU, + " dev menu: boot from USB"); + TEST_EQ(screens_displayed[i++], VB_SCREEN_DEVELOPER_MENU, + " dev menu: boot legacy"); + TEST_EQ(screens_displayed[i++], VB_SCREEN_ALT_FW_MENU, " altfw"); + TEST_EQ(screens_displayed[i++], VB_SCREEN_ALT_FW_MENU, " altfw"); + TEST_EQ(screens_displayed[i++], VB_SCREEN_BLANK, " blank (flash)"); + TEST_EQ(screens_displayed[i++], VB_SCREEN_ALT_FW_MENU, " altfw"); + TEST_EQ(screens_displayed[i++], VB_SCREEN_BLANK," final blank screen"); + TEST_EQ(screens_count, i, " no extra screens"); + TEST_EQ(beeps_count, 1, " error beep: legacy BIOS not found"); + TEST_EQ(beeps_played[0], 200, " low-frequency error beep"); +} + +int main(void) +{ + VbBootDevTest(); + VbBootRecTest(); + VbTestLanguageMenu(); + VbNavigationTest(); + + return gTestSuccess ? 0 : 255; +} |