yeahuh / rpms / qemu-kvm

Forked from rpms/qemu-kvm 2 years ago
Clone
902636
From 555ec3463b3dbfd6e08eac7840419d176f113e46 Mon Sep 17 00:00:00 2001
902636
From: "Dr. David Alan Gilbert" <dgilbert@redhat.com>
902636
Date: Tue, 5 May 2020 16:35:55 +0100
902636
Subject: [PATCH 4/9] virtiofsd: add --rlimit-nofile=NUM option
902636
MIME-Version: 1.0
902636
Content-Type: text/plain; charset=UTF-8
902636
Content-Transfer-Encoding: 8bit
902636
902636
RH-Author: Dr. David Alan Gilbert <dgilbert@redhat.com>
902636
Message-id: <20200505163600.22956-3-dgilbert@redhat.com>
902636
Patchwork-id: 96270
902636
O-Subject: [RHEL-AV-8.2.1 qemu-kvm PATCH 2/7] virtiofsd: add --rlimit-nofile=NUM option
902636
Bugzilla: 1817445
902636
RH-Acked-by: Stefan Hajnoczi <stefanha@redhat.com>
902636
RH-Acked-by: Philippe Mathieu-Daudé <philmd@redhat.com>
902636
RH-Acked-by: Max Reitz <mreitz@redhat.com>
902636
RH-Acked-by: Michael S. Tsirkin <mst@redhat.com>
902636
902636
From: Stefan Hajnoczi <stefanha@redhat.com>
902636
902636
Make it possible to specify the RLIMIT_NOFILE on the command-line.
902636
Users running multiple virtiofsd processes should allocate a certain
902636
number to each process so that the system-wide limit can never be
902636
exhausted.
902636
902636
When this option is set to 0 the rlimit is left at its current value.
902636
This is useful when a management tool wants to configure the rlimit
902636
itself.
902636
902636
The default behavior remains unchanged: try to set the limit to
902636
1,000,000 file descriptors if the current rlimit is lower.
902636
902636
Signed-off-by: Stefan Hajnoczi <stefanha@redhat.com>
902636
Reviewed-by: Dr. David Alan Gilbert <dgilbert@redhat.com>
902636
Message-Id: <20200501140644.220940-2-stefanha@redhat.com>
902636
Signed-off-by: Dr. David Alan Gilbert <dgilbert@redhat.com>
902636
(cherry picked from commit 6dbb716877728ce4eb51619885ef6ef4ada9565f)
902636
Signed-off-by: Danilo C. L. de Paula <ddepaula@redhat.com>
902636
---
902636
 tools/virtiofsd/fuse_lowlevel.h  |  1 +
902636
 tools/virtiofsd/helper.c         | 23 +++++++++++++++++++++++
902636
 tools/virtiofsd/passthrough_ll.c | 22 ++++++++--------------
902636
 3 files changed, 32 insertions(+), 14 deletions(-)
902636
902636
diff --git a/tools/virtiofsd/fuse_lowlevel.h b/tools/virtiofsd/fuse_lowlevel.h
902636
index 8f6d705..562fd52 100644
902636
--- a/tools/virtiofsd/fuse_lowlevel.h
902636
+++ b/tools/virtiofsd/fuse_lowlevel.h
902636
@@ -1777,6 +1777,7 @@ struct fuse_cmdline_opts {
902636
     int syslog;
902636
     int log_level;
902636
     unsigned int max_idle_threads;
902636
+    unsigned long rlimit_nofile;
902636
 };
902636
 
