1 // SPDX-License-Identifier: GPL-2.0+
2 
3 #include <common.h>
4 #include <dm.h>
5 #include <log.h>
6 #include <dm/of_extra.h>
7 #include <dm/test.h>
8 #include <test/test.h>
9 #include <test/ut.h>
10 
dm_test_ofnode_compatible(struct unit_test_state * uts)11 static int dm_test_ofnode_compatible(struct unit_test_state *uts)
12 {
13 	ofnode root_node = ofnode_path("/");
14 
15 	ut_assert(ofnode_valid(root_node));
16 	ut_assert(ofnode_device_is_compatible(root_node, "sandbox"));
17 
18 	return 0;
19 }
20 DM_TEST(dm_test_ofnode_compatible, UT_TESTF_SCAN_PDATA | UT_TESTF_SCAN_FDT);
21 
dm_test_ofnode_get_by_phandle(struct unit_test_state * uts)22 static int dm_test_ofnode_get_by_phandle(struct unit_test_state *uts)
23 {
24 	/* test invalid phandle */
25 	ut_assert(!ofnode_valid(ofnode_get_by_phandle(0)));
26 	ut_assert(!ofnode_valid(ofnode_get_by_phandle(-1)));
27 
28 	/* test first valid phandle */
29 	ut_assert(ofnode_valid(ofnode_get_by_phandle(1)));
30 
31 	/* test unknown phandle */
32 	ut_assert(!ofnode_valid(ofnode_get_by_phandle(0x1000000)));
33 
34 	return 0;
35 }
36 DM_TEST(dm_test_ofnode_get_by_phandle, UT_TESTF_SCAN_PDATA | UT_TESTF_SCAN_FDT);
37 
dm_test_ofnode_by_prop_value(struct unit_test_state * uts)38 static int dm_test_ofnode_by_prop_value(struct unit_test_state *uts)
39 {
40 	const char propname[] = "compatible";
41 	const char propval[] = "denx,u-boot-fdt-test";
42 	const char *str;
43 	ofnode node = ofnode_null();
44 
45 	/* Find first matching node, there should be at least one */
46 	node = ofnode_by_prop_value(node, propname, propval, sizeof(propval));
47 	ut_assert(ofnode_valid(node));
48 	str = ofnode_read_string(node, propname);
49 	ut_assert(str && !strcmp(str, propval));
50 
51 	/* Find the rest of the matching nodes */
52 	while (true) {
53 		node = ofnode_by_prop_value(node, propname, propval,
54 					    sizeof(propval));
55 		if (!ofnode_valid(node))
56 			break;
57 		str = ofnode_read_string(node, propname);
58 		ut_assert(str && !strcmp(str, propval));
59 	}
60 
61 	return 0;
62 }
63 DM_TEST(dm_test_ofnode_by_prop_value, UT_TESTF_SCAN_FDT);
64 
dm_test_ofnode_fmap(struct unit_test_state * uts)65 static int dm_test_ofnode_fmap(struct unit_test_state *uts)
66 {
67 	struct fmap_entry entry;
68 	ofnode node;
69 
70 	node = ofnode_path("/cros-ec/flash");
71 	ut_assert(ofnode_valid(node));
72 	ut_assertok(ofnode_read_fmap_entry(node, &entry));
73 	ut_asserteq(0x08000000, entry.offset);
74 	ut_asserteq(0x20000, entry.length);
75 
76 	return 0;
77 }
78 DM_TEST(dm_test_ofnode_fmap, UT_TESTF_SCAN_PDATA | UT_TESTF_SCAN_FDT);
79 
dm_test_ofnode_read(struct unit_test_state * uts)80 static int dm_test_ofnode_read(struct unit_test_state *uts)
81 {
82 	const u32 *val;
83 	ofnode node;
84 	int size;
85 
86 	node = ofnode_path("/a-test");
87 	ut_assert(ofnode_valid(node));
88 
89 	val = ofnode_read_prop(node, "int-value", &size);
90 	ut_assertnonnull(val);
91 	ut_asserteq(4, size);
92 	ut_asserteq(1234, fdt32_to_cpu(val[0]));
93 
94 	val = ofnode_read_prop(node, "missing", &size);
95 	ut_assertnull(val);
96 	ut_asserteq(-FDT_ERR_NOTFOUND, size);
97 
98 	/* Check it works without a size parameter */
99 	val = ofnode_read_prop(node, "missing", NULL);
100 	ut_assertnull(val);
101 
102 	return 0;
103 }
104 DM_TEST(dm_test_ofnode_read, UT_TESTF_SCAN_PDATA | UT_TESTF_SCAN_FDT);
105 
dm_test_ofnode_phandle(struct unit_test_state * uts)106 static int dm_test_ofnode_phandle(struct unit_test_state *uts)
107 {
108 	struct ofnode_phandle_args args;
109 	ofnode node;
110 	int ret;
111 	const char prop[] = "test-gpios";
112 	const char cell[] = "#gpio-cells";
113 	const char prop2[] = "phandle-value";
114 
115 	node = ofnode_path("/a-test");
116 	ut_assert(ofnode_valid(node));
117 
118 	/* Test ofnode_count_phandle_with_args with cell name */
119 	ret = ofnode_count_phandle_with_args(node, "missing", cell, 0);
120 	ut_asserteq(-ENOENT, ret);
121 	ret = ofnode_count_phandle_with_args(node, prop, "#invalid", 0);
122 	ut_asserteq(-EINVAL, ret);
123 	ret = ofnode_count_phandle_with_args(node, prop, cell, 0);
124 	ut_asserteq(5, ret);
125 
126 	/* Test ofnode_parse_phandle_with_args with cell name */
127 	ret = ofnode_parse_phandle_with_args(node, "missing", cell, 0, 0,
128 					     &args);
129 	ut_asserteq(-ENOENT, ret);
130 	ret = ofnode_parse_phandle_with_args(node, prop, "#invalid", 0, 0,
131 					     &args);
132 	ut_asserteq(-EINVAL, ret);
133 	ret = ofnode_parse_phandle_with_args(node, prop, cell, 0, 0, &args);
134 	ut_assertok(ret);
135 	ut_asserteq(1, args.args_count);
136 	ut_asserteq(1, args.args[0]);
137 	ret = ofnode_parse_phandle_with_args(node, prop, cell, 0, 1, &args);
138 	ut_assertok(ret);
139 	ut_asserteq(1, args.args_count);
140 	ut_asserteq(4, args.args[0]);
141 	ret = ofnode_parse_phandle_with_args(node, prop, cell, 0, 2, &args);
142 	ut_assertok(ret);
143 	ut_asserteq(5, args.args_count);
144 	ut_asserteq(5, args.args[0]);
145 	ut_asserteq(1, args.args[4]);
146 	ret = ofnode_parse_phandle_with_args(node, prop, cell, 0, 3, &args);
147 	ut_asserteq(-ENOENT, ret);
148 	ret = ofnode_parse_phandle_with_args(node, prop, cell, 0, 4, &args);
149 	ut_assertok(ret);
150 	ut_asserteq(1, args.args_count);
151 	ut_asserteq(12, args.args[0]);
152 	ret = ofnode_parse_phandle_with_args(node, prop, cell, 0, 5, &args);
153 	ut_asserteq(-ENOENT, ret);
154 
155 	/* Test ofnode_count_phandle_with_args with cell count */
156 	ret = ofnode_count_phandle_with_args(node, "missing", NULL, 2);
157 	ut_asserteq(-ENOENT, ret);
158 	ret = ofnode_count_phandle_with_args(node, prop2, NULL, 1);
159 	ut_asserteq(3, ret);
160 
161 	/* Test ofnode_parse_phandle_with_args with cell count */
162 	ret = ofnode_parse_phandle_with_args(node, prop2, NULL, 1, 0, &args);
163 	ut_assertok(ret);
164 	ut_asserteq(1, ofnode_valid(args.node));
165 	ut_asserteq(1, args.args_count);
166 	ut_asserteq(10, args.args[0]);
167 	ret = ofnode_parse_phandle_with_args(node, prop2, NULL, 1, 1, &args);
168 	ut_asserteq(-EINVAL, ret);
169 	ret = ofnode_parse_phandle_with_args(node, prop2, NULL, 1, 2, &args);
170 	ut_assertok(ret);
171 	ut_asserteq(1, ofnode_valid(args.node));
172 	ut_asserteq(1, args.args_count);
173 	ut_asserteq(30, args.args[0]);
174 	ret = ofnode_parse_phandle_with_args(node, prop2, NULL, 1, 3, &args);
175 	ut_asserteq(-ENOENT, ret);
176 
177 	return 0;
178 }
179 DM_TEST(dm_test_ofnode_phandle, UT_TESTF_SCAN_PDATA | UT_TESTF_SCAN_FDT);
180 
dm_test_ofnode_read_chosen(struct unit_test_state * uts)181 static int dm_test_ofnode_read_chosen(struct unit_test_state *uts)
182 {
183 	const char *str;
184 	const u32 *val;
185 	ofnode node;
186 	int size;
187 
188 	str = ofnode_read_chosen_string("setting");
189 	ut_assertnonnull(str);
190 	ut_asserteq_str("sunrise ohoka", str);
191 	ut_asserteq_ptr(NULL, ofnode_read_chosen_string("no-setting"));
192 
193 	node = ofnode_get_chosen_node("other-node");
194 	ut_assert(ofnode_valid(node));
195 	ut_asserteq_str("c-test@5", ofnode_get_name(node));
196 
197 	node = ofnode_get_chosen_node("setting");
198 	ut_assert(!ofnode_valid(node));
199 
200 	val = ofnode_read_chosen_prop("int-values", &size);
201 	ut_assertnonnull(val);
202 	ut_asserteq(8, size);
203 	ut_asserteq(0x1937, fdt32_to_cpu(val[0]));
204 	ut_asserteq(72993, fdt32_to_cpu(val[1]));
205 
206 	return 0;
207 }
208 DM_TEST(dm_test_ofnode_read_chosen, UT_TESTF_SCAN_PDATA | UT_TESTF_SCAN_FDT);
209 
dm_test_ofnode_read_aliases(struct unit_test_state * uts)210 static int dm_test_ofnode_read_aliases(struct unit_test_state *uts)
211 {
212 	const void *val;
213 	ofnode node;
214 	int size;
215 
216 	node = ofnode_get_aliases_node("ethernet3");
217 	ut_assert(ofnode_valid(node));
218 	ut_asserteq_str("sbe5", ofnode_get_name(node));
219 
220 	node = ofnode_get_aliases_node("unknown");
221 	ut_assert(!ofnode_valid(node));
222 
223 	val = ofnode_read_aliases_prop("spi0", &size);
224 	ut_assertnonnull(val);
225 	ut_asserteq(7, size);
226 	ut_asserteq_str("/spi@0", (const char *)val);
227 
228 	return 0;
229 }
230 DM_TEST(dm_test_ofnode_read_aliases, UT_TESTF_SCAN_PDATA | UT_TESTF_SCAN_FDT);
231 
dm_test_ofnode_get_child_count(struct unit_test_state * uts)232 static int dm_test_ofnode_get_child_count(struct unit_test_state *uts)
233 {
234 	ofnode node, child_node;
235 	u32 val;
236 
237 	node = ofnode_path("/i-test");
238 	ut_assert(ofnode_valid(node));
239 
240 	val = ofnode_get_child_count(node);
241 	ut_asserteq(3, val);
242 
243 	child_node = ofnode_first_subnode(node);
244 	ut_assert(ofnode_valid(child_node));
245 	val = ofnode_get_child_count(child_node);
246 	ut_asserteq(0, val);
247 
248 	return 0;
249 }
250 DM_TEST(dm_test_ofnode_get_child_count,
251 	UT_TESTF_SCAN_PDATA | UT_TESTF_SCAN_FDT);
252 
dm_test_ofnode_is_enabled(struct unit_test_state * uts)253 static int dm_test_ofnode_is_enabled(struct unit_test_state *uts)
254 {
255 	ofnode root_node = ofnode_path("/");
256 	ofnode node = ofnode_path("/usb@0");
257 
258 	ut_assert(ofnode_is_enabled(root_node));
259 	ut_assert(!ofnode_is_enabled(node));
260 
261 	return 0;
262 }
263 DM_TEST(dm_test_ofnode_is_enabled, UT_TESTF_SCAN_PDATA | UT_TESTF_SCAN_FDT);
264 
dm_test_ofnode_get_reg(struct unit_test_state * uts)265 static int dm_test_ofnode_get_reg(struct unit_test_state *uts)
266 {
267 	ofnode node;
268 	fdt_addr_t addr;
269 	fdt_size_t size;
270 
271 	node = ofnode_path("/translation-test@8000");
272 	ut_assert(ofnode_valid(node));
273 	addr = ofnode_get_addr(node);
274 	size = ofnode_get_size(node);
275 	ut_asserteq(0x8000, addr);
276 	ut_asserteq(0x4000, size);
277 
278 	node = ofnode_path("/translation-test@8000/dev@1,100");
279 	ut_assert(ofnode_valid(node));
280 	addr = ofnode_get_addr(node);
281 	size = ofnode_get_size(node);
282 	ut_asserteq(0x9000, addr);
283 	ut_asserteq(0x1000, size);
284 
285 	node = ofnode_path("/emul-mux-controller");
286 	ut_assert(ofnode_valid(node));
287 	addr = ofnode_get_addr(node);
288 	size = ofnode_get_size(node);
289 	ut_asserteq(FDT_ADDR_T_NONE, addr);
290 	ut_asserteq(FDT_SIZE_T_NONE, size);
291 
292 	return 0;
293 }
294 DM_TEST(dm_test_ofnode_get_reg, UT_TESTF_SCAN_PDATA | UT_TESTF_SCAN_FDT);
295