summaryrefslogtreecommitdiffstats
path: root/src/klammail/output.c
diff options
context:
space:
mode:
Diffstat (limited to 'src/klammail/output.c')
-rw-r--r--src/klammail/output.c369
1 files changed, 369 insertions, 0 deletions
diff --git a/src/klammail/output.c b/src/klammail/output.c
new file mode 100644
index 0000000..7efc77a
--- /dev/null
+++ b/src/klammail/output.c
@@ -0,0 +1,369 @@
+/*
+ * Copyright (C) 2002 - 2004 Tomasz Kojm <tkojm@clamav.net>
+ *
+ * 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.
+ */
+
+
+
+#include <stdio.h>
+#include <stdarg.h>
+#include <stdlib.h>
+#include <string.h>
+#include <ctype.h>
+#include <unistd.h>
+#include <fcntl.h>
+#include <time.h>
+#include <sys/stat.h>
+#include <errno.h>
+#include <sys/time.h>
+#include <sys/socket.h>
+#if HAVE_SYS_TYPES_H
+#include <sys/types.h>
+#endif
+
+#if defined(USE_SYSLOG) && !defined(C_AIX)
+#include <syslog.h>
+#endif
+
+#include "output.h"
+#include "memory.h"
+
+#ifdef CL_THREAD_SAFE
+#include <pthread.h>
+pthread_mutex_t logg_mutex = PTHREAD_MUTEX_INITIALIZER;
+#endif
+
+FILE *logg_fd = NULL;
+
+short int logg_verbose = 0, logg_lock = 0, logg_time = 0;
+int logg_size = 0;
+const char *logg_file = NULL;
+#if defined(USE_SYSLOG) && !defined(C_AIX)
+short logg_syslog = 0;
+#endif
+
+short int mprintf_disabled = 0, mprintf_verbose = 0, mprintf_quiet = 0,
+ mprintf_stdout = 0;
+
+int mdprintf(int desc, const char *str, ...)
+{
+ va_list args;
+ char buff[512];
+ int bytes;
+
+ va_start(args, str);
+ bytes = vsnprintf(buff, 512, str, args);
+ va_end(args);
+ write(desc, buff, bytes);
+ return bytes;
+}
+
+void logg_close(void) {
+
+#ifdef CL_THREAD_SAFE
+ pthread_mutex_lock(&logg_mutex);
+#endif
+ if (logg_fd) {
+ fclose(logg_fd);
+ logg_fd = NULL;
+ }
+#ifdef CL_THREAD_SAFE
+ pthread_mutex_unlock(&logg_mutex);
+#endif
+
+#if defined(USE_SYSLOG) && !defined(C_AIX)
+ if(logg_syslog) {
+ closelog();
+ }
+#endif
+}
+
+int logg(const char *str, ...)
+{
+ va_list args;
+ struct flock fl;
+ char *pt, *timestr, vbuff[1025];
+ time_t currtime;
+ struct stat sb;
+ mode_t old_umask;
+
+
+ va_start(args, str);
+
+ if(logg_file) {
+#ifdef CL_THREAD_SAFE
+ pthread_mutex_lock(&logg_mutex);
+#endif
+ if(!logg_fd) {
+ old_umask = umask(0037);
+ if((logg_fd = fopen(logg_file, "a")) == NULL) {
+ umask(old_umask);
+#ifdef CL_THREAD_SAFE
+ pthread_mutex_unlock(&logg_mutex);
+#endif
+ printf("ERROR: Can't open %s in append mode.\n", logg_file);
+ return -1;
+ } else umask(old_umask);
+
+ if(logg_lock) {
+ memset(&fl, 0, sizeof(fl));
+ fl.l_type = F_WRLCK;
+ if(fcntl(fileno(logg_fd), F_SETLK, &fl) == -1) {
+#ifdef CL_THREAD_SAFE
+ pthread_mutex_unlock(&logg_mutex);
+#endif
+ return -1;
+ }
+ }
+ }
+
+ /* Need to avoid logging time for verbose messages when logverbose
+ is not set or we get a bunch of timestamps in the log without
+ newlines... */
+ if(logg_time && ((*str != '*') || logg_verbose)) {
+ time(&currtime);
+ pt = ctime(&currtime);
+ timestr = mcalloc(strlen(pt), sizeof(char));
+ strncpy(timestr, pt, strlen(pt) - 1);
+ fprintf(logg_fd, "%s -> ", timestr);
+ free(timestr);
+ }
+
+ if(logg_size) {
+ if(stat(logg_file, &sb) != -1) {
+ if(sb.st_size > logg_size) {
+ logg_file = NULL;
+ fprintf(logg_fd, "Log size = %d, maximal = %d\n", (int) sb.st_size, logg_size);
+ fprintf(logg_fd, "LOGGING DISABLED (Maximal log file size exceeded).\n");
+ fclose(logg_fd);
+ logg_fd = NULL;
+#ifdef CL_THREAD_SAFE
+ pthread_mutex_unlock(&logg_mutex);
+#endif
+ return 0;
+ }
+ }
+ }
+
+
+ if(*str == '!') {
+ fprintf(logg_fd, "ERROR: ");
+ vfprintf(logg_fd, str + 1, args);
+ } else if(*str == '^') {
+ fprintf(logg_fd, "WARNING: ");
+ vfprintf(logg_fd, str + 1, args);
+ } else if(*str == '*') {
+ if(logg_verbose)
+ vfprintf(logg_fd, str + 1, args);
+ } else vfprintf(logg_fd, str, args);
+
+
+ fflush(logg_fd);
+
+#ifdef CL_THREAD_SAFE
+ pthread_mutex_unlock(&logg_mutex);
+#endif
+ }
+
+#if defined(USE_SYSLOG) && !defined(C_AIX)
+ if(logg_syslog) {
+
+ /* due to a problem with superfluous control characters (which
+ * vsnprintf() handles correctly) in (v)syslog we have to remove
+ * them in a final string
+ *
+ * FIXME: substitute %% instead of _
+ */
+ vsnprintf(vbuff, 1024, str, args);
+ vbuff[1024] = 0;
+
+ while((pt = strchr(vbuff, '%')))
+ *pt = '_';
+
+ if(vbuff[0] == '!') {
+ syslog(LOG_ERR, vbuff + 1);
+ } else if(vbuff[0] == '^') {
+ syslog(LOG_WARNING, vbuff + 1);
+ } else if(vbuff[0] == '*') {
+ if(logg_verbose) {
+ syslog(LOG_DEBUG, vbuff + 1);
+ }
+ } else syslog(LOG_INFO, vbuff);
+
+ }
+#endif
+
+ va_end(args);
+ return 0;
+}
+
+void mprintf(const char *str, ...)
+{
+ va_list args;
+ FILE *fd;
+ char logbuf[512];
+
+
+ if(mprintf_disabled) {
+ if(*str == '@') {
+ va_start(args, str);
+#ifdef NO_SNPRINTF
+ vsprintf(logbuf, ++str, args);
+#else
+ vsnprintf(logbuf, sizeof(logbuf), ++str, args);
+#endif
+ va_end(args);
+ logg("ERROR: %s", logbuf);
+ }
+ return;
+ }
+
+ if(mprintf_stdout)
+ fd = stdout;
+ else
+ fd = stderr;
+
+/* legend:
+ * ! - error
+ * @ - error with logging
+ * ...
+ */
+
+/*
+ * ERROR WARNING STANDARD
+ * normal yes yes yes
+ *
+ * verbose yes yes yes
+ *
+ * quiet yes no no
+ */
+
+
+ va_start(args, str);
+
+ if(*str == '!') {
+ fprintf(fd, "ERROR: ");
+ vfprintf(fd, ++str, args);
+ } else if(*str == '@') {
+ fprintf(fd, "ERROR: ");
+ vfprintf(fd, ++str, args);
+#ifdef NO_SNPRINTF
+ vsprintf(logbuf, str, args);
+#else
+ vsnprintf(logbuf, sizeof(logbuf), str, args);
+#endif
+ logg("ERROR: %s", logbuf);
+ } else if(!mprintf_quiet) {
+ if(*str == '^') {
+ fprintf(fd, "WARNING: ");
+ vfprintf(fd, ++str, args);
+ } else if(*str == '*') {
+ if(mprintf_verbose)
+ vfprintf(fd, ++str, args);
+ } else vfprintf(fd, str, args);
+ }
+
+ va_end(args);
+
+ if(fd == stdout)
+ fflush(stdout);
+
+}
+
+struct facstruct {
+ const char *name;
+ int code;
+};
+
+#if defined(USE_SYSLOG) && !defined(C_AIX)
+static const struct facstruct facilitymap[] = {
+#ifdef LOG_AUTH
+ { "LOG_AUTH", LOG_AUTH },
+#endif
+#ifdef LOG_AUTHPRIV
+ { "LOG_AUTHPRIV", LOG_AUTHPRIV },
+#endif
+#ifdef LOG_CRON
+ { "LOG_CRON", LOG_CRON },
+#endif
+#ifdef LOG_DAEMON
+ { "LOG_DAEMON", LOG_DAEMON },
+#endif
+#ifdef LOG_FTP
+ { "LOG_FTP", LOG_FTP },
+#endif
+#ifdef LOG_KERN
+ { "LOG_KERN", LOG_KERN },
+#endif
+#ifdef LOG_LPR
+ { "LOG_LPR", LOG_LPR },
+#endif
+#ifdef LOG_MAIL
+ { "LOG_MAIL", LOG_MAIL },
+#endif
+#ifdef LOG_NEWS
+ { "LOG_NEWS", LOG_NEWS },
+#endif
+#ifdef LOG_AUTH
+ { "LOG_AUTH", LOG_AUTH },
+#endif
+#ifdef LOG_SYSLOG
+ { "LOG_SYSLOG", LOG_SYSLOG },
+#endif
+#ifdef LOG_USER
+ { "LOG_USER", LOG_USER },
+#endif
+#ifdef LOG_UUCP
+ { "LOG_UUCP", LOG_UUCP },
+#endif
+#ifdef LOG_LOCAL0
+ { "LOG_LOCAL0", LOG_LOCAL0 },
+#endif
+#ifdef LOG_LOCAL1
+ { "LOG_LOCAL1", LOG_LOCAL1 },
+#endif
+#ifdef LOG_LOCAL2
+ { "LOG_LOCAL2", LOG_LOCAL2 },
+#endif
+#ifdef LOG_LOCAL3
+ { "LOG_LOCAL3", LOG_LOCAL3 },
+#endif
+#ifdef LOG_LOCAL4
+ { "LOG_LOCAL4", LOG_LOCAL4 },
+#endif
+#ifdef LOG_LOCAL5
+ { "LOG_LOCAL5", LOG_LOCAL5 },
+#endif
+#ifdef LOG_LOCAL6
+ { "LOG_LOCAL6", LOG_LOCAL6 },
+#endif
+#ifdef LOG_LOCAL7
+ { "LOG_LOCAL7", LOG_LOCAL7 },
+#endif
+ { NULL, -1 }
+};
+
+int logg_facility(const char *name)
+{
+ int i;
+
+ for(i = 0; facilitymap[i].name; i++)
+ if(!strcmp(facilitymap[i].name, name))
+ return facilitymap[i].code;
+
+ return -1;
+}
+#endif