diff options
| author | Peter Zhu <[email protected]> | 2025-01-14 12:16:59 -0500 |
|---|---|---|
| committer | git <[email protected]> | 2025-01-14 17:17:44 +0000 |
| commit | f5fa1ee5f6e0e29e15063e8b62eb0ce7042bb29b (patch) | |
| tree | d27cc0701923e5cd6b6e0471021e654ecc84aaaa /gc/mmtk/src | |
| parent | 1adcd960e2c0ef7e5debafa4245cd14420616d7b (diff) | |
[ruby/mmtk] Exit with error message if MMTK_PLAN is invalid
https://github.com/ruby/mmtk/commit/79ce2008a3
Diffstat (limited to 'gc/mmtk/src')
| -rw-r--r-- | gc/mmtk/src/api.rs | 29 |
1 files changed, 17 insertions, 12 deletions
diff --git a/gc/mmtk/src/api.rs b/gc/mmtk/src/api.rs index 5d1351ef9c..0145dcda08 100644 --- a/gc/mmtk/src/api.rs +++ b/gc/mmtk/src/api.rs @@ -91,6 +91,21 @@ fn mmtk_builder_default_parse_heap_mode(heap_min: usize, heap_max: usize) -> GCT } } +fn mmtk_builder_default_parse_plan() -> PlanSelector { + let plan_str = std::env::var("MMTK_PLAN") + .unwrap_or("MarkSweep".to_string()); + + match plan_str.as_str() { + "NoGC" => PlanSelector::NoGC, + "MarkSweep" => PlanSelector::MarkSweep, + "Immix" => PlanSelector::Immix, + _ => { + eprintln!("[FATAL] Invalid MMTK_PLAN {}", plan_str); + std::process::exit(1); + } + } +} + #[no_mangle] pub extern "C" fn mmtk_builder_default() -> *mut MMTKBuilder { let mut builder = MMTKBuilder::new_no_env_vars(); @@ -110,19 +125,9 @@ pub extern "C" fn mmtk_builder_default() -> *mut MMTKBuilder { std::process::exit(1); } - let heap_mode = mmtk_builder_default_parse_heap_mode(heap_min, heap_max); - - // Parse the env var, if it's not found set the plan name to MarkSweep - let plan_name = std::env::var("MMTK_PLAN") - .unwrap_or(String::from("MarkSweep")); - - // Parse the plan name into a valid MMTK Plan, if the name is not a valid plan use MarkSweep - let plan_selector = plan_name.parse::<PlanSelector>() - .unwrap_or("MarkSweep".parse::<PlanSelector>().unwrap()); - - builder.options.plan.set(plan_selector); + builder.options.gc_trigger.set(mmtk_builder_default_parse_heap_mode(heap_min, heap_max)); - builder.options.gc_trigger.set(heap_mode); + builder.options.plan.set(mmtk_builder_default_parse_plan()); Box::into_raw(Box::new(builder)) } |