yarr/RegexJIT.h

   1 /*
   2  * Copyright (C) 2009 Apple Inc. All rights reserved.
   3  *
   4  * Redistribution and use in source and binary forms, with or without
   5  * modification, are permitted provided that the following conditions
   6  * are met:
   7  * 1. Redistributions of source code must retain the above copyright
   8  *    notice, this list of conditions and the following disclaimer.
   9  * 2. Redistributions in binary form must reproduce the above copyright
  10  *    notice, this list of conditions and the following disclaimer in the
  11  *    documentation and/or other materials provided with the distribution.
  12  *
  13  * THIS SOFTWARE IS PROVIDED BY APPLE INC. ``AS IS'' AND ANY
  14  * EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE
  15  * IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR
  16  * PURPOSE ARE DISCLAIMED.  IN NO EVENT SHALL APPLE INC. OR
  17  * CONTRIBUTORS BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL,
  18  * EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT LIMITED TO,
  19  * PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE, DATA, OR
  20  * PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY THEORY
  21  * OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT
  22  * (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE
  23  * OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.
  24  */
  25
  26 #ifndef RegexJIT_h
  27 #define RegexJIT_h
  28
  29 #include <wtf/Platform.h>
  30
  31 #if ENABLE(YARR_JIT)
  32
  33 #include "MacroAssembler.h"
  34 #include "RegexPattern.h"
  35 #include <UString.h>
  36
  37 #include <pcre.h>
  38 struct JSRegExp; // temporary, remove when fallback is removed.
  39
  40 #if CPU(X86) && !COMPILER(MSVC)
  41 #define YARR_CALL __attribute__ ((regparm (3)))
  42 #else
  43 #define YARR_CALL
  44 #endif
  45
  46 namespace JSC {
  47
  48 class JSGlobalData;
  49 class ExecutablePool;
  50
  51 namespace Yarr {
  52
  53 class RegexCodeBlock {
  54     typedef int (*RegexJITCode)(const UChar* input, unsigned start, unsigned length, int* output) YARR_CALL;
  55
  56 public:
  57     RegexCodeBlock()
  58         : m_fallback(0)
  59     {
  60     }
  61
  62     ~RegexCodeBlock()
  63     {
  64         if (m_fallback)
  65             jsRegExpFree(m_fallback);
  66     }
  67
  68     JSRegExp* getFallback() { return m_fallback; }
  69     void setFallback(JSRegExp* fallback) { m_fallback = fallback; }
  70
  71     bool operator!() { return !m_ref.m_code.executableAddress(); }
  72     void set(MacroAssembler::CodeRef ref) { m_ref = ref; }
  73
  74     int execute(const UChar* input, unsigned start, unsigned length, int* output)
  75     {
  76         return reinterpret_cast<RegexJITCode>(m_ref.m_code.executableAddress())(input, start, length, output);
  77     }
  78
  79 private:
  80     MacroAssembler::CodeRef m_ref;
  81     JSRegExp* m_fallback;
  82 };
  83
  84 void jitCompileRegex(JSGlobalData* globalData, RegexCodeBlock& jitObject, const UString& pattern, unsigned& numSubpatterns, const char*& error, bool ignoreCase = false, bool multiline = false);
  85
  86 inline int executeRegex(RegexCodeBlock& jitObject, const UChar* input, unsigned start, unsigned length, int* output, int outputArraySize)
  87 {
  88     if (JSRegExp* fallback = jitObject.getFallback())
  89         return (jsRegExpExecute(fallback, input, length, start, output, outputArraySize) < 0) ? -1 : output[0];
  90
  91     return jitObject.execute(input, start, length, output);
  92 }
  93
  94 } } // namespace JSC::Yarr
  95
  96 #endif
  97
  98 #endif // RegexJIT_h