improve interface for cleaning up unused devices

1 files changed, 15 insertions, 7 deletions

diff --git a/device.c b/device.c
index c02335c..e206a76 100644
--- a/device.c
+++ b/device.c
@@ -278,15 +278,23 @@ void device_remove_user(struct device_user *dep)
 	dep->dev = NULL;
 }
 
+static void
+__device_free_unused(struct device *dev)
+{
+	if (!list_empty(&dev->users))
+		return;
+
+	device_free(dev);
+}
+
 void
-device_free_all(void)
+device_free_unused(struct device *dev)
 {
-	struct device *dev, *tmp;
+	struct device *tmp;
 
-	avl_for_each_element_safe(&devices, dev, avl, tmp) {
-		if (!list_empty(&dev->users))
-			continue;
+	if (dev)
+		return __device_free_unused(dev);
 
-		device_free(dev);
-	}
+	avl_for_each_element_safe(&devices, dev, avl, tmp)
+		__device_free_unused(dev);
 }