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