a753190a0c
To avoid duplicating code, create a new fit_util module which provides various utility functions for FIT. Move this code out from the existing test_fit.py and refactor it with addition parameters. Fix up pylint warnings in the conversion. This involves no functional change. Signed-off-by: Simon Glass <sjg@chromium.org>
85 lines
2.0 KiB
C
85 lines
2.0 KiB
C
// 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 <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 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);
|
|
}
|