From 2dbe1c54369c0d726b72120411e6dcbfb41a8d8d Mon Sep 17 00:00:00 2001 From: Clint Adams Date: Tue, 16 Mar 2004 21:44:40 +0000 Subject: * 19641: Src/Modules/pcre.c: implement -pcre-match conditional suggested by Oliver. --- ChangeLog | 5 ++++ Src/Modules/pcre.c | 80 ++++++++++++++++++++++++++++++++++++++++++++---------- 2 files changed, 71 insertions(+), 14 deletions(-) diff --git a/ChangeLog b/ChangeLog index d3b2f66a9..a9379dabf 100644 --- a/ChangeLog +++ b/ChangeLog @@ -1,3 +1,8 @@ +2004-03-16 Clint Adams + + * 19641: Src/Modules/pcre.c: implement -pcre-match conditional + suggested by Oliver. + 2004-03-16 Peter Stephenson * 19629: Src/Modules/zpty.c: implement Oliver's research into diff --git a/Src/Modules/pcre.c b/Src/Modules/pcre.c index 41c37b03c..b6c63f976 100644 --- a/Src/Modules/pcre.c +++ b/Src/Modules/pcre.c @@ -31,6 +31,8 @@ #include "pcre.mdh" #include "pcre.pro" +#define CPCRE_PLAIN 0 + /**/ #if defined(HAVE_PCRE_COMPILE) && defined(HAVE_PCRE_EXEC) #include @@ -97,12 +99,32 @@ bin_pcre_study(char *nam, char **args, Options ops, int func) /**/ #endif /* !HAVE_PCRE_STUDY */ +/**/ +static int +zpcre_get_substrings(char *arg, int *ovec, int ret, char *receptacle) +{ + char **captures, **matches; + + if(!pcre_get_substring_list(arg, ovec, ret, (const char ***)&captures)) { + + matches = zarrdup(&captures[1]); /* first one would be entire string */ + if (receptacle == NULL) + setaparam("match", matches); + else + setaparam(receptacle, matches); + + pcre_free_substring_list((const char **)captures); + } + + return 0; +} + /**/ static int bin_pcre_match(char *nam, char **args, Options ops, int func) { int ret, capcount, *ovec, ovecsize; - char **captures, **matches, *receptacle = NULL; + char *receptacle = NULL; if(OPT_ISSET(ops,'a')) { receptacle = *args++; @@ -112,7 +134,7 @@ bin_pcre_match(char *nam, char **args, Options ops, int func) } } - if (ret = pcre_fullinfo(pcre_pattern, pcre_hints, PCRE_INFO_CAPTURECOUNT, &capcount)) + if ((ret = pcre_fullinfo(pcre_pattern, pcre_hints, PCRE_INFO_CAPTURECOUNT, &capcount))) { zwarnnam(nam, "error %d in fullinfo", NULL, ret); return 1; @@ -126,17 +148,8 @@ bin_pcre_match(char *nam, char **args, Options ops, int func) if (ret==0) return 0; else if (ret==PCRE_ERROR_NOMATCH) return 1; /* no match */ else if (ret>0) { - if(!pcre_get_substring_list(*args, ovec, ret, (const char ***)&captures)) { - - matches = zarrdup(&captures[1]); /* first one would be entire string */ - if (receptacle == NULL) - setaparam("match", matches); - else - setaparam(receptacle, matches); - - pcre_free_substring_list((const char **)captures); - return 0; - } + zpcre_get_substrings(*args, ovec, ret, receptacle); + return 0; } else { zwarnnam(nam, "error in pcre_exec", NULL, 0); @@ -146,12 +159,44 @@ bin_pcre_match(char *nam, char **args, Options ops, int func) return 1; } +/**/ +static int +cond_pcre_match(char **a, int id) +{ + pcre *pcre_pat; + const char *pcre_err; + char *lhstr, *rhre; + int r = 0, pcre_opts = 0, pcre_errptr, capcnt, *ov, ovsize; + + lhstr = cond_str(a,0,0); + rhre = cond_str(a,1,0); + + switch(id) { + case CPCRE_PLAIN: + pcre_pat = pcre_compile(rhre, pcre_opts, &pcre_err, &pcre_errptr, NULL); + pcre_fullinfo(pcre_pat, NULL, PCRE_INFO_CAPTURECOUNT, &capcnt); + ovsize = (capcnt+1)*3; + ov = zalloc(ovsize*sizeof(int)); + r = pcre_exec(pcre_pat, NULL, lhstr, strlen(lhstr), 0, 0, ov, ovsize); + if (r==0) return 1; + else if (r==PCRE_ERROR_NOMATCH) return 0; /* no match */ + else if (r>0) { + zpcre_get_substrings(lhstr, ov, r, NULL); + return 1; + } + break; + } + + return 0; +} + /**/ #else /* !(HAVE_PCRE_COMPILE && HAVE_PCRE_EXEC) */ # define bin_pcre_compile bin_notavail # define bin_pcre_study bin_notavail # define bin_pcre_match bin_notavail +# define cond_pcre_match cond_match /**/ #endif /* !(HAVE_PCRE_COMPILE && HAVE_PCRE_EXEC) */ @@ -162,6 +207,11 @@ static struct builtin bintab[] = { BUILTIN("pcre_match", 0, bin_pcre_match, 1, 2, 0, "a", NULL) }; +static struct conddef cotab[] = { + CONDDEF("pcre-match", CONDF_INFIX, cond_pcre_match, 0, 0, CPCRE_PLAIN) +}; + + /**/ int setup_(Module m) @@ -173,7 +223,8 @@ setup_(Module m) int boot_(Module m) { - return !addbuiltins(m->nam, bintab, sizeof(bintab)/sizeof(*bintab)); + return !addbuiltins(m->nam, bintab, sizeof(bintab)/sizeof(*bintab)) || + !addconddefs(m->nam, cotab, sizeof(cotab)/sizeof(*cotab)); } /**/ @@ -181,6 +232,7 @@ int cleanup_(Module m) { deletebuiltins(m->nam, bintab, sizeof(bintab)/sizeof(*bintab)); + deleteconddefs(m->nam, cotab, sizeof(cotab)/sizeof(*cotab)); return 0; } -- cgit 1.4.1