Home | History | Annotate | Download | only in yarr 
      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
     99