Thought of another janitorial item for the list.
[oweals/busybox.git] / syslogd.c
index 9bedc84f29a7cd71145865f86027b23d9025bc2a..8049fc5d1675777c5b599b1631808f03f21e491e 100644 (file)
--- a/syslogd.c
+++ b/syslogd.c
@@ -2,7 +2,7 @@
 /*
  * Mini syslogd implementation for busybox
  *
- * Copyright (C) 1999,2000 by Lineo, inc.
+ * Copyright (C) 1999,2000,2001 by Lineo, inc.
  * Written by Erik Andersen <andersen@lineo.com>, <andersee@debian.org>
  *
  * Copyright (C) 2000 by Karl M. Hegbloom <karlheg@debian.org>
@@ -23,7 +23,6 @@
  *
  */
 
-#include "busybox.h"
 #include <stdio.h>
 #include <stdlib.h>
 #include <ctype.h>
@@ -34,6 +33,7 @@
 #include <signal.h>
 #include <stdarg.h>
 #include <time.h>
+#include <string.h>
 #include <unistd.h>
 #include <sys/socket.h>
 #include <sys/types.h>
@@ -41,7 +41,8 @@
 #include <sys/param.h>
 
 #if ! defined __GLIBC__ && ! defined __UCLIBC__
-
+#include <sys/syscall.h>
+#include <linux/unistd.h>
 typedef unsigned int socklen_t;
 
 #ifndef __alpha__
@@ -54,6 +55,7 @@ static inline _syscall3(int, klogctl, int, type, char *, b, int, len);
 #else
 # include <sys/klog.h>
 #endif
+#include "busybox.h"
 
 
 
@@ -79,13 +81,14 @@ static char LocalHostName[32];
 #ifdef BB_FEATURE_REMOTE_LOG
 #include <netinet/in.h>
 /* udp socket for logging to remote host */
-static int  remotefd = -1;
+static int remotefd = -1;
 /* where do we log? */
 static char *RemoteHost;
 /* what port to log to? */
-static int  RemotePort = 514;
+static int RemotePort = 514;
 /* To remote log or not to remote log, that is the question. */
-static int  doRemoteLog = FALSE;
+static int doRemoteLog = FALSE;
+static int local_logging = FALSE;
 #endif
 
 /* Note: There is also a function called "message()" in init.c */
