email.c

   1 /*
   2  * SARG Squid Analysis Report Generator      http://sarg.sourceforge.net
   3  *                                                            1998, 2010
   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 int geramail(const char *dirname, int debug, const char *outdir, const char *email, const char *TempDir)
  31 {
  32         FILE *fp_in, *fp_top1, *fp_top2, *fp_top3;
  33         long long int ttnbytes=0, ttnacc=0, tnacc=0;
  34         long long int tnbytes=0, ttnelap=0, tnelap=0;
  35         long long int nacc, nbytes, elap;
  36         long long int avgacc, avgelap;
  37         double perc=0.00;
  38         double perc2=0.00;
  39         int posicao=0;
  40         char olduser[MAX_USER_LEN], csort[MAXLEN];
  41         char wger[MAXLEN], top1[MAXLEN], top2[MAXLEN], top3[MAXLEN], user[MAX_USER_LEN];
  42         char strip1[MAXLEN], strip2[MAXLEN], strip3[MAXLEN], strip4[MAXLEN], strip5[MAXLEN], strip6[MAXLEN], strip7[MAXLEN];
  43         char *buf;
  44         char warea[MAXLEN];
  45         int totuser=0;
  46         time_t t;
  47         struct tm *local;
  48         int cstatus;
  49         struct getwordstruct gwarea;
  50         struct generalitemstruct item;
  51         longline line;
  52         const struct userinfostruct *uinfo;
  53
  54         snprintf(wger,sizeof(wger),"%s/sarg-general",dirname);
  55         if((fp_in=fopen(wger,"r"))==NULL) {
  56                 debuga(_("(email) Cannot open file %s\n"),wger);
  57                 exit(EXIT_FAILURE);
  58         }
  59
  60         snprintf(top1,sizeof(top1),"%s/top",dirname);
  61         if((fp_top1=fopen(top1,"w"))==NULL) {
  62                 debuga(_("(email) Cannot open file %s\n"),top1);
  63                 exit(EXIT_FAILURE);
  64         }
  65
  66         snprintf(top2,sizeof(top2),"%s/top.tmp",dirname);
  67         if((fp_top2=fopen(top2,"w"))==NULL) {
  68                 debuga(_("(email) Cannot open file %s\n"),top2);
  69                 exit(EXIT_FAILURE);
  70         }
  71
  72         olduser[0]='\0';
  73         totuser=0;
  74
  75         if ((line=longline_create())==NULL) {
  76                 debuga(_("Not enough memory to read file %s\n"),wger);
  77                 exit(EXIT_FAILURE);
  78         }
  79
  80         while((buf=longline_read(fp_in,line))!=NULL) {
  81                 ger_read(buf,&item,wger);
  82                 if(item.total) continue;
  83                 if(strcmp(olduser,item.user) != 0)
  84                 {
  85                         totuser++;
  86
  87                         if (olduser[0] != '\0') {
  88 #if defined(__FreeBSD__)
  89                                 fprintf(fp_top2,"%s\t%qu\t%qu\t%qu\n",olduser,tnbytes,tnacc,tnelap);
  90 #else
  91                                 fprintf(fp_top2,"%s\t%"PRIu64"\t%"PRIu64"\t%"PRIu64"\n",olduser,(uint64_t)tnbytes,(uint64_t)tnacc,(uint64_t)tnelap);
  92 #endif
  93                                 ttnbytes+=tnbytes;
  94                                 ttnacc+=tnacc;
  95                                 ttnelap+=tnelap;
  96                         }
  97                         strcpy(olduser,item.user);
  98                         tnbytes=0;
  99                         tnacc=0;
 100                         tnelap=0;
 101                 }
 102
 103                 tnbytes+=item.nbytes;
 104                 tnacc+=item.nacc;
 105                 tnelap+=item.nelap;
 106         }
 107         fclose(fp_in);
 108         longline_destroy(&line);
 109
 110         if (olduser[0] != '\0') {
 111 #if defined(__FreeBSD__)
 112                 fprintf(fp_top2,"%s\t%qu\t%qu\t%qu\n",olduser,tnbytes,tnacc,tnelap);
 113 #else
 114                 fprintf(fp_top2,"%s\t%"PRIu64"\t%"PRIu64"\t%"PRIu64"\n",olduser,(uint64_t)tnbytes,(uint64_t)tnacc,(uint64_t)tnelap);
 115 #endif
 116                 ttnbytes+=tnbytes;
 117                 ttnacc+=tnacc;
 118                 ttnelap+=tnelap;
 119         }
 120
 121         fclose(fp_top2);
 122
 123         sprintf(csort,"sort -n -T \"%s\" -r -k 2,2 -o \"%s\" \"%s\"", TempDir, top1, top2);
 124         cstatus=system(csort);
 125         if (!WIFEXITED(cstatus) || WEXITSTATUS(cstatus)) {
 126                 debuga(_("sort command return status %d\n"),WEXITSTATUS(cstatus));
 127                 debuga(_("sort command: %s\n"),csort);
 128                 exit(EXIT_FAILURE);
 129         }
 130
 131         unlink(top2);
 132
 133         if((fp_top1=fopen(top1,"r"))==NULL) {
 134                 debuga(_("(email) Cannot open file %s\n"),top1);
 135                 exit(EXIT_FAILURE);
 136         }
 137
 138         snprintf(top3,sizeof(top3),"%s/report",dirname);
 139         if((fp_top3=fopen(top3,"w"))==NULL) {
 140                 debuga(_("(email) Cannot open file %s\n"),top3);
 141                 exit(EXIT_FAILURE);
 142         }
 143
 144         strcpy(strip1,_("Squid User Access Report"));
 145         strip_latin(strip1);
 146         fprintf(fp_top3,"%s\n",strip1);
 147
 148         strcpy(strip1,_("Decreasing Access (bytes)"));
 149         strip_latin(strip1);
 150         fprintf(fp_top3,"%s\n",strip1);
 151
 152         strcpy(strip1,_("Period"));
 153         strip_latin(strip1);
 154         fprintf(fp_top3,"%s %s\n\n",strip1,period.text);
 155
 156         strcpy(strip1,_("NUM"));
 157         strip_latin(strip1);
 158         strcpy(strip2,_("USERID"));
 159         strip_latin(strip2);
 160         strcpy(strip3,_("CONNECT"));
 161         strip_latin(strip3);
 162         strcpy(strip4,_("BYTES"));
 163         strip_latin(strip4);
 164         strcpy(strip5,_("ELAPSED TIME"));
 165         strip_latin(strip5);
 166         strcpy(strip6,_("MILLISEC"));
 167         strip_latin(strip6);
 168         strcpy(strip7,_("TIME"));
 169         strip_latin(strip7);
 170
 171         fprintf(fp_top3,"%-7s %-20s %-8s %-15s %%%-6s %-10s %-10s %%%-7s\n------- -------------------- -------- --------------- ------- ---------- ---------- -------\n",strip1,strip2,strip3,strip4,strip4,strip5,strip6,strip7);
 172
 173         while(fgets(warea,sizeof(warea),fp_top1))
 174         {
 175                 fixendofline(warea);
 176                 getword_start(&gwarea,warea);
 177                 if (getword(user,sizeof(user),&gwarea,'\t')<0) {
 178                         debuga(_("There is an invalid user ID in file %s\n"),top1);
 179                         exit(EXIT_FAILURE);
 180                 }
 181                 if (getword_atoll(&nbytes,&gwarea,'\t')<0) {
 182                         debuga(_("There is an invalid number of bytes in file %s\n"),top1);
 183                         exit(EXIT_FAILURE);
 184                 }
 185                 if (getword_atoll(&nacc,&gwarea,'\t')<0) {
 186                         debuga(_("There is an invalid number of access in file %s\n"),top1);
 187                         exit(EXIT_FAILURE);
 188                 }
 189                 if (getword_atoll(&elap,&gwarea,'\0')<0) {
 190                         debuga(_("There is an invalid elapsed time in file %s\n"),top1);
 191                         exit(EXIT_FAILURE);
 192                 }
 193
 194                 uinfo=userinfo_find_from_id(user);
 195                 if (!uinfo) {
 196                         debuga(_("Unknown user ID %s in file %s\n"),user,top1);
 197                         exit(EXIT_FAILURE);
 198                 }
 199
 200                 perc=(ttnbytes) ? nbytes * 100. / ttnbytes : 0;
 201                 perc2=(ttnelap) ? elap * 100. / ttnelap : 0;
 202
 203                 posicao++;
 204
 205 #if defined(__FreeBSD__)
 206                 fprintf(fp_top3,"%7d %20s %8lld %15s %3.2lf%% %10s %10qu %3.2lf%%\n",posicao,uinfo->label,nacc,fixnum(nbytes,1),perc,buildtime(elap),elap,perc2);
 207 #else
 208                 fprintf(fp_top3,"%7d %20s %8"PRIu64" %15s %3.2lf%% %10s %10"PRIu64" %3.2lf%%\n",posicao,uinfo->label,(uint64_t)nacc,fixnum(nbytes,1),perc,buildtime(elap),(uint64_t)elap,perc2);
 209 #endif
 210         }
 211
 212         // output total
 213         fputs("------- -------------------- -------- --------------- ------- ---------- ---------- -------\n",fp_top3);
 214 #if defined(__FreeBSD__)
 215         fprintf(fp_top3,"%-7s %20s %8qu %15s %8s %9s %10qu\n",_("TOTAL")," ",ttnacc,fixnum(ttnbytes,1)," ",buildtime(ttnelap),ttnelap);
 216 #else
 217         fprintf(fp_top3,"%-7s %20s %8"PRIu64" %15s %8s %9s %10"PRIu64"\n",_("TOTAL")," ",(uint64_t)ttnacc,fixnum(ttnbytes,1)," ",buildtime(ttnelap),(uint64_t)ttnelap);
 218 #endif
 219
 220         // compute and write average
 221         if (totuser>0) {
 222                 tnbytes=(totuser) ? ttnbytes / totuser : 0;
 223                 avgacc=ttnacc/totuser;
 224                 avgelap=ttnelap/totuser;
 225         } else {
 226                 tnbytes=0;
 227                 avgacc=0;
 228                 avgelap=0;
 229         }
 230
 231         strcpy(strip1,_("AVERAGE"));
 232         strip_latin(strip1);
 233 #if defined(__FreeBSD__)
 234         fprintf(fp_top3,"%-7s %20s %8qu %15s %8s %9s %10qu\n",strip1," ",avgacc,fixnum(tnbytes,1)," ",buildtime(avgelap),avgelap);
 235 #else
 236         fprintf(fp_top3,"%-7s %20s %8"PRIu64" %15s %8s %9s %10"PRIu64"\n",strip1," ",(uint64_t)avgacc,fixnum(tnbytes,1)," ",buildtime(avgelap),(uint64_t)avgelap);
 237 #endif
 238
 239         fclose(fp_top1);
 240         unlink(top1);
 241
 242         t = time(NULL);
 243         local = localtime(&t);
 244         fprintf(fp_top3, "\n%s\n", asctime(local));
 245
 246         fclose(fp_top3);
 247
 248         if(strcmp(email,"stdout") == 0) {
 249                 if((fp_top3=fopen(top3,"r"))==NULL) {
 250                         debuga(_("(email) Cannot open file %s\n"),top3);
 251                         exit(EXIT_FAILURE);
 252                 }
 253
 254                 while(fgets(warea,sizeof(warea),fp_top3)!=NULL)
 255                         fputs(warea,stdout);
 256         } else {
 257                 snprintf(warea,sizeof(warea),"%s -s \"SARG %s, %s\" \"%s\" <\"%s\"",MailUtility,_("Report"),asctime(local),email,top3);
 258                 cstatus=system(warea);
 259                 if (!WIFEXITED(cstatus) || WEXITSTATUS(cstatus)) {
 260                         debuga(_("command return status %d\n"),WEXITSTATUS(cstatus));
 261                         debuga(_("command: %s\n"),warea);
 262                         exit(EXIT_FAILURE);
 263                 }
 264         }
 265
 266         //unlinkdir(TempDir,0);
 267
 268         return (0);
 269 }