All the sprintf/fputs have been replaced by fprintf to avoid an intermediary buffer...

[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"

int getdata(char *, FILE *);
void datashow(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, *fp_usr;
   char template1[255]="/var/tmp/sargtpl1.XXXXXX";
   char template2[255]="/var/tmp/sargtpl2.XXXXXX";
   char cmd[512];
   char buf[MAXLEN];
   int  fd1,fd2,nreg;
   int cstatus;

   if(UserTabFile[0] != '\0') {
      if(debug) {
         sprintf(msg,"%s: %s",text[86],UserTabFile);
         debuga(msg);
      }
      if((fp_usr=fopen(UserTabFile,"r"))==NULL) {
        fprintf(stderr, "SARG: (realtime) %s: %s - %s\n",text[45],UserTabFile,strerror(errno));
        exit(1);
      }
      nreg = lseek(fileno(fp_usr), 0, SEEK_END);
      lseek(fileno(fp_usr), 0, 0);
      if((userfile=(char *) malloc(nreg+100))==NULL){
         fprintf(stderr, "SARG ERROR: %s",text[87]);
         exit(1);
      }
      bzero(userfile,nreg+100);
      strncat(userfile,":",1);
      z1=0;
      z2=1;
      while(fgets(buf,sizeof(buf),fp_usr)!=NULL){
        buf[strlen(buf)-1]='\0';
        if(strstr(buf,"\r") != 0) buf[strlen(buf)-1]='\0';
        if (getword(bufy,sizeof(bufy),buf,' ')<0) {
           printf("SARG: Maybe you have a broken record or garbage in your %s file.\n",UserTabFile);
           exit(1);
        }
        for(z1=0; z1<=strlen(bufy); z1++) {
           userfile[z2]=bufy[z1];
           z2++;
        }
        strncat(userfile,":",1);
        for(z1=0; z1<=strlen(buf); z1++) {
           userfile[z2]=buf[z1];
           z2++;
        }
        strncat(userfile,":",1);
      }
      fclose(fp_usr);
   }

   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);
}

int getdata(char *rec, FILE *ftmp)
{
   time_t tt;
   struct tm *t;

   if (getword3(dat,sizeof(dat),rec,' ')<0) {
      fprintf(stderr,"SARG: The time stamp at column 1 is too long.\n");
      return(-1);
   }
   if (getword3(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 (getword3(warea,sizeof(warea),rec,' ')<0) {
         return(-1);
      }
   if (getword3(ip,sizeof(ip),rec,' ')<0) {
      fprintf(stderr,"SARG: The IP address at column 3 is too long.\n");
      return(-1);
   }
   if (getword3(warea,sizeof(warea),rec,' ')<0) {
      fprintf(stderr,"SARG: The status at column 4 is too long.\n");
      return(-1);
   }
   if (getword3(warea,sizeof(warea),rec,' ')<0) {
      fprintf(stderr,"SARG: The size at column 5 is too long.\n");
      return(-1);
   }
   if (getword3(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 (getword3(url,sizeof(url),rec,' ')<0) {
         return(-1);
      }
      if (getword3(user,sizeof(user),rec,' ')<0) {
         return(-1);
      }
   }else {
      if (getword3(url,sizeof(url),rec,'/')<0) {
         fprintf(stderr,"SARG: The URL at column 7 is too long.\n");
         return(-1);
      }
      if (getword3(url,sizeof(url),rec,'/')<0) {
         fprintf(stderr,"SARG: The URL at column 7 is too long.\n");
         return(-1);
      }
      if (getword3(url,sizeof(url),rec,'/')<0) {
         fprintf(stderr,"SARG: The URL at column 7 is too long.\n");
         return(-1);
      }
      if (getword3(user,sizeof(user),rec,' ')<0) {
         fprintf(stderr,"SARG: The data at column 8 is too long.\n");
         return(-1);
      }
      if (getword3(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, 127, "%Y-%m-%d %H:%M", t);
   else if(strncmp(DateFormat,"e",1) == 0)
      strftime(tbuf, 127, "%d-%m-%Y %H:%M", t);

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

void datashow(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, MAXLEN, fin)) {
      buf[strlen(buf)-1]='\0';
      if (getword3(dat,sizeof(dat),buf,' ')<0) {
         printf("SARG: Maybe you have a broken record or garbage in your %s file.\n",tmp);
         exit(1);
      }
      if (getword3(tim,sizeof(tim),buf,' ')<0) {
         printf("SARG: Maybe you have a broken record or garbage in your %s file.\n",tmp);
         exit(1);
      }
      if (getword3(ip,sizeof(ip),buf,' ')<0) {
         printf("SARG: Maybe you have a broken record or garbage in your %s file.\n",tmp);
         exit(1);
      }
      if (getword3(user,sizeof(user),buf,' ')<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 (getword3(url,sizeof(url),buf,' ')<0) {
         printf("SARG: Maybe you have a broken record or garbage in your %s file.\n",tmp);
         exit(1);
      }
      if (getword3(typ,sizeof(typ),buf,' ')<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));
      if(UserTabFile[0] != '\0') {
         sprintf(warea,":%s:",u2);
         if((str=(char *) strstr(userfile,warea)) != (char *) NULL ) {
            z1=0;
            str2=(char *) strstr(str+1,":");
            str2++;
            bzero(name, MAXLEN);
            while(str2[z1] != ':') {
               name[z1]=str2[z1];
               z1++;
            }
         } else strcpy(name,u2);
      } else strcpy(name,u2);

      if(dotinuser && strstr(name,"_")) {
         str2=(char *)subs(name,"_",".");
         strcpy(name,str2);
      }

      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</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("<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]);
}