shared/output.c
afb48b28
 /*
e1187134
  *  Copyright (C) 2002 - 2005 Tomasz Kojm <tkojm@clamav.net>
afb48b28
  *
  *  This program is free software; you can redistribute it and/or modify
bb34cb31
  *  it under the terms of the GNU General Public License version 2 as
  *  published by the Free Software Foundation.
afb48b28
  *
  *  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
48b7b4a7
  *  Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston,
  *  MA 02110-1301, USA.
afb48b28
  */
ace24e1f
 #ifdef _MSC_VER
 #include <windows.h>
 #include <winsock.h>
 #endif
 
afb48b28
 
 #if HAVE_CONFIG_H
 #include "clamav-config.h"
 #endif
 
 #include <stdio.h>
 #include <stdarg.h>
 #include <stdlib.h>
 #include <string.h>
 #include <ctype.h>
ace24e1f
 #ifdef HAVE_UNISTD_H
afb48b28
 #include <unistd.h>
ace24e1f
 #endif
afb48b28
 #include <fcntl.h>
 #include <time.h>
 #include <sys/stat.h>
 #include <errno.h>
ace24e1f
 #ifndef C_WINDOWS
afb48b28
 #include <sys/time.h>
 #include <sys/socket.h>
ace24e1f
 #endif
afb48b28
 #if HAVE_SYS_TYPES_H
 #include <sys/types.h>
 #endif
 
 #if defined(USE_SYSLOG) && !defined(C_AIX)
 #include <syslog.h>
 #endif
 
 #include "output.h"
 
2ed08e8a
 #ifdef CL_NOTHREADS
 #undef CL_THREAD_SAFE
 #endif
 
afb48b28
 #ifdef CL_THREAD_SAFE
 #include <pthread.h>
 pthread_mutex_t logg_mutex = PTHREAD_MUTEX_INITIALIZER;
 #endif
 
0ae41a2d
 #ifdef  C_LINUX
 #include <libintl.h>
 #include <locale.h>
 
 #define gettext_noop(s) s
 #define _(s)    gettext(s)
 #define N_(s)   gettext_noop(s)
 
 #else
 
 #define _(s)    s
 #define N_(s)   s
 
 #endif
 
afb48b28
 FILE *logg_fd = NULL;
 
7b8edc5c
 short int logg_verbose = 0, logg_lock = 1, logg_time = 0, logg_foreground = 1;
58bcf502
 int logg_size = 0;
 const char *logg_file = NULL;
 #if defined(USE_SYSLOG) && !defined(C_AIX)
0ae41a2d
 short logg_syslog;
58bcf502
 #endif
 
 short int mprintf_disabled = 0, mprintf_verbose = 0, mprintf_quiet = 0,
 	  mprintf_stdout = 0;
 
afb48b28
 int mdprintf(int desc, const char *str, ...)
 {
 	va_list args;
 	char buff[512];
 	int bytes;
 
     va_start(args, str);
2e81b220
     bytes = vsnprintf(buff, sizeof(buff), str, args);
afb48b28
     va_end(args);
2e81b220
 
     if(bytes == -1)
 	return bytes;
 
a889f40e
     if(bytes >= (int) sizeof(buff))
2e81b220
 	bytes = sizeof(buff) - 1;
 
c477e3ba
     return send(desc, buff, bytes, 0);
afb48b28
 }
 
 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, ...)
 {
0ae41a2d
 	va_list args, argscpy, argsout;
ace24e1f
 #ifdef F_WRLCK
afb48b28
 	struct flock fl;
ace24e1f
 #endif
fe7ee98f
 	char *pt, *timestr, vbuff[1025];
afb48b28
 	time_t currtime;
 	struct stat sb;
 	mode_t old_umask;
 
 
fe7ee98f
     va_start(args, str);
04d42cce
     /* va_copy is less portable so we just use va_start once more */
     va_start(argscpy, str);
0ae41a2d
     va_start(argsout, str);
fe7ee98f
 
afb48b28
 #ifdef CL_THREAD_SAFE
234582ae
     pthread_mutex_lock(&logg_mutex);
afb48b28
 #endif
234582ae
     if(logg_file) {
afb48b28
 	if(!logg_fd) {
 	    old_umask = umask(0037);
ace24e1f
 	    if((logg_fd = fopen(logg_file, "at")) == NULL) {
afb48b28
 		umask(old_umask);
 #ifdef CL_THREAD_SAFE
 		pthread_mutex_unlock(&logg_mutex);
 #endif
ce99b521
 		printf("ERROR: Can't open %s in append mode (check permissions!).\n", logg_file);
afb48b28
 		return -1;
 	    } else umask(old_umask);
 
ace24e1f
 #ifdef F_WRLCK
afb48b28
 	    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;
 		}
 	    }
ace24e1f
 #endif
afb48b28
 	}
 
 	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;
 		}
 	    }
 	}
 
