2 * Copyright (c) 2013 Google, Inc
4 * SPDX-License-Identifier: GPL-2.0+
15 #include <dm/uclass-internal.h>
19 DECLARE_GLOBAL_DATA_PTR
;
21 static int testfdt_drv_ping(struct udevice
*dev
, int pingval
, int *pingret
)
23 const struct dm_test_pdata
*pdata
= dev
->platdata
;
24 struct dm_test_priv
*priv
= dev_get_priv(dev
);
26 *pingret
= pingval
+ pdata
->ping_add
;
27 priv
->ping_total
+= *pingret
;
32 static const struct test_ops test_ops
= {
33 .ping
= testfdt_drv_ping
,
36 static int testfdt_ofdata_to_platdata(struct udevice
*dev
)
38 struct dm_test_pdata
*pdata
= dev_get_platdata(dev
);
40 pdata
->ping_add
= fdtdec_get_int(gd
->fdt_blob
, dev
->of_offset
,
42 pdata
->base
= fdtdec_get_addr(gd
->fdt_blob
, dev
->of_offset
,
48 static int testfdt_drv_probe(struct udevice
*dev
)
50 struct dm_test_priv
*priv
= dev_get_priv(dev
);
52 priv
->ping_total
+= DM_TEST_START_TOTAL
;
55 * If this device is on a bus, the uclass_flag will be set before
56 * calling this function. This is used by
57 * dm_test_bus_child_pre_probe_uclass().
59 priv
->uclass_total
+= priv
->uclass_flag
;
64 static const struct udevice_id testfdt_ids
[] = {
66 .compatible
= "denx,u-boot-fdt-test",
67 .data
= DM_TEST_TYPE_FIRST
},
69 .compatible
= "google,another-fdt-test",
70 .data
= DM_TEST_TYPE_SECOND
},
74 U_BOOT_DRIVER(testfdt_drv
) = {
75 .name
= "testfdt_drv",
76 .of_match
= testfdt_ids
,
77 .id
= UCLASS_TEST_FDT
,
78 .ofdata_to_platdata
= testfdt_ofdata_to_platdata
,
79 .probe
= testfdt_drv_probe
,
81 .priv_auto_alloc_size
= sizeof(struct dm_test_priv
),
82 .platdata_auto_alloc_size
= sizeof(struct dm_test_pdata
),
85 /* From here is the testfdt uclass code */
86 int testfdt_ping(struct udevice
*dev
, int pingval
, int *pingret
)
88 const struct test_ops
*ops
= device_get_ops(dev
);
93 return ops
->ping(dev
, pingval
, pingret
);
96 UCLASS_DRIVER(testfdt
) = {
98 .id
= UCLASS_TEST_FDT
,
99 .flags
= DM_UC_FLAG_SEQ_ALIAS
,
102 int dm_check_devices(struct unit_test_state
*uts
, int num_devices
)
109 * Now check that the ping adds are what we expect. This is using the
110 * ping-add property in each node.
112 for (i
= 0; i
< num_devices
; i
++) {
115 ret
= uclass_get_device(UCLASS_TEST_FDT
, i
, &dev
);
119 * Get the 'ping-expect' property, which tells us what the
120 * ping add should be. We don't use the platdata because we
121 * want to test the code that sets that up
122 * (testfdt_drv_probe()).
124 base
= fdtdec_get_addr(gd
->fdt_blob
, dev
->of_offset
,
126 debug("dev=%d, base=%d: %s\n", i
, base
,
127 fdt_get_name(gd
->fdt_blob
, dev
->of_offset
, NULL
));
129 ut_assert(!dm_check_operations(uts
, dev
, base
,
136 /* Test that FDT-based binding works correctly */
137 static int dm_test_fdt(struct unit_test_state
*uts
)
139 const int num_devices
= 6;
145 ret
= dm_scan_fdt(gd
->fdt_blob
, false);
148 ret
= uclass_get(UCLASS_TEST_FDT
, &uc
);
151 /* These are num_devices compatible root-level device tree nodes */
152 ut_asserteq(num_devices
, list_count_items(&uc
->dev_head
));
154 /* Each should have platform data but no private data */
155 for (i
= 0; i
< num_devices
; i
++) {
156 ret
= uclass_find_device(UCLASS_TEST_FDT
, i
, &dev
);
158 ut_assert(!dev_get_priv(dev
));
159 ut_assert(dev
->platdata
);
162 ut_assertok(dm_check_devices(uts
, num_devices
));
166 DM_TEST(dm_test_fdt
, 0);
168 static int dm_test_fdt_pre_reloc(struct unit_test_state
*uts
)
173 ret
= dm_scan_fdt(gd
->fdt_blob
, true);
176 ret
= uclass_get(UCLASS_TEST_FDT
, &uc
);
179 /* These is only one pre-reloc device */
180 ut_asserteq(1, list_count_items(&uc
->dev_head
));
184 DM_TEST(dm_test_fdt_pre_reloc
, 0);
186 /* Test that sequence numbers are allocated properly */
187 static int dm_test_fdt_uclass_seq(struct unit_test_state
*uts
)
191 /* A few basic santiy tests */
192 ut_assertok(uclass_find_device_by_seq(UCLASS_TEST_FDT
, 3, true, &dev
));
193 ut_asserteq_str("b-test", dev
->name
);
195 ut_assertok(uclass_find_device_by_seq(UCLASS_TEST_FDT
, 8, true, &dev
));
196 ut_asserteq_str("a-test", dev
->name
);
198 ut_asserteq(-ENODEV
, uclass_find_device_by_seq(UCLASS_TEST_FDT
, 5,
200 ut_asserteq_ptr(NULL
, dev
);
203 ut_assertok(uclass_get_device_by_seq(UCLASS_TEST_FDT
, 6, &dev
));
204 ut_asserteq_str("e-test", dev
->name
);
206 ut_asserteq(-ENODEV
, uclass_find_device_by_seq(UCLASS_TEST_FDT
, 7,
210 * Note that c-test nodes are not probed since it is not a top-level
213 ut_assertok(uclass_get_device_by_seq(UCLASS_TEST_FDT
, 3, &dev
));
214 ut_asserteq_str("b-test", dev
->name
);
217 * d-test wants sequence number 3 also, but it can't have it because
218 * b-test gets it first.
220 ut_assertok(uclass_get_device(UCLASS_TEST_FDT
, 2, &dev
));
221 ut_asserteq_str("d-test", dev
->name
);
223 /* d-test actually gets 0 */
224 ut_assertok(uclass_get_device_by_seq(UCLASS_TEST_FDT
, 0, &dev
));
225 ut_asserteq_str("d-test", dev
->name
);
227 /* initially no one wants seq 1 */
228 ut_asserteq(-ENODEV
, uclass_get_device_by_seq(UCLASS_TEST_FDT
, 1,
230 ut_assertok(uclass_get_device(UCLASS_TEST_FDT
, 0, &dev
));
231 ut_assertok(uclass_get_device(UCLASS_TEST_FDT
, 4, &dev
));
233 /* But now that it is probed, we can find it */
234 ut_assertok(uclass_get_device_by_seq(UCLASS_TEST_FDT
, 1, &dev
));
235 ut_asserteq_str("f-test", dev
->name
);
239 DM_TEST(dm_test_fdt_uclass_seq
, DM_TESTF_SCAN_PDATA
| DM_TESTF_SCAN_FDT
);
241 /* Test that we can find a device by device tree offset */
242 static int dm_test_fdt_offset(struct unit_test_state
*uts
)
244 const void *blob
= gd
->fdt_blob
;
248 node
= fdt_path_offset(blob
, "/e-test");
250 ut_assertok(uclass_get_device_by_of_offset(UCLASS_TEST_FDT
, node
,
252 ut_asserteq_str("e-test", dev
->name
);
254 /* This node should not be bound */
255 node
= fdt_path_offset(blob
, "/junk");
257 ut_asserteq(-ENODEV
, uclass_get_device_by_of_offset(UCLASS_TEST_FDT
,
260 /* This is not a top level node so should not be probed */
261 node
= fdt_path_offset(blob
, "/some-bus/c-test@5");
263 ut_asserteq(-ENODEV
, uclass_get_device_by_of_offset(UCLASS_TEST_FDT
,
268 DM_TEST(dm_test_fdt_offset
, DM_TESTF_SCAN_PDATA
| DM_TESTF_SCAN_FDT
);