45a6a3d63e886c5a47e5fcc30844f10e0739bc5a
[WebKit-https.git] / Source / JavaScriptCore / yarr / YarrJIT.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 #pragma once
27
28 #if ENABLE(YARR_JIT)
29
30 #include "VM.h"
31 #include "MacroAssemblerCodeRef.h"
32 #include "MatchResult.h"
33 #include "Yarr.h"
34 #include "YarrPattern.h"
35
36 #if CPU(X86) && !COMPILER(MSVC)
37 #define YARR_CALL __attribute__ ((regparm (3)))
38 #else
39 #define YARR_CALL
40 #endif
41
42 namespace JSC {
43
44 class VM;
45 class ExecutablePool;
46
47 namespace Yarr {
48
49 class YarrCodeBlock {
50 #if CPU(X86_64) || CPU(ARM64)
51     typedef MatchResult (*YarrJITCode8)(const LChar* input, unsigned start, unsigned length, int* output) YARR_CALL;
52     typedef MatchResult (*YarrJITCode16)(const UChar* input, unsigned start, unsigned length, int* output) YARR_CALL;
53     typedef MatchResult (*YarrJITCodeMatchOnly8)(const LChar* input, unsigned start, unsigned length) YARR_CALL;
54     typedef MatchResult (*YarrJITCodeMatchOnly16)(const UChar* input, unsigned start, unsigned length) YARR_CALL;
55 #else
56     typedef EncodedMatchResult (*YarrJITCode8)(const LChar* input, unsigned start, unsigned length, int* output) YARR_CALL;
57     typedef EncodedMatchResult (*YarrJITCode16)(const UChar* input, unsigned start, unsigned length, int* output) YARR_CALL;
58     typedef EncodedMatchResult (*YarrJITCodeMatchOnly8)(const LChar* input, unsigned start, unsigned length) YARR_CALL;
59     typedef EncodedMatchResult (*YarrJITCodeMatchOnly16)(const UChar* input, unsigned start, unsigned length) YARR_CALL;
60 #endif
61
62 public:
63     YarrCodeBlock()
64         : m_needFallBack(false)
65     {
66     }
67
68     ~YarrCodeBlock()
69     {
70     }
71
72     void setFallBack(bool fallback) { m_needFallBack = fallback; }
73     bool isFallBack() { return m_needFallBack; }
74
75     bool has8BitCode() { return m_ref8.size(); }
76     bool has16BitCode() { return m_ref16.size(); }
77     void set8BitCode(MacroAssemblerCodeRef ref) { m_ref8 = ref; }
78     void set16BitCode(MacroAssemblerCodeRef ref) { m_ref16 = ref; }
79
80     bool has8BitCodeMatchOnly() { return m_matchOnly8.size(); }
81     bool has16BitCodeMatchOnly() { return m_matchOnly16.size(); }
82     void set8BitCodeMatchOnly(MacroAssemblerCodeRef matchOnly) { m_matchOnly8 = matchOnly; }
83     void set16BitCodeMatchOnly(MacroAssemblerCodeRef matchOnly) { m_matchOnly16 = matchOnly; }
84
85     MatchResult execute(const LChar* input, unsigned start, unsigned length, int* output)
86     {
87         ASSERT(has8BitCode());
88         return MatchResult(reinterpret_cast<YarrJITCode8>(m_ref8.code().executableAddress())(input, start, length, output));
89     }
90
91     MatchResult execute(const UChar* input, unsigned start, unsigned length, int* output)
92     {
93         ASSERT(has16BitCode());
94         return MatchResult(reinterpret_cast<YarrJITCode16>(m_ref16.code().executableAddress())(input, start, length, output));
95     }
96
97     MatchResult execute(const LChar* input, unsigned start, unsigned length)
98     {
99         ASSERT(has8BitCodeMatchOnly());
100         return MatchResult(reinterpret_cast<YarrJITCodeMatchOnly8>(m_matchOnly8.code().executableAddress())(input, start, length));
101     }
102
103     MatchResult execute(const UChar* input, unsigned start, unsigned length)
104     {
105         ASSERT(has16BitCodeMatchOnly());
106         return MatchResult(reinterpret_cast<YarrJITCodeMatchOnly16>(m_matchOnly16.code().executableAddress())(input, start, length));
107     }
108
109 #if ENABLE(REGEXP_TRACING)
110     void *get8BitMatchOnlyAddr()
111     {
112         if (!has8BitCodeMatchOnly())
113             return 0;
114
115         return m_matchOnly8.code().executableAddress();
116     }
117
118     void *get16BitMatchOnlyAddr()
119     {
120         if (!has16BitCodeMatchOnly())
121             return 0;
122
123         return m_matchOnly16.code().executableAddress();
124     }
125
126     void *get8BitMatchAddr()
127     {
128         if (!has8BitCode())
129             return 0;
130
131         return m_ref8.code().executableAddress();
132     }
133
134     void *get16BitMatchAddr()
135     {
136         if (!has16BitCode())
137             return 0;
138
139         return m_ref16.code().executableAddress();
140     }
141 #endif
142
143     size_t size() const
144     {
145         return m_ref8.size() + m_ref16.size() + m_matchOnly8.size() + m_matchOnly16.size();
146     }
147
148     void clear()
149     {
150         m_ref8 = MacroAssemblerCodeRef();
151         m_ref16 = MacroAssemblerCodeRef();
152         m_matchOnly8 = MacroAssemblerCodeRef();
153         m_matchOnly16 = MacroAssemblerCodeRef();
154         m_needFallBack = false;
155     }
156
157 private:
158     MacroAssemblerCodeRef m_ref8;
159     MacroAssemblerCodeRef m_ref16;
160     MacroAssemblerCodeRef m_matchOnly8;
161     MacroAssemblerCodeRef m_matchOnly16;
162     bool m_needFallBack;
163 };
164
165 enum YarrJITCompileMode {
166     MatchOnly,
167     IncludeSubpatterns
168 };
169 void jitCompile(YarrPattern&, YarrCharSize, VM*, YarrCodeBlock& jitObject, YarrJITCompileMode = IncludeSubpatterns);
170
171 } } // namespace JSC::Yarr
172
173 #endif