2 * Copyright (C) 2015 Google, Inc
4 * SPDX-License-Identifier: GPL-2.0+
12 #include <asm/state.h>
14 #include <dm/device-internal.h>
16 #include <dm/uclass-internal.h>
19 DECLARE_GLOBAL_DATA_PTR;
21 /* Test that sandbox USB works correctly */
22 static int dm_test_usb_base(struct unit_test_state *uts)
26 ut_asserteq(-ENODEV, uclass_get_device_by_seq(UCLASS_USB, 0, &bus));
27 ut_assertok(uclass_get_device(UCLASS_USB, 0, &bus));
28 ut_asserteq(-ENODEV, uclass_get_device_by_seq(UCLASS_USB, 2, &bus));
32 DM_TEST(dm_test_usb_base, DM_TESTF_SCAN_PDATA | DM_TESTF_SCAN_FDT);
35 * Test that we can use the flash stick. This is more of a functional test. It
36 * covers scanning the bug, setting up a hub and a flash stick and reading
37 * data from the flash stick.
39 static int dm_test_usb_flash(struct unit_test_state *uts)
42 block_dev_desc_t *dev_desc;
45 state_set_skip_delays(true);
46 ut_assertok(usb_init());
47 ut_assertok(uclass_get_device(UCLASS_MASS_STORAGE, 0, &dev));
48 ut_assertok(get_device("usb", "0", &dev_desc));
50 /* Read a few blocks and look for the string we expect */
51 ut_asserteq(512, dev_desc->blksz);
52 memset(cmp, '\0', sizeof(cmp));
53 ut_asserteq(2, dev_desc->block_read(dev_desc->dev, 0, 2, cmp));
54 ut_assertok(strcmp(cmp, "this is a test"));
58 DM_TEST(dm_test_usb_flash, DM_TESTF_SCAN_PDATA | DM_TESTF_SCAN_FDT);
60 /* test that we can handle multiple storage devices */
61 static int dm_test_usb_multi(struct unit_test_state *uts)
65 state_set_skip_delays(true);
66 ut_assertok(usb_init());
67 ut_assertok(uclass_get_device(UCLASS_MASS_STORAGE, 0, &dev));
68 ut_assertok(uclass_get_device(UCLASS_MASS_STORAGE, 1, &dev));
69 ut_assertok(uclass_get_device(UCLASS_MASS_STORAGE, 2, &dev));
73 DM_TEST(dm_test_usb_multi, DM_TESTF_SCAN_PDATA | DM_TESTF_SCAN_FDT);
75 static int count_usb_devices(void)
82 ret = uclass_get(UCLASS_USB_HUB, &uc);
86 uclass_foreach_dev(hub, uc) {
90 for (device_find_first_child(hub, &dev);
92 device_find_next_child(&dev)) {
100 /* test that we can remove an emulated device and it is then not found */
101 static int dm_test_usb_remove(struct unit_test_state *uts)
103 struct udevice *dev, *emul;
105 /* Scan and check that all devices are present */
106 state_set_skip_delays(true);
107 ut_assertok(usb_init());
108 ut_assertok(uclass_get_device(UCLASS_MASS_STORAGE, 0, &dev));
109 ut_assertok(uclass_get_device(UCLASS_MASS_STORAGE, 1, &dev));
110 ut_assertok(uclass_get_device(UCLASS_MASS_STORAGE, 2, &dev));
111 ut_asserteq(5, count_usb_devices());
112 ut_assertok(usb_stop());
113 ut_asserteq(5, count_usb_devices());
115 /* Remove the second emulation device */
116 ut_assertok(uclass_find_device_by_name(UCLASS_USB_EMUL, "flash-stick@1",
118 ut_assertok(device_unbind(dev));
120 /* Rescan - only the first and third should be present */
121 ut_assertok(usb_init());
122 ut_assertok(uclass_get_device(UCLASS_MASS_STORAGE, 0, &dev));
123 ut_assertok(usb_emul_find_for_dev(dev, &emul));
124 ut_asserteq_str("flash-stick@0", emul->name);
125 ut_assertok(uclass_get_device(UCLASS_MASS_STORAGE, 1, &dev));
126 ut_assertok(usb_emul_find_for_dev(dev, &emul));
127 ut_asserteq_str("flash-stick@2", emul->name);
129 ut_asserteq(-ENODEV, uclass_get_device(UCLASS_MASS_STORAGE, 2, &dev));
131 ut_asserteq(4, count_usb_devices());
132 ut_assertok(usb_stop());
133 ut_asserteq(4, count_usb_devices());
137 DM_TEST(dm_test_usb_remove, DM_TESTF_SCAN_PDATA | DM_TESTF_SCAN_FDT);
139 const char usb_tree_base[] =
140 " 1 Hub (12 Mb/s, 100mA)\n"
141 " | sandbox hub 2345\n"
143 " |\b+-2 Mass Storage (12 Mb/s, 100mA)\n"
144 " | sandbox flash flash-stick@0\n"
146 " |\b+-3 Mass Storage (12 Mb/s, 100mA)\n"
147 " | sandbox flash flash-stick@1\n"
149 " |\b+-4 Mass Storage (12 Mb/s, 100mA)\n"
150 " sandbox flash flash-stick@2\n"
153 /* test that the 'usb tree' command output looks correct */
154 static int dm_test_usb_tree(struct unit_test_state *uts)
159 state_set_skip_delays(true);
160 ut_assertok(usb_init());
161 console_record_reset_enable();
163 len = membuff_getraw(&gd->console_out, -1, true, &data);
166 ut_asserteq_str(usb_tree_base, data);
167 ut_assertok(usb_stop());
171 DM_TEST(dm_test_usb_tree, DM_TESTF_SCAN_PDATA | DM_TESTF_SCAN_FDT);
173 const char usb_tree_remove[] =
174 " 1 Hub (12 Mb/s, 100mA)\n"
175 " | sandbox hub 2345\n"
177 " |\b+-2 Mass Storage (12 Mb/s, 100mA)\n"
178 " | sandbox flash flash-stick@0\n"
180 " |\b+-3 Mass Storage (12 Mb/s, 100mA)\n"
181 " sandbox flash flash-stick@2\n"
185 * test that the 'usb tree' command output looks correct when we remove a
188 static int dm_test_usb_tree_remove(struct unit_test_state *uts)
194 /* Remove the second emulation device */
195 ut_assertok(uclass_find_device_by_name(UCLASS_USB_EMUL, "flash-stick@1",
197 ut_assertok(device_unbind(dev));
199 state_set_skip_delays(true);
200 ut_assertok(usb_init());
201 console_record_reset_enable();
203 len = membuff_getraw(&gd->console_out, -1, true, &data);
206 ut_asserteq_str(usb_tree_remove, data);
207 ut_assertok(usb_stop());
211 DM_TEST(dm_test_usb_tree_remove, DM_TESTF_SCAN_PDATA | DM_TESTF_SCAN_FDT);
213 const char usb_tree_reorder[] =
214 " 1 Hub (12 Mb/s, 100mA)\n"
215 " | sandbox hub 2345\n"
217 " |\b+-2 Mass Storage (12 Mb/s, 100mA)\n"
218 " | sandbox flash flash-stick@0\n"
220 " |\b+-3 Mass Storage (12 Mb/s, 100mA)\n"
221 " | sandbox flash flash-stick@2\n"
223 " |\b+-4 Mass Storage (12 Mb/s, 100mA)\n"
224 " sandbox flash flash-stick@1\n"
228 * test that the 'usb tree' command output looks correct when we reorder two
231 static int dm_test_usb_tree_reorder(struct unit_test_state *uts)
233 struct udevice *dev, *parent;
237 /* Remove the second emulation device */
238 ut_assertok(uclass_find_device_by_name(UCLASS_USB_EMUL, "flash-stick@1",
240 parent = dev->parent;
242 /* Reorder the devices in the parent list and uclass list */
243 list_del(&dev->sibling_node);
244 list_add_tail(&dev->sibling_node, &parent->child_head);
246 list_del(&dev->uclass_node);
247 list_add_tail(&dev->uclass_node, &dev->uclass->dev_head);
249 state_set_skip_delays(true);
250 ut_assertok(usb_init());
251 console_record_reset_enable();
253 len = membuff_getraw(&gd->console_out, -1, true, &data);
256 ut_asserteq_str(usb_tree_reorder, data);
257 ut_assertok(usb_stop());
261 DM_TEST(dm_test_usb_tree_reorder, DM_TESTF_SCAN_PDATA | DM_TESTF_SCAN_FDT);
263 static int dm_test_usb_keyb(struct unit_test_state *uts)
267 state_set_skip_delays(true);
268 ut_assertok(usb_init());
270 /* Initially there should be no characters */
271 ut_asserteq(0, tstc());
273 ut_assertok(uclass_get_device_by_name(UCLASS_USB_EMUL, "keyb",
277 * Add a string to the USB keyboard buffer - it should appear in
280 ut_assertok(sandbox_usb_keyb_add_string(dev, "ab"));
281 ut_asserteq(1, tstc());
282 ut_asserteq('a', getc());
283 ut_asserteq(1, tstc());
284 ut_asserteq('b', getc());
285 ut_asserteq(0, tstc());
287 ut_assertok(usb_stop());
291 DM_TEST(dm_test_usb_keyb, DM_TESTF_SCAN_PDATA | DM_TESTF_SCAN_FDT);