]> granicus.if.org Git - apache/blob - server/util_pcre.c
Merge r1734006 from trunk:
[apache] / server / util_pcre.c
1 /*************************************************
2  *      Perl-Compatible Regular Expressions      *
3  *************************************************/
4
5 /*
6 This is a library of functions to support regular expressions whose syntax
7 and semantics are as close as possible to those of the Perl 5 language. See
8 the file Tech.Notes for some information on the internals.
9
10 This module is a wrapper that provides a POSIX API to the underlying PCRE
11 functions.
12
13 Written by: Philip Hazel <ph10@cam.ac.uk>
14
15            Copyright (c) 1997-2004 University of Cambridge
16
17 -----------------------------------------------------------------------------
18 Redistribution and use in source and binary forms, with or without
19 modification, are permitted provided that the following conditions are met:
20
21     * Redistributions of source code must retain the above copyright notice,
22       this list of conditions and the following disclaimer.
23
24     * Redistributions in binary form must reproduce the above copyright
25       notice, this list of conditions and the following disclaimer in the
26       documentation and/or other materials provided with the distribution.
27
28     * Neither the name of the University of Cambridge nor the names of its
29       contributors may be used to endorse or promote products derived from
30       this software without specific prior written permission.
31
32 THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS "AS IS"
33 AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE
34 IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE
35 ARE DISCLAIMED. IN NO EVENT SHALL THE COPYRIGHT OWNER OR CONTRIBUTORS BE
36 LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR
37 CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF
38 SUBSTITUTE GOODS OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS
39 INTERRUPTION) HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN
40 CONTRACT, STRICT LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE)
41 ARISING IN ANY WAY OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE
42 POSSIBILITY OF SUCH DAMAGE.
43 -----------------------------------------------------------------------------
44 */
45
46 #include "httpd.h"
47 #include "apr_strings.h"
48 #include "apr_tables.h"
49 #include "pcre.h"
50
51 #define APR_WANT_STRFUNC
52 #include "apr_want.h"
53
54 #ifndef POSIX_MALLOC_THRESHOLD
55 #define POSIX_MALLOC_THRESHOLD (10)
56 #endif
57
58 /* Table of error strings corresponding to POSIX error codes; must be
59  * kept in synch with include/ap_regex.h's AP_REG_E* definitions.
60  */
61
62 static const char *const pstring[] = {
63     "",                         /* Dummy for value 0 */
64     "internal error",           /* AP_REG_ASSERT */
65     "failed to get memory",     /* AP_REG_ESPACE */
66     "bad argument",             /* AP_REG_INVARG */
67     "match failed"              /* AP_REG_NOMATCH */
68 };
69
70 AP_DECLARE(apr_size_t) ap_regerror(int errcode, const ap_regex_t *preg,
71                                    char *errbuf, apr_size_t errbuf_size)
72 {
73     const char *message, *addmessage;
74     apr_size_t length, addlength;
75
76     message = (errcode >= (int)(sizeof(pstring) / sizeof(char *))) ?
77               "unknown error code" : pstring[errcode];
78     length = strlen(message) + 1;
79
80     addmessage = " at offset ";
81     addlength = (preg != NULL && (int)preg->re_erroffset != -1) ?
82                 strlen(addmessage) + 6 : 0;
83
84     if (errbuf_size > 0) {
85         if (addlength > 0 && errbuf_size >= length + addlength)
86             apr_snprintf(errbuf, errbuf_size, "%s%s%-6d", message, addmessage,
87                          (int)preg->re_erroffset);
88         else
89             apr_cpystrn(errbuf, message, errbuf_size);
90     }
91
92     return length + addlength;
93 }
94
95
96
97
98 /*************************************************
99  *           Free store held by a regex          *
100  *************************************************/
101
102 AP_DECLARE(void) ap_regfree(ap_regex_t *preg)
103 {
104     (pcre_free)(preg->re_pcre);
105 }
106
107
108
109
110 /*************************************************
111  *            Compile a regular expression       *
112  *************************************************/
113
114 /*
115  * Arguments:
116  *  preg        points to a structure for recording the compiled expression
117  *  pattern     the pattern to compile
118  *  cflags      compilation flags
119  *
120  * Returns:      0 on success
121  *               various non-zero codes on failure
122 */
123 AP_DECLARE(int) ap_regcomp(ap_regex_t * preg, const char *pattern, int cflags)
124 {
125     const char *errorptr;
126     int erroffset;
127     int errcode = 0;
128     int options = PCRE_DUPNAMES;
129
130     if ((cflags & AP_REG_ICASE) != 0)
131         options |= PCRE_CASELESS;
132     if ((cflags & AP_REG_NEWLINE) != 0)
133         options |= PCRE_MULTILINE;
134     if ((cflags & AP_REG_DOTALL) != 0)
135         options |= PCRE_DOTALL;
136
137     preg->re_pcre =
138         pcre_compile2(pattern, options, &errcode, &errorptr, &erroffset, NULL);
139     preg->re_erroffset = erroffset;
140
141     if (preg->re_pcre == NULL) {
142         /*
143          * There doesn't seem to be constants defined for compile time error
144          * codes. 21 is "failed to get memory" according to pcreapi(3).
145          */
146         if (errcode == 21)
147             return AP_REG_ESPACE;
148         return AP_REG_INVARG;
149     }
150
151     pcre_fullinfo((const pcre *)preg->re_pcre, NULL,
152                    PCRE_INFO_CAPTURECOUNT, &(preg->re_nsub));
153     return 0;
154 }
155
156
157
158
159 /*************************************************
160  *              Match a regular expression       *
161  *************************************************/
162
163 /* Unfortunately, PCRE requires 3 ints of working space for each captured
164  * substring, so we have to get and release working store instead of just using
165  * the POSIX structures as was done in earlier releases when PCRE needed only 2
166  * ints. However, if the number of possible capturing brackets is small, use a
167  * block of store on the stack, to reduce the use of malloc/free. The threshold
168  * is in a macro that can be changed at configure time.
169  */
170 AP_DECLARE(int) ap_regexec(const ap_regex_t *preg, const char *string,
171                            apr_size_t nmatch, ap_regmatch_t *pmatch,
172                            int eflags)
173 {
174     return ap_regexec_len(preg, string, strlen(string), nmatch, pmatch,
175                           eflags);
176 }
177
178 AP_DECLARE(int) ap_regexec_len(const ap_regex_t *preg, const char *buff,
179                                apr_size_t len, apr_size_t nmatch,
180                                ap_regmatch_t *pmatch, int eflags)
181 {
182     int rc;
183     int options = 0;
184     int *ovector = NULL;
185     int small_ovector[POSIX_MALLOC_THRESHOLD * 3];
186     int allocated_ovector = 0;
187
188     if ((eflags & AP_REG_NOTBOL) != 0)
189         options |= PCRE_NOTBOL;
190     if ((eflags & AP_REG_NOTEOL) != 0)
191         options |= PCRE_NOTEOL;
192
193     ((ap_regex_t *)preg)->re_erroffset = (apr_size_t)(-1);    /* Only has meaning after compile */
194
195     if (nmatch > 0) {
196         if (nmatch <= POSIX_MALLOC_THRESHOLD) {
197             ovector = &(small_ovector[0]);
198         }
199         else {
200             ovector = (int *)malloc(sizeof(int) * nmatch * 3);
201             if (ovector == NULL)
202                 return AP_REG_ESPACE;
203             allocated_ovector = 1;
204         }
205     }
206
207     rc = pcre_exec((const pcre *)preg->re_pcre, NULL, buff, (int)len,
208                    0, options, ovector, nmatch * 3);
209
210     if (rc == 0)
211         rc = nmatch;            /* All captured slots were filled in */
212
213     if (rc >= 0) {
214         apr_size_t i;
215         for (i = 0; i < (apr_size_t)rc; i++) {
216             pmatch[i].rm_so = ovector[i * 2];
217             pmatch[i].rm_eo = ovector[i * 2 + 1];
218         }
219         if (allocated_ovector)
220             free(ovector);
221         for (; i < nmatch; i++)
222             pmatch[i].rm_so = pmatch[i].rm_eo = -1;
223         return 0;
224     }
225
226     else {
227         if (allocated_ovector)
228             free(ovector);
229         switch (rc) {
230         case PCRE_ERROR_NOMATCH:
231             return AP_REG_NOMATCH;
232         case PCRE_ERROR_NULL:
233             return AP_REG_INVARG;
234         case PCRE_ERROR_BADOPTION:
235             return AP_REG_INVARG;
236         case PCRE_ERROR_BADMAGIC:
237             return AP_REG_INVARG;
238         case PCRE_ERROR_UNKNOWN_NODE:
239             return AP_REG_ASSERT;
240         case PCRE_ERROR_NOMEMORY:
241             return AP_REG_ESPACE;
242 #ifdef PCRE_ERROR_MATCHLIMIT
243         case PCRE_ERROR_MATCHLIMIT:
244             return AP_REG_ESPACE;
245 #endif
246 #ifdef PCRE_ERROR_BADUTF8
247         case PCRE_ERROR_BADUTF8:
248             return AP_REG_INVARG;
249 #endif
250 #ifdef PCRE_ERROR_BADUTF8_OFFSET
251         case PCRE_ERROR_BADUTF8_OFFSET:
252             return AP_REG_INVARG;
253 #endif
254         default:
255             return AP_REG_ASSERT;
256         }
257     }
258 }
259
260 AP_DECLARE(int) ap_regname(const ap_regex_t *preg,
261                            apr_array_header_t *names, const char *prefix,
262                            int upper)
263 {
264     int namecount;
265     int nameentrysize;
266     int i;
267     char *nametable;
268
269     pcre_fullinfo((const pcre *)preg->re_pcre, NULL,
270                        PCRE_INFO_NAMECOUNT, &namecount);
271     pcre_fullinfo((const pcre *)preg->re_pcre, NULL,
272                        PCRE_INFO_NAMEENTRYSIZE, &nameentrysize);
273     pcre_fullinfo((const pcre *)preg->re_pcre, NULL,
274                        PCRE_INFO_NAMETABLE, &nametable);
275
276     for (i = 0; i < namecount; i++) {
277         const char *offset = nametable + i * nameentrysize;
278         int capture = ((offset[0] << 8) + offset[1]);
279         while (names->nelts <= capture) {
280             apr_array_push(names);
281         }
282         if (upper || prefix) {
283             char *name = ((char **) names->elts)[capture] =
284                     prefix ? apr_pstrcat(names->pool, prefix, offset + 2,
285                             NULL) :
286                             apr_pstrdup(names->pool, offset + 2);
287             if (upper) {
288                 ap_str_toupper(name);
289             }
290         }
291         else {
292             ((const char **)names->elts)[capture] = offset + 2;
293         }
294     }
295
296     return namecount;
297 }
298
299 /* End of pcreposix.c */