mii/wasmer
1
0
Fork 0
mirror of https://github.com/mii443/wasmer.git synced 2025-12-10 14:48:27 +00:00
Code Issues Packages Projects Releases Wiki Activity

Add conversion logic for boxed_vec, simplify vec creation code

Browse Source
This commit is contained in:
Mark McCaskey
2020-10-02 16:24:46 -07:00
parent 554e8794ad
commit 34e6139570
2 changed files with 45 additions and 34 deletions
Download Patch File Download Diff File Expand all files Collapse all files

32
lib/c-api/src/wasm_c_api/module.rs
View File

@@ -38,18 +38,14 @@ pub unsafe extern "C" fn wasm_module_exports(
module: &wasm_module_t,
out: &mut wasm_exporttype_vec_t,
) {
let mut exports = module
let exports = module
.inner
.exports()
.map(Into::into)
.map(Box::new)
.map(Box::into_raw)
.collect::<Vec<*mut wasm_exporttype_t>>();
.collect::<Vec<Box<wasm_exporttype_t>>>();
debug_assert_eq!(exports.len(), exports.capacity());
out.size = exports.len();
out.data = exports.as_mut_ptr();
mem::forget(exports);
*out = exports.into();
}
#[no_mangle]
@@ -57,18 +53,14 @@ pub unsafe extern "C" fn wasm_module_imports(
module: &wasm_module_t,
out: &mut wasm_importtype_vec_t,
) {
let mut imports = module
let imports = module
.inner
.imports()
.map(Into::into)
.map(Box::new)
.map(Box::into_raw)
.collect::<Vec<*mut wasm_importtype_t>>();
.collect::<Vec<Box<wasm_importtype_t>>>();
debug_assert_eq!(imports.len(), imports.capacity());
out.size = imports.len();
out.data = imports.as_mut_ptr();
mem::forget(imports);
*out = imports.into();
}
#[no_mangle]
@@ -102,16 +94,8 @@ pub unsafe extern "C" fn wasm_module_serialize(
out_ptr: &mut wasm_byte_vec_t,
) {
let mut byte_vec = match module.inner.serialize() {
Ok(mut byte_vec) => {
byte_vec.shrink_to_fit();
byte_vec
}
Ok(byte_vec) => byte_vec,
Err(_) => return,
};
// ensure we won't leak memory
// TODO: use `Vec::into_raw_parts` when it becomes stable
debug_assert_eq!(byte_vec.capacity(), byte_vec.len());
out_ptr.size = byte_vec.len();
out_ptr.data = byte_vec.as_mut_ptr();
mem::forget(byte_vec);
*out_ptr = byte_vec.into();
}