Patch from Larry Doolittle (with minor touchups from me so everything compiles
[oweals/busybox.git] / syslogd.c
index d39cd6a0dc3a441390b0e7b7c59f8161c767362c..9bedc84f29a7cd71145865f86027b23d9025bc2a 100644 (file)
--- a/syslogd.c
+++ b/syslogd.c
@@ -2,9 +2,11 @@
 /*
  * Mini syslogd implementation for busybox
  *
- * Copyright (C) 1999 by Lineo, inc.
+ * Copyright (C) 1999,2000 by Lineo, inc.
  * Written by Erik Andersen <andersen@lineo.com>, <andersee@debian.org>
  *
+ * Copyright (C) 2000 by Karl M. Hegbloom <karlheg@debian.org>
+ *
  * This program is free software; you can redistribute it and/or modify
  * it under the terms of the GNU General Public License as published by
  * the Free Software Foundation; either version 2 of the License, or
  *
  */
 
-#include "internal.h"
+#include "busybox.h"
 #include <stdio.h>
-#include <stdarg.h>
-#include <sys/socket.h>
-#include <sys/un.h>
-#include <unistd.h>
-#include <time.h>
-#include <sys/types.h>
-#include <sys/stat.h>
-#include <fcntl.h>
-#include <signal.h>
+#include <stdlib.h>
 #include <ctype.h>
-#include <netdb.h>
-#include <sys/klog.h>
 #include <errno.h>
+#include <fcntl.h>
+#include <netdb.h>
 #include <paths.h>
+#include <signal.h>
+#include <stdarg.h>
+#include <time.h>
+#include <unistd.h>
+#include <sys/socket.h>
+#include <sys/types.h>
+#include <sys/un.h>
+#include <sys/param.h>
+
+#if ! defined __GLIBC__ && ! defined __UCLIBC__
+
+typedef unsigned int socklen_t;
+
+#ifndef __alpha__
+# define __NR_klogctl __NR_syslog
+static inline _syscall3(int, klogctl, int, type, char *, b, int, len);
+#else                                                  /* __alpha__ */
+#define klogctl syslog
+#endif
+
+#else
+# include <sys/klog.h>
+#endif
 
-#define ksyslog klogctl
-extern int ksyslog(int type, char *buf, int len);
 
 
 /* SYSLOG_NAMES defined to pull some extra junk from syslog.h */
 #define SYSLOG_NAMES
 #include <sys/syslog.h>
+#include <sys/uio.h>
 
 /* Path for the file where all log messages are written */
-#define __LOG_FILE             "/var/log/messages"
+#define __LOG_FILE "/var/log/messages"
 
+/* Path to the unix socket */
+char lfile[BUFSIZ] = "";
 
 static char *logFilePath = __LOG_FILE;
 
@@ -58,55 +76,61 @@ static int MarkInterval = 20 * 60;
 /* localhost's name */
 static char LocalHostName[32];
 
-static const char syslogd_usage[] =
-       "syslogd [OPTION]...\n\n"
-       "Linux system and kernel (provides klogd) logging utility.\n"
-       "Note that this version of syslogd/klogd ignores /etc/syslog.conf.\n\n"
-       "Options:\n"
-       "\t-m\tChange the mark timestamp interval. default=20min. 0=off\n"
-       "\t-n\tDo not fork into the background (for when run by init)\n"
-#ifdef BB_KLOGD
-       "\t-K\tDo not start up the klogd process (by default syslogd spawns klogd).\n"
+#ifdef BB_FEATURE_REMOTE_LOG
+#include <netinet/in.h>
+/* udp socket for logging to remote host */
+static int  remotefd = -1;
+/* where do we log? */
+static char *RemoteHost;
+/* what port to log to? */
+static int  RemotePort = 514;
+/* To remote log or not to remote log, that is the question. */
+static int  doRemoteLog = FALSE;
 #endif
-       "\t-O\tSpecify an alternate log file.  default=/var/log/messages\n";
-
 
-/* print a message to the log file */
-static void message(char *fmt, ...)
+/* Note: There is also a function called "message()" in init.c */
+/* Print a message to the log file. */
+static void message (char *fmt, ...) __attribute__ ((format (printf, 1, 2)));
+static void message (char *fmt, ...)
 {
        int fd;
+       struct flock fl;
        va_list arguments;
 
-       if (
-               (fd =
-                device_open(logFilePath,
-                                        O_WRONLY | O_CREAT | O_NOCTTY | O_APPEND |
-                                        O_NONBLOCK)) >= 0) {
-               va_start(arguments, fmt);
-               vdprintf(fd, fmt, arguments);
-               va_end(arguments);
-               close(fd);
+       fl.l_whence = SEEK_SET;
+       fl.l_start  = 0;
+       fl.l_len    = 1;
+
+       if ((fd = device_open (logFilePath,
+                                                  O_WRONLY | O_CREAT | O_NOCTTY | O_APPEND |
+                                                  O_NONBLOCK)) >= 0) {
+               fl.l_type = F_WRLCK;
+               fcntl (fd, F_SETLKW, &fl);
+               va_start (arguments, fmt);
+               vdprintf (fd, fmt, arguments);
+               va_end (arguments);
+               fl.l_type = F_UNLCK;
+               fcntl (fd, F_SETLKW, &fl);
+               close (fd);
        } else {
                /* Always send console messages to /dev/console so people will see them. */
-               if (
-                       (fd =
-                        device_open(_PATH_CONSOLE,
-                                                O_WRONLY | O_NOCTTY | O_NONBLOCK)) >= 0) {
-                       va_start(arguments, fmt);
-                       vdprintf(fd, fmt, arguments);
-                       va_end(arguments);
-                       close(fd);
+               if ((fd = device_open (_PATH_CONSOLE,
+                                                          O_WRONLY | O_NOCTTY | O_NONBLOCK)) >= 0) {
+                       va_start (arguments, fmt);
+                       vdprintf (fd, fmt, arguments);
+                       va_end (arguments);
+                       close (fd);
                } else {
-                       fprintf(stderr, "Bummer, can't print: ");
-                       va_start(arguments, fmt);
-                       vfprintf(stderr, fmt, arguments);
-                       fflush(stderr);
-                       va_end(arguments);
+                       fprintf (stderr, "Bummer, can't print: ");
+                       va_start (arguments, fmt);
+                       vfprintf (stderr, fmt, arguments);
+                       fflush (stderr);
+                       va_end (arguments);
                }
        }
 }
 
-static void logMessage(int pri, char *msg)
+static void logMessage (int pri, char *msg)
 {
        time_t now;
        char *timestamp;
@@ -139,12 +163,36 @@ static void logMessage(int pri, char *msg)
 
        /* 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);
+          }
+        }
+#endif
+
 }
 
 static void quit_signal(int sig)
 {
        logMessage(0, "System log daemon exiting.");
-       unlink(_PATH_LOG);
+       unlink(lfile);
        exit(TRUE);
 }
 
@@ -156,101 +204,200 @@ static void domark(int sig)
        }
 }
 
-static void doSyslogd(void)
+#define BUFSIZE 1023
+static int serveConnection (int conn)
 {
-       struct sockaddr_un sunx;
-       int fd, conn;
-       size_t addrLength;
-       char buf[1024];
-       char *q, *p = buf;
-       int readSize;
+       char   buf[ BUFSIZE + 1 ];
+       int    n_read;
 
-       /* Set up sig handlers */
-       signal(SIGINT, quit_signal);
-       signal(SIGTERM, quit_signal);
-       signal(SIGQUIT, quit_signal);
-       signal(SIGALRM, domark);
-       signal(SIGHUP, SIG_IGN);
-       alarm(MarkInterval);
+       while ((n_read = read (conn, buf, BUFSIZE )) > 0) {
 
-       /* Remove any preexisting socket/file */
-       unlink(_PATH_LOG);
+               int           pri = (LOG_USER | LOG_NOTICE);
+               char          line[ BUFSIZE + 1 ];
+               unsigned char c;
 
-       memset(&sunx, 0, sizeof(sunx));
-       sunx.sun_family = AF_UNIX;      /* Unix domain socket */
-       strncpy(sunx.sun_path, _PATH_LOG, sizeof(sunx.sun_path));
-       if ((fd = socket(AF_UNIX, SOCK_STREAM, 0)) < 0) {
-               perror("Couldn't obtain descriptor for socket " _PATH_LOG);
-               exit(FALSE);
-       }
+               char *p = buf, *q = line;
 
-       addrLength = sizeof(sunx.sun_family) + strlen(sunx.sun_path);
-       if ((bind(fd, (struct sockaddr *) &sunx, addrLength)) ||
-               (listen(fd, 5))) {
-               perror("Could not connect to socket " _PATH_LOG);
-               exit(FALSE);
-       }
+               buf[ n_read - 1 ] = '\0';
 
-       umask(0);
-       if (chmod(_PATH_LOG, 0666) < 0) {
-               perror("Could not set permission on " _PATH_LOG);
-               exit(FALSE);
+               while (p && (c = *p) && q < &line[ sizeof (line) - 1 ]) {
+                       if (c == '<') {
+                       /* Parse the magic priority number. */
+                               pri = 0;
+                               while (isdigit (*(++p))) {
+                                       pri = 10 * pri + (*p - '0');
+                               }
+                               if (pri & ~(LOG_FACMASK | LOG_PRIMASK)){
+                                       pri = (LOG_USER | LOG_NOTICE);
+                               }
+                       } else if (c == '\n') {
+                               *q++ = ' ';
+                       } else if (iscntrl (c) && (c < 0177)) {
+                               *q++ = '^';
+                               *q++ = c ^ 0100;
+                       } else {
+                               *q++ = c;
+                       }
+                       p++;
+               }
+               *q = '\0';
+               /* Now log it */
+               logMessage (pri, line);
        }
+       return (0);
+}
+
+
+#ifdef BB_FEATURE_REMOTE_LOG
+static void init_RemoteLog (void){
+
+  struct sockaddr_in remoteaddr;
+  struct hostent *hostinfo;
+  int len = sizeof(remoteaddr);
+
+  bzero(&remoteaddr, len);
+  
+  remotefd = socket(AF_INET, SOCK_DGRAM, 0);
+
+  if (remotefd < 0) {
+    fatalError("syslogd: cannot create socket\n");
+  }
+
+  hostinfo = (struct hostent *) gethostbyname(RemoteHost);
+
+  if (!hostinfo) {
+    fatalError("syslogd: cannot resolve remote host name [%s]\n", RemoteHost);
+  }
+
+  remoteaddr.sin_family = AF_INET;
+  remoteaddr.sin_addr = *(struct in_addr *) *hostinfo->h_addr_list;
+  remoteaddr.sin_port = htons(RemotePort);
+
+  /* 
+     Since we are using UDP sockets, connect just sets the default host and port 
+     for future operations
+  */
+  if ( 0 != (connect(remotefd, (struct sockaddr *) &remoteaddr, len))){
+    fatalError("syslogd: cannot connect to remote host %s:%d\n", RemoteHost, RemotePort);
+  }
+
+}
+#endif
+
+static void doSyslogd (void) __attribute__ ((noreturn));
+static void doSyslogd (void)
+{
+       struct sockaddr_un sunx;
+       socklen_t addrLength;
+
+
+       int sock_fd;
+       fd_set fds;
 
-       logMessage(0, "syslogd started: BusyBox v" BB_VER " (" BB_BT ")");
-
-
-       while ((conn = accept(fd, (struct sockaddr *) &sunx,
-                                                 &addrLength)) >= 0) {
-               while ((readSize = read(conn, buf, sizeof(buf))) > 0) {
-                       char line[1025];
-                       unsigned char c;
-                       int pri = (LOG_USER | LOG_NOTICE);
-
-                       memset(line, 0, sizeof(line));
-                       p = buf;
-                       q = line;
-                       while (p && (c = *p) && q < &line[sizeof(line) - 1]) {
-                               if (c == '<') {
-                                       /* Parse the magic priority number */
-                                       pri = 0;
-                                       while (isdigit(*(++p))) {
-                                               pri = 10 * pri + (*p - '0');
+       char lfile[BUFSIZ];
+
+       /* Set up signal handlers. */
+       signal (SIGINT,  quit_signal);
+       signal (SIGTERM, quit_signal);
+       signal (SIGQUIT, quit_signal);
+       signal (SIGHUP,  SIG_IGN);
+       signal (SIGCHLD,  SIG_IGN);
+#ifdef SIGCLD
+       signal (SIGCLD,  SIG_IGN);
+#endif
+       signal (SIGALRM, domark);
+       alarm (MarkInterval);
+
+       /* 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));
+
+       unlink (lfile);
+
+       memset (&sunx, 0, sizeof (sunx));
+       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));
+
+       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));
+
+       if (chmod (lfile, 0666) < 0)
+               fatalError ("Could not set permission on " _PATH_LOG ": %s\n", strerror (errno));
+
+       FD_ZERO (&fds);
+       FD_SET (sock_fd, &fds);
+
+        #ifdef BB_FEATURE_REMOTE_LOG
+        if (doRemoteLog == TRUE){
+          init_RemoteLog();
+        }
+        #endif
+
+       logMessage (0, "syslogd started: BusyBox v" BB_VER " (" BB_BT ")");
+
+       for (;;) {
+
+               fd_set readfds;
+               int    n_ready;
+               int    fd;
+
+               memcpy (&readfds, &fds, sizeof (fds));
+
+               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));
+               }
+
+               for (fd = 0; (n_ready > 0) && (fd < FD_SETSIZE); fd++) {
+                       if (FD_ISSET (fd, &readfds)) {
+
+                               --n_ready;
+
+                               if (fd == sock_fd) {
+
+                                       int   conn;
+                                       pid_t pid;
+
+                                       if ((conn = accept (sock_fd, (struct sockaddr *) &sunx, &addrLength)) < 0) {
+                                               fatalError ("accept error: %s\n", strerror (errno));
+                                       }
+
+                                       pid = fork();
+
+                                       if (pid < 0) {
+                                               perror ("syslogd: fork");
+                                               close (conn);
+                                               continue;
+                                       }
+
+                                       if (pid == 0) {
+                                               serveConnection (conn);
+                                               close (conn);
+                                               exit( TRUE);
                                        }
-                                       if (pri & ~(LOG_FACMASK | LOG_PRIMASK))
-                                               pri = (LOG_USER | LOG_NOTICE);
-                               } else if (c == '\n') {
-                                       *q++ = ' ';
-                               } else if (iscntrl(c) && (c < 0177)) {
-                                       *q++ = '^';
-                                       *q++ = c ^ 0100;
-                               } else {
-                                       *q++ = c;
+                                       close (conn);
                                }
-                               p++;
                        }
-                       *q = '\0';
-
-                       /* Now log it */
-                       logMessage(pri, line);
                }
-               close(conn);
        }
-
-       close(fd);
 }
 
-#ifdef BB_KLOGD
+#ifdef BB_FEATURE_KLOGD
 
 static void klogd_signal(int sig)
 {
-       ksyslog(7, NULL, 0);
-       ksyslog(0, 0, 0);
+       klogctl(7, NULL, 0);
+       klogctl(0, 0, 0);
        logMessage(0, "Kernel log daemon exiting.");
        exit(TRUE);
 }
 
-static void doKlogd(void)
+static void doKlogd (void) __attribute__ ((noreturn));
+static void doKlogd (void)
 {
        int priority = LOG_INFO;
        char log_buffer[4096];
@@ -261,15 +408,22 @@ static void doKlogd(void)
        signal(SIGKILL, klogd_signal);
        signal(SIGTERM, klogd_signal);
        signal(SIGHUP, SIG_IGN);
+
+#ifdef BB_FEATURE_REMOTE_LOG
+        if (doRemoteLog == TRUE){
+          init_RemoteLog();
+        }
+#endif
+
        logMessage(0, "klogd started: "
                           "BusyBox v" BB_VER " (" BB_BT ")");
 
-       ksyslog(1, NULL, 0);
+       klogctl(1, NULL, 0);
 
        while (1) {
                /* Use kernel syscalls */
                memset(log_buffer, '\0', sizeof(log_buffer));
-               if (ksyslog(2, log_buffer, sizeof(log_buffer)) < 0) {
+               if (klogctl(2, log_buffer, sizeof(log_buffer)) < 0) {
                        char message[80];
 
                        if (errno == EINTR)
@@ -316,12 +470,21 @@ static void doKlogd(void)
 
 #endif
 
+static void daemon_init (char **argv, char *dz, void fn (void))
+{
+       setsid();
+       chdir ("/");
+       strncpy(argv[0], dz, strlen(argv[0]));
+       fn();
+       exit(0);
+}
+
 extern int syslogd_main(int argc, char **argv)
 {
        int pid, klogd_pid;
        int doFork = TRUE;
 
-#ifdef BB_KLOGD
+#ifdef BB_FEATURE_KLOGD
        int startKlogd = TRUE;
 #endif
        int stopDoingThat = FALSE;
@@ -341,7 +504,7 @@ extern int syslogd_main(int argc, char **argv)
                        case 'n':
                                doFork = FALSE;
                                break;
-#ifdef BB_KLOGD
+#ifdef BB_FEATURE_KLOGD
                        case 'K':
                                startKlogd = FALSE;
                                break;
@@ -353,40 +516,65 @@ extern int syslogd_main(int argc, char **argv)
                                logFilePath = *(++argv1);
                                stopDoingThat = TRUE;
                                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;
+#endif
                        default:
                                usage(syslogd_usage);
                        }
                }
        }
 
+       if (argc > 0)
+               usage(syslogd_usage);
+
        /* Store away localhost's name before the fork */
        gethostname(LocalHostName, sizeof(LocalHostName));
        if ((p = strchr(LocalHostName, '.'))) {
                *p++ = '\0';
        }
 
+       umask(0);
+
+#ifdef BB_FEATURE_KLOGD
+       /* Start up the klogd process */
+       if (startKlogd == TRUE) {
+               klogd_pid = fork();
+               if (klogd_pid == 0) {
+                       daemon_init (argv, "klogd", doKlogd);
+               }
+       }
+#endif
+
        if (doFork == TRUE) {
                pid = fork();
                if (pid < 0)
                        exit(pid);
                else if (pid == 0) {
-                       strncpy(argv[0], "syslogd", strlen(argv[0]));
-                       doSyslogd();
+                       daemon_init (argv, "syslogd", doSyslogd);
                }
        } else {
                doSyslogd();
        }
 
-#ifdef BB_KLOGD
-       /* Start up the klogd process */
-       if (startKlogd == TRUE) {
-               klogd_pid = fork();
-               if (klogd_pid == 0) {
-                       strncpy(argv[0], "klogd", strlen(argv[0]));
-                       doKlogd();
-               }
-       }
-#endif
-
-       exit(TRUE);
+       return(TRUE);
 }
+
+/*
+Local Variables
+c-file-style: "linux"
+c-basic-offset: 4
+tab-width: 4
+End:
+*/