Source/JavaScriptCore/yarr/YarrJIT.h - WebKit - Git at Google

 /*
  * Copyright (C) 2009-2021 Apple Inc. All rights reserved.
  * Copyright (C) 2019 the V8 project authors. 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.
  */

 #pragma once

 #if ENABLE(YARR_JIT)

 #include "MacroAssemblerCodeRef.h"
 #include "MatchResult.h"
 #include "VM.h"
 #include "Yarr.h"
 #include "YarrPattern.h"
 #include <wtf/Bitmap.h>
 #include <wtf/FixedVector.h>
 #include <wtf/UniqueRef.h>

 #define YARR_CALL

 namespace JSC {

 class VM;
 class ExecutablePool;

 namespace Yarr {

 class MatchingContextHolder;
 class YarrCodeBlock;

 enum class JITFailureReason : uint8_t {
     DecodeSurrogatePair,
     BackReference,
     ForwardReference,
     VariableCountedParenthesisWithNonZeroMinimum,
     ParenthesizedSubpattern,
     FixedCountParenthesizedSubpattern,
     ParenthesisNestedTooDeep,
     ExecutableMemoryAllocationFailure,
 };

 class BoyerMooreFastCandidates {
     WTF_MAKE_FAST_ALLOCATED(BoyerMooreFastCandidates);
 public:
     static constexpr unsigned maxSize = 2;
     using CharacterVector = Vector<UChar32, maxSize>;

     BoyerMooreFastCandidates() = default;

     bool isValid() const { return m_isValid; }
     void invalidate()
     {
         m_characters.clear();
         m_isValid = false;
     }

     bool isEmpty() const { return m_characters.isEmpty(); }
     unsigned size() const { return m_characters.size(); }
     UChar32 at(unsigned index) const { return m_characters.at(index); }

     void add(UChar32 character)
     {
         if (!isValid())
             return;
         if (!m_characters.contains(character)) {
             if (m_characters.size() < maxSize)
                 m_characters.append(character);
             else
                 invalidate();
         }
     }

     void merge(const BoyerMooreFastCandidates& other)
     {
         if (!isValid())
             return;
         if (!other.isValid()) {
             invalidate();
             return;
         }
         for (unsigned index = 0; index < other.size(); ++index)
             add(other.at(index));
     }

     void dump(PrintStream&) const;

 private:
     CharacterVector m_characters;
     bool m_isValid { true };
 };

 class BoyerMooreBitmap {
     WTF_MAKE_NONCOPYABLE(BoyerMooreBitmap);
     WTF_MAKE_FAST_ALLOCATED(BoyerMooreBitmap);
 public:
     static constexpr unsigned mapSize = 128;
     static constexpr unsigned mapMask = 128 - 1;
     using Map = Bitmap<mapSize>;

     BoyerMooreBitmap() = default;

     unsigned count() const { return m_count; }
     const Map& map() const { return m_map; }
     const BoyerMooreFastCandidates& charactersFastPath() const { return m_charactersFastPath; }

     bool add(CharSize charSize, UChar32 character)
     {
         if (isAllSet())
             return false;
         if (charSize == CharSize::Char8 && character > 0xff)
             return true;
         m_charactersFastPath.add(character);
         unsigned position = character & mapMask;
         if (!m_map.get(position)) {
             m_map.set(position);
             ++m_count;
         }
         return !isAllSet();
     }

     void addCharacters(CharSize charSize, const Vector<UChar32>& characters)
     {
         if (isAllSet())
             return;
         ASSERT(std::is_sorted(characters.begin(), characters.end()));
         for (UChar32 character : characters) {
             // Early return since characters are sorted.
             if (charSize == CharSize::Char8 && character > 0xff)
                 return;
             if (!add(charSize, character))
                 return;
         }
     }

     void addRanges(CharSize charSize, const Vector<CharacterRange>& ranges)
     {
         if (isAllSet())
             return;
         ASSERT(std::is_sorted(ranges.begin(), ranges.end(), [](CharacterRange lhs, CharacterRange rhs) {
                 return lhs.begin < rhs.begin;
             }));
         for (CharacterRange range : ranges) {
             auto begin = range.begin;
             auto end = range.end;
             if (charSize == CharSize::Char8) {
                 // Early return since ranges are sorted.
                 if (begin > 0xff)
                     return;
                 if (end > 0xff)
                     end = 0xff;
             }
             if (static_cast<unsigned>(end - begin + 1) >= mapSize) {
                 setAll();
                 return;
             }
             for (UChar32 character = begin; character <= end; ++character) {
                 if (!add(charSize, character))
                     return;
             }
         }
     }

     void setAll()
     {
         m_count = mapSize;
     }

     bool isAllSet() const { return m_count == mapSize; }

 private:
     Map m_map { };
     BoyerMooreFastCandidates m_charactersFastPath;
     unsigned m_count { 0 };
 };

 #if CPU(ARM64E)
 extern "C" EncodedMatchResult vmEntryToYarrJIT(const void* input, UCPURegister start, UCPURegister length, int* output, MatchingContextHolder* matchingContext, const void* codePtr);
 extern "C" void vmEntryToYarrJITAfter(void);
 #endif

 class YarrCodeBlock {
     WTF_MAKE_FAST_ALLOCATED;
     WTF_MAKE_NONCOPYABLE(YarrCodeBlock);

     using YarrJITCode8 = EncodedMatchResult (*)(const LChar* input, UCPURegister start, UCPURegister length, int* output, MatchingContextHolder& matchingContext) YARR_CALL;
     using YarrJITCode16 = EncodedMatchResult (*)(const UChar* input, UCPURegister start, UCPURegister length, int* output, MatchingContextHolder& matchingContext) YARR_CALL;
     using YarrJITCodeMatchOnly8 = EncodedMatchResult (*)(const LChar* input, UCPURegister start, UCPURegister length, void*, MatchingContextHolder& matchingContext) YARR_CALL;
     using YarrJITCodeMatchOnly16 = EncodedMatchResult (*)(const UChar* input, UCPURegister start, UCPURegister length, void*, MatchingContextHolder& matchingContext) YARR_CALL;

 public:
     YarrCodeBlock() = default;

     void setFallBackWithFailureReason(JITFailureReason failureReason) { m_failureReason = failureReason; }
     std::optional<JITFailureReason> failureReason() { return m_failureReason; }

     bool has8BitCode() { return m_ref8.size(); }
     bool has16BitCode() { return m_ref16.size(); }
     void set8BitCode(MacroAssemblerCodeRef<Yarr8BitPtrTag> ref, Vector<UniqueRef<BoyerMooreBitmap::Map>> maps)
     {
         m_ref8 = ref;
         m_maps.reserveCapacity(m_maps.size() + maps.size());
         for (unsigned index = 0; index < maps.size(); ++index)
             m_maps.uncheckedAppend(WTFMove(maps[index]));
     }
     void set16BitCode(MacroAssemblerCodeRef<Yarr16BitPtrTag> ref, Vector<UniqueRef<BoyerMooreBitmap::Map>> maps)
     {
         m_ref16 = ref;
         m_maps.reserveCapacity(m_maps.size() + maps.size());
         for (unsigned index = 0; index < maps.size(); ++index)
             m_maps.uncheckedAppend(WTFMove(maps[index]));
     }

     bool has8BitCodeMatchOnly() { return m_matchOnly8.size(); }
     bool has16BitCodeMatchOnly() { return m_matchOnly16.size(); }
     void set8BitCodeMatchOnly(MacroAssemblerCodeRef<YarrMatchOnly8BitPtrTag> matchOnly, Vector<UniqueRef<BoyerMooreBitmap::Map>> maps)
     {
         m_matchOnly8 = matchOnly;
         m_maps.reserveCapacity(m_maps.size() + maps.size());
         for (unsigned index = 0; index < maps.size(); ++index)
             m_maps.uncheckedAppend(WTFMove(maps[index]));
     }
     void set16BitCodeMatchOnly(MacroAssemblerCodeRef<YarrMatchOnly16BitPtrTag> matchOnly, Vector<UniqueRef<BoyerMooreBitmap::Map>> maps)
     {
         m_matchOnly16 = matchOnly;
         m_maps.reserveCapacity(m_maps.size() + maps.size());
         for (unsigned index = 0; index < maps.size(); ++index)
             m_maps.uncheckedAppend(WTFMove(maps[index]));
     }

     bool usesPatternContextBuffer() { return m_usesPatternContextBuffer; }
 #if ENABLE(YARR_JIT_ALL_PARENS_EXPRESSIONS)
     void setUsesPatternContextBuffer() { m_usesPatternContextBuffer = true; }
 #endif

     MatchResult execute(const LChar* input, unsigned start, unsigned length, int* output, MatchingContextHolder& matchingContext)
     {
         ASSERT(has8BitCode());
 #if CPU(ARM64E)
         if (Options::useJITCage())
             return MatchResult(vmEntryToYarrJIT(input, start, length, output, &matchingContext, retagCodePtr<Yarr8BitPtrTag, YarrEntryPtrTag>(m_ref8.code().executableAddress())));
 #endif
         return MatchResult(untagCFunctionPtr<YarrJITCode8, Yarr8BitPtrTag>(m_ref8.code().executableAddress())(input, start, length, output, matchingContext));
     }

     MatchResult execute(const UChar* input, unsigned start, unsigned length, int* output, MatchingContextHolder& matchingContext)
     {
         ASSERT(has16BitCode());
 #if CPU(ARM64E)
         if (Options::useJITCage())
             return MatchResult(vmEntryToYarrJIT(input, start, length, output, &matchingContext, retagCodePtr<Yarr16BitPtrTag, YarrEntryPtrTag>(m_ref16.code().executableAddress())));
 #endif
         return MatchResult(untagCFunctionPtr<YarrJITCode16, Yarr16BitPtrTag>(m_ref16.code().executableAddress())(input, start, length, output, matchingContext));
     }

     MatchResult execute(const LChar* input, unsigned start, unsigned length, MatchingContextHolder& matchingContext)
     {
         ASSERT(has8BitCodeMatchOnly());
 #if CPU(ARM64E)
         if (Options::useJITCage())
             return MatchResult(vmEntryToYarrJIT(input, start, length, nullptr, &matchingContext, retagCodePtr<YarrMatchOnly8BitPtrTag, YarrEntryPtrTag>(m_matchOnly8.code().executableAddress())));
 #endif
         return MatchResult(untagCFunctionPtr<YarrJITCodeMatchOnly8, YarrMatchOnly8BitPtrTag>(m_matchOnly8.code().executableAddress())(input, start, length, nullptr, matchingContext));
     }

     MatchResult execute(const UChar* input, unsigned start, unsigned length, MatchingContextHolder& matchingContext)
     {
         ASSERT(has16BitCodeMatchOnly());
 #if CPU(ARM64E)
         if (Options::useJITCage())
             return MatchResult(vmEntryToYarrJIT(input, start, length, nullptr, &matchingContext, retagCodePtr<YarrMatchOnly16BitPtrTag, YarrEntryPtrTag>(m_matchOnly16.code().executableAddress())));
 #endif
         return MatchResult(untagCFunctionPtr<YarrJITCodeMatchOnly16, YarrMatchOnly16BitPtrTag>(m_matchOnly16.code().executableAddress())(input, start, length, nullptr, matchingContext));
     }

 #if ENABLE(REGEXP_TRACING)
     void *get8BitMatchOnlyAddr()
     {
         if (!has8BitCodeMatchOnly())
             return 0;

         return m_matchOnly8.code().executableAddress();
     }

     void *get16BitMatchOnlyAddr()
     {
         if (!has16BitCodeMatchOnly())
             return 0;

         return m_matchOnly16.code().executableAddress();
     }

     void *get8BitMatchAddr()
     {
         if (!has8BitCode())
             return 0;

         return m_ref8.code().executableAddress();
     }

     void *get16BitMatchAddr()
     {
         if (!has16BitCode())
             return 0;

         return m_ref16.code().executableAddress();
     }
 #endif

     size_t size() const
     {
         return m_ref8.size() + m_ref16.size() + m_matchOnly8.size() + m_matchOnly16.size();
     }

     void clear(const AbstractLocker&)
     {
         m_ref8 = MacroAssemblerCodeRef<Yarr8BitPtrTag>();
         m_ref16 = MacroAssemblerCodeRef<Yarr16BitPtrTag>();
         m_matchOnly8 = MacroAssemblerCodeRef<YarrMatchOnly8BitPtrTag>();
         m_matchOnly16 = MacroAssemblerCodeRef<YarrMatchOnly16BitPtrTag>();
         m_maps.clear();
         m_failureReason = std::nullopt;
     }

     const BoyerMooreBitmap::Map::WordType* tryReuseBoyerMooreBitmap(const BoyerMooreBitmap::Map& map) const
     {
         for (auto& stored : m_maps) {
             if (stored.get() == map)
                 return stored->storage();
         }
         return nullptr;
     }

 private:
     MacroAssemblerCodeRef<Yarr8BitPtrTag> m_ref8;
     MacroAssemblerCodeRef<Yarr16BitPtrTag> m_ref16;
     MacroAssemblerCodeRef<YarrMatchOnly8BitPtrTag> m_matchOnly8;
     MacroAssemblerCodeRef<YarrMatchOnly16BitPtrTag> m_matchOnly16;
     Vector<UniqueRef<BoyerMooreBitmap::Map>> m_maps;
     bool m_usesPatternContextBuffer { false };
     std::optional<JITFailureReason> m_failureReason;
 };

 enum class JITCompileMode : uint8_t {
     MatchOnly,
     IncludeSubpatterns
 };
 void jitCompile(YarrPattern&, String& patternString, CharSize, VM*, YarrCodeBlock& jitObject, JITCompileMode);

 } } // namespace JSC::Yarr

 #endif
	/*
	* Copyright (C) 2009-2021 Apple Inc. All rights reserved.
	* Copyright (C) 2019 the V8 project authors. 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.
	*/

	#pragma once

	#if ENABLE(YARR_JIT)

	#include "MacroAssemblerCodeRef.h"
	#include "MatchResult.h"
	#include "VM.h"
	#include "Yarr.h"
	#include "YarrPattern.h"
	#include <wtf/Bitmap.h>
	#include <wtf/FixedVector.h>
	#include <wtf/UniqueRef.h>

	#define YARR_CALL

	namespace JSC {

	class VM;
	class ExecutablePool;

	namespace Yarr {

	class MatchingContextHolder;
	class YarrCodeBlock;

	enum class JITFailureReason : uint8_t {
	DecodeSurrogatePair,
	BackReference,
	ForwardReference,
	VariableCountedParenthesisWithNonZeroMinimum,
	ParenthesizedSubpattern,
	FixedCountParenthesizedSubpattern,
	ParenthesisNestedTooDeep,
	ExecutableMemoryAllocationFailure,
	};

	class BoyerMooreFastCandidates {
	WTF_MAKE_FAST_ALLOCATED(BoyerMooreFastCandidates);
	public:
	static constexpr unsigned maxSize = 2;
	using CharacterVector = Vector<UChar32, maxSize>;

	BoyerMooreFastCandidates() = default;

	bool isValid() const { return m_isValid; }
	void invalidate()
	{
	m_characters.clear();
	m_isValid = false;
	}

	bool isEmpty() const { return m_characters.isEmpty(); }
	unsigned size() const { return m_characters.size(); }
	UChar32 at(unsigned index) const { return m_characters.at(index); }

	void add(UChar32 character)
	{
	if (!isValid())
	return;
	if (!m_characters.contains(character)) {
	if (m_characters.size() < maxSize)
	m_characters.append(character);
	else
	invalidate();
	}
	}

	void merge(const BoyerMooreFastCandidates& other)
	{
	if (!isValid())
	return;
	if (!other.isValid()) {
	invalidate();
	return;
	}
	for (unsigned index = 0; index < other.size(); ++index)
	add(other.at(index));
	}

	void dump(PrintStream&) const;

	private:
	CharacterVector m_characters;
	bool m_isValid { true };
	};

	class BoyerMooreBitmap {
	WTF_MAKE_NONCOPYABLE(BoyerMooreBitmap);
	WTF_MAKE_FAST_ALLOCATED(BoyerMooreBitmap);
	public:
	static constexpr unsigned mapSize = 128;
	static constexpr unsigned mapMask = 128 - 1;
	using Map = Bitmap<mapSize>;

	BoyerMooreBitmap() = default;

	unsigned count() const { return m_count; }
	const Map& map() const { return m_map; }
	const BoyerMooreFastCandidates& charactersFastPath() const { return m_charactersFastPath; }

	bool add(CharSize charSize, UChar32 character)
	{
	if (isAllSet())
	return false;
	if (charSize == CharSize::Char8 && character > 0xff)
	return true;
	m_charactersFastPath.add(character);
	unsigned position = character & mapMask;
	if (!m_map.get(position)) {
	m_map.set(position);
	++m_count;
	}
	return !isAllSet();
	}

	void addCharacters(CharSize charSize, const Vector<UChar32>& characters)
	{
	if (isAllSet())
	return;
	ASSERT(std::is_sorted(characters.begin(), characters.end()));
	for (UChar32 character : characters) {
	// Early return since characters are sorted.
	if (charSize == CharSize::Char8 && character > 0xff)
	return;
	if (!add(charSize, character))
	return;
	}
	}

	void addRanges(CharSize charSize, const Vector<CharacterRange>& ranges)
	{
	if (isAllSet())
	return;
	ASSERT(std::is_sorted(ranges.begin(), ranges.end(), [](CharacterRange lhs, CharacterRange rhs) {
	return lhs.begin < rhs.begin;
	}));
	for (CharacterRange range : ranges) {
	auto begin = range.begin;
	auto end = range.end;
	if (charSize == CharSize::Char8) {
	// Early return since ranges are sorted.
	if (begin > 0xff)
	return;
	if (end > 0xff)
	end = 0xff;
	}
	if (static_cast<unsigned>(end - begin + 1) >= mapSize) {
	setAll();
	return;
	}
	for (UChar32 character = begin; character <= end; ++character) {
	if (!add(charSize, character))
	return;
	}
	}
	}

	void setAll()
	{
	m_count = mapSize;
	}

	bool isAllSet() const { return m_count == mapSize; }

	private:
	Map m_map { };
	BoyerMooreFastCandidates m_charactersFastPath;
	unsigned m_count { 0 };
	};

	#if CPU(ARM64E)
	extern "C" EncodedMatchResult vmEntryToYarrJIT(const void* input, UCPURegister start, UCPURegister length, int* output, MatchingContextHolder* matchingContext, const void* codePtr);
	extern "C" void vmEntryToYarrJITAfter(void);
	#endif

	class YarrCodeBlock {
	WTF_MAKE_FAST_ALLOCATED;
	WTF_MAKE_NONCOPYABLE(YarrCodeBlock);

	using YarrJITCode8 = EncodedMatchResult ()(const LChar input, UCPURegister start, UCPURegister length, int* output, MatchingContextHolder& matchingContext) YARR_CALL;
	using YarrJITCode16 = EncodedMatchResult ()(const UChar input, UCPURegister start, UCPURegister length, int* output, MatchingContextHolder& matchingContext) YARR_CALL;
	using YarrJITCodeMatchOnly8 = EncodedMatchResult ()(const LChar input, UCPURegister start, UCPURegister length, void*, MatchingContextHolder& matchingContext) YARR_CALL;
	using YarrJITCodeMatchOnly16 = EncodedMatchResult ()(const UChar input, UCPURegister start, UCPURegister length, void*, MatchingContextHolder& matchingContext) YARR_CALL;

	public:
	YarrCodeBlock() = default;

	void setFallBackWithFailureReason(JITFailureReason failureReason) { m_failureReason = failureReason; }
	std::optional<JITFailureReason> failureReason() { return m_failureReason; }

	bool has8BitCode() { return m_ref8.size(); }
	bool has16BitCode() { return m_ref16.size(); }
	void set8BitCode(MacroAssemblerCodeRef<Yarr8BitPtrTag> ref, Vector<UniqueRef<BoyerMooreBitmap::Map>> maps)
	{
	m_ref8 = ref;
	m_maps.reserveCapacity(m_maps.size() + maps.size());
	for (unsigned index = 0; index < maps.size(); ++index)
	m_maps.uncheckedAppend(WTFMove(maps[index]));
	}
	void set16BitCode(MacroAssemblerCodeRef<Yarr16BitPtrTag> ref, Vector<UniqueRef<BoyerMooreBitmap::Map>> maps)
	{
	m_ref16 = ref;
	m_maps.reserveCapacity(m_maps.size() + maps.size());
	for (unsigned index = 0; index < maps.size(); ++index)
	m_maps.uncheckedAppend(WTFMove(maps[index]));
	}

	bool has8BitCodeMatchOnly() { return m_matchOnly8.size(); }
	bool has16BitCodeMatchOnly() { return m_matchOnly16.size(); }
	void set8BitCodeMatchOnly(MacroAssemblerCodeRef<YarrMatchOnly8BitPtrTag> matchOnly, Vector<UniqueRef<BoyerMooreBitmap::Map>> maps)
	{
	m_matchOnly8 = matchOnly;
	m_maps.reserveCapacity(m_maps.size() + maps.size());
	for (unsigned index = 0; index < maps.size(); ++index)
	m_maps.uncheckedAppend(WTFMove(maps[index]));
	}
	void set16BitCodeMatchOnly(MacroAssemblerCodeRef<YarrMatchOnly16BitPtrTag> matchOnly, Vector<UniqueRef<BoyerMooreBitmap::Map>> maps)
	{
	m_matchOnly16 = matchOnly;
	m_maps.reserveCapacity(m_maps.size() + maps.size());
	for (unsigned index = 0; index < maps.size(); ++index)
	m_maps.uncheckedAppend(WTFMove(maps[index]));
	}

	bool usesPatternContextBuffer() { return m_usesPatternContextBuffer; }
	#if ENABLE(YARR_JIT_ALL_PARENS_EXPRESSIONS)
	void setUsesPatternContextBuffer() { m_usesPatternContextBuffer = true; }
	#endif

	MatchResult execute(const LChar* input, unsigned start, unsigned length, int* output, MatchingContextHolder& matchingContext)
	{
	ASSERT(has8BitCode());
	#if CPU(ARM64E)
	if (Options::useJITCage())
	return MatchResult(vmEntryToYarrJIT(input, start, length, output, &matchingContext, retagCodePtr<Yarr8BitPtrTag, YarrEntryPtrTag>(m_ref8.code().executableAddress())));
	#endif
	return MatchResult(untagCFunctionPtr<YarrJITCode8, Yarr8BitPtrTag>(m_ref8.code().executableAddress())(input, start, length, output, matchingContext));
	}

	MatchResult execute(const UChar* input, unsigned start, unsigned length, int* output, MatchingContextHolder& matchingContext)
	{
	ASSERT(has16BitCode());
	#if CPU(ARM64E)
	if (Options::useJITCage())
	return MatchResult(vmEntryToYarrJIT(input, start, length, output, &matchingContext, retagCodePtr<Yarr16BitPtrTag, YarrEntryPtrTag>(m_ref16.code().executableAddress())));
	#endif
	return MatchResult(untagCFunctionPtr<YarrJITCode16, Yarr16BitPtrTag>(m_ref16.code().executableAddress())(input, start, length, output, matchingContext));
	}

	MatchResult execute(const LChar* input, unsigned start, unsigned length, MatchingContextHolder& matchingContext)
	{
	ASSERT(has8BitCodeMatchOnly());
	#if CPU(ARM64E)
	if (Options::useJITCage())
	return MatchResult(vmEntryToYarrJIT(input, start, length, nullptr, &matchingContext, retagCodePtr<YarrMatchOnly8BitPtrTag, YarrEntryPtrTag>(m_matchOnly8.code().executableAddress())));
	#endif
	return MatchResult(untagCFunctionPtr<YarrJITCodeMatchOnly8, YarrMatchOnly8BitPtrTag>(m_matchOnly8.code().executableAddress())(input, start, length, nullptr, matchingContext));
	}

	MatchResult execute(const UChar* input, unsigned start, unsigned length, MatchingContextHolder& matchingContext)
	{
	ASSERT(has16BitCodeMatchOnly());
	#if CPU(ARM64E)
	if (Options::useJITCage())
	return MatchResult(vmEntryToYarrJIT(input, start, length, nullptr, &matchingContext, retagCodePtr<YarrMatchOnly16BitPtrTag, YarrEntryPtrTag>(m_matchOnly16.code().executableAddress())));
	#endif
	return MatchResult(untagCFunctionPtr<YarrJITCodeMatchOnly16, YarrMatchOnly16BitPtrTag>(m_matchOnly16.code().executableAddress())(input, start, length, nullptr, matchingContext));
	}

	#if ENABLE(REGEXP_TRACING)
	void *get8BitMatchOnlyAddr()
	{
	if (!has8BitCodeMatchOnly())
	return 0;

	return m_matchOnly8.code().executableAddress();
	}

	void *get16BitMatchOnlyAddr()
	{
	if (!has16BitCodeMatchOnly())
	return 0;

	return m_matchOnly16.code().executableAddress();
	}

	void *get8BitMatchAddr()
	{
	if (!has8BitCode())
	return 0;

	return m_ref8.code().executableAddress();
	}

	void *get16BitMatchAddr()
	{
	if (!has16BitCode())
	return 0;

	return m_ref16.code().executableAddress();
	}
	#endif

	size_t size() const
	{
	return m_ref8.size() + m_ref16.size() + m_matchOnly8.size() + m_matchOnly16.size();
	}

	void clear(const AbstractLocker&)
	{
	m_ref8 = MacroAssemblerCodeRef<Yarr8BitPtrTag>();
	m_ref16 = MacroAssemblerCodeRef<Yarr16BitPtrTag>();
	m_matchOnly8 = MacroAssemblerCodeRef<YarrMatchOnly8BitPtrTag>();
	m_matchOnly16 = MacroAssemblerCodeRef<YarrMatchOnly16BitPtrTag>();
	m_maps.clear();
	m_failureReason = std::nullopt;
	}

	const BoyerMooreBitmap::Map::WordType* tryReuseBoyerMooreBitmap(const BoyerMooreBitmap::Map& map) const
	{
	for (auto& stored : m_maps) {
	if (stored.get() == map)
	return stored->storage();
	}
	return nullptr;
	}

	private:
	MacroAssemblerCodeRef<Yarr8BitPtrTag> m_ref8;
	MacroAssemblerCodeRef<Yarr16BitPtrTag> m_ref16;
	MacroAssemblerCodeRef<YarrMatchOnly8BitPtrTag> m_matchOnly8;
	MacroAssemblerCodeRef<YarrMatchOnly16BitPtrTag> m_matchOnly16;
	Vector<UniqueRef<BoyerMooreBitmap::Map>> m_maps;
	bool m_usesPatternContextBuffer { false };
	std::optional<JITFailureReason> m_failureReason;
	};

	enum class JITCompileMode : uint8_t {
	MatchOnly,
	IncludeSubpatterns
	};
	void jitCompile(YarrPattern&, String& patternString, CharSize, VM*, YarrCodeBlock& jitObject, JITCompileMode);

	} } // namespace JSC::Yarr

	#endif