eff613a9
 	if(logg_fd) {
             /* 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);
65d08d61
 		timestr = calloc(strlen(pt), 1);
eff613a9
 		strncpy(timestr, pt, strlen(pt) - 1);
 		fprintf(logg_fd, "%s -> ", timestr);
 		free(timestr);
 	    }
afb48b28
 
eff613a9
 	    _(str);
 	    if(*str == '!') {
 		fprintf(logg_fd, "ERROR: ");
fe7ee98f
 		vfprintf(logg_fd, str + 1, args);
eff613a9
 	    } 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 if(*str == '#') {
 		vfprintf(logg_fd, str + 1, args);
 	    } else vfprintf(logg_fd, str, args);
afb48b28
 
 
eff613a9
 	    fflush(logg_fd);
 	}
20de3381
     }
afb48b28
 
 #if defined(USE_SYSLOG) && !defined(C_AIX)
     if(logg_syslog) {
0ae41a2d
         _(str);
04d42cce
 	vsnprintf(vbuff, 1024, str, argscpy);
fe7ee98f
 	vbuff[1024] = 0;
 
 	if(vbuff[0] == '!') {
118920f8
 	    syslog(LOG_ERR, "%s", vbuff + 1);
fe7ee98f
 	} else if(vbuff[0] == '^') {
118920f8
 	    syslog(LOG_WARNING, "%s", vbuff + 1);
fe7ee98f
 	} else if(vbuff[0] == '*') {
afb48b28
 	    if(logg_verbose) {
118920f8
 		syslog(LOG_DEBUG, "%s", vbuff + 1);
afb48b28
 	    }
914ce12d
 	} else if(vbuff[0] == '#') {
118920f8
 	    syslog(LOG_INFO, "%s", vbuff + 1);
 	} else syslog(LOG_INFO, "%s", vbuff);
afb48b28
 
     }
 #endif
 
7b8edc5c
     if(logg_foreground) {
914ce12d
 	_(str);
         vsnprintf(vbuff, 1024, str, argsout);
 	vbuff[1024] = 0;
 	if(vbuff[0] != '#')
118920f8
 	    mprintf("%s", vbuff);
0ae41a2d
     }
 
234582ae
 #ifdef CL_THREAD_SAFE
     pthread_mutex_unlock(&logg_mutex);
 #endif
 
fe7ee98f
     va_end(args);
04d42cce
     va_end(argscpy);
0ae41a2d
     va_end(argsout);
afb48b28
     return 0;
 }
 
 void mprintf(const char *str, ...)
 {
0ae41a2d
 	va_list args;
afb48b28
 	FILE *fd;
118920f8
 	char buff[512];
afb48b28
 
 
0ae41a2d
     if(mprintf_disabled) 
afb48b28
 	return;
 
e1187134
     fd = stdout;
afb48b28
 
 /* legend:
  * ! - error
  * @ - error with logging
  * ...
  */
 
 /*
  *             ERROR    WARNING    STANDARD
e1187134
  * normal      stderr   stderr     stdout
afb48b28
  * 
e1187134
  * verbose     stderr   stderr     stdout
afb48b28
  * 
e1187134
  * quiet       stderr     no         no
afb48b28
  */
 
     va_start(args, str);
118920f8
     vsnprintf(buff, sizeof(buff), str, args);
     va_end(args);
afb48b28
 
4b25a0de
     if(buff[0] == '!') {
e1187134
        if(!mprintf_stdout)
            fd = stderr;
118920f8
 	fprintf(fd, "ERROR: %s", &buff[1]);
4b25a0de
     } else if(buff[0] == '@') {
e1187134
        if(!mprintf_stdout)
            fd = stderr;
118920f8
 	fprintf(fd, "ERROR: %s", &buff[1]);
afb48b28
     } else if(!mprintf_quiet) {
4b25a0de
 	if(buff[0] == '^') {
e1187134
            if(!mprintf_stdout)
                fd = stderr;
118920f8
 	    fprintf(fd, "WARNING: %s", &buff[1]);
4b25a0de
 	} else if(buff[0] == '*') {
afb48b28
 	    if(mprintf_verbose)
118920f8
 		fprintf(fd, "%s", &buff[1]);
 	} else fprintf(fd, "%s", buff);
afb48b28
     }
 
     if(fd == stdout)
 	fflush(stdout);
 }
