| /* |
| * Copyright (C) 2009 Apple Inc. All rights reserved. |
| * |
| * Redistribution and use in source and binary forms, with or without |
| * modification, are permitted provided that the following conditions |
| * are met: |
| * 1. Redistributions of source code must retain the above copyright |
| * notice, this list of conditions and the following disclaimer. |
| * 2. Redistributions in binary form must reproduce the above copyright |
| * notice, this list of conditions and the following disclaimer in the |
| * documentation and/or other materials provided with the distribution. |
| * |
| * THIS SOFTWARE IS PROVIDED BY APPLE INC. ``AS IS'' AND ANY |
| * EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE |
| * IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR |
| * PURPOSE ARE DISCLAIMED. IN NO EVENT SHALL APPLE INC. OR |
| * CONTRIBUTORS BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, |
| * EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT LIMITED TO, |
| * PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE, DATA, OR |
| * PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY THEORY |
| * OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT |
| * (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE |
| * OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE. |
| */ |
| |
| #ifndef YarrJIT_h |
| #define YarrJIT_h |
| |
| #if ENABLE(YARR_JIT) |
| |
| #include "JSGlobalData.h" |
| #include "MacroAssemblerCodeRef.h" |
| #include "MatchResult.h" |
| #include "Yarr.h" |
| #include "YarrPattern.h" |
| |
| #if CPU(X86) && !COMPILER(MSVC) |
| #define YARR_CALL __attribute__ ((regparm (3))) |
| #else |
| #define YARR_CALL |
| #endif |
| |
| namespace JSC { |
| |
| class JSGlobalData; |
| class ExecutablePool; |
| |
| namespace Yarr { |
| |
| class YarrCodeBlock { |
| #if CPU(X86_64) |
| typedef MatchResult (*YarrJITCode8)(const LChar* input, unsigned start, unsigned length, int* output) YARR_CALL; |
| typedef MatchResult (*YarrJITCode16)(const UChar* input, unsigned start, unsigned length, int* output) YARR_CALL; |
| typedef MatchResult (*YarrJITCodeMatchOnly8)(const LChar* input, unsigned start, unsigned length) YARR_CALL; |
| typedef MatchResult (*YarrJITCodeMatchOnly16)(const UChar* input, unsigned start, unsigned length) YARR_CALL; |
| #else |
| typedef EncodedMatchResult (*YarrJITCode8)(const LChar* input, unsigned start, unsigned length, int* output) YARR_CALL; |
| typedef EncodedMatchResult (*YarrJITCode16)(const UChar* input, unsigned start, unsigned length, int* output) YARR_CALL; |
| typedef EncodedMatchResult (*YarrJITCodeMatchOnly8)(const LChar* input, unsigned start, unsigned length) YARR_CALL; |
| typedef EncodedMatchResult (*YarrJITCodeMatchOnly16)(const UChar* input, unsigned start, unsigned length) YARR_CALL; |
| #endif |
| |
| public: |
| YarrCodeBlock() |
| : m_needFallBack(false) |
| { |
| } |
| |
| ~YarrCodeBlock() |
| { |
| } |
| |
| void setFallBack(bool fallback) { m_needFallBack = fallback; } |
| bool isFallBack() { return m_needFallBack; } |
| |
| bool has8BitCode() { return m_ref8.size(); } |
| bool has16BitCode() { return m_ref16.size(); } |
| void set8BitCode(MacroAssemblerCodeRef ref) { m_ref8 = ref; } |
| void set16BitCode(MacroAssemblerCodeRef ref) { m_ref16 = ref; } |
| |
| bool has8BitCodeMatchOnly() { return m_matchOnly8.size(); } |
| bool has16BitCodeMatchOnly() { return m_matchOnly16.size(); } |
| void set8BitCodeMatchOnly(MacroAssemblerCodeRef matchOnly) { m_matchOnly8 = matchOnly; } |
| void set16BitCodeMatchOnly(MacroAssemblerCodeRef matchOnly) { m_matchOnly16 = matchOnly; } |
| |
| MatchResult execute(const LChar* input, unsigned start, unsigned length, int* output) |
| { |
| ASSERT(has8BitCode()); |
| return MatchResult(reinterpret_cast<YarrJITCode8>(m_ref8.code().executableAddress())(input, start, length, output)); |
| } |
| |
| MatchResult execute(const UChar* input, unsigned start, unsigned length, int* output) |
| { |
| ASSERT(has16BitCode()); |
| return MatchResult(reinterpret_cast<YarrJITCode16>(m_ref16.code().executableAddress())(input, start, length, output)); |
| } |
| |
| MatchResult execute(const LChar* input, unsigned start, unsigned length) |
| { |
| ASSERT(has8BitCodeMatchOnly()); |
| return MatchResult(reinterpret_cast<YarrJITCodeMatchOnly8>(m_matchOnly8.code().executableAddress())(input, start, length)); |
| } |
| |
| MatchResult execute(const UChar* input, unsigned start, unsigned length) |
| { |
| ASSERT(has16BitCodeMatchOnly()); |
| return MatchResult(reinterpret_cast<YarrJITCodeMatchOnly16>(m_matchOnly16.code().executableAddress())(input, start, length)); |
| } |
| |
| #if ENABLE(REGEXP_TRACING) |
| void *getAddr() { return m_ref.code().executableAddress(); } |
| #endif |
| |
| void clear() |
| { |
| m_ref8 = MacroAssemblerCodeRef(); |
| m_ref16 = MacroAssemblerCodeRef(); |
| m_matchOnly8 = MacroAssemblerCodeRef(); |
| m_matchOnly16 = MacroAssemblerCodeRef(); |
| m_needFallBack = false; |
| } |
| |
| private: |
| MacroAssemblerCodeRef m_ref8; |
| MacroAssemblerCodeRef m_ref16; |
| MacroAssemblerCodeRef m_matchOnly8; |
| MacroAssemblerCodeRef m_matchOnly16; |
| bool m_needFallBack; |
| }; |
| |
| enum YarrJITCompileMode { |
| MatchOnly, |
| IncludeSubpatterns |
| }; |
| void jitCompile(YarrPattern&, YarrCharSize, JSGlobalData*, YarrCodeBlock& jitObject, YarrJITCompileMode = IncludeSubpatterns); |
| |
| } } // namespace JSC::Yarr |
| |
| #endif |
| |
| #endif // YarrJIT_h |