1d442bb
From: Stefan Hajnoczi <stefanha@redhat.com>
1d442bb
Date: Mon, 27 Jan 2020 19:01:07 +0000
1d442bb
Subject: [PATCH] virtiofsd: add --fd=FDNUM fd passing option
1d442bb
MIME-Version: 1.0
1d442bb
Content-Type: text/plain; charset=UTF-8
1d442bb
Content-Transfer-Encoding: 8bit
1d442bb
1d442bb
Although --socket-path=PATH is useful for manual invocations, management
1d442bb
tools typically create the UNIX domain socket themselves and pass it to
1d442bb
the vhost-user device backend.  This way QEMU can be launched
1d442bb
immediately with a valid socket.  No waiting for the vhost-user device
1d442bb
backend is required when fd passing is used.
1d442bb
1d442bb
Signed-off-by: Stefan Hajnoczi <stefanha@redhat.com>
1d442bb
Reviewed-by: Daniel P. Berrangé <berrange@redhat.com>
1d442bb
Signed-off-by: Dr. David Alan Gilbert <dgilbert@redhat.com>
1d442bb
(cherry picked from commit cee8e35d4386e34bf79c3ca2aab7f7b1bb48cf8d)
1d442bb
---
1d442bb
 tools/virtiofsd/fuse_i.h        |  1 +
1d442bb
 tools/virtiofsd/fuse_lowlevel.c | 16 ++++++++++++----
1d442bb
 tools/virtiofsd/fuse_virtio.c   | 31 +++++++++++++++++++++++++------
1d442bb
 3 files changed, 38 insertions(+), 10 deletions(-)
1d442bb
1d442bb
diff --git a/tools/virtiofsd/fuse_i.h b/tools/virtiofsd/fuse_i.h
1d442bb
index 1126723d18..45995f3246 100644
1d442bb
--- a/tools/virtiofsd/fuse_i.h
1d442bb
+++ b/tools/virtiofsd/fuse_i.h
1d442bb
@@ -68,6 +68,7 @@ struct fuse_session {
1d442bb
     size_t bufsize;
1d442bb
     int error;
1d442bb
     char *vu_socket_path;
1d442bb
+    int   vu_listen_fd;
1d442bb
     int   vu_socketfd;
1d442bb
     struct fv_VuDev *virtio_dev;
1d442bb
 };
1d442bb
diff --git a/tools/virtiofsd/fuse_lowlevel.c b/tools/virtiofsd/fuse_lowlevel.c
1d442bb
index 4f4684d942..95f4db8fcf 100644
1d442bb
--- a/tools/virtiofsd/fuse_lowlevel.c
1d442bb
+++ b/tools/virtiofsd/fuse_lowlevel.c
1d442bb
@@ -2130,6 +2130,7 @@ static const struct fuse_opt fuse_ll_opts[] = {
1d442bb
     LL_OPTION("--debug", debug, 1),
1d442bb
     LL_OPTION("allow_root", deny_others, 1),
1d442bb
     LL_OPTION("--socket-path=%s", vu_socket_path, 0),
1d442bb
+    LL_OPTION("--fd=%d", vu_listen_fd, 0),
1d442bb
     FUSE_OPT_END
1d442bb
 };
1d442bb
 
1d442bb
@@ -2147,7 +2148,8 @@ void fuse_lowlevel_help(void)
1d442bb
      */
1d442bb
     printf(
1d442bb
         "    -o allow_root              allow access by root\n"
1d442bb
-        "    --socket-path=PATH         path for the vhost-user socket\n");
1d442bb
+        "    --socket-path=PATH         path for the vhost-user socket\n"
1d442bb
+        "    --fd=FDNUM                 fd number of vhost-user socket\n");
1d442bb
 }
1d442bb
 
1d442bb
 void fuse_session_destroy(struct fuse_session *se)
1d442bb
@@ -2191,6 +2193,7 @@ struct fuse_session *fuse_session_new(struct fuse_args *args,
1d442bb
         goto out1;
1d442bb
     }
1d442bb
     se->fd = -1;
1d442bb
+    se->vu_listen_fd = -1;
1d442bb
     se->conn.max_write = UINT_MAX;
1d442bb
     se->conn.max_readahead = UINT_MAX;
1d442bb
 
1d442bb
@@ -2212,8 +2215,13 @@ struct fuse_session *fuse_session_new(struct fuse_args *args,
1d442bb
         goto out4;
1d442bb
     }
