1use std::fmt::{self, Debug};
2
3use anyhow::Result;
4use wasm_encoder::Encode;
5use wasmparser::{BinaryReader, NameSectionReader};
6
7use crate::utils::{indirect_name_map, name_map};
8
9/// Helper for rewriting a module's name section with a new module name.
10pub struct ModuleNames<'a> {
11 module_name: Option<String>,
12 names: Vec<wasmparser::Name<'a>>,
13}
14
15impl<'a> Debug for ModuleNames<'a> {
16 fn fmt(&self, f: &mut fmt::Formatter<'_>) -> fmt::Result {
17 f&mut DebugStruct<'_, '_>.debug_struct("ModuleNames")
18 .field(name:"module_name", &self.module_name)
19 .finish_non_exhaustive()
20 }
21}
22
23impl<'a> ModuleNames<'a> {
24 /// Create an empty name section.
25 pub fn empty() -> Self {
26 ModuleNames {
27 module_name: None,
28 names: Vec::new(),
29 }
30 }
31 /// Read a name section from a WebAssembly binary. Records the module name, and all other
32 /// contents of name section, for later serialization.
33 pub fn from_bytes(bytes: &'a [u8], offset: usize) -> Result<ModuleNames<'a>> {
34 let reader = BinaryReader::new(bytes, offset);
35 let section = NameSectionReader::new(reader);
36 let mut s = Self::empty();
37 for name in section.into_iter() {
38 let name = name?;
39 match name {
40 wasmparser::Name::Module { name, .. } => s.module_name = Some(name.to_owned()),
41 _ => s.names.push(name),
42 }
43 }
44 Ok(s)
45 }
46 /// Update module section according to [`AddMetadata`]
47 pub(crate) fn from_name(name: &Option<String>) -> Self {
48 let mut s = Self::empty();
49 s.module_name = name.clone();
50 s
51 }
52
53 /// Merge with another section
54 pub(crate) fn merge(&mut self, other: &Self) {
55 if other.module_name.is_some() {
56 self.module_name = other.module_name.clone();
57 }
58 self.names.extend_from_slice(&other.names);
59 }
60
61 /// Set module name
62 pub fn set_name(&mut self, name: &str) {
63 self.module_name = Some(name.to_owned())
64 }
65 /// Get module name
66 pub fn get_name(&self) -> Option<&String> {
67 self.module_name.as_ref()
68 }
69 /// Serialize into [`wasm_encoder::NameSection`].
70 pub(crate) fn section(&self) -> Result<wasm_encoder::NameSection> {
71 let mut section = wasm_encoder::NameSection::new();
72 if let Some(module_name) = &self.module_name {
73 section.module(&module_name);
74 }
75 for n in self.names.iter() {
76 match n {
77 wasmparser::Name::Module { .. } => unreachable!(),
78 wasmparser::Name::Function(m) => section.functions(&name_map(&m)?),
79 wasmparser::Name::Local(m) => section.locals(&indirect_name_map(&m)?),
80 wasmparser::Name::Label(m) => section.labels(&indirect_name_map(&m)?),
81 wasmparser::Name::Type(m) => section.types(&name_map(&m)?),
82 wasmparser::Name::Table(m) => section.tables(&name_map(&m)?),
83 wasmparser::Name::Memory(m) => section.memories(&name_map(&m)?),
84 wasmparser::Name::Global(m) => section.globals(&name_map(&m)?),
85 wasmparser::Name::Element(m) => section.elements(&name_map(&m)?),
86 wasmparser::Name::Data(m) => section.data(&name_map(&m)?),
87 wasmparser::Name::Field(m) => section.fields(&indirect_name_map(&m)?),
88 wasmparser::Name::Tag(m) => section.tags(&name_map(&m)?),
89 wasmparser::Name::Unknown { .. } => {} // wasm-encoder doesn't support it
90 }
91 }
92 Ok(section)
93 }
94
95 /// Serialize into the raw bytes of a wasm custom section.
96 pub fn raw_custom_section(&self) -> Result<Vec<u8>> {
97 let mut ret = Vec::new();
98 self.section()?.encode(&mut ret);
99 Ok(ret)
100 }
101}
102