@@ -139,17 +142,17 @@ static void logMessage (int pri, char *msg)
 
        if (pri != 0) {
                for (c_fac = facilitynames;
-                        c_fac->c_name && !(c_fac->c_val == LOG_FAC(pri) << 3); c_fac++);
+                               c_fac->c_name && !(c_fac->c_val == LOG_FAC(pri) << 3); c_fac++);
                for (c_pri = prioritynames;
-                        c_pri->c_name && !(c_pri->c_val == LOG_PRI(pri)); c_pri++);
-               if (*c_fac->c_name == '\0' || *c_pri->c_name == '\0')
+                               c_pri->c_name && !(c_pri->c_val == LOG_PRI(pri)); c_pri++);
+               if (c_fac->c_name == NULL || c_pri->c_name == NULL)
                        snprintf(res, sizeof(res), "<%d>", pri);
                else
                        snprintf(res, sizeof(res), "%s.%s", c_fac->c_name, c_pri->c_name);
        }
 
        if (strlen(msg) < 16 || msg[3] != ' ' || msg[6] != ' ' ||
-               msg[9] != ':' || msg[12] != ':' || msg[15] != ' ') {
+                       msg[9] != ':' || msg[12] != ':' || msg[15] != ' ') {
                time(&now);
                timestamp = ctime(&now) + 4;
                timestamp[15] = '\0';
@@ -161,31 +164,32 @@ static void logMessage (int pri, char *msg)
 
        /* todo: supress duplicates */
 
-       /* now spew out the message to wherever it is supposed to go */
-       message("%s %s %s %s\n", timestamp, LocalHostName, res, msg);
-
 #ifdef BB_FEATURE_REMOTE_LOG
        /* send message to remote logger */
-        if ( -1 != remotefd){
-#define IOV_COUNT 2
-          struct iovec iov[IOV_COUNT];
-          struct iovec *v = iov;
-
-          bzero(&res, sizeof(res));
-          snprintf(res, sizeof(res), "<%d>", pri);
-          v->iov_base = res ;
-          v->iov_len = strlen(res);          
-          v++;
-               
-          v->iov_base = msg;
-          v->iov_len = strlen(msg);          
-
-          if ( -1 == writev(remotefd,iov, IOV_COUNT)){
-            fatalError("syslogd: cannot write to remote file handle on" 
-                       "%s:%d\n",RemoteHost,RemotePort);
-          }
-        }
+       if ( -1 != remotefd){
+static const int IOV_COUNT = 2;
+               struct iovec iov[IOV_COUNT];
+               struct iovec *v = iov;
+
+               bzero(&res, sizeof(res));
+               snprintf(res, sizeof(res), "<%d>", pri);
+               v->iov_base = res ;
+               v->iov_len = strlen(res);          
+               v++;
+
+               v->iov_base = msg;
+               v->iov_len = strlen(msg);          
+
+               if ( -1 == writev(remotefd,iov, IOV_COUNT)){
+                       error_msg_and_die("syslogd: cannot write to remote file handle on" 
+                                       "%s:%d",RemoteHost,RemotePort);
+               }
+       }
+       if (local_logging == TRUE)
 #endif
+               /* now spew out the message to wherever it is supposed to go */
+               message("%s %s %s %s\n", timestamp, LocalHostName, res, msg);
+
 
 }
 
@@ -204,10 +208,10 @@ static void domark(int sig)
        }
 }
 
