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