blob: 5e4efa587c33a9e43fde6d5ce75ce1726b650ca8 [file] [log] [blame]
Sean Anderson535e7002021-04-12 18:53:07 -04001// SPDX-License-Identifier: GPL-2.0+
2/*
3 * Copyright (C) 2020 Sean Anderson <sean.anderson@seco.com>
4 */
5
6#include <common.h>
7#include <dm.h>
8#include <mmc.h>
9#include <part.h>
10#include <part_efi.h>
11#include <dm/test.h>
12#include <test/ut.h>
13
Simon Glassa0fb9de2023-01-17 10:47:13 -070014static int do_test(struct unit_test_state *uts, int expected,
15 const char *part_str, bool whole)
Sean Anderson535e7002021-04-12 18:53:07 -040016{
Sean Anderson535e7002021-04-12 18:53:07 -040017 struct blk_desc *mmc_dev_desc;
18 struct disk_partition part_info;
Sean Anderson26de4292021-05-15 14:13:54 -040019
20 ut_asserteq(expected,
21 part_get_info_by_dev_and_name_or_num("mmc", part_str,
22 &mmc_dev_desc,
23 &part_info, whole));
24 return 0;
25}
26
27static int dm_test_part(struct unit_test_state *uts)
28{
29 char *oldbootdevice;
30 char str_disk_guid[UUID_STR_LEN + 1];
31 int ret;
32 struct blk_desc *mmc_dev_desc;
Sean Anderson535e7002021-04-12 18:53:07 -040033 struct disk_partition parts[2] = {
34 {
35 .start = 48, /* GPT data takes up the first 34 blocks or so */
36 .size = 1,
37 .name = "test1",
38 },
39 {
40 .start = 49,
41 .size = 1,
42 .name = "test2",
43 },
44 };
45
46 ut_asserteq(1, blk_get_device_by_str("mmc", "1", &mmc_dev_desc));
47 if (CONFIG_IS_ENABLED(RANDOM_UUID)) {
48 gen_rand_uuid_str(parts[0].uuid, UUID_STR_FORMAT_STD);
49 gen_rand_uuid_str(parts[1].uuid, UUID_STR_FORMAT_STD);
50 gen_rand_uuid_str(str_disk_guid, UUID_STR_FORMAT_STD);
51 }
52 ut_assertok(gpt_restore(mmc_dev_desc, str_disk_guid, parts,
53 ARRAY_SIZE(parts)));
54
Sean Anderson26de4292021-05-15 14:13:54 -040055 oldbootdevice = env_get("bootdevice");
Sean Anderson535e7002021-04-12 18:53:07 -040056
Simon Glassa0fb9de2023-01-17 10:47:13 -070057#define test(expected, part_str, whole) \
58 ut_assertok(do_test(uts, expected, part_str, whole))
Sean Anderson26de4292021-05-15 14:13:54 -040059
60 env_set("bootdevice", NULL);
61 test(-ENODEV, NULL, true);
Sean Anderson535e7002021-04-12 18:53:07 -040062 test(-ENODEV, "", true);
63 env_set("bootdevice", "0");
Sean Anderson26de4292021-05-15 14:13:54 -040064 test(0, NULL, true);
Sean Anderson535e7002021-04-12 18:53:07 -040065 test(0, "", true);
66 env_set("bootdevice", "1");
Sean Anderson26de4292021-05-15 14:13:54 -040067 test(1, NULL, false);
Sean Anderson535e7002021-04-12 18:53:07 -040068 test(1, "", false);
69 test(1, "-", false);
70 env_set("bootdevice", "");
71 test(-EPROTONOSUPPORT, "0", false);
72 test(0, "0", true);
73 test(0, ":0", true);
74 test(0, ".0", true);
75 test(0, ".0:0", true);
76 test(-EINVAL, "#test1", true);
77 test(1, "1", false);
78 test(1, "1", true);
79 test(-ENOENT, "1:0", false);
80 test(0, "1:0", true);
81 test(1, "1:1", false);
82 test(2, "1:2", false);
83 test(1, "1.0", false);
84 test(0, "1.0:0", true);
85 test(1, "1.0:1", false);
86 test(2, "1.0:2", false);
87 test(-EINVAL, "1#bogus", false);
88 test(1, "1#test1", false);
89 test(2, "1#test2", false);
Sean Anderson26de4292021-05-15 14:13:54 -040090 ret = 0;
Sean Anderson535e7002021-04-12 18:53:07 -040091
Sean Anderson26de4292021-05-15 14:13:54 -040092 env_set("bootdevice", oldbootdevice);
93 return ret;
Sean Anderson535e7002021-04-12 18:53:07 -040094}
95DM_TEST(dm_test_part, UT_TESTF_SCAN_PDATA | UT_TESTF_SCAN_FDT);