1 // Copyright 2003-2009 Google Inc.  All rights reserved.
2 // Use of this source code is governed by a BSD-style
3 // license that can be found in the LICENSE file.
4 
5 // This is a variant of PCRE's pcrecpp.cc, originally written at Google.
6 // The main changes are the addition of the HitLimit method and
7 // compilation as PCRE in namespace re2.
8 
9 #include <assert.h>
10 #include <ctype.h>
11 #include <errno.h>
12 #include <stdlib.h>
13 #include <string.h>
14 #include <limits>
15 #include <string>
16 #include <utility>
17 
18 #include "util/util.h"
19 #include "util/flags.h"
20 #include "util/logging.h"
21 #include "util/pcre.h"
22 #include "util/strutil.h"
23 
24 // Silence warnings about the wacky formatting in the operator() functions.
25 #if !defined(__clang__) && defined(__GNUC__) && __GNUC__ >= 6
26 #pragma GCC diagnostic ignored "-Wmisleading-indentation"
27 #endif
28 
29 #define PCREPORT(level) LOG(level)
30 
31 // Default PCRE limits.
32 // Defaults chosen to allow a plausible amount of CPU and
33 // not exceed main thread stacks.  Note that other threads
34 // often have smaller stacks, and therefore tightening
35 // regexp_stack_limit may frequently be necessary.
36 DEFINE_FLAG(int, regexp_stack_limit, 256 << 10,
37             "default PCRE stack limit (bytes)");
38 DEFINE_FLAG(int, regexp_match_limit, 1000000,
39             "default PCRE match limit (function calls)");
40 
41 #ifndef USEPCRE
42 
43 // Fake just enough of the PCRE API to allow this file to build. :)
44 
45 struct pcre_extra {
46   int flags;
47   int match_limit;
48   int match_limit_recursion;
49 };
50 
51 #define PCRE_EXTRA_MATCH_LIMIT 0
52 #define PCRE_EXTRA_MATCH_LIMIT_RECURSION 0
53 #define PCRE_ANCHORED 0
54 #define PCRE_NOTEMPTY 0
55 #define PCRE_ERROR_NOMATCH 1
56 #define PCRE_ERROR_MATCHLIMIT 2
57 #define PCRE_ERROR_RECURSIONLIMIT 3
58 #define PCRE_INFO_CAPTURECOUNT 0
59 
pcre_free(void *)60 void pcre_free(void*) {
61 }
62 
pcre_compile(const char *,int,const char **,int *,const unsigned char *)63 pcre* pcre_compile(const char*, int, const char**, int*, const unsigned char*) {
64   return NULL;
65 }
66 
pcre_exec(const pcre *,const pcre_extra *,const char *,int,int,int,int *,int)67 int pcre_exec(const pcre*, const pcre_extra*, const char*, int, int, int, int*, int) {
68   return 0;
69 }
70 
pcre_fullinfo(const pcre *,const pcre_extra *,int,void *)71 int pcre_fullinfo(const pcre*, const pcre_extra*, int, void*) {
72   return 0;
73 }
74 
75 #endif
76 
77 namespace re2 {
78 
79 // Maximum number of args we can set
80 static const int kMaxArgs = 16;
81 static const int kVecSize = (1 + kMaxArgs) * 3;  // results + PCRE workspace
82 
83 // Approximate size of a recursive invocation of PCRE's
84 // internal "match()" frame.  This varies depending on the
85 // compiler and architecture, of course, so the constant is
86 // just a conservative estimate.  To find the exact number,
87 // run regexp_unittest with --regexp_stack_limit=0 under
88 // a debugger and look at the frames when it crashes.
89 // The exact frame size was 656 in production on 2008/02/03.
90 static const int kPCREFrameSize = 700;
91 
92 // Special name for missing C++ arguments.
93 PCRE::Arg PCRE::no_more_args((void*)NULL);
94 
95 const PCRE::PartialMatchFunctor PCRE::PartialMatch = { };
96 const PCRE::FullMatchFunctor PCRE::FullMatch = { } ;
97 const PCRE::ConsumeFunctor PCRE::Consume = { };
98 const PCRE::FindAndConsumeFunctor PCRE::FindAndConsume = { };
99 
100 // If a regular expression has no error, its error_ field points here
101 static const std::string empty_string;
102 
Init(const char * pattern,Option options,int match_limit,int stack_limit,bool report_errors)103 void PCRE::Init(const char* pattern, Option options, int match_limit,
104               int stack_limit, bool report_errors) {
105   pattern_ = pattern;
106   options_ = options;
107   match_limit_ = match_limit;
108   stack_limit_ = stack_limit;
109   hit_limit_ = false;
110   error_ = &empty_string;
111   report_errors_ = report_errors;
112   re_full_ = NULL;
113   re_partial_ = NULL;
114 
115   if (options & ~(EnabledCompileOptions | EnabledExecOptions)) {
116     error_ = new std::string("illegal regexp option");
117     PCREPORT(ERROR)
118         << "Error compiling '" << pattern << "': illegal regexp option";
119   } else {
120     re_partial_ = Compile(UNANCHORED);
121     if (re_partial_ != NULL) {
122       re_full_ = Compile(ANCHOR_BOTH);
123     }
124   }
125 }
126 
PCRE(const char * pattern)127 PCRE::PCRE(const char* pattern) {
128   Init(pattern, None, 0, 0, true);
129 }
PCRE(const char * pattern,Option option)130 PCRE::PCRE(const char* pattern, Option option) {
131   Init(pattern, option, 0, 0, true);
132 }
PCRE(const std::string & pattern)133 PCRE::PCRE(const std::string& pattern) {
134   Init(pattern.c_str(), None, 0, 0, true);
135 }
PCRE(const std::string & pattern,Option option)136 PCRE::PCRE(const std::string& pattern, Option option) {
137   Init(pattern.c_str(), option, 0, 0, true);
138 }
PCRE(const std::string & pattern,const PCRE_Options & re_option)139 PCRE::PCRE(const std::string& pattern, const PCRE_Options& re_option) {
140   Init(pattern.c_str(), re_option.option(), re_option.match_limit(),
141        re_option.stack_limit(), re_option.report_errors());
142 }
143 
PCRE(const char * pattern,const PCRE_Options & re_option)144 PCRE::PCRE(const char *pattern, const PCRE_Options& re_option) {
145   Init(pattern, re_option.option(), re_option.match_limit(),
146        re_option.stack_limit(), re_option.report_errors());
147 }
148 
~PCRE()149 PCRE::~PCRE() {
150   if (re_full_ != NULL)         pcre_free(re_full_);
151   if (re_partial_ != NULL)      pcre_free(re_partial_);
152   if (error_ != &empty_string)  delete error_;
153 }
154 
Compile(Anchor anchor)155 pcre* PCRE::Compile(Anchor anchor) {
156   // Special treatment for anchoring.  This is needed because at
157   // runtime pcre only provides an option for anchoring at the
158   // beginning of a string.
159   //
160   // There are three types of anchoring we want:
161   //    UNANCHORED      Compile the original pattern, and use
162   //                    a pcre unanchored match.
163   //    ANCHOR_START    Compile the original pattern, and use
164   //                    a pcre anchored match.
165   //    ANCHOR_BOTH     Tack a "\z" to the end of the original pattern
166   //                    and use a pcre anchored match.
167 
168   const char* error = "";
169   int eoffset;
170   pcre* re;
171   if (anchor != ANCHOR_BOTH) {
172     re = pcre_compile(pattern_.c_str(),
173                       (options_ & EnabledCompileOptions),
174                       &error, &eoffset, NULL);
175   } else {
176     // Tack a '\z' at the end of PCRE.  Parenthesize it first so that
177     // the '\z' applies to all top-level alternatives in the regexp.
178     std::string wrapped = "(?:";  // A non-counting grouping operator
179     wrapped += pattern_;
180     wrapped += ")\\z";
181     re = pcre_compile(wrapped.c_str(),
182                       (options_ & EnabledCompileOptions),
183                       &error, &eoffset, NULL);
184   }
185   if (re == NULL) {
186     if (error_ == &empty_string) error_ = new std::string(error);
187     PCREPORT(ERROR) << "Error compiling '" << pattern_ << "': " << error;
188   }
189   return re;
190 }
191 
192 /***** Convenience interfaces *****/
193 
operator ()(const StringPiece & text,const PCRE & re,const Arg & a0,const Arg & a1,const Arg & a2,const Arg & a3,const Arg & a4,const Arg & a5,const Arg & a6,const Arg & a7,const Arg & a8,const Arg & a9,const Arg & a10,const Arg & a11,const Arg & a12,const Arg & a13,const Arg & a14,const Arg & a15) const194 bool PCRE::FullMatchFunctor::operator ()(const StringPiece& text,
195                                        const PCRE& re,
196                                        const Arg& a0,
197                                        const Arg& a1,
198                                        const Arg& a2,
199                                        const Arg& a3,
200                                        const Arg& a4,
201                                        const Arg& a5,
202                                        const Arg& a6,
203                                        const Arg& a7,
204                                        const Arg& a8,
205                                        const Arg& a9,
206                                        const Arg& a10,
207                                        const Arg& a11,
208                                        const Arg& a12,
209                                        const Arg& a13,
210                                        const Arg& a14,
211                                        const Arg& a15) const {
212   const Arg* args[kMaxArgs];
213   int n = 0;
214   if (&a0 == &no_more_args)  goto done; args[n++] = &a0;
215   if (&a1 == &no_more_args)  goto done; args[n++] = &a1;
216   if (&a2 == &no_more_args)  goto done; args[n++] = &a2;
217   if (&a3 == &no_more_args)  goto done; args[n++] = &a3;
218   if (&a4 == &no_more_args)  goto done; args[n++] = &a4;
219   if (&a5 == &no_more_args)  goto done; args[n++] = &a5;
220   if (&a6 == &no_more_args)  goto done; args[n++] = &a6;
221   if (&a7 == &no_more_args)  goto done; args[n++] = &a7;
222   if (&a8 == &no_more_args)  goto done; args[n++] = &a8;
223   if (&a9 == &no_more_args)  goto done; args[n++] = &a9;
224   if (&a10 == &no_more_args) goto done; args[n++] = &a10;
225   if (&a11 == &no_more_args) goto done; args[n++] = &a11;
226   if (&a12 == &no_more_args) goto done; args[n++] = &a12;
227   if (&a13 == &no_more_args) goto done; args[n++] = &a13;
228   if (&a14 == &no_more_args) goto done; args[n++] = &a14;
229   if (&a15 == &no_more_args) goto done; args[n++] = &a15;
230 done:
231 
232   size_t consumed;
233   int vec[kVecSize] = {};
234   return re.DoMatchImpl(text, ANCHOR_BOTH, &consumed, args, n, vec, kVecSize);
235 }
236 
operator ()(const StringPiece & text,const PCRE & re,const Arg & a0,const Arg & a1,const Arg & a2,const Arg & a3,const Arg & a4,const Arg & a5,const Arg & a6,const Arg & a7,const Arg & a8,const Arg & a9,const Arg & a10,const Arg & a11,const Arg & a12,const Arg & a13,const Arg & a14,const Arg & a15) const237 bool PCRE::PartialMatchFunctor::operator ()(const StringPiece& text,
238                                           const PCRE& re,
239                                           const Arg& a0,
240                                           const Arg& a1,
241                                           const Arg& a2,
242                                           const Arg& a3,
243                                           const Arg& a4,
244                                           const Arg& a5,
245                                           const Arg& a6,
246                                           const Arg& a7,
247                                           const Arg& a8,
248                                           const Arg& a9,
249                                           const Arg& a10,
250                                           const Arg& a11,
251                                           const Arg& a12,
252                                           const Arg& a13,
253                                           const Arg& a14,
254                                           const Arg& a15) const {
255   const Arg* args[kMaxArgs];
256   int n = 0;
257   if (&a0 == &no_more_args)  goto done; args[n++] = &a0;
258   if (&a1 == &no_more_args)  goto done; args[n++] = &a1;
259   if (&a2 == &no_more_args)  goto done; args[n++] = &a2;
260   if (&a3 == &no_more_args)  goto done; args[n++] = &a3;
261   if (&a4 == &no_more_args)  goto done; args[n++] = &a4;
262   if (&a5 == &no_more_args)  goto done; args[n++] = &a5;
263   if (&a6 == &no_more_args)  goto done; args[n++] = &a6;
264   if (&a7 == &no_more_args)  goto done; args[n++] = &a7;
265   if (&a8 == &no_more_args)  goto done; args[n++] = &a8;
266   if (&a9 == &no_more_args)  goto done; args[n++] = &a9;
267   if (&a10 == &no_more_args) goto done; args[n++] = &a10;
268   if (&a11 == &no_more_args) goto done; args[n++] = &a11;
269   if (&a12 == &no_more_args) goto done; args[n++] = &a12;
270   if (&a13 == &no_more_args) goto done; args[n++] = &a13;
271   if (&a14 == &no_more_args) goto done; args[n++] = &a14;
272   if (&a15 == &no_more_args) goto done; args[n++] = &a15;
273 done:
274 
275   size_t consumed;
276   int vec[kVecSize] = {};
277   return re.DoMatchImpl(text, UNANCHORED, &consumed, args, n, vec, kVecSize);
278 }
279 
operator ()(StringPiece * input,const PCRE & pattern,const Arg & a0,const Arg & a1,const Arg & a2,const Arg & a3,const Arg & a4,const Arg & a5,const Arg & a6,const Arg & a7,const Arg & a8,const Arg & a9,const Arg & a10,const Arg & a11,const Arg & a12,const Arg & a13,const Arg & a14,const Arg & a15) const280 bool PCRE::ConsumeFunctor::operator ()(StringPiece* input,
281                                      const PCRE& pattern,
282                                      const Arg& a0,
283                                      const Arg& a1,
284                                      const Arg& a2,
285                                      const Arg& a3,
286                                      const Arg& a4,
287                                      const Arg& a5,
288                                      const Arg& a6,
289                                      const Arg& a7,
290                                      const Arg& a8,
291                                      const Arg& a9,
292                                      const Arg& a10,
293                                      const Arg& a11,
294                                      const Arg& a12,
295                                      const Arg& a13,
296                                      const Arg& a14,
297                                      const Arg& a15) const {
298   const Arg* args[kMaxArgs];
299   int n = 0;
300   if (&a0 == &no_more_args)  goto done; args[n++] = &a0;
301   if (&a1 == &no_more_args)  goto done; args[n++] = &a1;
302   if (&a2 == &no_more_args)  goto done; args[n++] = &a2;
303   if (&a3 == &no_more_args)  goto done; args[n++] = &a3;
304   if (&a4 == &no_more_args)  goto done; args[n++] = &a4;
305   if (&a5 == &no_more_args)  goto done; args[n++] = &a5;
306   if (&a6 == &no_more_args)  goto done; args[n++] = &a6;
307   if (&a7 == &no_more_args)  goto done; args[n++] = &a7;
308   if (&a8 == &no_more_args)  goto done; args[n++] = &a8;
309   if (&a9 == &no_more_args)  goto done; args[n++] = &a9;
310   if (&a10 == &no_more_args) goto done; args[n++] = &a10;
311   if (&a11 == &no_more_args) goto done; args[n++] = &a11;
312   if (&a12 == &no_more_args) goto done; args[n++] = &a12;
313   if (&a13 == &no_more_args) goto done; args[n++] = &a13;
314   if (&a14 == &no_more_args) goto done; args[n++] = &a14;
315   if (&a15 == &no_more_args) goto done; args[n++] = &a15;
316 done:
317 
318   size_t consumed;
319   int vec[kVecSize] = {};
320   if (pattern.DoMatchImpl(*input, ANCHOR_START, &consumed,
321                           args, n, vec, kVecSize)) {
322     input->remove_prefix(consumed);
323     return true;
324   } else {
325     return false;
326   }
327 }
328 
operator ()(StringPiece * input,const PCRE & pattern,const Arg & a0,const Arg & a1,const Arg & a2,const Arg & a3,const Arg & a4,const Arg & a5,const Arg & a6,const Arg & a7,const Arg & a8,const Arg & a9,const Arg & a10,const Arg & a11,const Arg & a12,const Arg & a13,const Arg & a14,const Arg & a15) const329 bool PCRE::FindAndConsumeFunctor::operator ()(StringPiece* input,
330                                             const PCRE& pattern,
331                                             const Arg& a0,
332                                             const Arg& a1,
333                                             const Arg& a2,
334                                             const Arg& a3,
335                                             const Arg& a4,
336                                             const Arg& a5,
337                                             const Arg& a6,
338                                             const Arg& a7,
339                                             const Arg& a8,
340                                             const Arg& a9,
341                                             const Arg& a10,
342                                             const Arg& a11,
343                                             const Arg& a12,
344                                             const Arg& a13,
345                                             const Arg& a14,
346                                             const Arg& a15) const {
347   const Arg* args[kMaxArgs];
348   int n = 0;
349   if (&a0 == &no_more_args)  goto done; args[n++] = &a0;
350   if (&a1 == &no_more_args)  goto done; args[n++] = &a1;
351   if (&a2 == &no_more_args)  goto done; args[n++] = &a2;
352   if (&a3 == &no_more_args)  goto done; args[n++] = &a3;
353   if (&a4 == &no_more_args)  goto done; args[n++] = &a4;
354   if (&a5 == &no_more_args)  goto done; args[n++] = &a5;
355   if (&a6 == &no_more_args)  goto done; args[n++] = &a6;
356   if (&a7 == &no_more_args)  goto done; args[n++] = &a7;
357   if (&a8 == &no_more_args)  goto done; args[n++] = &a8;
358   if (&a9 == &no_more_args)  goto done; args[n++] = &a9;
359   if (&a10 == &no_more_args) goto done; args[n++] = &a10;
360   if (&a11 == &no_more_args) goto done; args[n++] = &a11;
361   if (&a12 == &no_more_args) goto done; args[n++] = &a12;
362   if (&a13 == &no_more_args) goto done; args[n++] = &a13;
363   if (&a14 == &no_more_args) goto done; args[n++] = &a14;
364   if (&a15 == &no_more_args) goto done; args[n++] = &a15;
365 done:
366 
367   size_t consumed;
368   int vec[kVecSize] = {};
369   if (pattern.DoMatchImpl(*input, UNANCHORED, &consumed,
370                           args, n, vec, kVecSize)) {
371     input->remove_prefix(consumed);
372     return true;
373   } else {
374     return false;
375   }
376 }
377 
Replace(std::string * str,const PCRE & pattern,const StringPiece & rewrite)378 bool PCRE::Replace(std::string *str,
379                  const PCRE& pattern,
380                  const StringPiece& rewrite) {
381   int vec[kVecSize] = {};
382   int matches = pattern.TryMatch(*str, 0, UNANCHORED, true, vec, kVecSize);
383   if (matches == 0)
384     return false;
385 
386   std::string s;
387   if (!pattern.Rewrite(&s, rewrite, *str, vec, matches))
388     return false;
389 
390   assert(vec[0] >= 0);
391   assert(vec[1] >= 0);
392   str->replace(vec[0], vec[1] - vec[0], s);
393   return true;
394 }
395 
GlobalReplace(std::string * str,const PCRE & pattern,const StringPiece & rewrite)396 int PCRE::GlobalReplace(std::string *str,
397                       const PCRE& pattern,
398                       const StringPiece& rewrite) {
399   int count = 0;
400   int vec[kVecSize] = {};
401   std::string out;
402   size_t start = 0;
403   bool last_match_was_empty_string = false;
404 
405   while (start <= str->size()) {
406     // If the previous match was for the empty string, we shouldn't
407     // just match again: we'll match in the same way and get an
408     // infinite loop.  Instead, we do the match in a special way:
409     // anchored -- to force another try at the same position --
410     // and with a flag saying that this time, ignore empty matches.
411     // If this special match returns, that means there's a non-empty
412     // match at this position as well, and we can continue.  If not,
413     // we do what perl does, and just advance by one.
414     // Notice that perl prints '@@@' for this;
415     //    perl -le '$_ = "aa"; s/b*|aa/@/g; print'
416     int matches;
417     if (last_match_was_empty_string) {
418       matches = pattern.TryMatch(*str, start, ANCHOR_START, false,
419                                  vec, kVecSize);
420       if (matches <= 0) {
421         if (start < str->size())
422           out.push_back((*str)[start]);
423         start++;
424         last_match_was_empty_string = false;
425         continue;
426       }
427     } else {
428       matches = pattern.TryMatch(*str, start, UNANCHORED, true,
429                                  vec, kVecSize);
430       if (matches <= 0)
431         break;
432     }
433     size_t matchstart = vec[0], matchend = vec[1];
434     assert(matchstart >= start);
435     assert(matchend >= matchstart);
436 
437     out.append(*str, start, matchstart - start);
438     pattern.Rewrite(&out, rewrite, *str, vec, matches);
439     start = matchend;
440     count++;
441     last_match_was_empty_string = (matchstart == matchend);
442   }
443 
444   if (count == 0)
445     return 0;
446 
447   if (start < str->size())
448     out.append(*str, start, str->size() - start);
449   using std::swap;
450   swap(out, *str);
451   return count;
452 }
453 
Extract(const StringPiece & text,const PCRE & pattern,const StringPiece & rewrite,std::string * out)454 bool PCRE::Extract(const StringPiece &text,
455                  const PCRE& pattern,
456                  const StringPiece &rewrite,
457                  std::string *out) {
458   int vec[kVecSize] = {};
459   int matches = pattern.TryMatch(text, 0, UNANCHORED, true, vec, kVecSize);
460   if (matches == 0)
461     return false;
462   out->clear();
463   return pattern.Rewrite(out, rewrite, text, vec, matches);
464 }
465 
QuoteMeta(const StringPiece & unquoted)466 std::string PCRE::QuoteMeta(const StringPiece& unquoted) {
467   std::string result;
468   result.reserve(unquoted.size() << 1);
469 
470   // Escape any ascii character not in [A-Za-z_0-9].
471   //
472   // Note that it's legal to escape a character even if it has no
473   // special meaning in a regular expression -- so this function does
474   // that.  (This also makes it identical to the perl function of the
475   // same name except for the null-character special case;
476   // see `perldoc -f quotemeta`.)
477   for (size_t ii = 0; ii < unquoted.size(); ++ii) {
478     // Note that using 'isalnum' here raises the benchmark time from
479     // 32ns to 58ns:
480     if ((unquoted[ii] < 'a' || unquoted[ii] > 'z') &&
481         (unquoted[ii] < 'A' || unquoted[ii] > 'Z') &&
482         (unquoted[ii] < '0' || unquoted[ii] > '9') &&
483         unquoted[ii] != '_' &&
484         // If this is the part of a UTF8 or Latin1 character, we need
485         // to copy this byte without escaping.  Experimentally this is
486         // what works correctly with the regexp library.
487         !(unquoted[ii] & 128)) {
488       if (unquoted[ii] == '\0') {  // Special handling for null chars.
489         // Can't use "\\0" since the next character might be a digit.
490         result += "\\x00";
491         continue;
492       }
493       result += '\\';
494     }
495     result += unquoted[ii];
496   }
497 
498   return result;
499 }
500 
501 /***** Actual matching and rewriting code *****/
502 
HitLimit()503 bool PCRE::HitLimit() {
504   return hit_limit_ != 0;
505 }
506 
ClearHitLimit()507 void PCRE::ClearHitLimit() {
508   hit_limit_ = 0;
509 }
510 
TryMatch(const StringPiece & text,size_t startpos,Anchor anchor,bool empty_ok,int * vec,int vecsize) const511 int PCRE::TryMatch(const StringPiece& text,
512                    size_t startpos,
513                    Anchor anchor,
514                    bool empty_ok,
515                    int *vec,
516                    int vecsize) const {
517   pcre* re = (anchor == ANCHOR_BOTH) ? re_full_ : re_partial_;
518   if (re == NULL) {
519     PCREPORT(ERROR) << "Matching against invalid re: " << *error_;
520     return 0;
521   }
522 
523   int match_limit = match_limit_;
524   if (match_limit <= 0) {
525     match_limit = GetFlag(FLAGS_regexp_match_limit);
526   }
527 
528   int stack_limit = stack_limit_;
529   if (stack_limit <= 0) {
530     stack_limit = GetFlag(FLAGS_regexp_stack_limit);
531   }
532 
533   pcre_extra extra = { 0 };
534   if (match_limit > 0) {
535     extra.flags |= PCRE_EXTRA_MATCH_LIMIT;
536     extra.match_limit = match_limit;
537   }
538   if (stack_limit > 0) {
539     extra.flags |= PCRE_EXTRA_MATCH_LIMIT_RECURSION;
540     extra.match_limit_recursion = stack_limit / kPCREFrameSize;
541   }
542 
543   int options = 0;
544   if (anchor != UNANCHORED)
545     options |= PCRE_ANCHORED;
546   if (!empty_ok)
547     options |= PCRE_NOTEMPTY;
548 
549   int rc = pcre_exec(re,              // The regular expression object
550                      &extra,
551                      (text.data() == NULL) ? "" : text.data(),
552                      static_cast<int>(text.size()),
553                      static_cast<int>(startpos),
554                      options,
555                      vec,
556                      vecsize);
557 
558   // Handle errors
559   if (rc == 0) {
560     // pcre_exec() returns 0 as a special case when the number of
561     // capturing subpatterns exceeds the size of the vector.
562     // When this happens, there is a match and the output vector
563     // is filled, but we miss out on the positions of the extra subpatterns.
564     rc = vecsize / 2;
565   } else if (rc < 0) {
566     switch (rc) {
567       case PCRE_ERROR_NOMATCH:
568         return 0;
569       case PCRE_ERROR_MATCHLIMIT:
570         // Writing to hit_limit is not safe if multiple threads
571         // are using the PCRE, but the flag is only intended
572         // for use by unit tests anyway, so we let it go.
573         hit_limit_ = true;
574         PCREPORT(WARNING) << "Exceeded match limit of " << match_limit
575                         << " when matching '" << pattern_ << "'"
576                         << " against text that is " << text.size() << " bytes.";
577         return 0;
578       case PCRE_ERROR_RECURSIONLIMIT:
579         // See comment about hit_limit above.
580         hit_limit_ = true;
581         PCREPORT(WARNING) << "Exceeded stack limit of " << stack_limit
582                         << " when matching '" << pattern_ << "'"
583                         << " against text that is " << text.size() << " bytes.";
584         return 0;
585       default:
586         // There are other return codes from pcre.h :
587         // PCRE_ERROR_NULL           (-2)
588         // PCRE_ERROR_BADOPTION      (-3)
589         // PCRE_ERROR_BADMAGIC       (-4)
590         // PCRE_ERROR_UNKNOWN_NODE   (-5)
591         // PCRE_ERROR_NOMEMORY       (-6)
592         // PCRE_ERROR_NOSUBSTRING    (-7)
593         // ...
594         PCREPORT(ERROR) << "Unexpected return code: " << rc
595                       << " when matching '" << pattern_ << "'"
596                       << ", re=" << re
597                       << ", text=" << text
598                       << ", vec=" << vec
599                       << ", vecsize=" << vecsize;
600         return 0;
601     }
602   }
603 
604   return rc;
605 }
606 
DoMatchImpl(const StringPiece & text,Anchor anchor,size_t * consumed,const Arg * const * args,int n,int * vec,int vecsize) const607 bool PCRE::DoMatchImpl(const StringPiece& text,
608                        Anchor anchor,
609                        size_t* consumed,
610                        const Arg* const* args,
611                        int n,
612                        int* vec,
613                        int vecsize) const {
614   assert((1 + n) * 3 <= vecsize);  // results + PCRE workspace
615   if (NumberOfCapturingGroups() < n) {
616     // RE has fewer capturing groups than number of Arg pointers passed in.
617     return false;
618   }
619 
620   int matches = TryMatch(text, 0, anchor, true, vec, vecsize);
621   assert(matches >= 0);  // TryMatch never returns negatives
622   if (matches == 0)
623     return false;
624 
625   *consumed = vec[1];
626 
627   if (n == 0 || args == NULL) {
628     // We are not interested in results
629     return true;
630   }
631 
632   // If we got here, we must have matched the whole pattern.
633   // We do not need (can not do) any more checks on the value of 'matches' here
634   // -- see the comment for TryMatch.
635   for (int i = 0; i < n; i++) {
636     const int start = vec[2*(i+1)];
637     const int limit = vec[2*(i+1)+1];
638 
639     // Avoid invoking undefined behavior when text.data() happens
640     // to be null and start happens to be -1, the latter being the
641     // case for an unmatched subexpression. Even if text.data() is
642     // not null, pointing one byte before was a longstanding bug.
643     const char* addr = NULL;
644     if (start != -1) {
645       addr = text.data() + start;
646     }
647 
648     if (!args[i]->Parse(addr, limit-start)) {
649       // TODO: Should we indicate what the error was?
650       return false;
651     }
652   }
653 
654   return true;
655 }
656 
DoMatch(const StringPiece & text,Anchor anchor,size_t * consumed,const Arg * const args[],int n) const657 bool PCRE::DoMatch(const StringPiece& text,
658                    Anchor anchor,
659                    size_t* consumed,
660                    const Arg* const args[],
661                    int n) const {
662   assert(n >= 0);
663   const int vecsize = (1 + n) * 3;  // results + PCRE workspace
664                                     // (as for kVecSize)
665   int* vec = new int[vecsize];
666   bool b = DoMatchImpl(text, anchor, consumed, args, n, vec, vecsize);
667   delete[] vec;
668   return b;
669 }
670 
Rewrite(std::string * out,const StringPiece & rewrite,const StringPiece & text,int * vec,int veclen) const671 bool PCRE::Rewrite(std::string *out, const StringPiece &rewrite,
672                  const StringPiece &text, int *vec, int veclen) const {
673   int number_of_capturing_groups = NumberOfCapturingGroups();
674   for (const char *s = rewrite.data(), *end = s + rewrite.size();
675        s < end; s++) {
676     int c = *s;
677     if (c == '\\') {
678       c = *++s;
679       if (isdigit(c)) {
680         int n = (c - '0');
681         if (n >= veclen) {
682           if (n <= number_of_capturing_groups) {
683             // unmatched optional capturing group. treat
684             // its value as empty string; i.e., nothing to append.
685           } else {
686             PCREPORT(ERROR) << "requested group " << n
687                           << " in regexp " << rewrite.data();
688             return false;
689           }
690         }
691         int start = vec[2 * n];
692         if (start >= 0)
693           out->append(text.data() + start, vec[2 * n + 1] - start);
694       } else if (c == '\\') {
695         out->push_back('\\');
696       } else {
697         PCREPORT(ERROR) << "invalid rewrite pattern: " << rewrite.data();
698         return false;
699       }
700     } else {
701       out->push_back(c);
702     }
703   }
704   return true;
705 }
706 
CheckRewriteString(const StringPiece & rewrite,std::string * error) const707 bool PCRE::CheckRewriteString(const StringPiece& rewrite,
708                               std::string* error) const {
709   int max_token = -1;
710   for (const char *s = rewrite.data(), *end = s + rewrite.size();
711        s < end; s++) {
712     int c = *s;
713     if (c != '\\') {
714       continue;
715     }
716     if (++s == end) {
717       *error = "Rewrite schema error: '\\' not allowed at end.";
718       return false;
719     }
720     c = *s;
721     if (c == '\\') {
722       continue;
723     }
724     if (!isdigit(c)) {
725       *error = "Rewrite schema error: "
726                "'\\' must be followed by a digit or '\\'.";
727       return false;
728     }
729     int n = (c - '0');
730     if (max_token < n) {
731       max_token = n;
732     }
733   }
734 
735   if (max_token > NumberOfCapturingGroups()) {
736     *error = StringPrintf(
737         "Rewrite schema requests %d matches, but the regexp only has %d "
738         "parenthesized subexpressions.",
739         max_token, NumberOfCapturingGroups());
740     return false;
741   }
742   return true;
743 }
744 
745 
746 // Return the number of capturing subpatterns, or -1 if the
747 // regexp wasn't valid on construction.
NumberOfCapturingGroups() const748 int PCRE::NumberOfCapturingGroups() const {
749   if (re_partial_ == NULL) return -1;
750 
751   int result;
752   int rc = pcre_fullinfo(re_partial_,       // The regular expression object
753                          NULL,              // We did not study the pattern
754                          PCRE_INFO_CAPTURECOUNT,
755                          &result);
756   if (rc != 0) {
757     PCREPORT(ERROR) << "Unexpected return code: " << rc;
758     return -1;
759   }
760   return result;
761 }
762 
763 
764 /***** Parsers for various types *****/
765 
parse_null(const char * str,size_t n,void * dest)766 bool PCRE::Arg::parse_null(const char* str, size_t n, void* dest) {
767   // We fail if somebody asked us to store into a non-NULL void* pointer
768   return (dest == NULL);
769 }
770 
parse_string(const char * str,size_t n,void * dest)771 bool PCRE::Arg::parse_string(const char* str, size_t n, void* dest) {
772   if (dest == NULL) return true;
773   reinterpret_cast<std::string*>(dest)->assign(str, n);
774   return true;
775 }
776 
parse_stringpiece(const char * str,size_t n,void * dest)777 bool PCRE::Arg::parse_stringpiece(const char* str, size_t n, void* dest) {
778   if (dest == NULL) return true;
779   *(reinterpret_cast<StringPiece*>(dest)) = StringPiece(str, n);
780   return true;
781 }
782 
parse_char(const char * str,size_t n,void * dest)783 bool PCRE::Arg::parse_char(const char* str, size_t n, void* dest) {
784   if (n != 1) return false;
785   if (dest == NULL) return true;
786   *(reinterpret_cast<char*>(dest)) = str[0];
787   return true;
788 }
789 
parse_schar(const char * str,size_t n,void * dest)790 bool PCRE::Arg::parse_schar(const char* str, size_t n, void* dest) {
791   if (n != 1) return false;
792   if (dest == NULL) return true;
793   *(reinterpret_cast<signed char*>(dest)) = str[0];
794   return true;
795 }
796 
parse_uchar(const char * str,size_t n,void * dest)797 bool PCRE::Arg::parse_uchar(const char* str, size_t n, void* dest) {
798   if (n != 1) return false;
799   if (dest == NULL) return true;
800   *(reinterpret_cast<unsigned char*>(dest)) = str[0];
801   return true;
802 }
803 
804 // Largest number spec that we are willing to parse
805 static const int kMaxNumberLength = 32;
806 
807 // PCREQUIPCRES "buf" must have length at least kMaxNumberLength+1
808 // PCREQUIPCRES "n > 0"
809 // Copies "str" into "buf" and null-terminates if necessary.
810 // Returns one of:
811 //      a. "str" if no termination is needed
812 //      b. "buf" if the string was copied and null-terminated
813 //      c. "" if the input was invalid and has no hope of being parsed
TerminateNumber(char * buf,const char * str,size_t n)814 static const char* TerminateNumber(char* buf, const char* str, size_t n) {
815   if ((n > 0) && isspace(*str)) {
816     // We are less forgiving than the strtoxxx() routines and do not
817     // allow leading spaces.
818     return "";
819   }
820 
821   // See if the character right after the input text may potentially
822   // look like a digit.
823   if (isdigit(str[n]) ||
824       ((str[n] >= 'a') && (str[n] <= 'f')) ||
825       ((str[n] >= 'A') && (str[n] <= 'F'))) {
826     if (n > kMaxNumberLength) return ""; // Input too big to be a valid number
827     memcpy(buf, str, n);
828     buf[n] = '\0';
829     return buf;
830   } else {
831     // We can parse right out of the supplied string, so return it.
832     return str;
833   }
834 }
835 
parse_long_radix(const char * str,size_t n,void * dest,int radix)836 bool PCRE::Arg::parse_long_radix(const char* str,
837                                  size_t n,
838                                  void* dest,
839                                  int radix) {
840   if (n == 0) return false;
841   char buf[kMaxNumberLength+1];
842   str = TerminateNumber(buf, str, n);
843   char* end;
844   errno = 0;
845   long r = strtol(str, &end, radix);
846   if (end != str + n) return false;   // Leftover junk
847   if (errno) return false;
848   if (dest == NULL) return true;
849   *(reinterpret_cast<long*>(dest)) = r;
850   return true;
851 }
852 
parse_ulong_radix(const char * str,size_t n,void * dest,int radix)853 bool PCRE::Arg::parse_ulong_radix(const char* str,
854                                   size_t n,
855                                   void* dest,
856                                   int radix) {
857   if (n == 0) return false;
858   char buf[kMaxNumberLength+1];
859   str = TerminateNumber(buf, str, n);
860   if (str[0] == '-') {
861     // strtoul() will silently accept negative numbers and parse
862     // them.  This module is more strict and treats them as errors.
863     return false;
864   }
865 
866   char* end;
867   errno = 0;
868   unsigned long r = strtoul(str, &end, radix);
869   if (end != str + n) return false;   // Leftover junk
870   if (errno) return false;
871   if (dest == NULL) return true;
872   *(reinterpret_cast<unsigned long*>(dest)) = r;
873   return true;
874 }
875 
parse_short_radix(const char * str,size_t n,void * dest,int radix)876 bool PCRE::Arg::parse_short_radix(const char* str,
877                                   size_t n,
878                                   void* dest,
879                                   int radix) {
880   long r;
881   if (!parse_long_radix(str, n, &r, radix)) return false;  // Could not parse
882   if ((short)r != r) return false;                         // Out of range
883   if (dest == NULL) return true;
884   *(reinterpret_cast<short*>(dest)) = (short)r;
885   return true;
886 }
887 
parse_ushort_radix(const char * str,size_t n,void * dest,int radix)888 bool PCRE::Arg::parse_ushort_radix(const char* str,
889                                    size_t n,
890                                    void* dest,
891                                    int radix) {
892   unsigned long r;
893   if (!parse_ulong_radix(str, n, &r, radix)) return false;  // Could not parse
894   if ((unsigned short)r != r) return false;                 // Out of range
895   if (dest == NULL) return true;
896   *(reinterpret_cast<unsigned short*>(dest)) = (unsigned short)r;
897   return true;
898 }
899 
parse_int_radix(const char * str,size_t n,void * dest,int radix)900 bool PCRE::Arg::parse_int_radix(const char* str,
901                                 size_t n,
902                                 void* dest,
903                                 int radix) {
904   long r;
905   if (!parse_long_radix(str, n, &r, radix)) return false;  // Could not parse
906   if ((int)r != r) return false;                           // Out of range
907   if (dest == NULL) return true;
908   *(reinterpret_cast<int*>(dest)) = (int)r;
909   return true;
910 }
911 
parse_uint_radix(const char * str,size_t n,void * dest,int radix)912 bool PCRE::Arg::parse_uint_radix(const char* str,
913                                  size_t n,
914                                  void* dest,
915                                  int radix) {
916   unsigned long r;
917   if (!parse_ulong_radix(str, n, &r, radix)) return false;  // Could not parse
918   if ((unsigned int)r != r) return false;                   // Out of range
919   if (dest == NULL) return true;
920   *(reinterpret_cast<unsigned int*>(dest)) = (unsigned int)r;
921   return true;
922 }
923 
parse_longlong_radix(const char * str,size_t n,void * dest,int radix)924 bool PCRE::Arg::parse_longlong_radix(const char* str,
925                                      size_t n,
926                                      void* dest,
927                                      int radix) {
928   if (n == 0) return false;
929   char buf[kMaxNumberLength+1];
930   str = TerminateNumber(buf, str, n);
931   char* end;
932   errno = 0;
933   long long r = strtoll(str, &end, radix);
934   if (end != str + n) return false;   // Leftover junk
935   if (errno) return false;
936   if (dest == NULL) return true;
937   *(reinterpret_cast<long long*>(dest)) = r;
938   return true;
939 }
940 
parse_ulonglong_radix(const char * str,size_t n,void * dest,int radix)941 bool PCRE::Arg::parse_ulonglong_radix(const char* str,
942                                       size_t n,
943                                       void* dest,
944                                       int radix) {
945   if (n == 0) return false;
946   char buf[kMaxNumberLength+1];
947   str = TerminateNumber(buf, str, n);
948   if (str[0] == '-') {
949     // strtoull() will silently accept negative numbers and parse
950     // them.  This module is more strict and treats them as errors.
951     return false;
952   }
953   char* end;
954   errno = 0;
955   unsigned long long r = strtoull(str, &end, radix);
956   if (end != str + n) return false;   // Leftover junk
957   if (errno) return false;
958   if (dest == NULL) return true;
959   *(reinterpret_cast<unsigned long long*>(dest)) = r;
960   return true;
961 }
962 
parse_double_float(const char * str,size_t n,bool isfloat,void * dest)963 static bool parse_double_float(const char* str, size_t n, bool isfloat,
964                                void* dest) {
965   if (n == 0) return false;
966   static const int kMaxLength = 200;
967   char buf[kMaxLength];
968   if (n >= kMaxLength) return false;
969   memcpy(buf, str, n);
970   buf[n] = '\0';
971   char* end;
972   errno = 0;
973   double r;
974   if (isfloat) {
975     r = strtof(buf, &end);
976   } else {
977     r = strtod(buf, &end);
978   }
979   if (end != buf + n) return false;   // Leftover junk
980   if (errno) return false;
981   if (dest == NULL) return true;
982   if (isfloat) {
983     *(reinterpret_cast<float*>(dest)) = (float)r;
984   } else {
985     *(reinterpret_cast<double*>(dest)) = r;
986   }
987   return true;
988 }
989 
parse_double(const char * str,size_t n,void * dest)990 bool PCRE::Arg::parse_double(const char* str, size_t n, void* dest) {
991   return parse_double_float(str, n, false, dest);
992 }
993 
parse_float(const char * str,size_t n,void * dest)994 bool PCRE::Arg::parse_float(const char* str, size_t n, void* dest) {
995   return parse_double_float(str, n, true, dest);
996 }
997 
998 #define DEFINE_INTEGER_PARSER(name)                                           \
999   bool PCRE::Arg::parse_##name(const char* str, size_t n, void* dest) {       \
1000     return parse_##name##_radix(str, n, dest, 10);                            \
1001   }                                                                           \
1002   bool PCRE::Arg::parse_##name##_hex(const char* str, size_t n, void* dest) { \
1003     return parse_##name##_radix(str, n, dest, 16);                            \
1004   }                                                                           \
1005   bool PCRE::Arg::parse_##name##_octal(const char* str, size_t n,             \
1006                                        void* dest) {                          \
1007     return parse_##name##_radix(str, n, dest, 8);                             \
1008   }                                                                           \
1009   bool PCRE::Arg::parse_##name##_cradix(const char* str, size_t n,            \
1010                                         void* dest) {                         \
1011     return parse_##name##_radix(str, n, dest, 0);                             \
1012   }
1013 
1014 DEFINE_INTEGER_PARSER(short);
1015 DEFINE_INTEGER_PARSER(ushort);
1016 DEFINE_INTEGER_PARSER(int);
1017 DEFINE_INTEGER_PARSER(uint);
1018 DEFINE_INTEGER_PARSER(long);
1019 DEFINE_INTEGER_PARSER(ulong);
1020 DEFINE_INTEGER_PARSER(longlong);
1021 DEFINE_INTEGER_PARSER(ulonglong);
1022 
1023 #undef DEFINE_INTEGER_PARSER
1024 
1025 }  // namespace re2
1026