blob: cc97e255e5cb08052af42564a1790d63ec96fbef [file] [log] [blame]
// SPDX-License-Identifier: GPL-2.0+
/*
* Test for bootdev functions. All start with 'bootdev'
*
* Copyright 2021 Google LLC
* Written by Simon Glass <sjg@chromium.org>
*/
#include <common.h>
#include <bootdev.h>
#include <bootstd.h>
#include <dm.h>
#include <memalign.h>
#include <mmc.h>
#include <linux/log2.h>
#include <test/suites.h>
#include <test/ut.h>
#include <u-boot/crc.h>
#include "bootstd_common.h"
/* tracks whether bootstd_setup_for_tests() has been run yet */
bool vbe_setup_done;
/* set up MMC for VBE tests */
int bootstd_setup_for_tests(void)
{
ALLOC_CACHE_ALIGN_BUFFER(u8, buf, MMC_MAX_BLOCK_LEN);
struct udevice *mmc;
struct blk_desc *desc;
int ret;
if (vbe_setup_done)
return 0;
/* Set up the version string */
ret = uclass_get_device(UCLASS_MMC, 1, &mmc);
if (ret)
return log_msg_ret("mmc", -EIO);
desc = blk_get_by_device(mmc);
memset(buf, '\0', MMC_MAX_BLOCK_LEN);
strcpy(buf, TEST_VERSION);
if (blk_dwrite(desc, VERSION_START_BLK, 1, buf) != 1)
return log_msg_ret("wr1", -EIO);
/* Set up the nvdata */
memset(buf, '\0', MMC_MAX_BLOCK_LEN);
buf[1] = ilog2(0x40) << 4 | 1;
*(u32 *)(buf + 4) = TEST_VERNUM;
buf[0] = crc8(0, buf + 1, 0x3f);
if (blk_dwrite(desc, NVDATA_START_BLK, 1, buf) != 1)
return log_msg_ret("wr2", -EIO);
vbe_setup_done = true;
return 0;
}
int bootstd_test_drop_bootdev_order(struct unit_test_state *uts)
{
struct bootstd_priv *priv;
struct udevice *bootstd;
ut_assertok(uclass_first_device_err(UCLASS_BOOTSTD, &bootstd));
priv = dev_get_priv(bootstd);
priv->bootdev_order = NULL;
return 0;
}
int bootstd_test_check_mmc_hunter(struct unit_test_state *uts)
{
struct bootdev_hunter *start, *mmc;
struct bootstd_priv *std;
uint seq;
if (!IS_ENABLED(CONFIG_MMC))
return 0;
/* get access to the used hunters */
ut_assertok(bootstd_get_priv(&std));
/* check that the hunter was used */
start = ll_entry_start(struct bootdev_hunter, bootdev_hunter);
mmc = BOOTDEV_HUNTER_GET(mmc_bootdev_hunter);
seq = mmc - start;
ut_asserteq(BIT(seq), std->hunters_used);
return 0;
}
int do_ut_bootstd(struct cmd_tbl *cmdtp, int flag, int argc, char *const argv[])
{
struct unit_test *tests = UNIT_TEST_SUITE_START(bootstd_test);
const int n_ents = UNIT_TEST_SUITE_COUNT(bootstd_test);
int ret;
ret = bootstd_setup_for_tests();
if (ret) {
printf("Failed to set up for bootstd tests (err=%d)\n", ret);
return CMD_RET_FAILURE;
}
return cmd_ut_category("bootstd", "bootstd_test_",
tests, n_ents, argc, argv);
}