Loading...
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74 75 76 77 78 79 80 81 82 83 84 85 86 87 88 89 90 91 92 93 94 95 96 97 98 99 100 101 102 103 104 105 106 107 108 109 110 111 112 113 114 115 116 117 118 119 120 121 122 | // SPDX-License-Identifier: GPL-2.0+ /* * Copyright (C) 2018, Bin Meng <bmeng.cn@gmail.com> */ #include <common.h> #include <dm.h> #include <virtio_types.h> #include <virtio.h> #include <virtio_ring.h> #include <dm/device-internal.h> #include <dm/uclass-internal.h> #include <dm/root.h> #include <dm/test.h> #include <test/ut.h> /* Basic test of the virtio uclass */ static int dm_test_virtio_base(struct unit_test_state *uts) { struct udevice *bus, *dev; u8 status; /* check probe success */ ut_assertok(uclass_first_device(UCLASS_VIRTIO, &bus)); /* check the child virtio-blk device is bound */ ut_assertok(device_find_first_child(bus, &dev)); ut_assertok(strcmp(dev->name, "virtio-blk#0")); /* check driver status */ ut_assertok(virtio_get_status(dev, &status)); ut_asserteq(VIRTIO_CONFIG_S_ACKNOWLEDGE, status); return 0; } DM_TEST(dm_test_virtio_base, DM_TESTF_SCAN_PDATA | DM_TESTF_SCAN_FDT); /* Test all of the virtio uclass ops */ static int dm_test_virtio_all_ops(struct unit_test_state *uts) { struct udevice *bus, *dev; struct virtio_dev_priv *uc_priv; uint offset = 0, len = 0, nvqs = 1; void *buffer = NULL; u8 status; u32 counter; u64 features; struct virtqueue *vqs[2]; /* check probe success */ ut_assertok(uclass_first_device(UCLASS_VIRTIO, &bus)); /* check the child virtio-blk device is bound */ ut_assertok(device_find_first_child(bus, &dev)); /* * fake the virtio device probe by filling in uc_priv->vdev * which is used by virtio_find_vqs/virtio_del_vqs. */ uc_priv = dev_get_uclass_priv(bus); uc_priv->vdev = dev; /* test virtio_xxx APIs */ ut_assertok(virtio_get_config(dev, offset, buffer, len)); ut_assertok(virtio_set_config(dev, offset, buffer, len)); ut_asserteq(-ENOSYS, virtio_generation(dev, &counter)); ut_assertok(virtio_set_status(dev, VIRTIO_CONFIG_S_DRIVER_OK)); ut_assertok(virtio_get_status(dev, &status)); ut_asserteq(VIRTIO_CONFIG_S_DRIVER_OK, status); ut_assertok(virtio_reset(dev)); ut_assertok(virtio_get_status(dev, &status)); ut_asserteq(0, status); ut_assertok(virtio_get_features(dev, &features)); ut_asserteq(VIRTIO_F_VERSION_1, features); ut_assertok(virtio_set_features(dev)); ut_assertok(virtio_find_vqs(dev, nvqs, vqs)); ut_assertok(virtio_del_vqs(dev)); ut_assertok(virtio_notify(dev, vqs[0])); return 0; } DM_TEST(dm_test_virtio_all_ops, DM_TESTF_SCAN_PDATA | DM_TESTF_SCAN_FDT); /* Test of the virtio driver that does not have required driver ops */ static int dm_test_virtio_missing_ops(struct unit_test_state *uts) { struct udevice *bus; /* find the virtio device */ ut_assertok(uclass_find_device(UCLASS_VIRTIO, 1, &bus)); /* * Probe the device should fail with error -ENOENT. * See ops check in virtio_uclass_pre_probe(). */ ut_asserteq(-ENOENT, device_probe(bus)); return 0; } DM_TEST(dm_test_virtio_missing_ops, DM_TESTF_SCAN_PDATA | DM_TESTF_SCAN_FDT); /* Test removal of virtio device driver */ static int dm_test_virtio_remove(struct unit_test_state *uts) { struct udevice *bus, *dev; /* check probe success */ ut_assertok(uclass_first_device(UCLASS_VIRTIO, &bus)); /* check the child virtio-blk device is bound */ ut_assertok(device_find_first_child(bus, &dev)); /* set driver status to VIRTIO_CONFIG_S_DRIVER_OK */ ut_assertok(virtio_set_status(dev, VIRTIO_CONFIG_S_DRIVER_OK)); /* check the device can be successfully removed */ dev->flags |= DM_FLAG_ACTIVATED; ut_assertok(device_remove(bus, DM_REMOVE_ACTIVE_ALL)); return 0; } DM_TEST(dm_test_virtio_remove, DM_TESTF_SCAN_PDATA | DM_TESTF_SCAN_FDT); |