tools: dumpimage: Clarify help
[oweals/u-boot.git] / test / dm / blk.c
1 // SPDX-License-Identifier: GPL-2.0+
2 /*
3  * Copyright (C) 2015 Google, Inc
4  */
5
6 #include <common.h>
7 #include <dm.h>
8 #include <usb.h>
9 #include <asm/state.h>
10 #include <dm/test.h>
11 #include <test/ut.h>
12
13 DECLARE_GLOBAL_DATA_PTR;
14
15 /* Test that block devices can be created */
16 static int dm_test_blk_base(struct unit_test_state *uts)
17 {
18         struct udevice *blk1, *blk3, *dev;
19
20         /* Make sure there are no block devices */
21         ut_asserteq(-ENODEV, uclass_get_device_by_seq(UCLASS_BLK, 0, &dev));
22
23         /* Create two, one the parent of the other */
24         ut_assertok(blk_create_device(gd->dm_root, "sandbox_host_blk", "test",
25                                       IF_TYPE_HOST, 1, 512, 2, &blk1));
26         ut_assertok(blk_create_device(blk1, "sandbox_host_blk", "test",
27                                       IF_TYPE_HOST, 3, 512, 2, &blk3));
28
29         /* Check we can find them */
30         ut_asserteq(-ENODEV, blk_get_device(IF_TYPE_HOST, 0, &dev));
31         ut_assertok(blk_get_device(IF_TYPE_HOST, 1, &dev));
32         ut_asserteq_ptr(blk1, dev);
33         ut_assertok(blk_get_device(IF_TYPE_HOST, 3, &dev));
34         ut_asserteq_ptr(blk3, dev);
35
36         /* Check we can iterate */
37         ut_assertok(blk_first_device(IF_TYPE_HOST, &dev));
38         ut_asserteq_ptr(blk1, dev);
39         ut_assertok(blk_next_device(&dev));
40         ut_asserteq_ptr(blk3, dev);
41
42         return 0;
43 }
44 DM_TEST(dm_test_blk_base, DM_TESTF_SCAN_PDATA | DM_TESTF_SCAN_FDT);
45
46 static int count_blk_devices(void)
47 {
48         struct udevice *blk;
49         struct uclass *uc;
50         int count = 0;
51         int ret;
52
53         ret = uclass_get(UCLASS_BLK, &uc);
54         if (ret)
55                 return ret;
56
57         uclass_foreach_dev(blk, uc)
58                 count++;
59
60         return count;
61 }
62
63 /* Test that block devices work correctly with USB */
64 static int dm_test_blk_usb(struct unit_test_state *uts)
65 {
66         struct udevice *usb_dev, *dev;
67         struct blk_desc *dev_desc;
68
69         /* Get a flash device */
70         state_set_skip_delays(true);
71         ut_assertok(usb_init());
72         ut_assertok(uclass_get_device(UCLASS_MASS_STORAGE, 0, &usb_dev));
73         ut_assertok(blk_get_device_by_str("usb", "0", &dev_desc));
74
75         /* The parent should be a block device */
76         ut_assertok(blk_get_device(IF_TYPE_USB, 0, &dev));
77         ut_asserteq_ptr(usb_dev, dev_get_parent(dev));
78
79         /* Check we have one block device for each mass storage device */
80         ut_asserteq(6, count_blk_devices());
81
82         /* Now go around again, making sure the old devices were unbound */
83         ut_assertok(usb_stop());
84         ut_assertok(usb_init());
85         ut_asserteq(6, count_blk_devices());
86         ut_assertok(usb_stop());
87
88         return 0;
89 }
90 DM_TEST(dm_test_blk_usb, DM_TESTF_SCAN_PDATA | DM_TESTF_SCAN_FDT);
91
92 /* Test that we can find block devices without probing them */
93 static int dm_test_blk_find(struct unit_test_state *uts)
94 {
95         struct udevice *blk, *dev;
96
97         ut_assertok(blk_create_device(gd->dm_root, "sandbox_host_blk", "test",
98                                       IF_TYPE_HOST, 1, 512, 2, &blk));
99         ut_asserteq(-ENODEV, blk_find_device(IF_TYPE_HOST, 0, &dev));
100         ut_assertok(blk_find_device(IF_TYPE_HOST, 1, &dev));
101         ut_asserteq_ptr(blk, dev);
102         ut_asserteq(false, device_active(dev));
103
104         /* Now activate it */
105         ut_assertok(blk_get_device(IF_TYPE_HOST, 1, &dev));
106         ut_asserteq_ptr(blk, dev);
107         ut_asserteq(true, device_active(dev));
108
109         return 0;
110 }
111 DM_TEST(dm_test_blk_find, DM_TESTF_SCAN_PDATA | DM_TESTF_SCAN_FDT);
112
113 /* Test that block device numbering works as expected */
114 static int dm_test_blk_devnum(struct unit_test_state *uts)
115 {
116         struct udevice *dev, *mmc_dev, *parent;
117         int i;
118
119         /*
120          * Probe the devices, with the first one being probed last. This is the
121          * one with no alias / sequence numnber.
122          */
123         ut_assertok(uclass_get_device(UCLASS_MMC, 1, &dev));
124         ut_assertok(uclass_get_device(UCLASS_MMC, 2, &dev));
125         ut_assertok(uclass_get_device(UCLASS_MMC, 0, &dev));
126         for (i = 0; i < 3; i++) {
127                 struct blk_desc *desc;
128
129                 /* Check that the bblock device is attached */
130                 ut_assertok(uclass_get_device_by_seq(UCLASS_MMC, i, &mmc_dev));
131                 ut_assertok(blk_find_device(IF_TYPE_MMC, i, &dev));
132                 parent = dev_get_parent(dev);
133                 ut_asserteq_ptr(parent, mmc_dev);
134                 ut_asserteq(trailing_strtol(mmc_dev->name), i);
135
136                 /*
137                  * Check that the block device devnum matches its parent's
138                  * sequence number
139                  */
140                 desc = dev_get_uclass_platdata(dev);
141                 ut_asserteq(desc->devnum, i);
142         }
143
144         return 0;
145 }
146 DM_TEST(dm_test_blk_devnum, DM_TESTF_SCAN_PDATA | DM_TESTF_SCAN_FDT);
147
148 /* Test that we can get a block from its parent */
149 static int dm_test_blk_get_from_parent(struct unit_test_state *uts)
150 {
151         struct udevice *dev, *blk;
152
153         ut_assertok(uclass_get_device(UCLASS_MMC, 0, &dev));
154         ut_assertok(blk_get_from_parent(dev, &blk));
155
156         ut_assertok(uclass_get_device(UCLASS_I2C, 0, &dev));
157         ut_asserteq(-ENOTBLK, blk_get_from_parent(dev, &blk));
158
159         ut_assertok(uclass_get_device(UCLASS_GPIO, 0, &dev));
160         ut_asserteq(-ENODEV, blk_get_from_parent(dev, &blk));
161
162         return 0;
163 }
164 DM_TEST(dm_test_blk_get_from_parent, DM_TESTF_SCAN_PDATA | DM_TESTF_SCAN_FDT);