mirror of
				https://source.denx.de/u-boot/u-boot.git
				synced 2025-10-31 00:11:51 +01:00 
			
		
		
		
	If a disk has a bootable partition we are expected to use it to locate the boot files. Add a function to find it. To test this, update mmc1 to have two paritions, fixing up other tests accordingly. Signed-off-by: Simon Glass <sjg@chromium.org>
		
			
				
	
	
		
			109 lines
		
	
	
		
			2.6 KiB
		
	
	
	
		
			C
		
	
	
	
	
	
			
		
		
	
	
			109 lines
		
	
	
		
			2.6 KiB
		
	
	
	
		
			C
		
	
	
	
	
	
| // SPDX-License-Identifier: GPL-2.0+
 | |
| /*
 | |
|  * Copyright (C) 2020 Sean Anderson <sean.anderson@seco.com>
 | |
|  */
 | |
| 
 | |
| #include <common.h>
 | |
| #include <dm.h>
 | |
| #include <mmc.h>
 | |
| #include <part.h>
 | |
| #include <part_efi.h>
 | |
| #include <dm/test.h>
 | |
| #include <test/ut.h>
 | |
| 
 | |
| static int do_test(struct unit_test_state *uts, int expected,
 | |
| 		   const char *part_str, bool whole)
 | |
| {
 | |
| 	struct blk_desc *mmc_dev_desc;
 | |
| 	struct disk_partition part_info;
 | |
| 
 | |
| 	ut_asserteq(expected,
 | |
| 		    part_get_info_by_dev_and_name_or_num("mmc", part_str,
 | |
| 							 &mmc_dev_desc,
 | |
| 							 &part_info, whole));
 | |
| 	return 0;
 | |
| }
 | |
| 
 | |
| static int dm_test_part(struct unit_test_state *uts)
 | |
| {
 | |
| 	char *oldbootdevice;
 | |
| 	char str_disk_guid[UUID_STR_LEN + 1];
 | |
| 	int ret;
 | |
| 	struct blk_desc *mmc_dev_desc;
 | |
| 	struct disk_partition parts[2] = {
 | |
| 		{
 | |
| 			.start = 48, /* GPT data takes up the first 34 blocks or so */
 | |
| 			.size = 1,
 | |
| 			.name = "test1",
 | |
| 		},
 | |
| 		{
 | |
| 			.start = 49,
 | |
| 			.size = 1,
 | |
| 			.name = "test2",
 | |
| 		},
 | |
| 	};
 | |
| 
 | |
| 	ut_asserteq(2, blk_get_device_by_str("mmc", "2", &mmc_dev_desc));
 | |
| 	if (CONFIG_IS_ENABLED(RANDOM_UUID)) {
 | |
| 		gen_rand_uuid_str(parts[0].uuid, UUID_STR_FORMAT_STD);
 | |
| 		gen_rand_uuid_str(parts[1].uuid, UUID_STR_FORMAT_STD);
 | |
| 		gen_rand_uuid_str(str_disk_guid, UUID_STR_FORMAT_STD);
 | |
| 	}
 | |
| 	ut_assertok(gpt_restore(mmc_dev_desc, str_disk_guid, parts,
 | |
| 				ARRAY_SIZE(parts)));
 | |
| 
 | |
| 	oldbootdevice = env_get("bootdevice");
 | |
| 
 | |
| #define test(expected, part_str, whole) \
 | |
| 	ut_assertok(do_test(uts, expected, part_str, whole))
 | |
| 
 | |
| 	env_set("bootdevice", NULL);
 | |
| 	test(-ENODEV, NULL, true);
 | |
| 	test(-ENODEV, "", true);
 | |
| 	env_set("bootdevice", "0");
 | |
| 	test(0, NULL, true);
 | |
| 	test(0, "", true);
 | |
| 	env_set("bootdevice", "2");
 | |
| 	test(1, NULL, false);
 | |
| 	test(1, "", false);
 | |
| 	test(1, "-", false);
 | |
| 	env_set("bootdevice", "");
 | |
| 	test(-EPROTONOSUPPORT, "0", false);
 | |
| 	test(0, "0", true);
 | |
| 	test(0, ":0", true);
 | |
| 	test(0, ".0", true);
 | |
| 	test(0, ".0:0", true);
 | |
| 	test(-EINVAL, "#test1", true);
 | |
| 	test(1, "2", false);
 | |
| 	test(1, "2", true);
 | |
| 	test(-ENOENT, "1:0", false);
 | |
| 	test(0, "1:0", true);
 | |
| 	test(1, "1:1", false);
 | |
| 	test(2, "1:2", false);
 | |
| 	test(1, "1.0", false);
 | |
| 	test(0, "1.0:0", true);
 | |
| 	test(1, "1.0:1", false);
 | |
| 	test(2, "1.0:2", false);
 | |
| 	test(-EINVAL, "1#bogus", false);
 | |
| 	test(1, "2#test1", false);
 | |
| 	test(2, "2#test2", false);
 | |
| 	ret = 0;
 | |
| 
 | |
| 	env_set("bootdevice", oldbootdevice);
 | |
| 	return ret;
 | |
| }
 | |
| DM_TEST(dm_test_part, UT_TESTF_SCAN_PDATA | UT_TESTF_SCAN_FDT);
 | |
| 
 | |
| static int dm_test_part_bootable(struct unit_test_state *uts)
 | |
| {
 | |
| 	struct blk_desc *desc;
 | |
| 	struct udevice *dev;
 | |
| 
 | |
| 	ut_assertok(uclass_get_device_by_name(UCLASS_BLK, "mmc1.blk", &dev));
 | |
| 	desc = dev_get_uclass_plat(dev);
 | |
| 	ut_asserteq(1, part_get_bootable(desc));
 | |
| 
 | |
| 	return 0;
 | |
| }
 | |
| DM_TEST(dm_test_part_bootable, UT_TESTF_SCAN_FDT);
 |