Gitiles
Code Review Sign In
git01.mediatek.com / filogic / uboot / 18357b321931dbfb11ab52d98aa6fc34c859e336 / . / test / boot / bootmeth.c
blob: e498eee036eb993c6bea5601f8daf9ff207c5943 [file] [log] [blame]
Simon Glassb255efc2022-04-24 23:31:24 -0600[diff] [blame]1// SPDX-License-Identifier: GPL-2.0+
2/*
Simon Glassda192022022-07-30 15:52:17 -0600[diff] [blame]3 * Test for bootdev functions. All start with 'bootmeth'
Simon Glassb255efc2022-04-24 23:31:24 -0600[diff] [blame]4 *
5 * Copyright 2021 Google LLC
6 * Written by Simon Glass <sjg@chromium.org>
7 */
8
9#include <common.h>
Simon Glassf6d71a82022-07-30 15:52:19 -0600[diff] [blame]10#include <bootmeth.h>
Simon Glassb255efc2022-04-24 23:31:24 -0600[diff] [blame]11#include <bootstd.h>
Simon Glassf6d71a82022-07-30 15:52:19 -0600[diff] [blame]12#include <dm.h>
Simon Glassb255efc2022-04-24 23:31:24 -0600[diff] [blame]13#include <test/suites.h>
14#include <test/ut.h>
15#include "bootstd_common.h"
16
17/* Check 'bootmeth list' command */
18static int bootmeth_cmd_list(struct unit_test_state *uts)
19{
20 console_record_reset_enable();
21 ut_assertok(run_command("bootmeth list", 0));
22 ut_assert_nextline("Order Seq Name Description");
23 ut_assert_nextlinen("---");
Simon Glassb71d7f72023-05-10 16:34:46 -0600[diff] [blame]24 ut_assert_nextline(" 0 0 extlinux Extlinux boot from a block device");
Simon Glassb255efc2022-04-24 23:31:24 -0600[diff] [blame]25 ut_assert_nextline(" 1 1 efi EFI boot from an .efi file");
Simon Glass90b643d2022-07-30 15:52:36 -0600[diff] [blame]26 if (IS_ENABLED(CONFIG_BOOTMETH_GLOBAL))
27 ut_assert_nextline(" glob 2 firmware0 VBE simple");
Simon Glassb255efc2022-04-24 23:31:24 -0600[diff] [blame]28 ut_assert_nextlinen("---");
Simon Glass90b643d2022-07-30 15:52:36 -0600[diff] [blame]29 ut_assert_nextline(IS_ENABLED(CONFIG_BOOTMETH_GLOBAL) ?
30 "(3 bootmeths)" : "(2 bootmeths)");
Simon Glassb255efc2022-04-24 23:31:24 -0600[diff] [blame]31 ut_assert_console_end();
32
33 return 0;
34}
35BOOTSTD_TEST(bootmeth_cmd_list, UT_TESTF_DM | UT_TESTF_SCAN_FDT);
36
37/* Check 'bootmeth order' command */
38static int bootmeth_cmd_order(struct unit_test_state *uts)
39{
40 /* Select just one bootmethod */
41 console_record_reset_enable();
Simon Glassb71d7f72023-05-10 16:34:46 -0600[diff] [blame]42 ut_assertok(run_command("bootmeth order extlinux", 0));
Simon Glassb255efc2022-04-24 23:31:24 -0600[diff] [blame]43 ut_assert_console_end();
44 ut_assertnonnull(env_get("bootmeths"));
Simon Glassb71d7f72023-05-10 16:34:46 -0600[diff] [blame]45 ut_asserteq_str("extlinux", env_get("bootmeths"));
Simon Glassb255efc2022-04-24 23:31:24 -0600[diff] [blame]46
47 /* Only that one should be listed */
48 ut_assertok(run_command("bootmeth list", 0));
49 ut_assert_nextline("Order Seq Name Description");
50 ut_assert_nextlinen("---");
Simon Glassb71d7f72023-05-10 16:34:46 -0600[diff] [blame]51 ut_assert_nextline(" 0 0 extlinux Extlinux boot from a block device");
Simon Glassb255efc2022-04-24 23:31:24 -0600[diff] [blame]52 ut_assert_nextlinen("---");
53 ut_assert_nextline("(1 bootmeth)");
54 ut_assert_console_end();
55
56 /* Check the -a flag, efi should show as not in the order ("-") */
57 ut_assertok(run_command("bootmeth list -a", 0));
58 ut_assert_nextline("Order Seq Name Description");
59 ut_assert_nextlinen("---");
Simon Glassb71d7f72023-05-10 16:34:46 -0600[diff] [blame]60 ut_assert_nextline(" 0 0 extlinux Extlinux boot from a block device");
Simon Glassb255efc2022-04-24 23:31:24 -0600[diff] [blame]61 ut_assert_nextline(" - 1 efi EFI boot from an .efi file");
Simon Glass90b643d2022-07-30 15:52:36 -0600[diff] [blame]62 if (IS_ENABLED(CONFIG_BOOTMETH_GLOBAL))
63 ut_assert_nextline(" glob 2 firmware0 VBE simple");
Simon Glassb255efc2022-04-24 23:31:24 -0600[diff] [blame]64 ut_assert_nextlinen("---");
Simon Glass90b643d2022-07-30 15:52:36 -0600[diff] [blame]65 ut_assert_nextline(IS_ENABLED(CONFIG_BOOTMETH_GLOBAL) ?
66 "(3 bootmeths)" : "(2 bootmeths)");
Simon Glassb255efc2022-04-24 23:31:24 -0600[diff] [blame]67 ut_assert_console_end();
68
69 /* Check the -a flag with the reverse order */
Simon Glassb71d7f72023-05-10 16:34:46 -0600[diff] [blame]70 ut_assertok(run_command("bootmeth order \"efi extlinux\"", 0));
Simon Glassb255efc2022-04-24 23:31:24 -0600[diff] [blame]71 ut_assert_console_end();
72 ut_assertok(run_command("bootmeth list -a", 0));
73 ut_assert_nextline("Order Seq Name Description");
74 ut_assert_nextlinen("---");
Simon Glassb71d7f72023-05-10 16:34:46 -0600[diff] [blame]75 ut_assert_nextline(" 1 0 extlinux Extlinux boot from a block device");
Simon Glassb255efc2022-04-24 23:31:24 -0600[diff] [blame]76 ut_assert_nextline(" 0 1 efi EFI boot from an .efi file");
Simon Glass90b643d2022-07-30 15:52:36 -0600[diff] [blame]77 if (IS_ENABLED(CONFIG_BOOTMETH_GLOBAL))
78 ut_assert_nextline(" glob 2 firmware0 VBE simple");
Simon Glassb255efc2022-04-24 23:31:24 -0600[diff] [blame]79 ut_assert_nextlinen("---");
Simon Glass90b643d2022-07-30 15:52:36 -0600[diff] [blame]80 ut_assert_nextline(IS_ENABLED(CONFIG_BOOTMETH_GLOBAL) ?
81 "(3 bootmeths)" : "(2 bootmeths)");
Simon Glassb255efc2022-04-24 23:31:24 -0600[diff] [blame]82 ut_assert_console_end();
83
84 /* Now reset the order to empty, which should show all of them again */
85 ut_assertok(run_command("bootmeth order", 0));
86 ut_assert_console_end();
87 ut_assertnull(env_get("bootmeths"));
88 ut_assertok(run_command("bootmeth list", 0));
Simon Glass90b643d2022-07-30 15:52:36 -0600[diff] [blame]89 ut_assert_skip_to_line(IS_ENABLED(CONFIG_BOOTMETH_GLOBAL) ?
90 "(3 bootmeths)" : "(2 bootmeths)");
Simon Glassb255efc2022-04-24 23:31:24 -0600[diff] [blame]91
92 /* Try reverse order */
Simon Glassb71d7f72023-05-10 16:34:46 -0600[diff] [blame]93 ut_assertok(run_command("bootmeth order \"efi extlinux\"", 0));
Simon Glassb255efc2022-04-24 23:31:24 -0600[diff] [blame]94 ut_assert_console_end();
95 ut_assertok(run_command("bootmeth list", 0));
96 ut_assert_nextline("Order Seq Name Description");
97 ut_assert_nextlinen("---");
98 ut_assert_nextline(" 0 1 efi EFI boot from an .efi file");
Simon Glassb71d7f72023-05-10 16:34:46 -0600[diff] [blame]99 ut_assert_nextline(" 1 0 extlinux Extlinux boot from a block device");
Simon Glassb255efc2022-04-24 23:31:24 -0600[diff] [blame]100 ut_assert_nextlinen("---");
101 ut_assert_nextline("(2 bootmeths)");
102 ut_assertnonnull(env_get("bootmeths"));
Simon Glassb71d7f72023-05-10 16:34:46 -0600[diff] [blame]103 ut_asserteq_str("efi extlinux", env_get("bootmeths"));
Simon Glassb255efc2022-04-24 23:31:24 -0600[diff] [blame]104 ut_assert_console_end();
105
Simon Glassc7599442022-10-20 18:22:49 -0600[diff] [blame]106 return 0;
107}
108BOOTSTD_TEST(bootmeth_cmd_order, UT_TESTF_DM | UT_TESTF_SCAN_FDT);
109
110/* Check 'bootmeth order' command with global bootmeths */
111static int bootmeth_cmd_order_glob(struct unit_test_state *uts)
112{
Simon Glass90b643d2022-07-30 15:52:36 -0600[diff] [blame]113 if (!IS_ENABLED(CONFIG_BOOTMETH_GLOBAL))
Simon Glassc7599442022-10-20 18:22:49 -0600[diff] [blame]114 return -EAGAIN;
Simon Glass90b643d2022-07-30 15:52:36 -0600[diff] [blame]115
Simon Glassc7599442022-10-20 18:22:49 -0600[diff] [blame]116 console_record_reset_enable();
Simon Glassc8d37212022-07-30 15:52:34 -0600[diff] [blame]117 ut_assertok(run_command("bootmeth order \"efi firmware0\"", 0));
118 ut_assert_console_end();
119 ut_assertok(run_command("bootmeth list", 0));
120 ut_assert_nextline("Order Seq Name Description");
121 ut_assert_nextlinen("---");
122 ut_assert_nextline(" 0 1 efi EFI boot from an .efi file");
123 ut_assert_nextline(" glob 2 firmware0 VBE simple");
124 ut_assert_nextlinen("---");
125 ut_assert_nextline("(2 bootmeths)");
126 ut_assertnonnull(env_get("bootmeths"));
127 ut_asserteq_str("efi firmware0", env_get("bootmeths"));
128 ut_assert_console_end();
129
Simon Glassb255efc2022-04-24 23:31:24 -0600[diff] [blame]130 return 0;
131}
Simon Glassc7599442022-10-20 18:22:49 -0600[diff] [blame]132BOOTSTD_TEST(bootmeth_cmd_order_glob, UT_TESTF_DM | UT_TESTF_SCAN_FDT);
Simon Glassb255efc2022-04-24 23:31:24 -0600[diff] [blame]133
134/* Check 'bootmeths' env var */
135static int bootmeth_env(struct unit_test_state *uts)
136{
137 struct bootstd_priv *std;
138
139 ut_assertok(bootstd_get_priv(&std));
140
141 /* Select just one bootmethod */
142 console_record_reset_enable();
Simon Glassb71d7f72023-05-10 16:34:46 -0600[diff] [blame]143 ut_assertok(env_set("bootmeths", "extlinux"));
Simon Glassb255efc2022-04-24 23:31:24 -0600[diff] [blame]144 ut_asserteq(1, std->bootmeth_count);
145
146 /* Select an invalid bootmethod */
147 ut_asserteq(1, run_command("setenv bootmeths fred", 0));
148 ut_assert_nextline("Unknown bootmeth 'fred'");
149 ut_assert_nextlinen("## Error inserting");
150 ut_assert_console_end();
151
Simon Glassb71d7f72023-05-10 16:34:46 -0600[diff] [blame]152 ut_assertok(env_set("bootmeths", "efi extlinux"));
Simon Glassb255efc2022-04-24 23:31:24 -0600[diff] [blame]153 ut_asserteq(2, std->bootmeth_count);
154 ut_assert_console_end();
155
156 return 0;
157}
158BOOTSTD_TEST(bootmeth_env, UT_TESTF_DM | UT_TESTF_SCAN_FDT);
Simon Glassf6d71a82022-07-30 15:52:19 -0600[diff] [blame]159
160/* Check the get_state_desc() method */
161static int bootmeth_state(struct unit_test_state *uts)
162{
163 struct udevice *dev;
164 char buf[50];
165
Michal Suchanekac12a2f2022-10-12 21:57:59 +0200[diff] [blame]166 ut_assertok(uclass_first_device_err(UCLASS_BOOTMETH, &dev));
Simon Glassf6d71a82022-07-30 15:52:19 -0600[diff] [blame]167 ut_assertnonnull(dev);
168
169 ut_assertok(bootmeth_get_state_desc(dev, buf, sizeof(buf)));
170 ut_asserteq_str("OK", buf);
171
172 return 0;
173}
174BOOTSTD_TEST(bootmeth_state, UT_TESTF_DM | UT_TESTF_SCAN_FDT);