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
check_for_input(struct unit_test_state * uts,const char * in,bool correct)16 static 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 */
test_autoboot(struct unit_test_state * uts)45 static 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
96 COMMON_TEST(test_autoboot, 0);
97