diff options
author | Tim Schumacher <timschumi@gmx.de> | 2023-01-21 11:44:19 +0100 |
---|---|---|
committer | Ali Mohammad Pur <Ali.mpfard@gmail.com> | 2023-01-25 17:10:05 +0330 |
commit | 982ebbc304a8e2ed637009f9bcc4dac53b43e06c (patch) | |
tree | 5e671d853859510a19534cbfde184b6c683ba4b3 /Userland/Libraries | |
parent | 409fb0fe79071686590cfd1fdbc815193a15d795 (diff) | |
download | serenity-982ebbc304a8e2ed637009f9bcc4dac53b43e06c.zip |
LibWasm: Port the parser to `Core::Stream`
Diffstat (limited to 'Userland/Libraries')
-rw-r--r-- | Userland/Libraries/LibWasm/Parser/Parser.cpp | 319 | ||||
-rw-r--r-- | Userland/Libraries/LibWasm/Types.h | 186 | ||||
-rw-r--r-- | Userland/Libraries/LibWeb/WebAssembly/WebAssemblyObject.cpp | 10 |
3 files changed, 275 insertions, 240 deletions
diff --git a/Userland/Libraries/LibWasm/Parser/Parser.cpp b/Userland/Libraries/LibWasm/Parser/Parser.cpp index 093635b03a..45fd96390d 100644 --- a/Userland/Libraries/LibWasm/Parser/Parser.cpp +++ b/Userland/Libraries/LibWasm/Parser/Parser.cpp @@ -8,25 +8,27 @@ #include <AK/LEB128.h> #include <AK/ScopeGuard.h> #include <AK/ScopeLogger.h> +#include <LibCore/MemoryStream.h> #include <LibWasm/Types.h> namespace Wasm { -ParseError with_eof_check(InputStream const& stream, ParseError error_if_not_eof) +ParseError with_eof_check(Core::Stream::Stream const& stream, ParseError error_if_not_eof) { - if (stream.unreliable_eof()) + if (stream.is_eof()) return ParseError::UnexpectedEof; return error_if_not_eof; } template<typename T> -static auto parse_vector(InputStream& stream) +static auto parse_vector(Core::Stream::Stream& stream) { ScopeLogger<WASM_BINPARSER_DEBUG> logger; if constexpr (requires { T::parse(stream); }) { using ResultT = typename decltype(T::parse(stream))::ValueType; size_t count; - if (!LEB128::read_unsigned(stream, count)) + Core::Stream::WrapInAKInputStream wrapped_stream { stream }; + if (!LEB128::read_unsigned(wrapped_stream, count)) return ParseResult<Vector<ResultT>> { with_eof_check(stream, ParseError::ExpectedSize) }; Vector<ResultT> entries; @@ -39,26 +41,29 @@ static auto parse_vector(InputStream& stream) return ParseResult<Vector<ResultT>> { move(entries) }; } else { size_t count; - if (!LEB128::read_unsigned(stream, count)) + Core::Stream::WrapInAKInputStream wrapped_stream { stream }; + if (!LEB128::read_unsigned(wrapped_stream, count)) return ParseResult<Vector<T>> { with_eof_check(stream, ParseError::ExpectedSize) }; Vector<T> entries; for (size_t i = 0; i < count; ++i) { if constexpr (IsSame<T, size_t>) { size_t value; - if (!LEB128::read_unsigned(stream, value)) + Core::Stream::WrapInAKInputStream wrapped_stream { stream }; + if (!LEB128::read_unsigned(wrapped_stream, value)) return ParseResult<Vector<T>> { with_eof_check(stream, ParseError::ExpectedSize) }; entries.append(value); } else if constexpr (IsSame<T, ssize_t>) { ssize_t value; - if (!LEB128::read_signed(stream, value)) + Core::Stream::WrapInAKInputStream wrapped_stream { stream }; + if (!LEB128::read_signed(wrapped_stream, value)) return ParseResult<Vector<T>> { with_eof_check(stream, ParseError::ExpectedSize) }; entries.append(value); } else if constexpr (IsSame<T, u8>) { if (count > Constants::max_allowed_vector_size) return ParseResult<Vector<T>> { ParseError::HugeAllocationRequested }; entries.resize(count); - if (!stream.read_or_error({ entries.data(), entries.size() })) + if (stream.read_entire_buffer({ entries.data(), entries.size() }).is_error()) return ParseResult<Vector<T>> { with_eof_check(stream, ParseError::InvalidInput) }; break; // Note: We read this all in one go! } @@ -67,7 +72,7 @@ static auto parse_vector(InputStream& stream) } } -static ParseResult<DeprecatedString> parse_name(InputStream& stream) +static ParseResult<DeprecatedString> parse_name(Core::Stream::Stream& stream) { ScopeLogger<WASM_BINPARSER_DEBUG> logger; auto data = parse_vector<u8>(stream); @@ -83,24 +88,20 @@ struct ParseUntilAnyOfResult { Vector<T> values; }; template<typename T, u8... terminators, typename... Args> -static ParseResult<ParseUntilAnyOfResult<T>> parse_until_any_of(InputStream& stream, Args&... args) -requires(requires(InputStream& stream, Args... args) { T::parse(stream, args...); }) +static ParseResult<ParseUntilAnyOfResult<T>> parse_until_any_of(Core::Stream::Stream& stream, Args&... args) +requires(requires(Core::Stream::Stream& stream, Args... args) { T::parse(stream, args...); }) { ScopeLogger<WASM_BINPARSER_DEBUG> logger; ReconsumableStream new_stream { stream }; - ScopeGuard drain_errors { - [&] { - new_stream.handle_any_error(); - } - }; ParseUntilAnyOfResult<T> result; for (;;) { - u8 byte; - new_stream >> byte; - if (new_stream.has_any_error()) + auto byte_or_error = new_stream.read_value<u8>(); + if (byte_or_error.is_error()) return with_eof_check(stream, ParseError::ExpectedValueOrTerminator); + auto byte = byte_or_error.release_value(); + constexpr auto equals = [](auto&& a, auto&& b) { return a == b; }; if ((... || equals(byte, terminators))) { @@ -117,13 +118,15 @@ requires(requires(InputStream& stream, Args... args) { T::parse(stream, args...) } } -ParseResult<ValueType> ValueType::parse(InputStream& stream) +ParseResult<ValueType> ValueType::parse(Core::Stream::Stream& stream) { ScopeLogger<WASM_BINPARSER_DEBUG> logger("ValueType"sv); - u8 tag; - stream >> tag; - if (stream.has_any_error()) + auto tag_or_error = stream.read_value<u8>(); + if (tag_or_error.is_error()) return with_eof_check(stream, ParseError::ExpectedKindTag); + + auto tag = tag_or_error.release_value(); + switch (tag) { case Constants::i32_tag: return ValueType(I32); @@ -142,7 +145,7 @@ ParseResult<ValueType> ValueType::parse(InputStream& stream) } } -ParseResult<ResultType> ResultType::parse(InputStream& stream) +ParseResult<ResultType> ResultType::parse(Core::Stream::Stream& stream) { ScopeLogger<WASM_BINPARSER_DEBUG> logger("ResultType"sv); auto types = parse_vector<ValueType>(stream); @@ -151,14 +154,15 @@ ParseResult<ResultType> ResultType::parse(InputStream& stream) return ResultType { types.release_value() }; } -ParseResult<FunctionType> FunctionType::parse(InputStream& stream) +ParseResult<FunctionType> FunctionType::parse(Core::Stream::Stream& stream) { ScopeLogger<WASM_BINPARSER_DEBUG> logger("FunctionType"sv); - u8 tag; - stream >> tag; - if (stream.has_any_error()) + auto tag_or_error = stream.read_value<u8>(); + if (tag_or_error.is_error()) return with_eof_check(stream, ParseError::ExpectedKindTag); + auto tag = tag_or_error.release_value(); + if (tag != Constants::function_signature_tag) { dbgln("Expected 0x60, but found {:#x}", tag); return with_eof_check(stream, ParseError::InvalidTag); @@ -174,25 +178,28 @@ ParseResult<FunctionType> FunctionType::parse(InputStream& stream) return FunctionType { parameters_result.release_value(), results_result.release_value() }; } -ParseResult<Limits> Limits::parse(InputStream& stream) +ParseResult<Limits> Limits::parse(Core::Stream::Stream& stream) { ScopeLogger<WASM_BINPARSER_DEBUG> logger("Limits"sv); - u8 flag; - stream >> flag; - if (stream.has_any_error()) + auto flag_or_error = stream.read_value<u8>(); + if (flag_or_error.is_error()) return with_eof_check(stream, ParseError::ExpectedKindTag); + auto flag = flag_or_error.release_value(); + if (flag > 1) return with_eof_check(stream, ParseError::InvalidTag); size_t min; - if (!LEB128::read_unsigned(stream, min)) + Core::Stream::WrapInAKInputStream wrapped_stream { stream }; + if (!LEB128::read_unsigned(wrapped_stream, min)) return with_eof_check(stream, ParseError::ExpectedSize); Optional<u32> max; if (flag) { size_t value; - if (!LEB128::read_unsigned(stream, value)) + Core::Stream::WrapInAKInputStream wrapped_stream { stream }; + if (!LEB128::read_unsigned(wrapped_stream, value)) return with_eof_check(stream, ParseError::ExpectedSize); max = value; } @@ -200,7 +207,7 @@ ParseResult<Limits> Limits::parse(InputStream& stream) return Limits { static_cast<u32>(min), move(max) }; } -ParseResult<MemoryType> MemoryType::parse(InputStream& stream) +ParseResult<MemoryType> MemoryType::parse(Core::Stream::Stream& stream) { ScopeLogger<WASM_BINPARSER_DEBUG> logger("MemoryType"sv); auto limits_result = Limits::parse(stream); @@ -209,7 +216,7 @@ ParseResult<MemoryType> MemoryType::parse(InputStream& stream) return MemoryType { limits_result.release_value() }; } -ParseResult<TableType> TableType::parse(InputStream& stream) +ParseResult<TableType> TableType::parse(Core::Stream::Stream& stream) { ScopeLogger<WASM_BINPARSER_DEBUG> logger("TableType"sv); auto type_result = ValueType::parse(stream); @@ -223,50 +230,48 @@ ParseResult<TableType> TableType::parse(InputStream& stream) return TableType { type_result.release_value(), limits_result.release_value() }; } -ParseResult<GlobalType> GlobalType::parse(InputStream& stream) +ParseResult<GlobalType> GlobalType::parse(Core::Stream::Stream& stream) { ScopeLogger<WASM_BINPARSER_DEBUG> logger("GlobalType"sv); auto type_result = ValueType::parse(stream); if (type_result.is_error()) return type_result.error(); - u8 mutable_; - stream >> mutable_; - if (stream.has_any_error()) + auto mutable_or_error = stream.read_value<u8>(); + if (mutable_or_error.is_error()) return with_eof_check(stream, ParseError::ExpectedKindTag); + auto mutable_ = mutable_or_error.release_value(); + if (mutable_ > 1) return with_eof_check(stream, ParseError::InvalidTag); return GlobalType { type_result.release_value(), mutable_ == 0x01 }; } -ParseResult<BlockType> BlockType::parse(InputStream& stream) +ParseResult<BlockType> BlockType::parse(Core::Stream::Stream& stream) { ScopeLogger<WASM_BINPARSER_DEBUG> logger("BlockType"sv); - u8 kind; - stream >> kind; - if (stream.has_any_error()) + auto kind_or_error = stream.read_value<u8>(); + if (kind_or_error.is_error()) return with_eof_check(stream, ParseError::ExpectedKindTag); + + auto kind = kind_or_error.release_value(); if (kind == Constants::empty_block_tag) return BlockType {}; { - InputMemoryStream value_stream { ReadonlyBytes { &kind, 1 } }; - if (auto value_type = ValueType::parse(value_stream); !value_type.is_error()) + auto value_stream = Core::Stream::FixedMemoryStream::construct(ReadonlyBytes { &kind, 1 }).release_value_but_fixme_should_propagate_errors(); + if (auto value_type = ValueType::parse(*value_stream); !value_type.is_error()) return BlockType { value_type.release_value() }; } ReconsumableStream new_stream { stream }; new_stream.unread({ &kind, 1 }); - ScopeGuard drain_errors { - [&] { - new_stream.handle_any_error(); - } - }; ssize_t index_value; - if (!LEB128::read_signed(new_stream, index_value)) + Core::Stream::WrapInAKInputStream wrapped_new_stream { new_stream }; + if (!LEB128::read_signed(wrapped_new_stream, index_value)) return with_eof_check(stream, ParseError::ExpectedIndex); if (index_value < 0) { @@ -277,7 +282,7 @@ ParseResult<BlockType> BlockType::parse(InputStream& stream) return BlockType { TypeIndex(index_value) }; } -ParseResult<Vector<Instruction>> Instruction::parse(InputStream& stream, InstructionPointer& ip) +ParseResult<Vector<Instruction>> Instruction::parse(Core::Stream::Stream& stream, InstructionPointer& ip) { struct NestedInstructionState { Vector<Instruction> prior_instructions; @@ -291,11 +296,12 @@ ParseResult<Vector<Instruction>> Instruction::parse(InputStream& stream, Instruc do { ScopeLogger<WASM_BINPARSER_DEBUG> logger("Instruction"sv); - u8 byte; - stream >> byte; - if (stream.has_any_error()) + auto byte_or_error = stream.read_value<u8>(); + if (byte_or_error.is_error()) return with_eof_check(stream, ParseError::ExpectedKindTag); + auto byte = byte_or_error.release_value(); + if (!nested_instructions.is_empty()) { auto& nested_structure = nested_instructions.last(); if (byte == 0x0b) { @@ -408,13 +414,15 @@ ParseResult<Vector<Instruction>> Instruction::parse(InputStream& stream, Instruc case Instructions::i64_store8.value(): case Instructions::i64_store16.value(): case Instructions::i64_store32.value(): { + Core::Stream::WrapInAKInputStream wrapped_stream { stream }; + // op (align offset) size_t align; - if (!LEB128::read_unsigned(stream, align)) + if (!LEB128::read_unsigned(wrapped_stream, align)) return with_eof_check(stream, ParseError::InvalidInput); size_t offset; - if (!LEB128::read_unsigned(stream, offset)) + if (!LEB128::read_unsigned(wrapped_stream, offset)) return with_eof_check(stream, ParseError::InvalidInput); resulting_instructions.append(Instruction { opcode, MemoryArgument { static_cast<u32>(align), static_cast<u32>(offset) } }); @@ -443,10 +451,11 @@ ParseResult<Vector<Instruction>> Instruction::parse(InputStream& stream, Instruc case Instructions::memory_grow.value(): { // op 0x0 // The zero is currently unused. - u8 unused; - stream >> unused; - if (stream.has_any_error()) + auto unused_or_error = stream.read_value<u8>(); + if (unused_or_error.is_error()) return with_eof_check(stream, ParseError::ExpectedKindTag); + + auto unused = unused_or_error.release_value(); if (unused != 0x00) { dbgln("Invalid tag in memory_grow {}", unused); return with_eof_check(stream, ParseError::InvalidTag); @@ -457,7 +466,8 @@ ParseResult<Vector<Instruction>> Instruction::parse(InputStream& stream, Instruc } case Instructions::i32_const.value(): { i32 value; - if (!LEB128::read_signed(stream, value)) + Core::Stream::WrapInAKInputStream wrapped_stream { stream }; + if (!LEB128::read_signed(wrapped_stream, value)) return with_eof_check(stream, ParseError::ExpectedSignedImmediate); resulting_instructions.append(Instruction { opcode, value }); @@ -466,7 +476,8 @@ ParseResult<Vector<Instruction>> Instruction::parse(InputStream& stream, Instruc case Instructions::i64_const.value(): { // op literal i64 value; - if (!LEB128::read_signed(stream, value)) + Core::Stream::WrapInAKInputStream wrapped_stream { stream }; + if (!LEB128::read_signed(wrapped_stream, value)) return with_eof_check(stream, ParseError::ExpectedSignedImmediate); resulting_instructions.append(Instruction { opcode, value }); @@ -475,8 +486,7 @@ ParseResult<Vector<Instruction>> Instruction::parse(InputStream& stream, Instruc case Instructions::f32_const.value(): { // op literal LittleEndian<u32> value; - stream >> value; - if (stream.has_any_error()) + if (stream.read_entire_buffer(value.bytes()).is_error()) return with_eof_check(stream, ParseError::ExpectedFloatingImmediate); auto floating = bit_cast<float>(static_cast<u32>(value)); @@ -486,8 +496,7 @@ ParseResult<Vector<Instruction>> Instruction::parse(InputStream& stream, Instruc case Instructions::f64_const.value(): { // op literal LittleEndian<u64> value; - stream >> value; - if (stream.has_any_error()) + if (stream.read_entire_buffer(value.bytes()).is_error()) return with_eof_check(stream, ParseError::ExpectedFloatingImmediate); auto floating = bit_cast<double>(static_cast<u64>(value)); @@ -668,7 +677,8 @@ ParseResult<Vector<Instruction>> Instruction::parse(InputStream& stream, Instruc case 0xfc: { // These are multibyte instructions. u32 selector; - if (!LEB128::read_unsigned(stream, selector)) + Core::Stream::WrapInAKInputStream wrapped_stream { stream }; + if (!LEB128::read_unsigned(wrapped_stream, selector)) return with_eof_check(stream, ParseError::InvalidInput); switch (selector) { case Instructions::i32_trunc_sat_f32_s_second: @@ -685,10 +695,11 @@ ParseResult<Vector<Instruction>> Instruction::parse(InputStream& stream, Instruc auto index = GenericIndexParser<DataIndex>::parse(stream); if (index.is_error()) return index.error(); - u8 unused; - stream >> unused; - if (stream.has_any_error()) + auto unused_or_error = stream.read_value<u8>(); + if (unused_or_error.is_error()) return with_eof_check(stream, ParseError::InvalidInput); + + auto unused = unused_or_error.release_value(); if (unused != 0x00) return ParseError::InvalidImmediate; resulting_instructions.append(Instruction { OpCode { 0xfc00 | selector }, index.release_value() }); @@ -703,10 +714,11 @@ ParseResult<Vector<Instruction>> Instruction::parse(InputStream& stream, Instruc } case Instructions::memory_copy_second: { for (size_t i = 0; i < 2; ++i) { - u8 unused; - stream >> unused; - if (stream.has_any_error()) + auto unused_or_error = stream.read_value<u8>(); + if (unused_or_error.is_error()) return with_eof_check(stream, ParseError::InvalidInput); + + auto unused = unused_or_error.release_value(); if (unused != 0x00) return ParseError::InvalidImmediate; } @@ -714,10 +726,11 @@ ParseResult<Vector<Instruction>> Instruction::parse(InputStream& stream, Instruc break; } case Instructions::memory_fill_second: { - u8 unused; - stream >> unused; - if (stream.has_any_error()) + auto unused_or_error = stream.read_value<u8>(); + if (unused_or_error.is_error()) return with_eof_check(stream, ParseError::InvalidInput); + + auto unused = unused_or_error.release_value(); if (unused != 0x00) return ParseError::InvalidImmediate; resulting_instructions.append(Instruction { OpCode { 0xfc00 | selector } }); @@ -769,7 +782,7 @@ ParseResult<Vector<Instruction>> Instruction::parse(InputStream& stream, Instruc return resulting_instructions; } -ParseResult<CustomSection> CustomSection::parse(InputStream& stream) +ParseResult<CustomSection> CustomSection::parse(Core::Stream::Stream& stream) { ScopeLogger<WASM_BINPARSER_DEBUG> logger("CustomSection"sv); auto name = parse_name(stream); @@ -780,9 +793,12 @@ ParseResult<CustomSection> CustomSection::parse(InputStream& stream) if (data_buffer.try_resize(64).is_error()) return ParseError::OutOfMemory; - while (!stream.has_any_error() && !stream.unreliable_eof()) { + while (!stream.is_eof()) { char buf[16]; - auto size = stream.read({ buf, 16 }); + auto span_or_error = stream.read({ buf, 16 }); + if (span_or_error.is_error()) + break; + auto size = span_or_error.release_value().size(); if (size == 0) break; if (data_buffer.try_append(buf, size).is_error()) @@ -792,7 +808,7 @@ ParseResult<CustomSection> CustomSection::parse(InputStream& stream) return CustomSection(name.release_value(), move(data_buffer)); } -ParseResult<TypeSection> TypeSection::parse(InputStream& stream) +ParseResult<TypeSection> TypeSection::parse(Core::Stream::Stream& stream) { ScopeLogger<WASM_BINPARSER_DEBUG> logger("TypeSection"sv); auto types = parse_vector<FunctionType>(stream); @@ -801,7 +817,7 @@ ParseResult<TypeSection> TypeSection::parse(InputStream& stream) return TypeSection { types.release_value() }; } -ParseResult<ImportSection::Import> ImportSection::Import::parse(InputStream& stream) +ParseResult<ImportSection::Import> ImportSection::Import::parse(Core::Stream::Stream& stream) { ScopeLogger<WASM_BINPARSER_DEBUG> logger("Import"sv); auto module = parse_name(stream); @@ -810,11 +826,12 @@ ParseResult<ImportSection::Import> ImportSection::Import::parse(InputStream& str auto name = parse_name(stream); if (name.is_error()) return name.error(); - u8 tag; - stream >> tag; - if (stream.has_any_error()) + auto tag_or_error = stream.read_value<u8>(); + if (tag_or_error.is_error()) return with_eof_check(stream, ParseError::ExpectedKindTag); + auto tag = tag_or_error.release_value(); + switch (tag) { case Constants::extern_function_tag: { auto index = GenericIndexParser<TypeIndex>::parse(stream); @@ -833,7 +850,7 @@ ParseResult<ImportSection::Import> ImportSection::Import::parse(InputStream& str } } -ParseResult<ImportSection> ImportSection::parse(InputStream& stream) +ParseResult<ImportSection> ImportSection::parse(Core::Stream::Stream& stream) { ScopeLogger<WASM_BINPARSER_DEBUG> logger("ImportSection"sv); auto imports = parse_vector<Import>(stream); @@ -842,7 +859,7 @@ ParseResult<ImportSection> ImportSection::parse(InputStream& stream) return ImportSection { imports.release_value() }; } -ParseResult<FunctionSection> FunctionSection::parse(InputStream& stream) +ParseResult<FunctionSection> FunctionSection::parse(Core::Stream::Stream& stream) { ScopeLogger<WASM_BINPARSER_DEBUG> logger("FunctionSection"sv); auto indices = parse_vector<size_t>(stream); @@ -857,7 +874,7 @@ ParseResult<FunctionSection> FunctionSection::parse(InputStream& stream) return FunctionSection { move(typed_indices) }; } -ParseResult<TableSection::Table> TableSection::Table::parse(InputStream& stream) +ParseResult<TableSection::Table> TableSection::Table::parse(Core::Stream::Stream& stream) { ScopeLogger<WASM_BINPARSER_DEBUG> logger("Table"sv); auto type = TableType::parse(stream); @@ -866,7 +883,7 @@ ParseResult<TableSection::Table> TableSection::Table::parse(InputStream& stream) return Table { type.release_value() }; } -ParseResult<TableSection> TableSection::parse(InputStream& stream) +ParseResult<TableSection> TableSection::parse(Core::Stream::Stream& stream) { ScopeLogger<WASM_BINPARSER_DEBUG> logger("TableSection"sv); auto tables = parse_vector<Table>(stream); @@ -875,7 +892,7 @@ ParseResult<TableSection> TableSection::parse(InputStream& stream) return TableSection { tables.release_value() }; } -ParseResult<MemorySection::Memory> MemorySection::Memory::parse(InputStream& stream) +ParseResult<MemorySection::Memory> MemorySection::Memory::parse(Core::Stream::Stream& stream) { ScopeLogger<WASM_BINPARSER_DEBUG> logger("Memory"sv); auto type = MemoryType::parse(stream); @@ -884,7 +901,7 @@ ParseResult<MemorySection::Memory> MemorySection::Memory::parse(InputStream& str return Memory { type.release_value() }; } -ParseResult<MemorySection> MemorySection::parse(InputStream& stream) +ParseResult<MemorySection> MemorySection::parse(Core::Stream::Stream& stream) { ScopeLogger<WASM_BINPARSER_DEBUG> logger("MemorySection"sv); auto memories = parse_vector<Memory>(stream); @@ -893,7 +910,7 @@ ParseResult<MemorySection> MemorySection::parse(InputStream& stream) return MemorySection { memories.release_value() }; } -ParseResult<Expression> Expression::parse(InputStream& stream) +ParseResult<Expression> Expression::parse(Core::Stream::Stream& stream) { ScopeLogger<WASM_BINPARSER_DEBUG> logger("Expression"sv); InstructionPointer ip { 0 }; @@ -904,7 +921,7 @@ ParseResult<Expression> Expression::parse(InputStream& stream) return Expression { move(instructions.value().values) }; } -ParseResult<GlobalSection::Global> GlobalSection::Global::parse(InputStream& stream) +ParseResult<GlobalSection::Global> GlobalSection::Global::parse(Core::Stream::Stream& stream) { ScopeLogger<WASM_BINPARSER_DEBUG> logger("Global"sv); auto type = GlobalType::parse(stream); @@ -916,7 +933,7 @@ ParseResult<GlobalSection::Global> GlobalSection::Global::parse(InputStream& str return Global { type.release_value(), exprs.release_value() }; } -ParseResult<GlobalSection> GlobalSection::parse(InputStream& stream) +ParseResult<GlobalSection> GlobalSection::parse(Core::Stream::Stream& stream) { ScopeLogger<WASM_BINPARSER_DEBUG> logger("GlobalSection"sv); auto result = parse_vector<Global>(stream); @@ -925,19 +942,21 @@ ParseResult<GlobalSection> GlobalSection::parse(InputStream& stream) return GlobalSection { result.release_value() }; } -ParseResult<ExportSection::Export> ExportSection::Export::parse(InputStream& stream) +ParseResult<ExportSection::Export> ExportSection::Export::parse(Core::Stream::Stream& stream) { ScopeLogger<WASM_BINPARSER_DEBUG> logger("Export"sv); auto name = parse_name(stream); if (name.is_error()) return name.error(); - u8 tag; - stream >> tag; - if (stream.has_any_error()) + auto tag_or_error = stream.read_value<u8>(); + if (tag_or_error.is_error()) return with_eof_check(stream, ParseError::ExpectedKindTag); + auto tag = tag_or_error.release_value(); + size_t index; - if (!LEB128::read_unsigned(stream, index)) + Core::Stream::WrapInAKInputStream wrapped_stream { stream }; + if (!LEB128::read_unsigned(wrapped_stream, index)) return with_eof_check(stream, ParseError::ExpectedIndex); switch (tag) { @@ -954,7 +973,7 @@ ParseResult<ExportSection::Export> ExportSection::Export::parse(InputStream& str } } -ParseResult<ExportSection> ExportSection::parse(InputStream& stream) +ParseResult<ExportSection> ExportSection::parse(Core::Stream::Stream& stream) { ScopeLogger<WASM_BINPARSER_DEBUG> logger("ExportSection"sv); auto result = parse_vector<Export>(stream); @@ -963,7 +982,7 @@ ParseResult<ExportSection> ExportSection::parse(InputStream& stream) return ExportSection { result.release_value() }; } -ParseResult<StartSection::StartFunction> StartSection::StartFunction::parse(InputStream& stream) +ParseResult<StartSection::StartFunction> StartSection::StartFunction::parse(Core::Stream::Stream& stream) { ScopeLogger<WASM_BINPARSER_DEBUG> logger("StartFunction"sv); auto index = GenericIndexParser<FunctionIndex>::parse(stream); @@ -972,7 +991,7 @@ ParseResult<StartSection::StartFunction> StartSection::StartFunction::parse(Inpu return StartFunction { index.release_value() }; } -ParseResult<StartSection> StartSection::parse(InputStream& stream) +ParseResult<StartSection> StartSection::parse(Core::Stream::Stream& stream) { ScopeLogger<WASM_BINPARSER_DEBUG> logger("StartSection"sv); auto result = StartFunction::parse(stream); @@ -981,7 +1000,7 @@ ParseResult<StartSection> StartSection::parse(InputStream& stream) return StartSection { result.release_value() }; } -ParseResult<ElementSection::SegmentType0> ElementSection::SegmentType0::parse(InputStream& stream) +ParseResult<ElementSection::SegmentType0> ElementSection::SegmentType0::parse(Core::Stream::Stream& stream) { auto expression = Expression::parse(stream); if (expression.is_error()) @@ -993,12 +1012,13 @@ ParseResult<ElementSection::SegmentType0> ElementSection::SegmentType0::parse(In return SegmentType0 { indices.release_value(), Active { 0, expression.release_value() } }; } -ParseResult<ElementSection::SegmentType1> ElementSection::SegmentType1::parse(InputStream& stream) +ParseResult<ElementSection::SegmentType1> ElementSection::SegmentType1::parse(Core::Stream::Stream& stream) { - u8 kind; - stream >> kind; - if (stream.has_any_error()) + auto kind_or_error = stream.read_value<u8>(); + if (kind_or_error.is_error()) return with_eof_check(stream, ParseError::ExpectedKindTag); + + auto kind = kind_or_error.release_value(); if (kind != 0) return ParseError::InvalidTag; auto indices = parse_vector<GenericIndexParser<FunctionIndex>>(stream); @@ -1008,56 +1028,57 @@ ParseResult<ElementSection::SegmentType1> ElementSection::SegmentType1::parse(In return SegmentType1 { indices.release_value() }; } -ParseResult<ElementSection::SegmentType2> ElementSection::SegmentType2::parse(InputStream& stream) +ParseResult<ElementSection::SegmentType2> ElementSection::SegmentType2::parse(Core::Stream::Stream& stream) { dbgln("Type 2"); (void)stream; return ParseError::NotImplemented; } -ParseResult<ElementSection::SegmentType3> ElementSection::SegmentType3::parse(InputStream& stream) +ParseResult<ElementSection::SegmentType3> ElementSection::SegmentType3::parse(Core::Stream::Stream& stream) { dbgln("Type 3"); (void)stream; return ParseError::NotImplemented; } -ParseResult<ElementSection::SegmentType4> ElementSection::SegmentType4::parse(InputStream& stream) +ParseResult<ElementSection::SegmentType4> ElementSection::SegmentType4::parse(Core::Stream::Stream& stream) { dbgln("Type 4"); (void)stream; return ParseError::NotImplemented; } -ParseResult<ElementSection::SegmentType5> ElementSection::SegmentType5::parse(InputStream& stream) +ParseResult<ElementSection::SegmentType5> ElementSection::SegmentType5::parse(Core::Stream::Stream& stream) { dbgln("Type 5"); (void)stream; return ParseError::NotImplemented; } -ParseResult<ElementSection::SegmentType6> ElementSection::SegmentType6::parse(InputStream& stream) +ParseResult<ElementSection::SegmentType6> ElementSection::SegmentType6::parse(Core::Stream::Stream& stream) { dbgln("Type 6"); (void)stream; return ParseError::NotImplemented; } -ParseResult<ElementSection::SegmentType7> ElementSection::SegmentType7::parse(InputStream& stream) +ParseResult<ElementSection::SegmentType7> ElementSection::SegmentType7::parse(Core::Stream::Stream& stream) { dbgln("Type 7"); (void)stream; return ParseError::NotImplemented; } -ParseResult<ElementSection::Element> ElementSection::Element::parse(InputStream& stream) +ParseResult<ElementSection::Element> ElementSection::Element::parse(Core::Stream::Stream& stream) { ScopeLogger<WASM_BINPARSER_DEBUG> logger("Element"sv); - u8 tag; - stream >> tag; - if (stream.has_any_error()) + auto tag_or_error = stream.read_value<u8>(); + if (tag_or_error.is_error()) return with_eof_check(stream, ParseError::ExpectedKindTag); + auto tag = tag_or_error.release_value(); + switch (tag) { case 0x00: if (auto result = SegmentType0::parse(stream); result.is_error()) { @@ -1118,7 +1139,7 @@ ParseResult<ElementSection::Element> ElementSection::Element::parse(InputStream& } } -ParseResult<ElementSection> ElementSection::parse(InputStream& stream) +ParseResult<ElementSection> ElementSection::parse(Core::Stream::Stream& stream) { ScopeLogger<WASM_BINPARSER_DEBUG> logger("ElementSection"sv); auto result = parse_vector<Element>(stream); @@ -1127,11 +1148,12 @@ ParseResult<ElementSection> ElementSection::parse(InputStream& stream) return ElementSection { result.release_value() }; } -ParseResult<Locals> Locals::parse(InputStream& stream) +ParseResult<Locals> Locals::parse(Core::Stream::Stream& stream) { ScopeLogger<WASM_BINPARSER_DEBUG> logger("Locals"sv); size_t count; - if (!LEB128::read_unsigned(stream, count)) + Core::Stream::WrapInAKInputStream wrapped_stream { stream }; + if (!LEB128::read_unsigned(wrapped_stream, count)) return with_eof_check(stream, ParseError::InvalidSize); if (count > Constants::max_allowed_function_locals_per_type) @@ -1144,7 +1166,7 @@ ParseResult<Locals> Locals::parse(InputStream& stream) return Locals { static_cast<u32>(count), type.release_value() }; } -ParseResult<CodeSection::Func> CodeSection::Func::parse(InputStream& stream) +ParseResult<CodeSection::Func> CodeSection::Func::parse(Core::Stream::Stream& stream) { ScopeLogger<WASM_BINPARSER_DEBUG> logger("Func"sv); auto locals = parse_vector<Locals>(stream); @@ -1156,19 +1178,15 @@ ParseResult<CodeSection::Func> CodeSection::Func::parse(InputStream& stream) return Func { locals.release_value(), body.release_value() }; } -ParseResult<CodeSection::Code> CodeSection::Code::parse(InputStream& stream) +ParseResult<CodeSection::Code> CodeSection::Code::parse(Core::Stream::Stream& stream) { ScopeLogger<WASM_BINPARSER_DEBUG> logger("Code"sv); size_t size; - if (!LEB128::read_unsigned(stream, size)) + Core::Stream::WrapInAKInputStream wrapped_stream { stream }; + if (!LEB128::read_unsigned(wrapped_stream, size)) return with_eof_check(stream, ParseError::InvalidSize); auto constrained_stream = ConstrainedStream { stream, size }; - ScopeGuard drain_errors { - [&] { - constrained_stream.handle_any_error(); - } - }; auto func = Func::parse(constrained_stream); if (func.is_error()) @@ -1177,7 +1195,7 @@ ParseResult<CodeSection::Code> CodeSection::Code::parse(InputStream& stream) return Code { static_cast<u32>(size), func.release_value() }; } -ParseResult<CodeSection> CodeSection::parse(InputStream& stream) +ParseResult<CodeSection> CodeSection::parse(Core::Stream::Stream& stream) { ScopeLogger<WASM_BINPARSER_DEBUG> logger("CodeSection"sv); auto result = parse_vector<Code>(stream); @@ -1186,14 +1204,15 @@ ParseResult<CodeSection> CodeSection::parse(InputStream& stream) return CodeSection { result.release_value() }; } -ParseResult<DataSection::Data> DataSection::Data::parse(InputStream& stream) +ParseResult<DataSection::Data> DataSection::Data::parse(Core::Stream::Stream& stream) { ScopeLogger<WASM_BINPARSER_DEBUG> logger("Data"sv); - u8 tag; - stream >> tag; - if (stream.has_any_error()) + auto tag_or_error = stream.read_value<u8>(); + if (tag_or_error.is_error()) return with_eof_check(stream, ParseError::ExpectedKindTag); + auto tag = tag_or_error.release_value(); + if (tag > 0x02) return with_eof_check(stream, ParseError::InvalidTag); @@ -1227,7 +1246,7 @@ ParseResult<DataSection::Data> DataSection::Data::parse(InputStream& stream) VERIFY_NOT_REACHED(); } -ParseResult<DataSection> DataSection::parse(InputStream& stream) +ParseResult<DataSection> DataSection::parse(Core::Stream::Stream& stream) { ScopeLogger<WASM_BINPARSER_DEBUG> logger("DataSection"sv); auto data = parse_vector<Data>(stream); @@ -1237,12 +1256,13 @@ ParseResult<DataSection> DataSection::parse(InputStream& stream) return DataSection { data.release_value() }; } -ParseResult<DataCountSection> DataCountSection::parse([[maybe_unused]] InputStream& stream) +ParseResult<DataCountSection> DataCountSection::parse([[maybe_unused]] Core::Stream::Stream& stream) { ScopeLogger<WASM_BINPARSER_DEBUG> logger("DataCountSection"sv); u32 value; - if (!LEB128::read_unsigned(stream, value)) { - if (stream.unreliable_eof()) { + Core::Stream::WrapInAKInputStream wrapped_stream { stream }; + if (!LEB128::read_unsigned(wrapped_stream, value)) { + if (stream.is_eof()) { // The section simply didn't contain anything. return DataCountSection { {} }; } @@ -1252,41 +1272,36 @@ ParseResult<DataCountSection> DataCountSection::parse([[maybe_unused]] InputStre return DataCountSection { value }; } -ParseResult<Module> Module::parse(InputStream& stream) +ParseResult<Module> Module::parse(Core::Stream::Stream& stream) { ScopeLogger<WASM_BINPARSER_DEBUG> logger("Module"sv); u8 buf[4]; - if (!stream.read_or_error({ buf, 4 })) + if (stream.read_entire_buffer({ buf, 4 }).is_error()) return with_eof_check(stream, ParseError::InvalidInput); if (Bytes { buf, 4 } != wasm_magic.span()) return with_eof_check(stream, ParseError::InvalidModuleMagic); - if (!stream.read_or_error({ buf, 4 })) + if (stream.read_entire_buffer({ buf, 4 }).is_error()) return with_eof_check(stream, ParseError::InvalidInput); if (Bytes { buf, 4 } != wasm_version.span()) return with_eof_check(stream, ParseError::InvalidModuleVersion); Vector<AnySection> sections; for (;;) { - u8 section_id; - stream >> section_id; - if (stream.unreliable_eof()) { - stream.handle_any_error(); + auto section_id_or_error = stream.read_value<u8>(); + if (stream.is_eof()) break; - } - if (stream.has_any_error()) + if (section_id_or_error.is_error()) return with_eof_check(stream, ParseError::ExpectedIndex); + auto section_id = section_id_or_error.release_value(); + size_t section_size; - if (!LEB128::read_unsigned(stream, section_size)) + Core::Stream::WrapInAKInputStream wrapped_stream { stream }; + if (!LEB128::read_unsigned(wrapped_stream, section_size)) return with_eof_check(stream, ParseError::ExpectedSize); auto section_stream = ConstrainedStream { stream, section_size }; - ScopeGuard drain_errors { - [&] { - section_stream.handle_any_error(); - } - }; switch (section_id) { case CustomSection::section_id: diff --git a/Userland/Libraries/LibWasm/Types.h b/Userland/Libraries/LibWasm/Types.h index 95a7b1526d..f221e84750 100644 --- a/Userland/Libraries/LibWasm/Types.h +++ b/Userland/Libraries/LibWasm/Types.h @@ -13,6 +13,7 @@ #include <AK/NonnullOwnPtrVector.h> #include <AK/Result.h> #include <AK/Variant.h> +#include <LibCore/Stream.h> #include <LibWasm/Constants.h> #include <LibWasm/Forward.h> #include <LibWasm/Opcode.h> @@ -58,22 +59,23 @@ AK_TYPEDEF_DISTINCT_ORDERED_ID(size_t, LabelIndex); AK_TYPEDEF_DISTINCT_ORDERED_ID(size_t, DataIndex); AK_TYPEDEF_DISTINCT_NUMERIC_GENERAL(u64, InstructionPointer, Arithmetic, Comparison, Flags, Increment); -ParseError with_eof_check(InputStream const& stream, ParseError error_if_not_eof); +ParseError with_eof_check(Core::Stream::Stream const& stream, ParseError error_if_not_eof); template<typename T> struct GenericIndexParser { - static ParseResult<T> parse(InputStream& stream) + static ParseResult<T> parse(Core::Stream::Stream& stream) { size_t value; - if (!LEB128::read_unsigned(stream, value)) + Core::Stream::WrapInAKInputStream wrapped_stream { stream }; + if (!LEB128::read_unsigned(wrapped_stream, value)) return with_eof_check(stream, ParseError::ExpectedIndex); return T { value }; } }; -class ReconsumableStream : public InputStream { +class ReconsumableStream : public Core::Stream::Stream { public: - explicit ReconsumableStream(InputStream& stream) + explicit ReconsumableStream(Core::Stream::Stream& stream) : m_stream(stream) { } @@ -81,8 +83,10 @@ public: void unread(ReadonlyBytes data) { m_buffer.append(data.data(), data.size()); } private: - size_t read(Bytes bytes) override + virtual ErrorOr<Bytes> read(Bytes bytes) override { + auto original_bytes = bytes; + size_t bytes_read_from_buffer = 0; if (!m_buffer.is_empty()) { auto read_size = min(bytes.size(), m_buffer.size()); @@ -93,20 +97,15 @@ private: bytes_read_from_buffer = read_size; } - return m_stream.read(bytes) + bytes_read_from_buffer; - } - bool unreliable_eof() const override - { - return m_buffer.is_empty() && m_stream.unreliable_eof(); + return original_bytes.trim(TRY(m_stream.read(bytes)).size() + bytes_read_from_buffer); } - bool read_or_error(Bytes bytes) override + + virtual bool is_eof() const override { - if (read(bytes)) - return true; - set_recoverable_error(); - return false; + return m_buffer.is_empty() && m_stream.is_eof(); } - bool discard_or_error(size_t count) override + + virtual ErrorOr<void> discard(size_t count) override { size_t bytes_discarded_from_buffer = 0; if (!m_buffer.is_empty()) { @@ -116,49 +115,74 @@ private: bytes_discarded_from_buffer = read_size; } - return m_stream.discard_or_error(count - bytes_discarded_from_buffer); + return m_stream.discard(count - bytes_discarded_from_buffer); + } + + virtual ErrorOr<size_t> write(ReadonlyBytes) override + { + return Error::from_errno(EBADF); + } + + virtual bool is_open() const override + { + return m_stream.is_open(); + } + + virtual void close() override + { + m_stream.close(); } - InputStream& m_stream; + Core::Stream::Stream& m_stream; Vector<u8, 8> m_buffer; }; -class ConstrainedStream : public InputStream { +class ConstrainedStream : public Core::Stream::Stream { public: - explicit ConstrainedStream(InputStream& stream, size_t size) + explicit ConstrainedStream(Core::Stream::Stream& stream, size_t size) : m_stream(stream) , m_bytes_left(size) { } private: - size_t read(Bytes bytes) override + ErrorOr<Bytes> read(Bytes bytes) override { auto to_read = min(m_bytes_left, bytes.size()); - auto nread = m_stream.read(bytes.slice(0, to_read)); - m_bytes_left -= nread; - return nread; + auto read_bytes = TRY(m_stream.read(bytes.slice(0, to_read))); + m_bytes_left -= read_bytes.size(); + return read_bytes; + } + + bool is_eof() const override + { + return m_bytes_left == 0 || m_stream.is_eof(); + } + + ErrorOr<void> discard(size_t count) override + { + if (count > m_bytes_left) + return Error::from_string_literal("Trying to discard more bytes than allowed"); + + return m_stream.discard(count); } - bool unreliable_eof() const override + + virtual ErrorOr<size_t> write(ReadonlyBytes) override { - return m_bytes_left == 0 || m_stream.unreliable_eof(); + return Error::from_errno(EBADF); } - bool read_or_error(Bytes bytes) override + + virtual bool is_open() const override { - if (read(bytes)) - return true; - set_recoverable_error(); - return false; + return m_stream.is_open(); } - bool discard_or_error(size_t count) override + + virtual void close() override { - auto to_discard = min(m_bytes_left, count); - if (m_stream.discard_or_error(to_discard)) - m_bytes_left -= to_discard; - return to_discard; + m_stream.close(); } - InputStream& m_stream; + Core::Stream::Stream& m_stream; size_t m_bytes_left { 0 }; }; @@ -187,7 +211,7 @@ public: auto is_numeric() const { return !is_reference(); } auto kind() const { return m_kind; } - static ParseResult<ValueType> parse(InputStream& stream); + static ParseResult<ValueType> parse(Core::Stream::Stream& stream); static DeprecatedString kind_name(Kind kind) { @@ -226,7 +250,7 @@ public: auto const& types() const { return m_types; } - static ParseResult<ResultType> parse(InputStream& stream); + static ParseResult<ResultType> parse(Core::Stream::Stream& stream); private: Vector<ValueType> m_types; @@ -244,7 +268,7 @@ public: auto& parameters() const { return m_parameters; } auto& results() const { return m_results; } - static ParseResult<FunctionType> parse(InputStream& stream); + static ParseResult<FunctionType> parse(Core::Stream::Stream& stream); private: Vector<ValueType> m_parameters; @@ -263,7 +287,7 @@ public: auto min() const { return m_min; } auto& max() const { return m_max; } - static ParseResult<Limits> parse(InputStream& stream); + static ParseResult<Limits> parse(Core::Stream::Stream& stream); private: u32 m_min { 0 }; @@ -280,7 +304,7 @@ public: auto& limits() const { return m_limits; } - static ParseResult<MemoryType> parse(InputStream& stream); + static ParseResult<MemoryType> parse(Core::Stream::Stream& stream); private: Limits m_limits; @@ -299,7 +323,7 @@ public: auto& limits() const { return m_limits; } auto& element_type() const { return m_element_type; } - static ParseResult<TableType> parse(InputStream& stream); + static ParseResult<TableType> parse(Core::Stream::Stream& stream); private: ValueType m_element_type; @@ -318,7 +342,7 @@ public: auto& type() const { return m_type; } auto is_mutable() const { return m_is_mutable; } - static ParseResult<GlobalType> parse(InputStream& stream); + static ParseResult<GlobalType> parse(Core::Stream::Stream& stream); private: ValueType m_type; @@ -364,7 +388,7 @@ public: return m_type_index; } - static ParseResult<BlockType> parse(InputStream& stream); + static ParseResult<BlockType> parse(Core::Stream::Stream& stream); private: Kind m_kind { Empty }; @@ -428,7 +452,7 @@ public: { } - static ParseResult<Vector<Instruction>> parse(InputStream& stream, InstructionPointer& ip); + static ParseResult<Vector<Instruction>> parse(Core::Stream::Stream& stream, InstructionPointer& ip); auto& opcode() const { return m_opcode; } auto& arguments() const { return m_arguments; } @@ -475,7 +499,7 @@ public: auto& name() const { return m_name; } auto& contents() const { return m_contents; } - static ParseResult<CustomSection> parse(InputStream& stream); + static ParseResult<CustomSection> parse(Core::Stream::Stream& stream); private: DeprecatedString m_name; @@ -493,7 +517,7 @@ public: auto& types() const { return m_types; } - static ParseResult<TypeSection> parse(InputStream& stream); + static ParseResult<TypeSection> parse(Core::Stream::Stream& stream); private: Vector<FunctionType> m_types; @@ -515,7 +539,7 @@ public: auto& name() const { return m_name; } auto& description() const { return m_description; } - static ParseResult<Import> parse(InputStream& stream); + static ParseResult<Import> parse(Core::Stream::Stream& stream); private: template<typename T> @@ -542,7 +566,7 @@ public: auto& imports() const { return m_imports; } - static ParseResult<ImportSection> parse(InputStream& stream); + static ParseResult<ImportSection> parse(Core::Stream::Stream& stream); private: Vector<Import> m_imports; @@ -559,7 +583,7 @@ public: auto& types() const { return m_types; } - static ParseResult<FunctionSection> parse(InputStream& stream); + static ParseResult<FunctionSection> parse(Core::Stream::Stream& stream); private: Vector<TypeIndex> m_types; @@ -576,7 +600,7 @@ public: auto& type() const { return m_type; } - static ParseResult<Table> parse(InputStream& stream); + static ParseResult<Table> parse(Core::Stream::Stream& stream); private: TableType m_type; @@ -592,7 +616,7 @@ public: auto& tables() const { return m_tables; }; - static ParseResult<TableSection> parse(InputStream& stream); + static ParseResult<TableSection> parse(Core::Stream::Stream& stream); private: Vector<Table> m_tables; @@ -609,7 +633,7 @@ public: auto& type() const { return m_type; } - static ParseResult<Memory> parse(InputStream& stream); + static ParseResult<Memory> parse(Core::Stream::Stream& stream); private: MemoryType m_type; @@ -625,7 +649,7 @@ public: auto& memories() const { return m_memories; } - static ParseResult<MemorySection> parse(InputStream& stream); + static ParseResult<MemorySection> parse(Core::Stream::Stream& stream); private: Vector<Memory> m_memories; @@ -640,7 +664,7 @@ public: auto& instructions() const { return m_instructions; } - static ParseResult<Expression> parse(InputStream& stream); + static ParseResult<Expression> parse(Core::Stream::Stream& stream); private: Vector<Instruction> m_instructions; @@ -659,7 +683,7 @@ public: auto& type() const { return m_type; } auto& expression() const { return m_expression; } - static ParseResult<Global> parse(InputStream& stream); + static ParseResult<Global> parse(Core::Stream::Stream& stream); private: GlobalType m_type; @@ -676,7 +700,7 @@ public: auto& entries() const { return m_entries; } - static ParseResult<GlobalSection> parse(InputStream& stream); + static ParseResult<GlobalSection> parse(Core::Stream::Stream& stream); private: Vector<Global> m_entries; @@ -698,7 +722,7 @@ public: auto& name() const { return m_name; } auto& description() const { return m_description; } - static ParseResult<Export> parse(InputStream& stream); + static ParseResult<Export> parse(Core::Stream::Stream& stream); private: DeprecatedString m_name; @@ -714,7 +738,7 @@ public: auto& entries() const { return m_entries; } - static ParseResult<ExportSection> parse(InputStream& stream); + static ParseResult<ExportSection> parse(Core::Stream::Stream& stream); private: Vector<Export> m_entries; @@ -731,7 +755,7 @@ public: auto& index() const { return m_index; } - static ParseResult<StartFunction> parse(InputStream& stream); + static ParseResult<StartFunction> parse(Core::Stream::Stream& stream); private: FunctionIndex m_index; @@ -746,7 +770,7 @@ public: auto& function() const { return m_function; } - static ParseResult<StartSection> parse(InputStream& stream); + static ParseResult<StartSection> parse(Core::Stream::Stream& stream); private: StartFunction m_function; @@ -765,43 +789,43 @@ public: struct SegmentType0 { // FIXME: Implement me! - static ParseResult<SegmentType0> parse(InputStream& stream); + static ParseResult<SegmentType0> parse(Core::Stream::Stream& stream); Vector<FunctionIndex> function_indices; Active mode; }; struct SegmentType1 { - static ParseResult<SegmentType1> parse(InputStream& stream); + static ParseResult<SegmentType1> parse(Core::Stream::Stream& stream); Vector<FunctionIndex> function_indices; }; struct SegmentType2 { // FIXME: Implement me! - static ParseResult<SegmentType2> parse(InputStream& stream); + static ParseResult<SegmentType2> parse(Core::Stream::Stream& stream); }; struct SegmentType3 { // FIXME: Implement me! - static ParseResult<SegmentType3> parse(InputStream& stream); + static ParseResult<SegmentType3> parse(Core::Stream::Stream& stream); }; struct SegmentType4 { // FIXME: Implement me! - static ParseResult<SegmentType4> parse(InputStream& stream); + static ParseResult<SegmentType4> parse(Core::Stream::Stream& stream); }; struct SegmentType5 { // FIXME: Implement me! - static ParseResult<SegmentType5> parse(InputStream& stream); + static ParseResult<SegmentType5> parse(Core::Stream::Stream& stream); }; struct SegmentType6 { // FIXME: Implement me! - static ParseResult<SegmentType6> parse(InputStream& stream); + static ParseResult<SegmentType6> parse(Core::Stream::Stream& stream); }; struct SegmentType7 { // FIXME: Implement me! - static ParseResult<SegmentType7> parse(InputStream& stream); + static ParseResult<SegmentType7> parse(Core::Stream::Stream& stream); }; struct Element { - static ParseResult<Element> parse(InputStream&); + static ParseResult<Element> parse(Core::Stream::Stream&); ValueType type; Vector<Expression> init; @@ -817,7 +841,7 @@ public: auto& segments() const { return m_segments; } - static ParseResult<ElementSection> parse(InputStream& stream); + static ParseResult<ElementSection> parse(Core::Stream::Stream& stream); private: Vector<Element> m_segments; @@ -835,7 +859,7 @@ public: auto n() const { return m_n; } auto& type() const { return m_type; } - static ParseResult<Locals> parse(InputStream& stream); + static ParseResult<Locals> parse(Core::Stream::Stream& stream); private: u32 m_n { 0 }; @@ -856,7 +880,7 @@ public: auto& locals() const { return m_locals; } auto& body() const { return m_body; } - static ParseResult<Func> parse(InputStream& stream); + static ParseResult<Func> parse(Core::Stream::Stream& stream); private: Vector<Locals> m_locals; @@ -873,7 +897,7 @@ public: auto size() const { return m_size; } auto& func() const { return m_func; } - static ParseResult<Code> parse(InputStream& stream); + static ParseResult<Code> parse(Core::Stream::Stream& stream); private: u32 m_size { 0 }; @@ -889,7 +913,7 @@ public: auto& functions() const { return m_functions; } - static ParseResult<CodeSection> parse(InputStream& stream); + static ParseResult<CodeSection> parse(Core::Stream::Stream& stream); private: Vector<Code> m_functions; @@ -916,7 +940,7 @@ public: auto& value() const { return m_value; } - static ParseResult<Data> parse(InputStream& stream); + static ParseResult<Data> parse(Core::Stream::Stream& stream); private: Value m_value; @@ -931,7 +955,7 @@ public: auto& data() const { return m_data; } - static ParseResult<DataSection> parse(InputStream& stream); + static ParseResult<DataSection> parse(Core::Stream::Stream& stream); private: Vector<Data> m_data; @@ -948,7 +972,7 @@ public: auto& count() const { return m_count; } - static ParseResult<DataCountSection> parse(InputStream& stream); + static ParseResult<DataCountSection> parse(Core::Stream::Stream& stream); private: Optional<u32> m_count; @@ -1043,7 +1067,7 @@ public: StringView validation_error() const { return *m_validation_error; } void set_validation_error(DeprecatedString error) { m_validation_error = move(error); } - static ParseResult<Module> parse(InputStream& stream); + static ParseResult<Module> parse(Core::Stream::Stream& stream); private: bool populate_sections(); diff --git a/Userland/Libraries/LibWeb/WebAssembly/WebAssemblyObject.cpp b/Userland/Libraries/LibWeb/WebAssembly/WebAssemblyObject.cpp index 5d1592bce1..e7557025c1 100644 --- a/Userland/Libraries/LibWeb/WebAssembly/WebAssemblyObject.cpp +++ b/Userland/Libraries/LibWeb/WebAssembly/WebAssemblyObject.cpp @@ -12,6 +12,7 @@ #include "WebAssemblyTableObject.h" #include "WebAssemblyTablePrototype.h" #include <AK/ScopeGuard.h> +#include <LibCore/MemoryStream.h> #include <LibJS/Runtime/Array.h> #include <LibJS/Runtime/ArrayBuffer.h> #include <LibJS/Runtime/BigInt.h> @@ -118,13 +119,8 @@ JS::ThrowCompletionOr<size_t> parse_module(JS::VM& vm, JS::Object* buffer_object } else { return vm.throw_completion<JS::TypeError>("Not a BufferSource"); } - InputMemoryStream stream { data }; - auto module_result = Wasm::Module::parse(stream); - ScopeGuard drain_errors { - [&] { - stream.handle_any_error(); - } - }; + auto stream = Core::Stream::FixedMemoryStream::construct(data).release_value_but_fixme_should_propagate_errors(); + auto module_result = Wasm::Module::parse(*stream); if (module_result.is_error()) { // FIXME: Throw CompileError instead. return vm.throw_completion<JS::TypeError>(Wasm::parse_error_to_deprecated_string(module_result.error())); |