From 15a71e60592afbb9200d1b9c3adcffb532ba7890 Mon Sep 17 00:00:00 2001
From: Tang Chen <tangchen@cn.fujitsu.com>
Date: Wed, 22 Aug 2012 12:10:24 +0800
Subject: [PATCH] Introduce virNetlinkEventServiceStopAll() to stop all netlink
 services.

This patch introduce virNetlinkEventServiceStopAll() to stop
all the monitors to receive netlink messages for libvirtd.

Signed-off-by: Tang Chen <tangchen@cn.fujitsu.com>
---
 daemon/libvirtd.c        |  2 +-
 src/libvirt_private.syms |  1 +
 src/util/virnetlink.c    | 50 ++++++++++++++++++++++++++++++++++++++++
 src/util/virnetlink.h    |  5 ++++
 4 files changed, 57 insertions(+), 1 deletion(-)

diff --git a/daemon/libvirtd.c b/daemon/libvirtd.c
index 96c8dddf2c..c7f8d3b740 100644
--- a/daemon/libvirtd.c
+++ b/daemon/libvirtd.c
@@ -1327,7 +1327,7 @@ int main(int argc, char **argv) {
                 0, "shutdown", NULL, NULL);
 
 cleanup:
-    virNetlinkEventServiceStop(NETLINK_ROUTE);
+    virNetlinkEventServiceStopAll();
     virObjectUnref(remoteProgram);
     virObjectUnref(qemuProgram);
     virNetServerClose(srv);
diff --git a/src/libvirt_private.syms b/src/libvirt_private.syms
index 8962de2ed9..d91f492cc3 100644
--- a/src/libvirt_private.syms
+++ b/src/libvirt_private.syms
@@ -1483,6 +1483,7 @@ virNetlinkEventRemoveClient;
 virNetlinkEventServiceIsRunning;
 virNetlinkEventServiceLocalPid;
 virNetlinkEventServiceStop;
+virNetlinkEventServiceStopAll;
 virNetlinkEventServiceStart;
 virNetlinkShutdown;
 virNetlinkStartup;
diff --git a/src/util/virnetlink.c b/src/util/virnetlink.c
index 19619f5edc..2c5e9f315b 100644
--- a/src/util/virnetlink.c
+++ b/src/util/virnetlink.c
@@ -404,6 +404,46 @@ virNetlinkEventServiceStop(unsigned int protocol)
     return 0;
 }
 
+/**
+ * virNetlinkEventServiceStopAll:
+ *
+ * Stop all the monitors to receive netlink messages for libvirtd.
+ *
+ * Returns -1 if any monitor cannot be unregistered, 0 upon success
+ */
+int
+virNetlinkEventServiceStopAll(void)
+{
+    unsigned int i, j;
+    virNetlinkEventSrvPrivatePtr srv = NULL;
+
+    VIR_INFO("stopping all netlink event services");
+
+    for (i = 0; i < MAX_LINKS; i++) {
+        srv = server[i];
+        if (!srv)
+            continue;
+
+        virNetlinkEventServerLock(srv);
+        nl_close(srv->netlinknh);
+        virNetlinkFree(srv->netlinknh);
+        virEventRemoveHandle(srv->eventwatch);
+
+        for (j = 0; j < srv->handlesCount; j++) {
+            if (srv->handles[j].deleted == VIR_NETLINK_HANDLE_VALID)
+                virNetlinkEventRemoveClientPrimitive(j, i);
+        }
+
+        server[i] = NULL;
+        virNetlinkEventServerUnlock(srv);
+
+        virMutexDestroy(&srv->lock);
+        VIR_FREE(srv);
+    }
+
+    return 0;
+}
+
 /**
  * virNetlinkEventServiceIsRunning:
  *
@@ -731,6 +771,16 @@ int virNetlinkEventServiceStop(unsigned int protocol ATTRIBUTE_UNUSED)
     return 0;
 }
 
+/**
+ * stopNetlinkEventServerAll: stop all the monitors to receive netlink
+ * messages for libvirtd
+ */
+int virNetlinkEventServiceStopAll(void)
+{
+    VIR_DEBUG("%s", _(unsupported));
+    return 0;
+}
+
 /**
  * startNetlinkEventServer: start a monitor to receive netlink
  * messages for libvirtd
diff --git a/src/util/virnetlink.h b/src/util/virnetlink.h
index 2e18af403f..1982dae5b1 100644
--- a/src/util/virnetlink.h
+++ b/src/util/virnetlink.h
@@ -53,6 +53,11 @@ typedef void (*virNetlinkEventRemoveCallback)(int watch, const virMacAddrPtr mac
  */
 int virNetlinkEventServiceStop(unsigned int protocol);
 
+/**
+ * stopNetlinkEventServerAll: stop all the monitors to receive netlink messages for libvirtd
+ */
+int virNetlinkEventServiceStopAll(void);
+
 /**
  * startNetlinkEventServer: start a monitor to receive netlink messages for libvirtd
  */