#include "include/conf.h"
#include "include/defs.h"
+#ifdef HAVE_PCRE_H
+#include <pcre.h>
+#define USE_PCRE
+#else
+#error "PCRE not compiled in"
+#endif
/*!
A host name and the name to report.
const char *Alias;
};
+#ifdef USE_PCRE
+/*!
+A regular expression.
+*/
+struct hostalias_regex
+{
+ //! The next regular expression to test.
+ struct hostalias_regex *Next;
+ //! The regular expression to match against the host name.
+ pcre *Re;
+ //! The replacement name.
+ const char *Alias;
+};
+#endif
+
//! The first host name.
static struct hostalias_name *FirstAliasName=NULL;
//! The first IPv4 address.
//! The first IPv§ address.
static struct hostalias_ipv6 *FirstAliasIpv6=NULL;
+#ifdef USE_PCRE
+static struct hostalias_regex *FirstAliasRe=NULL;
+#endif
+
/*!
Store a name to alias.
return(1);
}
+#ifdef USE_PCRE
/*!
-Read the file containing the host names to alias in the report.
+Store a regular expression to match the alias.
-\param Filename The name of the file.
+\retval 1 Alias added.
+\retval 0 Ignore the line.
+\retval -1 Error.
*/
-void read_hostalias(const char *Filename)
+static int Alias_StoreRegexp(char *buf)
+{
+ char Delimiter;
+ char *End;
+ struct hostalias_regex *alias;
+ struct hostalias_regex *new_alias;
+ struct hostalias_regex **prev_alias;
+ const char *PcreError;
+ int ErrorOffset;
+ char *Replace;
+ int len;
+ char *tmp;
+
+ // find the pattern
+ Delimiter=*buf++;
+ for (End=buf ; *End && *End!=Delimiter ; End++) {
+ if (*End=='\\') {
+ if (End[1]=='\0') {
+ debuga(_("Invalid NUL character found in regular expression\n"));
+ return(-1);
+ }
+ End++; //ignore the escaped character
+ }
+ }
+ if (*End!=Delimiter) {
+ debuga(_("Unterminated regular expression\n"));
+ return(-1);
+ }
+ *End++='\0';
+
+ // find the alias
+ for (Replace=End ; *Replace==' ' || *Replace=='\t' ; Replace++);
+ for (End=Replace ; *End && (unsigned char)*End>' ' ; End++);
+ *End='\0';
+
+ // store it
+ new_alias=malloc(sizeof(*new_alias));
+ if (!new_alias) {
+ debuga(_("Not enough memory to store the host name aliasing directives\n"));
+ return(-1);
+ }
+ new_alias->Next=NULL;
+ new_alias->Re=pcre_compile(buf,0,&PcreError,&ErrorOffset,NULL);
+ if (new_alias->Re==NULL) {
+ debuga(_("Failed to compile the regular expression: %s\n"),PcreError);
+ free(new_alias);
+ return(-1);
+ }
+ len=strlen(Replace);
+ tmp=malloc(len+2);
+ if (!tmp) {
+ debuga(_("Not enough memory to store the host name aliasing directives\n"));
+ pcre_free(new_alias->Re);
+ return(-1);
+ }
+ tmp[0]=ALIAS_PREFIX;
+ memcpy(tmp+1,Replace,len);
+ tmp[len+1]='\0';
+ new_alias->Alias=tmp;
+
+ // chain it
+ prev_alias=&FirstAliasRe;
+ for (alias=FirstAliasRe ; alias ; alias=alias->Next)
+ prev_alias=&alias->Next;
+ *prev_alias=new_alias;
+
+ return(1);
+}
+#endif
+
+/*!
+Store an alias in the corresponding list.
+
+\param buf The string to parse and store.
+
+\retval 0 No error.
+\retval -1 Error in file.
+*/
+static int Alias_Store(char *buf)
{
- FILE *fi;
- longline line;
- char *buf;
int type;
const char *name;
unsigned char ipv4[4];
unsigned short int ipv6[8];
int nbits;
const char *next;
+ int Error;
+
+#ifdef USE_PCRE
+ if (strncasecmp(buf,"re:",3)==0) {
+ if (Alias_StoreRegexp(buf+3)<0)
+ return(-1);
+ return(0);
+ }
+#endif
+ type=extract_address_mask(buf,&name,ipv4,ipv6,&nbits,&next);
+ if (type<0) {
+ return(-1);
+ }
+
+ if (type==1) {
+ Error=Alias_StoreName(name,next);
+ } else if (type==2) {
+ Error=Alias_StoreIpv4(ipv4,nbits,next);
+ } else if (type==3) {
+ Error=Alias_StoreIpv6(ipv6,nbits,next);
+ }
+ if (Error<0) return(-1);
+ return(0);
+}
+
+/*!
+Read the file containing the host names to alias in the report.
+
+\param Filename The name of the file.
+*/
+void read_hostalias(const char *Filename)
+{
+ FILE *fi;
+ longline line;
+ char *buf;
if (debug) debuga(_("Reading host alias file \"%s\"\n"),Filename);
fi=fopen(Filename,"rt");
}
while ((buf=longline_read(fi,line)) != NULL) {
- type=extract_address_mask(buf,&name,ipv4,ipv6,&nbits,&next);
- if (type<0) {
+ if (Alias_Store(buf)<0) {
debuga(_("While reading \"%s\"\n"),Filename);
exit(EXIT_FAILURE);
}
-
- if (type==1) {
- Alias_StoreName(name,next);
- } else if (type==2) {
- Alias_StoreIpv4(ipv4,nbits,next);
- } else if (type==3) {
- Alias_StoreIpv6(ipv6,nbits,next);
- }
}
longline_destroy(&line);
*/
void free_hostalias(void)
{
- struct hostalias_name *alias1;
- struct hostalias_name *next1;
- struct hostalias_ipv4 *alias4;
- struct hostalias_ipv4 *next4;
- struct hostalias_ipv6 *alias6;
- struct hostalias_ipv6 *next6;
-
- for (alias1=FirstAliasName ; alias1 ; alias1=next1) {
- next1=alias1->Next;
- if (alias1->HostName_Prefix) free((void *)alias1->HostName_Prefix);
- if (alias1->HostName_Suffix) free((void *)alias1->HostName_Suffix);
- free((void *)alias1->Alias);
- free(alias1);
- }
- FirstAliasName=NULL;
- for (alias4=FirstAliasIpv4 ; alias4 ; alias4=next4) {
- next4=alias4->Next;
- free((void *)alias4->Alias);
- free(alias4);
- }
- FirstAliasIpv4=NULL;
- for (alias6=FirstAliasIpv6 ; alias6 ; alias6=next6) {
- next6=alias6->Next;
- free((void *)alias6->Alias);
- free(alias6);
- }
- FirstAliasIpv6=NULL;
+ {
+ struct hostalias_name *alias1;
+ struct hostalias_name *next1;
+
+ for (alias1=FirstAliasName ; alias1 ; alias1=next1) {
+ next1=alias1->Next;
+ if (alias1->HostName_Prefix) free((void *)alias1->HostName_Prefix);
+ if (alias1->HostName_Suffix) free((void *)alias1->HostName_Suffix);
+ free((void *)alias1->Alias);
+ free(alias1);
+ }
+ FirstAliasName=NULL;
+ }
+ {
+ struct hostalias_ipv4 *alias4;
+ struct hostalias_ipv4 *next4;
+
+ for (alias4=FirstAliasIpv4 ; alias4 ; alias4=next4) {
+ next4=alias4->Next;
+ free((void *)alias4->Alias);
+ free(alias4);
+ }
+ FirstAliasIpv4=NULL;
+ }
+ {
+ struct hostalias_ipv6 *alias6;
+ struct hostalias_ipv6 *next6;
+
+ for (alias6=FirstAliasIpv6 ; alias6 ; alias6=next6) {
+ next6=alias6->Next;
+ free((void *)alias6->Alias);
+ free(alias6);
+ }
+ FirstAliasIpv6=NULL;
+ }
+#ifdef USE_PCRE
+ {
+ struct hostalias_regex *alias;
+ struct hostalias_regex *next;
+
+ for (alias=FirstAliasRe ; alias ; alias=next) {
+ next=alias->Next;
+ pcre_free(alias->Re);
+ free((void *)alias->Alias);
+ free(alias);
+ }
+ FirstAliasRe=NULL;
+ }
+#endif
}
/*!
\return The pointer to the host name or its alias.
*/
-const char *alias_url_name(const char *url,const char *next)
+static const char *alias_url_name(const char *url,const char *next)
{
struct hostalias_name *alias;
int len;
\return The pointer to the host name or its alias.
*/
-const char *alias_url_ipv4(const char *url,unsigned char *ipv4)
+static const char *alias_url_ipv4(const char *url,unsigned char *ipv4)
{
struct hostalias_ipv4 *alias;
int len;
\return The pointer to the host name or its alias.
*/
-const char *alias_url_ipv6(const char *url,unsigned short int *ipv6)
+static const char *alias_url_ipv6(const char *url,unsigned short int *ipv6)
{
struct hostalias_ipv6 *alias;
int len;
return(url);
}
+#ifdef USE_PCRE
+/*!
+Replace the host name by its alias if it is in our list.
+
+\param url The host name.
+
+\return The pointer to the host name or its alias.
+*/
+static const char *alias_url_regex(const char *url)
+{
+ struct hostalias_regex *alias;
+
+ for (alias=FirstAliasRe ; alias ; alias=alias->Next) {
+ if (pcre_exec(alias->Re,NULL,url,strlen(url),0,0,NULL,0)==0) {
+ return(alias->Alias);
+ }
+ }
+ return(url);
+}
+#endif
+
/*!
Find the beginning of the URL beyond the scheme://
if (!full_url) {
for (str=(char *)start ; *str && *str!='/' && *str!='?' ; str++);
*str='\0';
+#ifdef USE_PCRE
+ if (FirstAliasRe) {
+ address=alias_url_regex(start);
+ if (address!=start) return(address);
+ }
+#endif
type=extract_address_mask(start,&address,ipv4,ipv6,NULL,&next);
if (type==1) {
if (FirstAliasName)
projects / thirdparty / sarg.git / commitdiff
