email.c

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