1// SPDX-License-Identifier: GPL-2.0+ 2/* 3 * Test for bootdev functions. All start with 'bootdev' 4 * 5 * Copyright 2021 Google LLC 6 * Written by Simon Glass <sjg@chromium.org> 7 */ 8 9#include <common.h> 10#include <bootdev.h> 11#include <bootstd.h> 12#include <dm.h> 13#include <memalign.h> 14#include <mmc.h> 15#include <linux/log2.h> 16#include <test/suites.h> 17#include <test/ut.h> 18#include <u-boot/crc.h> 19#include "bootstd_common.h" 20 21/* tracks whether bootstd_setup_for_tests() has been run yet */ 22bool vbe_setup_done; 23 24/* set up MMC for VBE tests */ 25int bootstd_setup_for_tests(void) 26{ 27 ALLOC_CACHE_ALIGN_BUFFER(u8, buf, MMC_MAX_BLOCK_LEN); 28 struct udevice *mmc; 29 struct blk_desc *desc; 30 int ret; 31 32 if (vbe_setup_done) 33 return 0; 34 35 /* Set up the version string */ 36 ret = uclass_get_device(UCLASS_MMC, 1, &mmc); 37 if (ret) 38 return log_msg_ret("mmc", -EIO); 39 desc = blk_get_by_device(mmc); 40 41 memset(buf, '\0', MMC_MAX_BLOCK_LEN); 42 strcpy(buf, TEST_VERSION); 43 if (blk_dwrite(desc, VERSION_START_BLK, 1, buf) != 1) 44 return log_msg_ret("wr1", -EIO); 45 46 /* Set up the nvdata */ 47 memset(buf, '\0', MMC_MAX_BLOCK_LEN); 48 buf[1] = ilog2(0x40) << 4 | 1; 49 *(u32 *)(buf + 4) = TEST_VERNUM; 50 buf[0] = crc8(0, buf + 1, 0x3f); 51 if (blk_dwrite(desc, NVDATA_START_BLK, 1, buf) != 1) 52 return log_msg_ret("wr2", -EIO); 53 54 vbe_setup_done = true; 55 56 return 0; 57} 58 59int bootstd_test_drop_bootdev_order(struct unit_test_state *uts) 60{ 61 struct bootstd_priv *priv; 62 struct udevice *bootstd; 63 64 ut_assertok(uclass_first_device_err(UCLASS_BOOTSTD, &bootstd)); 65 priv = dev_get_priv(bootstd); 66 priv->bootdev_order = NULL; 67 68 return 0; 69} 70 71int bootstd_test_check_mmc_hunter(struct unit_test_state *uts) 72{ 73 struct bootdev_hunter *start, *mmc; 74 struct bootstd_priv *std; 75 uint seq; 76 77 if (!IS_ENABLED(CONFIG_MMC)) 78 return 0; 79 80 /* get access to the used hunters */ 81 ut_assertok(bootstd_get_priv(&std)); 82 83 /* check that the hunter was used */ 84 start = ll_entry_start(struct bootdev_hunter, bootdev_hunter); 85 mmc = BOOTDEV_HUNTER_GET(mmc_bootdev_hunter); 86 seq = mmc - start; 87 ut_asserteq(BIT(seq), std->hunters_used); 88 89 return 0; 90} 91 92int do_ut_bootstd(struct cmd_tbl *cmdtp, int flag, int argc, char *const argv[]) 93{ 94 struct unit_test *tests = UNIT_TEST_SUITE_START(bootstd_test); 95 const int n_ents = UNIT_TEST_SUITE_COUNT(bootstd_test); 96 int ret; 97 98 ret = bootstd_setup_for_tests(); 99 if (ret) { 100 printf("Failed to set up for bootstd tests (err=%d)\n", ret); 101 return CMD_RET_FAILURE; 102 } 103 104 return cmd_ut_category("bootstd", "bootstd_test_", 105 tests, n_ents, argc, argv); 106} 107