test/dm/part.c

   1 // 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
  14 static inline int do_test(struct unit_test_state *uts, int expected,
  15                           const char *part_str, bool whole)
  16 {
  17         struct blk_desc *mmc_dev_desc;
  18         struct disk_partition part_info;
  19
  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
  27 static 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;
  33         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
  55         oldbootdevice = env_get("bootdevice");
  56
  57 #define test(expected, part_str, whole) do { \
  58         ret = do_test(uts, expected, part_str, whole); \
  59         if (ret) \
  60                 goto out; \
  61 } while (0)
  62
  63         env_set("bootdevice", NULL);
  64         test(-ENODEV, NULL, true);
  65         test(-ENODEV, "", true);
  66         env_set("bootdevice", "0");
  67         test(0, NULL, true);
  68         test(0, "", true);
  69         env_set("bootdevice", "1");
  70         test(1, NULL, false);
  71         test(1, "", false);
  72         test(1, "-", false);
  73         env_set("bootdevice", "");
  74         test(-EPROTONOSUPPORT, "0", false);
  75         test(0, "0", true);
  76         test(0, ":0", true);
  77         test(0, ".0", true);
  78         test(0, ".0:0", true);
  79         test(-EINVAL, "#test1", true);
  80         test(1, "1", false);
  81         test(1, "1", true);
  82         test(-ENOENT, "1:0", false);
  83         test(0, "1:0", true);
  84         test(1, "1:1", false);
  85         test(2, "1:2", false);
  86         test(1, "1.0", false);
  87         test(0, "1.0:0", true);
  88         test(1, "1.0:1", false);
  89         test(2, "1.0:2", false);
  90         test(-EINVAL, "1#bogus", false);
  91         test(1, "1#test1", false);
  92         test(2, "1#test2", false);
  93         ret = 0;
  94
  95 out:
  96         env_set("bootdevice", oldbootdevice);
  97         return ret;
  98 }
  99 DM_TEST(dm_test_part, UT_TESTF_SCAN_PDATA | UT_TESTF_SCAN_FDT);