Add an option to create the per_user_limit files

[thirdparty/sarg.git] / html.c

diff --git a/html.c b/html.c
index 54d953f25970196b9975198254947bec6b574440..ad5be738e80adcea141bdf66d228aa911fa98070 100644 (file)
--- a/html.c
+++ b/html.c
@@ -27,6 +27,13 @@
 #include "include/conf.h"
 #include "include/defs.h"
 
+//! Number of limits.
+int PerUserLimitsNumber=0;
+//! Log user's who downloaded more than the limit.
+struct PerUserLimitStruct PerUserLimits[MAX_USER_LIMITS];
+//! How to create a per user file.
+enum PerUserFileCreationEnum PerUserFileCreation=PUFC_Always;
+
 extern struct globalstatstruct globstat;
 
 void htmlrel(void)
@@ -51,13 +58,15 @@ void htmlrel(void)
        double perc=0, perc2=0, ouperc=0, inperc=0;
        int count;
        int cstatus;
+       int i;
+       unsigned int user_limit[(MAX_USER_LIMITS+sizeof(unsigned int)-1)/sizeof(unsigned int)];
        bool have_denied_report;
        const char *sort_field;
        const char *sort_order;
        char siteind[MAX_TRUNCATED_URL];
        struct getwordstruct gwarea;
        longline line,line1;
-       struct userinfostruct *uinfo;
+       const struct userinfostruct *uinfo;
        userscan uscan;
 
        if (snprintf(tmp2,sizeof(tmp2),"%s/sargtmp.int_unsort",tmp)>=sizeof(tmp2)) {
@@ -77,6 +86,31 @@ void htmlrel(void)
 
        sort_labels(&sort_field,&sort_order);
 
+       switch (PerUserFileCreation)
+       {
+               case PUFC_Always:
+                       for (i=0 ; i<PerUserLimitsNumber ; i++) {
+                               FILE *fp_usr=fopen(PerUserLimits[i].File,"wt");
+                               if (fp_usr==NULL) {
+                                       debuga(_("Cannot create empty per_user_limit file \"%s\": %s"),PerUserLimits[i].File,
+                                                  strerror(errno));
+                                       exit(EXIT_FAILURE);
+                               }
+                               fclose(fp_usr);
+                       }
+                       break;
+
+               case PUFC_AsRequired:
+                       for (i=0 ; i<PerUserLimitsNumber ; i++) {
+                               if (access(PerUserLimits[i].File,R_OK)==0 && unlink(PerUserLimits[i].File)==-1) {
+                                       debuga(_("Cannot delete per_user_limit file \"%s\": %s"),PerUserLimits[i].File,
+                                                  strerror(errno));
+                                       exit(EXIT_FAILURE);
+                               }
+                       }
+                       break;
+       }
+
        uscan=userinfo_startscan();
        if (uscan == NULL) {
                debuga(_("Cannot enumerate the user list\n"));
@@ -129,6 +163,9 @@ void htmlrel(void)
                        exit(EXIT_FAILURE);
                }
 
+               for (i=0 ; i<sizeof(user_limit)/sizeof(user_limit[0]) ; i++)
+                       user_limit[i]=0;
+
                tnacc=0;
                tnbytes=0;
                tnelap=0;
@@ -527,22 +564,35 @@ void htmlrel(void)
                        fputs("</tr>\n",fp_ou);
                }
 
-               if (PerUserLimit > 0 && (int)(tnbytes/1000000LLU) > PerUserLimit && !uinfo->user_limit) {
-                       FILE *fp_usr;
+               if (PerUserLimitsNumber>0) {
+                       int limit=(int)(tnbytes/1000000LLU);
+                       int maskid;
+                       int mask;
+                       for (i=0 ; i<PerUserLimitsNumber ; i++) {
+                               maskid=i/sizeof(unsigned int);
+                               mask=0x1U << (i % sizeof(unsigned int));
+                               if (limit>PerUserLimits[i].Limit && (user_limit[maskid] & mask)==0) {
+                                       FILE *fp_usr;
+
+                                       if((fp_usr = fopen(PerUserLimits[i].File, "at")) == 0) {
+                                               debuga(_("(html10) Cannot open per user limit file %s: %s\n"),PerUserLimits[i].File,strerror(errno));
+                                               exit(EXIT_FAILURE);
+                                       }
+                                       if (PerUserLimits[i].UserId)
+                                               fprintf(fp_usr,"%s\n",uinfo->label);
+                                       else
+                                               fprintf(fp_usr,"%s\n",uinfo->ip);
+                                       if (fclose(fp_usr)==EOF) {
+                                               debuga(_("Write error in per user limit file %s: %s\n"),PerUserLimits[i].File,strerror(errno));
+                                               exit(EXIT_FAILURE);
+                                       }
+                                       user_limit[maskid]|=mask;
 
-                       if((fp_usr = fopen(PerUserLimitFile, "a")) == 0) {
-                               debuga(_("(html10) Cannot open file %s: %s\n"),PerUserLimitFile,strerror(errno));
-                               exit(EXIT_FAILURE);
-                       }
-                       fprintf(fp_usr,"%s\n",uinfo->ip);
-                       if (fclose(fp_usr)==EOF) {
-                               debuga(_("Write error in %s: %s\n"),PerUserLimitFile,strerror(errno));
-                               exit(EXIT_FAILURE);
+                                       if(debug)
+                                               debuga(_("User %s limit exceeded (%d MB). Added to file %s\n"),uinfo->label,
+                                                          PerUserLimits[i].Limit,PerUserLimits[i].File);
+                               }
                        }
-                       uinfo->user_limit=1;
-
-                       if(debug)
-                               debuga(_("User %s limit exceeded (%d MB). Added to file %s\n"),uinfo->label,PerUserLimit,PerUserLimitFile);
                }
 
                if ((ReportType & REPORT_TYPE_TOPUSERS) != 0 && (UserReportFields & USERREPORTFIELDS_AVERAGE) != 0) {