1// SPDX-License-Identifier: GPL-2.0+
2/*
3 * Copyright (C) 2015 Google, Inc
4 */
5
6#include <common.h>
7#include <dm.h>
8#include <fastboot.h>
9#include <fb_mmc.h>
10#include <mmc.h>
11#include <part.h>
12#include <part_efi.h>
13#include <dm/test.h>
14#include <test/ut.h>
15#include <linux/stringify.h>
16
17#define FB_ALIAS_PREFIX "fastboot_partition_alias_"
18
19static int dm_test_fastboot_mmc_part(struct unit_test_state *uts)
20{
21	char response[FASTBOOT_RESPONSE_LEN] = {0};
22	char str_disk_guid[UUID_STR_LEN + 1];
23	struct blk_desc *mmc_dev_desc, *fb_dev_desc;
24	struct disk_partition part_info;
25	struct disk_partition parts[2] = {
26		{
27			.start = 48, /* GPT data takes up the first 34 blocks or so */
28			.size = 1,
29			.name = "test1",
30		},
31		{
32			.start = 49,
33			.size = 1,
34			.name = "test2",
35		},
36	};
37
38	/*
39	 * There are a lot of literal 0s I don't want to have to construct from
40	 * MMC_DEV.
41	 */
42	ut_asserteq(0, CONFIG_FASTBOOT_FLASH_MMC_DEV);
43	ut_assertok(blk_get_device_by_str("mmc", "0", &mmc_dev_desc));
44	if (CONFIG_IS_ENABLED(RANDOM_UUID)) {
45		gen_rand_uuid_str(parts[0].uuid, UUID_STR_FORMAT_STD);
46		gen_rand_uuid_str(parts[1].uuid, UUID_STR_FORMAT_STD);
47		gen_rand_uuid_str(str_disk_guid, UUID_STR_FORMAT_STD);
48	}
49	ut_assertok(gpt_restore(mmc_dev_desc, str_disk_guid, parts,
50				ARRAY_SIZE(parts)));
51
52	/* "Classic" partition labels */
53	ut_asserteq(1, fastboot_mmc_get_part_info("test1", &fb_dev_desc,
54						  &part_info, response));
55	ut_asserteq(2, fastboot_mmc_get_part_info("test2", &fb_dev_desc,
56						  &part_info, response));
57
58	/* Test aliases */
59	ut_assertnull(env_get(FB_ALIAS_PREFIX "test3"));
60	ut_assertok(env_set(FB_ALIAS_PREFIX "test3", "test1"));
61	ut_asserteq(1, fastboot_mmc_get_part_info("test3", &fb_dev_desc,
62						  &part_info, response));
63	ut_assertok(env_set(FB_ALIAS_PREFIX "test3", NULL));
64
65	/* "New" partition labels */
66	ut_asserteq(1, fastboot_mmc_get_part_info("#test1", &fb_dev_desc,
67						  &part_info, response));
68	ut_asserteq(1, fastboot_mmc_get_part_info("0#test1", &fb_dev_desc,
69						  &part_info, response));
70	ut_asserteq(1, fastboot_mmc_get_part_info("0.0#test1", &fb_dev_desc,
71						  &part_info, response));
72	ut_asserteq(1, fastboot_mmc_get_part_info("0:1", &fb_dev_desc,
73						  &part_info, response));
74	ut_asserteq(1, fastboot_mmc_get_part_info("0.0:1", &fb_dev_desc,
75						  &part_info, response));
76	ut_asserteq(1, fastboot_mmc_get_part_info("0", &fb_dev_desc,
77						  &part_info, response));
78	ut_asserteq(1, fastboot_mmc_get_part_info("0.0", &fb_dev_desc,
79						  &part_info, response));
80	ut_asserteq(0, fastboot_mmc_get_part_info("0:0", &fb_dev_desc,
81						  &part_info, response));
82	ut_asserteq(0, fastboot_mmc_get_part_info("0.0:0", &fb_dev_desc,
83						  &part_info, response));
84	ut_asserteq(0, fastboot_mmc_get_part_info("2", &fb_dev_desc,
85						  &part_info, response));
86	ut_asserteq(0, fastboot_mmc_get_part_info("2.0", &fb_dev_desc,
87						  &part_info, response));
88	ut_asserteq(1, fastboot_mmc_get_part_info(":1", &fb_dev_desc,
89						  &part_info, response));
90	ut_asserteq(0, fastboot_mmc_get_part_info(":0", &fb_dev_desc,
91						  &part_info, response));
92
93	return 0;
94}
95DM_TEST(dm_test_fastboot_mmc_part, UT_TESTF_SCAN_PDATA | UT_TESTF_SCAN_FDT);
96