blob: 3672b744e5fb729e56a0a276a70ca4d4531dbd02 [file] [log] [blame]
Simon Glasse5a7b772022-07-30 15:52:37 -06001// SPDX-License-Identifier: GPL-2.0+
2/*
3 * Test for vbe-simple bootmeth. All start with 'vbe_simple'
4 *
5 * Copyright 2023 Google LLC
6 * Written by Simon Glass <sjg@chromium.org>
7 */
8
Simon Glasse5a7b772022-07-30 15:52:37 -06009#include <bootmeth.h>
10#include <dm.h>
11#include <image.h>
Simon Glasse5a7b772022-07-30 15:52:37 -060012#include <of_live.h>
13#include <vbe.h>
Simon Glasse5a7b772022-07-30 15:52:37 -060014#include <test/suites.h>
15#include <test/ut.h>
Simon Glasse5a7b772022-07-30 15:52:37 -060016#include "bootstd_common.h"
17
Simon Glassa9289612022-10-20 18:23:14 -060018/*
19 * Basic test of reading nvdata and updating a fwupd node in the device tree
20 *
21 * This sets up its own VBE info in the device, using bootstd_setup_for_tests()
22 * then does a VBE fixup and checks that everything is present.
23 */
Simon Glasse5a7b772022-07-30 15:52:37 -060024static int vbe_simple_test_base(struct unit_test_state *uts)
25{
Simon Glasse5a7b772022-07-30 15:52:37 -060026 const char *version, *bl_version;
27 struct event_ft_fixup fixup;
Simon Glass80253462022-10-11 09:47:13 -060028 struct udevice *dev;
Simon Glasse5a7b772022-07-30 15:52:37 -060029 struct device_node *np;
Simon Glasse5a7b772022-07-30 15:52:37 -060030 char fdt_buf[0x400];
31 char info[100];
32 int node_ofs;
33 ofnode node;
34 u32 vernum;
35
Simon Glass80253462022-10-11 09:47:13 -060036 /* Set up the VBE info */
37 ut_assertok(bootstd_setup_for_tests());
Simon Glasse5a7b772022-07-30 15:52:37 -060038
39 /* Read the version back */
40 ut_assertok(vbe_find_by_any("firmware0", &dev));
41 ut_assertok(bootmeth_get_state_desc(dev, info, sizeof(info)));
42 ut_asserteq_str("Version: " TEST_VERSION "\nVernum: 1/2", info);
43
44 ut_assertok(fdt_create_empty_tree(fdt_buf, sizeof(fdt_buf)));
45 node_ofs = fdt_add_subnode(fdt_buf, 0, "chosen");
46 ut_assert(node_ofs > 0);
47
48 node_ofs = fdt_add_subnode(fdt_buf, node_ofs, "fwupd");
49 ut_assert(node_ofs > 0);
50
51 node_ofs = fdt_add_subnode(fdt_buf, node_ofs, "firmware0");
52 ut_assert(node_ofs > 0);
53
Simon Glasseda440b2022-09-06 20:27:31 -060054 if (of_live_active()) {
55 ut_assertok(unflatten_device_tree(fdt_buf, &np));
56 fixup.tree = oftree_from_np(np);
57 } else {
58 fixup.tree = oftree_from_fdt(fdt_buf);
59 }
Simon Glasse5a7b772022-07-30 15:52:37 -060060
61 /*
62 * It would be better to call image_setup_libfdt() here, but that
63 * function does not allow passing an ofnode. We can pass fdt_buf but
Simon Glasseda440b2022-09-06 20:27:31 -060064 * when it comes to send the event, it creates an ofnode that uses the
Simon Glasse5a7b772022-07-30 15:52:37 -060065 * control FDT, since it has no way of accessing the live tree created
66 * here.
67 *
Simon Glasseda440b2022-09-06 20:27:31 -060068 * Two fix this we need image_setup_libfdt() is updated to use ofnode
Simon Glasse5a7b772022-07-30 15:52:37 -060069 */
Simon Glass80253462022-10-11 09:47:13 -060070 fixup.images = NULL;
Simon Glasse5a7b772022-07-30 15:52:37 -060071 ut_assertok(event_notify(EVT_FT_FIXUP, &fixup, sizeof(fixup)));
72
Simon Glass45ae59d2022-09-06 20:27:24 -060073 node = oftree_path(fixup.tree, "/chosen/fwupd/firmware0");
Simon Glasse5a7b772022-07-30 15:52:37 -060074
75 version = ofnode_read_string(node, "cur-version");
76 ut_assertnonnull(version);
77 ut_asserteq_str(TEST_VERSION, version);
78
79 ut_assertok(ofnode_read_u32(node, "cur-vernum", &vernum));
80 ut_asserteq(TEST_VERNUM, vernum);
81
82 bl_version = ofnode_read_string(node, "bootloader-version");
83 ut_assertnonnull(bl_version);
Simon Glassa6a06cc2022-10-20 18:23:12 -060084 ut_asserteq_str(version_string + 7, bl_version);
Simon Glasse5a7b772022-07-30 15:52:37 -060085
86 return 0;
87}
Simon Glasseda440b2022-09-06 20:27:31 -060088BOOTSTD_TEST(vbe_simple_test_base, UT_TESTF_DM | UT_TESTF_SCAN_FDT);