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/UCPlatform.h"
  46 #include "CyteKit/stringWithUTF8Bytes.h"
  47
  48 class Pcre {
  49   private:
  50     pcre *code_;
  51     pcre_extra *study_;
  52     int capture_;
  53     int *matches_;
  54     const char *data_;
  55
  56   public:
  57     Pcre() :
  58         code_(NULL),
  59         study_(NULL),
  60         data_(NULL)
  61     {
  62     }
  63
  64     Pcre(const char *regex, NSString *data = nil) :
  65         code_(NULL),
  66         study_(NULL),
  67         data_(NULL)
  68     {
  69         this->operator =(regex);
  70
  71         if (data != nil)
  72             this->operator ()(data);
  73     }
  74
  75     void operator =(const char *regex) {
  76         _assert(code_ == NULL);
  77
  78         const char *error;
  79         int offset;
  80         code_ = pcre_compile(regex, 0, &error, &offset, NULL);
  81
  82         if (code_ == NULL) {
  83             fprintf(stderr, "%d:%s\n", offset, error);
  84             _assert(false);
  85         }
  86
  87         pcre_fullinfo(code_, study_, PCRE_INFO_CAPTURECOUNT, &capture_);
  88         _assert(capture_ >= 0);
  89         matches_ = new int[(capture_ + 1) * 3];
  90     }
  91
  92     ~Pcre() {
  93         pcre_free(code_);
  94         delete matches_;
  95     }
  96
  97     NSString *operator [](size_t match) const {
  98         return [NSString stringWithUTF8Bytes:(data_ + matches_[match * 2]) length:(matches_[match * 2 + 1] - matches_[match * 2])];
  99     }
 100
 101     _finline bool operator ()(NSString *data) {
 102         // XXX: length is for characters, not for bytes
 103         return operator ()([data UTF8String], [data length]);
 104     }
 105
 106     _finline bool operator ()(const char *data) {
 107         return operator ()(data, strlen(data));
 108     }
 109
 110     bool operator ()(const char *data, size_t size) {
 111         if (pcre_exec(code_, study_, data, size, 0, 0, matches_, (capture_ + 1) * 3) >= 0) {
 112             data_ = data;
 113             return true;
 114         } else {
 115             data_ = NULL;
 116             return false;
 117         }
 118     }
 119
 120     operator bool() const {
 121         return data_ != NULL;
 122     }
 123
 124     NSString *operator ->*(NSString *format) const {
 125         id values[capture_];
 126         for (int i(0); i != capture_; ++i)
 127             values[i] = this->operator [](i + 1);
 128
 129         return [[[NSString alloc] initWithFormat:format arguments:reinterpret_cast<va_list>(values)] autorelease];
 130     }
 131 };
 132
 133 #endif//Cydia_PerlCompatibleRegEx_HPP