Add foreground option to daemon
[guacd.git] / src / daemon.c
index 0f692d2..018f624 100644 (file)
@@ -20,6 +20,7 @@
  * the Initial Developer. All Rights Reserved.
  *
  * Contributor(s):
+ *  David PHAM-VAN <d.pham-van@ulteo.com> Ulteo SAS - http://www.ulteo.com
  *
  * Alternatively, the contents of this file may be used under the terms of
  * either the GNU General Public License Version 2 or later (the "GPL"), or
@@ -48,6 +49,7 @@
 
 #include <errno.h>
 #include <syslog.h>
+#include <libgen.h>
 
 #include <guacamole/client.h>
 #include <guacamole/error.h>
@@ -82,7 +84,7 @@ void guacd_handle_connection(int fd) {
     if (select->argc != 1) {
 
         /* Log error */
-        syslog(LOG_ERR, "Bad number of arguments to \"select\" (%i)",
+        guacd_log_error("Bad number of arguments to \"select\" (%i)",
                 select->argc);
 
         /* Free resources */
@@ -90,7 +92,7 @@ void guacd_handle_connection(int fd) {
         return;
     }
 
-    syslog(LOG_INFO, "Protocol \"%s\" selected", select->argv[0]);
+    guacd_log_info("Protocol \"%s\" selected", select->argv[0]);
 
     /* Get plugin from protocol in select */
     plugin = guac_client_plugin_open(select->argv[0]);
@@ -138,7 +140,7 @@ void guacd_handle_connection(int fd) {
     /* Load and init client */
     client = guac_client_plugin_get_client(plugin, socket,
             connect->argc, connect->argv,
-            guacd_log_info, guacd_log_error);
+            guacd_client_log_info, guacd_client_log_error);
 
     guac_instruction_free(connect);
 
@@ -154,16 +156,16 @@ void guacd_handle_connection(int fd) {
     }
 
     /* Start client threads */
-    syslog(LOG_INFO, "Starting client");
+    guacd_log_info("Starting client");
     if (guacd_client_start(client))
-        syslog(LOG_ERR, "Client finished abnormally");
+        guacd_log_error("Client finished abnormally");
     else
-        syslog(LOG_INFO, "Client finished normally");
+        guacd_log_info("Client finished normally");
 
     /* Clean up */
     guac_client_free(client);
     if (guac_client_plugin_close(plugin))
-        syslog(LOG_ERR, "Error closing client plugin");
+        guacd_log_error("Error closing client plugin");
 
     /* Close socket */
     guac_socket_close(socket);
@@ -198,6 +200,7 @@ int main(int argc, char* argv[]) {
     char* listen_port = "4822";  /* Default port */
     char* pidfile = NULL;
     int opt;
+    int foreground = 0;
 
     /* General */
     int retval;
@@ -206,13 +209,16 @@ int main(int argc, char* argv[]) {
     pid_t daemon_pid;
 
     /* Parse arguments */
-    while ((opt = getopt(argc, argv, "l:b:p:")) != -1) {
+    while ((opt = getopt(argc, argv, "l:b:p:f")) != -1) {
         if (opt == 'l') {
             listen_port = strdup(optarg);
         }
         else if (opt == 'b') {
             listen_address = strdup(optarg);
         }
+        else if (opt == 'f') {
+            foreground = 1;
+        }
         else if (opt == 'p') {
             pidfile = strdup(optarg);
         }
@@ -221,15 +227,20 @@ int main(int argc, char* argv[]) {
             fprintf(stderr, "USAGE: %s"
                     " [-l LISTENPORT]"
                     " [-b LISTENADDRESS]"
+                    " [-f foreground]"
                     " [-p PIDFILE]\n", argv[0]);
 
             exit(EXIT_FAILURE);
         }
     }
 
+    /* Set up logging prefix */
+    strncpy(log_prefix, basename(argv[0]), sizeof(log_prefix));
+
+
     /* Get addresses for binding */
     if ((retval = getaddrinfo(listen_address, listen_port, &hints, &addresses))) {
-        fprintf(stderr, "Error parsing given address or port: %s\n",
+        guacd_log_error("Error parsing given address or port: %s",
                 gai_strerror(retval));
         exit(EXIT_FAILURE);
     }
@@ -237,13 +248,13 @@ int main(int argc, char* argv[]) {
     /* Get socket */
     socket_fd = socket(AF_INET, SOCK_STREAM, 0);
     if (socket_fd < 0) {
-        fprintf(stderr, "Error opening socket: %s\n", strerror(errno));
+        guacd_log_error("Error opening socket: %s", strerror(errno));
         exit(EXIT_FAILURE);
     }
 
     /* Allow socket reuse */
     if (setsockopt(socket_fd, SOL_SOCKET, SO_REUSEADDR, (void*) &opt_on, sizeof(opt_on))) {
-        fprintf(stderr, "Warning: Unable to set socket options for reuse: %s\n", strerror(errno));
+        guacd_log_info("Unable to set socket options for reuse: %s", strerror(errno));
     }
 
     /* Attempt binding of each address until success */
@@ -258,7 +269,7 @@ int main(int argc, char* argv[]) {
                 bound_address, sizeof(bound_address),
                 bound_port, sizeof(bound_port),
                 NI_NUMERICHOST | NI_NUMERICSERV)))
-            fprintf(stderr, "Unable to resolve host: %s\n",
+            guacd_log_error("Unable to resolve host: %s",
                     gai_strerror(retval));
 
         /* Attempt to bind socket to address */
@@ -266,18 +277,18 @@ int main(int argc, char* argv[]) {
                     current_address->ai_addr,
                     current_address->ai_addrlen) == 0) {
 
-            fprintf(stderr, "Successfully bound socket to "
-                    "host %s, port %s\n", bound_address, bound_port);
+            guacd_log_info("Successfully bound socket to "
+                    "host %s, port %s", bound_address, bound_port);
 
             /* Done if successful bind */
             break;
 
         }
 
-        /* Otherwise log error */
+        /* Otherwise log information regarding bind failure */
         else
-            fprintf(stderr, "Error binding socket to "
-                    "host %s, port %s: %s\n",
+            guacd_log_info("Unable to bind socket to "
+                    "host %s, port %s: %s",
                     bound_address, bound_port, strerror(errno));
 
         current_address = current_address->ai_next;
@@ -286,40 +297,42 @@ int main(int argc, char* argv[]) {
 
     /* If unable to bind to anything, fail */
     if (current_address == NULL) {
-        fprintf(stderr, "Unable to bind socket to any addresses.\n");
+        guacd_log_error("Unable to bind socket to any addresses.");
         exit(EXIT_FAILURE);
     }
 
-    /* Fork into background */
-    daemon_pid = fork();
+    if (!foreground) {
+        /* Fork into background */
+        daemon_pid = fork();
 
-    /* If error, fail */
-    if (daemon_pid == -1) {
-        fprintf(stderr, "Error forking daemon process: %s\n", strerror(errno));
-        exit(EXIT_FAILURE);
-    }
+        /* If error, fail */
+        if (daemon_pid == -1) {
+            guacd_log_error("Error forking daemon process: %s", strerror(errno));
+            exit(EXIT_FAILURE);
+        }
 
-    /* If parent, write PID file and exit */
-    else if (daemon_pid != 0) {
+        /* If parent, write PID file and exit */
+        else if (daemon_pid != 0) {
 
-        if (pidfile != NULL) {
+            if (pidfile != NULL) {
 
-            /* Attempt to open pidfile and write PID */
-            FILE* pidf = fopen(pidfile, "w");
-            if (pidf) {
-                fprintf(pidf, "%d\n", daemon_pid);
-                fclose(pidf);
-            }
+                /* Attempt to open pidfile and write PID */
+                FILE* pidf = fopen(pidfile, "w");
+                if (pidf) {
+                    fprintf(pidf, "%d\n", daemon_pid);
+                    fclose(pidf);
+                }
+
+                /* Warn on failure */
+                else {
+                    guacd_log_error("Could not write PID file: %s", strerror(errno));
+                    exit(EXIT_FAILURE);
+                }
 
-            /* Warn on failure */
-            else {
-                fprintf(stderr, "WARNING: Could not write PID file: %s\n", strerror(errno));
-                exit(EXIT_FAILURE);
             }
 
+            exit(EXIT_SUCCESS);
         }
-
-        exit(EXIT_SUCCESS);
     }
 
     /* Open log */
@@ -327,12 +340,12 @@ int main(int argc, char* argv[]) {
 
     /* Ignore SIGPIPE */
     if (signal(SIGPIPE, SIG_IGN) == SIG_ERR) {
-        syslog(LOG_ERR, "Could not set handler for SIGPIPE to ignore. SIGPIPE may cause termination of the daemon.");
+        guacd_log_info("Could not set handler for SIGPIPE to ignore. SIGPIPE may cause termination of the daemon.");
     }
 
     /* Ignore SIGCHLD (force automatic removal of children) */
     if (signal(SIGCHLD, SIG_IGN) == SIG_ERR) {
-        syslog(LOG_ERR, "Could not set handler for SIGCHLD to ignore. Child processes may pile up in the process table.");
+        guacd_log_info("Could not set handler for SIGCHLD to ignore. Child processes may pile up in the process table.");
     }
 
     /* Log listening status */
@@ -349,7 +362,7 @@ int main(int argc, char* argv[]) {
 
         /* Listen for connections */
         if (listen(socket_fd, 5) < 0) {
-            syslog(LOG_ERR, "Could not listen on socket: %s", strerror(errno));
+            guacd_log_error("Could not listen on socket: %s", strerror(errno));
             return 3;
         }
 
@@ -357,7 +370,7 @@ int main(int argc, char* argv[]) {
         client_addr_len = sizeof(client_addr);
         connected_socket_fd = accept(socket_fd, (struct sockaddr*) &client_addr, &client_addr_len);
         if (connected_socket_fd < 0) {
-            syslog(LOG_ERR, "Could not accept client connection: %s", strerror(errno));
+            guacd_log_error("Could not accept client connection: %s", strerror(errno));
             return 3;
         }
 
@@ -374,7 +387,7 @@ int main(int argc, char* argv[]) {
 
         /* If error, log */
         if (child_pid == -1)
-            syslog(LOG_ERR, "Error forking child process: %s\n", strerror(errno));
+            guacd_log_error("Error forking child process: %s", strerror(errno));
 
         /* If child, start client, and exit when finished */
         else if (child_pid == 0) {
@@ -385,14 +398,14 @@ int main(int argc, char* argv[]) {
 
         /* If parent, close reference to child's descriptor */
         else if (close(connected_socket_fd) < 0) {
-            syslog(LOG_ERR, "Error closing daemon reference to child descriptor: %s", strerror(errno));
+            guacd_log_error("Error closing daemon reference to child descriptor: %s", strerror(errno));
         }
 
     }
 
     /* Close socket */
     if (close(socket_fd) < 0) {
-        syslog(LOG_ERR, "Could not close socket: %s", strerror(errno));
+        guacd_log_error("Could not close socket: %s", strerror(errno));
         return 3;
     }