1276201c337f97cbaa7a1f10d806e246eba558c0
[people/mcb30/busybox.git] / sysklogd / syslogd.c
1 /* vi: set sw=4 ts=4: */
2 /*
3  * Mini syslogd implementation for busybox
4  *
5  * Copyright (C) 1999,2000,2001 by Lineo, inc.
6  * Written by Erik Andersen <andersen@lineo.com>, <andersee@debian.org>
7  *
8  * Copyright (C) 2000 by Karl M. Hegbloom <karlheg@debian.org>
9  *
10  * This program is free software; you can redistribute it and/or modify
11  * it under the terms of the GNU General Public License as published by
12  * the Free Software Foundation; either version 2 of the License, or
13  * (at your option) any later version.
14  *
15  * This program is distributed in the hope that it will be useful,
16  * but WITHOUT ANY WARRANTY; without even the implied warranty of
17  * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU
18  * General Public License for more details.
19  *
20  * You should have received a copy of the GNU General Public License
21  * along with this program; if not, write to the Free Software
22  * Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA 02111-1307 USA
23  *
24  */
25
26 #include <stdio.h>
27 #include <stdlib.h>
28 #include <ctype.h>
29 #include <errno.h>
30 #include <fcntl.h>
31 #include <netdb.h>
32 #include <paths.h>
33 #include <signal.h>
34 #include <stdarg.h>
35 #include <time.h>
36 #include <string.h>
37 #include <unistd.h>
38 #include <sys/socket.h>
39 #include <sys/types.h>
40 #include <sys/un.h>
41 #include <sys/param.h>
42
43 #if ! defined __GLIBC__ && ! defined __UCLIBC__
44 #include <sys/syscall.h>
45 #include <linux/unistd.h>
46 typedef unsigned int socklen_t;
47
48 #ifndef __alpha__
49 # define __NR_klogctl __NR_syslog
50 static inline _syscall3(int, klogctl, int, type, char *, b, int, len);
51 #else                                                   /* __alpha__ */
52 #define klogctl syslog
53 #endif
54
55 #else
56 # include <sys/klog.h>
57 #endif
58 #include "busybox.h"
59
60
61
62 /* SYSLOG_NAMES defined to pull some extra junk from syslog.h */
63 #define SYSLOG_NAMES
64 #include <sys/syslog.h>
65 #include <sys/uio.h>
66
67 /* Path for the file where all log messages are written */
68 #define __LOG_FILE "/var/log/messages"
69
70 /* Path to the unix socket */
71 char lfile[BUFSIZ] = "";
72
73 static char *logFilePath = __LOG_FILE;
74
75 /* interval between marks in seconds */
76 static int MarkInterval = 20 * 60;
77
78 /* localhost's name */
79 static char LocalHostName[32];
80
81 #ifdef BB_FEATURE_REMOTE_LOG
82 #include <netinet/in.h>
83 /* udp socket for logging to remote host */
84 static int remotefd = -1;
85 /* where do we log? */
86 static char *RemoteHost;
87 /* what port to log to? */
88 static int RemotePort = 514;
89 /* To remote log or not to remote log, that is the question. */
90 static int doRemoteLog = FALSE;
91 static int local_logging = FALSE;
92 #endif
93
94 /* Note: There is also a function called "message()" in init.c */
95 /* Print a message to the log file. */
96 static void message (char *fmt, ...) __attribute__ ((format (printf, 1, 2)));
97 static void message (char *fmt, ...)
98 {
99         int fd;
100         struct flock fl;
101         va_list arguments;
102
103         fl.l_whence = SEEK_SET;
104         fl.l_start  = 0;
105         fl.l_len    = 1;
106
107         if ((fd = device_open (logFilePath,
108                                                    O_WRONLY | O_CREAT | O_NOCTTY | O_APPEND |
109                                                    O_NONBLOCK)) >= 0) {
110                 fl.l_type = F_WRLCK;
111                 fcntl (fd, F_SETLKW, &fl);
112                 va_start (arguments, fmt);
113                 vdprintf (fd, fmt, arguments);
114                 va_end (arguments);
115                 fl.l_type = F_UNLCK;
116                 fcntl (fd, F_SETLKW, &fl);
117                 close (fd);
118         } else {
119                 /* Always send console messages to /dev/console so people will see them. */
120                 if ((fd = device_open (_PATH_CONSOLE,
121                                                            O_WRONLY | O_NOCTTY | O_NONBLOCK)) >= 0) {
122                         va_start (arguments, fmt);
123                         vdprintf (fd, fmt, arguments);
124                         va_end (arguments);
125                         close (fd);
126                 } else {
127                         fprintf (stderr, "Bummer, can't print: ");
128                         va_start (arguments, fmt);
129                         vfprintf (stderr, fmt, arguments);
130                         fflush (stderr);
131                         va_end (arguments);
132                 }
133         }
134 }
135
136 static void logMessage (int pri, char *msg)
137 {
138         time_t now;
139         char *timestamp;
140         static char res[20] = "";
141         CODE *c_pri, *c_fac;
142
143         if (pri != 0) {
144                 for (c_fac = facilitynames;
145                                 c_fac->c_name && !(c_fac->c_val == LOG_FAC(pri) << 3); c_fac++);
146                 for (c_pri = prioritynames;
147                                 c_pri->c_name && !(c_pri->c_val == LOG_PRI(pri)); c_pri++);
148                 if (c_fac->c_name == NULL || c_pri->c_name == NULL)
149                         snprintf(res, sizeof(res), "<%d>", pri);
150                 else
151                         snprintf(res, sizeof(res), "%s.%s", c_fac->c_name, c_pri->c_name);
152         }
153
154         if (strlen(msg) < 16 || msg[3] != ' ' || msg[6] != ' ' ||
155                         msg[9] != ':' || msg[12] != ':' || msg[15] != ' ') {
156                 time(&now);
157                 timestamp = ctime(&now) + 4;
158                 timestamp[15] = '\0';
159         } else {
160                 timestamp = msg;
161                 timestamp[15] = '\0';
162                 msg += 16;
163         }
164
165         /* todo: supress duplicates */
166
167 #ifdef BB_FEATURE_REMOTE_LOG
168         /* send message to remote logger */
169         if ( -1 != remotefd){
170 static const int IOV_COUNT = 2;
171                 struct iovec iov[IOV_COUNT];
172                 struct iovec *v = iov;
173
174                 bzero(&res, sizeof(res));
175                 snprintf(res, sizeof(res), "<%d>", pri);
176                 v->iov_base = res ;
177                 v->iov_len = strlen(res);          
178                 v++;
179
180                 v->iov_base = msg;
181                 v->iov_len = strlen(msg);          
182
183                 if ( -1 == writev(remotefd,iov, IOV_COUNT)){
184                         error_msg_and_die("syslogd: cannot write to remote file handle on" 
185                                         "%s:%d",RemoteHost,RemotePort);
186                 }
187         }
188         if (local_logging == TRUE)
189 #endif
190                 /* now spew out the message to wherever it is supposed to go */
191                 message("%s %s %s %s\n", timestamp, LocalHostName, res, msg);
192
193
194 }
195
196 static void quit_signal(int sig)
197 {
198         logMessage(0, "System log daemon exiting.");
199         unlink(lfile);
200         exit(TRUE);
201 }
202
203 static void domark(int sig)
204 {
205         if (MarkInterval > 0) {
206                 logMessage(LOG_SYSLOG | LOG_INFO, "-- MARK --");
207                 alarm(MarkInterval);
208         }
209 }
210
211 static const int BUFSIZE = 1023;
212 static int serveConnection (int conn)
213 {
214         RESERVE_BB_BUFFER(buf, BUFSIZE + 1);
215         int    n_read;
216
217         while ((n_read = read (conn, buf, BUFSIZE )) > 0) {
218
219                 int           pri = (LOG_USER | LOG_NOTICE);
220                 char          line[ BUFSIZE + 1 ];
221                 unsigned char c;
222
223                 char *p = buf, *q = line;
224
225                 buf[ n_read - 1 ] = '\0';
226
227                 while (p && (c = *p) && q < &line[ sizeof (line) - 1 ]) {
228                         if (c == '<') {
229                         /* Parse the magic priority number. */
230                                 pri = 0;
231                                 while (isdigit (*(++p))) {
232                                         pri = 10 * pri + (*p - '0');
233                                 }
234                                 if (pri & ~(LOG_FACMASK | LOG_PRIMASK)){
235                                         pri = (LOG_USER | LOG_NOTICE);
236                                 }
237                         } else if (c == '\n') {
238                                 *q++ = ' ';
239                         } else if (iscntrl (c) && (c < 0177)) {
240                                 *q++ = '^';
241                                 *q++ = c ^ 0100;
242                         } else {
243                                 *q++ = c;
244                         }
245                         p++;
246                 }
247                 *q = '\0';
248                 /* Now log it */
249                 logMessage (pri, line);
250         }
251         return (0);
252 }
253
254
255 #ifdef BB_FEATURE_REMOTE_LOG
256 static void init_RemoteLog (void){
257
258   struct sockaddr_in remoteaddr;
259   struct hostent *hostinfo;
260   int len = sizeof(remoteaddr);
261
262   bzero(&remoteaddr, len);
263   
264   remotefd = socket(AF_INET, SOCK_DGRAM, 0);
265
266   if (remotefd < 0) {
267     error_msg_and_die("syslogd: cannot create socket");
268   }
269
270   hostinfo = (struct hostent *) gethostbyname(RemoteHost);
271
272   if (!hostinfo) {
273     error_msg_and_die("syslogd: cannot resolve remote host name [%s]", RemoteHost);
274   }
275
276   remoteaddr.sin_family = AF_INET;
277   remoteaddr.sin_addr = *(struct in_addr *) *hostinfo->h_addr_list;
278   remoteaddr.sin_port = htons(RemotePort);
279
280   /* 
281      Since we are using UDP sockets, connect just sets the default host and port 
282      for future operations
283   */
284   if ( 0 != (connect(remotefd, (struct sockaddr *) &remoteaddr, len))){
285     error_msg_and_die("syslogd: cannot connect to remote host %s:%d", RemoteHost, RemotePort);
286   }
287
288 }
289 #endif
290
291 static void doSyslogd (void) __attribute__ ((noreturn));
292 static void doSyslogd (void)
293 {
294         struct sockaddr_un sunx;
295         socklen_t addrLength;
296
297
298         int sock_fd;
299         fd_set fds;
300
301         RESERVE_BB_BUFFER(lfile, BUFSIZ);
302
303         /* Set up signal handlers. */
304         signal (SIGINT,  quit_signal);
305         signal (SIGTERM, quit_signal);
306         signal (SIGQUIT, quit_signal);
307         signal (SIGHUP,  SIG_IGN);
308         signal (SIGCHLD,  SIG_IGN);
309 #ifdef SIGCLD
310         signal (SIGCLD,  SIG_IGN);
311 #endif
312         signal (SIGALRM, domark);
313         alarm (MarkInterval);
314
315         /* Create the syslog file so realpath() can work. */
316         close (open (_PATH_LOG, O_RDWR | O_CREAT, 0644));
317         if (realpath (_PATH_LOG, lfile) == NULL)
318                 perror_msg_and_die ("Could not resolve path to " _PATH_LOG);
319
320         unlink (lfile);
321
322         memset (&sunx, 0, sizeof (sunx));
323         sunx.sun_family = AF_UNIX;
324         strncpy (sunx.sun_path, lfile, sizeof (sunx.sun_path));
325         if ((sock_fd = socket (AF_UNIX, SOCK_STREAM, 0)) < 0)
326                 perror_msg_and_die ("Couldn't obtain descriptor for socket " _PATH_LOG);
327
328         addrLength = sizeof (sunx.sun_family) + strlen (sunx.sun_path);
329         if ((bind (sock_fd, (struct sockaddr *) &sunx, addrLength)) || (listen (sock_fd, 5)))
330                 perror_msg_and_die ("Could not connect to socket " _PATH_LOG);
331
332         if (chmod (lfile, 0666) < 0)
333                 perror_msg_and_die ("Could not set permission on " _PATH_LOG);
334
335         FD_ZERO (&fds);
336         FD_SET (sock_fd, &fds);
337
338         #ifdef BB_FEATURE_REMOTE_LOG
339         if (doRemoteLog == TRUE){
340           init_RemoteLog();
341         }
342         #endif
343
344         logMessage (0, "syslogd started: BusyBox v" BB_VER " (" BB_BT ")");
345
346         for (;;) {
347
348                 fd_set readfds;
349                 int    n_ready;
350                 int    fd;
351
352                 memcpy (&readfds, &fds, sizeof (fds));
353
354                 if ((n_ready = select (FD_SETSIZE, &readfds, NULL, NULL, NULL)) < 0) {
355                         if (errno == EINTR) continue; /* alarm may have happened. */
356                         perror_msg_and_die ("select error");
357                 }
358
359                 for (fd = 0; (n_ready > 0) && (fd < FD_SETSIZE); fd++) {
360                         if (FD_ISSET (fd, &readfds)) {
361
362                                 --n_ready;
363
364                                 if (fd == sock_fd) {
365                                         int   conn;
366
367                                         //printf("New Connection request.\n");
368                                         if ((conn = accept (sock_fd, (struct sockaddr *) &sunx, &addrLength)) < 0) {
369                                                 perror_msg_and_die ("accept error");
370                                         }
371
372                                         FD_SET(conn, &fds);
373                                         //printf("conn: %i, set_size: %i\n",conn,FD_SETSIZE);
374                                 } else {                
375                                         //printf("Serving connection: %i\n",fd);
376                                         serveConnection (fd);
377                                         close (fd);
378                                         FD_CLR(fd, &fds);
379                                 } /* fd == sock_fd */
380                         }/* FD_ISSET() */
381                 }/* for */
382         } /* for main loop */
383 }
384
385 #ifdef BB_FEATURE_KLOGD
386
387 static void klogd_signal(int sig)
388 {
389         klogctl(7, NULL, 0);
390         klogctl(0, 0, 0);
391         logMessage(0, "Kernel log daemon exiting.");
392         exit(TRUE);
393 }
394
395 static void doKlogd (void) __attribute__ ((noreturn));
396 static void doKlogd (void)
397 {
398         int priority = LOG_INFO;
399         char log_buffer[4096];
400         int i, n, lastc;
401         char *start;
402
403         /* Set up sig handlers */
404         signal(SIGINT, klogd_signal);
405         signal(SIGKILL, klogd_signal);
406         signal(SIGTERM, klogd_signal);
407         signal(SIGHUP, SIG_IGN);
408
409 #ifdef BB_FEATURE_REMOTE_LOG
410         if (doRemoteLog == TRUE){
411           init_RemoteLog();
412         }
413 #endif
414
415         logMessage(0, "klogd started: "
416                            "BusyBox v" BB_VER " (" BB_BT ")");
417
418         /* "Open the log. Currently a NOP." */
419         klogctl(1, NULL, 0);
420
421         while (1) {
422                 /* Use kernel syscalls */
423                 memset(log_buffer, '\0', sizeof(log_buffer));
424                 n = klogctl(2, log_buffer, sizeof(log_buffer));
425                 if (n < 0) {
426                         char message[80];
427
428                         if (errno == EINTR)
429                                 continue;
430                         snprintf(message, 79, "klogd: Error return from sys_sycall: " \
431                                          "%d - %s.\n", errno, strerror(errno));
432                         logMessage(LOG_SYSLOG | LOG_ERR, message);
433                         exit(1);
434                 }
435
436                 /* klogctl buffer parsing modelled after code in dmesg.c */
437                 start=&log_buffer[0];
438                 lastc='\0';
439                 for (i=0; i<n; i++) {
440                         if (lastc == '\0' && log_buffer[i] == '<') {
441                                 priority = 0;
442                                 i++;
443                                 while (isdigit(log_buffer[i])) {
444                                         priority = priority*10+(log_buffer[i]-'0');
445                                         i++;
446                                 }
447                                 if (log_buffer[i] == '>') i++;
448                                 start = &log_buffer[i];
449                         }
450                         if (log_buffer[i] == '\n') {
451                                 log_buffer[i] = '\0';  /* zero terminate this message */
452                                 logMessage(LOG_KERN | priority, start);
453                                 start = &log_buffer[i+1];
454                                 priority = LOG_INFO;
455                         }
456                         lastc = log_buffer[i];
457                 }
458         }
459
460 }
461
462 #endif
463
464 static void daemon_init (char **argv, char *dz, void fn (void))
465 {
466         setsid();
467         chdir ("/");
468         strncpy(argv[0], dz, strlen(argv[0]));
469         fn();
470         exit(0);
471 }
472
473 extern int syslogd_main(int argc, char **argv)
474 {
475         int opt, pid, klogd_pid;
476         int doFork = TRUE;
477
478 #ifdef BB_FEATURE_KLOGD
479         int startKlogd = TRUE;
480 #endif
481         char *p;
482
483         /* do normal option parsing */
484         while ((opt = getopt(argc, argv, "m:nKO:R:L")) > 0) {
485                 switch (opt) {
486                         case 'm':
487                                 MarkInterval = atoi(optarg) * 60;
488                                 break;
489                         case 'n':
490                                 doFork = FALSE;
491                                 break;
492 #ifdef BB_FEATURE_KLOGD
493                         case 'K':
494                                 startKlogd = FALSE;
495                                 break;
496 #endif
497                         case 'O':
498                                 logFilePath = strdup(optarg);
499                                 break;
500 #ifdef BB_FEATURE_REMOTE_LOG
501                         case 'R':
502                                 RemoteHost = strdup(optarg);
503                                 if ( (p = strchr(RemoteHost, ':'))){
504                                         RemotePort = atoi(p+1);
505                                         *p = '\0';
506                                 }          
507                                 doRemoteLog = TRUE;
508                                 break;
509                         case 'L':
510                                 local_logging = TRUE;
511                                 break;
512 #endif
513                         default:
514                                 show_usage();
515                 }
516         }
517
518 #ifdef BB_FEATURE_REMOTE_LOG
519         /* If they have not specified remote logging, then log locally */
520         if (doRemoteLog == FALSE)
521                 local_logging = TRUE;
522 #endif
523
524         /* Store away localhost's name before the fork */
525         gethostname(LocalHostName, sizeof(LocalHostName));
526         if ((p = strchr(LocalHostName, '.'))) {
527                 *p++ = '\0';
528         }
529
530         umask(0);
531
532 #ifdef BB_FEATURE_KLOGD
533         /* Start up the klogd process */
534         if (startKlogd == TRUE) {
535                 klogd_pid = fork();
536                 if (klogd_pid == 0) {
537                         daemon_init (argv, "klogd", doKlogd);
538                 }
539         }
540 #endif
541
542         if (doFork == TRUE) {
543                 pid = fork();
544                 if (pid < 0)
545                         exit(pid);
546                 else if (pid == 0) {
547                         daemon_init (argv, "syslogd", doSyslogd);
548                 }
549         } else {
550                 doSyslogd();
551         }
552
553         return EXIT_SUCCESS;
554 }
555
556 /*
557 Local Variables
558 c-file-style: "linux"
559 c-basic-offset: 4
560 tab-width: 4
561 End:
562 */