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> 13#include <of_live.h> 14#include <vbe.h> 15#include <test/suites.h> 16#include <test/ut.h> 17#include "bootstd_common.h" 18 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 */ 25static int vbe_simple_test_base(struct unit_test_state *uts) 26{ 27 const char *version, *bl_version; 28 struct event_ft_fixup fixup; 29 struct udevice *dev; 30 struct device_node *np; 31 char fdt_buf[0x400]; 32 char info[100]; 33 int node_ofs; 34 ofnode node; 35 u32 vernum; 36 37 /* Set up the VBE info */ 38 ut_assertok(bootstd_setup_for_tests()); 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 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 } 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 65 * when it comes to send the event, it creates an ofnode that uses the 66 * control FDT, since it has no way of accessing the live tree created 67 * here. 68 * 69 * Two fix this we need image_setup_libfdt() is updated to use ofnode 70 */ 71 fixup.images = NULL; 72 ut_assertok(event_notify(EVT_FT_FIXUP, &fixup, sizeof(fixup))); 73 74 node = oftree_path(fixup.tree, "/chosen/fwupd/firmware0"); 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); 85 ut_asserteq_str(version_string + 7, bl_version); 86 87 return 0; 88} 89BOOTSTD_TEST(vbe_simple_test_base, UT_TESTF_DM | UT_TESTF_SCAN_FDT); 90