/*
* SARG Squid Analysis Report Generator http://sarg.sourceforge.net
- * 1998, 2010
+ * 1998, 2015
*
* SARG donations:
* please look at http://sarg.sourceforge.net/donations.php
#include "include/conf.h"
#include "include/defs.h"
+#ifdef ENABLE_DOUBLE_CHECK_DATA
+extern struct globalstatstruct globstat;
+#endif
+
int geramail(const char *dirname, int debug, const char *outdir, const char *email, const char *TempDir)
{
- FILE *fp_in, *fp_top1, *fp_top2, *fp_top3;
- long long int ttnbytes=0, ttnacc=0, tnacc=0;
- long long int tnbytes=0, ttnelap=0, tnelap=0;
- long long int nacc, nbytes, elap;
- long long int avgacc, avgelap;
- double perc=0.00;
- double perc2=0.00;
- int posicao=0;
- char olduser[MAX_USER_LEN], csort[MAXLEN];
- char wger[MAXLEN], top1[MAXLEN], top2[MAXLEN], top3[MAXLEN], user[MAX_USER_LEN];
- char strip1[MAXLEN], strip2[MAXLEN], strip3[MAXLEN], strip4[MAXLEN], strip5[MAXLEN], strip6[MAXLEN], strip7[MAXLEN];
- char buf[MAXLEN];
- int totuser=0;
- time_t t;
- struct tm *local;
- int cstatus;
- char warea[MAXLEN];
- struct getwordstruct gwarea;
- struct generalitemstruct item;
-
- snprintf(wger,sizeof(wger),"%s/sarg-general",dirname);
- if((fp_in=fopen(wger,"r"))==NULL) {
- debuga(_("(email) Cannot open file %s\n"),wger);
- exit(EXIT_FAILURE);
- }
-
- snprintf(top1,sizeof(top1),"%s/top",dirname);
- if((fp_top1=fopen(top1,"w"))==NULL) {
- debuga(_("(email) Cannot open file %s\n"),top1);
- exit(EXIT_FAILURE);
- }
-
- snprintf(top2,sizeof(top2),"%s/top.tmp",dirname);
- if((fp_top2=fopen(top2,"w"))==NULL) {
- debuga(_("(email) Cannot open file %s\n"),top2);
- exit(EXIT_FAILURE);
- }
-
- olduser[0]='\0';
- totuser=0;
-
- while(fgets(warea,sizeof(warea),fp_in)) {
- ger_read(warea,&item,wger);
- if(item.total) continue;
- if(strcmp(olduser,item.user) != 0)
- {
- totuser++;
-
- if (olduser[0] != '\0') {
+ FILE *fp_in, *fp_top1, *fp_top2, *fp_top3;
+ long long int ttnbytes=0, ttnacc=0, tnacc=0;
+ long long int tnbytes=0, ttnelap=0, tnelap=0;
+ long long int nacc, nbytes, elap;
+ long long int avgacc, avgelap;
+ double perc=0.00;
+ double perc2=0.00;
+ int posicao=0;
+ char olduser[MAX_USER_LEN], csort[MAXLEN];
+ char wger[MAXLEN], top1[MAXLEN], top2[MAXLEN], top3[MAXLEN], user[MAX_USER_LEN];
+ char strip1[MAXLEN], strip2[MAXLEN], strip3[MAXLEN], strip4[MAXLEN], strip5[MAXLEN], strip6[MAXLEN], strip7[MAXLEN];
+ char *buf;
+ char warea[MAXLEN];
+ char Subject[120];
+ int totuser=0;
+ time_t t;
+ struct tm *local;
+ int cstatus;
+ struct getwordstruct gwarea;
+ struct generalitemstruct item;
+ longline line;
+ const struct userinfostruct *uinfo;
+
+ snprintf(wger,sizeof(wger),"%s/sarg-general",dirname);
+ if((fp_in=fopen(wger,"r"))==NULL) {
+ debuga(__FILE__,__LINE__,_("Cannot open file \"%s\": %s\n"),wger,strerror(errno));
+ exit(EXIT_FAILURE);
+ }
+
+ snprintf(top2,sizeof(top2),"%s/top.tmp",dirname);
+ if((fp_top2=fopen(top2,"w"))==NULL) {
+ debuga(__FILE__,__LINE__,_("Cannot open file \"%s\": %s\n"),top2,strerror(errno));
+ exit(EXIT_FAILURE);
+ }
+
+ olduser[0]='\0';
+ totuser=0;
+
+ if ((line=longline_create())==NULL) {
+ debuga(__FILE__,__LINE__,_("Not enough memory to read file \"%s\"\n"),wger);
+ exit(EXIT_FAILURE);
+ }
+
+ while((buf=longline_read(fp_in,line))!=NULL) {
+ ger_read(buf,&item,wger);
+ if(item.total) continue;
+ if(strcmp(olduser,item.user) != 0) {
+ totuser++;
+
+ if (olduser[0] != '\0') {
#if defined(__FreeBSD__)
- fprintf(fp_top2,"%s\t%qu\t%qu\t%qu\n",olduser,tnbytes,tnacc,tnelap);
+ fprintf(fp_top2,"%s\t%qu\t%qu\t%qu\n",olduser,tnbytes,tnacc,tnelap);
#else
- fprintf(fp_top2,"%s\t%"PRIu64"\t%"PRIu64"\t%"PRIu64"\n",olduser,(uint64_t)tnbytes,(uint64_t)tnacc,(uint64_t)tnelap);
+ fprintf(fp_top2,"%s\t%"PRIu64"\t%"PRIu64"\t%"PRIu64"\n",olduser,(uint64_t)tnbytes,(uint64_t)tnacc,(uint64_t)tnelap);
#endif
- ttnbytes+=tnbytes;
- ttnacc+=tnacc;
- ttnelap+=tnelap;
- }
- strcpy(olduser,item.user);
- tnbytes=0;
- tnacc=0;
- tnelap=0;
- }
-
- tnbytes+=item.nbytes;
- tnacc+=item.nacc;
- tnelap+=item.nelap;
- }
-
- if (olduser[0] != '\0') {
+ ttnbytes+=tnbytes;
+ ttnacc+=tnacc;
+ ttnelap+=tnelap;
+ }
+ strcpy(olduser,item.user);
+ tnbytes=0;
+ tnacc=0;
+ tnelap=0;
+ }
+
+ tnbytes+=item.nbytes;
+ tnacc+=item.nacc;
+ tnelap+=item.nelap;
+ }
+ if (fclose(fp_in)==EOF) {
+ debuga(__FILE__,__LINE__,_("Read error in \"%s\": %s\n"),wger,strerror(errno));
+ exit(EXIT_FAILURE);
+ }
+ longline_destroy(&line);
+
+ if (olduser[0] != '\0') {
#if defined(__FreeBSD__)
- fprintf(fp_top2,"%s\t%qu\t%qu\t%qu\n",olduser,tnbytes,tnacc,tnelap);
+ fprintf(fp_top2,"%s\t%qu\t%qu\t%qu\n",olduser,tnbytes,tnacc,tnelap);
#else
- fprintf(fp_top2,"%s\t%"PRIu64"\t%"PRIu64"\t%"PRIu64"\n",olduser,(uint64_t)tnbytes,(uint64_t)tnacc,(uint64_t)tnelap);
+ fprintf(fp_top2,"%s\t%"PRIu64"\t%"PRIu64"\t%"PRIu64"\n",olduser,(uint64_t)tnbytes,(uint64_t)tnacc,(uint64_t)tnelap);
#endif
- ttnbytes+=tnbytes;
- ttnacc+=tnacc;
- ttnelap+=tnelap;
- }
-
- fclose(fp_in);
- fclose(fp_top2);
-
- sprintf(csort,"sort -n -T \"%s\" -r -k 2,2 -o \"%s\" \"%s\"", TempDir, top1, top2);
- cstatus=system(csort);
- if (!WIFEXITED(cstatus) || WEXITSTATUS(cstatus)) {
- debuga(_("sort command return status %d\n"),WEXITSTATUS(cstatus));
- debuga(_("sort command: %s\n"),csort);
- exit(EXIT_FAILURE);
- }
-
- unlink(top2);
-
- if((fp_top1=fopen(top1,"a"))==NULL) {
- debuga(_("(email) Cannot open file %s\n"),top1);
- exit(EXIT_FAILURE);
- }
-#if defined(__FreeBSD__)
- fprintf(fp_top1,"TOTAL\t%qu\t%qu\t%qu\n",ttnbytes,ttnacc,ttnelap);
-#else
- fprintf(fp_top1,"TOTAL\t%"PRIu64"\t%"PRIu64"\t%"PRIu64"\n",(uint64_t)ttnbytes,(uint64_t)ttnacc,(uint64_t)ttnelap);
+ ttnbytes+=tnbytes;
+ ttnacc+=tnacc;
+ ttnelap+=tnelap;
+ }
+
+ if (fclose(fp_top2)==EOF) {
+ debuga(__FILE__,__LINE__,_("Write error in \"%s\": %s\n"),top2,strerror(errno));
+ exit(EXIT_FAILURE);
+ }
+
+#ifdef ENABLE_DOUBLE_CHECK_DATA
+ if (ttnacc!=globstat.nacc || ttnbytes!=globstat.nbytes || ttnelap!=globstat.elap) {
+ debuga(__FILE__,__LINE__,_("Total statistics mismatch when reading \"%s\" to produce the email report\n"),wger);
+ exit(EXIT_FAILURE);
+ }
#endif
- fclose(fp_top1);
-
- if((fp_top1=fopen(top1,"r"))==NULL) {
- debuga(_("(email) Cannot open file %s\n"),top1);
- exit(EXIT_FAILURE);
- }
-
- snprintf(top3,sizeof(top3),"%s/report",dirname);
- if((fp_top3=fopen(top3,"w"))==NULL) {
- debuga(_("(email) Cannot open file %s\n"),top3);
- exit(EXIT_FAILURE);
- }
-
- strcpy(strip1,_("Squid User Access Report"));
- strip_latin(strip1);
- fprintf(fp_top3,"%s\n",strip1);
-
- strcpy(strip1,_("Decreasing Access (bytes)"));
- strip_latin(strip1);
- fprintf(fp_top3,"%s\n",strip1);
-
- strcpy(strip1,_("Period"));
- strip_latin(strip1);
- fprintf(fp_top3,"%s %s\n\n",strip1,period.text);
-
- strcpy(strip1,_("NUM"));
- strip_latin(strip1);
- strcpy(strip2,_("USERID"));
- strip_latin(strip2);
- strcpy(strip3,_("CONNECT"));
- strip_latin(strip3);
- strcpy(strip4,_("BYTES"));
- strip_latin(strip4);
- strcpy(strip5,_("ELAPSED TIME"));
- strip_latin(strip5);
- strcpy(strip6,_("MILLISEC"));
- strip_latin(strip6);
- strcpy(strip7,_("TIME"));
- strip_latin(strip7);
-
- fprintf(fp_top3,"%-7s %-20s %-8s %-15s %%%-6s %-10s %-10s %%%-7s\n------- -------------------- -------- --------------- ------- ---------- ---------- -------\n",strip1,strip2,strip3,strip4,strip4,strip5,strip6,strip7);
-
- while(fgets(warea,sizeof(warea),fp_top1))
- {
- getword_start(&gwarea,warea);
- if (getword(user,sizeof(user),&gwarea,'\t')<0) {
- debuga(_("There is an invalid user ID in file %s\n"),top1);
- exit(EXIT_FAILURE);
- }
- if (getword_atoll(&nbytes,&gwarea,'\t')<0) {
- debuga(_("There is an invalid number of bytes in file %s\n"),top1);
- exit(EXIT_FAILURE);
- }
- if (getword_atoll(&nacc,&gwarea,'\t')<0) {
- debuga(_("There is an invalid number of access in file %s\n"),top1);
- exit(EXIT_FAILURE);
- }
- if (getword_atoll(&elap,&gwarea,'\t')<0) {
- debuga(_("There is an invalid elapsed time in file %s\n"),top1);
- exit(EXIT_FAILURE);
- }
-
- perc=(ttnbytes) ? nbytes * 100. / ttnbytes : 0;
- perc2=(ttnelap) ? elap * 100. / ttnelap : 0;
-
- posicao++;
-
- if(strcmp(user,"TOTAL") == 0){
- fputs("------- -------------------- -------- --------------- ------- ---------- ---------- -------\n",fp_top3);
+
+ if (snprintf(csort,sizeof(csort),"sort -n -T \"%s\" -t \"\t\" -r -k 2,2 -o \"%s\" \"%s\"", TempDir, top1, top2)>=sizeof(csort)) {
+ debuga(__FILE__,__LINE__,_("Sort command too long when sorting file \"%s\" to \"%s\"\n"),top2,top1);
+ exit(EXIT_FAILURE);
+ }
+ cstatus=system(csort);
+ if (!WIFEXITED(cstatus) || WEXITSTATUS(cstatus)) {
+ debuga(__FILE__,__LINE__,_("sort command return status %d\n"),WEXITSTATUS(cstatus));
+ debuga(__FILE__,__LINE__,_("sort command: %s\n"),csort);
+ exit(EXIT_FAILURE);
+ }
+
+ if (!KeepTempLog && unlink(top2)) {
+ debuga(__FILE__,__LINE__,_("Cannot delete \"%s\": %s\n"),top2,strerror(errno));
+ exit(EXIT_FAILURE);
+ }
+
+ if((fp_top1=fopen(top1,"r"))==NULL) {
+ debuga(__FILE__,__LINE__,_("Cannot open file \"%s\": %s\n"),top1,strerror(errno));
+ exit(EXIT_FAILURE);
+ }
+
+ snprintf(top3,sizeof(top3),"%s/report",dirname);
+ if((fp_top3=fopen(top3,"w"))==NULL) {
+ debuga(__FILE__,__LINE__,_("Cannot open file \"%s\": %s\n"),top3,strerror(errno));
+ exit(EXIT_FAILURE);
+ }
+
+ safe_strcpy(strip1,_("Squid User Access Report"),sizeof(strip1));
+ strip_latin(strip1);
+ fprintf(fp_top3,"%s\n",strip1);
+
+ safe_strcpy(strip1,_("Decreasing Access (bytes)"),sizeof(strip1));
+ strip_latin(strip1);
+ fprintf(fp_top3,"%s\n",strip1);
+
+ safe_strcpy(strip1,_("Period"),sizeof(strip1));
+ strip_latin(strip1);
+ fprintf(fp_top3,"%s %s\n\n",strip1,period.text);
+
+ safe_strcpy(strip1,_("NUM"),sizeof(strip1));
+ strip_latin(strip1);
+ safe_strcpy(strip2,_("USERID"),sizeof(strip2));
+ strip_latin(strip2);
+ safe_strcpy(strip3,_("CONNECT"),sizeof(strip3));
+ strip_latin(strip3);
+ safe_strcpy(strip4,_("BYTES"),sizeof(strip4));
+ strip_latin(strip4);
+ safe_strcpy(strip5,_("ELAPSED TIME"),sizeof(strip5));
+ strip_latin(strip5);
+ safe_strcpy(strip6,_("MILLISEC"),sizeof(strip6));
+ strip_latin(strip6);
+ safe_strcpy(strip7,pgettext("duration","TIME"),sizeof(strip7));
+ strip_latin(strip7);
+
+ fprintf(fp_top3,"%-7s %-20s %-8s %-15s %%%-6s %-10s %-10s %%%-7s\n------- -------------------- -------- --------------- ------- ---------- ---------- -------\n",strip1,strip2,strip3,strip4,strip4,strip5,strip6,strip7);
+
+ while(fgets(warea,sizeof(warea),fp_top1)) {
+ fixendofline(warea);
+ getword_start(&gwarea,warea);
+ if (getword(user,sizeof(user),&gwarea,'\t')<0) {
+ debuga(__FILE__,__LINE__,_("Invalid user ID in file \"%s\"\n"),top1);
+ exit(EXIT_FAILURE);
+ }
+ if (getword_atoll(&nbytes,&gwarea,'\t')<0) {
+ debuga(__FILE__,__LINE__,_("Invalid number of bytes in file \"%s\"\n"),top1);
+ exit(EXIT_FAILURE);
+ }
+ if (getword_atoll(&nacc,&gwarea,'\t')<0) {
+ debuga(__FILE__,__LINE__,_("Invalid number of accesses in file \"%s\"\n"),top1);
+ exit(EXIT_FAILURE);
+ }
+ if (getword_atoll(&elap,&gwarea,'\0')<0) {
+ debuga(__FILE__,__LINE__,_("Invalid elapsed time in file \"%s\"\n"),top1);
+ exit(EXIT_FAILURE);
+ }
+
+ uinfo=userinfo_find_from_id(user);
+ if (!uinfo) {
+ debuga(__FILE__,__LINE__,_("Unknown user ID %s in file \"%s\"\n"),user,top1);
+ exit(EXIT_FAILURE);
+ }
+
+ perc=(ttnbytes) ? nbytes * 100. / ttnbytes : 0;
+ perc2=(ttnelap) ? elap * 100. / ttnelap : 0;
+
+ posicao++;
+
#if defined(__FreeBSD__)
- fprintf(fp_top3,"%-7s %20s %8qu %15s %8s %9s %10qu\n",_("TOTAL")," ",ttnacc,fixnum(ttnbytes,1)," ",buildtime(ttnelap),ttnelap);
+ 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);
#else
- 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);
+ 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);
#endif
- } else {
+ }
+
+ // output total
+ fputs("------- -------------------- -------- --------------- ------- ---------- ---------- -------\n",fp_top3);
#if defined(__FreeBSD__)
- fprintf(fp_top3,"%7d %20s %8lld %15s %3.2lf%% %10s %10qu %3.2lf%%\n",posicao,user,nacc,fixnum(nbytes,1),perc,buildtime(elap),elap,perc2);
+ fprintf(fp_top3,"%-7s %20s %8qu %15s %8s %9s %10qu\n",_("TOTAL")," ",ttnacc,fixnum(ttnbytes,1)," ",buildtime(ttnelap),ttnelap);
#else
- fprintf(fp_top3,"%7d %20s %8"PRIu64" %15s %3.2lf%% %10s %10"PRIu64" %3.2lf%%\n",posicao,user,(uint64_t)nacc,fixnum(nbytes,1),perc,buildtime(elap),(uint64_t)elap,perc2);
+ 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);
#endif
- }
- }
-
- if (totuser>0) {
- tnbytes=(totuser) ? ttnbytes / totuser : 0;
- avgacc=ttnacc/totuser;
- avgelap=ttnelap/totuser;
- } else {
- tnbytes=0;
- avgacc=0;
- avgelap=0;
- }
-
- strcpy(strip1,_("AVERAGE"));
- strip_latin(strip1);
+
+ // compute and write average
+ if (totuser>0) {
+ tnbytes=(totuser) ? ttnbytes / totuser : 0;
+ avgacc=ttnacc/totuser;
+ avgelap=ttnelap/totuser;
+ } else {
+ tnbytes=0;
+ avgacc=0;
+ avgelap=0;
+ }
+
+ safe_strcpy(strip1,_("AVERAGE"),sizeof(strip1));
+ strip_latin(strip1);
#if defined(__FreeBSD__)
- fprintf(fp_top3,"%-7s %20s %8qu %15s %8s %9s %10qu\n",strip1," ",avgacc,fixnum(tnbytes,1)," ",buildtime(avgelap),avgelap);
+ fprintf(fp_top3,"%-7s %20s %8qu %15s %8s %9s %10qu\n",strip1," ",avgacc,fixnum(tnbytes,1)," ",buildtime(avgelap),avgelap);
#else
- 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);
+ 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);
#endif
- fclose(fp_top1);
- unlink(top1);
-
- t = time(NULL);
- local = localtime(&t);
- fprintf(fp_top3, "\n%s\n", asctime(local));
-
- fclose(fp_top3);
-
- if(strcmp(email,"stdout") == 0) {
- if((fp_top3=fopen(top3,"r"))==NULL) {
- debuga(_("(email) Cannot open file %s\n"),top3);
- exit(EXIT_FAILURE);
- }
-
- while(fgets(buf,sizeof(buf),fp_top3)!=NULL)
- fputs(buf,stdout);
- } else {
- snprintf(buf,sizeof(buf),"\"%s\" -s \"SARG %s, %s\" \"%s\" <\"%s\"",MailUtility,_("Report"),asctime(local),email,top3);
- cstatus=system(buf);
- if (!WIFEXITED(cstatus) || WEXITSTATUS(cstatus)) {
- debuga(_("command return status %d\n"),WEXITSTATUS(cstatus));
- debuga(_("command: %s\n"),buf);
- exit(EXIT_FAILURE);
- }
- }
-
- unlinkdir(TempDir,0);
-
- return (0);
+ if (fclose(fp_top1)==EOF) {
+ debuga(__FILE__,__LINE__,_("Read error in \"%s\": %s\n"),top1,strerror(errno));
+ exit(EXIT_FAILURE);
+ }
+ if (!KeepTempLog && unlink(top1)) {
+ debuga(__FILE__,__LINE__,_("Cannot delete \"%s\": %s\n"),top1,strerror(errno));
+ exit(EXIT_FAILURE);
+ }
+
+ t = time(NULL);
+ local = localtime(&t);
+ fprintf(fp_top3, "\n%s\n", asctime(local));
+
+ if (fclose(fp_top3)==EOF) {
+ debuga(__FILE__,__LINE__,_("Write error in \"%s\": %s\n"),top3,strerror(errno));
+ exit(EXIT_FAILURE);
+ }
+
+ if(strcmp(email,"stdout") == 0) {
+ if((fp_top3=fopen(top3,"r"))==NULL) {
+ debuga(__FILE__,__LINE__,_("Cannot open file \"%s\": %s\n"),top3,strerror(errno));
+ exit(EXIT_FAILURE);
+ }
+
+ while(fgets(warea,sizeof(warea),fp_top3)!=NULL)
+ fputs(warea,stdout);
+ if (fclose(fp_top3)==EOF) {
+ debuga(__FILE__,__LINE__,_("Read error in \"%s\": %s\n"),top3,strerror(errno));
+ exit(EXIT_FAILURE);
+ }
+ } else {
+ /* TRANSLATORS: The string is formatted using strftime. You can use
+ any string formatting marker allowed by strftime. */
+ strftime(Subject,sizeof(Subject),_("SARG report, %c"),local);
+ snprintf(warea,sizeof(warea),"%s -s \"%s\" \"%s\" <\"%s\"",MailUtility,Subject,email,top3);
+ if (debug)
+ debuga(__FILE__,__LINE__,_("Sending mail with command: %s\n"),warea);
+ cstatus=system(warea);
+ if (!WIFEXITED(cstatus) || WEXITSTATUS(cstatus)) {
+ debuga(__FILE__,__LINE__,_("command return status %d\n"),WEXITSTATUS(cstatus));
+ debuga(__FILE__,__LINE__,_("command: %s\n"),warea);
+ exit(EXIT_FAILURE);
+ }
+ }
+
+ //unlinkdir(TempDir,0);
+
+ return (0);
}