From 5ec5449194a8f9d977d3ab0be7a1f7e5d41cb638 Mon Sep 17 00:00:00 2001
From: Mukund Sivaraman <muks@banu.com>
Date: Tue, 15 Sep 2009 01:34:30 +0530
Subject: Extract cmdline parsing code into a function

---
 src/main.c | 59 +++++++++++++++++++++++++++++++++++++----------------------
 src/main.h |  1 +
 2 files changed, 38 insertions(+), 22 deletions(-)

diff --git a/src/main.c b/src/main.c
index bd5babf..ff14c56 100644
--- a/src/main.c
+++ b/src/main.c
@@ -161,36 +161,32 @@ static int get_id (char *str)
         return atoi (str);
 }
 
-int main (int argc, char **argv)
+/**
+ * process_cmdline:
+ * @argc: argc as passed to main()
+ * @argv: argv as passed to main()
+ *
+ * This function parses command line arguments.
+ **/
+static void
+process_cmdline (int argc, char **argv)
 {
-        int optch;
-        unsigned int godaemon = TRUE;   /* boolean */
-        struct passwd *thisuser = NULL;
-        struct group *thisgroup = NULL;
-        FILE *config_file;
-
-        /* Only allow u+rw bits. This may be required for some versions
-         * of glibc so that mkstemp() doesn't make us vulnerable.
-         */
-        umask (0177);
-
-        /* Default configuration file location */
-        config.config_file = DEFAULT_CONF_FILE;
+        int opt;
 
-        /*
-         * Process the various options
-         */
-        while ((optch = getopt (argc, argv, "c:vldh")) != EOF) {
-                switch (optch) {
+        while ((opt = getopt (argc, argv, "c:vldh")) != EOF) {
+                switch (opt) {
                 case 'v':
                         display_version ();
                         exit (EX_OK);
+
                 case 'l':
                         display_license ();
                         exit (EX_OK);
+
                 case 'd':
-                        godaemon = FALSE;
+                        config.godaemon = FALSE;
                         break;
+
                 case 'c':
                         config.config_file = safestrdup (optarg);
                         if (!config.config_file) {
@@ -200,12 +196,31 @@ int main (int argc, char **argv)
                                 exit (EX_SOFTWARE);
                         }
                         break;
+
                 case 'h':
                 default:
                         display_usage ();
                         exit (EX_OK);
                 }
         }
+}
+
+int
+main (int argc, char **argv)
+{
+        struct passwd *thisuser = NULL;
+        struct group *thisgroup = NULL;
+        FILE *config_file;
+
+        /* Only allow u+rw bits. This may be required for some versions
+         * of glibc so that mkstemp() doesn't make us vulnerable.
+         */
+        umask (0177);
+
+        config.config_file = DEFAULT_CONF_FILE;
+        config.godaemon = TRUE;
+
+        process_cmdline (argc, argv);
 
         log_message (LOG_INFO, "Initializing " PACKAGE " ...");
 
@@ -244,7 +259,7 @@ int main (int argc, char **argv)
                 }
                 config.syslog = FALSE;  /* disable syslog */
         } else if (config.syslog) {
-                if (godaemon == TRUE)
+                if (config.godaemon == TRUE)
                         openlog ("tinyproxy", LOG_PID, LOG_DAEMON);
                 else
                         openlog ("tinyproxy", LOG_PID, LOG_USER);
@@ -299,7 +314,7 @@ int main (int argc, char **argv)
                 anonymous_insert ("Content-Type");
         }
 
-        if (godaemon == TRUE)
+        if (config.godaemon == TRUE)
                 makedaemon ();
 
         if (config.pidpath) {
diff --git a/src/main.h b/src/main.h
index 243962e..0c65fd6 100644
--- a/src/main.h
+++ b/src/main.h
@@ -50,6 +50,7 @@ struct config_s {
         unsigned int syslog;    /* boolean */
         int port;
         char *stathost;
+        unsigned int godaemon;  /* boolean */
         unsigned int quit;      /* boolean */
         char *user;
         char *group;
-- 
cgit v1.2.3