c695dab4
 
 struct facstruct {
     const char *name;
     int code;
 };
 
 #if defined(USE_SYSLOG) && !defined(C_AIX)
 static const struct facstruct facilitymap[] = {
45ae238c
 #ifdef LOG_AUTH
c695dab4
     { "LOG_AUTH",	LOG_AUTH },
45ae238c
 #endif
 #ifdef LOG_AUTHPRIV
c695dab4
     { "LOG_AUTHPRIV",	LOG_AUTHPRIV },
45ae238c
 #endif
 #ifdef LOG_CRON
c695dab4
     { "LOG_CRON",	LOG_CRON },
45ae238c
 #endif
 #ifdef LOG_DAEMON
c695dab4
     { "LOG_DAEMON",	LOG_DAEMON },
45ae238c
 #endif
 #ifdef LOG_FTP
c695dab4
     { "LOG_FTP",	LOG_FTP },
45ae238c
 #endif
 #ifdef LOG_KERN
c695dab4
     { "LOG_KERN",	LOG_KERN },
45ae238c
 #endif
 #ifdef LOG_LPR
c695dab4
     { "LOG_LPR",	LOG_LPR },
45ae238c
 #endif
 #ifdef LOG_MAIL
c695dab4
     { "LOG_MAIL",	LOG_MAIL },
45ae238c
 #endif
 #ifdef LOG_NEWS
c695dab4
     { "LOG_NEWS",	LOG_NEWS },
45ae238c
 #endif
 #ifdef LOG_AUTH
c695dab4
     { "LOG_AUTH",	LOG_AUTH },
45ae238c
 #endif
 #ifdef LOG_SYSLOG
c695dab4
     { "LOG_SYSLOG",	LOG_SYSLOG },
45ae238c
 #endif
 #ifdef LOG_USER
c695dab4
     { "LOG_USER",	LOG_USER },
45ae238c
 #endif
 #ifdef LOG_UUCP
c695dab4
     { "LOG_UUCP",	LOG_UUCP },
45ae238c
 #endif
 #ifdef LOG_LOCAL0
c695dab4
     { "LOG_LOCAL0",	LOG_LOCAL0 },
45ae238c
 #endif
 #ifdef LOG_LOCAL1
c695dab4
     { "LOG_LOCAL1",	LOG_LOCAL1 },
45ae238c
 #endif
 #ifdef LOG_LOCAL2
c695dab4
     { "LOG_LOCAL2",	LOG_LOCAL2 },
45ae238c
 #endif
 #ifdef LOG_LOCAL3
c695dab4
     { "LOG_LOCAL3",	LOG_LOCAL3 },
45ae238c
 #endif
 #ifdef LOG_LOCAL4
c695dab4
     { "LOG_LOCAL4",	LOG_LOCAL4 },
45ae238c
 #endif
 #ifdef LOG_LOCAL5
c695dab4
     { "LOG_LOCAL5",	LOG_LOCAL5 },
45ae238c
 #endif
 #ifdef LOG_LOCAL6
c695dab4
     { "LOG_LOCAL6",	LOG_LOCAL6 },
45ae238c
 #endif
 #ifdef LOG_LOCAL7
c695dab4
     { "LOG_LOCAL7",	LOG_LOCAL7 },
45ae238c
 #endif
c695dab4
     { 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