1d442bb
 
1d442bb
-    if (!se->vu_socket_path) {
1d442bb
-        fprintf(stderr, "fuse: missing -o vhost_user_socket option\n");
1d442bb
+    if (!se->vu_socket_path && se->vu_listen_fd < 0) {
1d442bb
+        fuse_log(FUSE_LOG_ERR, "fuse: missing --socket-path or --fd option\n");
1d442bb
+        goto out4;
1d442bb
+    }
1d442bb
+    if (se->vu_socket_path && se->vu_listen_fd >= 0) {
1d442bb
+        fuse_log(FUSE_LOG_ERR,
1d442bb
+                 "fuse: --socket-path and --fd cannot be given together\n");
1d442bb
         goto out4;
1d442bb
     }
1d442bb
 
1d442bb
@@ -2253,7 +2261,7 @@ void fuse_session_unmount(struct fuse_session *se)
1d442bb
 
1d442bb
 int fuse_lowlevel_is_virtio(struct fuse_session *se)
1d442bb
 {
1d442bb
-    return se->vu_socket_path != NULL;
1d442bb
+    return !!se->virtio_dev;
1d442bb
 }
1d442bb
 
1d442bb
 #ifdef linux
1d442bb
diff --git a/tools/virtiofsd/fuse_virtio.c b/tools/virtiofsd/fuse_virtio.c
1d442bb
index 7e2711b504..635f87756a 100644
1d442bb
--- a/tools/virtiofsd/fuse_virtio.c
1d442bb
+++ b/tools/virtiofsd/fuse_virtio.c
1d442bb
@@ -638,18 +638,21 @@ int virtio_loop(struct fuse_session *se)
1d442bb
     return 0;
1d442bb
 }
1d442bb
 
1d442bb
-int virtio_session_mount(struct fuse_session *se)
1d442bb
+static int fv_create_listen_socket(struct fuse_session *se)
1d442bb
 {
1d442bb
     struct sockaddr_un un;
1d442bb
     mode_t old_umask;
1d442bb
 
1d442bb
+    /* Nothing to do if fd is already initialized */
1d442bb
+    if (se->vu_listen_fd >= 0) {
1d442bb
+        return 0;
1d442bb
+    }
1d442bb
+
1d442bb
     if (strlen(se->vu_socket_path) >= sizeof(un.sun_path)) {
1d442bb
         fuse_log(FUSE_LOG_ERR, "Socket path too long\n");
1d442bb
         return -1;
1d442bb
     }
1d442bb
 
1d442bb
-    se->fd = -1;
1d442bb
-
1d442bb
     /*
1d442bb
      * Create the Unix socket to communicate with qemu
1d442bb
      * based on QEMU's vhost-user-bridge
1d442bb
@@ -682,15 +685,31 @@ int virtio_session_mount(struct fuse_session *se)
1d442bb
         return -1;
1d442bb
     }
1d442bb
 
1d442bb
+    se->vu_listen_fd = listen_sock;
1d442bb
+    return 0;
1d442bb
+}
1d442bb
+
1d442bb
+int virtio_session_mount(struct fuse_session *se)
1d442bb
+{
1d442bb
+    int ret;
1d442bb
+
1d442bb
+    ret = fv_create_listen_socket(se);
1d442bb
+    if (ret < 0) {
1d442bb
+        return ret;
1d442bb
+    }
1d442bb
+
1d442bb
+    se->fd = -1;
1d442bb
+
1d442bb
     fuse_log(FUSE_LOG_INFO, "%s: Waiting for vhost-user socket connection...\n",
1d442bb
              __func__);
1d442bb
-    int data_sock = accept(listen_sock, NULL, NULL);
1d442bb
+    int data_sock = accept(se->vu_listen_fd, NULL, NULL);
1d442bb
     if (data_sock == -1) {
1d442bb
         fuse_log(FUSE_LOG_ERR, "vhost socket accept: %m\n");
1d442bb
-        close(listen_sock);
1d442bb
+        close(se->vu_listen_fd);
1d442bb
         return -1;
1d442bb
     }
1d442bb
-    close(listen_sock);
1d442bb
+    close(se->vu_listen_fd);
1d442bb
+    se->vu_listen_fd = -1;
1d442bb
     fuse_log(FUSE_LOG_INFO, "%s: Received vhost-user socket connection\n",
1d442bb
              __func__);
1d442bb