Clean up fdreceiver sockets on exit

xapi-project · May 28, 2012 · 5772631 · 5772631
1 parent 0515cd3
commit 5772631
Show file tree

Hide file tree

Showing 2 changed files with 9 additions and 0 deletions.
diff --git a/drivers/block-nbd.c b/drivers/block-nbd.c
@@ -185,6 +185,13 @@ tdnbd_fdreceiver_start()
 
 }
 
+void
+tdnbd_fdreceiver_stop()
+{
+	if (fdreceiver)
+		td_fdreceiver_stop(fdreceiver);
+}
+
 static void
 __cancel_req(int i, struct td_nbd_request *pos, int e)
 {

diff --git a/drivers/tapdisk2.c b/drivers/tapdisk2.c
@@ -42,6 +42,7 @@
 #include "tapdisk-control.h"
 
 void tdnbd_fdreceiver_start();
+void tdnbd_fdreceiver_stop();
 
 static void
 usage(const char *app, int err)
@@ -149,6 +150,7 @@ main(int argc, char *argv[])
 	err = tapdisk_server_run();
 
 out:
+	tdnbd_fdreceiver_stop();
 	tapdisk_control_close();
 	tapdisk_stop_logging();
 	return -err;