examples/vhost: fix driver unregistering

[dpdk.git] / examples / vhost / main.c

diff --git a/examples/vhost/main.c b/examples/vhost/main.c
index 56a5c70..1b137b9 100644 (file)
--- a/examples/vhost/main.c
+++ b/examples/vhost/main.c
@@ -2871,6 +2871,16 @@ setup_mempool_tbl(int socket, uint32_t index, char *pool_name,
        }
 }
 
+/* When we receive a INT signal, unregister vhost driver */
+static void
+sigint_handler(__rte_unused int signum)
+{
+       /* Unregister vhost driver. */
+       int ret = rte_vhost_driver_unregister((char *)&dev_basename);
+       if (ret != 0)
+               rte_exit(EXIT_FAILURE, "vhost driver unregister failure.\n");
+       exit(0);
+}
 
 /*
  * Main function, does initialisation and calls the per-lcore functions. The CUSE
@@ -2887,6 +2897,8 @@ main(int argc, char *argv[])
        uint16_t queue_id;
        static pthread_t tid;
 
+       signal(SIGINT, sigint_handler);
+
        /* init EAL */
        ret = rte_eal_init(argc, argv);
        if (ret < 0)