902636
 /**
902636
diff --git a/tools/virtiofsd/helper.c b/tools/virtiofsd/helper.c
902636
index 0801cf7..9b3eddc 100644
902636
--- a/tools/virtiofsd/helper.c
902636
+++ b/tools/virtiofsd/helper.c
902636
@@ -23,6 +23,8 @@
902636
 #include <stdlib.h>
902636
 #include <string.h>
902636
 #include <sys/param.h>
902636
+#include <sys/time.h>
902636
+#include <sys/resource.h>
902636
 #include <unistd.h>
902636
 
902636
 #define FUSE_HELPER_OPT(t, p)                       \
902636
@@ -53,6 +55,7 @@ static const struct fuse_opt fuse_helper_opts[] = {
902636
     FUSE_HELPER_OPT("subtype=", nodefault_subtype),
902636
     FUSE_OPT_KEY("subtype=", FUSE_OPT_KEY_KEEP),
902636
     FUSE_HELPER_OPT("max_idle_threads=%u", max_idle_threads),
902636
+    FUSE_HELPER_OPT("--rlimit-nofile=%lu", rlimit_nofile),
902636
     FUSE_HELPER_OPT("--syslog", syslog),
902636
     FUSE_HELPER_OPT_VALUE("log_level=debug", log_level, FUSE_LOG_DEBUG),
902636
     FUSE_HELPER_OPT_VALUE("log_level=info", log_level, FUSE_LOG_INFO),
902636
@@ -171,6 +174,9 @@ void fuse_cmdline_help(void)
902636
            "                               default: no_writeback\n"
902636
            "    -o xattr|no_xattr          enable/disable xattr\n"
902636
            "                               default: no_xattr\n"
902636
+           "    --rlimit-nofile=<num>      set maximum number of file descriptors\n"
902636
+           "                               (0 leaves rlimit unchanged)\n"
902636
+           "                               default: 1,000,000 if the current rlimit is lower\n"
902636
            );
902636
 }
902636
 
902636
@@ -191,11 +197,28 @@ static int fuse_helper_opt_proc(void *data, const char *arg, int key,
902636
     }
902636
 }
902636
 
902636
+static unsigned long get_default_rlimit_nofile(void)
902636
+{
902636
+    rlim_t max_fds = 1000000; /* our default RLIMIT_NOFILE target */
902636
+    struct rlimit rlim;
902636
+
902636
+    if (getrlimit(RLIMIT_NOFILE, &rlim) < 0) {
902636
+        fuse_log(FUSE_LOG_ERR, "getrlimit(RLIMIT_NOFILE): %m\n");
902636
+        exit(1);
902636
+    }
902636
+
902636
+    if (rlim.rlim_cur >= max_fds) {
902636
+        return 0; /* we have more fds available than required! */
902636
+    }
902636
+    return max_fds;
902636
+}
902636
+
902636
 int fuse_parse_cmdline(struct fuse_args *args, struct fuse_cmdline_opts *opts)
902636
 {
902636
     memset(opts, 0, sizeof(struct fuse_cmdline_opts));
902636
 
902636
     opts->max_idle_threads = 10;
902636
+    opts->rlimit_nofile = get_default_rlimit_nofile();
902636
     opts->foreground = 1;
902636
 
902636
     if (fuse_opt_parse(args, opts, fuse_helper_opts, fuse_helper_opt_proc) ==
902636
diff --git a/tools/virtiofsd/passthrough_ll.c b/tools/virtiofsd/passthrough_ll.c
902636
index 50ff672..184ad0f 100644
902636
--- a/tools/virtiofsd/passthrough_ll.c
902636
+++ b/tools/virtiofsd/passthrough_ll.c
902636
@@ -2711,24 +2711,18 @@ static void setup_sandbox(struct lo_data *lo, struct fuse_session *se,
902636
     setup_seccomp(enable_syslog);
902636
 }
902636
 
902636
-/* Raise the maximum number of open file descriptors */
902636
-static void setup_nofile_rlimit(void)
902636
+/* Set the maximum number of open file descriptors */
902636
+static void setup_nofile_rlimit(unsigned long rlimit_nofile)
902636
 {
902636
-    const rlim_t max_fds = 1000000;
902636
-    struct rlimit rlim;
902636
-
902636
-    if (getrlimit(RLIMIT_NOFILE, &rlim) < 0) {
902636
-        fuse_log(FUSE_LOG_ERR, "getrlimit(RLIMIT_NOFILE): %m\n");
902636
-        exit(1);
902636
-    }
902636
+    struct rlimit rlim = {
902636
+        .rlim_cur = rlimit_nofile,
902636
+        .rlim_max = rlimit_nofile,
902636
+    };
902636
 
902636
-    if (rlim.rlim_cur >= max_fds) {
902636
+    if (rlimit_nofile == 0) {
902636
         return; /* nothing to do */
902636
     }
902636
 
902636
-    rlim.rlim_cur = max_fds;
902636
-    rlim.rlim_max = max_fds;
902636
-
902636
     if (setrlimit(RLIMIT_NOFILE, &rlim) < 0) {
902636
         /* Ignore SELinux denials */
902636
         if (errno == EPERM) {
902636
@@ -2981,7 +2975,7 @@ int main(int argc, char *argv[])
902636
 
902636
     fuse_daemonize(opts.foreground);
902636
 
902636
-    setup_nofile_rlimit();
902636
+    setup_nofile_rlimit(opts.rlimit_nofile);
902636
 
902636
     /* Must be before sandbox since it wants /proc */
902636
     setup_capng();
902636
-- 
902636
1.8.3.1
902636