1use anyhow::{Context, Result};
4use clap::Parser;
5use serde::Deserialize;
6use std::{
7 fmt, fs,
8 path::{Path, PathBuf},
9 time::Duration,
10};
11use wasmtime::Config;
12
13pub mod opt;
14
15#[cfg(feature = "logging")]
16fn init_file_per_thread_logger(prefix: &'static str) {
17 file_per_thread_logger::initialize(prefix);
18 file_per_thread_logger::allow_uninitialized();
19
20 #[cfg(feature = "parallel-compilation")]
25 rayon::ThreadPoolBuilder::new()
26 .spawn_handler(move |thread| {
27 let mut b = std::thread::Builder::new();
28 if let Some(name) = thread.name() {
29 b = b.name(name.to_owned());
30 }
31 if let Some(stack_size) = thread.stack_size() {
32 b = b.stack_size(stack_size);
33 }
34 b.spawn(move || {
35 file_per_thread_logger::initialize(prefix);
36 thread.run()
37 })?;
38 Ok(())
39 })
40 .build_global()
41 .unwrap();
42}
43
44wasmtime_option_group! {
45 #[derive(PartialEq, Clone, Deserialize)]
46 #[serde(rename_all = "kebab-case", deny_unknown_fields)]
47 pub struct OptimizeOptions {
48 #[serde(default)]
50 #[serde(deserialize_with = "crate::opt::cli_parse_wrapper")]
51 pub opt_level: Option<wasmtime::OptLevel>,
52
53 #[serde(default)]
55 #[serde(deserialize_with = "crate::opt::cli_parse_wrapper")]
56 pub regalloc_algorithm: Option<wasmtime::RegallocAlgorithm>,
57
58 pub memory_may_move: Option<bool>,
61
62 pub memory_reservation: Option<u64>,
64
65 pub memory_reservation_for_growth: Option<u64>,
67
68 pub memory_guard_size: Option<u64>,
70
71 pub guard_before_linear_memory: Option<bool>,
74
75 pub table_lazy_init: Option<bool>,
80
81 pub pooling_allocator: Option<bool>,
83
84 pub pooling_decommit_batch_size: Option<usize>,
87
88 pub pooling_memory_keep_resident: Option<usize>,
91
92 pub pooling_table_keep_resident: Option<usize>,
95
96 #[serde(default)]
99 #[serde(deserialize_with = "crate::opt::cli_parse_wrapper")]
100 pub pooling_memory_protection_keys: Option<wasmtime::MpkEnabled>,
101
102 pub pooling_max_memory_protection_keys: Option<usize>,
105
106 pub memory_init_cow: Option<bool>,
109
110 pub memory_guaranteed_dense_image_size: Option<u64>,
113
114 pub pooling_total_core_instances: Option<u32>,
117
118 pub pooling_total_component_instances: Option<u32>,
121
122 pub pooling_total_memories: Option<u32>,
125
126 pub pooling_total_tables: Option<u32>,
129
130 pub pooling_total_stacks: Option<u32>,
133
134 pub pooling_max_memory_size: Option<usize>,
137
138 pub pooling_table_elements: Option<usize>,
141
142 pub pooling_max_core_instance_size: Option<usize>,
145
146 pub pooling_max_unused_warm_slots: Option<u32>,
149
150 pub pooling_async_stack_keep_resident: Option<usize>,
153
154 pub pooling_max_component_instance_size: Option<usize>,
157
158 pub pooling_max_core_instances_per_component: Option<u32>,
161
162 pub pooling_max_memories_per_component: Option<u32>,
165
166 pub pooling_max_tables_per_component: Option<u32>,
169
170 pub pooling_max_tables_per_module: Option<u32>,
172
173 pub pooling_max_memories_per_module: Option<u32>,
175
176 pub pooling_total_gc_heaps: Option<u32>,
178
179 pub signals_based_traps: Option<bool>,
181
182 pub dynamic_memory_guard_size: Option<u64>,
184
185 pub static_memory_guard_size: Option<u64>,
187
188 pub static_memory_forced: Option<bool>,
190
191 pub static_memory_maximum_size: Option<u64>,
193
194 pub dynamic_memory_reserved_for_growth: Option<u64>,
196 }
197
198 enum Optimize {
199 ...
200 }
201}
202
203wasmtime_option_group! {
204 #[derive(PartialEq, Clone, Deserialize)]
205 #[serde(rename_all = "kebab-case", deny_unknown_fields)]
206 pub struct CodegenOptions {
207 #[serde(default)]
212 #[serde(deserialize_with = "crate::opt::cli_parse_wrapper")]
213 pub compiler: Option<wasmtime::Strategy>,
214 #[serde(default)]
224 #[serde(deserialize_with = "crate::opt::cli_parse_wrapper")]
225 pub collector: Option<wasmtime::Collector>,
226 pub cranelift_debug_verifier: Option<bool>,
228 pub cache: Option<bool>,
230 pub cache_config: Option<String>,
232 pub parallel_compilation: Option<bool>,
234 pub pcc: Option<bool>,
236 pub native_unwind_info: Option<bool>,
239
240 #[prefixed = "cranelift"]
241 #[serde(default)]
242 pub cranelift: Vec<(String, Option<String>)>,
245 }
246
247 enum Codegen {
248 ...
249 }
250}
251
252wasmtime_option_group! {
253 #[derive(PartialEq, Clone, Deserialize)]
254 #[serde(rename_all = "kebab-case", deny_unknown_fields)]
255 pub struct DebugOptions {
256 pub debug_info: Option<bool>,
258 pub address_map: Option<bool>,
260 pub logging: Option<bool>,
262 pub log_to_files: Option<bool>,
264 pub coredump: Option<String>,
266 }
267
268 enum Debug {
269 ...
270 }
271}
272
273wasmtime_option_group! {
274 #[derive(PartialEq, Clone, Deserialize)]
275 #[serde(rename_all = "kebab-case", deny_unknown_fields)]
276 pub struct WasmOptions {
277 pub nan_canonicalization: Option<bool>,
279 pub fuel: Option<u64>,
287 pub epoch_interruption: Option<bool>,
290 pub max_wasm_stack: Option<usize>,
293 pub async_stack_size: Option<usize>,
299 pub async_stack_zeroing: Option<bool>,
302 pub unknown_exports_allow: Option<bool>,
304 pub unknown_imports_trap: Option<bool>,
307 pub unknown_imports_default: Option<bool>,
310 pub wmemcheck: Option<bool>,
312 pub max_memory_size: Option<usize>,
317 pub max_table_elements: Option<usize>,
319 pub max_instances: Option<usize>,
321 pub max_tables: Option<usize>,
323 pub max_memories: Option<usize>,
325 pub trap_on_grow_failure: Option<bool>,
332 pub timeout: Option<Duration>,
334 pub all_proposals: Option<bool>,
336 pub bulk_memory: Option<bool>,
338 pub multi_memory: Option<bool>,
340 pub multi_value: Option<bool>,
342 pub reference_types: Option<bool>,
344 pub simd: Option<bool>,
346 pub relaxed_simd: Option<bool>,
348 pub relaxed_simd_deterministic: Option<bool>,
357 pub tail_call: Option<bool>,
359 pub threads: Option<bool>,
361 pub shared_everything_threads: Option<bool>,
363 pub memory64: Option<bool>,
365 pub component_model: Option<bool>,
367 pub component_model_async: Option<bool>,
369 pub component_model_async_builtins: Option<bool>,
372 pub component_model_async_stackful: Option<bool>,
375 pub component_model_error_context: Option<bool>,
378 pub function_references: Option<bool>,
380 pub gc: Option<bool>,
382 pub custom_page_sizes: Option<bool>,
384 pub wide_arithmetic: Option<bool>,
386 pub extended_const: Option<bool>,
388 pub exceptions: Option<bool>,
390 pub legacy_exceptions: Option<bool>,
392 }
393
394 enum Wasm {
395 ...
396 }
397}
398
399wasmtime_option_group! {
400 #[derive(PartialEq, Clone, Deserialize)]
401 #[serde(rename_all = "kebab-case", deny_unknown_fields)]
402 pub struct WasiOptions {
403 pub cli: Option<bool>,
405 pub cli_exit_with_code: Option<bool>,
407 pub common: Option<bool>,
409 pub nn: Option<bool>,
411 pub threads: Option<bool>,
413 pub http: Option<bool>,
415 pub http_outgoing_body_buffer_chunks: Option<usize>,
419 pub http_outgoing_body_chunk_size: Option<usize>,
422 pub config: Option<bool>,
424 pub keyvalue: Option<bool>,
426 pub listenfd: Option<bool>,
430 #[serde(default)]
433 pub tcplisten: Vec<String>,
434 pub tls: Option<bool>,
436 pub preview2: Option<bool>,
439 #[serde(skip)]
448 pub nn_graph: Vec<WasiNnGraph>,
449 pub inherit_network: Option<bool>,
452 pub allow_ip_name_lookup: Option<bool>,
454 pub tcp: Option<bool>,
456 pub udp: Option<bool>,
458 pub network_error_code: Option<bool>,
460 pub preview0: Option<bool>,
462 pub inherit_env: Option<bool>,
466 #[serde(skip)]
468 pub config_var: Vec<KeyValuePair>,
469 #[serde(skip)]
471 pub keyvalue_in_memory_data: Vec<KeyValuePair>,
472 }
473
474 enum Wasi {
475 ...
476 }
477}
478
479#[derive(Debug, Clone, PartialEq)]
480pub struct WasiNnGraph {
481 pub format: String,
482 pub dir: String,
483}
484
485#[derive(Debug, Clone, PartialEq)]
486pub struct KeyValuePair {
487 pub key: String,
488 pub value: String,
489}
490
491#[derive(Parser, Clone, Deserialize)]
493#[serde(deny_unknown_fields)]
494pub struct CommonOptions {
495 #[arg(short = 'O', long = "optimize", value_name = "KEY[=VAL[,..]]")]
505 #[serde(skip)]
506 opts_raw: Vec<opt::CommaSeparated<Optimize>>,
507
508 #[arg(short = 'C', long = "codegen", value_name = "KEY[=VAL[,..]]")]
510 #[serde(skip)]
511 codegen_raw: Vec<opt::CommaSeparated<Codegen>>,
512
513 #[arg(short = 'D', long = "debug", value_name = "KEY[=VAL[,..]]")]
515 #[serde(skip)]
516 debug_raw: Vec<opt::CommaSeparated<Debug>>,
517
518 #[arg(short = 'W', long = "wasm", value_name = "KEY[=VAL[,..]]")]
521 #[serde(skip)]
522 wasm_raw: Vec<opt::CommaSeparated<Wasm>>,
523
524 #[arg(short = 'S', long = "wasi", value_name = "KEY[=VAL[,..]]")]
526 #[serde(skip)]
527 wasi_raw: Vec<opt::CommaSeparated<Wasi>>,
528
529 #[arg(skip)]
532 #[serde(skip)]
533 configured: bool,
534
535 #[arg(skip)]
536 #[serde(rename = "optimize", default)]
537 pub opts: OptimizeOptions,
538
539 #[arg(skip)]
540 #[serde(rename = "codegen", default)]
541 pub codegen: CodegenOptions,
542
543 #[arg(skip)]
544 #[serde(rename = "debug", default)]
545 pub debug: DebugOptions,
546
547 #[arg(skip)]
548 #[serde(rename = "wasm", default)]
549 pub wasm: WasmOptions,
550
551 #[arg(skip)]
552 #[serde(rename = "wasi", default)]
553 pub wasi: WasiOptions,
554
555 #[arg(long, value_name = "TARGET")]
557 #[serde(skip)]
558 pub target: Option<String>,
559
560 #[arg(long = "config", value_name = "FILE")]
567 #[serde(skip)]
568 pub config: Option<PathBuf>,
569}
570
571macro_rules! match_feature {
572 (
573 [$feat:tt : $config:expr]
574 $val:ident => $e:expr,
575 $p:pat => err,
576 ) => {
577 #[cfg(feature = $feat)]
578 {
579 if let Some($val) = $config {
580 $e;
581 }
582 }
583 #[cfg(not(feature = $feat))]
584 {
585 if let Some($p) = $config {
586 anyhow::bail!(concat!("support for ", $feat, " disabled at compile time"));
587 }
588 }
589 };
590}
591
592impl CommonOptions {
593 pub fn new() -> CommonOptions {
595 CommonOptions {
596 opts_raw: Vec::new(),
597 codegen_raw: Vec::new(),
598 debug_raw: Vec::new(),
599 wasm_raw: Vec::new(),
600 wasi_raw: Vec::new(),
601 configured: true,
602 opts: Default::default(),
603 codegen: Default::default(),
604 debug: Default::default(),
605 wasm: Default::default(),
606 wasi: Default::default(),
607 target: None,
608 config: None,
609 }
610 }
611
612 fn configure(&mut self) -> Result<()> {
613 if self.configured {
614 return Ok(());
615 }
616 self.configured = true;
617 if let Some(toml_config_path) = &self.config {
618 let toml_options = CommonOptions::from_file(toml_config_path)?;
619 self.opts = toml_options.opts;
620 self.codegen = toml_options.codegen;
621 self.debug = toml_options.debug;
622 self.wasm = toml_options.wasm;
623 self.wasi = toml_options.wasi;
624 }
625 self.opts.configure_with(&self.opts_raw);
626 self.codegen.configure_with(&self.codegen_raw);
627 self.debug.configure_with(&self.debug_raw);
628 self.wasm.configure_with(&self.wasm_raw);
629 self.wasi.configure_with(&self.wasi_raw);
630 Ok(())
631 }
632
633 pub fn init_logging(&mut self) -> Result<()> {
634 self.configure()?;
635 if self.debug.logging == Some(false) {
636 return Ok(());
637 }
638 #[cfg(feature = "logging")]
639 if self.debug.log_to_files == Some(true) {
640 let prefix = "wasmtime.dbg.";
641 init_file_per_thread_logger(prefix);
642 } else {
643 use std::io::IsTerminal;
644 use tracing_subscriber::{EnvFilter, FmtSubscriber};
645 let builder = FmtSubscriber::builder()
646 .with_writer(std::io::stderr)
647 .with_env_filter(EnvFilter::from_env("WASMTIME_LOG"))
648 .with_ansi(std::io::stderr().is_terminal());
649 if std::env::var("WASMTIME_LOG_NO_CONTEXT").is_ok_and(|value| value.eq("1")) {
650 builder
651 .with_level(false)
652 .with_target(false)
653 .without_time()
654 .init()
655 } else {
656 builder.init();
657 }
658 }
659 #[cfg(not(feature = "logging"))]
660 if self.debug.log_to_files == Some(true) || self.debug.logging == Some(true) {
661 anyhow::bail!("support for logging disabled at compile time");
662 }
663 Ok(())
664 }
665
666 pub fn config(&mut self, pooling_allocator_default: Option<bool>) -> Result<Config> {
667 self.configure()?;
668 let mut config = Config::new();
669
670 match_feature! {
671 ["cranelift" : self.codegen.compiler]
672 strategy => config.strategy(strategy),
673 _ => err,
674 }
675 match_feature! {
676 ["gc" : self.codegen.collector]
677 collector => config.collector(collector),
678 _ => err,
679 }
680 if let Some(target) = &self.target {
681 config.target(target)?;
682 }
683 match_feature! {
684 ["cranelift" : self.codegen.cranelift_debug_verifier]
685 enable => config.cranelift_debug_verifier(enable),
686 true => err,
687 }
688 if let Some(enable) = self.debug.debug_info {
689 config.debug_info(enable);
690 }
691 if self.debug.coredump.is_some() {
692 #[cfg(feature = "coredump")]
693 config.coredump_on_trap(true);
694 #[cfg(not(feature = "coredump"))]
695 anyhow::bail!("support for coredumps disabled at compile time");
696 }
697 match_feature! {
698 ["cranelift" : self.opts.opt_level]
699 level => config.cranelift_opt_level(level),
700 _ => err,
701 }
702 match_feature! {
703 ["cranelift": self.opts.regalloc_algorithm]
704 algo => config.cranelift_regalloc_algorithm(algo),
705 _ => err,
706 }
707 match_feature! {
708 ["cranelift" : self.wasm.nan_canonicalization]
709 enable => config.cranelift_nan_canonicalization(enable),
710 true => err,
711 }
712 match_feature! {
713 ["cranelift" : self.codegen.pcc]
714 enable => config.cranelift_pcc(enable),
715 true => err,
716 }
717
718 self.enable_wasm_features(&mut config)?;
719
720 #[cfg(feature = "cranelift")]
721 for (name, value) in self.codegen.cranelift.iter() {
722 let name = name.replace('-', "_");
723 unsafe {
724 match value {
725 Some(val) => {
726 config.cranelift_flag_set(&name, val);
727 }
728 None => {
729 config.cranelift_flag_enable(&name);
730 }
731 }
732 }
733 }
734 #[cfg(not(feature = "cranelift"))]
735 if !self.codegen.cranelift.is_empty() {
736 anyhow::bail!("support for cranelift disabled at compile time");
737 }
738
739 #[cfg(feature = "cache")]
740 if self.codegen.cache != Some(false) {
741 use wasmtime::Cache;
742 let cache = match &self.codegen.cache_config {
743 Some(path) => Cache::from_file(Some(Path::new(path)))?,
744 None => Cache::from_file(None)?,
745 };
746 config.cache(Some(cache));
747 }
748 #[cfg(not(feature = "cache"))]
749 if self.codegen.cache == Some(true) {
750 anyhow::bail!("support for caching disabled at compile time");
751 }
752
753 match_feature! {
754 ["parallel-compilation" : self.codegen.parallel_compilation]
755 enable => config.parallel_compilation(enable),
756 true => err,
757 }
758
759 let memory_reservation = self
760 .opts
761 .memory_reservation
762 .or(self.opts.static_memory_maximum_size);
763 if let Some(size) = memory_reservation {
764 config.memory_reservation(size);
765 }
766
767 if let Some(enable) = self.opts.static_memory_forced {
768 config.memory_may_move(!enable);
769 }
770 if let Some(enable) = self.opts.memory_may_move {
771 config.memory_may_move(enable);
772 }
773
774 let memory_guard_size = self
775 .opts
776 .static_memory_guard_size
777 .or(self.opts.dynamic_memory_guard_size)
778 .or(self.opts.memory_guard_size);
779 if let Some(size) = memory_guard_size {
780 config.memory_guard_size(size);
781 }
782
783 let mem_for_growth = self
784 .opts
785 .memory_reservation_for_growth
786 .or(self.opts.dynamic_memory_reserved_for_growth);
787 if let Some(size) = mem_for_growth {
788 config.memory_reservation_for_growth(size);
789 }
790 if let Some(enable) = self.opts.guard_before_linear_memory {
791 config.guard_before_linear_memory(enable);
792 }
793 if let Some(enable) = self.opts.table_lazy_init {
794 config.table_lazy_init(enable);
795 }
796
797 if self.wasm.fuel.is_some() {
799 config.consume_fuel(true);
800 }
801
802 if let Some(enable) = self.wasm.epoch_interruption {
803 config.epoch_interruption(enable);
804 }
805 if let Some(enable) = self.debug.address_map {
806 config.generate_address_map(enable);
807 }
808 if let Some(enable) = self.opts.memory_init_cow {
809 config.memory_init_cow(enable);
810 }
811 if let Some(size) = self.opts.memory_guaranteed_dense_image_size {
812 config.memory_guaranteed_dense_image_size(size);
813 }
814 if let Some(enable) = self.opts.signals_based_traps {
815 config.signals_based_traps(enable);
816 }
817 if let Some(enable) = self.codegen.native_unwind_info {
818 config.native_unwind_info(enable);
819 }
820
821 match_feature! {
822 ["pooling-allocator" : self.opts.pooling_allocator.or(pooling_allocator_default)]
823 enable => {
824 if enable {
825 let mut cfg = wasmtime::PoolingAllocationConfig::default();
826 if let Some(size) = self.opts.pooling_memory_keep_resident {
827 cfg.linear_memory_keep_resident(size);
828 }
829 if let Some(size) = self.opts.pooling_table_keep_resident {
830 cfg.table_keep_resident(size);
831 }
832 if let Some(limit) = self.opts.pooling_total_core_instances {
833 cfg.total_core_instances(limit);
834 }
835 if let Some(limit) = self.opts.pooling_total_component_instances {
836 cfg.total_component_instances(limit);
837 }
838 if let Some(limit) = self.opts.pooling_total_memories {
839 cfg.total_memories(limit);
840 }
841 if let Some(limit) = self.opts.pooling_total_tables {
842 cfg.total_tables(limit);
843 }
844 if let Some(limit) = self.opts.pooling_table_elements
845 .or(self.wasm.max_table_elements)
846 {
847 cfg.table_elements(limit);
848 }
849 if let Some(limit) = self.opts.pooling_max_core_instance_size {
850 cfg.max_core_instance_size(limit);
851 }
852 match_feature! {
853 ["async" : self.opts.pooling_total_stacks]
854 limit => cfg.total_stacks(limit),
855 _ => err,
856 }
857 if let Some(max) = self.opts.pooling_max_memory_size
858 .or(self.wasm.max_memory_size)
859 {
860 cfg.max_memory_size(max);
861 }
862 if let Some(size) = self.opts.pooling_decommit_batch_size {
863 cfg.decommit_batch_size(size);
864 }
865 if let Some(max) = self.opts.pooling_max_unused_warm_slots {
866 cfg.max_unused_warm_slots(max);
867 }
868 match_feature! {
869 ["async" : self.opts.pooling_async_stack_keep_resident]
870 size => cfg.async_stack_keep_resident(size),
871 _ => err,
872 }
873 if let Some(max) = self.opts.pooling_max_component_instance_size {
874 cfg.max_component_instance_size(max);
875 }
876 if let Some(max) = self.opts.pooling_max_core_instances_per_component {
877 cfg.max_core_instances_per_component(max);
878 }
879 if let Some(max) = self.opts.pooling_max_memories_per_component {
880 cfg.max_memories_per_component(max);
881 }
882 if let Some(max) = self.opts.pooling_max_tables_per_component {
883 cfg.max_tables_per_component(max);
884 }
885 if let Some(max) = self.opts.pooling_max_tables_per_module {
886 cfg.max_tables_per_module(max);
887 }
888 if let Some(max) = self.opts.pooling_max_memories_per_module {
889 cfg.max_memories_per_module(max);
890 }
891 match_feature! {
892 ["memory-protection-keys" : self.opts.pooling_memory_protection_keys]
893 enable => cfg.memory_protection_keys(enable),
894 _ => err,
895 }
896 match_feature! {
897 ["memory-protection-keys" : self.opts.pooling_max_memory_protection_keys]
898 max => cfg.max_memory_protection_keys(max),
899 _ => err,
900 }
901 match_feature! {
902 ["gc" : self.opts.pooling_total_gc_heaps]
903 max => cfg.total_gc_heaps(max),
904 _ => err,
905 }
906 config.allocation_strategy(wasmtime::InstanceAllocationStrategy::Pooling(cfg));
907 }
908 },
909 true => err,
910 }
911
912 if self.opts.pooling_memory_protection_keys.is_some()
913 && !self.opts.pooling_allocator.unwrap_or(false)
914 {
915 anyhow::bail!("memory protection keys require the pooling allocator");
916 }
917
918 if self.opts.pooling_max_memory_protection_keys.is_some()
919 && !self.opts.pooling_memory_protection_keys.is_some()
920 {
921 anyhow::bail!(
922 "max memory protection keys requires memory protection keys to be enabled"
923 );
924 }
925
926 match_feature! {
927 ["async" : self.wasm.async_stack_size]
928 size => config.async_stack_size(size),
929 _ => err,
930 }
931 match_feature! {
932 ["async" : self.wasm.async_stack_zeroing]
933 enable => config.async_stack_zeroing(enable),
934 _ => err,
935 }
936
937 if let Some(max) = self.wasm.max_wasm_stack {
938 config.max_wasm_stack(max);
939
940 #[cfg(feature = "async")]
944 if self.wasm.async_stack_size.is_none() {
945 const DEFAULT_HOST_STACK: usize = 512 << 10;
946 config.async_stack_size(max + DEFAULT_HOST_STACK);
947 }
948 }
949
950 if let Some(enable) = self.wasm.relaxed_simd_deterministic {
951 config.relaxed_simd_deterministic(enable);
952 }
953 match_feature! {
954 ["cranelift" : self.wasm.wmemcheck]
955 enable => config.wmemcheck(enable),
956 true => err,
957 }
958
959 Ok(config)
960 }
961
962 pub fn enable_wasm_features(&self, config: &mut Config) -> Result<()> {
963 let all = self.wasm.all_proposals;
964
965 if let Some(enable) = self.wasm.simd.or(all) {
966 config.wasm_simd(enable);
967 }
968 if let Some(enable) = self.wasm.relaxed_simd.or(all) {
969 config.wasm_relaxed_simd(enable);
970 }
971 if let Some(enable) = self.wasm.bulk_memory.or(all) {
972 config.wasm_bulk_memory(enable);
973 }
974 if let Some(enable) = self.wasm.multi_value.or(all) {
975 config.wasm_multi_value(enable);
976 }
977 if let Some(enable) = self.wasm.tail_call.or(all) {
978 config.wasm_tail_call(enable);
979 }
980 if let Some(enable) = self.wasm.multi_memory.or(all) {
981 config.wasm_multi_memory(enable);
982 }
983 if let Some(enable) = self.wasm.memory64.or(all) {
984 config.wasm_memory64(enable);
985 }
986 if let Some(enable) = self.wasm.custom_page_sizes.or(all) {
987 config.wasm_custom_page_sizes(enable);
988 }
989 if let Some(enable) = self.wasm.wide_arithmetic.or(all) {
990 config.wasm_wide_arithmetic(enable);
991 }
992 if let Some(enable) = self.wasm.extended_const.or(all) {
993 config.wasm_extended_const(enable);
994 }
995 if let Some(enable) = self.wasm.exceptions.or(all) {
996 config.wasm_exceptions(enable);
997 }
998 if let Some(enable) = self.wasm.legacy_exceptions.or(all) {
999 #[expect(deprecated, reason = "forwarding CLI flag")]
1000 config.wasm_legacy_exceptions(enable);
1001 }
1002
1003 macro_rules! handle_conditionally_compiled {
1004 ($(($feature:tt, $field:tt, $method:tt))*) => ($(
1005 if let Some(enable) = self.wasm.$field.or(all) {
1006 #[cfg(feature = $feature)]
1007 config.$method(enable);
1008 #[cfg(not(feature = $feature))]
1009 if enable && all.is_none() {
1010 anyhow::bail!("support for {} was disabled at compile-time", $feature);
1011 }
1012 }
1013 )*)
1014 }
1015
1016 handle_conditionally_compiled! {
1017 ("component-model", component_model, wasm_component_model)
1018 ("component-model-async", component_model_async, wasm_component_model_async)
1019 ("component-model-async", component_model_async_builtins, wasm_component_model_async_builtins)
1020 ("component-model-async", component_model_async_stackful, wasm_component_model_async_stackful)
1021 ("component-model", component_model_error_context, wasm_component_model_error_context)
1022 ("threads", threads, wasm_threads)
1023 ("gc", gc, wasm_gc)
1024 ("gc", reference_types, wasm_reference_types)
1025 ("gc", function_references, wasm_function_references)
1026 }
1027 Ok(())
1028 }
1029
1030 pub fn from_file<P: AsRef<Path>>(path: P) -> Result<Self> {
1031 let path_ref = path.as_ref();
1032 let file_contents = fs::read_to_string(path_ref)
1033 .with_context(|| format!("failed to read config file: {path_ref:?}"))?;
1034 toml::from_str::<CommonOptions>(&file_contents)
1035 .with_context(|| format!("failed to parse TOML config file {path_ref:?}"))
1036 }
1037}
1038
1039#[cfg(test)]
1040mod tests {
1041 use wasmtime::{OptLevel, RegallocAlgorithm};
1042
1043 use super::*;
1044
1045 #[test]
1046 fn from_toml() {
1047 let empty_toml = "";
1049 let mut common_options: CommonOptions = toml::from_str(empty_toml).unwrap();
1050 common_options.config(None).unwrap();
1051
1052 let basic_toml = r#"
1054 [optimize]
1055 [codegen]
1056 [debug]
1057 [wasm]
1058 [wasi]
1059 "#;
1060 let mut common_options: CommonOptions = toml::from_str(basic_toml).unwrap();
1061 common_options.config(None).unwrap();
1062
1063 for (opt_value, expected) in [
1065 ("0", Some(OptLevel::None)),
1066 ("1", Some(OptLevel::Speed)),
1067 ("2", Some(OptLevel::Speed)),
1068 ("\"s\"", Some(OptLevel::SpeedAndSize)),
1069 ("\"hello\"", None), ("3", None), ] {
1072 let toml = format!(
1073 r#"
1074 [optimize]
1075 opt-level = {opt_value}
1076 "#,
1077 );
1078 let parsed_opt_level = toml::from_str::<CommonOptions>(&toml)
1079 .ok()
1080 .and_then(|common_options| common_options.opts.opt_level);
1081
1082 assert_eq!(
1083 parsed_opt_level, expected,
1084 "Mismatch for input '{opt_value}'. Parsed: {parsed_opt_level:?}, Expected: {expected:?}"
1085 );
1086 }
1087
1088 for (regalloc_value, expected) in [
1090 ("\"backtracking\"", Some(RegallocAlgorithm::Backtracking)),
1091 ("\"hello\"", None), ("3", None), ("true", None), ] {
1095 let toml = format!(
1096 r#"
1097 [optimize]
1098 regalloc-algorithm = {regalloc_value}
1099 "#,
1100 );
1101 let parsed_regalloc_algorithm = toml::from_str::<CommonOptions>(&toml)
1102 .ok()
1103 .and_then(|common_options| common_options.opts.regalloc_algorithm);
1104 assert_eq!(
1105 parsed_regalloc_algorithm, expected,
1106 "Mismatch for input '{regalloc_value}'. Parsed: {parsed_regalloc_algorithm:?}, Expected: {expected:?}"
1107 );
1108 }
1109
1110 for (strategy_value, expected) in [
1112 ("\"cranelift\"", Some(wasmtime::Strategy::Cranelift)),
1113 ("\"winch\"", Some(wasmtime::Strategy::Winch)),
1114 ("\"hello\"", None), ("5", None), ("true", None), ] {
1118 let toml = format!(
1119 r#"
1120 [codegen]
1121 compiler = {strategy_value}
1122 "#,
1123 );
1124 let parsed_strategy = toml::from_str::<CommonOptions>(&toml)
1125 .ok()
1126 .and_then(|common_options| common_options.codegen.compiler);
1127 assert_eq!(
1128 parsed_strategy, expected,
1129 "Mismatch for input '{strategy_value}'. Parsed: {parsed_strategy:?}, Expected: {expected:?}",
1130 );
1131 }
1132
1133 for (collector_value, expected) in [
1135 (
1136 "\"drc\"",
1137 Some(wasmtime::Collector::DeferredReferenceCounting),
1138 ),
1139 ("\"null\"", Some(wasmtime::Collector::Null)),
1140 ("\"hello\"", None), ("5", None), ("true", None), ] {
1144 let toml = format!(
1145 r#"
1146 [codegen]
1147 collector = {collector_value}
1148 "#,
1149 );
1150 let parsed_collector = toml::from_str::<CommonOptions>(&toml)
1151 .ok()
1152 .and_then(|common_options| common_options.codegen.collector);
1153 assert_eq!(
1154 parsed_collector, expected,
1155 "Mismatch for input '{collector_value}'. Parsed: {parsed_collector:?}, Expected: {expected:?}",
1156 );
1157 }
1158 }
1159}
1160
1161impl Default for CommonOptions {
1162 fn default() -> CommonOptions {
1163 CommonOptions::new()
1164 }
1165}
1166
1167impl fmt::Display for CommonOptions {
1168 fn fmt(&self, f: &mut fmt::Formatter<'_>) -> fmt::Result {
1169 let CommonOptions {
1170 codegen_raw,
1171 codegen,
1172 debug_raw,
1173 debug,
1174 opts_raw,
1175 opts,
1176 wasm_raw,
1177 wasm,
1178 wasi_raw,
1179 wasi,
1180 configured,
1181 target,
1182 config,
1183 } = self;
1184 if let Some(target) = target {
1185 write!(f, "--target {target} ")?;
1186 }
1187 if let Some(config) = config {
1188 write!(f, "--config {} ", config.display())?;
1189 }
1190
1191 let codegen_flags;
1192 let opts_flags;
1193 let wasi_flags;
1194 let wasm_flags;
1195 let debug_flags;
1196
1197 if *configured {
1198 codegen_flags = codegen.to_options();
1199 debug_flags = debug.to_options();
1200 wasi_flags = wasi.to_options();
1201 wasm_flags = wasm.to_options();
1202 opts_flags = opts.to_options();
1203 } else {
1204 codegen_flags = codegen_raw
1205 .iter()
1206 .flat_map(|t| t.0.iter())
1207 .cloned()
1208 .collect();
1209 debug_flags = debug_raw.iter().flat_map(|t| t.0.iter()).cloned().collect();
1210 wasi_flags = wasi_raw.iter().flat_map(|t| t.0.iter()).cloned().collect();
1211 wasm_flags = wasm_raw.iter().flat_map(|t| t.0.iter()).cloned().collect();
1212 opts_flags = opts_raw.iter().flat_map(|t| t.0.iter()).cloned().collect();
1213 }
1214
1215 for flag in codegen_flags {
1216 write!(f, "-C{flag} ")?;
1217 }
1218 for flag in opts_flags {
1219 write!(f, "-O{flag} ")?;
1220 }
1221 for flag in wasi_flags {
1222 write!(f, "-S{flag} ")?;
1223 }
1224 for flag in wasm_flags {
1225 write!(f, "-W{flag} ")?;
1226 }
1227 for flag in debug_flags {
1228 write!(f, "-D{flag} ")?;
1229 }
1230
1231 Ok(())
1232 }
1233}