| 1 | //! Conversions from `wasmparser` to `wasm-encoder` to [`Reencode`] parsed wasm. |
| 2 | //! |
| 3 | //! The [`RoundtripReencoder`] allows encoding identical wasm to the parsed |
| 4 | //! input. |
| 5 | |
| 6 | #[cfg (all(not(feature = "std" ), core_error))] |
| 7 | use core::error::Error as StdError; |
| 8 | #[cfg (feature = "std" )] |
| 9 | use std::error::Error as StdError; |
| 10 | |
| 11 | use crate::CoreTypeEncoder; |
| 12 | use core::convert::Infallible; |
| 13 | |
| 14 | #[cfg (feature = "component-model" )] |
| 15 | mod component; |
| 16 | |
| 17 | #[cfg (feature = "component-model" )] |
| 18 | pub use self::component::*; |
| 19 | |
| 20 | #[allow (missing_docs)] // FIXME |
| 21 | pub trait Reencode { |
| 22 | type Error; |
| 23 | |
| 24 | fn data_index(&mut self, data: u32) -> u32 { |
| 25 | utils::data_index(self, data) |
| 26 | } |
| 27 | |
| 28 | fn element_index(&mut self, element: u32) -> u32 { |
| 29 | utils::element_index(self, element) |
| 30 | } |
| 31 | |
| 32 | fn function_index(&mut self, func: u32) -> u32 { |
| 33 | utils::function_index(self, func) |
| 34 | } |
| 35 | |
| 36 | fn global_index(&mut self, global: u32) -> u32 { |
| 37 | utils::global_index(self, global) |
| 38 | } |
| 39 | |
| 40 | fn memory_index(&mut self, memory: u32) -> u32 { |
| 41 | utils::memory_index(self, memory) |
| 42 | } |
| 43 | |
| 44 | fn table_index(&mut self, table: u32) -> u32 { |
| 45 | utils::table_index(self, table) |
| 46 | } |
| 47 | |
| 48 | fn tag_index(&mut self, tag: u32) -> u32 { |
| 49 | utils::tag_index(self, tag) |
| 50 | } |
| 51 | |
| 52 | fn type_index(&mut self, ty: u32) -> u32 { |
| 53 | utils::type_index(self, ty) |
| 54 | } |
| 55 | |
| 56 | fn type_index_unpacked( |
| 57 | &mut self, |
| 58 | ty: wasmparser::UnpackedIndex, |
| 59 | ) -> Result<u32, Error<Self::Error>> { |
| 60 | utils::type_index_unpacked(self, ty) |
| 61 | } |
| 62 | |
| 63 | fn external_index(&mut self, kind: wasmparser::ExternalKind, index: u32) -> u32 { |
| 64 | match kind { |
| 65 | wasmparser::ExternalKind::Func => self.function_index(index), |
| 66 | wasmparser::ExternalKind::Table => self.table_index(index), |
| 67 | wasmparser::ExternalKind::Memory => self.memory_index(index), |
| 68 | wasmparser::ExternalKind::Global => self.global_index(index), |
| 69 | wasmparser::ExternalKind::Tag => self.tag_index(index), |
| 70 | } |
| 71 | } |
| 72 | |
| 73 | fn abstract_heap_type( |
| 74 | &mut self, |
| 75 | value: wasmparser::AbstractHeapType, |
| 76 | ) -> crate::AbstractHeapType { |
| 77 | utils::abstract_heap_type(self, value) |
| 78 | } |
| 79 | |
| 80 | fn array_type( |
| 81 | &mut self, |
| 82 | array_ty: wasmparser::ArrayType, |
| 83 | ) -> Result<crate::ArrayType, Error<Self::Error>> { |
| 84 | utils::array_type(self, array_ty) |
| 85 | } |
| 86 | |
| 87 | fn block_type( |
| 88 | &mut self, |
| 89 | arg: wasmparser::BlockType, |
| 90 | ) -> Result<crate::BlockType, Error<Self::Error>> { |
| 91 | utils::block_type(self, arg) |
| 92 | } |
| 93 | |
| 94 | fn const_expr( |
| 95 | &mut self, |
| 96 | const_expr: wasmparser::ConstExpr, |
| 97 | ) -> Result<crate::ConstExpr, Error<Self::Error>> { |
| 98 | utils::const_expr(self, const_expr) |
| 99 | } |
| 100 | |
| 101 | fn catch(&mut self, arg: wasmparser::Catch) -> crate::Catch { |
| 102 | utils::catch(self, arg) |
| 103 | } |
| 104 | |
| 105 | fn composite_type( |
| 106 | &mut self, |
| 107 | composite_ty: wasmparser::CompositeType, |
| 108 | ) -> Result<crate::CompositeType, Error<Self::Error>> { |
| 109 | utils::composite_type(self, composite_ty) |
| 110 | } |
| 111 | |
| 112 | fn entity_type( |
| 113 | &mut self, |
| 114 | type_ref: wasmparser::TypeRef, |
| 115 | ) -> Result<crate::EntityType, Error<Self::Error>> { |
| 116 | utils::entity_type(self, type_ref) |
| 117 | } |
| 118 | |
| 119 | fn export_kind(&mut self, external_kind: wasmparser::ExternalKind) -> crate::ExportKind { |
| 120 | utils::export_kind(self, external_kind) |
| 121 | } |
| 122 | |
| 123 | fn field_type( |
| 124 | &mut self, |
| 125 | field_ty: wasmparser::FieldType, |
| 126 | ) -> Result<crate::FieldType, Error<Self::Error>> { |
| 127 | utils::field_type(self, field_ty) |
| 128 | } |
| 129 | |
| 130 | fn func_type( |
| 131 | &mut self, |
| 132 | func_ty: wasmparser::FuncType, |
| 133 | ) -> Result<crate::FuncType, Error<Self::Error>> { |
| 134 | utils::func_type(self, func_ty) |
| 135 | } |
| 136 | |
| 137 | fn cont_type( |
| 138 | &mut self, |
| 139 | cont_ty: wasmparser::ContType, |
| 140 | ) -> Result<crate::ContType, Error<Self::Error>> { |
| 141 | utils::cont_type(self, cont_ty) |
| 142 | } |
| 143 | |
| 144 | fn global_type( |
| 145 | &mut self, |
| 146 | global_ty: wasmparser::GlobalType, |
| 147 | ) -> Result<crate::GlobalType, Error<Self::Error>> { |
| 148 | utils::global_type(self, global_ty) |
| 149 | } |
| 150 | |
| 151 | fn handle(&mut self, on: wasmparser::Handle) -> crate::Handle { |
| 152 | utils::handle(self, on) |
| 153 | } |
| 154 | |
| 155 | fn heap_type( |
| 156 | &mut self, |
| 157 | heap_type: wasmparser::HeapType, |
| 158 | ) -> Result<crate::HeapType, Error<Self::Error>> { |
| 159 | utils::heap_type(self, heap_type) |
| 160 | } |
| 161 | |
| 162 | fn instruction<'a>( |
| 163 | &mut self, |
| 164 | arg: wasmparser::Operator<'a>, |
| 165 | ) -> Result<crate::Instruction<'a>, Error<Self::Error>> { |
| 166 | utils::instruction(self, arg) |
| 167 | } |
| 168 | |
| 169 | fn memory_type(&mut self, memory_ty: wasmparser::MemoryType) -> crate::MemoryType { |
| 170 | utils::memory_type(self, memory_ty) |
| 171 | } |
| 172 | |
| 173 | fn mem_arg(&mut self, arg: wasmparser::MemArg) -> crate::MemArg { |
| 174 | utils::mem_arg(self, arg) |
| 175 | } |
| 176 | |
| 177 | fn ordering(&mut self, arg: wasmparser::Ordering) -> crate::Ordering { |
| 178 | utils::ordering(self, arg) |
| 179 | } |
| 180 | |
| 181 | fn ref_type( |
| 182 | &mut self, |
| 183 | ref_type: wasmparser::RefType, |
| 184 | ) -> Result<crate::RefType, Error<Self::Error>> { |
| 185 | utils::ref_type(self, ref_type) |
| 186 | } |
| 187 | |
| 188 | fn storage_type( |
| 189 | &mut self, |
| 190 | storage_ty: wasmparser::StorageType, |
| 191 | ) -> Result<crate::StorageType, Error<Self::Error>> { |
| 192 | utils::storage_type(self, storage_ty) |
| 193 | } |
| 194 | |
| 195 | fn struct_type( |
| 196 | &mut self, |
| 197 | struct_ty: wasmparser::StructType, |
| 198 | ) -> Result<crate::StructType, Error<Self::Error>> { |
| 199 | utils::struct_type(self, struct_ty) |
| 200 | } |
| 201 | |
| 202 | fn sub_type( |
| 203 | &mut self, |
| 204 | sub_ty: wasmparser::SubType, |
| 205 | ) -> Result<crate::SubType, Error<Self::Error>> { |
| 206 | utils::sub_type(self, sub_ty) |
| 207 | } |
| 208 | |
| 209 | fn table_type( |
| 210 | &mut self, |
| 211 | table_ty: wasmparser::TableType, |
| 212 | ) -> Result<crate::TableType, Error<Self::Error>> { |
| 213 | utils::table_type(self, table_ty) |
| 214 | } |
| 215 | |
| 216 | fn tag_kind(&mut self, kind: wasmparser::TagKind) -> crate::TagKind { |
| 217 | utils::tag_kind(self, kind) |
| 218 | } |
| 219 | |
| 220 | fn tag_type(&mut self, tag_ty: wasmparser::TagType) -> crate::TagType { |
| 221 | utils::tag_type(self, tag_ty) |
| 222 | } |
| 223 | |
| 224 | fn val_type( |
| 225 | &mut self, |
| 226 | val_ty: wasmparser::ValType, |
| 227 | ) -> Result<crate::ValType, Error<Self::Error>> { |
| 228 | utils::val_type(self, val_ty) |
| 229 | } |
| 230 | |
| 231 | /// Parses the input `section` given from the `wasmparser` crate and |
| 232 | /// adds the custom section to the `module`. |
| 233 | fn parse_custom_section( |
| 234 | &mut self, |
| 235 | module: &mut crate::Module, |
| 236 | section: wasmparser::CustomSectionReader<'_>, |
| 237 | ) -> Result<(), Error<Self::Error>> { |
| 238 | utils::parse_custom_section(self, module, section) |
| 239 | } |
| 240 | |
| 241 | /// Converts the input `section` given from the `wasmparser` crate into an |
| 242 | /// encoded custom section. |
| 243 | fn custom_section<'a>( |
| 244 | &mut self, |
| 245 | section: wasmparser::CustomSectionReader<'a>, |
| 246 | ) -> crate::CustomSection<'a> { |
| 247 | utils::custom_section(self, section) |
| 248 | } |
| 249 | |
| 250 | /// Parses the input `section` given from the `wasmparser` crate and adds |
| 251 | /// all the code to the `code` section. |
| 252 | fn parse_code_section( |
| 253 | &mut self, |
| 254 | code: &mut crate::CodeSection, |
| 255 | section: wasmparser::CodeSectionReader<'_>, |
| 256 | ) -> Result<(), Error<Self::Error>> { |
| 257 | utils::parse_code_section(self, code, section) |
| 258 | } |
| 259 | |
| 260 | /// Parses a single [`wasmparser::FunctionBody`] and adds it to the `code` section. |
| 261 | fn parse_function_body( |
| 262 | &mut self, |
| 263 | code: &mut crate::CodeSection, |
| 264 | func: wasmparser::FunctionBody<'_>, |
| 265 | ) -> Result<(), Error<Self::Error>> { |
| 266 | utils::parse_function_body(self, code, func) |
| 267 | } |
| 268 | |
| 269 | /// Create a new [`crate::Function`] by parsing the locals declarations from the |
| 270 | /// provided [`wasmparser::FunctionBody`]. |
| 271 | fn new_function_with_parsed_locals( |
| 272 | &mut self, |
| 273 | func: &wasmparser::FunctionBody<'_>, |
| 274 | ) -> Result<crate::Function, Error<Self::Error>> { |
| 275 | utils::new_function_with_parsed_locals(self, func) |
| 276 | } |
| 277 | |
| 278 | /// Parses a single instruction from `reader` and adds it to `function`. |
| 279 | fn parse_instruction<'a>( |
| 280 | &mut self, |
| 281 | reader: &mut wasmparser::OperatorsReader<'a>, |
| 282 | ) -> Result<crate::Instruction<'a>, Error<Self::Error>> { |
| 283 | utils::parse_instruction(self, reader) |
| 284 | } |
| 285 | |
| 286 | /// Parses the input `section` given from the `wasmparser` crate and adds |
| 287 | /// all the data to the `data` section. |
| 288 | fn parse_data_section( |
| 289 | &mut self, |
| 290 | data: &mut crate::DataSection, |
| 291 | section: wasmparser::DataSectionReader<'_>, |
| 292 | ) -> Result<(), Error<Self::Error>> { |
| 293 | utils::parse_data_section(self, data, section) |
| 294 | } |
| 295 | |
| 296 | /// Parses a single [`wasmparser::Data`] and adds it to the `data` section. |
| 297 | fn parse_data( |
| 298 | &mut self, |
| 299 | data: &mut crate::DataSection, |
| 300 | datum: wasmparser::Data<'_>, |
| 301 | ) -> Result<(), Error<Self::Error>> { |
| 302 | utils::parse_data(self, data, datum) |
| 303 | } |
| 304 | |
| 305 | /// Parses the input `section` given from the `wasmparser` crate and adds |
| 306 | /// all the elements to the `element` section. |
| 307 | fn parse_element_section( |
| 308 | &mut self, |
| 309 | elements: &mut crate::ElementSection, |
| 310 | section: wasmparser::ElementSectionReader<'_>, |
| 311 | ) -> Result<(), Error<Self::Error>> { |
| 312 | utils::parse_element_section(self, elements, section) |
| 313 | } |
| 314 | |
| 315 | /// Parses the single [`wasmparser::Element`] provided and adds it to the |
| 316 | /// `element` section. |
| 317 | fn parse_element( |
| 318 | &mut self, |
| 319 | elements: &mut crate::ElementSection, |
| 320 | element: wasmparser::Element<'_>, |
| 321 | ) -> Result<(), Error<Self::Error>> { |
| 322 | utils::parse_element(self, elements, element) |
| 323 | } |
| 324 | |
| 325 | fn element_items<'a>( |
| 326 | &mut self, |
| 327 | items: wasmparser::ElementItems<'a>, |
| 328 | ) -> Result<crate::Elements<'a>, Error<Self::Error>> { |
| 329 | utils::element_items(self, items) |
| 330 | } |
| 331 | |
| 332 | /// Parses the input `section` given from the `wasmparser` crate and adds |
| 333 | /// all the exports to the `exports` section. |
| 334 | fn parse_export_section( |
| 335 | &mut self, |
| 336 | exports: &mut crate::ExportSection, |
| 337 | section: wasmparser::ExportSectionReader<'_>, |
| 338 | ) -> Result<(), Error<Self::Error>> { |
| 339 | utils::parse_export_section(self, exports, section) |
| 340 | } |
| 341 | |
| 342 | /// Parses the single [`wasmparser::Export`] provided and adds it to the |
| 343 | /// `exports` section. |
| 344 | fn parse_export(&mut self, exports: &mut crate::ExportSection, export: wasmparser::Export<'_>) { |
| 345 | utils::parse_export(self, exports, export) |
| 346 | } |
| 347 | |
| 348 | /// Parses the input `section` given from the `wasmparser` crate and adds |
| 349 | /// all the functions to the `functions` section. |
| 350 | fn parse_function_section( |
| 351 | &mut self, |
| 352 | functions: &mut crate::FunctionSection, |
| 353 | section: wasmparser::FunctionSectionReader<'_>, |
| 354 | ) -> Result<(), Error<Self::Error>> { |
| 355 | utils::parse_function_section(self, functions, section) |
| 356 | } |
| 357 | |
| 358 | /// Parses the input `section` given from the `wasmparser` crate and adds |
| 359 | /// all the globals to the `globals` section. |
| 360 | fn parse_global_section( |
| 361 | &mut self, |
| 362 | globals: &mut crate::GlobalSection, |
| 363 | section: wasmparser::GlobalSectionReader<'_>, |
| 364 | ) -> Result<(), Error<Self::Error>> { |
| 365 | utils::parse_global_section(self, globals, section) |
| 366 | } |
| 367 | |
| 368 | /// Parses the single [`wasmparser::Global`] provided and adds it to the |
| 369 | /// `globals` section. |
| 370 | fn parse_global( |
| 371 | &mut self, |
| 372 | globals: &mut crate::GlobalSection, |
| 373 | global: wasmparser::Global<'_>, |
| 374 | ) -> Result<(), Error<Self::Error>> { |
| 375 | utils::parse_global(self, globals, global) |
| 376 | } |
| 377 | |
| 378 | /// Parses the input `section` given from the `wasmparser` crate and adds |
| 379 | /// all the imports to the `import` section. |
| 380 | fn parse_import_section( |
| 381 | &mut self, |
| 382 | imports: &mut crate::ImportSection, |
| 383 | section: wasmparser::ImportSectionReader<'_>, |
| 384 | ) -> Result<(), Error<Self::Error>> { |
| 385 | utils::parse_import_section(self, imports, section) |
| 386 | } |
| 387 | |
| 388 | /// Parses the single [`wasmparser::Import`] provided and adds it to the |
| 389 | /// `import` section. |
| 390 | fn parse_import( |
| 391 | &mut self, |
| 392 | imports: &mut crate::ImportSection, |
| 393 | import: wasmparser::Import<'_>, |
| 394 | ) -> Result<(), Error<Self::Error>> { |
| 395 | utils::parse_import(self, imports, import) |
| 396 | } |
| 397 | |
| 398 | /// Parses the input `section` given from the `wasmparser` crate and adds |
| 399 | /// all the memories to the `memories` section. |
| 400 | fn parse_memory_section( |
| 401 | &mut self, |
| 402 | memories: &mut crate::MemorySection, |
| 403 | section: wasmparser::MemorySectionReader<'_>, |
| 404 | ) -> Result<(), Error<Self::Error>> { |
| 405 | utils::parse_memory_section(self, memories, section) |
| 406 | } |
| 407 | |
| 408 | /// Parses the input `section` given from the `wasmparser` crate and adds |
| 409 | /// all the tables to the `tables` section. |
| 410 | fn parse_table_section( |
| 411 | &mut self, |
| 412 | tables: &mut crate::TableSection, |
| 413 | section: wasmparser::TableSectionReader<'_>, |
| 414 | ) -> Result<(), Error<Self::Error>> { |
| 415 | utils::parse_table_section(self, tables, section) |
| 416 | } |
| 417 | |
| 418 | /// Parses a single [`wasmparser::Table`] and adds it to the `tables` section. |
| 419 | fn parse_table( |
| 420 | &mut self, |
| 421 | tables: &mut crate::TableSection, |
| 422 | table: wasmparser::Table<'_>, |
| 423 | ) -> Result<(), Error<Self::Error>> { |
| 424 | utils::parse_table(self, tables, table) |
| 425 | } |
| 426 | |
| 427 | /// Parses the input `section` given from the `wasmparser` crate and adds |
| 428 | /// all the tags to the `tags` section. |
| 429 | fn parse_tag_section( |
| 430 | &mut self, |
| 431 | tags: &mut crate::TagSection, |
| 432 | section: wasmparser::TagSectionReader<'_>, |
| 433 | ) -> Result<(), Error<Self::Error>> { |
| 434 | utils::parse_tag_section(self, tags, section) |
| 435 | } |
| 436 | |
| 437 | /// Parses the input `section` given from the `wasmparser` crate and adds |
| 438 | /// all the types to the `types` section. |
| 439 | fn parse_type_section( |
| 440 | &mut self, |
| 441 | types: &mut crate::TypeSection, |
| 442 | section: wasmparser::TypeSectionReader<'_>, |
| 443 | ) -> Result<(), Error<Self::Error>> { |
| 444 | utils::parse_type_section(self, types, section) |
| 445 | } |
| 446 | |
| 447 | /// Parses a single [`wasmparser::RecGroup`] and adds it to the `types` section. |
| 448 | fn parse_recursive_type_group( |
| 449 | &mut self, |
| 450 | encoder: CoreTypeEncoder, |
| 451 | rec_group: wasmparser::RecGroup, |
| 452 | ) -> Result<(), Error<Self::Error>> { |
| 453 | utils::parse_recursive_type_group(self, encoder, rec_group) |
| 454 | } |
| 455 | |
| 456 | fn parse_unknown_section( |
| 457 | &mut self, |
| 458 | module: &mut crate::Module, |
| 459 | id: u8, |
| 460 | contents: &[u8], |
| 461 | ) -> Result<(), Error<Self::Error>> { |
| 462 | utils::parse_unknown_section(self, module, id, contents) |
| 463 | } |
| 464 | |
| 465 | /// A hook method that is called inside [`Reencode::parse_core_module`] |
| 466 | /// before and after every non-custom core wasm section. |
| 467 | /// |
| 468 | /// This method can be used to insert new custom sections in between those |
| 469 | /// sections, or to detect when a non-custom section is missing and insert |
| 470 | /// it in the [proper order]. |
| 471 | /// |
| 472 | /// The `after` parameter is `None` iff the hook is called before the first |
| 473 | /// non-custom section, and `Some(s)` afterwards, where `s` is the |
| 474 | /// [`SectionId`] of the previous non-custom section. |
| 475 | /// |
| 476 | /// The `before` parameter is `None` iff the hook is called after the last |
| 477 | /// non-custom section, and `Some(s)` beforehand, where `s` is the |
| 478 | /// [`SectionId`] of the following non-custom section. |
| 479 | /// |
| 480 | /// [proper order]: https://webassembly.github.io/spec/core/binary/modules.html#binary-module |
| 481 | /// [`SectionId`]: crate::SectionId |
| 482 | fn intersperse_section_hook( |
| 483 | &mut self, |
| 484 | module: &mut crate::Module, |
| 485 | after: Option<crate::SectionId>, |
| 486 | before: Option<crate::SectionId>, |
| 487 | ) -> Result<(), Error<Self::Error>> { |
| 488 | utils::intersperse_section_hook(self, module, after, before) |
| 489 | } |
| 490 | |
| 491 | fn parse_core_module( |
| 492 | &mut self, |
| 493 | module: &mut crate::Module, |
| 494 | parser: wasmparser::Parser, |
| 495 | data: &[u8], |
| 496 | ) -> Result<(), Error<Self::Error>> { |
| 497 | utils::parse_core_module(self, module, parser, data) |
| 498 | } |
| 499 | |
| 500 | fn custom_name_section( |
| 501 | &mut self, |
| 502 | section: wasmparser::NameSectionReader<'_>, |
| 503 | ) -> Result<crate::NameSection, Error<Self::Error>> { |
| 504 | utils::custom_name_section(self, section) |
| 505 | } |
| 506 | |
| 507 | fn parse_custom_name_subsection( |
| 508 | &mut self, |
| 509 | names: &mut crate::NameSection, |
| 510 | section: wasmparser::Name<'_>, |
| 511 | ) -> Result<(), Error<Self::Error>> { |
| 512 | utils::parse_custom_name_subsection(self, names, section) |
| 513 | } |
| 514 | |
| 515 | fn data_count(&mut self, count: u32) -> u32 { |
| 516 | count |
| 517 | } |
| 518 | |
| 519 | fn start_section(&mut self, start: u32) -> u32 { |
| 520 | self.function_index(start) |
| 521 | } |
| 522 | } |
| 523 | |
| 524 | /// An error when re-encoding from `wasmparser` to `wasm-encoder`. |
| 525 | #[derive (Debug)] |
| 526 | pub enum Error<E = Infallible> { |
| 527 | /// There was a type reference that was canonicalized and no longer |
| 528 | /// references an index into a module's types space, so we cannot encode it |
| 529 | /// into a Wasm binary again. |
| 530 | CanonicalizedHeapTypeReference, |
| 531 | /// The const expression is invalid: not actually constant or something like |
| 532 | /// that. |
| 533 | InvalidConstExpr, |
| 534 | /// The code section size listed was not valid for the wasm binary provided. |
| 535 | InvalidCodeSectionSize, |
| 536 | /// There was a section that does not belong into a core wasm module. |
| 537 | UnexpectedNonCoreModuleSection, |
| 538 | /// There was a section that does not belong into a compoennt module. |
| 539 | UnexpectedNonComponentSection, |
| 540 | /// A core type definition was found in a component that's not supported. |
| 541 | UnsupportedCoreTypeInComponent, |
| 542 | /// There was an error when parsing. |
| 543 | ParseError(wasmparser::BinaryReaderError), |
| 544 | /// There was a user-defined error when re-encoding. |
| 545 | UserError(E), |
| 546 | } |
| 547 | |
| 548 | impl<E> From<wasmparser::BinaryReaderError> for Error<E> { |
| 549 | fn from(err: wasmparser::BinaryReaderError) -> Self { |
| 550 | Self::ParseError(err) |
| 551 | } |
| 552 | } |
| 553 | |
| 554 | impl<E: core::fmt::Display> core::fmt::Display for Error<E> { |
| 555 | fn fmt(&self, fmt: &mut core::fmt::Formatter) -> core::fmt::Result { |
| 556 | match self { |
| 557 | Self::ParseError(_e) => { |
| 558 | write!(fmt, "There was an error when parsing" ) |
| 559 | } |
| 560 | Self::UserError(e) => write!(fmt, " {e}" ), |
| 561 | Self::InvalidConstExpr => write!(fmt, "The const expression was invalid" ), |
| 562 | Self::UnexpectedNonCoreModuleSection => write!( |
| 563 | fmt, |
| 564 | "There was a section that does not belong into a core wasm module" |
| 565 | ), |
| 566 | Self::UnexpectedNonComponentSection => write!( |
| 567 | fmt, |
| 568 | "There was a section that does not belong into a component" |
| 569 | ), |
| 570 | Self::CanonicalizedHeapTypeReference => write!( |
| 571 | fmt, |
| 572 | "There was a canonicalized heap type reference without type index information" |
| 573 | ), |
| 574 | Self::UnsupportedCoreTypeInComponent => { |
| 575 | fmt.write_str("unsupported core type in a component" ) |
| 576 | } |
| 577 | Self::InvalidCodeSectionSize => fmt.write_str("invalid code section size" ), |
| 578 | } |
| 579 | } |
| 580 | } |
| 581 | |
| 582 | #[cfg (any(feature = "std" , core_error))] |
| 583 | impl<E: 'static + StdError> StdError for Error<E> { |
| 584 | fn source(&self) -> Option<&(dyn StdError + 'static)> { |
| 585 | match self { |
| 586 | Self::ParseError(e: &BinaryReaderError) => Some(e), |
| 587 | Self::UserError(e: &E) => Some(e), |
| 588 | Self::InvalidConstExpr |
| 589 | | Self::CanonicalizedHeapTypeReference |
| 590 | | Self::UnexpectedNonCoreModuleSection |
| 591 | | Self::UnexpectedNonComponentSection |
| 592 | | Self::UnsupportedCoreTypeInComponent |
| 593 | | Self::InvalidCodeSectionSize => None, |
| 594 | } |
| 595 | } |
| 596 | } |
| 597 | |
| 598 | /// Reencodes `wasmparser` into `wasm-encoder` so that the encoded wasm is |
| 599 | /// identical to the input and can be parsed and encoded again. |
| 600 | #[derive (Debug)] |
| 601 | pub struct RoundtripReencoder; |
| 602 | |
| 603 | impl Reencode for RoundtripReencoder { |
| 604 | type Error = Infallible; |
| 605 | } |
| 606 | |
| 607 | #[allow (missing_docs)] // FIXME |
| 608 | pub mod utils { |
| 609 | use super::{Error, Reencode}; |
| 610 | use crate::{CoreTypeEncoder, Encode}; |
| 611 | use alloc::vec::Vec; |
| 612 | use core::ops::Range; |
| 613 | |
| 614 | pub fn parse_core_module<T: ?Sized + Reencode>( |
| 615 | reencoder: &mut T, |
| 616 | module: &mut crate::Module, |
| 617 | parser: wasmparser::Parser, |
| 618 | data: &[u8], |
| 619 | ) -> Result<(), Error<T::Error>> { |
| 620 | fn handle_intersperse_section_hook<T: ?Sized + Reencode>( |
| 621 | reencoder: &mut T, |
| 622 | module: &mut crate::Module, |
| 623 | last_section: &mut Option<crate::SectionId>, |
| 624 | next_section: Option<crate::SectionId>, |
| 625 | ) -> Result<(), Error<T::Error>> { |
| 626 | let after = core::mem::replace(last_section, next_section); |
| 627 | let before = next_section; |
| 628 | reencoder.intersperse_section_hook(module, after, before) |
| 629 | } |
| 630 | |
| 631 | // Convert from `range` to a byte range within `data` while |
| 632 | // accounting for various offsets. Then create a |
| 633 | // `CodeSectionReader` (which notably the payload does not |
| 634 | // give us here) and recurse with that. This means that |
| 635 | // users overridding `parse_code_section` always get that |
| 636 | // function called. |
| 637 | let orig_offset = parser.offset() as usize; |
| 638 | let get_original_section = |range: Range<usize>| { |
| 639 | data.get(range.start - orig_offset..range.end - orig_offset) |
| 640 | .ok_or(Error::InvalidCodeSectionSize) |
| 641 | }; |
| 642 | let mut last_section = None; |
| 643 | |
| 644 | for section in parser.parse_all(data) { |
| 645 | match section? { |
| 646 | wasmparser::Payload::Version { |
| 647 | encoding: wasmparser::Encoding::Module, |
| 648 | .. |
| 649 | } => (), |
| 650 | wasmparser::Payload::Version { .. } => { |
| 651 | return Err(Error::UnexpectedNonCoreModuleSection) |
| 652 | } |
| 653 | wasmparser::Payload::TypeSection(section) => { |
| 654 | handle_intersperse_section_hook( |
| 655 | reencoder, |
| 656 | module, |
| 657 | &mut last_section, |
| 658 | Some(crate::SectionId::Type), |
| 659 | )?; |
| 660 | let mut types = crate::TypeSection::new(); |
| 661 | reencoder.parse_type_section(&mut types, section)?; |
| 662 | module.section(&types); |
| 663 | } |
| 664 | wasmparser::Payload::ImportSection(section) => { |
| 665 | handle_intersperse_section_hook( |
| 666 | reencoder, |
| 667 | module, |
| 668 | &mut last_section, |
| 669 | Some(crate::SectionId::Import), |
| 670 | )?; |
| 671 | let mut imports = crate::ImportSection::new(); |
| 672 | reencoder.parse_import_section(&mut imports, section)?; |
| 673 | module.section(&imports); |
| 674 | } |
| 675 | wasmparser::Payload::FunctionSection(section) => { |
| 676 | handle_intersperse_section_hook( |
| 677 | reencoder, |
| 678 | module, |
| 679 | &mut last_section, |
| 680 | Some(crate::SectionId::Function), |
| 681 | )?; |
| 682 | let mut functions = crate::FunctionSection::new(); |
| 683 | reencoder.parse_function_section(&mut functions, section)?; |
| 684 | module.section(&functions); |
| 685 | } |
| 686 | wasmparser::Payload::TableSection(section) => { |
| 687 | handle_intersperse_section_hook( |
| 688 | reencoder, |
| 689 | module, |
| 690 | &mut last_section, |
| 691 | Some(crate::SectionId::Table), |
| 692 | )?; |
| 693 | let mut tables = crate::TableSection::new(); |
| 694 | reencoder.parse_table_section(&mut tables, section)?; |
| 695 | module.section(&tables); |
| 696 | } |
| 697 | wasmparser::Payload::MemorySection(section) => { |
| 698 | handle_intersperse_section_hook( |
| 699 | reencoder, |
| 700 | module, |
| 701 | &mut last_section, |
| 702 | Some(crate::SectionId::Memory), |
| 703 | )?; |
| 704 | let mut memories = crate::MemorySection::new(); |
| 705 | reencoder.parse_memory_section(&mut memories, section)?; |
| 706 | module.section(&memories); |
| 707 | } |
| 708 | wasmparser::Payload::TagSection(section) => { |
| 709 | handle_intersperse_section_hook( |
| 710 | reencoder, |
| 711 | module, |
| 712 | &mut last_section, |
| 713 | Some(crate::SectionId::Tag), |
| 714 | )?; |
| 715 | let mut tags = crate::TagSection::new(); |
| 716 | reencoder.parse_tag_section(&mut tags, section)?; |
| 717 | module.section(&tags); |
| 718 | } |
| 719 | wasmparser::Payload::GlobalSection(section) => { |
| 720 | handle_intersperse_section_hook( |
| 721 | reencoder, |
| 722 | module, |
| 723 | &mut last_section, |
| 724 | Some(crate::SectionId::Global), |
| 725 | )?; |
| 726 | let mut globals = crate::GlobalSection::new(); |
| 727 | reencoder.parse_global_section(&mut globals, section)?; |
| 728 | module.section(&globals); |
| 729 | } |
| 730 | wasmparser::Payload::ExportSection(section) => { |
| 731 | handle_intersperse_section_hook( |
| 732 | reencoder, |
| 733 | module, |
| 734 | &mut last_section, |
| 735 | Some(crate::SectionId::Export), |
| 736 | )?; |
| 737 | let mut exports = crate::ExportSection::new(); |
| 738 | reencoder.parse_export_section(&mut exports, section)?; |
| 739 | module.section(&exports); |
| 740 | } |
| 741 | wasmparser::Payload::StartSection { func, .. } => { |
| 742 | handle_intersperse_section_hook( |
| 743 | reencoder, |
| 744 | module, |
| 745 | &mut last_section, |
| 746 | Some(crate::SectionId::Start), |
| 747 | )?; |
| 748 | module.section(&crate::StartSection { |
| 749 | function_index: reencoder.start_section(func), |
| 750 | }); |
| 751 | } |
| 752 | wasmparser::Payload::ElementSection(section) => { |
| 753 | handle_intersperse_section_hook( |
| 754 | reencoder, |
| 755 | module, |
| 756 | &mut last_section, |
| 757 | Some(crate::SectionId::Element), |
| 758 | )?; |
| 759 | let mut elements = crate::ElementSection::new(); |
| 760 | reencoder.parse_element_section(&mut elements, section)?; |
| 761 | module.section(&elements); |
| 762 | } |
| 763 | wasmparser::Payload::DataCountSection { count, .. } => { |
| 764 | handle_intersperse_section_hook( |
| 765 | reencoder, |
| 766 | module, |
| 767 | &mut last_section, |
| 768 | Some(crate::SectionId::DataCount), |
| 769 | )?; |
| 770 | let count = reencoder.data_count(count); |
| 771 | module.section(&crate::DataCountSection { count }); |
| 772 | } |
| 773 | wasmparser::Payload::DataSection(section) => { |
| 774 | handle_intersperse_section_hook( |
| 775 | reencoder, |
| 776 | module, |
| 777 | &mut last_section, |
| 778 | Some(crate::SectionId::Data), |
| 779 | )?; |
| 780 | let mut data = crate::DataSection::new(); |
| 781 | reencoder.parse_data_section(&mut data, section)?; |
| 782 | module.section(&data); |
| 783 | } |
| 784 | wasmparser::Payload::CodeSectionStart { range, .. } => { |
| 785 | handle_intersperse_section_hook( |
| 786 | reencoder, |
| 787 | module, |
| 788 | &mut last_section, |
| 789 | Some(crate::SectionId::Code), |
| 790 | )?; |
| 791 | let mut codes = crate::CodeSection::new(); |
| 792 | |
| 793 | // Convert from `range` to a byte range within `data` while |
| 794 | // accounting for various offsets. Then create a |
| 795 | // `CodeSectionReader` (which notably the payload does not |
| 796 | // give us here) and recurse with that. This means that |
| 797 | // users overridding `parse_code_section` always get that |
| 798 | // function called. |
| 799 | let section = get_original_section(range.clone())?; |
| 800 | let reader = wasmparser::BinaryReader::new(section, range.start); |
| 801 | let section = wasmparser::CodeSectionReader::new(reader)?; |
| 802 | reencoder.parse_code_section(&mut codes, section)?; |
| 803 | module.section(&codes); |
| 804 | } |
| 805 | |
| 806 | // Parsing of code section entries (function bodies) happens |
| 807 | // above during the handling of the code section. That means |
| 808 | // that we just skip all these payloads. |
| 809 | wasmparser::Payload::CodeSectionEntry(_) => {} |
| 810 | |
| 811 | #[cfg (feature = "component-model" )] |
| 812 | wasmparser::Payload::ModuleSection { .. } |
| 813 | | wasmparser::Payload::InstanceSection(_) |
| 814 | | wasmparser::Payload::CoreTypeSection(_) |
| 815 | | wasmparser::Payload::ComponentSection { .. } |
| 816 | | wasmparser::Payload::ComponentInstanceSection(_) |
| 817 | | wasmparser::Payload::ComponentAliasSection(_) |
| 818 | | wasmparser::Payload::ComponentTypeSection(_) |
| 819 | | wasmparser::Payload::ComponentCanonicalSection(_) |
| 820 | | wasmparser::Payload::ComponentStartSection { .. } |
| 821 | | wasmparser::Payload::ComponentImportSection(_) |
| 822 | | wasmparser::Payload::ComponentExportSection(_) => { |
| 823 | return Err(Error::UnexpectedNonCoreModuleSection) |
| 824 | } |
| 825 | wasmparser::Payload::CustomSection(section) => { |
| 826 | reencoder.parse_custom_section(module, section)?; |
| 827 | } |
| 828 | wasmparser::Payload::End(_) => { |
| 829 | handle_intersperse_section_hook(reencoder, module, &mut last_section, None)?; |
| 830 | } |
| 831 | |
| 832 | other => match other.as_section() { |
| 833 | Some((id, range)) => { |
| 834 | let section = get_original_section(range)?; |
| 835 | reencoder.parse_unknown_section(module, id, section)?; |
| 836 | } |
| 837 | None => unreachable!(), |
| 838 | }, |
| 839 | } |
| 840 | } |
| 841 | |
| 842 | Ok(()) |
| 843 | } |
| 844 | |
| 845 | /// A hook method that is called inside [`Reencode::parse_core_module`] |
| 846 | /// before and after every non-custom core wasm section. |
| 847 | /// |
| 848 | /// This method can be used to insert new custom sections in between those |
| 849 | /// sections, or to detect when a non-custom section is missing and insert |
| 850 | /// it in the [proper order]. |
| 851 | /// |
| 852 | /// The `after` parameter is `None` iff the hook is called before the first |
| 853 | /// non-custom section, and `Some(s)` afterwards, where `s` is the |
| 854 | /// [`SectionId`] of the previous non-custom section. |
| 855 | /// |
| 856 | /// The `before` parameter is `None` iff the hook is called after the last |
| 857 | /// non-custom section, and `Some(s)` beforehand, where `s` is the |
| 858 | /// [`SectionId`] of the following non-custom section. |
| 859 | /// |
| 860 | /// [proper order]: https://webassembly.github.io/spec/core/binary/modules.html#binary-module |
| 861 | /// [`SectionId`]: crate::SectionId |
| 862 | pub fn intersperse_section_hook<T: ?Sized + Reencode>( |
| 863 | _reencoder: &mut T, |
| 864 | _module: &mut crate::Module, |
| 865 | _after: Option<crate::SectionId>, |
| 866 | _before: Option<crate::SectionId>, |
| 867 | ) -> Result<(), Error<T::Error>> { |
| 868 | Ok(()) |
| 869 | } |
| 870 | |
| 871 | pub fn memory_index<T: ?Sized + Reencode>(_reencoder: &mut T, memory: u32) -> u32 { |
| 872 | memory |
| 873 | } |
| 874 | |
| 875 | pub fn mem_arg<T: ?Sized + Reencode>( |
| 876 | reencoder: &mut T, |
| 877 | arg: wasmparser::MemArg, |
| 878 | ) -> crate::MemArg { |
| 879 | crate::MemArg { |
| 880 | offset: arg.offset, |
| 881 | align: arg.align.into(), |
| 882 | memory_index: reencoder.memory_index(arg.memory), |
| 883 | } |
| 884 | } |
| 885 | |
| 886 | pub fn ordering<T: ?Sized + Reencode>( |
| 887 | _reencoder: &mut T, |
| 888 | arg: wasmparser::Ordering, |
| 889 | ) -> crate::Ordering { |
| 890 | match arg { |
| 891 | wasmparser::Ordering::SeqCst => crate::Ordering::SeqCst, |
| 892 | wasmparser::Ordering::AcqRel => crate::Ordering::AcqRel, |
| 893 | } |
| 894 | } |
| 895 | |
| 896 | pub fn function_index<T: ?Sized + Reencode>(_reencoder: &mut T, func: u32) -> u32 { |
| 897 | func |
| 898 | } |
| 899 | |
| 900 | pub fn tag_index<T: ?Sized + Reencode>(_reencoder: &mut T, tag: u32) -> u32 { |
| 901 | tag |
| 902 | } |
| 903 | |
| 904 | pub fn catch<T: ?Sized + Reencode>(reencoder: &mut T, arg: wasmparser::Catch) -> crate::Catch { |
| 905 | match arg { |
| 906 | wasmparser::Catch::One { tag, label } => crate::Catch::One { |
| 907 | tag: reencoder.tag_index(tag), |
| 908 | label, |
| 909 | }, |
| 910 | wasmparser::Catch::OneRef { tag, label } => crate::Catch::OneRef { |
| 911 | tag: reencoder.tag_index(tag), |
| 912 | label, |
| 913 | }, |
| 914 | wasmparser::Catch::All { label } => crate::Catch::All { label }, |
| 915 | wasmparser::Catch::AllRef { label } => crate::Catch::AllRef { label }, |
| 916 | } |
| 917 | } |
| 918 | |
| 919 | pub fn handle<T: ?Sized + Reencode>( |
| 920 | reencoder: &mut T, |
| 921 | arg: wasmparser::Handle, |
| 922 | ) -> crate::Handle { |
| 923 | match arg { |
| 924 | wasmparser::Handle::OnLabel { tag, label } => crate::Handle::OnLabel { |
| 925 | tag: reencoder.tag_index(tag), |
| 926 | label, |
| 927 | }, |
| 928 | wasmparser::Handle::OnSwitch { tag } => crate::Handle::OnSwitch { |
| 929 | tag: reencoder.tag_index(tag), |
| 930 | }, |
| 931 | } |
| 932 | } |
| 933 | |
| 934 | /// Parses the input `section` given from the `wasmparser` crate and |
| 935 | /// adds the custom section to the `module`. |
| 936 | pub fn parse_custom_section<T: ?Sized + Reencode>( |
| 937 | reencoder: &mut T, |
| 938 | module: &mut crate::Module, |
| 939 | section: wasmparser::CustomSectionReader<'_>, |
| 940 | ) -> Result<(), Error<T::Error>> { |
| 941 | match section.as_known() { |
| 942 | wasmparser::KnownCustom::Name(name) => { |
| 943 | module.section(&reencoder.custom_name_section(name)?); |
| 944 | } |
| 945 | _ => { |
| 946 | module.section(&reencoder.custom_section(section)); |
| 947 | } |
| 948 | } |
| 949 | Ok(()) |
| 950 | } |
| 951 | |
| 952 | /// Converts the input `section` given from the `wasmparser` crate into an |
| 953 | /// encoded custom section. |
| 954 | pub fn custom_section<'a, T: ?Sized + Reencode>( |
| 955 | _reencoder: &mut T, |
| 956 | section: wasmparser::CustomSectionReader<'a>, |
| 957 | ) -> crate::CustomSection<'a> { |
| 958 | crate::CustomSection { |
| 959 | data: section.data().into(), |
| 960 | name: section.name().into(), |
| 961 | } |
| 962 | } |
| 963 | |
| 964 | pub fn export_kind<T: ?Sized + Reencode>( |
| 965 | _reencoder: &mut T, |
| 966 | external_kind: wasmparser::ExternalKind, |
| 967 | ) -> crate::ExportKind { |
| 968 | match external_kind { |
| 969 | wasmparser::ExternalKind::Func => crate::ExportKind::Func, |
| 970 | wasmparser::ExternalKind::Table => crate::ExportKind::Table, |
| 971 | wasmparser::ExternalKind::Memory => crate::ExportKind::Memory, |
| 972 | wasmparser::ExternalKind::Global => crate::ExportKind::Global, |
| 973 | wasmparser::ExternalKind::Tag => crate::ExportKind::Tag, |
| 974 | } |
| 975 | } |
| 976 | |
| 977 | pub fn memory_type<T: ?Sized + Reencode>( |
| 978 | _reencoder: &mut T, |
| 979 | memory_ty: wasmparser::MemoryType, |
| 980 | ) -> crate::MemoryType { |
| 981 | crate::MemoryType { |
| 982 | minimum: memory_ty.initial, |
| 983 | maximum: memory_ty.maximum, |
| 984 | memory64: memory_ty.memory64, |
| 985 | shared: memory_ty.shared, |
| 986 | page_size_log2: memory_ty.page_size_log2, |
| 987 | } |
| 988 | } |
| 989 | |
| 990 | pub fn tag_kind<T: ?Sized + Reencode>( |
| 991 | _reencoder: &mut T, |
| 992 | kind: wasmparser::TagKind, |
| 993 | ) -> crate::TagKind { |
| 994 | match kind { |
| 995 | wasmparser::TagKind::Exception => crate::TagKind::Exception, |
| 996 | } |
| 997 | } |
| 998 | |
| 999 | pub fn type_index<T: ?Sized + Reencode>(_reencoder: &mut T, ty: u32) -> u32 { |
| 1000 | ty |
| 1001 | } |
| 1002 | |
| 1003 | pub fn type_index_unpacked<T: ?Sized + Reencode>( |
| 1004 | reencoder: &mut T, |
| 1005 | ty: wasmparser::UnpackedIndex, |
| 1006 | ) -> Result<u32, Error<T::Error>> { |
| 1007 | ty.as_module_index() |
| 1008 | .map(|ty| reencoder.type_index(ty)) |
| 1009 | .ok_or(Error::CanonicalizedHeapTypeReference) |
| 1010 | } |
| 1011 | |
| 1012 | pub fn tag_type<T: ?Sized + Reencode>( |
| 1013 | reencoder: &mut T, |
| 1014 | tag_ty: wasmparser::TagType, |
| 1015 | ) -> crate::TagType { |
| 1016 | crate::TagType { |
| 1017 | kind: reencoder.tag_kind(tag_ty.kind), |
| 1018 | func_type_idx: reencoder.type_index(tag_ty.func_type_idx), |
| 1019 | } |
| 1020 | } |
| 1021 | |
| 1022 | pub fn abstract_heap_type<T: ?Sized + Reencode>( |
| 1023 | _reencoder: &mut T, |
| 1024 | value: wasmparser::AbstractHeapType, |
| 1025 | ) -> crate::AbstractHeapType { |
| 1026 | use wasmparser::AbstractHeapType::*; |
| 1027 | match value { |
| 1028 | Func => crate::AbstractHeapType::Func, |
| 1029 | Extern => crate::AbstractHeapType::Extern, |
| 1030 | Any => crate::AbstractHeapType::Any, |
| 1031 | None => crate::AbstractHeapType::None, |
| 1032 | NoExtern => crate::AbstractHeapType::NoExtern, |
| 1033 | NoFunc => crate::AbstractHeapType::NoFunc, |
| 1034 | Eq => crate::AbstractHeapType::Eq, |
| 1035 | Struct => crate::AbstractHeapType::Struct, |
| 1036 | Array => crate::AbstractHeapType::Array, |
| 1037 | I31 => crate::AbstractHeapType::I31, |
| 1038 | Exn => crate::AbstractHeapType::Exn, |
| 1039 | NoExn => crate::AbstractHeapType::NoExn, |
| 1040 | Cont => crate::AbstractHeapType::Cont, |
| 1041 | NoCont => crate::AbstractHeapType::NoCont, |
| 1042 | } |
| 1043 | } |
| 1044 | |
| 1045 | /// Parses the input `section` given from the `wasmparser` crate and adds |
| 1046 | /// all the types to the `types` section. |
| 1047 | pub fn parse_type_section<T: ?Sized + Reencode>( |
| 1048 | reencoder: &mut T, |
| 1049 | types: &mut crate::TypeSection, |
| 1050 | section: wasmparser::TypeSectionReader<'_>, |
| 1051 | ) -> Result<(), Error<T::Error>> { |
| 1052 | for rec_group in section { |
| 1053 | reencoder.parse_recursive_type_group(types.ty(), rec_group?)?; |
| 1054 | } |
| 1055 | Ok(()) |
| 1056 | } |
| 1057 | |
| 1058 | /// Parses a single [`wasmparser::RecGroup`] and adds it to the `types` section. |
| 1059 | pub fn parse_recursive_type_group<T: ?Sized + Reencode>( |
| 1060 | reencoder: &mut T, |
| 1061 | encoder: CoreTypeEncoder, |
| 1062 | rec_group: wasmparser::RecGroup, |
| 1063 | ) -> Result<(), Error<T::Error>> { |
| 1064 | if rec_group.is_explicit_rec_group() { |
| 1065 | let subtypes = rec_group |
| 1066 | .into_types() |
| 1067 | .map(|t| reencoder.sub_type(t)) |
| 1068 | .collect::<Result<Vec<_>, _>>()?; |
| 1069 | encoder.rec(subtypes); |
| 1070 | } else { |
| 1071 | let ty = rec_group.into_types().next().unwrap(); |
| 1072 | encoder.subtype(&reencoder.sub_type(ty)?); |
| 1073 | } |
| 1074 | Ok(()) |
| 1075 | } |
| 1076 | |
| 1077 | pub fn sub_type<T: ?Sized + Reencode>( |
| 1078 | reencoder: &mut T, |
| 1079 | sub_ty: wasmparser::SubType, |
| 1080 | ) -> Result<crate::SubType, Error<T::Error>> { |
| 1081 | Ok(crate::SubType { |
| 1082 | is_final: sub_ty.is_final, |
| 1083 | supertype_idx: sub_ty |
| 1084 | .supertype_idx |
| 1085 | .map(|i| reencoder.type_index_unpacked(i.unpack())) |
| 1086 | .transpose()?, |
| 1087 | composite_type: reencoder.composite_type(sub_ty.composite_type)?, |
| 1088 | }) |
| 1089 | } |
| 1090 | |
| 1091 | pub fn composite_type<T: ?Sized + Reencode>( |
| 1092 | reencoder: &mut T, |
| 1093 | composite_ty: wasmparser::CompositeType, |
| 1094 | ) -> Result<crate::CompositeType, Error<T::Error>> { |
| 1095 | let inner = match composite_ty.inner { |
| 1096 | wasmparser::CompositeInnerType::Func(f) => { |
| 1097 | crate::CompositeInnerType::Func(reencoder.func_type(f)?) |
| 1098 | } |
| 1099 | wasmparser::CompositeInnerType::Array(a) => { |
| 1100 | crate::CompositeInnerType::Array(reencoder.array_type(a)?) |
| 1101 | } |
| 1102 | wasmparser::CompositeInnerType::Struct(s) => { |
| 1103 | crate::CompositeInnerType::Struct(reencoder.struct_type(s)?) |
| 1104 | } |
| 1105 | wasmparser::CompositeInnerType::Cont(c) => { |
| 1106 | crate::CompositeInnerType::Cont(reencoder.cont_type(c)?) |
| 1107 | } |
| 1108 | }; |
| 1109 | Ok(crate::CompositeType { |
| 1110 | inner, |
| 1111 | shared: composite_ty.shared, |
| 1112 | }) |
| 1113 | } |
| 1114 | |
| 1115 | pub fn func_type<T: ?Sized + Reencode>( |
| 1116 | reencoder: &mut T, |
| 1117 | func_ty: wasmparser::FuncType, |
| 1118 | ) -> Result<crate::FuncType, Error<T::Error>> { |
| 1119 | let mut buf = Vec::with_capacity(func_ty.params().len() + func_ty.results().len()); |
| 1120 | for ty in func_ty.params().iter().chain(func_ty.results()).copied() { |
| 1121 | buf.push(reencoder.val_type(ty)?); |
| 1122 | } |
| 1123 | Ok(crate::FuncType::from_parts( |
| 1124 | buf.into(), |
| 1125 | func_ty.params().len(), |
| 1126 | )) |
| 1127 | } |
| 1128 | |
| 1129 | pub fn array_type<T: ?Sized + Reencode>( |
| 1130 | reencoder: &mut T, |
| 1131 | array_ty: wasmparser::ArrayType, |
| 1132 | ) -> Result<crate::ArrayType, Error<T::Error>> { |
| 1133 | Ok(crate::ArrayType(reencoder.field_type(array_ty.0)?)) |
| 1134 | } |
| 1135 | |
| 1136 | pub fn struct_type<T: ?Sized + Reencode>( |
| 1137 | reencoder: &mut T, |
| 1138 | struct_ty: wasmparser::StructType, |
| 1139 | ) -> Result<crate::StructType, Error<T::Error>> { |
| 1140 | Ok(crate::StructType { |
| 1141 | fields: struct_ty |
| 1142 | .fields |
| 1143 | .iter() |
| 1144 | .map(|field_ty| reencoder.field_type(*field_ty)) |
| 1145 | .collect::<Result<_, _>>()?, |
| 1146 | }) |
| 1147 | } |
| 1148 | |
| 1149 | pub fn field_type<T: ?Sized + Reencode>( |
| 1150 | reencoder: &mut T, |
| 1151 | field_ty: wasmparser::FieldType, |
| 1152 | ) -> Result<crate::FieldType, Error<T::Error>> { |
| 1153 | Ok(crate::FieldType { |
| 1154 | element_type: reencoder.storage_type(field_ty.element_type)?, |
| 1155 | mutable: field_ty.mutable, |
| 1156 | }) |
| 1157 | } |
| 1158 | |
| 1159 | pub fn storage_type<T: ?Sized + Reencode>( |
| 1160 | reencoder: &mut T, |
| 1161 | storage_ty: wasmparser::StorageType, |
| 1162 | ) -> Result<crate::StorageType, Error<T::Error>> { |
| 1163 | Ok(match storage_ty { |
| 1164 | wasmparser::StorageType::I8 => crate::StorageType::I8, |
| 1165 | wasmparser::StorageType::I16 => crate::StorageType::I16, |
| 1166 | wasmparser::StorageType::Val(v) => crate::StorageType::Val(reencoder.val_type(v)?), |
| 1167 | }) |
| 1168 | } |
| 1169 | |
| 1170 | pub fn cont_type<T: ?Sized + Reencode>( |
| 1171 | reencoder: &mut T, |
| 1172 | cont_ty: wasmparser::ContType, |
| 1173 | ) -> Result<crate::ContType, Error<T::Error>> { |
| 1174 | Ok(crate::ContType( |
| 1175 | reencoder.type_index_unpacked(cont_ty.0.unpack())?, |
| 1176 | )) |
| 1177 | } |
| 1178 | |
| 1179 | pub fn val_type<T: ?Sized + Reencode>( |
| 1180 | reencoder: &mut T, |
| 1181 | val_ty: wasmparser::ValType, |
| 1182 | ) -> Result<crate::ValType, Error<T::Error>> { |
| 1183 | Ok(match val_ty { |
| 1184 | wasmparser::ValType::I32 => crate::ValType::I32, |
| 1185 | wasmparser::ValType::I64 => crate::ValType::I64, |
| 1186 | wasmparser::ValType::F32 => crate::ValType::F32, |
| 1187 | wasmparser::ValType::F64 => crate::ValType::F64, |
| 1188 | wasmparser::ValType::V128 => crate::ValType::V128, |
| 1189 | wasmparser::ValType::Ref(r) => crate::ValType::Ref(reencoder.ref_type(r)?), |
| 1190 | }) |
| 1191 | } |
| 1192 | |
| 1193 | pub fn ref_type<T: ?Sized + Reencode>( |
| 1194 | reencoder: &mut T, |
| 1195 | ref_type: wasmparser::RefType, |
| 1196 | ) -> Result<crate::RefType, Error<T::Error>> { |
| 1197 | Ok(crate::RefType { |
| 1198 | nullable: ref_type.is_nullable(), |
| 1199 | heap_type: reencoder.heap_type(ref_type.heap_type())?, |
| 1200 | }) |
| 1201 | } |
| 1202 | |
| 1203 | pub fn heap_type<T: ?Sized + Reencode>( |
| 1204 | reencoder: &mut T, |
| 1205 | heap_type: wasmparser::HeapType, |
| 1206 | ) -> Result<crate::HeapType, Error<T::Error>> { |
| 1207 | Ok(match heap_type { |
| 1208 | wasmparser::HeapType::Concrete(i) => { |
| 1209 | crate::HeapType::Concrete(reencoder.type_index_unpacked(i)?) |
| 1210 | } |
| 1211 | wasmparser::HeapType::Abstract { shared, ty } => crate::HeapType::Abstract { |
| 1212 | shared, |
| 1213 | ty: reencoder.abstract_heap_type(ty), |
| 1214 | }, |
| 1215 | }) |
| 1216 | } |
| 1217 | |
| 1218 | /// Parses the input `section` given from the `wasmparser` crate and adds |
| 1219 | /// all the tables to the `tables` section. |
| 1220 | pub fn parse_table_section<T: ?Sized + Reencode>( |
| 1221 | reencoder: &mut T, |
| 1222 | tables: &mut crate::TableSection, |
| 1223 | section: wasmparser::TableSectionReader<'_>, |
| 1224 | ) -> Result<(), Error<T::Error>> { |
| 1225 | for table in section { |
| 1226 | reencoder.parse_table(tables, table?)?; |
| 1227 | } |
| 1228 | Ok(()) |
| 1229 | } |
| 1230 | |
| 1231 | /// Parses a single [`wasmparser::Table`] and adds it to the `tables` section. |
| 1232 | pub fn parse_table<T: ?Sized + Reencode>( |
| 1233 | reencoder: &mut T, |
| 1234 | tables: &mut crate::TableSection, |
| 1235 | table: wasmparser::Table<'_>, |
| 1236 | ) -> Result<(), Error<T::Error>> { |
| 1237 | let ty = reencoder.table_type(table.ty)?; |
| 1238 | match table.init { |
| 1239 | wasmparser::TableInit::RefNull => { |
| 1240 | tables.table(ty); |
| 1241 | } |
| 1242 | wasmparser::TableInit::Expr(e) => { |
| 1243 | tables.table_with_init(ty, &reencoder.const_expr(e)?); |
| 1244 | } |
| 1245 | } |
| 1246 | Ok(()) |
| 1247 | } |
| 1248 | |
| 1249 | pub fn table_type<T: ?Sized + Reencode>( |
| 1250 | reencoder: &mut T, |
| 1251 | table_ty: wasmparser::TableType, |
| 1252 | ) -> Result<crate::TableType, Error<T::Error>> { |
| 1253 | Ok(crate::TableType { |
| 1254 | element_type: reencoder.ref_type(table_ty.element_type)?, |
| 1255 | minimum: table_ty.initial, |
| 1256 | maximum: table_ty.maximum, |
| 1257 | table64: table_ty.table64, |
| 1258 | shared: table_ty.shared, |
| 1259 | }) |
| 1260 | } |
| 1261 | |
| 1262 | /// Parses the input `section` given from the `wasmparser` crate and adds |
| 1263 | /// all the tags to the `tags` section. |
| 1264 | pub fn parse_tag_section<T: ?Sized + Reencode>( |
| 1265 | reencoder: &mut T, |
| 1266 | tags: &mut crate::TagSection, |
| 1267 | section: wasmparser::TagSectionReader<'_>, |
| 1268 | ) -> Result<(), Error<T::Error>> { |
| 1269 | for tag in section { |
| 1270 | let tag = tag?; |
| 1271 | tags.tag(reencoder.tag_type(tag)); |
| 1272 | } |
| 1273 | Ok(()) |
| 1274 | } |
| 1275 | |
| 1276 | /// Parses the input `section` given from the `wasmparser` crate and adds |
| 1277 | /// all the exports to the `exports` section. |
| 1278 | pub fn parse_export_section<T: ?Sized + Reencode>( |
| 1279 | reencoder: &mut T, |
| 1280 | exports: &mut crate::ExportSection, |
| 1281 | section: wasmparser::ExportSectionReader<'_>, |
| 1282 | ) -> Result<(), Error<T::Error>> { |
| 1283 | for export in section { |
| 1284 | reencoder.parse_export(exports, export?); |
| 1285 | } |
| 1286 | Ok(()) |
| 1287 | } |
| 1288 | |
| 1289 | /// Parses the single [`wasmparser::Export`] provided and adds it to the |
| 1290 | /// `exports` section. |
| 1291 | pub fn parse_export<T: ?Sized + Reencode>( |
| 1292 | reencoder: &mut T, |
| 1293 | exports: &mut crate::ExportSection, |
| 1294 | export: wasmparser::Export<'_>, |
| 1295 | ) { |
| 1296 | exports.export( |
| 1297 | export.name, |
| 1298 | reencoder.export_kind(export.kind), |
| 1299 | reencoder.external_index(export.kind, export.index), |
| 1300 | ); |
| 1301 | } |
| 1302 | |
| 1303 | /// Parses the input `section` given from the `wasmparser` crate and adds |
| 1304 | /// all the globals to the `globals` section. |
| 1305 | pub fn parse_global_section<T: ?Sized + Reencode>( |
| 1306 | reencoder: &mut T, |
| 1307 | globals: &mut crate::GlobalSection, |
| 1308 | section: wasmparser::GlobalSectionReader<'_>, |
| 1309 | ) -> Result<(), Error<T::Error>> { |
| 1310 | for global in section { |
| 1311 | reencoder.parse_global(globals, global?)?; |
| 1312 | } |
| 1313 | Ok(()) |
| 1314 | } |
| 1315 | |
| 1316 | /// Parses the single [`wasmparser::Global`] provided and adds it to the |
| 1317 | /// `globals` section. |
| 1318 | pub fn parse_global<T: ?Sized + Reencode>( |
| 1319 | reencoder: &mut T, |
| 1320 | globals: &mut crate::GlobalSection, |
| 1321 | global: wasmparser::Global<'_>, |
| 1322 | ) -> Result<(), Error<T::Error>> { |
| 1323 | globals.global( |
| 1324 | reencoder.global_type(global.ty)?, |
| 1325 | &reencoder.const_expr(global.init_expr)?, |
| 1326 | ); |
| 1327 | Ok(()) |
| 1328 | } |
| 1329 | |
| 1330 | pub fn global_type<T: ?Sized + Reencode>( |
| 1331 | reencoder: &mut T, |
| 1332 | global_ty: wasmparser::GlobalType, |
| 1333 | ) -> Result<crate::GlobalType, Error<T::Error>> { |
| 1334 | Ok(crate::GlobalType { |
| 1335 | val_type: reencoder.val_type(global_ty.content_type)?, |
| 1336 | mutable: global_ty.mutable, |
| 1337 | shared: global_ty.shared, |
| 1338 | }) |
| 1339 | } |
| 1340 | |
| 1341 | pub fn entity_type<T: ?Sized + Reencode>( |
| 1342 | reencoder: &mut T, |
| 1343 | type_ref: wasmparser::TypeRef, |
| 1344 | ) -> Result<crate::EntityType, Error<T::Error>> { |
| 1345 | Ok(match type_ref { |
| 1346 | wasmparser::TypeRef::Func(i) => crate::EntityType::Function(reencoder.type_index(i)), |
| 1347 | wasmparser::TypeRef::Table(t) => crate::EntityType::Table(reencoder.table_type(t)?), |
| 1348 | wasmparser::TypeRef::Memory(m) => crate::EntityType::Memory(reencoder.memory_type(m)), |
| 1349 | wasmparser::TypeRef::Global(g) => crate::EntityType::Global(reencoder.global_type(g)?), |
| 1350 | wasmparser::TypeRef::Tag(t) => crate::EntityType::Tag(reencoder.tag_type(t)), |
| 1351 | }) |
| 1352 | } |
| 1353 | |
| 1354 | /// Parses the input `section` given from the `wasmparser` crate and adds |
| 1355 | /// all the imports to the `import` section. |
| 1356 | pub fn parse_import_section<T: ?Sized + Reencode>( |
| 1357 | reencoder: &mut T, |
| 1358 | imports: &mut crate::ImportSection, |
| 1359 | section: wasmparser::ImportSectionReader<'_>, |
| 1360 | ) -> Result<(), Error<T::Error>> { |
| 1361 | for import in section { |
| 1362 | reencoder.parse_import(imports, import?)?; |
| 1363 | } |
| 1364 | Ok(()) |
| 1365 | } |
| 1366 | |
| 1367 | /// Parses the single [`wasmparser::Import`] provided and adds it to the |
| 1368 | /// `import` section. |
| 1369 | pub fn parse_import<T: ?Sized + Reencode>( |
| 1370 | reencoder: &mut T, |
| 1371 | imports: &mut crate::ImportSection, |
| 1372 | import: wasmparser::Import<'_>, |
| 1373 | ) -> Result<(), Error<T::Error>> { |
| 1374 | imports.import( |
| 1375 | import.module, |
| 1376 | import.name, |
| 1377 | reencoder.entity_type(import.ty)?, |
| 1378 | ); |
| 1379 | Ok(()) |
| 1380 | } |
| 1381 | |
| 1382 | /// Parses the input `section` given from the `wasmparser` crate and adds |
| 1383 | /// all the memories to the `memories` section. |
| 1384 | pub fn parse_memory_section<T: ?Sized + Reencode>( |
| 1385 | reencoder: &mut T, |
| 1386 | memories: &mut crate::MemorySection, |
| 1387 | section: wasmparser::MemorySectionReader<'_>, |
| 1388 | ) -> Result<(), Error<T::Error>> { |
| 1389 | for memory in section { |
| 1390 | let memory = memory?; |
| 1391 | memories.memory(reencoder.memory_type(memory)); |
| 1392 | } |
| 1393 | Ok(()) |
| 1394 | } |
| 1395 | |
| 1396 | /// Parses the input `section` given from the `wasmparser` crate and adds |
| 1397 | /// all the functions to the `functions` section. |
| 1398 | pub fn parse_function_section<T: ?Sized + Reencode>( |
| 1399 | reencoder: &mut T, |
| 1400 | functions: &mut crate::FunctionSection, |
| 1401 | section: wasmparser::FunctionSectionReader<'_>, |
| 1402 | ) -> Result<(), Error<T::Error>> { |
| 1403 | for func in section { |
| 1404 | functions.function(reencoder.type_index(func?)); |
| 1405 | } |
| 1406 | Ok(()) |
| 1407 | } |
| 1408 | |
| 1409 | /// Parses the input `section` given from the `wasmparser` crate and adds |
| 1410 | /// all the data to the `data` section. |
| 1411 | pub fn parse_data_section<T: ?Sized + Reencode>( |
| 1412 | reencoder: &mut T, |
| 1413 | data: &mut crate::DataSection, |
| 1414 | section: wasmparser::DataSectionReader<'_>, |
| 1415 | ) -> Result<(), Error<T::Error>> { |
| 1416 | for datum in section { |
| 1417 | reencoder.parse_data(data, datum?)?; |
| 1418 | } |
| 1419 | Ok(()) |
| 1420 | } |
| 1421 | |
| 1422 | /// Parses a single [`wasmparser::Data`] and adds it to the `data` section. |
| 1423 | pub fn parse_data<T: ?Sized + Reencode>( |
| 1424 | reencoder: &mut T, |
| 1425 | data: &mut crate::DataSection, |
| 1426 | datum: wasmparser::Data<'_>, |
| 1427 | ) -> Result<(), Error<T::Error>> { |
| 1428 | match datum.kind { |
| 1429 | wasmparser::DataKind::Active { |
| 1430 | memory_index, |
| 1431 | offset_expr, |
| 1432 | } => data.active( |
| 1433 | reencoder.memory_index(memory_index), |
| 1434 | &reencoder.const_expr(offset_expr)?, |
| 1435 | datum.data.iter().copied(), |
| 1436 | ), |
| 1437 | wasmparser::DataKind::Passive => data.passive(datum.data.iter().copied()), |
| 1438 | }; |
| 1439 | Ok(()) |
| 1440 | } |
| 1441 | |
| 1442 | /// Parses the input `section` given from the `wasmparser` crate and adds |
| 1443 | /// all the elements to the `element` section. |
| 1444 | pub fn parse_element_section<T: ?Sized + Reencode>( |
| 1445 | reencoder: &mut T, |
| 1446 | elements: &mut crate::ElementSection, |
| 1447 | section: wasmparser::ElementSectionReader<'_>, |
| 1448 | ) -> Result<(), Error<T::Error>> { |
| 1449 | for element in section { |
| 1450 | reencoder.parse_element(elements, element?)?; |
| 1451 | } |
| 1452 | Ok(()) |
| 1453 | } |
| 1454 | |
| 1455 | /// Parses the single [`wasmparser::Element`] provided and adds it to the |
| 1456 | /// `element` section. |
| 1457 | pub fn parse_element<T: ?Sized + Reencode>( |
| 1458 | reencoder: &mut T, |
| 1459 | elements: &mut crate::ElementSection, |
| 1460 | element: wasmparser::Element<'_>, |
| 1461 | ) -> Result<(), Error<T::Error>> { |
| 1462 | let elems = reencoder.element_items(element.items)?; |
| 1463 | match element.kind { |
| 1464 | wasmparser::ElementKind::Active { |
| 1465 | table_index, |
| 1466 | offset_expr, |
| 1467 | } => elements.active( |
| 1468 | // Inform the reencoder that a table index is being used even if |
| 1469 | // it's not actually present here. That helps wasm-mutate for |
| 1470 | // example which wants to track uses to know when it's ok to |
| 1471 | // remove a table. |
| 1472 | // |
| 1473 | // If the table index started at `None` and is still zero then |
| 1474 | // preserve this encoding and keep it at `None`. Otherwise if |
| 1475 | // the result is nonzero or it was previously nonzero then keep |
| 1476 | // that encoding too. |
| 1477 | match (table_index, reencoder.table_index(table_index.unwrap_or(0))) { |
| 1478 | (None, 0) => None, |
| 1479 | (_, n) => Some(n), |
| 1480 | }, |
| 1481 | &reencoder.const_expr(offset_expr)?, |
| 1482 | elems, |
| 1483 | ), |
| 1484 | wasmparser::ElementKind::Passive => elements.passive(elems), |
| 1485 | wasmparser::ElementKind::Declared => elements.declared(elems), |
| 1486 | }; |
| 1487 | Ok(()) |
| 1488 | } |
| 1489 | |
| 1490 | pub fn element_items<'a, T: ?Sized + Reencode>( |
| 1491 | reencoder: &mut T, |
| 1492 | items: wasmparser::ElementItems<'a>, |
| 1493 | ) -> Result<crate::Elements<'a>, Error<T::Error>> { |
| 1494 | Ok(match items { |
| 1495 | wasmparser::ElementItems::Functions(f) => { |
| 1496 | let mut funcs = Vec::new(); |
| 1497 | for func in f { |
| 1498 | funcs.push(reencoder.function_index(func?)); |
| 1499 | } |
| 1500 | crate::Elements::Functions(funcs.into()) |
| 1501 | } |
| 1502 | wasmparser::ElementItems::Expressions(ty, e) => { |
| 1503 | let mut exprs = Vec::new(); |
| 1504 | for expr in e { |
| 1505 | exprs.push(reencoder.const_expr(expr?)?); |
| 1506 | } |
| 1507 | crate::Elements::Expressions(reencoder.ref_type(ty)?, exprs.into()) |
| 1508 | } |
| 1509 | }) |
| 1510 | } |
| 1511 | |
| 1512 | pub fn table_index<T: ?Sized + Reencode>(_reencoder: &mut T, table: u32) -> u32 { |
| 1513 | table |
| 1514 | } |
| 1515 | |
| 1516 | pub fn global_index<T: ?Sized + Reencode>(_reencoder: &mut T, global: u32) -> u32 { |
| 1517 | global |
| 1518 | } |
| 1519 | |
| 1520 | pub fn data_index<T: ?Sized + Reencode>(_reencoder: &mut T, data: u32) -> u32 { |
| 1521 | data |
| 1522 | } |
| 1523 | |
| 1524 | pub fn element_index<T: ?Sized + Reencode>(_reencoder: &mut T, element: u32) -> u32 { |
| 1525 | element |
| 1526 | } |
| 1527 | |
| 1528 | pub fn const_expr<T: ?Sized + Reencode>( |
| 1529 | reencoder: &mut T, |
| 1530 | const_expr: wasmparser::ConstExpr, |
| 1531 | ) -> Result<crate::ConstExpr, Error<T::Error>> { |
| 1532 | let mut ops = const_expr.get_operators_reader(); |
| 1533 | let mut bytes = Vec::new(); |
| 1534 | |
| 1535 | while !ops.is_end_then_eof() { |
| 1536 | let insn = reencoder.parse_instruction(&mut ops)?; |
| 1537 | insn.encode(&mut bytes); |
| 1538 | } |
| 1539 | |
| 1540 | Ok(crate::ConstExpr::raw(bytes)) |
| 1541 | } |
| 1542 | |
| 1543 | pub fn block_type<T: ?Sized + Reencode>( |
| 1544 | reencoder: &mut T, |
| 1545 | arg: wasmparser::BlockType, |
| 1546 | ) -> Result<crate::BlockType, Error<T::Error>> { |
| 1547 | match arg { |
| 1548 | wasmparser::BlockType::Empty => Ok(crate::BlockType::Empty), |
| 1549 | wasmparser::BlockType::FuncType(n) => { |
| 1550 | Ok(crate::BlockType::FunctionType(reencoder.type_index(n))) |
| 1551 | } |
| 1552 | wasmparser::BlockType::Type(t) => Ok(crate::BlockType::Result(reencoder.val_type(t)?)), |
| 1553 | } |
| 1554 | } |
| 1555 | |
| 1556 | pub fn instruction<'a, T: ?Sized + Reencode>( |
| 1557 | reencoder: &mut T, |
| 1558 | arg: wasmparser::Operator<'a>, |
| 1559 | ) -> Result<crate::Instruction<'a>, Error<T::Error>> { |
| 1560 | use crate::Instruction; |
| 1561 | |
| 1562 | macro_rules! translate { |
| 1563 | ($( @$proposal:ident $op:ident $({ $($arg:ident: $argty:ty),* })? => $visit:ident ($($ann:tt)*))*) => { |
| 1564 | Ok(match arg { |
| 1565 | $( |
| 1566 | wasmparser::Operator::$op $({ $($arg),* })? => { |
| 1567 | $( |
| 1568 | $(let $arg = translate!(map $arg $arg);)* |
| 1569 | )? |
| 1570 | translate!(build $op $($($arg)*)?) |
| 1571 | } |
| 1572 | )* |
| 1573 | unexpected => unreachable!("encountered unexpected Wasm operator: {unexpected:?}" ), |
| 1574 | }) |
| 1575 | }; |
| 1576 | |
| 1577 | // This case is used to map, based on the name of the field, from the |
| 1578 | // wasmparser payload type to the wasm-encoder payload type through |
| 1579 | // `Translator` as applicable. |
| 1580 | (map $arg:ident tag_index) => (reencoder.tag_index($arg)); |
| 1581 | (map $arg:ident function_index) => (reencoder.function_index($arg)); |
| 1582 | (map $arg:ident table) => (reencoder.table_index($arg)); |
| 1583 | (map $arg:ident table_index) => (reencoder.table_index($arg)); |
| 1584 | (map $arg:ident dst_table) => (reencoder.table_index($arg)); |
| 1585 | (map $arg:ident src_table) => (reencoder.table_index($arg)); |
| 1586 | (map $arg:ident type_index) => (reencoder.type_index($arg)); |
| 1587 | (map $arg:ident array_type_index) => (reencoder.type_index($arg)); |
| 1588 | (map $arg:ident array_type_index_dst) => (reencoder.type_index($arg)); |
| 1589 | (map $arg:ident array_type_index_src) => (reencoder.type_index($arg)); |
| 1590 | (map $arg:ident struct_type_index) => (reencoder.type_index($arg)); |
| 1591 | (map $arg:ident global_index) => (reencoder.global_index($arg)); |
| 1592 | (map $arg:ident mem) => (reencoder.memory_index($arg)); |
| 1593 | (map $arg:ident src_mem) => (reencoder.memory_index($arg)); |
| 1594 | (map $arg:ident dst_mem) => (reencoder.memory_index($arg)); |
| 1595 | (map $arg:ident data_index) => (reencoder.data_index($arg)); |
| 1596 | (map $arg:ident elem_index) => (reencoder.element_index($arg)); |
| 1597 | (map $arg:ident array_data_index) => (reencoder.data_index($arg)); |
| 1598 | (map $arg:ident array_elem_index) => (reencoder.element_index($arg)); |
| 1599 | (map $arg:ident blockty) => (reencoder.block_type($arg)?); |
| 1600 | (map $arg:ident relative_depth) => ($arg); |
| 1601 | (map $arg:ident targets) => (( |
| 1602 | $arg |
| 1603 | .targets() |
| 1604 | .collect::<Result<Vec<_>, wasmparser::BinaryReaderError>>()? |
| 1605 | .into(), |
| 1606 | $arg.default(), |
| 1607 | )); |
| 1608 | (map $arg:ident ty) => (reencoder.val_type($arg)?); |
| 1609 | (map $arg:ident hty) => (reencoder.heap_type($arg)?); |
| 1610 | (map $arg:ident from_ref_type) => (reencoder.ref_type($arg)?); |
| 1611 | (map $arg:ident to_ref_type) => (reencoder.ref_type($arg)?); |
| 1612 | (map $arg:ident memarg) => (reencoder.mem_arg($arg)); |
| 1613 | (map $arg:ident ordering) => (reencoder.ordering($arg)); |
| 1614 | (map $arg:ident local_index) => ($arg); |
| 1615 | (map $arg:ident value) => ($arg); |
| 1616 | (map $arg:ident lane) => ($arg); |
| 1617 | (map $arg:ident lanes) => ($arg); |
| 1618 | (map $arg:ident array_size) => ($arg); |
| 1619 | (map $arg:ident field_index) => ($arg); |
| 1620 | (map $arg:ident try_table) => ($arg); |
| 1621 | (map $arg:ident argument_index) => (reencoder.type_index($arg)); |
| 1622 | (map $arg:ident result_index) => (reencoder.type_index($arg)); |
| 1623 | (map $arg:ident cont_type_index) => (reencoder.type_index($arg)); |
| 1624 | (map $arg:ident resume_table) => (( |
| 1625 | $arg.handlers.into_iter().map(|h| reencoder.handle(h)).collect::<Vec<_>>().into() |
| 1626 | )); |
| 1627 | |
| 1628 | // This case takes the arguments of a wasmparser instruction and creates |
| 1629 | // a wasm-encoder instruction. There are a few special cases for where |
| 1630 | // the structure of a wasmparser instruction differs from that of |
| 1631 | // wasm-encoder. |
| 1632 | (build $op:ident) => (Instruction::$op); |
| 1633 | (build BrTable $arg:ident) => (Instruction::BrTable($arg.0, $arg.1)); |
| 1634 | (build I32Const $arg:ident) => (Instruction::I32Const($arg)); |
| 1635 | (build I64Const $arg:ident) => (Instruction::I64Const($arg)); |
| 1636 | (build F32Const $arg:ident) => (Instruction::F32Const(f32::from_bits($arg.bits()))); |
| 1637 | (build F64Const $arg:ident) => (Instruction::F64Const(f64::from_bits($arg.bits()))); |
| 1638 | (build V128Const $arg:ident) => (Instruction::V128Const($arg.i128())); |
| 1639 | (build TryTable $table:ident) => (Instruction::TryTable(reencoder.block_type($table.ty)?, { |
| 1640 | $table.catches.into_iter().map(|c| reencoder.catch(c)).collect::<Vec<_>>().into() |
| 1641 | })); |
| 1642 | (build $op:ident $arg:ident) => (Instruction::$op($arg)); |
| 1643 | (build $op:ident $($arg:ident)*) => (Instruction::$op { $($arg),* }); |
| 1644 | } |
| 1645 | |
| 1646 | wasmparser::for_each_operator!(translate) |
| 1647 | } |
| 1648 | |
| 1649 | /// Parses the input `section` given from the `wasmparser` crate and adds |
| 1650 | /// all the code to the `code` section. |
| 1651 | pub fn parse_code_section<T: ?Sized + Reencode>( |
| 1652 | reencoder: &mut T, |
| 1653 | code: &mut crate::CodeSection, |
| 1654 | section: wasmparser::CodeSectionReader<'_>, |
| 1655 | ) -> Result<(), Error<T::Error>> { |
| 1656 | for func in section { |
| 1657 | reencoder.parse_function_body(code, func?)?; |
| 1658 | } |
| 1659 | Ok(()) |
| 1660 | } |
| 1661 | |
| 1662 | /// Parses a single [`wasmparser::FunctionBody`] and adds it to the `code` section. |
| 1663 | pub fn parse_function_body<T: ?Sized + Reencode>( |
| 1664 | reencoder: &mut T, |
| 1665 | code: &mut crate::CodeSection, |
| 1666 | func: wasmparser::FunctionBody<'_>, |
| 1667 | ) -> Result<(), Error<T::Error>> { |
| 1668 | let mut f = reencoder.new_function_with_parsed_locals(&func)?; |
| 1669 | let mut reader = func.get_operators_reader()?; |
| 1670 | while !reader.eof() { |
| 1671 | f.instruction(&reencoder.parse_instruction(&mut reader)?); |
| 1672 | } |
| 1673 | code.function(&f); |
| 1674 | Ok(()) |
| 1675 | } |
| 1676 | |
| 1677 | /// Create a new [`crate::Function`] by parsing the locals declarations from the |
| 1678 | /// provided [`wasmparser::FunctionBody`]. |
| 1679 | pub fn new_function_with_parsed_locals<T: ?Sized + Reencode>( |
| 1680 | reencoder: &mut T, |
| 1681 | func: &wasmparser::FunctionBody<'_>, |
| 1682 | ) -> Result<crate::Function, Error<T::Error>> { |
| 1683 | let mut locals = Vec::new(); |
| 1684 | for pair in func.get_locals_reader()? { |
| 1685 | let (cnt, ty) = pair?; |
| 1686 | locals.push((cnt, reencoder.val_type(ty)?)); |
| 1687 | } |
| 1688 | Ok(crate::Function::new(locals)) |
| 1689 | } |
| 1690 | |
| 1691 | /// Parses a single instruction from `reader` and adds it to `function`. |
| 1692 | pub fn parse_instruction<'a, T: ?Sized + Reencode>( |
| 1693 | reencoder: &mut T, |
| 1694 | reader: &mut wasmparser::OperatorsReader<'a>, |
| 1695 | ) -> Result<crate::Instruction<'a>, Error<T::Error>> { |
| 1696 | let instruction = reencoder.instruction(reader.read()?)?; |
| 1697 | Ok(instruction) |
| 1698 | } |
| 1699 | |
| 1700 | pub fn parse_unknown_section<T: ?Sized + Reencode>( |
| 1701 | _reencoder: &mut T, |
| 1702 | module: &mut crate::Module, |
| 1703 | id: u8, |
| 1704 | contents: &[u8], |
| 1705 | ) -> Result<(), Error<T::Error>> { |
| 1706 | module.section(&crate::RawSection { id, data: contents }); |
| 1707 | Ok(()) |
| 1708 | } |
| 1709 | |
| 1710 | pub fn custom_name_section<T: ?Sized + Reencode>( |
| 1711 | reencoder: &mut T, |
| 1712 | section: wasmparser::NameSectionReader<'_>, |
| 1713 | ) -> Result<crate::NameSection, Error<T::Error>> { |
| 1714 | let mut ret = crate::NameSection::new(); |
| 1715 | for subsection in section { |
| 1716 | reencoder.parse_custom_name_subsection(&mut ret, subsection?)?; |
| 1717 | } |
| 1718 | Ok(ret) |
| 1719 | } |
| 1720 | |
| 1721 | pub fn parse_custom_name_subsection<T: ?Sized + Reencode>( |
| 1722 | reencoder: &mut T, |
| 1723 | names: &mut crate::NameSection, |
| 1724 | section: wasmparser::Name<'_>, |
| 1725 | ) -> Result<(), Error<T::Error>> { |
| 1726 | match section { |
| 1727 | wasmparser::Name::Module { name, .. } => { |
| 1728 | names.module(name); |
| 1729 | } |
| 1730 | wasmparser::Name::Function(map) => { |
| 1731 | names.functions(&name_map(map, |i| reencoder.function_index(i))?); |
| 1732 | } |
| 1733 | wasmparser::Name::Type(map) => { |
| 1734 | names.types(&name_map(map, |i| reencoder.type_index(i))?); |
| 1735 | } |
| 1736 | wasmparser::Name::Local(map) => { |
| 1737 | names.locals(&indirect_name_map(map, |i| reencoder.function_index(i))?); |
| 1738 | } |
| 1739 | wasmparser::Name::Label(map) => { |
| 1740 | names.labels(&indirect_name_map(map, |i| reencoder.function_index(i))?); |
| 1741 | } |
| 1742 | wasmparser::Name::Table(map) => { |
| 1743 | names.tables(&name_map(map, |i| reencoder.table_index(i))?); |
| 1744 | } |
| 1745 | wasmparser::Name::Memory(map) => { |
| 1746 | names.memories(&name_map(map, |i| reencoder.memory_index(i))?); |
| 1747 | } |
| 1748 | wasmparser::Name::Global(map) => { |
| 1749 | names.globals(&name_map(map, |i| reencoder.global_index(i))?); |
| 1750 | } |
| 1751 | wasmparser::Name::Element(map) => { |
| 1752 | names.elements(&name_map(map, |i| reencoder.element_index(i))?); |
| 1753 | } |
| 1754 | wasmparser::Name::Data(map) => { |
| 1755 | names.data(&name_map(map, |i| reencoder.data_index(i))?); |
| 1756 | } |
| 1757 | wasmparser::Name::Tag(map) => { |
| 1758 | names.tags(&name_map(map, |i| reencoder.tag_index(i))?); |
| 1759 | } |
| 1760 | wasmparser::Name::Field(map) => { |
| 1761 | names.fields(&indirect_name_map(map, |i| reencoder.type_index(i))?); |
| 1762 | } |
| 1763 | wasmparser::Name::Unknown { ty, data, .. } => { |
| 1764 | names.raw(ty, data); |
| 1765 | } |
| 1766 | } |
| 1767 | Ok(()) |
| 1768 | } |
| 1769 | |
| 1770 | pub fn name_map( |
| 1771 | map: wasmparser::NameMap<'_>, |
| 1772 | mut map_index: impl FnMut(u32) -> u32, |
| 1773 | ) -> wasmparser::Result<crate::NameMap> { |
| 1774 | let mut ret = crate::NameMap::new(); |
| 1775 | for naming in map { |
| 1776 | let naming = naming?; |
| 1777 | ret.append(map_index(naming.index), naming.name); |
| 1778 | } |
| 1779 | Ok(ret) |
| 1780 | } |
| 1781 | |
| 1782 | pub fn indirect_name_map( |
| 1783 | map: wasmparser::IndirectNameMap<'_>, |
| 1784 | mut map_index: impl FnMut(u32) -> u32, |
| 1785 | ) -> wasmparser::Result<crate::IndirectNameMap> { |
| 1786 | let mut ret = crate::IndirectNameMap::new(); |
| 1787 | for naming in map { |
| 1788 | let naming = naming?; |
| 1789 | ret.append(map_index(naming.index), &name_map(naming.names, |i| i)?); |
| 1790 | } |
| 1791 | Ok(ret) |
| 1792 | } |
| 1793 | } |
| 1794 | |
| 1795 | impl From<wasmparser::MemArg> for crate::MemArg { |
| 1796 | fn from(arg: wasmparser::MemArg) -> Self { |
| 1797 | RoundtripReencoder.mem_arg(arg) |
| 1798 | } |
| 1799 | } |
| 1800 | |
| 1801 | impl From<wasmparser::Ordering> for crate::Ordering { |
| 1802 | fn from(arg: wasmparser::Ordering) -> Self { |
| 1803 | RoundtripReencoder.ordering(arg) |
| 1804 | } |
| 1805 | } |
| 1806 | |
| 1807 | impl TryFrom<wasmparser::BlockType> for crate::BlockType { |
| 1808 | type Error = Error; |
| 1809 | |
| 1810 | fn try_from(arg: wasmparser::BlockType) -> Result<Self, Self::Error> { |
| 1811 | RoundtripReencoder.block_type(arg) |
| 1812 | } |
| 1813 | } |
| 1814 | |
| 1815 | impl<'a> TryFrom<wasmparser::Operator<'a>> for crate::Instruction<'a> { |
| 1816 | type Error = Error; |
| 1817 | |
| 1818 | fn try_from(arg: wasmparser::Operator<'a>) -> Result<Self, Self::Error> { |
| 1819 | RoundtripReencoder.instruction(arg) |
| 1820 | } |
| 1821 | } |
| 1822 | |
| 1823 | impl From<wasmparser::Catch> for crate::Catch { |
| 1824 | fn from(arg: wasmparser::Catch) -> Self { |
| 1825 | RoundtripReencoder.catch(arg) |
| 1826 | } |
| 1827 | } |
| 1828 | |
| 1829 | impl<'a> TryFrom<wasmparser::ConstExpr<'a>> for crate::ConstExpr { |
| 1830 | type Error = Error; |
| 1831 | |
| 1832 | fn try_from(const_expr: wasmparser::ConstExpr) -> Result<Self, Self::Error> { |
| 1833 | RoundtripReencoder.const_expr(const_expr) |
| 1834 | } |
| 1835 | } |
| 1836 | |
| 1837 | impl<'a> From<wasmparser::CustomSectionReader<'a>> for crate::CustomSection<'a> { |
| 1838 | fn from(section: wasmparser::CustomSectionReader<'a>) -> Self { |
| 1839 | RoundtripReencoder.custom_section(section) |
| 1840 | } |
| 1841 | } |
| 1842 | |
| 1843 | impl From<wasmparser::ExternalKind> for crate::ExportKind { |
| 1844 | fn from(external_kind: wasmparser::ExternalKind) -> Self { |
| 1845 | RoundtripReencoder.export_kind(external_kind) |
| 1846 | } |
| 1847 | } |
| 1848 | |
| 1849 | impl TryFrom<wasmparser::GlobalType> for crate::GlobalType { |
| 1850 | type Error = Error; |
| 1851 | |
| 1852 | fn try_from(global_ty: wasmparser::GlobalType) -> Result<Self, Self::Error> { |
| 1853 | RoundtripReencoder.global_type(global_ty) |
| 1854 | } |
| 1855 | } |
| 1856 | |
| 1857 | impl From<wasmparser::Handle> for crate::Handle { |
| 1858 | fn from(arg: wasmparser::Handle) -> Self { |
| 1859 | RoundtripReencoder.handle(on:arg) |
| 1860 | } |
| 1861 | } |
| 1862 | |
| 1863 | impl TryFrom<wasmparser::TypeRef> for crate::EntityType { |
| 1864 | type Error = Error; |
| 1865 | |
| 1866 | fn try_from(type_ref: wasmparser::TypeRef) -> Result<Self, Self::Error> { |
| 1867 | RoundtripReencoder.entity_type(type_ref) |
| 1868 | } |
| 1869 | } |
| 1870 | |
| 1871 | impl From<wasmparser::MemoryType> for crate::MemoryType { |
| 1872 | fn from(memory_ty: wasmparser::MemoryType) -> Self { |
| 1873 | RoundtripReencoder.memory_type(memory_ty) |
| 1874 | } |
| 1875 | } |
| 1876 | |
| 1877 | impl TryFrom<wasmparser::TableType> for crate::TableType { |
| 1878 | type Error = Error; |
| 1879 | |
| 1880 | fn try_from(table_ty: wasmparser::TableType) -> Result<Self, Self::Error> { |
| 1881 | RoundtripReencoder.table_type(table_ty) |
| 1882 | } |
| 1883 | } |
| 1884 | |
| 1885 | impl From<wasmparser::TagKind> for crate::TagKind { |
| 1886 | fn from(kind: wasmparser::TagKind) -> Self { |
| 1887 | RoundtripReencoder.tag_kind(kind) |
| 1888 | } |
| 1889 | } |
| 1890 | |
| 1891 | impl From<wasmparser::TagType> for crate::TagType { |
| 1892 | fn from(tag_ty: wasmparser::TagType) -> Self { |
| 1893 | RoundtripReencoder.tag_type(tag_ty) |
| 1894 | } |
| 1895 | } |
| 1896 | |
| 1897 | impl TryFrom<wasmparser::SubType> for crate::SubType { |
| 1898 | type Error = Error; |
| 1899 | |
| 1900 | fn try_from(sub_ty: wasmparser::SubType) -> Result<Self, Self::Error> { |
| 1901 | RoundtripReencoder.sub_type(sub_ty) |
| 1902 | } |
| 1903 | } |
| 1904 | |
| 1905 | impl TryFrom<wasmparser::CompositeType> for crate::CompositeType { |
| 1906 | type Error = Error; |
| 1907 | |
| 1908 | fn try_from(composite_ty: wasmparser::CompositeType) -> Result<Self, Self::Error> { |
| 1909 | RoundtripReencoder.composite_type(composite_ty) |
| 1910 | } |
| 1911 | } |
| 1912 | |
| 1913 | impl TryFrom<wasmparser::FuncType> for crate::FuncType { |
| 1914 | type Error = Error; |
| 1915 | |
| 1916 | fn try_from(func_ty: wasmparser::FuncType) -> Result<Self, Self::Error> { |
| 1917 | RoundtripReencoder.func_type(func_ty) |
| 1918 | } |
| 1919 | } |
| 1920 | |
| 1921 | impl TryFrom<wasmparser::ArrayType> for crate::ArrayType { |
| 1922 | type Error = Error; |
| 1923 | |
| 1924 | fn try_from(array_ty: wasmparser::ArrayType) -> Result<Self, Self::Error> { |
| 1925 | RoundtripReencoder.array_type(array_ty) |
| 1926 | } |
| 1927 | } |
| 1928 | |
| 1929 | impl TryFrom<wasmparser::StructType> for crate::StructType { |
| 1930 | type Error = Error; |
| 1931 | |
| 1932 | fn try_from(struct_ty: wasmparser::StructType) -> Result<Self, Self::Error> { |
| 1933 | RoundtripReencoder.struct_type(struct_ty) |
| 1934 | } |
| 1935 | } |
| 1936 | |
| 1937 | impl TryFrom<wasmparser::FieldType> for crate::FieldType { |
| 1938 | type Error = Error; |
| 1939 | |
| 1940 | fn try_from(field_ty: wasmparser::FieldType) -> Result<Self, Self::Error> { |
| 1941 | RoundtripReencoder.field_type(field_ty) |
| 1942 | } |
| 1943 | } |
| 1944 | |
| 1945 | impl TryFrom<wasmparser::StorageType> for crate::StorageType { |
| 1946 | type Error = Error; |
| 1947 | |
| 1948 | fn try_from(storage_ty: wasmparser::StorageType) -> Result<Self, Self::Error> { |
| 1949 | RoundtripReencoder.storage_type(storage_ty) |
| 1950 | } |
| 1951 | } |
| 1952 | |
| 1953 | impl TryFrom<wasmparser::ValType> for crate::ValType { |
| 1954 | type Error = Error; |
| 1955 | |
| 1956 | fn try_from(val_ty: wasmparser::ValType) -> Result<Self, Self::Error> { |
| 1957 | RoundtripReencoder.val_type(val_ty) |
| 1958 | } |
| 1959 | } |
| 1960 | |
| 1961 | impl TryFrom<wasmparser::RefType> for crate::RefType { |
| 1962 | type Error = Error; |
| 1963 | |
| 1964 | fn try_from(ref_type: wasmparser::RefType) -> Result<Self, Self::Error> { |
| 1965 | RoundtripReencoder.ref_type(ref_type) |
| 1966 | } |
| 1967 | } |
| 1968 | |
| 1969 | impl TryFrom<wasmparser::HeapType> for crate::HeapType { |
| 1970 | type Error = Error; |
| 1971 | |
| 1972 | fn try_from(heap_type: wasmparser::HeapType) -> Result<Self, Self::Error> { |
| 1973 | crate::reencode::utils::heap_type(&mut crate::reencode::RoundtripReencoder, heap_type) |
| 1974 | } |
| 1975 | } |
| 1976 | |
| 1977 | impl From<wasmparser::AbstractHeapType> for crate::AbstractHeapType { |
| 1978 | fn from(value: wasmparser::AbstractHeapType) -> Self { |
| 1979 | RoundtripReencoder.abstract_heap_type(value) |
| 1980 | } |
| 1981 | } |
| 1982 | |