1// SPDX-License-Identifier: GPL-2.0+ 2/* 3 * Copyright (c) 2021 Steffen Jaeckel 4 * 5 * Unit tests for autoboot functionality 6 */ 7 8#include <autoboot.h> 9#include <common.h> 10#include <test/common.h> 11#include <test/test.h> 12#include <test/ut.h> 13 14#include <crypt.h> 15 16static int check_for_input(struct unit_test_state *uts, const char *in, 17 bool correct) 18{ 19 bool old_val; 20 /* The bootdelay is set to 1 second in test_autoboot() */ 21 const char *autoboot_prompt = 22 "Enter password \"a\" in 1 seconds to stop autoboot"; 23 24 console_record_reset_enable(); 25 console_in_puts(in); 26 27 /* turn on keyed autoboot for the test, if possible */ 28 old_val = autoboot_set_keyed(true); 29 autoboot_command("echo Autoboot password unlock not successful"); 30 old_val = autoboot_set_keyed(old_val); 31 32 ut_assert_nextline(autoboot_prompt); 33 if (!correct) 34 ut_assert_nextline("Autoboot password unlock not successful"); 35 ut_assert_console_end(); 36 return 0; 37} 38 39/** 40 * test_autoboot() - unit test for autoboot 41 * 42 * @uts: unit test state 43 * Return: 0 = success, 1 = failure 44 */ 45static int test_autoboot(struct unit_test_state *uts) 46{ 47 /* make sure that the bootdelay is set to something, 48 * otherwise the called functions will time out 49 */ 50 ut_assertok(env_set("bootdelay", "1")); 51 bootdelay_process(); 52 53 /* unset all relevant environment variables */ 54 env_set("bootstopusesha256", NULL); 55 env_set("bootstopkeycrypt", NULL); 56 env_set("bootstopkeysha256", NULL); 57 58 if (IS_ENABLED(CONFIG_CRYPT_PW_SHA256)) { 59 /* test the default password from CONFIG_AUTOBOOT_STOP_STR_CRYPT */ 60 ut_assertok(check_for_input(uts, "a\n", true)); 61 /* test a password from the `bootstopkeycrypt` environment variable */ 62 ut_assertok(env_set( 63 "bootstopkeycrypt", 64 "$5$rounds=640000$ycgRgpnRq4lmu.eb$aZ6YJWdklvyLML13w7mEHMHJnJOux6aptnp6VlsR5a9")); 65 66 ut_assertok(check_for_input(uts, "test\n", true)); 67 68 /* verify that the `bootstopusesha256` variable is treated correctly */ 69 ut_assertok(env_set("bootstopusesha256", "false")); 70 ut_assertok(check_for_input(uts, "test\n", true)); 71 } 72 73 if (IS_ENABLED(CONFIG_AUTOBOOT_ENCRYPTION)) { 74 /* test the `bootstopusesha256` and `bootstopkeysha256` features */ 75 ut_assertok(env_set("bootstopusesha256", "true")); 76 ut_assertok(env_set( 77 "bootstopkeysha256", 78 "edeaaff3f1774ad2888673770c6d64097e391bc362d7d6fb34982ddf0efd18cb")); 79 80 ut_assertok(check_for_input(uts, "abc\n", true)); 81 82 ut_assertok(env_set( 83 "bootstopkeysha256", 84 "ba7816bf8f01cfea414140de5dae2223b00361a396177a9cb410ff61f20015ad")); 85 86 ut_assertok(check_for_input(uts, "abc", true)); 87 88 ut_assertok(check_for_input(uts, "abc\n", true)); 89 90 ut_assertok(check_for_input(uts, "abd", false)); 91 } 92 93 return CMD_RET_SUCCESS; 94} 95 96COMMON_TEST(test_autoboot, 0); 97