src/libshared/src/RX.cpp

   1 ////////////////////////////////////////////////////////////////////////////////
   2 //
   3 // Copyright 2006 - 2017, Paul Beckingham, Federico Hernandez.
   4 //
   5 // Permission is hereby granted, free of charge, to any person obtaining a copy
   6 // of this software and associated documentation files (the "Software"), to deal
   7 // in the Software without restriction, including without limitation the rights
   8 // to use, copy, modify, merge, publish, distribute, sublicense, and/or sell
   9 // copies of the Software, and to permit persons to whom the Software is
  10 // furnished to do so, subject to the following conditions:
  11 //
  12 // The above copyright notice and this permission notice shall be included
  13 // in all copies or substantial portions of the Software.
  14 //
  15 // THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS
  16 // OR IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
  17 // FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL
  18 // THE AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
  19 // LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,
  20 // OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE
  21 // SOFTWARE.
  22 //
  23 // http://www.opensource.org/licenses/mit-license.php
  24 //
  25 ////////////////////////////////////////////////////////////////////////////////
  26
  27 #include <cmake.h>
  28 #include <RX.h>
  29 #include <cstdlib>
  30 #include <cstring>
  31
  32 ////////////////////////////////////////////////////////////////////////////////
  33 RX::RX ()
  34 {
  35 }
  36
  37 ////////////////////////////////////////////////////////////////////////////////
  38 RX::RX (
  39   const std::string& pattern,
  40   bool case_sensitive /* = true */)
  41 : _compiled (false)
  42 , _pattern (pattern)
  43 , _case_sensitive (case_sensitive)
  44 {
  45   compile ();
  46 }
  47
  48 ////////////////////////////////////////////////////////////////////////////////
  49 RX::RX (const RX& other)
  50 {
  51   _compiled       = false;
  52   _pattern        = other._pattern;
  53   _case_sensitive = other._case_sensitive;
  54 }
  55
  56 ////////////////////////////////////////////////////////////////////////////////
  57 RX::~RX ()
  58 {
  59   if (_compiled)
  60     regfree (&_regex);
  61 }
  62
  63 ////////////////////////////////////////////////////////////////////////////////
  64 RX& RX::operator= (const RX& other)
  65 {
  66   _compiled       = false;
  67   _pattern        = other._pattern;
  68   _case_sensitive = other._case_sensitive;
  69
  70   return *this;
  71 }
  72
  73 ////////////////////////////////////////////////////////////////////////////////
  74 void RX::compile ()
  75 {
  76   if (! _compiled)
  77   {
  78     memset (&_regex, 0, sizeof (regex_t));
  79
  80     int result;
  81     if ((result = regcomp (&_regex, _pattern.c_str (),
  82 #if defined REG_ENHANCED
  83                            REG_ENHANCED | REG_EXTENDED | REG_NEWLINE |
  84 #else
  85                            REG_EXTENDED | REG_NEWLINE |
  86 #endif
  87                            (_case_sensitive ? 0 : REG_ICASE))) != 0)
  88     {
  89       char message[256];
  90       regerror (result, &_regex, message, 256);
  91       throw std::string (message);
  92     }
  93
  94     _compiled = true;
  95   }
  96 }
  97
  98 ////////////////////////////////////////////////////////////////////////////////
  99 bool RX::match (const std::string& in)
 100 {
 101   if (! _compiled)
 102     compile ();
 103
 104   return regexec (&_regex, in.c_str (), 0, nullptr, 0) == 0 ? true : false;
 105 }
 106
 107 ////////////////////////////////////////////////////////////////////////////////
 108 bool RX::match (
 109   std::vector<std::string>& matches,
 110   const std::string& in)
 111 {
 112   if (! _compiled)
 113     compile ();
 114
 115   regmatch_t rm[2];
 116   int offset = 0;
 117   int length = in.length ();
 118   while (regexec (&_regex, in.c_str () + offset, 2, &rm[0], 0) == 0 &&
 119          offset < length)
 120   {
 121     matches.push_back (in.substr (rm[0].rm_so + offset, rm[0].rm_eo - rm[0].rm_so));
 122     offset += rm[0].rm_eo;
 123
 124     // Protection against zero-width patterns causing infinite loops.
 125     if (rm[0].rm_so == rm[0].rm_eo)
 126       ++offset;
 127   }
 128
 129   return matches.size () ? true : false;
 130 }
 131
 132 ////////////////////////////////////////////////////////////////////////////////
 133 bool RX::match (
 134   std::vector <int>& start,
 135   std::vector <int>& end,
 136   const std::string& in)
 137 {
 138   if (! _compiled)
 139     compile ();
 140
 141   regmatch_t rm[2];
 142   int offset = 0;
 143   int length = in.length ();
 144   while (regexec (&_regex, in.c_str () + offset, 2, &rm[0], 0) == 0 &&
 145          offset < length)
 146   {
 147     start.push_back (rm[0].rm_so + offset);
 148     end.push_back   (rm[0].rm_eo + offset);
 149     offset += rm[0].rm_eo;
 150
 151     // Protection against zero-width patterns causing infinite loops.
 152     if (rm[0].rm_so == rm[0].rm_eo)
 153       ++offset;
 154   }
 155
 156   return start.size () ? true : false;
 157 }
 158
 159 ////////////////////////////////////////////////////////////////////////////////