]> git.ipfire.org Git - thirdparty/sarg.git/blob - url.c
ca2318cafeb6c14423fe3a0370296a420b8899f7
[thirdparty/sarg.git] / url.c
1 /*
2 * SARG Squid Analysis Report Generator http://sarg.sourceforge.net
3 * 1998, 2013
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 #ifdef HAVE_PCRE_H
30 #include <pcre.h>
31 #define USE_PCRE
32 #endif
33
34 /*!
35 A host name and the name to report.
36 */
37 struct hostalias_name
38 {
39 //! The next host name in the list or NULL for the last item.
40 struct hostalias_name *Next;
41 //! The minimum length of a candidate host name.
42 int MinLen;
43 //! The length of the constant part at the beginning of the mask.
44 int PrefixLen;
45 //! The length of the constant part at the end of the mask.
46 int SuffixLen;
47 //! The first part of the mask of the host name.
48 const char *HostName_Prefix;
49 //! The second part of the mask of the host name.
50 const char *HostName_Suffix;
51 //! The replacement name.
52 const char *Alias;
53 };
54
55 /*!
56 An IPv4 address and the name to report.
57 */
58 struct hostalias_ipv4
59 {
60 //! The next host name in the list or NULL for the last item.
61 struct hostalias_ipv4 *Next;
62 //! The IP address.
63 unsigned char Ip[4];
64 //! The number of bits in the prefix.
65 int NBits;
66 //! The replacement name.
67 const char *Alias;
68 };
69
70 /*!
71 An IPv6 address and the name to report.
72 */
73 struct hostalias_ipv6
74 {
75 //! The next host name in the list or NULL for the last item.
76 struct hostalias_ipv6 *Next;
77 //! The IP address.
78 unsigned short Ip[8];
79 //! The number of bits in the prefix.
80 int NBits;
81 //! The replacement name.
82 const char *Alias;
83 };
84
85 #ifdef USE_PCRE
86 /*!
87 A regular expression.
88 */
89 struct hostalias_regex
90 {
91 //! The next regular expression to test.
92 struct hostalias_regex *Next;
93 //! The regular expression to match against the host name.
94 pcre *Re;
95 //! The replacement name.
96 const char *Alias;
97 //! \c True if this regular expression contains at least one subpattern
98 bool SubPartern;
99 };
100 #endif
101
102 //! The first host name.
103 static struct hostalias_name *FirstAliasName=NULL;
104 //! The first IPv4 address.
105 static struct hostalias_ipv4 *FirstAliasIpv4=NULL;
106 //! The first IPvĀ§ address.
107 static struct hostalias_ipv6 *FirstAliasIpv6=NULL;
108
109 #ifdef USE_PCRE
110 static struct hostalias_regex *FirstAliasRe=NULL;
111 #endif
112
113 /*!
114 Store a name to alias.
115
116 \param name The name to match including the wildcard.
117 \param next A pointer to the first character after the name.
118
119 \retval 1 Alias added.
120 \retval 0 Ignore the line.
121 \retval -1 Error.
122 */
123 static int Alias_StoreName(const char *name,const char *next)
124 {
125 const char *NameBegin;
126 const char *NameBeginE;
127 const char *NameEnd;
128 const char *NameEndE;
129 const char *Replace;
130 const char *ReplaceE;
131 const char *str;
132 char sep;
133 struct hostalias_name *alias;
134 struct hostalias_name *new_alias;
135 struct hostalias_name *prev_alias;
136 char *tmp;
137 int len;
138
139 if (*name=='#' || *name==';') return(0);
140
141 // get host name and split at the wildcard
142 NameBegin=name;
143 for (str=NameBegin ; str<next && (unsigned char)*str>' ' && *str!='*' ; str++);
144 NameBeginE=str;
145 if (NameBegin==NameBeginE) NameBegin=NULL;
146 if (str<next && *str=='*') {
147 NameEnd=++str;
148 while (str<next && (unsigned char)*str>' ') {
149 if (*str=='*') {
150 debuga(_("Host name alias \"%s*%s\" contains too many wildcards (*)\n"),NameBegin,NameEnd);
151 return(-1);
152 }
153 str++;
154 }
155 NameEndE=str;
156 if (NameEnd==NameEndE) {
157 debuga(_("Host name alias \"%*s\" must not end with a wildcard\n"),(int)(next-name),name);
158 return(-1);
159 }
160 } else {
161 NameEnd=NULL;
162 }
163 while (str<next && (unsigned char)*str<=' ') str++;
164 if (!NameBegin && !NameEnd) return(0);
165
166 // get the alias
167 sep=*next;
168 if (sep==' ' || sep=='\t') {
169 Replace=next;
170 while (*Replace==' ' || *Replace=='\t') Replace++;
171 if ((unsigned char)*Replace<' ') {
172 Replace=NULL;
173 } else {
174 for (str=Replace ; *str && (unsigned char)*str>=' ' ; str++);
175 ReplaceE=str;
176 }
177 } else
178 Replace=NULL;
179
180 if (NameBegin) {
181 len=(int)(NameBeginE-NameBegin);
182 tmp=malloc(len+1);
183 if (!tmp) {
184 debuga(_("Not enough memory to store the host name aliasing directives\n"));
185 return(-1);
186 }
187 memcpy(tmp,NameBegin,len);
188 tmp[len]='\0';
189 NameBegin=tmp;
190 }
191 if (NameEnd) {
192 len=(int)(NameEndE-NameEnd);
193 tmp=malloc(len+1);
194 if (!tmp) {
195 if (NameBegin) free((void*)NameBegin);
196 debuga(_("Not enough memory to store the host name aliasing directives\n"));
197 return(-1);
198 }
199 memcpy(tmp,NameEnd,len);
200 tmp[len]='\0';
201 NameEnd=tmp;
202 }
203
204 // ignore duplicates
205 prev_alias=NULL;
206 for (alias=FirstAliasName ; alias ; alias=alias->Next) {
207 if (((NameBegin && alias->HostName_Prefix && !strcmp(NameBegin,alias->HostName_Prefix)) || (!NameBegin && !alias->HostName_Prefix)) &&
208 ((NameEnd && alias->HostName_Suffix && !strcmp(NameEnd,alias->HostName_Suffix)) || (!NameEnd && !alias->HostName_Suffix))) {
209 if (NameBegin) free((void*)NameBegin);
210 return(0);
211 }
212 prev_alias=alias;
213 }
214
215 // insert into the list
216 new_alias=malloc(sizeof(*new_alias));
217 if (!new_alias) {
218 if (NameBegin) free((void*)NameBegin);
219 if (NameEnd) free((void*)NameEnd);
220 debuga(_("Not enough memory to store the host name aliasing directives\n"));
221 return(-1);
222 }
223 new_alias->MinLen=0;
224 if (NameBegin) {
225 new_alias->HostName_Prefix=NameBegin;
226 new_alias->MinLen+=strlen(NameBegin);
227 new_alias->PrefixLen=strlen(NameBegin);
228 } else {
229 new_alias->HostName_Prefix=NULL;
230 new_alias->PrefixLen=0;
231 }
232 if (NameEnd) {
233 new_alias->HostName_Suffix=NameEnd;
234 new_alias->MinLen+=strlen(NameEnd)+1;
235 new_alias->SuffixLen=strlen(NameEnd);
236 } else {
237 new_alias->HostName_Suffix=NULL;
238 new_alias->SuffixLen=0;
239 }
240 if (Replace) {
241 len=(int)(ReplaceE-Replace);
242 tmp=malloc(len+2);
243 if (!tmp) {
244 debuga(_("Not enough memory to store the host name aliasing directives\n"));
245 return(-1);
246 }
247 tmp[0]=ALIAS_PREFIX;
248 memcpy(tmp+1,Replace,len);
249 tmp[len+1]='\0';
250 new_alias->Alias=tmp;
251 } else {
252 tmp=malloc(new_alias->MinLen+2);
253 if (!tmp) {
254 debuga(_("Not enough memory to store the host name aliasing directives\n"));
255 return(-1);
256 }
257 tmp[0]=ALIAS_PREFIX;
258 if (new_alias->HostName_Prefix) strcpy(tmp+1,new_alias->HostName_Prefix);
259 if (new_alias->HostName_Suffix) {
260 tmp[new_alias->PrefixLen+1]='*';
261 strcpy(tmp+new_alias->PrefixLen+2,new_alias->HostName_Suffix);
262 }
263 new_alias->Alias=tmp;
264 }
265
266 new_alias->Next=NULL;
267 if (prev_alias)
268 prev_alias->Next=new_alias;
269 else
270 FirstAliasName=new_alias;
271 return(1);
272 }
273
274 /*!
275 Store a IPv4 to alias.
276
277 \param ipv4 The IPv4 to match.
278 \param nbits The number of bits in the prefix
279 \param next A pointer to the first character after the address.
280
281 \retval 1 Alias added.
282 \retval 0 Ignore the line.
283 \retval -1 Error.
284 */
285 static int Alias_StoreIpv4(unsigned char *ipv4,int nbits,const char *next)
286 {
287 const char *Replace;
288 const char *ReplaceE;
289 const char *str;
290 struct hostalias_ipv4 *alias;
291 struct hostalias_ipv4 *new_alias;
292 struct hostalias_ipv4 *prev_alias;
293 int i;
294 char *tmp;
295 int len;
296
297 // get the alias
298 Replace=next;
299 while (*Replace==' ' || *Replace=='\t') Replace++;
300 if ((unsigned char)*Replace<' ') {
301 Replace=NULL;
302 } else {
303 for (str=Replace ; *str && (unsigned char)*str>=' ' ; str++);
304 ReplaceE=str;
305 }
306
307 // store more restrictive range first
308 prev_alias=NULL;
309 for (alias=FirstAliasIpv4 ; alias ; alias=alias->Next) {
310 i=(nbits<alias->NBits) ? nbits : alias->NBits;
311 if ((i<8 || memcmp(ipv4,alias->Ip,i/8)==0) && ((i%8)==0 || (ipv4[i/8] ^ alias->Ip[i/8]) & (0xFFU<<(8-i%8)))==0) {
312 break;
313 }
314 prev_alias=alias;
315 }
316
317 // insert into the list
318 new_alias=malloc(sizeof(*new_alias));
319 if (!new_alias) {
320 debuga(_("Not enough memory to store the host name aliasing directives\n"));
321 return(-1);
322 }
323 memcpy(new_alias->Ip,ipv4,4);
324 new_alias->NBits=nbits;
325 if (Replace) {
326 len=(int)(ReplaceE-Replace);
327 tmp=malloc(len+2);
328 if (!tmp) {
329 debuga(_("Not enough memory to store the host name aliasing directives\n"));
330 return(-1);
331 }
332 tmp[0]=ALIAS_PREFIX;
333 memcpy(tmp+1,Replace,len);
334 tmp[len+1]='\0';
335 new_alias->Alias=tmp;
336 } else {
337 tmp=malloc(5*4+1);
338 if (!tmp) {
339 debuga(_("Not enough memory to store the host name aliasing directives\n"));
340 return(-1);
341 }
342 sprintf(tmp,"%c%d.%d.%d.%d/%d",ALIAS_PREFIX,ipv4[0],ipv4[1],ipv4[2],ipv4[3],nbits);
343 new_alias->Alias=tmp;
344 }
345
346 if (prev_alias) {
347 new_alias->Next=prev_alias->Next;
348 prev_alias->Next=new_alias;
349 } else {
350 new_alias->Next=NULL;
351 FirstAliasIpv4=new_alias;
352 }
353 return(1);
354 }
355
356 /*!
357 Store a IPv6 to alias.
358
359 \param ipv6 The IPv6 to match.
360 \param nbits The number of bits in the prefix
361 \param next A pointer to the first character after the address.
362
363 \retval 1 Alias added.
364 \retval 0 Ignore the line.
365 \retval -1 Error.
366 */
367 static int Alias_StoreIpv6(unsigned short *ipv6,int nbits,const char *next)
368 {
369 const char *Replace;
370 const char *ReplaceE;
371 const char *str;
372 struct hostalias_ipv6 *alias;
373 struct hostalias_ipv6 *new_alias;
374 struct hostalias_ipv6 *prev_alias;
375 int i;
376 char *tmp;
377 int len;
378
379 // get the alias
380 Replace=next;
381 while (*Replace==' ' || *Replace=='\t') Replace++;
382 if ((unsigned char)*Replace<' ') {
383 Replace=NULL;
384 } else {
385 for (str=Replace ; *str && (unsigned char)*str>=' ' ; str++);
386 ReplaceE=str;
387 }
388
389 // store more restrictive range first
390 prev_alias=NULL;
391 for (alias=FirstAliasIpv6 ; alias ; alias=alias->Next) {
392 i=(nbits<alias->NBits) ? nbits : alias->NBits;
393 if ((i<16 || memcmp(ipv6,alias->Ip,i/16*2)==0) && ((i%16)==0 || (ipv6[i/16] ^ alias->Ip[i/16]) & (0xFFFFU<<(16-i%16)))==0) {
394 break;
395 }
396 prev_alias=alias;
397 }
398
399 // insert into the list
400 new_alias=malloc(sizeof(*new_alias));
401 if (!new_alias) {
402 debuga(_("Not enough memory to store the host name aliasing directives\n"));
403 return(-1);
404 }
405 memcpy(new_alias->Ip,ipv6,8*sizeof(unsigned short int));
406 new_alias->NBits=nbits;
407 if (Replace) {
408 len=ReplaceE-Replace;
409 tmp=malloc(len+2);
410 if (!tmp) {
411 debuga(_("Not enough memory to store the host name aliasing directives\n"));
412 return(-1);
413 }
414 tmp[0]=ALIAS_PREFIX;
415 memcpy(tmp+1,Replace,len);
416 tmp[len+1]='\0';
417 new_alias->Alias=tmp;
418 } else {
419 tmp=malloc(5*8+5);
420 if (!tmp) {
421 debuga(_("Not enough memory to store the host name aliasing directives\n"));
422 return(-1);
423 }
424 sprintf(tmp,"%c%x:%x:%x:%x:%x:%x:%x:%x/%d",ALIAS_PREFIX,ipv6[0],ipv6[1],ipv6[2],ipv6[3],ipv6[4],ipv6[5],ipv6[6],ipv6[7],nbits);
425 new_alias->Alias=tmp;
426 }
427
428 if (prev_alias) {
429 new_alias->Next=prev_alias->Next;
430 prev_alias->Next=new_alias;
431 } else {
432 new_alias->Next=NULL;
433 FirstAliasIpv6=new_alias;
434 }
435 return(1);
436 }
437
438 #ifdef USE_PCRE
439 /*!
440 Store a regular expression to match the alias.
441
442 \retval 1 Alias added.
443 \retval 0 Ignore the line.
444 \retval -1 Error.
445 */
446 static int Alias_StoreRegexp(char *buf)
447 {
448 char Delimiter;
449 char *End;
450 struct hostalias_regex *alias;
451 struct hostalias_regex *new_alias;
452 struct hostalias_regex **prev_alias;
453 const char *PcreError;
454 int ErrorOffset;
455 char *Replace;
456 int len;
457 char *tmp;
458 int i;
459
460 // find the pattern
461 Delimiter=*buf++;
462 for (End=buf ; *End && *End!=Delimiter ; End++) {
463 if (*End=='\\') {
464 if (End[1]=='\0') {
465 debuga(_("Invalid NUL character found in regular expression\n"));
466 return(-1);
467 }
468 End++; //ignore the escaped character
469 }
470 }
471 if (*End!=Delimiter) {
472 debuga(_("Unterminated regular expression\n"));
473 return(-1);
474 }
475 *End++='\0';
476
477 // find the alias
478 for (Replace=End ; *Replace==' ' || *Replace=='\t' ; Replace++);
479 for (End=Replace ; *End && (unsigned char)*End>' ' ; End++);
480 *End='\0';
481
482 // store it
483 new_alias=malloc(sizeof(*new_alias));
484 if (!new_alias) {
485 debuga(_("Not enough memory to store the host name aliasing directives\n"));
486 return(-1);
487 }
488 new_alias->Next=NULL;
489 new_alias->Re=pcre_compile(buf,0,&PcreError,&ErrorOffset,NULL);
490 if (new_alias->Re==NULL) {
491 debuga(_("Failed to compile the regular expression \"%s\": %s\n"),buf,PcreError);
492 free(new_alias);
493 return(-1);
494 }
495 len=strlen(Replace);
496 tmp=malloc(len+2);
497 if (!tmp) {
498 debuga(_("Not enough memory to store the host name aliasing directives\n"));
499 pcre_free(new_alias->Re);
500 return(-1);
501 }
502 tmp[0]=ALIAS_PREFIX;
503 memcpy(tmp+1,Replace,len);
504 tmp[len+1]='\0';
505 new_alias->Alias=tmp;
506
507 new_alias->SubPartern=false;
508 for (i=1 ; tmp[i] ; i++)
509 // both the sed \1 and the perl $1 replacement operators are accepted
510 if ((tmp[i]=='\\' || tmp[i]=='$') && isdigit(tmp[i+1])) {
511 new_alias->SubPartern=true;
512 break;
513 }
514
515 // chain it
516 prev_alias=&FirstAliasRe;
517 for (alias=FirstAliasRe ; alias ; alias=alias->Next)
518 prev_alias=&alias->Next;
519 *prev_alias=new_alias;
520
521 return(1);
522 }
523 #endif
524
525 /*!
526 Store an alias in the corresponding list.
527
528 \param buf The string to parse and store.
529
530 \retval 0 No error.
531 \retval -1 Error in file.
532 \retval -2 Unknown string type to store.
533 */
534 static int Alias_Store(char *buf)
535 {
536 int type;
537 const char *name;
538 unsigned char ipv4[4];
539 unsigned short int ipv6[8];
540 int nbits;
541 const char *next;
542 int Error=-2;
543
544 if (strncasecmp(buf,"re:",3)==0) {
545 #ifdef USE_PCRE
546 if (Alias_StoreRegexp(buf+3)<0)
547 return(-1);
548 return(0);
549 #else
550 debuga(_("PCRE not compiled in therefore the regular expressions are not available in the host alias file\n"));
551 return(-1);
552 #endif
553 }
554 type=extract_address_mask(buf,&name,ipv4,ipv6,&nbits,&next);
555 if (type<0) {
556 return(-1);
557 }
558
559 if (type==1) {
560 Error=Alias_StoreName(name,next);
561 } else if (type==2) {
562 Error=Alias_StoreIpv4(ipv4,nbits,next);
563 } else if (type==3) {
564 Error=Alias_StoreIpv6(ipv6,nbits,next);
565 }
566 if (Error<0) return(-1);
567 return(0);
568 }
569
570 /*!
571 Read the file containing the host names to alias in the report.
572
573 \param Filename The name of the file.
574 */
575 void read_hostalias(const char *Filename)
576 {
577 FILE *fi;
578 longline line;
579 char *buf;
580
581 if (debug) debuga(_("Reading host alias file \"%s\"\n"),Filename);
582 fi=fopen(Filename,"rt");
583 if (!fi) {
584 debuga(_("Cannot read host name alias file \"%s\": %s\n"),Filename,strerror(errno));
585 exit(EXIT_FAILURE);
586 }
587
588 if ((line=longline_create())==NULL) {
589 debuga(_("Not enough memory to read the host name aliases\n"));
590 exit(EXIT_FAILURE);
591 }
592
593 while ((buf=longline_read(fi,line)) != NULL) {
594 if (Alias_Store(buf)<0) {
595 debuga(_("While reading \"%s\"\n"),Filename);
596 exit(EXIT_FAILURE);
597 }
598 }
599
600 longline_destroy(&line);
601 fclose(fi);
602
603 if (debug) {
604 struct hostalias_name *alias1;
605 struct hostalias_ipv4 *alias4;
606 struct hostalias_ipv6 *alias6;
607
608 debuga(_("List of host names to alias:\n"));
609 for (alias1=FirstAliasName ; alias1 ; alias1=alias1->Next) {
610 if (alias1->HostName_Prefix && alias1->HostName_Suffix)
611 debuga(_(" %s*%s => %s\n"),alias1->HostName_Prefix,alias1->HostName_Suffix,alias1->Alias);
612 else if (alias1->HostName_Prefix)
613 debuga(_(" %s => %s\n"),alias1->HostName_Prefix,alias1->Alias);
614 else
615 debuga(_(" *%s => %s\n"),alias1->HostName_Suffix,alias1->Alias);
616 }
617 for (alias4=FirstAliasIpv4 ; alias4 ; alias4=alias4->Next) {
618 debuga(_(" %d.%d.%d.%d/%d => %s\n"),alias4->Ip[0],alias4->Ip[1],alias4->Ip[2],alias4->Ip[3],alias4->NBits,alias4->Alias);
619 }
620 for (alias6=FirstAliasIpv6 ; alias6 ; alias6=alias6->Next) {
621 debuga(_(" %x:%x:%x:%x:%x:%x:%x:%x/%d => %s\n"),alias6->Ip[0],alias6->Ip[1],alias6->Ip[2],alias6->Ip[3],
622 alias6->Ip[4],alias6->Ip[5],alias6->Ip[6],alias6->Ip[7],alias6->NBits,alias6->Alias);
623 }
624 }
625 }
626
627 /*!
628 Free the memory allocated by read_hostalias().
629 */
630 void free_hostalias(void)
631 {
632 {
633 struct hostalias_name *alias1;
634 struct hostalias_name *next1;
635
636 for (alias1=FirstAliasName ; alias1 ; alias1=next1) {
637 next1=alias1->Next;
638 if (alias1->HostName_Prefix) free((void *)alias1->HostName_Prefix);
639 if (alias1->HostName_Suffix) free((void *)alias1->HostName_Suffix);
640 free((void *)alias1->Alias);
641 free(alias1);
642 }
643 FirstAliasName=NULL;
644 }
645 {
646 struct hostalias_ipv4 *alias4;
647 struct hostalias_ipv4 *next4;
648
649 for (alias4=FirstAliasIpv4 ; alias4 ; alias4=next4) {
650 next4=alias4->Next;
651 free((void *)alias4->Alias);
652 free(alias4);
653 }
654 FirstAliasIpv4=NULL;
655 }
656 {
657 struct hostalias_ipv6 *alias6;
658 struct hostalias_ipv6 *next6;
659
660 for (alias6=FirstAliasIpv6 ; alias6 ; alias6=next6) {
661 next6=alias6->Next;
662 free((void *)alias6->Alias);
663 free(alias6);
664 }
665 FirstAliasIpv6=NULL;
666 }
667 #ifdef USE_PCRE
668 {
669 struct hostalias_regex *alias;
670 struct hostalias_regex *next;
671
672 for (alias=FirstAliasRe ; alias ; alias=next) {
673 next=alias->Next;
674 pcre_free(alias->Re);
675 free((void *)alias->Alias);
676 free(alias);
677 }
678 FirstAliasRe=NULL;
679 }
680 #endif
681 }
682
683 /*!
684 Replace the host name by its alias if it is in our list.
685
686 \param url The host name.
687
688 \return The pointer to the host name or its alias.
689 */
690 static const char *alias_url_name(const char *url,const char *next)
691 {
692 struct hostalias_name *alias;
693 int len;
694
695 len=(int)(next-url);
696 for (alias=FirstAliasName ; alias ; alias=alias->Next) {
697 if (len<alias->MinLen) continue;
698 if (alias->HostName_Prefix) {
699 if (alias->HostName_Suffix) {
700 if (strncasecmp(url,alias->HostName_Prefix,alias->PrefixLen)==0 &&
701 strncasecmp(url+(len-alias->SuffixLen),alias->HostName_Suffix,len)==0) {
702 return(alias->Alias);
703 }
704 } else {
705 if (len==alias->PrefixLen && strncasecmp(url,alias->HostName_Prefix,len)==0) {
706 return(alias->Alias);
707 }
708 }
709 } else if (strncasecmp(url+(len-alias->SuffixLen),alias->HostName_Suffix,len)==0) {
710 return(alias->Alias);
711 }
712 }
713 return(url);
714 }
715
716 /*!
717 Replace the IPv4 address by its alias if it is in our list.
718
719 \param url The host name.
720 \param ipv4 The address.
721
722 \return The pointer to the host name or its alias.
723 */
724 static const char *alias_url_ipv4(const char *url,unsigned char *ipv4)
725 {
726 struct hostalias_ipv4 *alias;
727 int len;
728
729 for (alias=FirstAliasIpv4 ; alias ; alias=alias->Next) {
730 len=alias->NBits;
731 if ((len<8 || memcmp(ipv4,alias->Ip,len/8)==0) && ((len%8)==0 || (ipv4[len/8] ^ alias->Ip[len/8]) & (0xFFU<<(8-len%8)))==0) {
732 return(alias->Alias);
733 }
734 }
735 return(url);
736 }
737
738 /*!
739 Replace the IPv6 address by its alias if it is in our list.
740
741 \param url The host name.
742 \param ipv6 The address.
743
744 \return The pointer to the host name or its alias.
745 */
746 static const char *alias_url_ipv6(const char *url,unsigned short int *ipv6)
747 {
748 struct hostalias_ipv6 *alias;
749 int len;
750 int i;
751
752 for (alias=FirstAliasIpv6 ; alias ; alias=alias->Next) {
753 len=alias->NBits;
754 for (i=len/16-1 ; i>=0 && ipv6[i]==alias->Ip[i] ; i--);
755 if (i>=0) continue;
756 i=len/16;
757 if (i>=8 || len%16==0 || ((ipv6[i] ^ alias->Ip[i]) & (0xFFFF<<(len-i*16)))==0) {
758 return(alias->Alias);
759 }
760 }
761 return(url);
762 }
763
764 #ifdef USE_PCRE
765 /*!
766 Replace the host name by its alias if it is in our list.
767
768 \param url_ptr A pointer to the host name to match. It is replaced
769 by a pointer to the alias if a match is found.
770
771 \return \c True if a match is found or \c false if it failed.
772
773 \warning The function is not thread safe as it may return a static
774 internal buffer.
775 */
776 static bool alias_url_regex(const char **url_ptr)
777 {
778 struct hostalias_regex *alias;
779 int nmatches;
780 const char *url;
781 int url_len;
782 int ovector[30];//size must be a multiple of 3
783 static char Replacement[1024];
784 const char *str;
785 int i;
786 int sub;
787 int repl_idx;
788
789 url=*url_ptr;
790 url_len=strlen(url);
791 for (alias=FirstAliasRe ; alias ; alias=alias->Next) {
792 nmatches=pcre_exec(alias->Re,NULL,url,url_len,0,0,ovector,sizeof(ovector)/sizeof(ovector[0]));
793 if (nmatches>=0) {
794 if (nmatches==0) nmatches=(int)(sizeof(ovector)/sizeof(ovector[0]))/3*2; //only 2/3 of the vector is used by pcre_exec
795 if (nmatches==1 || !alias->SubPartern) { //no subpattern to replace
796 *url_ptr=alias->Alias;
797 } else {
798 repl_idx=0;
799 str=alias->Alias;
800 for (i=0 ; str[i] ; i++) {
801 // both the sed \1 and the perl $1 replacement operators are accepted
802 if ((str[i]=='\\' || str[i]=='$') && isdigit(str[i+1])) {
803 sub=str[++i]-'0';
804 if (sub>=1 && sub<=nmatches) {
805 /*
806 * ovector[sub] is the start position of the match.
807 * ovector[sub+1] is the end position of the match.
808 */
809 sub<<=1;
810 if (repl_idx+ovector[sub+1]-ovector[sub]>=sizeof(Replacement)-1) break;
811 memcpy(Replacement+repl_idx,url+ovector[sub],ovector[sub+1]-ovector[sub]);
812 repl_idx+=ovector[sub+1]-ovector[sub];
813 continue;
814 }
815 }
816 if (repl_idx>=sizeof(Replacement)-1) break;
817 Replacement[repl_idx++]=str[i];
818 }
819 Replacement[repl_idx]='\0';
820 *url_ptr=Replacement;
821 }
822 return(true);
823 }
824 }
825 return(false);
826 }
827 #endif
828
829 /*!
830 Find the beginning of the URL beyond the scheme://
831
832 \param url The url possibly containing a scheme.
833
834 \return The beginning of the url beyond the scheme.
835 */
836 const char *skip_scheme(const char *url)
837 {
838 const char *str;
839
840 /*
841 Skip any scheme:// at the beginning of the URL (see rfc2396 section 3.1).
842 The underscore is not part of the standard but is found in the squid logs as cache_object://.
843 */
844 for (str=url ; *str && (isalnum(*str) || *str=='+' || *str=='-' || *str=='.' || *str=='_') ; str++);
845 if (str[0]==':' && str[1]=='/' && str[2]=='/') {
846 url=str+3;
847 while (*url=='/') url++;
848 }
849 return(url);
850 }
851
852 /*!
853 Get the part of the URL necessary to generate the report.
854
855 \param url The URL as extracted from the report.
856 \param full_url \c True to keep the whole URL. If \c false,
857 the URL is truncated to only keep the host name and port number.
858 */
859 const char *process_url(const char *url,bool full_url)
860 {
861 static char short_url[1024];
862 int i;
863 const char *start;
864 int type;
865 unsigned char ipv4[4];
866 unsigned short int ipv6[8];
867 const char *next;
868
869 start=skip_scheme(url);
870 if (!full_url) {
871 for (i=0 ; i<sizeof(short_url)-1 && start[i] && start[i]!='/' && start[i]!='?' ; i++)
872 short_url[i]=start[i];
873 short_url[i]='\0';
874 start=short_url;
875 #ifdef USE_PCRE
876 if (FirstAliasRe) {
877 if (alias_url_regex(&start)) return(start);
878 }
879 #endif
880 type=extract_address_mask(start,NULL,ipv4,ipv6,NULL,&next);
881 if (type==1) {
882 if (FirstAliasName)
883 start=alias_url_name(start,next);
884 } else if (type==2) {
885 if (FirstAliasIpv4)
886 start=alias_url_ipv4(start,ipv4);
887 } else if (type==3) {
888 if (FirstAliasIpv6)
889 start=alias_url_ipv6(start,ipv6);
890 }
891 }
892 return(start);
893 }
894
895 /*!
896 Extract the host name from the URL.
897
898 \param url The url whose host name must be extracted.
899 \param hostname The buffer to store the host name.
900 \param hostsize The size of the host name buffer.
901
902 \note The function is stupid at this time. It just searches for the first slash
903 in the URL and truncates the URL there. It doesn't take the protocol into account
904 nor the port number nor any user or password information.
905 */
906 void url_hostname(const char *url,char *hostname,int hostsize)
907 {
908 int i;
909
910 hostsize--;
911 for (i=0 ; i<hostsize && url[i] && url[i]!='/' ; i++)
912 hostname[i]=url[i];
913 hostname[i]='\0';
914 }
915