[libvirt PATCH v5 13/30] nodedev: handle mdevs that disappear from mdevctl

[Jonathon Jongsma]

mdevctl does not currently provide any events when the list of defined
devices changes, so we will need to poll mdevctl for the list of defined
devices periodically. When a mediated device no longer exists from one
iteration to the next, we need to treat it as an "undefine" event.

When we get such an event, we remove the device from the list if it's
not active. Otherwise, we simply mark it as non-persistent.

Signed-off-by: Jonathon Jongsma <jjongsma at redhat.com>
---
 src/node_device/node_device_driver.c | 67 ++++++++++++++++++++++++++--
 1 file changed, 64 insertions(+), 3 deletions(-)

diff --git a/src/node_device/node_device_driver.c b/src/node_device/node_device_driver.c
index ad21565adb..318ad43b9f 100644
--- a/src/node_device/node_device_driver.c
+++ b/src/node_device/node_device_driver.c
@@ -1183,20 +1183,81 @@ virMdevctlListDefined(virNodeDeviceDefPtr **devs)
 }
 
 
+typedef struct _virMdevctlForEachData virMdevctlForEachData;
+struct _virMdevctlForEachData {
+    int ndefs;
+    virNodeDeviceDefPtr *defs;
+};
+
+
+/* This function keeps the list of persistent mediated devices consistent
+ * between the nodedev driver and mdevctl.
+ * @obj is a device that is currently known by the nodedev driver, and @opaque
+ * contains the most recent list of devices defined by mdevctl. If @obj is no
+ * longer defined in mdevctl, mark it as undefined and possibly remove it from
+ * the driver as well. Returning 'true' from this function indicates that the
+ * device should be removed from the nodedev driver list. */
+static bool
+removeMissingPersistentMdevs(virNodeDeviceObjPtr obj,
+                             const void *opaque)
+{
+    bool remove = false;
+    const virMdevctlForEachData *data = opaque;
+    size_t i;
+    virNodeDeviceDefPtr def = virNodeDeviceObjGetDef(obj);
+    virObjectEventPtr event;
+
+    if (def->caps->data.type != VIR_NODE_DEV_CAP_MDEV)
+        return false;
+
+    /* transient mdevs are populated via udev, so don't remove them from the
+     * nodedev driver just because they are not reported by by mdevctl */
+    if (!virNodeDeviceObjIsPersistent(obj))
+        return false;
+
+    for (i = 0; i < data->ndefs; i++) {
+        /* OK, this mdev is still defined by mdevctl */
+        if (STREQ(data->defs[i]->name, def->name))
+            return false;
+    }
+
+    event = virNodeDeviceEventLifecycleNew(def->name,
+                                           VIR_NODE_DEVICE_EVENT_UNDEFINED,
+                                           0);
+
+    /* The device is active, but no longer defined by mdevctl. Keep the device
+     * in the list, but mark it as non-persistent */
+    if (virNodeDeviceObjIsActive(obj))
+        virNodeDeviceObjSetPersistent(obj, false);
+    else
+        remove = true;
+
+    virObjectEventStateQueue(driver->nodeDeviceEventState, event);
+
+    return remove;
+}
+
+
 int
 nodeDeviceUpdateMediatedDevices(void)
 {
     g_autofree virNodeDeviceDefPtr *defs = NULL;
-    int ndefs;
+    virMdevctlForEachData data = { 0, };
     size_t i;
 
-    if ((ndefs = virMdevctlListDefined(&defs)) < 0) {
+    if ((data.ndefs = virMdevctlListDefined(&defs)) < 0) {
         virReportSystemError(errno, "%s",
                              _("failed to query mdevs from mdevctl"));
         return -1;
     }
 
-    for (i = 0; i < ndefs; i++) {
+    /* Any mdevs that were previously defined but were not returned in the
+     * latest mdevctl query should be removed from the device list */
+    data.defs = defs;
+    virNodeDeviceObjListForEachRemove(driver->devs,
+                                      removeMissingPersistentMdevs, &data);
+
+    for (i = 0; i < data.ndefs; i++) {
         virNodeDeviceObjPtr obj;
         virObjectEventPtr event;
         g_autoptr(virNodeDeviceDef) def = defs[i];
-- 
2.26.2