CyteKit/PerlCompatibleRegEx.hpp

   1 /* Cydia - iPhone UIKit Front-End for Debian APT
   2  * Copyright (C) 2008-2011  Jay Freeman (saurik)
   3 */
   4
   5 /* Modified BSD License {{{ */
   6 /*
   7  *        Redistribution and use in source and binary
   8  * forms, with or without modification, are permitted
   9  * provided that the following conditions are met:
  10  *
  11  * 1. Redistributions of source code must retain the
  12  *    above copyright notice, this list of conditions
  13  *    and the following disclaimer.
  14  * 2. Redistributions in binary form must reproduce the
  15  *    above copyright notice, this list of conditions
  16  *    and the following disclaimer in the documentation
  17  *    and/or other materials provided with the
  18  *    distribution.
  19  * 3. The name of the author may not be used to endorse
  20  *    or promote products derived from this software
  21  *    without specific prior written permission.
  22  *
  23  * THIS SOFTWARE IS PROVIDED BY THE AUTHOR ``AS IS''
  24  * AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING,
  25  * BUT NOT LIMITED TO, THE IMPLIED WARRANTIES OF
  26  * MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE
  27  * ARE DISCLAIMED. IN NO EVENT SHALL THE AUTHOR BE
  28  * LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL,
  29  * EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT
  30  * NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR
  31  * SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS
  32  * INTERRUPTION) HOWEVER CAUSED AND ON ANY THEORY OF
  33  * LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR
  34  * TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN
  35  * ANY WAY OUT OF THE USE OF THIS SOFTWARE, EVEN IF
  36  * ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.
  37 */
  38 /* }}} */
  39
  40 #ifndef Cydia_PerlCompatibleRegEx_HPP
  41 #define Cydia_PerlCompatibleRegEx_HPP
  42
  43 #include <pcre.h>
  44
  45 #include "CyteKit/stringWithUTF8Bytes.h"
  46
  47 class Pcre {
  48   private:
  49     pcre *code_;
  50     pcre_extra *study_;
  51     int capture_;
  52     int *matches_;
  53     const char *data_;
  54
  55   public:
  56     Pcre() :
  57         code_(NULL),
  58         study_(NULL)
  59     {
  60     }
  61
  62     Pcre(const char *regex) :
  63         code_(NULL),
  64         study_(NULL)
  65     {
  66         this->operator =(regex);
  67     }
  68
  69     void operator =(const char *regex) {
  70         _assert(code_ == NULL);
  71
  72         const char *error;
  73         int offset;
  74         code_ = pcre_compile(regex, 0, &error, &offset, NULL);
  75
  76         if (code_ == NULL) {
  77             fprintf(stderr, "%d:%s\n", offset, error);
  78             _assert(false);
  79         }
  80
  81         pcre_fullinfo(code_, study_, PCRE_INFO_CAPTURECOUNT, &capture_);
  82         matches_ = new int[(capture_ + 1) * 3];
  83     }
  84
  85     ~Pcre() {
  86         pcre_free(code_);
  87         delete matches_;
  88     }
  89
  90     NSString *operator [](size_t match) const {
  91         return [NSString stringWithUTF8Bytes:(data_ + matches_[match * 2]) length:(matches_[match * 2 + 1] - matches_[match * 2])];
  92     }
  93
  94     _finline bool operator ()(NSString *data) {
  95         // XXX: length is for characters, not for bytes
  96         return operator ()([data UTF8String], [data length]);
  97     }
  98
  99     _finline bool operator ()(const char *data) {
 100         return operator ()(data, strlen(data));
 101     }
 102
 103     bool operator ()(const char *data, size_t size) {
 104         data_ = data;
 105         return pcre_exec(code_, study_, data, size, 0, 0, matches_, (capture_ + 1) * 3) >= 0;
 106     }
 107
 108     NSString *operator ->*(NSString *format) const {
 109         id values[capture_];
 110         for (int i(0); i != capture_; ++i)
 111             values[i] = this->operator [](i + 1);
 112
 113         return [[[NSString alloc] initWithFormat:format arguments:reinterpret_cast<va_list>(values)] autorelease];
 114     }
 115 };
 116
 117 #endif//Cydia_PerlCompatibleRegEx_HPP