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