| /* -*- Mode: C++; tab-width: 8; indent-tabs-mode: nil; c-basic-offset: 4 -*- |
| * vim: set ts=8 sts=4 et sw=4 tw=99: |
| * |
| * 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 yarr_YarrJIT_h |
| #define yarr_YarrJIT_h |
| |
| #include "assembler/wtf/Platform.h" |
| |
| #if ENABLE_YARR_JIT |
| |
| #include "assembler/assembler/MacroAssembler.h" |
| |
| #include "MatchResult.h" |
| #include "Yarr.h" |
| #include "YarrPattern.h" |
| |
| #if WTF_CPU_X86 && !WTF_COMPILER_MSVC && !WTF_COMPILER_SUNCC |
| #define YARR_CALL __attribute__ ((regparm (3))) |
| #else |
| #define YARR_CALL |
| #endif |
| |
| #if JS_TRACE_LOGGING |
| #include "TraceLogging.h" |
| #endif |
| |
| namespace JSC { |
| |
| class JSGlobalData; |
| class ExecutablePool; |
| |
| namespace Yarr { |
| |
| class YarrCodeBlock { |
| #if defined(WTF_CPU_X86_64) && !defined(_WIN64) |
| typedef MatchResult JITMatchResult; |
| #else |
| typedef EncodedMatchResult JITMatchResult; |
| #endif |
| |
| typedef JITMatchResult (*YarrJITCode8)(const LChar* input, unsigned start, unsigned length, int* output) YARR_CALL; |
| typedef JITMatchResult (*YarrJITCode16)(const UChar* input, unsigned start, unsigned length, int* output) YARR_CALL; |
| typedef JITMatchResult (*YarrJITCodeMatchOnly8)(const LChar* input, unsigned start, unsigned length) YARR_CALL; |
| typedef JITMatchResult (*YarrJITCodeMatchOnly16)(const UChar* input, unsigned start, unsigned length) YARR_CALL; |
| |
| public: |
| YarrCodeBlock() |
| : m_needFallBack(false) |
| { |
| } |
| |
| ~YarrCodeBlock() |
| { |
| } |
| |
| void setFallBack(bool fallback) { m_needFallBack = fallback; } |
| bool isFallBack() { return m_needFallBack; } |
| |
| #ifdef YARR_8BIT_CHAR_SUPPORT |
| bool has8BitCode() const { return m_ref8.size(); } |
| void set8BitCode(MacroAssemblerCodeRef ref) { m_ref8 = ref; } |
| bool has8BitCodeMatchOnly() const { return m_matchOnly8.size(); } |
| void set8BitCodeMatchOnly(MacroAssemblerCodeRef matchOnly) { m_matchOnly8 = matchOnly; } |
| #endif |
| |
| bool has16BitCode() const { return m_ref16.size(); } |
| void set16BitCode(MacroAssemblerCodeRef ref) { m_ref16 = ref; } |
| |
| bool has16BitCodeMatchOnly() const { return m_matchOnly16.size(); } |
| void set16BitCodeMatchOnly(MacroAssemblerCodeRef matchOnly) { m_matchOnly16 = matchOnly; } |
| |
| #if YARR_8BIT_CHAR_SUPPORT |
| MatchResult execute(const LChar* input, unsigned start, unsigned length, int* output) |
| { |
| ASSERT(has8BitCode()); |
| |
| #if JS_TRACE_LOGGING |
| js::AutoTraceLog logger(js::TraceLogging::defaultLogger(), |
| js::TraceLogging::YARR_JIT_START, |
| js::TraceLogging::YARR_JIT_STOP); |
| #endif |
| |
| return MatchResult(reinterpret_cast<YarrJITCode8>(m_ref8.code().executableAddress())(input, start, length, output)); |
| } |
| |
| MatchResult execute(const LChar* input, unsigned start, unsigned length) |
| { |
| ASSERT(has8BitCodeMatchOnly()); |
| |
| #if JS_TRACE_LOGGING |
| js::AutoTraceLog logger(js::TraceLogging::defaultLogger(), |
| js::TraceLogging::YARR_JIT_START, |
| js::TraceLogging::YARR_JIT_STOP); |
| #endif |
| |
| return MatchResult(reinterpret_cast<YarrJITCodeMatchOnly8>(m_matchOnly8.code().executableAddress())(input, start, length)); |
| } |
| #endif |
| |
| MatchResult execute(const UChar* input, unsigned start, unsigned length, int* output) |
| { |
| ASSERT(has16BitCode()); |
| |
| #if JS_TRACE_LOGGING |
| js::AutoTraceLog logger(js::TraceLogging::defaultLogger(), |
| js::TraceLogging::YARR_JIT_START, |
| js::TraceLogging::YARR_JIT_STOP); |
| #endif |
| |
| YarrJITCode16 fn = JS_FUNC_TO_DATA_PTR(YarrJITCode16, m_ref16.code().executableAddress()); |
| return MatchResult(fn(input, start, length, output)); |
| } |
| |
| MatchResult execute(const UChar* input, unsigned start, unsigned length) |
| { |
| ASSERT(has16BitCodeMatchOnly()); |
| |
| #if JS_TRACE_LOGGING |
| js::AutoTraceLog logger(js::TraceLogging::defaultLogger(), |
| js::TraceLogging::YARR_JIT_START, |
| js::TraceLogging::YARR_JIT_STOP); |
| #endif |
| |
| YarrJITCodeMatchOnly16 fn = JS_FUNC_TO_DATA_PTR(YarrJITCodeMatchOnly16, m_matchOnly16.code().executableAddress()); |
| return MatchResult(fn(input, start, length)); |
| } |
| |
| #if ENABLE_REGEXP_TRACING |
| void *getAddr() { return m_ref.code().executableAddress(); } |
| #endif |
| |
| void clear() |
| { |
| #ifdef YARR_8BIT_CHAR_SUPPORT |
| m_ref8 = MacroAssemblerCodeRef(); |
| m_matchOnly8 = MacroAssemblerCodeRef(); |
| #endif |
| |
| m_ref16 = MacroAssemblerCodeRef(); |
| m_matchOnly16 = MacroAssemblerCodeRef(); |
| m_needFallBack = false; |
| } |
| |
| void release() { |
| #ifdef YARR_8BIT_CHAR_SUPPORT |
| m_ref8.release(); |
| m_matchOnly8.release(); |
| #endif |
| |
| m_ref16.release(); |
| m_matchOnly16.release(); |
| } |
| |
| private: |
| #ifdef YARR_8BIT_CHAR_SUPPORT |
| MacroAssemblerCodeRef m_ref8; |
| MacroAssemblerCodeRef m_matchOnly8; |
| #endif |
| |
| MacroAssemblerCodeRef m_ref16; |
| MacroAssemblerCodeRef m_matchOnly16; |
| bool m_needFallBack; |
| }; |
| |
| enum YarrJITCompileMode { |
| MatchOnly, |
| IncludeSubpatterns |
| }; |
| void jitCompile(YarrPattern&, YarrCharSize, JSGlobalData*, YarrCodeBlock& jitObject, YarrJITCompileMode = IncludeSubpatterns); |
| |
| } } // namespace JSC::Yarr |
| |
| #endif |
| |
| #endif /* yarr_YarrJIT_h */ |