LDAP usertab feature added

[thirdparty/sarg.git] / realtime.c

/*
 * AUTHOR: Pedro Lineu Orso                      orso@penguintech.com.br
 *                                                            1998, 2005
 * SARG Squid Analysis Report Generator      http://sarg.sourceforge.net
 *
 * SARG donations:
 *      please look at http://sarg.sourceforge.net/donations.php
 * ---------------------------------------------------------------------
 *
 *  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., 59 Temple Place, Suite 330, Boston, MA 02111, USA.
 *
 */

#include "include/conf.h"
#include "include/defs.h"

static int getdata(char*, FILE*);
static void datashow(const char *);
static void getlog(void);
static void header(void);

char dat[128];
char tim[128];
char typ[128];
char ouser[MAXLEN]="";
char ourl[MAXLEN]="";

void realtime(void)
{

   getlog();

}

static void getlog(void)
{
   FILE *tmp, *fp;
   char template1[255]="/var/tmp/sargtpl1.XXXXXX";
   char template2[255]="/var/tmp/sargtpl2.XXXXXX";
   char cmd[512];
   char buf[MAXLEN];
   int  fd1,fd2;
   int cstatus;

   read_usertab(UserTabFile);

   fd1 = mkstemp(template1);
   fd2 = mkstemp(template2);

   if((fd1 == -1 ) || ((tmp = fdopen (fd1, "w+" )) == NULL)  ) {    /* failure, bail out */
       fprintf(stderr, "SARG: (realtime) mkstemp error - %s\n",strerror(errno));
       exit(1);
   }

   sprintf(cmd,"tail -%d %s",realtime_access_log_lines,AccessLog);
   fp = popen(cmd, "r");
   while(fgets(buf,sizeof(buf),fp) != NULL )
      if (getdata(buf,tmp)<0) {
         fprintf(stderr,"SARG: Maybe a broken record or garbage was returned by %s.\n",cmd);
         exit(1);
      }
   pclose(fp);
   fclose(tmp);

   sprintf(cmd,"sort -r -k 1,1 -k 2,2 -o \"%s\" \"%s\"",template2,template1);
   cstatus=system(cmd);
   if (!WIFEXITED(cstatus) || WEXITSTATUS(cstatus)) {
      fprintf(stderr, "SARG: sort command return status %d\n",WEXITSTATUS(cstatus));
      fprintf(stderr, "SARG: sort command: %s\n",cmd);
      exit(1);
   }
   unlink(template1);
   datashow(template2);
}

static int getdata(char *rec, FILE *ftmp)
{
   time_t tt;
   struct tm *t;
   char tbuf[128];

   if (getword(dat,sizeof(dat),rec,' ')<0) {
      fprintf(stderr,"SARG: The time stamp at column 1 is too long.\n");
      return(-1);
   }
   if (getword(warea,sizeof(warea),rec,' ')<0) {
      fprintf(stderr,"SARG: The connection duration at column 2 is too long.\n");
      return(-1);
   }
   while(strcmp(warea,"") == 0 && strlen(rec) > 0)
      if (getword(warea,sizeof(warea),rec,' ')<0) {
         return(-1);
      }
   if (getword(ip,sizeof(ip),rec,' ')<0) {
      fprintf(stderr,"SARG: The IP address at column 3 is too long.\n");
      return(-1);
   }
   if (getword(warea,sizeof(warea),rec,' ')<0) {
      fprintf(stderr,"SARG: The status at column 4 is too long.\n");
      return(-1);
   }
   if (getword(warea,sizeof(warea),rec,' ')<0) {
      fprintf(stderr,"SARG: The size at column 5 is too long.\n");
      return(-1);
   }
   if (getword(typ,sizeof(typ),rec,' ')<0) {
      fprintf(stderr,"SARG: The action at column 6 is too long.\n");
      return(-1);
   }
   if(strncmp(typ,"CONNECT",7) == 0) {
      if (getword(url,sizeof(url),rec,' ')<0) {
         return(-1);
      }
      if (getword(user,sizeof(user),rec,' ')<0) {
         return(-1);
      }
   }else {
      if (getword(url,sizeof(url),rec,'/')<0) {
         fprintf(stderr,"SARG: The URL at column 7 is too long.\n");
         return(-1);
      }
      if (getword(url,sizeof(url),rec,'/')<0) {
         fprintf(stderr,"SARG: The URL at column 7 is too long.\n");
         return(-1);
      }
      if (getword(url,sizeof(url),rec,'/')<0) {
         fprintf(stderr,"SARG: The URL at column 7 is too long.\n");
         return(-1);
      }
      if (getword(user,sizeof(user),rec,' ')<0) {
         fprintf(stderr,"SARG: The data at column 8 is too long.\n");
         return(-1);
      }
      if (getword(user,sizeof(user),rec,' ')<0) {
         fprintf(stderr,"SARG: The user at column 9 is too long.\n");
         return(-1);
      }
   }

   if(strncmp(user,"-",1) == 0 && strcmp(RealtimeUnauthRec,"ignore") == 0)
      return(0);

   tt=atoi(dat);
   t=localtime(&tt);
   if(strncmp(DateFormat,"u",1) == 0)
      strftime(tbuf, sizeof(tbuf), "%Y-%m-%d %H:%M", t);
   else if(strncmp(DateFormat,"e",1) == 0)
      strftime(tbuf, sizeof(tbuf), "%d-%m-%Y %H:%M", t);

   fprintf(ftmp,"%s\t%s\t%s\t%s\t%s\n",tbuf,ip,user,url,typ);
   return(0);
}

