lib/fdtdec_test.c

   1 // SPDX-License-Identifier: GPL-2.0+
   2 /*
   3  * Some very basic tests for fdtdec, accessed through test_fdtdec command.
   4  * They are easiest to use with sandbox.
   5  *
   6  * Copyright (c) 2011 The Chromium OS Authors.
   7  */
   8
   9 #include <common.h>
  10 #include <fdtdec.h>
  11 #include <linux/libfdt.h>
  12 #include <malloc.h>
  13 #include <os.h>
  14
  15 /* The size of our test fdt blob */
  16 #define FDT_SIZE        (16 * 1024)
  17
  18 #define CHECK(op) ({                                                    \
  19                 int err = op;                                           \
  20                 if (err < 0) {                                          \
  21                         printf("%s: %s: %s\n", __func__, #op,           \
  22                                fdt_strerror(err));                      \
  23                         return err;                                     \
  24                 }                                                       \
  25                                                                         \
  26                 err;                                                    \
  27         })
  28
  29 #define CHECKVAL(op, expected) ({                                       \
  30                 int err = op;                                           \
  31                 if (err != expected) {                                  \
  32                         printf("%s: %s: expected %d, but returned %d\n",\
  33                                __func__, #op, expected, err);           \
  34                         return err;                                     \
  35                 }                                                       \
  36                                                                         \
  37                 err;                                                    \
  38         })
  39
  40 #define CHECKOK(op)     CHECKVAL(op, 0)
  41
  42 /* maximum number of nodes / aliases to generate */
  43 #define MAX_NODES       20
  44
  45 /*
  46  * Make a test fdt
  47  *
  48  * @param fdt           Device tree pointer
  49  * @param size          Size of device tree blob
  50  * @param aliases       Specifies alias assignments. Format is a list of items
  51  *                      separated by space. Items are #a where
  52  *                              # is the alias number
  53  *                              a is the node to point to
  54  * @param nodes         Specifies nodes to generate (a=0, b=1), upper case
  55  *                      means to create a disabled node
  56  */
  57 static int make_fdt(void *fdt, int size, const char *aliases,
  58                     const char *nodes)
  59 {
  60         char name[20], value[20];
  61         const char *s;
  62 #if defined(DEBUG) && defined(CONFIG_SANDBOX)
  63         int fd;
  64 #endif
  65
  66         CHECK(fdt_create(fdt, size));
  67         CHECK(fdt_finish_reservemap(fdt));
  68         CHECK(fdt_begin_node(fdt, ""));
  69
  70         CHECK(fdt_begin_node(fdt, "aliases"));
  71         for (s = aliases; *s;) {
  72                 sprintf(name, "i2c%c", *s);
  73                 sprintf(value, "/i2c%d@0", s[1] - 'a');
  74                 CHECK(fdt_property_string(fdt, name, value));
  75                 s += 2 + (s[2] != '\0');
  76         }
  77         CHECK(fdt_end_node(fdt));
  78
  79         for (s = nodes; *s; s++) {
  80                 sprintf(value, "i2c%d@0", (*s & 0xdf) - 'A');
  81                 CHECK(fdt_begin_node(fdt, value));
  82                 CHECK(fdt_property_string(fdt, "compatible",
  83                         fdtdec_get_compatible(COMPAT_UNKNOWN)));
  84                 if (*s <= 'Z')
  85                         CHECK(fdt_property_string(fdt, "status", "disabled"));
  86                 CHECK(fdt_end_node(fdt));
  87         }
  88
  89         CHECK(fdt_end_node(fdt));
  90         CHECK(fdt_finish(fdt));
  91         CHECK(fdt_pack(fdt));
  92 #if defined(DEBUG) && defined(CONFIG_SANDBOX)
  93         fd = os_open("/tmp/fdtdec-text.dtb", OS_O_CREAT | OS_O_WRONLY);
  94         if (fd == -1) {
  95                 printf("Could not open .dtb file to write\n");
  96                 return -1;
  97         }
  98         os_write(fd, fdt, size);
  99         os_close(fd);
 100 #endif
 101         return 0;
 102 }
 103
 104 static int run_test(const char *aliases, const char *nodes, const char *expect)
 105 {
 106         int list[MAX_NODES];
 107         const char *s;
 108         void *blob;
 109         int i;
 110
 111         blob = malloc(FDT_SIZE);
 112         if (!blob) {
 113                 printf("%s: out of memory\n", __func__);
 114                 return 1;
 115         }
 116
 117         printf("aliases=%s, nodes=%s, expect=%s: ", aliases, nodes, expect);
 118         CHECKVAL(make_fdt(blob, FDT_SIZE, aliases, nodes), 0);
 119         CHECKVAL(fdtdec_find_aliases_for_id(blob, "i2c",
 120                         COMPAT_UNKNOWN,
 121                         list, ARRAY_SIZE(list)), (int)strlen(expect));
 122
 123         /* Check we got the right ones */
 124         for (i = 0, s = expect; *s; s++, i++) {
 125                 int want = *s;
 126                 const char *name;
 127                 int got = ' ';
 128
 129                 name = list[i] ? fdt_get_name(blob, list[i], NULL) : NULL;
 130                 if (name)
 131                         got = name[3] + 'a' - '0';
 132
 133                 if (got != want) {
 134                         printf("Position %d: Expected '%c', got '%c' ('%s')\n",
 135                                i, want, got, name);
 136                         return 1;
 137                 }
 138         }
 139
 140         printf("pass\n");
 141         return 0;
 142 }
 143
 144 static int do_test_fdtdec(cmd_tbl_t *cmdtp, int flag, int argc,
 145                           char * const argv[])
 146 {
 147         /* basic tests */
 148         CHECKOK(run_test("", "", ""));
 149         CHECKOK(run_test("1e 3d", "", ""));
 150
 151         /*
 152          * 'a' represents 0, 'b' represents 1, etc.
 153          * The first character is the alias number, the second is the node
 154          * number. So the params mean:
 155          * 0a 1b        : point alias 0 to node 0 (a), alias 1 to node 1(b)
 156          * ab           : to create nodes 0 and 1 (a and b)
 157          * ab           : we expect the function to return two nodes, in
 158          *                the order 0, 1
 159          */
 160         CHECKOK(run_test("0a 1b", "ab", "ab"));
 161
 162         CHECKOK(run_test("0a 1c", "ab", "ab"));
 163         CHECKOK(run_test("1c", "ab", "ab"));
 164         CHECKOK(run_test("1b", "ab", "ab"));
 165         CHECKOK(run_test("0b", "ab", "ba"));
 166         CHECKOK(run_test("0b 2d", "dbc", "bcd"));
 167         CHECKOK(run_test("0d 3a 1c 2b", "dbac", "dcba"));
 168
 169         /* things with holes */
 170         CHECKOK(run_test("1b 3d", "dbc", "cb d"));
 171         CHECKOK(run_test("1e 3d", "dbc", "bc d"));
 172
 173         /* no aliases */
 174         CHECKOK(run_test("", "dbac", "dbac"));
 175
 176         /* disabled nodes */
 177         CHECKOK(run_test("0d 3a 1c 2b", "dBac", "dc a"));
 178         CHECKOK(run_test("0b 2d", "DBc", "c"));
 179         CHECKOK(run_test("0b 4d 2c", "DBc", "  c"));
 180
 181         /* conflicting aliases - first one gets it */
 182         CHECKOK(run_test("2a 1a 0a", "a", "  a"));
 183         CHECKOK(run_test("0a 1a 2a", "a", "a"));
 184
 185         printf("Test passed\n");
 186         return 0;
 187 }
 188
 189 U_BOOT_CMD(
 190         test_fdtdec, 3, 1, do_test_fdtdec,
 191         "test_fdtdec",
 192         "Run tests for fdtdec library");