pwasm_utils/
build.rs

1use super::{
2	externalize_mem, inject_runtime_type, optimize, pack_instance, shrink_unknown_stack, std::fmt,
3	ununderscore_funcs, OptimizerError, PackingError, TargetRuntime,
4};
5use parity_wasm::elements;
6
7#[derive(Debug)]
8pub enum Error {
9	Encoding(elements::Error),
10	Packing(PackingError),
11	Optimizer,
12}
13
14impl From<OptimizerError> for Error {
15	fn from(_err: OptimizerError) -> Self {
16		Error::Optimizer
17	}
18}
19
20impl From<PackingError> for Error {
21	fn from(err: PackingError) -> Self {
22		Error::Packing(err)
23	}
24}
25
26#[derive(Debug, Clone, Copy)]
27pub enum SourceTarget {
28	Emscripten,
29	Unknown,
30}
31
32impl fmt::Display for Error {
33	fn fmt(&self, f: &mut fmt::Formatter) -> Result<(), fmt::Error> {
34		use self::Error::*;
35		match self {
36			Encoding(err) => write!(f, "Encoding error ({})", err),
37			Optimizer => write!(f, "Optimization error due to missing export section. Pointed wrong file?"),
38			Packing(e) => write!(f, "Packing failed due to module structure error: {}. Sure used correct libraries for building contracts?", e),
39		}
40	}
41}
42
43fn has_ctor(module: &elements::Module, target_runtime: &TargetRuntime) -> bool {
44	if let Some(section) = module.export_section() {
45		section.entries().iter().any(|e| target_runtime.symbols().create == e.field())
46	} else {
47		false
48	}
49}
50
51#[allow(clippy::too_many_arguments)]
52pub fn build(
53	mut module: elements::Module,
54	source_target: SourceTarget,
55	runtime_type_version: Option<([u8; 4], u32)>,
56	public_api_entries: &[&str],
57	enforce_stack_adjustment: bool,
58	stack_size: u32,
59	skip_optimization: bool,
60	target_runtime: &TargetRuntime,
61) -> Result<(elements::Module, Option<elements::Module>), Error> {
62	if let SourceTarget::Emscripten = source_target {
63		module = ununderscore_funcs(module);
64	}
65
66	if let SourceTarget::Unknown = source_target {
67		// 49152 is 48kb!
68		if enforce_stack_adjustment {
69			assert!(stack_size <= 1024 * 1024);
70			let (new_module, new_stack_top) =
71				shrink_unknown_stack(module, 1024 * 1024 - stack_size);
72			module = new_module;
73			let mut stack_top_page = new_stack_top / 65536;
74			if new_stack_top % 65536 > 0 {
75				stack_top_page += 1
76			};
77			module = externalize_mem(module, Some(stack_top_page), 16);
78		} else {
79			module = externalize_mem(module, None, 16);
80		}
81	}
82
83	if let Some(runtime_type_version) = runtime_type_version {
84		let (runtime_type, runtime_version) = runtime_type_version;
85		module = inject_runtime_type(module, runtime_type, runtime_version);
86	}
87
88	let mut ctor_module = module.clone();
89
90	let mut public_api_entries = public_api_entries.to_vec();
91	public_api_entries.push(target_runtime.symbols().call);
92	if !skip_optimization {
93		optimize(&mut module, public_api_entries)?;
94	}
95
96	if !has_ctor(&ctor_module, target_runtime) {
97		return Ok((module, None))
98	}
99
100	if !skip_optimization {
101		let preserved_exports = match target_runtime {
102			TargetRuntime::PWasm(_) => vec![target_runtime.symbols().create],
103			TargetRuntime::Substrate(_) => {
104				vec![target_runtime.symbols().call, target_runtime.symbols().create]
105			},
106		};
107		optimize(&mut ctor_module, preserved_exports)?;
108	}
109
110	if let TargetRuntime::PWasm(_) = target_runtime {
111		ctor_module = pack_instance(
112			parity_wasm::serialize(module.clone()).map_err(Error::Encoding)?,
113			ctor_module.clone(),
114			target_runtime,
115		)?;
116	}
117
118	Ok((module, Some(ctor_module)))
119}
pwasm_utils/build.rs

pwasm_utils/
build.rs