static void datashow(const char *tmp)
{
   FILE *fin;
   char buf[MAXLEN];

   if((fin=fopen(tmp,"r"))==NULL) {
      fprintf(stderr, "SARG: (realtime) open error %s - %s\n",tmp,strerror(errno));
      exit(1);
   }

   header();

   while(fgets(buf, sizeof(buf), fin)) {
      fixendofline(buf);
      if (getword(dat,sizeof(dat),buf,'\t')<0) {
         printf("SARG: Maybe you have a broken record or garbage in your %s file.\n",tmp);
         exit(1);
      }
      if (getword(tim,sizeof(tim),buf,'\t')<0) {
         printf("SARG: Maybe you have a broken record or garbage in your %s file.\n",tmp);
         exit(1);
      }
      if (getword(ip,sizeof(ip),buf,'\t')<0) {
         printf("SARG: Maybe you have a broken record or garbage in your %s file.\n",tmp);
         exit(1);
      }
      if (getword(user,sizeof(user),buf,'\t')<0) {
         printf("SARG: Maybe you have a broken record or garbage in your %s file.\n",tmp);
         exit(1);
      }
      if(strlen(dat) < 3 || strlen(user) < 1) continue;
      if (getword(url,sizeof(url),buf,'\t')<0) {
         printf("SARG: Maybe you have a broken record or garbage in your %s file.\n",tmp);
         exit(1);
      }
      if (getword(typ,sizeof(typ),buf,'\t')<0) {
         printf("SARG: Maybe you have a broken record or garbage in your %s file.\n",tmp);
         exit(1);
      }
      if(strstr(RealtimeTypes,typ) == 0)
         continue;

      if(strcmp(ouser,user) == 0 && strcmp(ourl,url) == 0)
         continue;

      strcpy(u2,user);
      if(strcmp(Ip2Name,"yes") == 0)
         ip2name(u2,sizeof(u2));
//      get_usertab_name(u2,name,sizeof(name));
      user_find(name, u2);

      if(dotinuser && strchr(name,'_')) {
         subs(name,sizeof(name),"_",".");
      }

      printf("<tr><td class=\"data\">%s %s</td><td class=\"data3\">%s</td><td class=\"data3\">%s</td><td class=\"data3\">%s</td><td class=\"data2\"><a href=\"http://%s\">%s</td></tr>\n",dat,tim,ip,name,typ,url,url);
      strcpy(ouser,user);
      strcpy(ourl,url);
   }

   puts("</table>\n</div>\n</body>\n</html>\n");
   fclose(fin);
   unlink(tmp);
   fflush(NULL);

}

static void header(void)
{
   puts("<!DOCTYPE HTML PUBLIC \"-//W3C//DTD HTML 4.01 Transitional//EN\"");
   puts(" \"http://www.w3.org/TR/html4/loose.dtd\">\n");
   puts("<html>\n");
   puts("<head>\n");
   if(realtime_refresh)
      printf("  <meta http-equiv=refresh content=\"%d\" url=\"sarg-php/sarg-realtime.php\"; charset=\"%s\">\n",realtime_refresh,CharSet);
   else
      printf("  <meta http-equiv=\"Content-Type\" content=\"text/html; charset=%s\">\n",CharSet);
   css(stdout);
   puts("</head>\n");
   printf(buf,"<body style=\"font-family:%s;font-size:%s;background-color:%s;background-image:url(%s)\">\n",FontFace,TitleFontSize,BgColor,BgImage);
   puts("<div align=\"center\"><table cellpadding=\"1\" cellspacing=\"1\">\n");
   printf("<tr><th class=\"title2\" colspan=\"10\">SARG %s</th></tr>\n",text[134]);
   printf("<tr><th class=\"text\" colspan=\"10\">%s: %d s</th></tr>\n",text[136],realtime_refresh);
   printf("<tr><th class=\"header3\">%s</th><th class=\"header3\">%s</th><th class=\"header3\">%s</th><th class=\"header3\">%s</th><th class=\"header\">%s</th></tr>\n",text[110],text[111],text[98],text[135],text[91]);
}