summaryrefslogtreecommitdiff
path: root/test
diff options
context:
space:
mode:
Diffstat (limited to 'test')
-rw-r--r--test/cmd/Makefile1
-rw-r--r--test/cmd/part.c227
-rw-r--r--test/dm/fwu_mdata.c48
-rw-r--r--test/dm/part.c53
4 files changed, 329 insertions, 0 deletions
diff --git a/test/cmd/Makefile b/test/cmd/Makefile
index 5f2815b1bb6..8d6932f1176 100644
--- a/test/cmd/Makefile
+++ b/test/cmd/Makefile
@@ -26,6 +26,7 @@ obj-$(CONFIG_CMD_LOADM) += loadm.o
obj-$(CONFIG_CMD_MEMINFO) += meminfo.o
obj-$(CONFIG_CMD_MEMORY) += mem_copy.o
obj-$(CONFIG_CMD_MEM_SEARCH) += mem_search.o
+obj-$(CONFIG_CMD_PART) += part.o
ifdef CONFIG_CMD_PCI
obj-$(CONFIG_CMD_PCI_MPS) += pci_mps.o
endif
diff --git a/test/cmd/part.c b/test/cmd/part.c
new file mode 100644
index 00000000000..e0149011476
--- /dev/null
+++ b/test/cmd/part.c
@@ -0,0 +1,227 @@
+// SPDX-License-Identifier: GPL-2.0+
+/*
+ * Test for part command
+ *
+ * Copyright (C) 2026 Amarula Solutions
+ * Written by Dario Binacchi <[email protected]>
+ */
+
+#include <command.h>
+#include <dm.h>
+#include <env.h>
+#include <part.h>
+#include <vsprintf.h>
+#include <dm/test.h>
+#include <test/cmd.h>
+#include <test/test.h>
+#include <test/ut.h>
+
+static struct disk_partition gpt_parts[] = {
+ {
+ .start = 48,
+ .size = 1,
+ .name = "test1",
+ .uuid = "c5bce7a2-03f0-4d03-9048-01ff23b9d527",
+ },
+ {
+ .start = 49,
+ .size = 2,
+ .name = "test2",
+ .uuid = "9df346e8-2c53-4cd8-b9ac-3af83f9a9b74",
+ },
+};
+
+static char disk_guid[UUID_STR_LEN + 1] =
+ "8d60b397-1bb6-4d33-80ee-b1587d24c2f8";
+
+static int setup_gpt_partitions(struct unit_test_state *uts,
+ unsigned int mmc_dev_num)
+{
+ struct blk_desc *mmc_dev_desc;
+ char dev_str[10];
+ int i, ret;
+
+ if (!CONFIG_IS_ENABLED(MMC))
+ return -EAGAIN;
+
+ snprintf(dev_str, sizeof(dev_str), "%u", mmc_dev_num);
+
+ ret = blk_get_device_by_str("mmc", dev_str, &mmc_dev_desc);
+ if (ret == -ENODEV)
+ return -EAGAIN;
+
+ ut_asserteq(mmc_dev_num, ret);
+
+ if (CONFIG_IS_ENABLED(RANDOM_UUID)) {
+ for (i = 0; i < ARRAY_SIZE(gpt_parts); i++)
+ gen_rand_uuid_str(gpt_parts[i].uuid,
+ UUID_STR_FORMAT_STD);
+
+ gen_rand_uuid_str(disk_guid, UUID_STR_FORMAT_STD);
+ }
+
+ ut_assertok(gpt_restore(mmc_dev_desc, disk_guid, gpt_parts,
+ ARRAY_SIZE(gpt_parts)));
+ return 0;
+}
+
+static int cmd_test_part_number(struct unit_test_state *uts)
+{
+ unsigned int mmc_dev_num = 2;
+ char expected[10];
+ int i, ret;
+
+ ret = setup_gpt_partitions(uts, mmc_dev_num);
+ if (ret == -EAGAIN)
+ return ret;
+
+ ut_assertok(ret);
+
+ for (i = 0; i < ARRAY_SIZE(gpt_parts); i++) {
+ env_set("partnum", NULL);
+ ut_assertok(run_commandf("part number mmc %u %s partnum",
+ mmc_dev_num, gpt_parts[i].name));
+ snprintf(expected, sizeof(expected), "0x%x", i + 1);
+ ut_asserteq_str(expected, env_get("partnum"));
+ }
+
+ env_set("partnum", NULL);
+ ut_asserteq(1, run_commandf("part number mmc %u bogus partnum",
+ mmc_dev_num));
+ ut_assertnull(env_get("partnum"));
+
+ for (i = 0; i < ARRAY_SIZE(gpt_parts); i++) {
+ env_set("partnum", NULL);
+ ut_assertok(run_commandf("part number mmc %u %s partnum",
+ mmc_dev_num, gpt_parts[i].uuid));
+ snprintf(expected, sizeof(expected), "0x%x", i + 1);
+ ut_asserteq_str(expected, env_get("partnum"));
+ }
+
+ env_set("partnum", NULL);
+ ut_asserteq(1, run_commandf("part number mmc %u %s partnum",
+ mmc_dev_num,
+ "00000000-0000-0000-0000-000000000000"));
+ ut_assertnull(env_get("partnum"));
+
+ return 0;
+}
+CMD_TEST(cmd_test_part_number, UTF_CONSOLE);
+
+static int cmd_test_part_start(struct unit_test_state *uts)
+{
+ unsigned int mmc_dev_num = 2;
+ char expected[32];
+ int i, ret;
+
+ ret = setup_gpt_partitions(uts, mmc_dev_num);
+ if (ret == -EAGAIN)
+ return ret;
+
+ ut_assertok(ret);
+
+ for (i = 0; i < ARRAY_SIZE(gpt_parts); i++) {
+ env_set("partstart", NULL);
+ ut_assertok(run_commandf("part start mmc %u %d partstart",
+ mmc_dev_num, i + 1));
+ snprintf(expected, sizeof(expected), "%lx",
+ (unsigned long)gpt_parts[i].start);
+ ut_asserteq_str(expected, env_get("partstart"));
+ }
+
+ env_set("partstart", NULL);
+ ut_asserteq(1, run_commandf("part start mmc %u 3 partstart",
+ mmc_dev_num));
+ ut_assertnull(env_get("partstart"));
+
+ for (i = 0; i < ARRAY_SIZE(gpt_parts); i++) {
+ env_set("partstart", NULL);
+ ut_assertok(run_commandf("part start mmc %u %s partstart",
+ mmc_dev_num, gpt_parts[i].name));
+ snprintf(expected, sizeof(expected), "%lx",
+ (unsigned long)gpt_parts[i].start);
+ ut_asserteq_str(expected, env_get("partstart"));
+ }
+
+ env_set("partstart", NULL);
+ ut_asserteq(1, run_commandf("part start mmc %u bogus partstart",
+ mmc_dev_num));
+ ut_assertnull(env_get("partstart"));
+
+ for (i = 0; i < ARRAY_SIZE(gpt_parts); i++) {
+ env_set("partstart", NULL);
+ ut_assertok(run_commandf("part start mmc %u %s partstart",
+ mmc_dev_num, gpt_parts[i].uuid));
+ snprintf(expected, sizeof(expected), "%lx",
+ (unsigned long)gpt_parts[i].start);
+ ut_asserteq_str(expected, env_get("partstart"));
+ }
+
+ env_set("partstart", NULL);
+ ut_asserteq(1, run_commandf("part start mmc %u %s partstart",
+ mmc_dev_num,
+ "00000000-0000-0000-0000-000000000000"));
+ ut_assertnull(env_get("partstart"));
+
+ return 0;
+}
+CMD_TEST(cmd_test_part_start, UTF_CONSOLE);
+
+static int cmd_test_part_size(struct unit_test_state *uts)
+{
+ unsigned int mmc_dev_num = 2;
+ char expected[32];
+ int i, ret;
+
+ ret = setup_gpt_partitions(uts, mmc_dev_num);
+ if (ret == -EAGAIN)
+ return ret;
+
+ ut_assertok(ret);
+
+ for (i = 0; i < ARRAY_SIZE(gpt_parts); i++) {
+ env_set("partsize", NULL);
+ ut_assertok(run_commandf("part size mmc %u %d partsize",
+ mmc_dev_num, i + 1));
+ snprintf(expected, sizeof(expected), "%lx",
+ (unsigned long)gpt_parts[i].size);
+ ut_asserteq_str(expected, env_get("partsize"));
+ }
+
+ env_set("partsize", NULL);
+ ut_asserteq(1, run_commandf("part size mmc %u 3 partsize",
+ mmc_dev_num));
+ ut_assertnull(env_get("partsize"));
+
+ for (i = 0; i < ARRAY_SIZE(gpt_parts); i++) {
+ env_set("partsize", NULL);
+ ut_assertok(run_commandf("part size mmc %u %s partsize",
+ mmc_dev_num, gpt_parts[i].name));
+ snprintf(expected, sizeof(expected), "%lx",
+ (unsigned long)gpt_parts[i].size);
+ ut_asserteq_str(expected, env_get("partsize"));
+ }
+
+ env_set("partsize", NULL);
+ ut_asserteq(1, run_commandf("part size mmc %u bogus partsize",
+ mmc_dev_num));
+ ut_assertnull(env_get("partsize"));
+
+ for (i = 0; i < ARRAY_SIZE(gpt_parts); i++) {
+ env_set("partsize", NULL);
+ ut_assertok(run_commandf("part size mmc %u %s partsize",
+ mmc_dev_num, gpt_parts[i].uuid));
+ snprintf(expected, sizeof(expected), "%lx",
+ (unsigned long)gpt_parts[i].size);
+ ut_asserteq_str(expected, env_get("partsize"));
+ }
+
+ env_set("partsize", NULL);
+ ut_asserteq(1, run_commandf("part size mmc %u %s partsize",
+ mmc_dev_num,
+ "00000000-0000-0000-0000-000000000000"));
+ ut_assertnull(env_get("partsize"));
+
+ return 0;
+}
+CMD_TEST(cmd_test_part_size, UTF_CONSOLE);
diff --git a/test/dm/fwu_mdata.c b/test/dm/fwu_mdata.c
index cfe543d8a23..8624ccf61f7 100644
--- a/test/dm/fwu_mdata.c
+++ b/test/dm/fwu_mdata.c
@@ -143,3 +143,51 @@ static int dm_test_fwu_mdata_write(struct unit_test_state *uts)
return 0;
}
DM_TEST(dm_test_fwu_mdata_write, UTF_SCAN_FDT);
+
+static int dm_test_fwu_mdata_get_image_guid(struct unit_test_state *uts)
+{
+ efi_guid_t image_type_guid =
+ EFI_GUID(0x09d7cf52, 0x0720, 0x4710, \
+ 0x91, 0xd1, 0x08, 0x46, 0x9b, 0x7f, 0xe9, 0xc8);
+ efi_guid_t bank_0_image_guid =
+ EFI_GUID(0x10057a86, 0xdaf1, 0x4f93, \
+ 0xba, 0x7f, 0xb1, 0x95, 0xf7, 0xfa, 0x41, 0x70);
+ efi_guid_t bank_1_image_guid =
+ EFI_GUID(0xdb62ed3e, 0x6237, 0x4fb4, \
+ 0x80, 0xc4, 0x1b, 0x74, 0xd8, 0x46, 0xa8, 0xe7);
+ efi_guid_t wrong_image_type_guid =
+ EFI_GUID(0x12345678, 0x1302, 0x133f, \
+ 0x18, 0x0a, 0x14, 0x05, 0x18, 0x05, 0x14, 0x0b);
+ struct udevice *dev;
+ efi_guid_t image_guid;
+
+ ut_assertok(setup_blk_device(uts));
+ ut_assertok(populate_mmc_disk_image(uts));
+ ut_assertok(write_mmc_blk_device(uts));
+
+ /*
+ * Trigger lib/fwu_updates/fwu.c fwu_boottime_checks()
+ * to populate g_dev global pointer in that library.
+ */
+ ut_assertok(event_notify_null(EVT_POST_PREBOOT));
+
+ ut_assertok(uclass_first_device_err(UCLASS_FWU_MDATA, &dev));
+
+ ut_assertok(fwu_init());
+
+ ut_assertok(fwu_mdata_get_image_guid(&image_guid, &image_type_guid, 0));
+ ut_assertok(guidcmp(&image_guid, &bank_0_image_guid));
+
+ ut_assertok(fwu_mdata_get_image_guid(&image_guid, &image_type_guid, 1));
+ ut_assertok(guidcmp(&image_guid, &bank_1_image_guid));
+
+ ut_asserteq(-EINVAL, fwu_mdata_get_image_guid(&image_guid,
+ &image_type_guid, 2));
+
+ ut_asserteq(-ENOENT, fwu_mdata_get_image_guid(&image_guid,
+ &wrong_image_type_guid,
+ 0));
+
+ return 0;
+}
+DM_TEST(dm_test_fwu_mdata_get_image_guid, UTF_SCAN_FDT);
diff --git a/test/dm/part.c b/test/dm/part.c
index caae23bd4aa..ad37d7f406f 100644
--- a/test/dm/part.c
+++ b/test/dm/part.c
@@ -195,3 +195,56 @@ static int dm_test_part_get_info_by_type(struct unit_test_state *uts)
return 0;
}
DM_TEST(dm_test_part_get_info_by_type, UTF_SCAN_PDATA | UTF_SCAN_FDT);
+
+static int dm_test_part_get_info_by_uuid(struct unit_test_state *uts)
+{
+ struct disk_partition parts[] = {
+ {
+ .start = 48,
+ .size = 1,
+ .name = "test1",
+ .uuid = "c5bce7a2-03f0-4d03-9048-01ff23b9d527",
+ },
+ {
+ .start = 49,
+ .size = 1,
+ .name = "test2",
+ .uuid = "9df346e8-2c53-4cd8-b9ac-3af83f9a9b74",
+ },
+ };
+ char disk_guid[UUID_STR_LEN + 1] =
+ "8d60b397-1bb6-4d33-80ee-b1587d24c2f8";
+ struct blk_desc *mmc_dev_desc;
+ struct disk_partition info;
+ int part, i;
+
+ ut_asserteq(2, blk_get_device_by_str("mmc", "2", &mmc_dev_desc));
+
+ if (CONFIG_IS_ENABLED(RANDOM_UUID)) {
+ for (i = 0; i < ARRAY_SIZE(parts); i++)
+ gen_rand_uuid_str(parts[i].uuid, UUID_STR_FORMAT_STD);
+
+ gen_rand_uuid_str(disk_guid, UUID_STR_FORMAT_STD);
+ }
+
+ ut_assertok(gpt_restore(mmc_dev_desc, disk_guid, parts,
+ ARRAY_SIZE(parts)));
+
+ for (i = 0; i < ARRAY_SIZE(parts); i++) {
+ part = part_get_info_by_uuid(mmc_dev_desc, parts[i].uuid,
+ &info);
+
+ ut_asserteq(i + 1, part);
+ ut_asserteq_str(parts[i].name, info.name);
+ ut_asserteq(parts[i].start, info.start);
+ ut_asserteq(parts[i].size, info.size);
+ }
+
+ part = part_get_info_by_uuid(mmc_dev_desc,
+ "00000000-0000-0000-0000-000000000000",
+ &info);
+ ut_assert(part < 0);
+
+ return 0;
+}
+DM_TEST(dm_test_part_get_info_by_uuid, UTF_SCAN_PDATA | UTF_SCAN_FDT);
generated by cgit v1.2.3 (git 2.25.1) at 2026-05-16 13:34:09 +0000