diff options
Diffstat (limited to 'tests/vboot_ui_legacy_menu_tests.c')
| -rw-r--r-- | tests/vboot_ui_legacy_menu_tests.c | 2349 |
1 files changed, 0 insertions, 2349 deletions
diff --git a/tests/vboot_ui_legacy_menu_tests.c b/tests/vboot_ui_legacy_menu_tests.c deleted file mode 100644 index 1dc92535..00000000 --- a/tests/vboot_ui_legacy_menu_tests.c +++ /dev/null @@ -1,2349 +0,0 @@ -/* 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_kernel.h" -#include "vboot_struct.h" -#include "vboot_test.h" -#include "vboot_ui_legacy.h" -#include "vboot_ui_legacy_menu_private.h" - -/* Mock 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_fail; -static uint32_t mock_keypress[64]; -static uint32_t mock_keyflags[64]; -static uint32_t mock_keypress_count; -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(&lkp, 0, sizeof(lkp)); - - TEST_SUCC(vb2api_init(workbuf, sizeof(workbuf), &ctx), - "vb2api_init failed"); - vb2_nv_init(ctx); - - sd = vb2_get_sd(ctx); - - /* 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_fail = 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; - - 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; -} - -uint32_t vb2ex_get_locale_count(void) { - return 1; -} - -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; -} - -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; - VB2_DEBUG("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 - VB2_DEBUG("Ran out of screens_displayed entries!\n"); - VB2_DEBUG("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; -} - -void vb2ex_beep(uint32_t msec, uint32_t frequency) -{ - if (beeps_count < ARRAY_SIZE(beeps_played)) - beeps_played[beeps_count++] = frequency; - VB2_DEBUG("beeps_played[%d] = %dHz for %dms\n", - beeps_count - 1, frequency, msec); -} - -void vb2_enable_developer_mode(struct vb2_context *c) -{ - VB2_ASSERT(!virtdev_fail); - virtdev_set = 1; -} - -/* Tests */ - -static void VbBootDevTest(void) -{ - int i; - - VB2_DEBUG("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_TARGET_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_TARGET_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_TARGET_EXTERNAL); - vb2_nv_set(ctx, VB2_NV_DEV_BOOT_EXTERNAL, 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_TARGET_EXTERNAL); - 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_EXTERNAL, 1); - vb2_nv_set(ctx, VB2_NV_DEV_DEFAULT_BOOT, - VB2_DEV_DEFAULT_BOOT_TARGET_EXTERNAL); - 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), VB2_REQUEST_SHUTDOWN, - "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), VB2_REQUEST_SHUTDOWN, - "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), VB2_REQUEST_SHUTDOWN, - "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), VB2_REQUEST_SHUTDOWN, - "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), VB2_REQUEST_SHUTDOWN, - "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), VB2_REQUEST_SHUTDOWN, - "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), VB2_REQUEST_REBOOT, - "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), VB2_REQUEST_SHUTDOWN, - "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_EXTERNAL, 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_EXTERNAL, 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_EXTERNAL; - 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_EXTERNAL, 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_EXTERNAL, 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_TARGET_EXTERNAL); - 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_EXTERNAL, 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_EXTERNAL, 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_TARGET_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_EXTERNAL, 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_EXTERNAL, 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_EXTERNAL, 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_EXTERNAL, 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), VB2_REQUEST_REBOOT, - "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), VB2_REQUEST_REBOOT, - "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), VB2_REQUEST_SHUTDOWN, - "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), VB2_REQUEST_SHUTDOWN, - "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"); - - - VB2_DEBUG("...done.\n"); -} - -static void VbBootRecTest(void) -{ - int i; - - VB2_DEBUG("Testing VbBootRecoveryLegacyMenu()...\n"); - - /* Shutdown requested in BROKEN */ - ResetMocks(); - TEST_EQ(VbBootRecoveryLegacyMenu(ctx), VB2_REQUEST_SHUTDOWN, - "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), VB2_REQUEST_SHUTDOWN, - "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), VB2_REQUEST_SHUTDOWN, - "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), VB2_REQUEST_SHUTDOWN, - "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), VB2_REQUEST_SHUTDOWN, - "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), VB2_REQUEST_SHUTDOWN, - "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), VB2_REQUEST_SHUTDOWN, - "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), VB2_REQUEST_SHUTDOWN, - "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), VB2_REQUEST_SHUTDOWN, - "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), VB2_REQUEST_SHUTDOWN, - "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), VB2_REQUEST_SHUTDOWN, - "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), VB2_REQUEST_SHUTDOWN, - "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), VB2_REQUEST_SHUTDOWN, - "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), VB2_REQUEST_SHUTDOWN, - "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), VB2_REQUEST_SHUTDOWN, - "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), VB2_REQUEST_SHUTDOWN, - "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), VB2_REQUEST_REBOOT_EC_TO_RO, - "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), VB2_REQUEST_SHUTDOWN, - "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), VB2_REQUEST_SHUTDOWN, - "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"); - - /* Don't 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_fail = 1; - TEST_ABORT(VbBootRecoveryLegacyMenu(ctx), "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, 0, " virtual dev mode still off"); - 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_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), VB2_REQUEST_SHUTDOWN, - "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), VB2_REQUEST_SHUTDOWN, - "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), VB2_REQUEST_SHUTDOWN, - "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"); - - VB2_DEBUG("...done.\n"); -} - -static void VbTestLanguageMenu(void) -{ - int i; - - VB2_DEBUG("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), VB2_REQUEST_SHUTDOWN, - "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), VB2_REQUEST_SHUTDOWN, - "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), VB2_REQUEST_SHUTDOWN, - " 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"); - - VB2_DEBUG("...done.\n"); -} - -static void VbNavigationTest(void) -{ - int i; - - VB2_DEBUG("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), VB2_REQUEST_SHUTDOWN, - "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; -} |