]>
Commit | Line | Data |
---|---|---|
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 | //! The daily statistics for one user. | |
31 | struct DayStruct | |
32 | { | |
33 | int ndaylist; | |
34 | int maxindex; | |
35 | int daylist[MAX_DATETIME_DAYS]; | |
36 | long long int bytes[MAX_DATETIME_DAYS*24]; | |
37 | long long int elap[MAX_DATETIME_DAYS*24]; | |
38 | }; | |
39 | ||
40 | /*! | |
41 | Prepare the object to store the daily statistics of one user. | |
42 | ||
43 | \return The object to pass to other functions in this module. | |
44 | The object must be freed with a call to day_cleanup(). | |
45 | */ | |
46 | DayObject day_prepare(void) | |
47 | { | |
48 | DayObject ddata; | |
49 | ||
50 | ddata=(DayObject)malloc(sizeof(*ddata)); | |
51 | if (!ddata) | |
52 | { | |
53 | debuga(__FILE__,__LINE__,_("Not enough memory to store the daily statistics\n")); | |
54 | exit(EXIT_FAILURE); | |
55 | } | |
56 | ||
57 | return(ddata); | |
58 | } | |
59 | ||
60 | /*! | |
61 | Free the memory allocated by day_prepare(). | |
62 | ||
63 | \param ddata The object returned by day_prepare(). | |
64 | */ | |
65 | void day_cleanup(DayObject ddata) | |
66 | { | |
67 | if (ddata) free(ddata); | |
68 | } | |
69 | ||
70 | /*! | |
71 | Prepare the object for a new user. | |
72 | ||
73 | \param ddata The object created by day_prepare(). | |
74 | */ | |
75 | void day_newuser(DayObject ddata) | |
76 | { | |
77 | if (ddata) | |
78 | { | |
79 | ddata->ndaylist=0; | |
80 | ddata->maxindex=0; | |
81 | memset(ddata->bytes,0,sizeof(ddata->bytes)); | |
82 | memset(ddata->elap,0,sizeof(ddata->elap)); | |
83 | } | |
84 | } | |
85 | ||
86 | /*! | |
87 | Store one data point in the statistics. | |
88 | ||
89 | \param ddata The object to store the statistics. | |
90 | \param date The date of the data point formated as day/month/year. | |
91 | \param time The time of the data point. | |
92 | \param elap The time spent processing the user's request on the proxy. | |
93 | \param bytes The number of bytes transfered by the user. | |
94 | */ | |
95 | void day_addpoint(DayObject ddata,const char *date, const char *time, long long int elap, long long int bytes) | |
96 | { | |
97 | int hour; | |
98 | int day,month,year; | |
99 | int daynum; | |
100 | int dayidx; | |
101 | int i; | |
102 | ||
103 | if (!ddata) return; | |
104 | if (sscanf(date,"%d/%d/%d",&day,&month,&year)!=3) { | |
105 | debuga(__FILE__,__LINE__,_("Invalid date \"%s\" for the hourly statistics\n"),date); | |
106 | exit(EXIT_FAILURE); | |
107 | } | |
108 | if (day<1 || day>31 || month<1 || month>12 || year>9999) { | |
109 | debuga(__FILE__,__LINE__,_("Invalid date component in \"%s\" for the hourly statistics\n"),date); | |
110 | exit(EXIT_FAILURE); | |
111 | } | |
112 | hour=atoi(time); | |
113 | if (hour<0 || hour>=24) { | |
114 | debuga(__FILE__,__LINE__,_("Invalid hour %d for the hourly statistics\n"),hour); | |
115 | exit(EXIT_FAILURE); | |
116 | } | |
117 | daynum=(year*10000)+(month*100)+day; | |
118 | for (dayidx=ddata->ndaylist-1 ; dayidx>=0 && daynum!=ddata->daylist[dayidx] ; dayidx--); | |
119 | if (dayidx<0) { | |
120 | dayidx=ddata->ndaylist++; | |
121 | if (dayidx>=sizeof(ddata->daylist)/sizeof(*ddata->daylist)) { | |
122 | debuga(__FILE__,__LINE__,_("Too many different dates for the hourly statistics\n")); | |
123 | exit(EXIT_FAILURE); | |
124 | } | |
125 | ddata->daylist[dayidx]=daynum; | |
126 | } | |
127 | i=dayidx*24+hour; | |
128 | if (i>=ddata->maxindex) ddata->maxindex=i+1; | |
129 | ddata->bytes[i]+=bytes; | |
130 | ddata->elap[i]+=elap; | |
131 | } | |
132 | ||
133 | /*! | |
134 | Store the dayly statistics in the file. | |
135 | ||
136 | \param ddata The object containing the statistics. | |
137 | \param tmp The temporary directory to store the file into. | |
138 | \param uinfo The user's data. | |
139 | */ | |
140 | void day_totalize(DayObject ddata,const char *tmp, const struct userinfostruct *uinfo) | |
141 | { | |
142 | FILE *fp_ou; | |
143 | int hour; | |
144 | int day,month,year; | |
145 | int i; | |
146 | int daynum; | |
147 | int dayidx; | |
148 | char arqout[2048]; | |
149 | ||
150 | if (datetimeby==0) return; | |
151 | if (!ddata) return; | |
152 | ||
153 | if (snprintf(arqout,sizeof(arqout),"%s/%s.day",tmp,uinfo->filename)>=sizeof(arqout)) { | |
154 | debuga(__FILE__,__LINE__,_("Path too long: ")); | |
155 | debuga_more("%s/%s%s\n",tmp,uinfo->filename,".day"); | |
156 | exit(EXIT_FAILURE); | |
157 | } | |
158 | ||
159 | if((fp_ou=fopen(arqout,"w"))==NULL) { | |
160 | debuga(__FILE__,__LINE__,_("Cannot open file \"%s\": %s\n"),arqout,strerror(errno)); | |
161 | exit(EXIT_FAILURE); | |
162 | } | |
163 | ||
164 | for (i=0 ; i<ddata->maxindex ; i++) { | |
165 | if (ddata->bytes[i]==0 && ddata->elap[i]==0) continue; | |
166 | dayidx=i/24; | |
167 | if (dayidx>=sizeof(ddata->daylist)/sizeof(*ddata->daylist)) { | |
168 | debuga(__FILE__,__LINE__,_("Invalid day index found in the hourly statistics\n")); | |
169 | exit(EXIT_FAILURE); | |
170 | } | |
171 | hour=i%24; | |
172 | daynum=ddata->daylist[dayidx]; | |
173 | day=daynum%100; | |
174 | month=(daynum/100)%100; | |
175 | year=daynum/10000; | |
176 | fprintf(fp_ou,"%d/%d/%d\t%d",day,month,year,hour); | |
177 | if ((datetimeby & DATETIME_BYTE)!=0) fprintf(fp_ou,"\t%"PRIu64"",(uint64_t)ddata->bytes[i]); | |
178 | if ((datetimeby & DATETIME_ELAP)!=0) fprintf(fp_ou,"\t%"PRIu64"",(uint64_t)ddata->elap[i]); | |
179 | fputs("\n",fp_ou); | |
180 | } | |
181 | ||
182 | if (fclose(fp_ou)==EOF) { | |
183 | debuga(__FILE__,__LINE__,_("Write error in \"%s\": %s\n"),arqout,strerror(errno)); | |
184 | exit(EXIT_FAILURE); | |
185 | } | |
186 | return; | |
187 | } | |
188 | ||
189 | /*! | |
190 | Delete the temporary file generated by day_totalize(). | |
191 | ||
192 | \param uinfo The user whose daily statistics are to be deleted. | |
193 | */ | |
194 | void day_deletefile(const struct userinfostruct *uinfo) | |
195 | { | |
196 | char arqout[2048]; | |
197 | ||
198 | if (KeepTempLog) return; | |
199 | ||
200 | if (snprintf(arqout,sizeof(arqout),"%s/%s.day",tmp,uinfo->filename)>=sizeof(arqout)) { | |
201 | debuga(__FILE__,__LINE__,_("Path too long: ")); | |
202 | debuga_more("%s/%s%s\n",tmp,uinfo->filename,".day"); | |
203 | exit(EXIT_FAILURE); | |
204 | } | |
205 | ||
206 | if (unlink(arqout)) | |
207 | debuga(__FILE__,__LINE__,_("Cannot delete \"%s\": %s\n"),arqout,strerror(errno)); | |
208 | } | |
209 |