2 * Copyright (c) 2013 Google, Inc
4 * SPDX-License-Identifier: GPL-2.0+
15 #include <dm/device-internal.h>
16 #include <dm/uclass-internal.h>
20 DECLARE_GLOBAL_DATA_PTR
;
22 static int testfdt_drv_ping(struct udevice
*dev
, int pingval
, int *pingret
)
24 const struct dm_test_pdata
*pdata
= dev
->platdata
;
25 struct dm_test_priv
*priv
= dev_get_priv(dev
);
27 *pingret
= pingval
+ pdata
->ping_add
;
28 priv
->ping_total
+= *pingret
;
33 static const struct test_ops test_ops
= {
34 .ping
= testfdt_drv_ping
,
37 static int testfdt_ofdata_to_platdata(struct udevice
*dev
)
39 struct dm_test_pdata
*pdata
= dev_get_platdata(dev
);
41 pdata
->ping_add
= fdtdec_get_int(gd
->fdt_blob
, dev_of_offset(dev
),
43 pdata
->base
= fdtdec_get_addr(gd
->fdt_blob
, dev_of_offset(dev
),
49 static int testfdt_drv_probe(struct udevice
*dev
)
51 struct dm_test_priv
*priv
= dev_get_priv(dev
);
53 priv
->ping_total
+= DM_TEST_START_TOTAL
;
56 * If this device is on a bus, the uclass_flag will be set before
57 * calling this function. This is used by
58 * dm_test_bus_child_pre_probe_uclass().
60 priv
->uclass_total
+= priv
->uclass_flag
;
65 static const struct udevice_id testfdt_ids
[] = {
67 .compatible
= "denx,u-boot-fdt-test",
68 .data
= DM_TEST_TYPE_FIRST
},
70 .compatible
= "google,another-fdt-test",
71 .data
= DM_TEST_TYPE_SECOND
},
75 U_BOOT_DRIVER(testfdt_drv
) = {
76 .name
= "testfdt_drv",
77 .of_match
= testfdt_ids
,
78 .id
= UCLASS_TEST_FDT
,
79 .ofdata_to_platdata
= testfdt_ofdata_to_platdata
,
80 .probe
= testfdt_drv_probe
,
82 .priv_auto_alloc_size
= sizeof(struct dm_test_priv
),
83 .platdata_auto_alloc_size
= sizeof(struct dm_test_pdata
),
86 /* From here is the testfdt uclass code */
87 int testfdt_ping(struct udevice
*dev
, int pingval
, int *pingret
)
89 const struct test_ops
*ops
= device_get_ops(dev
);
94 return ops
->ping(dev
, pingval
, pingret
);
97 UCLASS_DRIVER(testfdt
) = {
99 .id
= UCLASS_TEST_FDT
,
100 .flags
= DM_UC_FLAG_SEQ_ALIAS
,
103 struct dm_testprobe_pdata
{
107 static int testprobe_drv_probe(struct udevice
*dev
)
109 struct dm_testprobe_pdata
*pdata
= dev_get_platdata(dev
);
111 return pdata
->probe_err
;
114 static const struct udevice_id testprobe_ids
[] = {
115 { .compatible
= "denx,u-boot-probe-test" },
119 U_BOOT_DRIVER(testprobe_drv
) = {
120 .name
= "testprobe_drv",
121 .of_match
= testprobe_ids
,
122 .id
= UCLASS_TEST_PROBE
,
123 .probe
= testprobe_drv_probe
,
124 .platdata_auto_alloc_size
= sizeof(struct dm_testprobe_pdata
),
127 UCLASS_DRIVER(testprobe
) = {
129 .id
= UCLASS_TEST_PROBE
,
130 .flags
= DM_UC_FLAG_SEQ_ALIAS
,
133 int dm_check_devices(struct unit_test_state
*uts
, int num_devices
)
140 * Now check that the ping adds are what we expect. This is using the
141 * ping-add property in each node.
143 for (i
= 0; i
< num_devices
; i
++) {
146 ret
= uclass_get_device(UCLASS_TEST_FDT
, i
, &dev
);
150 * Get the 'ping-expect' property, which tells us what the
151 * ping add should be. We don't use the platdata because we
152 * want to test the code that sets that up
153 * (testfdt_drv_probe()).
155 base
= fdtdec_get_addr(gd
->fdt_blob
, dev_of_offset(dev
),
157 debug("dev=%d, base=%d: %s\n", i
, base
,
158 fdt_get_name(gd
->fdt_blob
, dev_of_offset(dev
), NULL
));
160 ut_assert(!dm_check_operations(uts
, dev
, base
,
167 /* Test that FDT-based binding works correctly */
168 static int dm_test_fdt(struct unit_test_state
*uts
)
170 const int num_devices
= 7;
176 ret
= dm_scan_fdt(gd
->fdt_blob
, false);
179 ret
= uclass_get(UCLASS_TEST_FDT
, &uc
);
182 /* These are num_devices compatible root-level device tree nodes */
183 ut_asserteq(num_devices
, list_count_items(&uc
->dev_head
));
185 /* Each should have platform data but no private data */
186 for (i
= 0; i
< num_devices
; i
++) {
187 ret
= uclass_find_device(UCLASS_TEST_FDT
, i
, &dev
);
189 ut_assert(!dev_get_priv(dev
));
190 ut_assert(dev
->platdata
);
193 ut_assertok(dm_check_devices(uts
, num_devices
));
197 DM_TEST(dm_test_fdt
, 0);
199 static int dm_test_fdt_pre_reloc(struct unit_test_state
*uts
)
204 ret
= dm_scan_fdt(gd
->fdt_blob
, true);
207 ret
= uclass_get(UCLASS_TEST_FDT
, &uc
);
210 /* These is only one pre-reloc device */
211 ut_asserteq(1, list_count_items(&uc
->dev_head
));
215 DM_TEST(dm_test_fdt_pre_reloc
, 0);
217 /* Test that sequence numbers are allocated properly */
218 static int dm_test_fdt_uclass_seq(struct unit_test_state
*uts
)
222 /* A few basic santiy tests */
223 ut_assertok(uclass_find_device_by_seq(UCLASS_TEST_FDT
, 3, true, &dev
));
224 ut_asserteq_str("b-test", dev
->name
);
226 ut_assertok(uclass_find_device_by_seq(UCLASS_TEST_FDT
, 8, true, &dev
));
227 ut_asserteq_str("a-test", dev
->name
);
229 ut_asserteq(-ENODEV
, uclass_find_device_by_seq(UCLASS_TEST_FDT
, 5,
231 ut_asserteq_ptr(NULL
, dev
);
234 ut_assertok(uclass_get_device_by_seq(UCLASS_TEST_FDT
, 6, &dev
));
235 ut_asserteq_str("e-test", dev
->name
);
237 ut_asserteq(-ENODEV
, uclass_find_device_by_seq(UCLASS_TEST_FDT
, 7,
241 * Note that c-test nodes are not probed since it is not a top-level
244 ut_assertok(uclass_get_device_by_seq(UCLASS_TEST_FDT
, 3, &dev
));
245 ut_asserteq_str("b-test", dev
->name
);
248 * d-test wants sequence number 3 also, but it can't have it because
249 * b-test gets it first.
251 ut_assertok(uclass_get_device(UCLASS_TEST_FDT
, 2, &dev
));
252 ut_asserteq_str("d-test", dev
->name
);
254 /* d-test actually gets 0 */
255 ut_assertok(uclass_get_device_by_seq(UCLASS_TEST_FDT
, 0, &dev
));
256 ut_asserteq_str("d-test", dev
->name
);
258 /* initially no one wants seq 1 */
259 ut_asserteq(-ENODEV
, uclass_get_device_by_seq(UCLASS_TEST_FDT
, 1,
261 ut_assertok(uclass_get_device(UCLASS_TEST_FDT
, 0, &dev
));
262 ut_assertok(uclass_get_device(UCLASS_TEST_FDT
, 4, &dev
));
264 /* But now that it is probed, we can find it */
265 ut_assertok(uclass_get_device_by_seq(UCLASS_TEST_FDT
, 1, &dev
));
266 ut_asserteq_str("f-test", dev
->name
);
270 DM_TEST(dm_test_fdt_uclass_seq
, DM_TESTF_SCAN_PDATA
| DM_TESTF_SCAN_FDT
);
272 /* Test that we can find a device by device tree offset */
273 static int dm_test_fdt_offset(struct unit_test_state
*uts
)
275 const void *blob
= gd
->fdt_blob
;
279 node
= fdt_path_offset(blob
, "/e-test");
281 ut_assertok(uclass_get_device_by_of_offset(UCLASS_TEST_FDT
, node
,
283 ut_asserteq_str("e-test", dev
->name
);
285 /* This node should not be bound */
286 node
= fdt_path_offset(blob
, "/junk");
288 ut_asserteq(-ENODEV
, uclass_get_device_by_of_offset(UCLASS_TEST_FDT
,
291 /* This is not a top level node so should not be probed */
292 node
= fdt_path_offset(blob
, "/some-bus/c-test@5");
294 ut_asserteq(-ENODEV
, uclass_get_device_by_of_offset(UCLASS_TEST_FDT
,
299 DM_TEST(dm_test_fdt_offset
,
300 DM_TESTF_SCAN_PDATA
| DM_TESTF_SCAN_FDT
| DM_TESTF_FLAT_TREE
);
303 * Test various error conditions with uclass_first_device() and
304 * uclass_next_device()
306 static int dm_test_first_next_device(struct unit_test_state
*uts
)
308 struct dm_testprobe_pdata
*pdata
;
309 struct udevice
*dev
, *parent
= NULL
;
313 /* There should be 4 devices */
314 for (ret
= uclass_first_device(UCLASS_TEST_PROBE
, &dev
), count
= 0;
316 ret
= uclass_next_device(&dev
)) {
318 parent
= dev_get_parent(dev
);
321 ut_asserteq(4, count
);
323 /* Remove them and try again, with an error on the second one */
324 ut_assertok(uclass_get_device(UCLASS_TEST_PROBE
, 1, &dev
));
325 pdata
= dev_get_platdata(dev
);
326 pdata
->probe_err
= -ENOMEM
;
327 device_remove(parent
, DM_REMOVE_NORMAL
);
328 ut_assertok(uclass_first_device(UCLASS_TEST_PROBE
, &dev
));
329 ut_asserteq(-ENOMEM
, uclass_next_device(&dev
));
330 ut_asserteq_ptr(dev
, NULL
);
332 /* Now an error on the first one */
333 ut_assertok(uclass_get_device(UCLASS_TEST_PROBE
, 0, &dev
));
334 pdata
= dev_get_platdata(dev
);
335 pdata
->probe_err
= -ENOENT
;
336 device_remove(parent
, DM_REMOVE_NORMAL
);
337 ut_asserteq(-ENOENT
, uclass_first_device(UCLASS_TEST_PROBE
, &dev
));
341 DM_TEST(dm_test_first_next_device
, DM_TESTF_SCAN_PDATA
| DM_TESTF_SCAN_FDT
);
344 * check_devices() - Check return values and pointers
346 * This runs through a full sequence of uclass_first_device_check()...
347 * uclass_next_device_check() checking that the return values and devices
351 * @devlist: List of expected devices
352 * @mask: Indicates which devices should return an error. Device n should
353 * return error (-NOENT - n) if bit n is set, or no error (i.e. 0) if
356 static int check_devices(struct unit_test_state
*uts
,
357 struct udevice
*devlist
[], int mask
)
363 expected_ret
= (mask
& 1) ? -ENOENT
: 0;
365 ut_asserteq(expected_ret
,
366 uclass_first_device_check(UCLASS_TEST_PROBE
, &dev
));
367 for (i
= 0; i
< 4; i
++) {
368 ut_asserteq_ptr(devlist
[i
], dev
);
369 expected_ret
= (mask
& 1) ? -ENOENT
- (i
+ 1) : 0;
371 ut_asserteq(expected_ret
, uclass_next_device_check(&dev
));
373 ut_asserteq_ptr(NULL
, dev
);
378 /* Test uclass_first_device_check() and uclass_next_device_check() */
379 static int dm_test_first_next_ok_device(struct unit_test_state
*uts
)
381 struct dm_testprobe_pdata
*pdata
;
382 struct udevice
*dev
, *parent
= NULL
, *devlist
[4];
386 /* There should be 4 devices */
388 for (ret
= uclass_first_device_check(UCLASS_TEST_PROBE
, &dev
);
390 ret
= uclass_next_device_check(&dev
)) {
392 devlist
[count
++] = dev
;
393 parent
= dev_get_parent(dev
);
395 ut_asserteq(4, count
);
396 ut_assertok(uclass_first_device_check(UCLASS_TEST_PROBE
, &dev
));
397 ut_assertok(check_devices(uts
, devlist
, 0));
399 /* Remove them and try again, with an error on the second one */
400 pdata
= dev_get_platdata(devlist
[1]);
401 pdata
->probe_err
= -ENOENT
- 1;
402 device_remove(parent
, DM_REMOVE_NORMAL
);
403 ut_assertok(check_devices(uts
, devlist
, 1 << 1));
405 /* Now an error on the first one */
406 pdata
= dev_get_platdata(devlist
[0]);
407 pdata
->probe_err
= -ENOENT
- 0;
408 device_remove(parent
, DM_REMOVE_NORMAL
);
409 ut_assertok(check_devices(uts
, devlist
, 3 << 0));
411 /* Now errors on all */
412 pdata
= dev_get_platdata(devlist
[2]);
413 pdata
->probe_err
= -ENOENT
- 2;
414 pdata
= dev_get_platdata(devlist
[3]);
415 pdata
->probe_err
= -ENOENT
- 3;
416 device_remove(parent
, DM_REMOVE_NORMAL
);
417 ut_assertok(check_devices(uts
, devlist
, 0xf << 0));
421 DM_TEST(dm_test_first_next_ok_device
, DM_TESTF_SCAN_PDATA
| DM_TESTF_SCAN_FDT
);
423 static const struct udevice_id fdt_dummy_ids
[] = {
424 { .compatible
= "denx,u-boot-fdt-dummy", },
428 UCLASS_DRIVER(fdt_dummy
) = {
430 .id
= UCLASS_TEST_DUMMY
,
431 .flags
= DM_UC_FLAG_SEQ_ALIAS
,
434 U_BOOT_DRIVER(fdt_dummy_drv
) = {
435 .name
= "fdt_dummy_drv",
436 .of_match
= fdt_dummy_ids
,
437 .id
= UCLASS_TEST_DUMMY
,
440 static int dm_test_fdt_translation(struct unit_test_state
*uts
)
444 /* Some simple translations */
445 ut_assertok(uclass_find_device_by_seq(UCLASS_TEST_DUMMY
, 0, true, &dev
));
446 ut_asserteq_str("dev@0,0", dev
->name
);
447 ut_asserteq(0x8000, dev_read_addr(dev
));
449 ut_assertok(uclass_find_device_by_seq(UCLASS_TEST_DUMMY
, 1, true, &dev
));
450 ut_asserteq_str("dev@1,100", dev
->name
);
451 ut_asserteq(0x9000, dev_read_addr(dev
));
453 ut_assertok(uclass_find_device_by_seq(UCLASS_TEST_DUMMY
, 2, true, &dev
));
454 ut_asserteq_str("dev@2,200", dev
->name
);
455 ut_asserteq(0xA000, dev_read_addr(dev
));
457 /* No translation for busses with #size-cells == 0 */
458 ut_assertok(uclass_find_device_by_seq(UCLASS_TEST_DUMMY
, 3, true, &dev
));
459 ut_asserteq_str("dev@42", dev
->name
);
460 ut_asserteq(0x42, dev_read_addr(dev
));
464 DM_TEST(dm_test_fdt_translation
, DM_TESTF_SCAN_PDATA
| DM_TESTF_SCAN_FDT
);