-#define BUFSIZE 1023
+static const int BUFSIZE = 1023;
 static int serveConnection (int conn)
 {
-       char   buf[ BUFSIZE + 1 ];
+       RESERVE_BB_BUFFER(buf, BUFSIZE + 1);
        int    n_read;
 
        while ((n_read = read (conn, buf, BUFSIZE )) > 0) {
@@ -260,13 +264,13 @@ static void init_RemoteLog (void){
   remotefd = socket(AF_INET, SOCK_DGRAM, 0);
 
   if (remotefd < 0) {
-    fatalError("syslogd: cannot create socket\n");
+    error_msg_and_die("syslogd: cannot create socket");
   }
 
   hostinfo = (struct hostent *) gethostbyname(RemoteHost);
 
   if (!hostinfo) {
-    fatalError("syslogd: cannot resolve remote host name [%s]\n", RemoteHost);
+    error_msg_and_die("syslogd: cannot resolve remote host name [%s]", RemoteHost);
   }
 
   remoteaddr.sin_family = AF_INET;
@@ -278,7 +282,7 @@ static void init_RemoteLog (void){
      for future operations
   */
   if ( 0 != (connect(remotefd, (struct sockaddr *) &remoteaddr, len))){
-    fatalError("syslogd: cannot connect to remote host %s:%d\n", RemoteHost, RemotePort);
+    error_msg_and_die("syslogd: cannot connect to remote host %s:%d", RemoteHost, RemotePort);
   }
 
 }
@@ -294,7 +298,7 @@ static void doSyslogd (void)
        int sock_fd;
        fd_set fds;
 
-       char lfile[BUFSIZ];
+       RESERVE_BB_BUFFER(lfile, BUFSIZ);
 
        /* Set up signal handlers. */
        signal (SIGINT,  quit_signal);
@@ -311,7 +315,7 @@ static void doSyslogd (void)
        /* Create the syslog file so realpath() can work. */
        close (open (_PATH_LOG, O_RDWR | O_CREAT, 0644));
        if (realpath (_PATH_LOG, lfile) == NULL)
-               fatalError ("Could not resolve path to " _PATH_LOG ": %s\n", strerror (errno));
+               perror_msg_and_die ("Could not resolve path to " _PATH_LOG);
 
        unlink (lfile);
 
@@ -319,14 +323,14 @@ static void doSyslogd (void)
        sunx.sun_family = AF_UNIX;
        strncpy (sunx.sun_path, lfile, sizeof (sunx.sun_path));
        if ((sock_fd = socket (AF_UNIX, SOCK_STREAM, 0)) < 0)
-               fatalError ("Couldn't obtain descriptor for socket " _PATH_LOG ": %s\n", strerror (errno));
+               perror_msg_and_die ("Couldn't obtain descriptor for socket " _PATH_LOG);
 
        addrLength = sizeof (sunx.sun_family) + strlen (sunx.sun_path);
        if ((bind (sock_fd, (struct sockaddr *) &sunx, addrLength)) || (listen (sock_fd, 5)))
-               fatalError ("Could not connect to socket " _PATH_LOG ": %s\n", strerror (errno));
+               perror_msg_and_die ("Could not connect to socket " _PATH_LOG);
 
        if (chmod (lfile, 0666) < 0)
-               fatalError ("Could not set permission on " _PATH_LOG ": %s\n", strerror (errno));
+               perror_msg_and_die ("Could not set permission on " _PATH_LOG);
 
        FD_ZERO (&fds);
        FD_SET (sock_fd, &fds);
@@ -349,7 +353,7 @@ static void doSyslogd (void)
 
                if ((n_ready = select (FD_SETSIZE, &readfds, NULL, NULL, NULL)) < 0) {
                        if (errno == EINTR) continue; /* alarm may have happened. */
-                       fatalError ("select error: %s\n", strerror (errno));
+                       perror_msg_and_die ("select error");
                }
 
                for (fd = 0; (n_ready > 0) && (fd < FD_SETSIZE); fd++) {
@@ -363,7 +367,7 @@ static void doSyslogd (void)
                                        pid_t pid;
 
                                        if ((conn = accept (sock_fd, (struct sockaddr *) &sunx, &addrLength)) < 0) {
-                                               fatalError ("accept error: %s\n", strerror (errno));
+                                               perror_msg_and_die ("accept error");
                                        }
 
                                        pid = fork();
@@ -401,7 +405,8 @@ static void doKlogd (void)
 {
        int priority = LOG_INFO;
        char log_buffer[4096];
-       char *logp;
+       int i, n, lastc;
+       char *start;
 
        /* Set up sig handlers */
        signal(SIGINT, klogd_signal);
@@ -418,12 +423,14 @@ static void doKlogd (void)
        logMessage(0, "klogd started: "
                           "BusyBox v" BB_VER " (" BB_BT ")");
 
+       /* "Open the log. Currently a NOP." */
        klogctl(1, NULL, 0);
 
        while (1) {
                /* Use kernel syscalls */
                memset(log_buffer, '\0', sizeof(log_buffer));
-               if (klogctl(2, log_buffer, sizeof(log_buffer)) < 0) {
+               n = klogctl(2, log_buffer, sizeof(log_buffer));
+               if (n < 0) {
                        char message[80];
 
                        if (errno == EINTR)
@@ -433,37 +440,29 @@ static void doKlogd (void)
                        logMessage(LOG_SYSLOG | LOG_ERR, message);
                        exit(1);
                }
-               logp = log_buffer;
-               if (*log_buffer == '<') {
-                       switch (*(log_buffer + 1)) {
-                       case '0':
-                               priority = LOG_EMERG;
-                               break;
-                       case '1':
-                               priority = LOG_ALERT;
-                               break;
-                       case '2':
-                               priority = LOG_CRIT;
-                               break;
-                       case '3':
-                               priority = LOG_ERR;
-                               break;
-                       case '4':
-                               priority = LOG_WARNING;
-                               break;
-                       case '5':
-                               priority = LOG_NOTICE;
-                               break;
-                       case '6':
+
+               /* klogctl buffer parsing modelled after code in dmesg.c */
+               start=&log_buffer[0];
+               lastc='\0';
+               for (i=0; i<n; i++) {
+                       if (lastc == '\0' && log_buffer[i] == '<') {
+                               priority = 0;
+                               i++;
+                               while (isdigit(log_buffer[i])) {
+                                       priority = priority*10+(log_buffer[i]-'0');
+                                       i++;
+                               }
+                               if (log_buffer[i] == '>') i++;
+                               start = &log_buffer[i];
+                       }
+                       if (log_buffer[i] == '\n') {
+                               log_buffer[i] = '\0';  /* zero terminate this message */
+                               logMessage(LOG_KERN | priority, start);
+                               start = &log_buffer[i+1];
                                priority = LOG_INFO;
-                               break;
-                       case '7':
-                       default:
-                               priority = LOG_DEBUG;
                        }
-                       logp += 3;
+                       lastc = log_buffer[i];
                }
-               logMessage(LOG_KERN | priority, logp);
        }
 
 }
@@ -481,25 +480,19 @@ static void daemon_init (char **argv, char *dz, void fn (void))
 
 extern int syslogd_main(int argc, char **argv)
 {
-       int pid, klogd_pid;
+       int opt, pid, klogd_pid;
        int doFork = TRUE;
 
 #ifdef BB_FEATURE_KLOGD
        int startKlogd = TRUE;
 #endif
-       int stopDoingThat = FALSE;
        char *p;
-       char **argv1 = argv;
 
-       while (--argc > 0 && **(++argv1) == '-') {
-               stopDoingThat = FALSE;
-               while (stopDoingThat == FALSE && *(++(*argv1))) {
-                       switch (**argv1) {
+       /* do normal option parsing */
+       while ((opt = getopt(argc, argv, "m:nKO:R:L")) > 0) {
+               switch (opt) {
                        case 'm':
-                               if (--argc == 0) {
-                                       usage(syslogd_usage);
-                               }
-                               MarkInterval = atoi(*(++argv1)) * 60;
+                               MarkInterval = atoi(optarg) * 60;
                                break;
                        case 'n':
                                doFork = FALSE;
@@ -510,34 +503,31 @@ extern int syslogd_main(int argc, char **argv)
                                break;
 #endif
                        case 'O':
-                               if (--argc == 0) {
-                                       usage(syslogd_usage);
-                               }
-                               logFilePath = *(++argv1);
-                               stopDoingThat = TRUE;
+                               logFilePath = strdup(optarg);
                                break;
 #ifdef BB_FEATURE_REMOTE_LOG
                        case 'R':
-                          if (--argc == 0) {
-                            usage(syslogd_usage);
-                          }
-                          RemoteHost = *(++argv1);
-                          if ( (p = strchr(RemoteHost, ':'))){
-                            RemotePort = atoi(p+1);
-                            *p = '\0';
-                          }          
-                          doRemoteLog = TRUE;
-                          stopDoingThat = TRUE;
-                          break;
+                               RemoteHost = strdup(optarg);
+                               if ( (p = strchr(RemoteHost, ':'))){
+                                       RemotePort = atoi(p+1);
+                                       *p = '\0';
+                               }          
+                               doRemoteLog = TRUE;
+                               break;
+                       case 'L':
+                               local_logging = TRUE;
+                               break;
 #endif
                        default:
-                               usage(syslogd_usage);
-                       }
+                               show_usage();
                }
        }
 
-       if (argc > 0)
-               usage(syslogd_usage);
+#ifdef BB_FEATURE_REMOTE_LOG
+       /* If they have not specified remote logging, then log locally */
+       if (doRemoteLog == FALSE)
+               local_logging = TRUE;
+#endif
 
        /* Store away localhost's name before the fork */
        gethostname(LocalHostName, sizeof(LocalHostName));
@@ -568,7 +558,7 @@ extern int syslogd_main(int argc, char **argv)
                doSyslogd();
        }
 
-       return(TRUE);
+       return EXIT_SUCCESS;
 }
 
 /*