Simon Glass | 5fe76d4 | 2022-07-30 15:52:37 -0600 | [diff] [blame] | 1 | // 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 | |
| 9 | #include <common.h> |
| 10 | #include <bootmeth.h> |
| 11 | #include <dm.h> |
| 12 | #include <image.h> |
Simon Glass | 5fe76d4 | 2022-07-30 15:52:37 -0600 | [diff] [blame] | 13 | #include <of_live.h> |
| 14 | #include <vbe.h> |
Simon Glass | 5fe76d4 | 2022-07-30 15:52:37 -0600 | [diff] [blame] | 15 | #include <test/suites.h> |
| 16 | #include <test/ut.h> |
Simon Glass | 5fe76d4 | 2022-07-30 15:52:37 -0600 | [diff] [blame] | 17 | #include "bootstd_common.h" |
| 18 | |
Simon Glass | a56f663 | 2022-10-20 18:23:14 -0600 | [diff] [blame] | 19 | /* |
| 20 | * Basic test of reading nvdata and updating a fwupd node in the device tree |
| 21 | * |
| 22 | * This sets up its own VBE info in the device, using bootstd_setup_for_tests() |
| 23 | * then does a VBE fixup and checks that everything is present. |
| 24 | */ |
Simon Glass | 5fe76d4 | 2022-07-30 15:52:37 -0600 | [diff] [blame] | 25 | static int vbe_simple_test_base(struct unit_test_state *uts) |
| 26 | { |
Simon Glass | 5fe76d4 | 2022-07-30 15:52:37 -0600 | [diff] [blame] | 27 | const char *version, *bl_version; |
| 28 | struct event_ft_fixup fixup; |
Simon Glass | bfdfc5d | 2022-10-11 09:47:13 -0600 | [diff] [blame] | 29 | struct udevice *dev; |
Simon Glass | 5fe76d4 | 2022-07-30 15:52:37 -0600 | [diff] [blame] | 30 | struct device_node *np; |
Simon Glass | 5fe76d4 | 2022-07-30 15:52:37 -0600 | [diff] [blame] | 31 | char fdt_buf[0x400]; |
| 32 | char info[100]; |
| 33 | int node_ofs; |
| 34 | ofnode node; |
| 35 | u32 vernum; |
| 36 | |
Simon Glass | bfdfc5d | 2022-10-11 09:47:13 -0600 | [diff] [blame] | 37 | /* Set up the VBE info */ |
| 38 | ut_assertok(bootstd_setup_for_tests()); |
Simon Glass | 5fe76d4 | 2022-07-30 15:52:37 -0600 | [diff] [blame] | 39 | |
| 40 | /* Read the version back */ |
| 41 | ut_assertok(vbe_find_by_any("firmware0", &dev)); |
| 42 | ut_assertok(bootmeth_get_state_desc(dev, info, sizeof(info))); |
| 43 | ut_asserteq_str("Version: " TEST_VERSION "\nVernum: 1/2", info); |
| 44 | |
| 45 | ut_assertok(fdt_create_empty_tree(fdt_buf, sizeof(fdt_buf))); |
| 46 | node_ofs = fdt_add_subnode(fdt_buf, 0, "chosen"); |
| 47 | ut_assert(node_ofs > 0); |
| 48 | |
| 49 | node_ofs = fdt_add_subnode(fdt_buf, node_ofs, "fwupd"); |
| 50 | ut_assert(node_ofs > 0); |
| 51 | |
| 52 | node_ofs = fdt_add_subnode(fdt_buf, node_ofs, "firmware0"); |
| 53 | ut_assert(node_ofs > 0); |
| 54 | |
Simon Glass | 0d63213 | 2022-09-06 20:27:31 -0600 | [diff] [blame] | 55 | if (of_live_active()) { |
| 56 | ut_assertok(unflatten_device_tree(fdt_buf, &np)); |
| 57 | fixup.tree = oftree_from_np(np); |
| 58 | } else { |
| 59 | fixup.tree = oftree_from_fdt(fdt_buf); |
| 60 | } |
Simon Glass | 5fe76d4 | 2022-07-30 15:52:37 -0600 | [diff] [blame] | 61 | |
| 62 | /* |
| 63 | * It would be better to call image_setup_libfdt() here, but that |
| 64 | * function does not allow passing an ofnode. We can pass fdt_buf but |
Simon Glass | 0d63213 | 2022-09-06 20:27:31 -0600 | [diff] [blame] | 65 | * when it comes to send the event, it creates an ofnode that uses the |
Simon Glass | 5fe76d4 | 2022-07-30 15:52:37 -0600 | [diff] [blame] | 66 | * control FDT, since it has no way of accessing the live tree created |
| 67 | * here. |
| 68 | * |
Simon Glass | 0d63213 | 2022-09-06 20:27:31 -0600 | [diff] [blame] | 69 | * Two fix this we need image_setup_libfdt() is updated to use ofnode |
Simon Glass | 5fe76d4 | 2022-07-30 15:52:37 -0600 | [diff] [blame] | 70 | */ |
Simon Glass | bfdfc5d | 2022-10-11 09:47:13 -0600 | [diff] [blame] | 71 | fixup.images = NULL; |
Simon Glass | 5fe76d4 | 2022-07-30 15:52:37 -0600 | [diff] [blame] | 72 | ut_assertok(event_notify(EVT_FT_FIXUP, &fixup, sizeof(fixup))); |
| 73 | |
Simon Glass | b7bd94f | 2022-09-06 20:27:24 -0600 | [diff] [blame] | 74 | node = oftree_path(fixup.tree, "/chosen/fwupd/firmware0"); |
Simon Glass | 5fe76d4 | 2022-07-30 15:52:37 -0600 | [diff] [blame] | 75 | |
| 76 | version = ofnode_read_string(node, "cur-version"); |
| 77 | ut_assertnonnull(version); |
| 78 | ut_asserteq_str(TEST_VERSION, version); |
| 79 | |
| 80 | ut_assertok(ofnode_read_u32(node, "cur-vernum", &vernum)); |
| 81 | ut_asserteq(TEST_VERNUM, vernum); |
| 82 | |
| 83 | bl_version = ofnode_read_string(node, "bootloader-version"); |
| 84 | ut_assertnonnull(bl_version); |
Simon Glass | e45d226 | 2022-10-20 18:23:12 -0600 | [diff] [blame] | 85 | ut_asserteq_str(version_string + 7, bl_version); |
Simon Glass | 5fe76d4 | 2022-07-30 15:52:37 -0600 | [diff] [blame] | 86 | |
| 87 | return 0; |
| 88 | } |
Simon Glass | 0d63213 | 2022-09-06 20:27:31 -0600 | [diff] [blame] | 89 | BOOTSTD_TEST(vbe_simple_test_base, UT_TESTF_DM | UT_TESTF_SCAN_FDT); |