LibRegex: Account for uppercase characters in insensitive patterns

This commit is contained in:
Ali Mohammad Pur 2025-07-11 15:06:40 +02:00 committed by Jelle Raaijmakers
commit 5b45223d5f
Notes: github-actions[bot] 2025-07-12 09:27:33 +00:00
6 changed files with 73 additions and 31 deletions

View file

@ -6,6 +6,7 @@
#include "RegexByteCode.h"
#include "RegexDebug.h"
#include <AK/BinarySearch.h>
#include <AK/CharacterTypes.h>
#include <AK/StringBuilder.h>
@ -539,25 +540,24 @@ ALWAYS_INLINE ExecutionResult OpCode_Compare::execute(MatchInput const& input, M
if (input.view.length() <= state.string_position)
return ExecutionResult::Failed_ExecuteLowPrioForks;
auto count = m_bytecode->at(offset++);
auto range_data = m_bytecode->flat_data().slice(offset, count);
offset += count;
auto count_sensitive = m_bytecode->at(offset++);
auto count_insensitive = m_bytecode->at(offset++);
auto sensitive_range_data = m_bytecode->flat_data().slice(offset, count_sensitive);
offset += count_sensitive;
auto insensitive_range_data = m_bytecode->flat_data().slice(offset, count_insensitive);
offset += count_insensitive;
bool const insensitive = input.regex_options & AllFlags::Insensitive;
auto ch = input.view[state.string_position_in_code_units];
if (insensitive)
ch = to_ascii_lowercase(ch);
auto const* matching_range = binary_search(range_data, ch, nullptr, [insensitive = input.regex_options & AllFlags::Insensitive](auto needle, CharRange range) {
auto upper_case_needle = needle;
auto lower_case_needle = needle;
if (insensitive) {
upper_case_needle = to_ascii_uppercase(needle);
lower_case_needle = to_ascii_lowercase(needle);
}
if (lower_case_needle >= range.from && lower_case_needle <= range.to)
auto const ranges = insensitive && !insensitive_range_data.is_empty() ? insensitive_range_data : sensitive_range_data;
auto const* matching_range = binary_search(ranges, ch, nullptr, [](auto needle, CharRange range) {
if (needle >= range.from && needle <= range.to)
return 0;
if (upper_case_needle >= range.from && upper_case_needle <= range.to)
return 0;
if (lower_case_needle > range.to || upper_case_needle > range.to)
if (needle > range.to)
return 1;
return -1;
});
@ -934,9 +934,11 @@ Vector<CompareTypeAndValuePair> OpCode_Compare::flat_compares() const
auto value = m_bytecode->at(offset++);
result.append({ compare_type, value });
} else if (compare_type == CharacterCompareType::LookupTable) {
auto count = m_bytecode->at(offset++);
for (size_t i = 0; i < count; ++i)
auto count_sensitive = m_bytecode->at(offset++);
auto count_insensitive = m_bytecode->at(offset++);
for (size_t i = 0; i < count_sensitive; ++i)
result.append({ CharacterCompareType::CharRange, m_bytecode->at(offset++) });
offset += count_insensitive; // Skip insensitive ranges
} else if (compare_type == CharacterCompareType::GeneralCategory
|| compare_type == CharacterCompareType::Property
|| compare_type == CharacterCompareType::Script
@ -1027,11 +1029,21 @@ Vector<ByteString> OpCode_Compare::variable_arguments_to_byte_string(Optional<Ma
" compare against: '{}'",
input.value().view.substring_view(string_start_offset, state().string_position > view.length() ? 0 : 1).to_byte_string()));
} else if (compare_type == CharacterCompareType::LookupTable) {
auto count = m_bytecode->at(offset++);
for (size_t j = 0; j < count; ++j) {
auto count_sensitive = m_bytecode->at(offset++);
auto count_insensitive = m_bytecode->at(offset++);
for (size_t j = 0; j < count_sensitive; ++j) {
auto range = (CharRange)m_bytecode->at(offset++);
result.append(ByteString::formatted(" {:x}-{:x}", range.from, range.to));
}
if (count_insensitive > 0) {
result.append(" [insensitive ranges:");
for (size_t j = 0; j < count_insensitive; ++j) {
auto range = (CharRange)m_bytecode->at(offset++);
result.append(ByteString::formatted(" {:x}-{:x}", range.from, range.to));
}
result.append(" ]");
}
if (!view.is_null() && view.length() > state().string_position)
result.empend(ByteString::formatted(
" compare against: '{}'",

View file

@ -117,8 +117,8 @@ enum class BoundaryCheckType : ByteCodeValueType {
};
struct CharRange {
u32 const from;
u32 const to;
u32 from;
u32 to;
CharRange(u64 value)
: from(value >> 32)
@ -213,6 +213,12 @@ public:
Base::extend(other);
}
template<SameAs<Vector<ByteCodeValueType>> T>
void extend(T other)
{
Base::append(move(other));
}
template<typename... Args>
void empend(Args&&... args)
{

View file

@ -288,8 +288,14 @@ RegexResult Matcher<Parser>::match(Vector<RegexStringView> const& views, Optiona
if (match_length_minimum && match_length_minimum > view_length - view_index)
break;
auto const insensitive = input.regex_options.has_flag_set(AllFlags::Insensitive);
if (auto& starting_ranges = m_pattern->parser_result.optimization_data.starting_ranges; !starting_ranges.is_empty()) {
if (!binary_search(starting_ranges, input.view.code_unit_at(view_index), nullptr, compare_range))
auto ranges = insensitive ? m_pattern->parser_result.optimization_data.starting_ranges_insensitive.span() : starting_ranges.span();
auto ch = input.view.code_unit_at(view_index);
if (insensitive)
ch = to_ascii_lowercase(ch);
if (!binary_search(ranges, ch, nullptr, compare_range))
goto done_matching;
}

View file

@ -223,8 +223,11 @@ void Regex<Parser>::fill_optimization_data(BasicBlockList const& blocks)
if (!compares.char_classes.is_empty() || !compares.negated_char_classes.is_empty() || !compares.negated_ranges.is_empty())
return;
for (auto it = compares.ranges.begin(); it != compares.ranges.end(); ++it)
for (auto it = compares.ranges.begin(); it != compares.ranges.end(); ++it) {
parser_result.optimization_data.starting_ranges.append({ it.key(), *it });
parser_result.optimization_data.starting_ranges_insensitive.append({ to_ascii_lowercase(it.key()), to_ascii_lowercase(*it) });
quick_sort(parser_result.optimization_data.starting_ranges_insensitive, [](CharRange a, CharRange b) { return a.from < b.from; });
}
return;
}
case OpCodeId::CheckBegin:
@ -1863,10 +1866,13 @@ void Optimizer::append_character_class(ByteCode& target, Vector<CompareTypeAndVa
auto append_table = [&](auto& table) {
++argument_count;
arguments.append(to_underlying(CharacterCompareType::LookupTable));
auto size_index = arguments.size();
auto sensitive_size_index = arguments.size();
auto insensitive_size_index = sensitive_size_index + 1;
arguments.append(0);
arguments.append(0);
Optional<CharRange> active_range;
size_t range_count = 0;
Vector<ByteCodeValueType> range_data;
for (auto& range : table) {
if (!active_range.has_value()) {
active_range = range;
@ -1876,16 +1882,25 @@ void Optimizer::append_character_class(ByteCode& target, Vector<CompareTypeAndVa
if (range.from <= active_range->to + 1 && range.to + 1 >= active_range->from) {
active_range = CharRange { min(range.from, active_range->from), max(range.to, active_range->to) };
} else {
++range_count;
arguments.append(active_range.release_value());
range_data.append(active_range.release_value());
active_range = range;
}
}
if (active_range.has_value()) {
++range_count;
arguments.append(active_range.release_value());
if (active_range.has_value())
range_data.append(active_range.release_value());
arguments.extend(range_data);
arguments[sensitive_size_index] = range_data.size();
if (!all_of(range_data, [](CharRange range) { return range.from == to_ascii_lowercase(range.from) && range.to == to_ascii_lowercase(range.to); })) {
Vector<ByteCodeValueType> insensitive_data;
insensitive_data.ensure_capacity(range_data.size());
for (CharRange range : range_data)
insensitive_data.append(CharRange { to_ascii_lowercase(range.from), to_ascii_lowercase(range.to) });
quick_sort(insensitive_data, [](CharRange a, CharRange b) { return a.from < b.from; });
arguments.extend(insensitive_data);
arguments[insensitive_size_index] = insensitive_data.size();
}
arguments[size_index] = range_count;
};
auto contains_regular_table = !table.is_empty();

View file

@ -66,6 +66,7 @@ public:
Optional<ByteString> pure_substring_search;
// If populated, the pattern only accepts strings that start with a character in these ranges.
Vector<CharRange> starting_ranges;
Vector<CharRange> starting_ranges_insensitive;
bool only_start_of_line = false;
} optimization_data {};
};

View file

@ -736,6 +736,8 @@ TEST_CASE(ECMA262_match)
{ "(?!(b))\\1"sv, "a"sv, false },
// String table merge bug: inverse map should be merged regardless of available direct mappings.
{ "((?<x>a)|(?<x>b))"sv, "aa"sv, false },
// Insensitive charclasses should accept upper/lowercase in pattern (lookup table should still be ordered if insensitive lookup is used), ladybird#5399.
{ "[aBc]"sv, "b"sv, true, ECMAScriptFlags::Insensitive },
};
for (auto& test : tests) {