totday.c

   1 /*
   2  * SARG Squid Analysis Report Generator      http://sarg.sourceforge.net
   3  *                                                            1998, 2011
   4  *
   5  * SARG donations:
   6  *      please look at http://sarg.sourceforge.net/donations.php
   7  * Support:
   8  *     http://sourceforge.net/projects/sarg/forums/forum/363374
   9  * ---------------------------------------------------------------------
  10  *
  11  *  This program is free software; you can redistribute it and/or modify
  12  *  it under the terms of the GNU General Public License as published by
  13  *  the Free Software Foundation; either version 2 of the License, or
  14  *  (at your option) any later version.
  15  *
  16  *  This program is distributed in the hope that it will be useful,
  17  *  but WITHOUT ANY WARRANTY; without even the implied warranty of
  18  *  MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
  19  *  GNU General Public License for more details.
  20  *
  21  *  You should have received a copy of the GNU General Public License
  22  *  along with this program; if not, write to the Free Software
  23  *  Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA 02111, USA.
  24  *
  25  */
  26
  27 #include "include/conf.h"
  28 #include "include/defs.h"
  29
  30 void day_totalize(const char *tmp, const struct userinfostruct *uinfo)
  31 {
  32         FILE *fp_in, *fp_ou;
  33
  34         char buf[200];
  35         char date[20];
  36         long long int hour;
  37         long long int bytes;
  38         long long int elap;
  39         long long int tbytes[MAX_DATETIME_DAYS*24];
  40         long long int telap[MAX_DATETIME_DAYS*24];
  41         int day,month,year;
  42         int daylist[MAX_DATETIME_DAYS];
  43         int ndaylist;
  44         int daynum;
  45         int dayidx;
  46         int i;
  47         char wdirname[2048];
  48         char arqout[2048];
  49         char colsep;
  50         struct getwordstruct gwarea;
  51
  52         if((ReportType & REPORT_TYPE_USERS_SITES) == 0) return;
  53         if (datetimeby==0) return;
  54
  55         if (snprintf(wdirname,sizeof(wdirname),"%s/%s.htmp",tmp,uinfo->filename)>=sizeof(wdirname)) {
  56                 debuga(_("File name too long: %s/%s%s\n"),tmp,uinfo->filename,".htmp");
  57                 exit(EXIT_FAILURE);
  58         }
  59         if (snprintf(arqout,sizeof(arqout),"%s/%s.day",tmp,uinfo->filename)>=sizeof(arqout)) {
  60                 debuga(_("File name too long: %s/%s%s\n"),tmp,uinfo->filename,".day");
  61                 exit(EXIT_FAILURE);
  62         }
  63
  64         if((fp_in=fopen(wdirname,"r"))==NULL) {
  65                 debuga(_("(totday) Cannot open log file %s\n"),wdirname);
  66                 exit(EXIT_FAILURE);
  67         }
  68
  69         memset(tbytes,0,sizeof(tbytes));
  70         memset(telap,0,sizeof(tbytes));
  71         ndaylist=0;
  72
  73         while(fgets(buf,sizeof(buf),fp_in)!=NULL) {
  74                 fixendofline(buf);
  75                 getword_start(&gwarea,buf);
  76                 if (getword(date,sizeof(date),&gwarea,'\t')<0 || getword_atoll(&hour,&gwarea,'\t')<0) {
  77                         debuga(_("There is a broken record or garbage in file %s\n"),wdirname);
  78                         exit(EXIT_FAILURE);
  79                 }
  80                 if (sscanf(date,"%d/%d/%d",&day,&month,&year)!=3) continue;
  81                 if (day<1 || day>31 || month<1 || month>12 || year>9999) continue;
  82                 if (hour<0 || hour>=24) continue;
  83                 daynum=(year*10000)+(month*100)+day;
  84                 for (dayidx=0 ; dayidx<ndaylist && daynum!=daylist[dayidx] ; dayidx++);
  85                 if (dayidx>=ndaylist) {
  86                         if (dayidx>=sizeof(daylist)/sizeof(*daylist)) {
  87                                 debuga(_("Too many different dates in %s\n"),wdirname);
  88                                 exit(EXIT_FAILURE);
  89                         }
  90                         daylist[ndaylist++]=daynum;
  91                 }
  92                 i=dayidx*24+hour;
  93                 if ((datetimeby & DATETIME_BYTE)!=0) {
  94                         colsep=((datetimeby & DATETIME_ELAP)!=0) ? '\t' : '\0';
  95                         if (getword_atoll(&bytes,&gwarea,colsep)<0) {
  96                                 debuga(_("Invalid number of bytes in file %s\n"),wdirname);
  97                                 exit(EXIT_FAILURE);
  98                         }
  99                         tbytes[i]+=bytes;
 100                 }
 101                 if ((datetimeby & DATETIME_ELAP)!=0) {
 102                         if (getword_atoll(&elap,&gwarea,'\0')<0) {
 103                                 debuga(_("Invalid elapsed time in file %s\n"),wdirname);
 104                                 exit(EXIT_FAILURE);
 105                         }
 106                         telap[i]+=elap;
 107                 }
 108         }
 109         fclose(fp_in);
 110
 111         if((fp_ou=fopen(arqout,"w"))==NULL) {
 112                 debuga(_("(totday) Cannot open log file %s\n"),arqout);
 113                 exit(EXIT_FAILURE);
 114         }
 115
 116         for (i=0 ; i<sizeof(tbytes)/sizeof(*tbytes) ; i++) {
 117                 if (tbytes[i]==0 && telap[i]==0) continue;
 118                 dayidx=i/24;
 119                 if (dayidx>=sizeof(daylist)/sizeof(*daylist)) continue;
 120                 hour=i%24;
 121                 daynum=daylist[dayidx];
 122                 day=daynum%100;
 123                 month=(daynum/100)%100;
 124                 year=daynum/10000;
 125                 fprintf(fp_ou,"%d/%d/%d\t%d",day,month,year,(int)hour);
 126                 if ((datetimeby & DATETIME_BYTE)!=0) fprintf(fp_ou,"\t%"PRIu64"",(uint64_t)tbytes[i]);
 127                 if ((datetimeby & DATETIME_ELAP)!=0) fprintf(fp_ou,"\t%"PRIu64"",(uint64_t)telap[i]);
 128                 fputs("\n",fp_ou);
 129         }
 130
 131         if (fclose(fp_ou)==EOF) {
 132                 debuga(_("Failed to close file %s - %s\n"),arqout,strerror(errno));
 133                 exit(EXIT_FAILURE);
 134         }
 135
 136         if (unlink(wdirname)==-1) {
 137                 debuga(_("Cannot delete temporary file %s - %s\n"),wdirname,strerror(errno));
 138                 exit(EXIT_FAILURE);
 139         }
 140         return;
 141 }