From: Long Li <lon...@microsoft.com>

Implement VMBUS cleanup callback from eal_cleanup().

Signed-off-by: Long Li <lon...@microsoft.com>
---
 drivers/bus/vmbus/vmbus_common.c | 29 +++++++++++++++++++++++++++++
 1 file changed, 29 insertions(+)

diff --git a/drivers/bus/vmbus/vmbus_common.c b/drivers/bus/vmbus/vmbus_common.c
index 8d32d66504..a6277bfc06 100644
--- a/drivers/bus/vmbus/vmbus_common.c
+++ b/drivers/bus/vmbus/vmbus_common.c
@@ -188,6 +188,34 @@ rte_vmbus_probe(void)
        return (probed && probed == failed) ? -1 : 0;
 }
 
+static int
+rte_vmbus_cleanup(void)
+{
+       struct rte_vmbus_device *dev, *tmp_dev;
+       int error = 0;
+
+       RTE_TAILQ_FOREACH_SAFE(dev, &rte_vmbus_bus.device_list, next, tmp_dev) {
+
+               const struct rte_vmbus_driver *drv = dev->driver;
+               int ret;
+
+               if (!drv || !drv->remove)
+                       continue;
+
+               ret = drv->remove(dev);
+               if (ret < 0)
+                       error = -1;
+
+               rte_vmbus_unmap_device(dev);
+
+               dev->driver = NULL;
+               dev->device.driver = NULL;
+               free(dev);
+       }
+
+       return error;
+}
+
 static int
 vmbus_parse(const char *name, void *addr)
 {
@@ -285,6 +313,7 @@ struct rte_vmbus_bus rte_vmbus_bus = {
        .bus = {
                .scan = rte_vmbus_scan,
                .probe = rte_vmbus_probe,
+               .cleanup = rte_vmbus_cleanup,
                .find_device = vmbus_find_device,
                .parse = vmbus_parse,
        },
-- 
2.32.0

Reply via email to