tools: hv: introduce -n/--no-daemon option

All tools/hv daemons do mandatory daemon() on startup. However, no pidfile
is created, this make it difficult for an init system to track such daemons.
Modern linux distros use systemd as their init system. It can handle the
daemonizing by itself, however, it requires a daemon to stay in foreground
for that. Some distros already carry distro-specific patch for hv tools
which switches off daemon().

Introduce -n/--no-daemon option for all 3 daemons in hv/tools. Parse options
with getopt() to make this part easily expandable.

Signed-off-by: Vitaly Kuznetsov <vkuznets@redhat.com>
Signed-off-by: K. Y. Srinivasan <kys@microsoft.com>
Signed-off-by: Greg Kroah-Hartman <gregkh@linuxfoundation.org>

1 files changed, 31 insertions, 2 deletions

diff --git a/tools/hv/hv_fcopy_daemon.c b/tools/hv/hv_fcopy_daemon.c
index 8f96b3ee0724..f437d739f37d 100644
--- a/tools/hv/hv_fcopy_daemon.c
+++ b/tools/hv/hv_fcopy_daemon.c
@@ -33,6 +33,7 @@
 #include <sys/stat.h>
 #include <fcntl.h>
 #include <dirent.h>
+#include <getopt.h>
 
 static int target_fd;
 static char target_fname[W_MAX_PATH];
@@ -126,15 +127,43 @@ static int hv_copy_cancel(void)
 
 }
 
-int main(void)
+void print_usage(char *argv[])
+{
+        fprintf(stderr, "Usage: %s [options]\n"
+                "Options are:\n"
+                "  -n, --no-daemon        stay in foreground, don't daemonize\n"
+                "  -h, --help             print this help\n", argv[0]);
+}
+
+int main(int argc, char *argv[])
 {
         int fd, fcopy_fd, len;
         int error;
+        int daemonize = 1, long_index = 0, opt;
         int version = FCOPY_CURRENT_VERSION;
         char *buffer[4096 * 2];
         struct hv_fcopy_hdr *in_msg;
 
-        if (daemon(1, 0)) {
+        static struct option long_options[] = {
+                {"help",        no_argument,       0,  'h' },
+                {"no-daemon",   no_argument,       0,  'n' },
+                {0,             0,                 0,  0   }
+        };
+
+        while ((opt = getopt_long(argc, argv, "hn", long_options,
+                                  &long_index)) != -1) {
+                switch (opt) {
+                case 'n':
+                        daemonize = 0;
+                        break;
+                case 'h':
+                default:
+                        print_usage(argv);
+                        exit(EXIT_FAILURE);
+                }
+        }
+
+        if (daemonize && daemon(1, 0)) {
                 syslog(LOG_ERR, "daemon() failed; error: %s", strerror(errno));
                 exit(EXIT_FAILURE);
         }