Merge pull request #2 from older/masterHEADmaster

Add command line option to daemonize iiid process

1 files changed, 42 insertions, 2 deletions

diff --git a/src/iiid.cc b/src/iiid.cc
index b3dd3bf..29acd06 100644
--- a/src/iiid.cc
+++ b/src/iiid.cc
@@ -16,35 +16,38 @@
         PACKAGE " Version " VERSION "\n" \
         "Copyright (c) 2009 Klever Group"
 
 int main(int argc,char **argv) try {
 
     int port = 59278;
+    bool daemon_mode = false;
 
     while(true) {
         static struct option opts[] = {
             { "help", no_argument, 0, 'h' },
             { "usage", no_argument, 0, 'h' },
             { "version", no_argument, 0, 'V' },
             { "license", no_argument, 0, 'L' },
             { "port", required_argument, 0, 'p' },
+            { "daemon", no_argument, 0, 'd' },
             { NULL, 0, 0, 0 }
         };
-        int c = getopt_long(argc,argv,"hVLp:",opts,NULL);
+        int c = getopt_long(argc,argv,"hVLp:d",opts,NULL);
         if(c==-1) break;
         switch(c) {
             case 'h':
                 std::cerr << PHEADER << std::endl << std::endl
                     << " " << argv[0] << " [options]" << std::endl
                     << std::endl <<
                     " -h, --help,\n"
                     " --usage                    display this text\n"
                     " -V, --version              display version information\n"
                     " -L, --license              show license\n"
                     " -p <port>, --port=<port>   port to listen to\n"
                     "           (you're not likely to ever need it)\n"
+                    " -d, --daemon               run as daemon\n"
                     << std::endl << std::endl;
                 exit(0);
                 break;
             case 'V':
                 std::cerr << VERSION << std::endl;
                 exit(0);
@@ -58,12 +61,15 @@ int main(int argc,char **argv) try {
                 port = 0xffff&strtol(optarg,0,0);
                 if(errno) {
                     std::cerr << "Failed to parse port number" << std::endl;
                     exit(1);
                 }
                 break;
+            case 'd':
+                daemon_mode = true;
+                break;
             default:
                 std::cerr << "Huh?" << std::endl;
                 exit(1);
                 break;
         }
     }
@@ -71,23 +77,57 @@ int main(int argc,char **argv) try {
     const char *ident = rindex(*argv,'/');
     if(ident)
         ++ident;
     else
         ident = *argv;
     openlog(ident,LOG_PERROR|LOG_PID,LOG_DAEMON);
-    syslog(LOG_INFO,"Starting iii eye-fi manager");
+    syslog(LOG_INFO,"Starting iii eye-fi manager on port %d", port);
 
     struct stat st;
     if(stat(EYEKIN_CONF_DIR,&st) || !S_ISDIR(st.st_mode))
         syslog(LOG_WARNING,"configuration directory '%s' does not exist or is not a directory",EYEKIN_CONF_DIR);
     glob_t g; int rg = glob(EYEKIN_CONF_DIR"/????????????.conf",GLOB_NOSORT,NULL,&g);
     if(rg || !g.gl_pathc)
         syslog(LOG_WARNING,"I see nothing resembling a card config in '%s'",EYEKIN_CONF_DIR);
     else
         globfree(&g);
 
+    if(daemon_mode) {
+        pid_t pid, sid;
+        
+        /* Fork off the parent process */
+        pid = fork();
+        if (pid < 0) {
+            syslog(LOG_ERR, "Error forking the parent process");
+            exit(EXIT_FAILURE);
+        }
+
+        /* If we got a good PID, then we can exit the parent process. */
+        if (pid > 0) {
+            exit(EXIT_SUCCESS);
+        }
+
+        /* Create a new SID for the child process */
+        sid = setsid();
+        if (sid < 0) {
+            syslog(LOG_ERR, "Error creating a new SID for the child process");
+            exit(EXIT_FAILURE);
+        }
+        
+        /* Change the current working directory */
+        if ((chdir("/")) < 0) {
+            syslog(LOG_ERR, "Error changing current working directory to /");
+            exit(EXIT_FAILURE);
+        }
+        
+        /* Close out the standard file descriptors */
+        close(STDIN_FILENO);
+        close(STDOUT_FILENO);
+                close(STDERR_FILENO);   
+    }
+
     eyefiworker().run(port);
 
     closelog();
     return 0;
 } catch(const throwable_exit& e) {
     return e.rc;