Implement optional syslog logging using ordinary
[oweals/busybox.git] / networking / udhcp / common.c
index c01cd576506421fc1b6a63718f15705a857fe733..1ae65f75085c9dee64b103a59fa238811f75b225 100644 (file)
@@ -1,23 +1,13 @@
+/* vi: set sw=4 ts=4: */
 /* common.c
  *
- * Functions to assist in the writing and removing of pidfiles.
+ * Functions for debugging and logging as well as some other
+ * simple helper functions.
  *
- * Russ Dill <Russ.Dill@asu.edu> Soptember 2001
- * Rewrited by Vladimir Oleynik <dzo@simtreas.ru> (C) 2003
+ * Russ Dill <Russ.Dill@asu.edu> 2001-2003
+ * Rewritten by Vladimir Oleynik <dzo@simtreas.ru> (C) 2003
  *
- * 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
- * (at your option) any later version.
- *
- * This program is distributed in the hope that it will be useful,
- * but WITHOUT ANY WARRANTY; without even the implied warranty of
- * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
- * GNU General Public License for more details.
- *
- * You should have received a copy of the GNU General Public License
- * along with this program; if not, write to the Free Software
- * Foundation, Inc., 675 Mass Ave, Cambridge, MA 02139, USA.
+ * Licensed under GPLv2 or later, see file LICENSE in this tarball for details.
  */
 
 #include <fcntl.h>
 #include <string.h>
 #include <stdlib.h>
 #include <signal.h>
+#include <paths.h>
 #include <sys/socket.h>
+#include <stdarg.h>
+#include <syslog.h>
 
 #include "common.h"
+#include "pidfile.h"
 
 
 static int daemonized;
 
-#ifdef CONFIG_FEATURE_UDHCP_SYSLOG
-
-void udhcp_logging(int level, const char *fmt, ...)
+long uptime(void)
 {
-       int e = errno;
-       va_list p;
-       va_list p2;
-
-       va_start(p, fmt);
-       __va_copy(p2, p);
-       if(!daemonized) {
-               vprintf(fmt, p);
-               putchar('\n');
-               errno = e;
-       }
-       vsyslog(level, fmt, p2);
-       va_end(p);
+       struct sysinfo info;
+       sysinfo(&info);
+       return info.uptime;
 }
 
-void start_log(const char *client_server)
+/*
+ * This function makes sure our first socket calls
+ * aren't going to fd 1 (printf badness...) and are
+ * not later closed by daemon()
+ */
+static inline void sanitize_fds(void)
 {
-       openlog(bb_applet_name, LOG_PID | LOG_CONS, LOG_LOCAL0);
-       udhcp_logging(LOG_INFO, "%s (v%s) started", client_server, VERSION);
+       int fd = open(bb_dev_null, O_RDWR, 0);
+       if (fd < 0)
+               return;
+       while (fd < 3)
+               fd = dup(fd);
+       close(fd);
 }
 
-#else
-
-static char *syslog_level_msg[] = {
-       [LOG_EMERG]   = "EMERGENCY!",
-       [LOG_ALERT]   = "ALERT!",
-       [LOG_CRIT]    = "critical!",
-       [LOG_WARNING] = "warning",
-       [LOG_ERR]     = "error",
-       [LOG_INFO]    = "info",
-       [LOG_DEBUG]   = "debug"
-};
-
-void udhcp_logging(int level, const char *fmt, ...)
-{
-       int e = errno;
-       va_list p;
-
-       va_start(p, fmt);
-       if(!daemonized) {
-               printf("%s, ", syslog_level_msg[level]);
-               errno = e;
-               vprintf(fmt, p);
-               putchar('\n');
-       }
-       va_end(p);
-}
 
-void start_log(const char *client_server)
+void udhcp_background(const char *pidfile)
 {
-       udhcp_logging(LOG_INFO, "%s (v%s) started", client_server, VERSION);
+#ifdef __uClinux__
+       bb_error_msg("Cannot background in uclinux (yet)");
+#else /* __uClinux__ */
+       int pid_fd;
+
+       /* hold lock during fork. */
+       pid_fd = pidfile_acquire(pidfile);
+       setsid();
+       xdaemon(0, 0);
+       daemonized++;
+       logmode &= ~LOGMODE_STDIO;
+       pidfile_write_release(pid_fd);
+#endif /* __uClinux__ */
 }
-#endif
-
-static const char *saved_pidfile;
 
-static void exit_fun(void)
+void udhcp_start_log_and_pid(const char *client_server, const char *pidfile)
 {
-       if (saved_pidfile) unlink(saved_pidfile);
-}
+       int pid_fd;
 
-void background(const char *pidfile)
-{
-       int pid_fd = -1;
+       /* Make sure our syslog fd isn't overwritten */
+       sanitize_fds();
 
-       if (pidfile) {
-               pid_fd = open(pidfile, O_CREAT | O_WRONLY, 0644);
-               if (pid_fd < 0) {
-                       LOG(LOG_ERR, "Unable to open pidfile %s: %m", pidfile);
-               } else {
-                       lockf(pid_fd, F_LOCK, 0);
-                       if(!saved_pidfile)
-                               atexit(exit_fun);       /* set atexit one only */
-                       saved_pidfile = pidfile;        /* but may be rewrite */
-               }
-       }
-       while (pid_fd >= 0 && pid_fd < 3) pid_fd = dup(pid_fd); /* don't let daemon close it */
-       if (daemon(0, 0) == -1) {
-               perror("fork");
-               exit(1);
-       }
-       daemonized++;
-       if (pid_fd >= 0) {
-               FILE *out;
+       /* do some other misc startup stuff while we are here to save bytes */
+       pid_fd = pidfile_acquire(pidfile);
+       pidfile_write_release(pid_fd);
 
-               if ((out = fdopen(pid_fd, "w")) != NULL) {
-                       fprintf(out, "%d\n", getpid());
-                       fclose(out);
-               }
-               lockf(pid_fd, F_UNLCK, 0);
-               close(pid_fd);
-       }
-}
+       /* equivelent of doing a fflush after every \n */
+       setlinebuf(stdout);
 
-/* Signal handler */
-int udhcp_signal_pipe[2];
-static void signal_handler(int sig)
-{
-       if (send(udhcp_signal_pipe[1], &sig, sizeof(sig), MSG_DONTWAIT) < 0) {
-               LOG(LOG_ERR, "Could not send signal: %m");
+       if (ENABLE_FEATURE_UDHCP_SYSLOG) {
+               openlog(client_server, LOG_PID, LOG_LOCAL0);
+               logmode |= LOGMODE_SYSLOG;
        }
-}
 
-void udhcp_set_signal_pipe(int sig_add)
-{
-       socketpair(AF_UNIX, SOCK_STREAM, 0, udhcp_signal_pipe);
-       signal(SIGUSR1, signal_handler);
-       signal(SIGTERM, signal_handler);
-       if(sig_add)
-               signal(sig_add, signal_handler);
+       bb_info_msg("%s (v%s) started", client_server, BB_VER);
 }