// Don't worry about unused imports if we're frobbing features, only worry about // them with the default set of features enabled. #![cfg_attr(not(feature = "cache"), allow(unused_imports))] use crate::{handle_result, wasm_memorytype_t, wasmtime_error_t}; use std::os::raw::c_char; use std::ptr; use std::{ffi::CStr, sync::Arc}; use wasmtime::{ Config, InstanceAllocationStrategy, LinearMemory, MemoryCreator, OptLevel, ProfilingStrategy, Result, Strategy, }; #[cfg(feature = "pooling-allocator")] use wasmtime::PoolingAllocationConfig; #[repr(C)] #[derive(Clone)] pub struct wasm_config_t { pub(crate) config: Config, } wasmtime_c_api_macros::declare_own!(wasm_config_t); #[repr(u8)] #[derive(Clone)] pub enum wasmtime_strategy_t { WASMTIME_STRATEGY_AUTO, WASMTIME_STRATEGY_CRANELIFT, } #[repr(u8)] #[derive(Clone)] pub enum wasmtime_opt_level_t { WASMTIME_OPT_LEVEL_NONE, WASMTIME_OPT_LEVEL_SPEED, WASMTIME_OPT_LEVEL_SPEED_AND_SIZE, } #[repr(u8)] #[derive(Clone)] pub enum wasmtime_profiling_strategy_t { WASMTIME_PROFILING_STRATEGY_NONE, WASMTIME_PROFILING_STRATEGY_JITDUMP, WASMTIME_PROFILING_STRATEGY_VTUNE, WASMTIME_PROFILING_STRATEGY_PERFMAP, } #[unsafe(no_mangle)] pub extern "C" fn wasm_config_new() -> Box { Box::new(wasm_config_t { config: Config::default(), }) } #[unsafe(no_mangle)] pub extern "C" fn wasmtime_config_debug_info_set(c: &mut wasm_config_t, enable: bool) { c.config.debug_info(enable); } #[unsafe(no_mangle)] pub extern "C" fn wasmtime_config_consume_fuel_set(c: &mut wasm_config_t, enable: bool) { c.config.consume_fuel(enable); } #[unsafe(no_mangle)] pub extern "C" fn wasmtime_config_epoch_interruption_set(c: &mut wasm_config_t, enable: bool) { c.config.epoch_interruption(enable); } #[unsafe(no_mangle)] pub extern "C" fn wasmtime_config_max_wasm_stack_set(c: &mut wasm_config_t, size: usize) { c.config.max_wasm_stack(size); } #[unsafe(no_mangle)] #[cfg(feature = "threads")] pub extern "C" fn wasmtime_config_wasm_threads_set(c: &mut wasm_config_t, enable: bool) { c.config.wasm_threads(enable); } #[unsafe(no_mangle)] pub extern "C" fn wasmtime_config_shared_memory_set(c: &mut wasm_config_t, enable: bool) { c.config.shared_memory(enable); } #[unsafe(no_mangle)] pub extern "C" fn wasmtime_config_wasm_tail_call_set(c: &mut wasm_config_t, enable: bool) { c.config.wasm_tail_call(enable); } #[unsafe(no_mangle)] pub extern "C" fn wasmtime_config_wasm_reference_types_set(c: &mut wasm_config_t, enable: bool) { c.config.wasm_reference_types(enable); } #[unsafe(no_mangle)] pub extern "C" fn wasmtime_config_wasm_function_references_set( c: &mut wasm_config_t, enable: bool, ) { c.config.wasm_function_references(enable); } #[unsafe(no_mangle)] pub extern "C" fn wasmtime_config_wasm_gc_set(c: &mut wasm_config_t, enable: bool) { c.config.wasm_gc(enable); } #[unsafe(no_mangle)] pub extern "C" fn wasmtime_config_gc_support_set(c: &mut wasm_config_t, enable: bool) { c.config.gc_support(enable); } #[unsafe(no_mangle)] pub extern "C" fn wasmtime_config_wasm_simd_set(c: &mut wasm_config_t, enable: bool) { c.config.wasm_simd(enable); } #[unsafe(no_mangle)] pub extern "C" fn wasmtime_config_wasm_relaxed_simd_set(c: &mut wasm_config_t, enable: bool) { c.config.wasm_relaxed_simd(enable); } #[unsafe(no_mangle)] pub extern "C" fn wasmtime_config_wasm_relaxed_simd_deterministic_set( c: &mut wasm_config_t, enable: bool, ) { c.config.relaxed_simd_deterministic(enable); } #[unsafe(no_mangle)] pub extern "C" fn wasmtime_config_wasm_bulk_memory_set(c: &mut wasm_config_t, enable: bool) { c.config.wasm_bulk_memory(enable); } #[unsafe(no_mangle)] pub extern "C" fn wasmtime_config_wasm_multi_value_set(c: &mut wasm_config_t, enable: bool) { c.config.wasm_multi_value(enable); } #[unsafe(no_mangle)] pub extern "C" fn wasmtime_config_wasm_multi_memory_set(c: &mut wasm_config_t, enable: bool) { c.config.wasm_multi_memory(enable); } #[unsafe(no_mangle)] pub extern "C" fn wasmtime_config_wasm_memory64_set(c: &mut wasm_config_t, enable: bool) { c.config.wasm_memory64(enable); } #[unsafe(no_mangle)] pub extern "C" fn wasmtime_config_wasm_stack_switching_set(c: &mut wasm_config_t, enable: bool) { c.config.wasm_stack_switching(enable); } #[unsafe(no_mangle)] #[cfg(any(feature = "cranelift", feature = "winch"))] pub extern "C" fn wasmtime_config_strategy_set( c: &mut wasm_config_t, strategy: wasmtime_strategy_t, ) { use wasmtime_strategy_t::*; c.config.strategy(match strategy { WASMTIME_STRATEGY_AUTO => Strategy::Auto, WASMTIME_STRATEGY_CRANELIFT => Strategy::Cranelift, }); } #[unsafe(no_mangle)] #[cfg(feature = "parallel-compilation")] pub extern "C" fn wasmtime_config_parallel_compilation_set(c: &mut wasm_config_t, enable: bool) { c.config.parallel_compilation(enable); } #[unsafe(no_mangle)] #[cfg(any(feature = "cranelift", feature = "winch"))] pub extern "C" fn wasmtime_config_cranelift_debug_verifier_set( c: &mut wasm_config_t, enable: bool, ) { c.config.cranelift_debug_verifier(enable); } #[unsafe(no_mangle)] #[cfg(any(feature = "cranelift", feature = "winch"))] pub extern "C" fn wasmtime_config_cranelift_nan_canonicalization_set( c: &mut wasm_config_t, enable: bool, ) { c.config.cranelift_nan_canonicalization(enable); } #[unsafe(no_mangle)] #[cfg(any(feature = "cranelift", feature = "winch"))] pub extern "C" fn wasmtime_config_cranelift_opt_level_set( c: &mut wasm_config_t, opt_level: wasmtime_opt_level_t, ) { use wasmtime_opt_level_t::*; c.config.cranelift_opt_level(match opt_level { WASMTIME_OPT_LEVEL_NONE => OptLevel::None, WASMTIME_OPT_LEVEL_SPEED => OptLevel::Speed, WASMTIME_OPT_LEVEL_SPEED_AND_SIZE => OptLevel::SpeedAndSize, }); } #[unsafe(no_mangle)] pub extern "C" fn wasmtime_config_profiler_set( c: &mut wasm_config_t, strategy: wasmtime_profiling_strategy_t, ) { use wasmtime_profiling_strategy_t::*; c.config.profiler(match strategy { WASMTIME_PROFILING_STRATEGY_NONE => ProfilingStrategy::None, WASMTIME_PROFILING_STRATEGY_JITDUMP => ProfilingStrategy::JitDump, WASMTIME_PROFILING_STRATEGY_VTUNE => ProfilingStrategy::VTune, WASMTIME_PROFILING_STRATEGY_PERFMAP => ProfilingStrategy::PerfMap, }); } #[unsafe(no_mangle)] #[cfg(feature = "cache")] pub unsafe extern "C" fn wasmtime_config_cache_config_load( c: &mut wasm_config_t, filename: *const c_char, ) -> Option> { use std::path::Path; use wasmtime::Cache; handle_result( if filename.is_null() { Cache::from_file(None).map(|cache| c.config.cache(Some(cache))) } else { match CStr::from_ptr(filename).to_str() { Ok(s) => { Cache::from_file(Some(&Path::new(s))).map(|cache| c.config.cache(Some(cache))) } Err(e) => Err(e.into()), } }, |_cfg| {}, ) } #[unsafe(no_mangle)] pub extern "C" fn wasmtime_config_memory_may_move_set(c: &mut wasm_config_t, enable: bool) { c.config.memory_may_move(enable); } #[unsafe(no_mangle)] pub extern "C" fn wasmtime_config_memory_reservation_set(c: &mut wasm_config_t, size: u64) { c.config.memory_reservation(size); } #[unsafe(no_mangle)] pub extern "C" fn wasmtime_config_memory_guard_size_set(c: &mut wasm_config_t, size: u64) { c.config.memory_guard_size(size); } #[unsafe(no_mangle)] pub extern "C" fn wasmtime_config_memory_reservation_for_growth_set( c: &mut wasm_config_t, size: u64, ) { c.config.memory_reservation_for_growth(size); } #[unsafe(no_mangle)] pub extern "C" fn wasmtime_config_native_unwind_info_set(c: &mut wasm_config_t, enabled: bool) { c.config.native_unwind_info(enabled); } #[unsafe(no_mangle)] pub unsafe extern "C" fn wasmtime_config_target_set( c: &mut wasm_config_t, target: *const c_char, ) -> Option> { let target = CStr::from_ptr(target).to_str().expect("not valid utf-8"); handle_result(c.config.target(target), |_cfg| {}) } #[unsafe(no_mangle)] pub extern "C" fn wasmtime_config_macos_use_mach_ports_set(c: &mut wasm_config_t, enabled: bool) { c.config.macos_use_mach_ports(enabled); } #[unsafe(no_mangle)] pub extern "C" fn wasmtime_config_signals_based_traps_set(c: &mut wasm_config_t, enable: bool) { c.config.signals_based_traps(enable); } #[unsafe(no_mangle)] #[cfg(any(feature = "cranelift", feature = "winch"))] pub unsafe extern "C" fn wasmtime_config_cranelift_flag_enable( c: &mut wasm_config_t, flag: *const c_char, ) { let flag = CStr::from_ptr(flag).to_str().expect("not valid utf-8"); c.config.cranelift_flag_enable(flag); } #[unsafe(no_mangle)] #[cfg(any(feature = "cranelift", feature = "winch"))] pub unsafe extern "C" fn wasmtime_config_cranelift_flag_set( c: &mut wasm_config_t, flag: *const c_char, value: *const c_char, ) { let flag = CStr::from_ptr(flag).to_str().expect("not valid utf-8"); let value = CStr::from_ptr(value).to_str().expect("not valid utf-8"); c.config.cranelift_flag_set(flag, value); } pub type wasmtime_memory_get_callback_t = extern "C" fn( env: *mut std::ffi::c_void, byte_size: &mut usize, maximum_byte_size: &mut usize, ) -> *mut u8; pub type wasmtime_memory_grow_callback_t = extern "C" fn(env: *mut std::ffi::c_void, new_size: usize) -> Option>; #[repr(C)] pub struct wasmtime_linear_memory_t { env: *mut std::ffi::c_void, get_memory: wasmtime_memory_get_callback_t, grow_memory: wasmtime_memory_grow_callback_t, finalizer: Option, } pub type wasmtime_new_memory_callback_t = extern "C" fn( env: *mut std::ffi::c_void, ty: &wasm_memorytype_t, minimum: usize, maximum: usize, reserved_size_in_bytes: usize, guard_size_in_bytes: usize, memory_ret: *mut wasmtime_linear_memory_t, ) -> Option>; struct CHostLinearMemory { foreign: crate::ForeignData, get_memory: wasmtime_memory_get_callback_t, grow_memory: wasmtime_memory_grow_callback_t, } unsafe impl LinearMemory for CHostLinearMemory { fn byte_size(&self) -> usize { let mut byte_size = 0; let mut byte_capacity = 0; let cb = self.get_memory; cb(self.foreign.data, &mut byte_size, &mut byte_capacity); return byte_size; } fn byte_capacity(&self) -> usize { let mut byte_size = 0; let mut byte_capacity = 0; let cb = self.get_memory; cb(self.foreign.data, &mut byte_size, &mut byte_capacity); byte_capacity } fn as_ptr(&self) -> *mut u8 { let mut byte_size = 0; let mut byte_capacity = 0; let cb = self.get_memory; cb(self.foreign.data, &mut byte_size, &mut byte_capacity) } fn grow_to(&mut self, new_size: usize) -> Result<()> { let cb = self.grow_memory; let error = cb(self.foreign.data, new_size); if let Some(err) = error { Err((*err).into()) } else { Ok(()) } } } #[repr(C)] pub struct wasmtime_memory_creator_t { env: *mut std::ffi::c_void, new_memory: wasmtime_new_memory_callback_t, finalizer: Option, } struct CHostMemoryCreator { foreign: crate::ForeignData, new_memory: wasmtime_new_memory_callback_t, } unsafe impl Send for CHostMemoryCreator {} unsafe impl Sync for CHostMemoryCreator {} unsafe impl MemoryCreator for CHostMemoryCreator { fn new_memory( &self, ty: wasmtime::MemoryType, minimum: usize, maximum: Option, reserved_size_in_bytes: Option, guard_size_in_bytes: usize, ) -> Result, String> { extern "C" fn panic_get_callback( _env: *mut std::ffi::c_void, _byte_size: &mut usize, _maximum_byte_size: &mut usize, ) -> *mut u8 { panic!("a callback must be set"); } extern "C" fn panic_grow_callback( _env: *mut std::ffi::c_void, _size: usize, ) -> Option> { panic!("a callback must be set"); } let mut memory = wasmtime_linear_memory_t { env: ptr::null_mut(), get_memory: panic_get_callback, grow_memory: panic_grow_callback, finalizer: None, }; let cb = self.new_memory; let error = cb( self.foreign.data, &wasm_memorytype_t::new(ty), minimum, maximum.unwrap_or(usize::MAX), reserved_size_in_bytes.unwrap_or(0), guard_size_in_bytes, &mut memory, ); match error { None => { let foreign = crate::ForeignData { data: memory.env, finalizer: memory.finalizer, }; Ok(Box::new(CHostLinearMemory { foreign, get_memory: memory.get_memory, grow_memory: memory.grow_memory, })) } Some(err) => { let err: wasmtime::Error = (*err).into(); Err(format!("{err}")) } } } } #[unsafe(no_mangle)] pub unsafe extern "C" fn wasmtime_config_host_memory_creator_set( c: &mut wasm_config_t, creator: &wasmtime_memory_creator_t, ) { c.config.with_host_memory(Arc::new(CHostMemoryCreator { foreign: crate::ForeignData { data: creator.env, finalizer: creator.finalizer, }, new_memory: creator.new_memory, })); } #[unsafe(no_mangle)] pub extern "C" fn wasmtime_config_memory_init_cow_set(c: &mut wasm_config_t, enable: bool) { c.config.memory_init_cow(enable); } #[unsafe(no_mangle)] pub extern "C" fn wasmtime_config_wasm_wide_arithmetic_set(c: &mut wasm_config_t, enable: bool) { c.config.wasm_wide_arithmetic(enable); } #[unsafe(no_mangle)] pub extern "C" fn wasmtime_config_wasm_exceptions_set(c: &mut wasm_config_t, enable: bool) { c.config.wasm_exceptions(enable); } #[unsafe(no_mangle)] pub extern "C" fn wasmtime_config_wasm_custom_page_sizes_set(c: &mut wasm_config_t, enable: bool) { c.config.wasm_custom_page_sizes(enable); } #[repr(C)] #[derive(Clone)] #[cfg(feature = "pooling-allocator")] pub struct wasmtime_pooling_allocation_config_t { pub(crate) config: PoolingAllocationConfig, } #[unsafe(no_mangle)] #[cfg(feature = "pooling-allocator")] pub extern "C" fn wasmtime_pooling_allocation_config_new() -> Box { Box::new(wasmtime_pooling_allocation_config_t { config: PoolingAllocationConfig::default(), }) } #[unsafe(no_mangle)] #[cfg(feature = "pooling-allocator")] pub extern "C" fn wasmtime_pooling_allocation_config_delete( _: Box, ) { } #[unsafe(no_mangle)] #[cfg(feature = "pooling-allocator")] pub extern "C" fn wasmtime_pooling_allocation_config_max_unused_warm_slots_set( c: &mut wasmtime_pooling_allocation_config_t, max: u32, ) { c.config.max_unused_warm_slots(max); } #[unsafe(no_mangle)] #[cfg(feature = "pooling-allocator")] pub extern "C" fn wasmtime_pooling_allocation_config_decommit_batch_size_set( c: &mut wasmtime_pooling_allocation_config_t, batch_size: usize, ) { c.config.decommit_batch_size(batch_size); } #[unsafe(no_mangle)] #[cfg(all(feature = "pooling-allocator", feature = "async"))] pub extern "C" fn wasmtime_pooling_allocation_config_async_stack_keep_resident_set( c: &mut wasmtime_pooling_allocation_config_t, size: usize, ) { c.config.async_stack_keep_resident(size); } #[unsafe(no_mangle)] #[cfg(feature = "pooling-allocator")] pub extern "C" fn wasmtime_pooling_allocation_config_linear_memory_keep_resident_set( c: &mut wasmtime_pooling_allocation_config_t, size: usize, ) { c.config.linear_memory_keep_resident(size); } #[unsafe(no_mangle)] #[cfg(feature = "pooling-allocator")] pub extern "C" fn wasmtime_pooling_allocation_config_table_keep_resident_set( c: &mut wasmtime_pooling_allocation_config_t, size: usize, ) { c.config.table_keep_resident(size); } #[unsafe(no_mangle)] #[cfg(feature = "pooling-allocator")] pub extern "C" fn wasmtime_pooling_allocation_config_total_component_instances_set( c: &mut wasmtime_pooling_allocation_config_t, count: u32, ) { c.config.total_component_instances(count); } #[unsafe(no_mangle)] #[cfg(feature = "pooling-allocator")] pub extern "C" fn wasmtime_pooling_allocation_config_max_component_instance_size_set( c: &mut wasmtime_pooling_allocation_config_t, size: usize, ) { c.config.max_component_instance_size(size); } #[unsafe(no_mangle)] #[cfg(feature = "pooling-allocator")] pub extern "C" fn wasmtime_pooling_allocation_config_max_core_instances_per_component_set( c: &mut wasmtime_pooling_allocation_config_t, count: u32, ) { c.config.max_core_instances_per_component(count); } #[unsafe(no_mangle)] #[cfg(feature = "pooling-allocator")] pub extern "C" fn wasmtime_pooling_allocation_config_max_memories_per_component_set( c: &mut wasmtime_pooling_allocation_config_t, count: u32, ) { c.config.max_memories_per_component(count); } #[unsafe(no_mangle)] #[cfg(feature = "pooling-allocator")] pub extern "C" fn wasmtime_pooling_allocation_config_max_tables_per_component_set( c: &mut wasmtime_pooling_allocation_config_t, count: u32, ) { c.config.max_tables_per_component(count); } #[unsafe(no_mangle)] #[cfg(feature = "pooling-allocator")] pub extern "C" fn wasmtime_pooling_allocation_config_total_memories_set( c: &mut wasmtime_pooling_allocation_config_t, count: u32, ) { c.config.total_memories(count); } #[unsafe(no_mangle)] #[cfg(feature = "pooling-allocator")] pub extern "C" fn wasmtime_pooling_allocation_config_total_tables_set( c: &mut wasmtime_pooling_allocation_config_t, count: u32, ) { c.config.total_tables(count); } #[unsafe(no_mangle)] #[cfg(all(feature = "pooling-allocator", feature = "async"))] pub extern "C" fn wasmtime_pooling_allocation_config_total_stacks_set( c: &mut wasmtime_pooling_allocation_config_t, count: u32, ) { c.config.total_stacks(count); } #[unsafe(no_mangle)] #[cfg(feature = "pooling-allocator")] pub extern "C" fn wasmtime_pooling_allocation_config_total_core_instances_set( c: &mut wasmtime_pooling_allocation_config_t, count: u32, ) { c.config.total_core_instances(count); } #[unsafe(no_mangle)] #[cfg(feature = "pooling-allocator")] pub extern "C" fn wasmtime_pooling_allocation_config_max_core_instance_size_set( c: &mut wasmtime_pooling_allocation_config_t, size: usize, ) { c.config.max_core_instance_size(size); } #[unsafe(no_mangle)] #[cfg(feature = "pooling-allocator")] pub extern "C" fn wasmtime_pooling_allocation_config_max_tables_per_module_set( c: &mut wasmtime_pooling_allocation_config_t, tables: u32, ) { c.config.max_tables_per_module(tables); } #[unsafe(no_mangle)] #[cfg(feature = "pooling-allocator")] pub extern "C" fn wasmtime_pooling_allocation_config_table_elements_set( c: &mut wasmtime_pooling_allocation_config_t, elements: usize, ) { c.config.table_elements(elements); } #[unsafe(no_mangle)] #[cfg(feature = "pooling-allocator")] pub extern "C" fn wasmtime_pooling_allocation_config_max_memories_per_module_set( c: &mut wasmtime_pooling_allocation_config_t, memories: u32, ) { c.config.max_memories_per_module(memories); } #[unsafe(no_mangle)] #[cfg(feature = "pooling-allocator")] pub extern "C" fn wasmtime_pooling_allocation_config_max_memory_size_set( c: &mut wasmtime_pooling_allocation_config_t, bytes: usize, ) { c.config.max_memory_size(bytes); } #[unsafe(no_mangle)] #[cfg(all(feature = "pooling-allocator", feature = "gc"))] pub extern "C" fn wasmtime_pooling_allocation_config_total_gc_heaps_set( c: &mut wasmtime_pooling_allocation_config_t, count: u32, ) { c.config.total_gc_heaps(count); } #[unsafe(no_mangle)] #[cfg(feature = "pooling-allocator")] pub extern "C" fn wasmtime_pooling_allocation_strategy_set( c: &mut wasm_config_t, pc: &wasmtime_pooling_allocation_config_t, ) { c.config .allocation_strategy(InstanceAllocationStrategy::Pooling(pc.config.clone())); } #[unsafe(no_mangle)] pub extern "C" fn wasmtime_config_concurrency_support_set(c: &mut wasm_config_t, enable: bool) { c.config.concurrency_support(enable); }