mirrors/ladybird
0
0
Fork 0
mirror of https://github.com/LadybirdBrowser/ladybird.git synced 2025-09-07 01:56:38 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions 9

LibWasm: Validate number of data sections

Browse source
This commit is contained in:
Diego 2024-06-10 12:42:13 -07:00 • committed by Ali Mohammad Pur
commit e64ac8c177
Notes: sideshowbarker 2024-07-17 06:33:00 +09:00
2 changed files with 7 additions and 0 deletions
Download patch file Download diff file Expand all files Collapse all files

6
Userland/Libraries/LibWasm/AbstractMachine/Validator.cpp
View file

@ -41,6 +41,10 @@ ErrorOr<void, ValidationError> Validator::validate(Module& module)
m_context.types.extend(section.types()); m_context.types.extend(section.types());
}); });
module.for_each_section_of_type<DataCountSection>([this](DataCountSection const& section) {
m_context.data_count = section.count();
});
module.for_each_section_of_type<ImportSection>([&](ImportSection const& section) { module.for_each_section_of_type<ImportSection>([&](ImportSection const& section) {
for (auto& import_ : section.imports()) { for (auto& import_ : section.imports()) {
import_.description().visit( import_.description().visit(
@ -177,6 +181,8 @@ ErrorOr<void, ValidationError> Validator::validate(StartSection const& section)
ErrorOr<void, ValidationError> Validator::validate(DataSection const& section) ErrorOr<void, ValidationError> Validator::validate(DataSection const& section)
{ {
if (m_context.data_count.has_value() && section.data().size() != m_context.data_count)
return Errors::invalid("data count does not match segment count"sv);
for (auto& entry : section.data()) { for (auto& entry : section.data()) {
TRY(entry.value().visit( TRY(entry.value().visit(
[](DataSection::Data::Passive const&) { return ErrorOr<void, ValidationError> {}; }, [](DataSection::Data::Passive const&) { return ErrorOr<void, ValidationError> {}; },

1
Userland/Libraries/LibWasm/AbstractMachine/Validator.h
View file

@ -27,6 +27,7 @@ struct Context {
COWVector<bool> datas; COWVector<bool> datas;
COWVector<ValueType> locals; COWVector<ValueType> locals;
COWVector<ResultType> labels; COWVector<ResultType> labels;
Optional<u32> data_count;
Optional<ResultType> return_; Optional<ResultType> return_;
AK::HashTable<FunctionIndex> references; AK::HashTable<FunctionIndex> references;
size_t imported_function_count { 0 }; size_t imported_function_count { 0 };