Skip to main content

synth_backend/
arm_backend.rs

1//! ARM Backend — wraps the instruction selector + optimizer + encoder as a Backend
2//!
3//! This is Synth's custom ARM compiler targeting Cortex-M (Thumb-2).
4//! It's the only backend that supports per-rule formal verification (ASIL D path).
5
6use crate::ArmEncoder;
7use synth_core::backend::{
8    Backend, BackendCapabilities, BackendError, CodeRelocation, CompilationResult, CompileConfig,
9    CompiledFunction, SafetyBounds,
10};
11use synth_core::target::{IsaVariant, TargetSpec};
12use synth_core::wasm_decoder::DecodedModule;
13use synth_core::wasm_op::WasmOp;
14use synth_synthesis::{
15    ArmInstruction, ArmOp, BoundsCheckConfig, InstructionSelector, OptimizationConfig,
16    OptimizerBridge, RuleDatabase, validate_instructions,
17};
18
19/// ARM Cortex-M backend using Synth's custom compiler pipeline
20pub struct ArmBackend;
21
22impl ArmBackend {
23    pub fn new() -> Self {
24        Self
25    }
26}
27
28impl Default for ArmBackend {
29    fn default() -> Self {
30        Self::new()
31    }
32}
33
34impl Backend for ArmBackend {
35    fn name(&self) -> &str {
36        "arm"
37    }
38
39    fn capabilities(&self) -> BackendCapabilities {
40        BackendCapabilities {
41            produces_elf: false,
42            supports_rule_verification: true,
43            supports_binary_verification: true,
44            is_external: false,
45        }
46    }
47
48    fn supported_targets(&self) -> Vec<TargetSpec> {
49        vec![
50            TargetSpec::cortex_m3(),
51            TargetSpec::cortex_m4(),
52            TargetSpec::cortex_m4f(),
53            TargetSpec::cortex_m7(),
54            TargetSpec::cortex_m7dp(),
55        ]
56    }
57
58    fn compile_module(
59        &self,
60        module: &DecodedModule,
61        config: &CompileConfig,
62    ) -> Result<CompilationResult, BackendError> {
63        let exports: Vec<_> = module
64            .functions
65            .iter()
66            .filter(|f| f.export_name.is_some())
67            .collect();
68
69        if exports.is_empty() {
70            return Err(BackendError::CompilationFailed(
71                "no exported functions found".into(),
72            ));
73        }
74
75        let mut functions = Vec::new();
76        for func in &exports {
77            let name = func.export_name.clone().unwrap();
78            let compiled = self.compile_function(&name, &func.ops, config)?;
79            functions.push(compiled);
80        }
81
82        Ok(CompilationResult {
83            functions,
84            elf: None,
85            backend_name: self.name().to_string(),
86        })
87    }
88
89    fn compile_function(
90        &self,
91        name: &str,
92        ops: &[WasmOp],
93        config: &CompileConfig,
94    ) -> Result<CompiledFunction, BackendError> {
95        let (code, relocations) =
96            compile_wasm_to_arm(ops, config).map_err(BackendError::CompilationFailed)?;
97
98        Ok(CompiledFunction {
99            name: name.to_string(),
100            code,
101            wasm_ops: ops.to_vec(),
102            relocations,
103        })
104    }
105
106    fn is_available(&self) -> bool {
107        true // Always available — it's a library backend
108    }
109}
110
111/// Count the number of function parameters by analyzing LocalGet patterns
112fn count_params(wasm_ops: &[WasmOp]) -> u32 {
113    let mut first_access: std::collections::HashMap<u32, bool> = std::collections::HashMap::new();
114    for op in wasm_ops {
115        match op {
116            WasmOp::LocalGet(idx) => {
117                first_access.entry(*idx).or_insert(true);
118            }
119            WasmOp::LocalSet(idx) | WasmOp::LocalTee(idx) => {
120                first_access.entry(*idx).or_insert(false);
121            }
122            _ => {}
123        }
124    }
125
126    first_access
127        .iter()
128        .filter_map(
129            |(&idx, &is_read_first)| {
130                if is_read_first { Some(idx + 1) } else { None }
131            },
132        )
133        .max()
134        .unwrap_or(0)
135}
136
137/// Core compilation: WASM ops → ARM machine code bytes + relocations
138///
139/// Returns (code_bytes, relocations) where relocations record BL instructions
140/// that target external symbols (e.g., `__meld_dispatch_import` for import calls).
141fn compile_wasm_to_arm(
142    wasm_ops: &[WasmOp],
143    config: &CompileConfig,
144) -> Result<(Vec<u8>, Vec<CodeRelocation>), String> {
145    let num_params = count_params(wasm_ops);
146
147    let bounds_config = match config.effective_safety_bounds() {
148        SafetyBounds::None => BoundsCheckConfig::None,
149        SafetyBounds::Mpu => BoundsCheckConfig::Mpu,
150        SafetyBounds::Software => BoundsCheckConfig::Software,
151        SafetyBounds::Mask => BoundsCheckConfig::Masking,
152    };
153
154    // The non-optimized (direct) instruction-selection path. Handles f32 via
155    // VFP/FPU. Used directly when `--no-optimize` is set, and as the fallback
156    // when the optimized path declines a module (see issue #120 below).
157    let select_direct = || -> Result<Vec<ArmInstruction>, String> {
158        let db = RuleDatabase::with_standard_rules();
159        let mut selector =
160            InstructionSelector::with_bounds_check(db.rules().to_vec(), bounds_config);
161        selector.set_target(config.target.fpu, &config.target.triple);
162        if config.num_imports > 0 {
163            selector.set_num_imports(config.num_imports);
164        }
165        // #195: plumb the callee argument-count tables so the direct selector can
166        // marshal call arguments into R0–R3 per AAPCS.
167        selector.set_func_arg_counts(
168            config.func_arg_counts.clone(),
169            config.type_arg_counts.clone(),
170        );
171        // #197: in relocatable host-link mode, emit direct `func_N` BLs for
172        // imports (rewritten to the wasm field name by build_relocatable_elf)
173        // instead of `__meld_dispatch_import`.
174        selector.set_relocatable(config.relocatable);
175        selector
176            .select_with_stack(wasm_ops, num_params)
177            .map_err(|e| format!("instruction selection failed: {}", e))
178    };
179
180    // Instruction selection: optimized or direct.
181    //
182    // #197: `--relocatable` (host-link ET_REL) forces the direct selector. The
183    // optimized path materializes an absolute linmem base (0x20000100) and does
184    // not preserve caller-saved registers across calls — both wrong for a
185    // host-linked object, where the linmem base arrives via `fp` at runtime and
186    // callees follow AAPCS. `select_with_stack` (now i64-spill capable after
187    // #171) handles fp-relative memory + caller-saved preservation correctly.
188    let arm_instrs = if config.no_optimize || config.relocatable {
189        select_direct()?
190    } else {
191        let opt_config = if config.loom_compat {
192            OptimizationConfig::loom_compat()
193        } else {
194            OptimizationConfig::all()
195        };
196
197        let mut bridge = OptimizerBridge::with_config(opt_config);
198        // #188: tell the bridge how many imports there are so it declines only
199        // LOCAL calls (and leaves import calls on the optimized path, keeping
200        // the #173 field-name relocation rewrite intact).
201        bridge.set_num_imports(config.num_imports);
202        // `ir_to_arm` now returns `Result` — an `Err` means the optimized path
203        // hit an unmapped vreg (issue-#93-class). Treat it identically to an
204        // `optimize_full` failure: fall back to the direct selector rather
205        // than propagating, so the function still compiles correctly.
206        match bridge
207            .optimize_full(wasm_ops)
208            .and_then(|(opt_ir, _cfg, _stats)| bridge.ir_to_arm(&opt_ir, num_params as usize))
209        {
210            Ok(arm_ops) => arm_ops
211                .into_iter()
212                .map(|op| ArmInstruction {
213                    op,
214                    source_line: None,
215                })
216                .collect(),
217            // Issue #120: the optimized path declines modules it cannot lower
218            // (notably scalar f32/f64 ops — the IR has no float opcodes). Fall
219            // back to the direct instruction selector, which handles f32 via
220            // VFP/FPU. This is honest degradation: the function still compiles
221            // correctly, just without IR-level optimization.
222            Err(_) => select_direct()?,
223        }
224    };
225
226    // ISA feature gate: validate that all generated instructions are supported
227    // by the target. This catches FPU instructions on no-FPU targets, double-precision
228    // instructions on single-precision targets, etc.
229    validate_instructions(&arm_instrs, config.target.fpu, &config.target.triple)
230        .map_err(|e| format!("ISA validation failed: {}", e))?;
231
232    // Encode to binary — use Thumb-2 for Cortex-M targets
233    let use_thumb2 = matches!(config.target.isa, IsaVariant::Thumb2 | IsaVariant::Thumb);
234
235    let encoder = if use_thumb2 {
236        ArmEncoder::new_thumb2_with_fpu(config.target.fpu)
237    } else {
238        ArmEncoder::new_arm32()
239    };
240
241    let mut code = Vec::new();
242    let mut relocations = Vec::new();
243
244    for instr in &arm_instrs {
245        // Record a relocation for every BL: the encoder emits `bl #0` and
246        // relies on a relocation to patch the target. This covers BOTH import
247        // dispatch stubs (`__meld_*`, undefined externals) AND internal calls
248        // (`func_N`, defined in this object). Previously only `__meld_*` was
249        // recorded, so internal `BL func_N` calls were left as unpatched
250        // `bl #0` placeholders branching to a garbage address (#167).
251        if let ArmOp::Bl { label } = &instr.op {
252            relocations.push(CodeRelocation {
253                offset: code.len() as u32,
254                symbol: label.clone(),
255            });
256        }
257
258        let encoded = encoder
259            .encode(&instr.op)
260            .map_err(|e| format!("ARM encoding failed: {}", e))?;
261        code.extend_from_slice(&encoded);
262    }
263
264    Ok((code, relocations))
265}
266
267#[cfg(test)]
268mod tests {
269    use super::*;
270
271    #[test]
272    fn test_arm_backend_name() {
273        let backend = ArmBackend::new();
274        assert_eq!(backend.name(), "arm");
275        assert!(backend.is_available());
276    }
277
278    #[test]
279    fn test_arm_backend_capabilities() {
280        let backend = ArmBackend::new();
281        let caps = backend.capabilities();
282        assert!(!caps.produces_elf);
283        assert!(caps.supports_rule_verification);
284        assert!(!caps.is_external);
285    }
286
287    #[test]
288    fn test_compile_add_function() {
289        let backend = ArmBackend::new();
290        let ops = vec![WasmOp::LocalGet(0), WasmOp::LocalGet(1), WasmOp::I32Add];
291        let config = CompileConfig::default();
292
293        let result = backend.compile_function("add", &ops, &config);
294        assert!(result.is_ok());
295
296        let func = result.unwrap();
297        assert_eq!(func.name, "add");
298        assert!(!func.code.is_empty());
299        assert_eq!(func.wasm_ops, ops);
300    }
301
302    #[test]
303    fn test_count_params() {
304        let ops = vec![WasmOp::LocalGet(0), WasmOp::LocalGet(1), WasmOp::I32Add];
305        assert_eq!(count_params(&ops), 2);
306
307        let no_params = vec![WasmOp::I32Const(5), WasmOp::I32Const(3), WasmOp::I32Add];
308        assert_eq!(count_params(&no_params), 0);
309    }
310
311    #[test]
312    fn test_arm_backend_register() {
313        let mut registry = synth_core::BackendRegistry::new();
314        registry.register(Box::new(ArmBackend::new()));
315        assert!(registry.get("arm").is_some());
316        assert_eq!(registry.available().len(), 1);
317    }
318
319    #[test]
320    fn test_compile_import_call_produces_relocations() {
321        let backend = ArmBackend::new();
322        // Simulate a WASM module where func index 0 is an import.
323        // Call(0) should generate MOV R0, #0; BL __meld_dispatch_import
324        let ops = vec![WasmOp::Call(0)];
325        let config = CompileConfig {
326            num_imports: 1,
327            no_optimize: true, // Direct instruction selection to preserve Call semantics
328            ..CompileConfig::default()
329        };
330
331        let result = backend.compile_function("caller", &ops, &config);
332        assert!(result.is_ok());
333
334        let func = result.unwrap();
335        assert!(!func.code.is_empty());
336        assert_eq!(func.relocations.len(), 1);
337        assert_eq!(func.relocations[0].symbol, "__meld_dispatch_import");
338        // The BL is the second instruction (after MOV R0, #0), so offset should be > 0
339        assert!(func.relocations[0].offset > 0);
340    }
341
342    /// Regression test for #197: in `relocatable` mode, an import call must
343    /// relocate against the direct `func_N` symbol (rewritten to the wasm field
344    /// name by `build_relocatable_elf`), NOT `__meld_dispatch_import`. This is
345    /// the ABI half of the #197 fix — without it, a host linker cannot resolve
346    /// the call to the real kernel symbol (e.g. `k_spin_lock`).
347    #[test]
348    fn test_compile_relocatable_import_uses_direct_func_symbol_197() {
349        let backend = ArmBackend::new();
350        let ops = vec![WasmOp::Call(0)]; // func 0 is an import
351        let config = CompileConfig {
352            num_imports: 1,
353            relocatable: true,
354            ..CompileConfig::default()
355        };
356
357        let func = backend
358            .compile_function("caller", &ops, &config)
359            .expect("relocatable import call compiles");
360
361        assert_eq!(func.relocations.len(), 1);
362        assert_eq!(
363            func.relocations[0].symbol, "func_0",
364            "#197: relocatable import must relocate against func_0 (→ field name), not Meld dispatch"
365        );
366    }
367
368    #[test]
369    fn test_compile_no_imports_no_relocations() {
370        let backend = ArmBackend::new();
371        let ops = vec![WasmOp::LocalGet(0), WasmOp::LocalGet(1), WasmOp::I32Add];
372        let config = CompileConfig::default();
373
374        let func = backend.compile_function("add", &ops, &config).unwrap();
375        assert!(func.relocations.is_empty());
376    }
377
378    /// Regression test for #167: a call to an INTERNAL function
379    /// (index `>= num_imports`) must record a relocation against `func_{index}`.
380    /// Before the fix, only `__meld_*` (import) BLs were relocated, so
381    /// internal `BL func_N` was emitted as an unpatched `bl #0` branching
382    /// to a garbage address — making the object non-linkable. This test
383    /// would have caught that regression.
384    #[test]
385    fn test_compile_internal_call_produces_relocation_167() {
386        let backend = ArmBackend::new();
387        // num_imports = 1, so Call(2) is an INTERNAL call → `BL func_2`.
388        let ops = vec![WasmOp::Call(2)];
389        let config = CompileConfig {
390            num_imports: 1,
391            no_optimize: true,
392            ..CompileConfig::default()
393        };
394
395        let func = backend
396            .compile_function("caller", &ops, &config)
397            .expect("internal call compiles");
398
399        assert_eq!(
400            func.relocations.len(),
401            1,
402            "an internal call must emit exactly one relocation (#167)"
403        );
404        assert_eq!(
405            func.relocations[0].symbol, "func_2",
406            "internal call must relocate against the callee's func_{{index}} symbol (#167)"
407        );
408    }
409
410    // ─── Phase 1 safety-bounds plumbing for ARM ──────────────────────────
411
412    #[test]
413    fn arm_safety_bounds_mpu_emits_same_code_as_none() {
414        // Mpu mode must not introduce any inline check on ARM — the MPU
415        // handles faults via hardware. The encoded bytes for an i32.load
416        // should be identical between None and Mpu.
417        let backend = ArmBackend::new();
418        let ops = vec![
419            WasmOp::LocalGet(0),
420            WasmOp::I32Load {
421                offset: 0,
422                align: 2,
423            },
424        ];
425        let cfg_none = CompileConfig {
426            no_optimize: true,
427            ..Default::default()
428        };
429        let cfg_mpu = CompileConfig {
430            no_optimize: true,
431            safety_bounds: SafetyBounds::Mpu,
432            ..Default::default()
433        };
434        let n = backend.compile_function("ld", &ops, &cfg_none).unwrap();
435        let m = backend.compile_function("ld", &ops, &cfg_mpu).unwrap();
436        assert_eq!(
437            n.code, m.code,
438            "Mpu and None should produce identical ARM bytes (Mpu relies on hardware)"
439        );
440    }
441
442    #[test]
443    fn arm_legacy_bounds_check_still_emits_software_check() {
444        // Legacy CLI users with `--bounds-check` should keep getting the
445        // software path even though the new SafetyBounds field defaults to None.
446        let backend = ArmBackend::new();
447        let ops = vec![
448            WasmOp::LocalGet(0),
449            WasmOp::I32Load {
450                offset: 0,
451                align: 2,
452            },
453        ];
454        let cfg_legacy = CompileConfig {
455            no_optimize: true,
456            bounds_check: true,
457            ..Default::default()
458        };
459        let cfg_software = CompileConfig {
460            no_optimize: true,
461            safety_bounds: SafetyBounds::Software,
462            ..Default::default()
463        };
464        let l = backend.compile_function("ld", &ops, &cfg_legacy).unwrap();
465        let s = backend.compile_function("ld", &ops, &cfg_software).unwrap();
466        assert_eq!(
467            l.code, s.code,
468            "--bounds-check should produce the same bytes as --safety-bounds=software"
469        );
470    }
471
472    // ========================================================================
473    // ISA feature gate tests — ensure the compiler never emits unsupported
474    // instructions for a given target
475    // ========================================================================
476
477    #[test]
478    fn test_f32_rejected_on_cortex_m3_no_fpu() {
479        let backend = ArmBackend::new();
480        let ops = vec![WasmOp::F32Const(1.0), WasmOp::F32Const(2.0), WasmOp::F32Add];
481        let config = CompileConfig {
482            target: TargetSpec::cortex_m3(),
483            no_optimize: true,
484            ..CompileConfig::default()
485        };
486
487        let result = backend.compile_function("fadd", &ops, &config);
488        assert!(
489            result.is_err(),
490            "f32 operations should fail on Cortex-M3 (no FPU)"
491        );
492    }
493
494    #[test]
495    fn test_f32_accepted_on_cortex_m4f() {
496        let backend = ArmBackend::new();
497        let ops = vec![WasmOp::F32Const(1.0), WasmOp::F32Const(2.0), WasmOp::F32Add];
498        let config = CompileConfig {
499            target: TargetSpec::cortex_m4f(),
500            no_optimize: true,
501            ..CompileConfig::default()
502        };
503
504        let result = backend.compile_function("fadd", &ops, &config);
505        assert!(
506            result.is_ok(),
507            "f32 operations should succeed on Cortex-M4F, got: {:?}",
508            result.unwrap_err()
509        );
510    }
511
512    #[test]
513    fn test_i32_works_on_all_targets() {
514        let backend = ArmBackend::new();
515        let ops = vec![WasmOp::LocalGet(0), WasmOp::LocalGet(1), WasmOp::I32Add];
516
517        // Cortex-M3 (no FPU)
518        let config_m3 = CompileConfig {
519            target: TargetSpec::cortex_m3(),
520            no_optimize: true,
521            ..CompileConfig::default()
522        };
523        assert!(
524            backend.compile_function("add", &ops, &config_m3).is_ok(),
525            "i32 ops should work on Cortex-M3"
526        );
527
528        // Cortex-M4F (single FPU)
529        let config_m4f = CompileConfig {
530            target: TargetSpec::cortex_m4f(),
531            no_optimize: true,
532            ..CompileConfig::default()
533        };
534        assert!(
535            backend.compile_function("add", &ops, &config_m4f).is_ok(),
536            "i32 ops should work on Cortex-M4F"
537        );
538
539        // Cortex-M7DP (double FPU)
540        let config_m7dp = CompileConfig {
541            target: TargetSpec::cortex_m7dp(),
542            no_optimize: true,
543            ..CompileConfig::default()
544        };
545        assert!(
546            backend.compile_function("add", &ops, &config_m7dp).is_ok(),
547            "i32 ops should work on Cortex-M7DP"
548        );
549    }
550
551    #[test]
552    fn test_f32_rejected_on_cortex_m4_no_fpu() {
553        // Cortex-M4 (without F suffix) has no FPU
554        let backend = ArmBackend::new();
555        let ops = vec![WasmOp::F32Const(1.5), WasmOp::F32Const(2.5), WasmOp::F32Mul];
556        let config = CompileConfig {
557            target: TargetSpec::cortex_m4(),
558            no_optimize: true,
559            ..CompileConfig::default()
560        };
561
562        let result = backend.compile_function("fmul", &ops, &config);
563        assert!(
564            result.is_err(),
565            "f32 operations should fail on Cortex-M4 (no FPU)"
566        );
567    }
568
569    // ========================================================================
570    // Issue #120 — f32 ops in the optimized lowering path
571    //
572    // `OptimizerBridge::wasm_to_ir` has no handlers for f32/f64 ops, so a
573    // value-producing float op fell through to `Opcode::Nop`, leaving a
574    // downstream consumer with an unmapped vreg and tripping the PR #101
575    // defensive panic in `ir_to_arm`. Customer reproducer: `compiler_builtins
576    // float::div` and `gale_compute_ipi_mask` in the `falcon-rate-component`
577    // module.
578    //
579    // Fix: `optimize_full` declines float modules with a typed `Err`;
580    // `compile_wasm_to_arm` falls back to the non-optimized `select_with_stack`
581    // path, which handles f32 via VFP/FPU. These tests use the *default*
582    // (optimized) config — `no_optimize` is NOT set — which is the exact
583    // configuration that panicked pre-fix.
584    // ========================================================================
585
586    /// Pre-fix: this panicked with "vreg vN has no assigned ARM register and
587    /// no spill slot" inside `ir_to_arm`. Post-fix: the optimized path declines
588    /// the module and the backend falls back to direct selection, producing a
589    /// non-empty f32.div lowering on a Cortex-M4F.
590    #[test]
591    fn test_issue120_f32_div_compiles_via_optimized_default() {
592        let backend = ArmBackend::new();
593        let ops = vec![WasmOp::LocalGet(0), WasmOp::LocalGet(1), WasmOp::F32Div];
594        let config = CompileConfig {
595            target: TargetSpec::cortex_m4f(),
596            // no_optimize NOT set — this exercises the optimized path that
597            // panicked in issue #120, then the fallback to direct selection.
598            ..CompileConfig::default()
599        };
600
601        let result = backend.compile_function("fdiv", &ops, &config);
602        assert!(
603            result.is_ok(),
604            "f32.div must compile on Cortex-M4F via the optimized->direct \
605             fallback (issue #120), got: {:?}",
606            result.as_ref().err()
607        );
608        assert!(
609            !result.unwrap().code.is_empty(),
610            "f32.div must produce non-empty machine code"
611        );
612    }
613
614    /// A spread of f32 ops, all through the optimized (default) config, must
615    /// compile via the fallback on an FPU target without panicking.
616    #[test]
617    fn test_issue120_assorted_f32_ops_compile_via_optimized_default() {
618        let backend = ArmBackend::new();
619        let config = CompileConfig {
620            target: TargetSpec::cortex_m4f(),
621            ..CompileConfig::default()
622        };
623
624        let cases: Vec<(&str, Vec<WasmOp>)> = vec![
625            (
626                "fadd",
627                vec![WasmOp::LocalGet(0), WasmOp::LocalGet(1), WasmOp::F32Add],
628            ),
629            (
630                "fmul",
631                vec![WasmOp::LocalGet(0), WasmOp::LocalGet(1), WasmOp::F32Mul],
632            ),
633            (
634                "fsub",
635                vec![WasmOp::LocalGet(0), WasmOp::LocalGet(1), WasmOp::F32Sub],
636            ),
637        ];
638
639        for (name, ops) in cases {
640            let result = backend.compile_function(name, &ops, &config);
641            assert!(
642                result.is_ok(),
643                "{name} must compile via the optimized->direct fallback \
644                 (issue #120), got: {:?}",
645                result.as_ref().err()
646            );
647            assert!(
648                !result.unwrap().code.is_empty(),
649                "{name} must produce non-empty machine code"
650            );
651        }
652    }
653
654    /// The fallback must still honor the ISA feature gate: f32 on a no-FPU
655    /// target must fail cleanly (not panic) even on the optimized path.
656    #[test]
657    fn test_issue120_f32_div_rejected_on_no_fpu_via_optimized() {
658        let backend = ArmBackend::new();
659        let ops = vec![WasmOp::LocalGet(0), WasmOp::LocalGet(1), WasmOp::F32Div];
660        let config = CompileConfig {
661            target: TargetSpec::cortex_m3(),
662            ..CompileConfig::default()
663        };
664
665        let result = backend.compile_function("fdiv", &ops, &config);
666        assert!(
667            result.is_err(),
668            "f32.div must be rejected on Cortex-M3 (no FPU), not panic"
669        );
670    }
671
672    /// Issue #94: end-to-end byte-size check for the canonical u64-packed
673    /// FFI-return hi32 extract pattern. Compiles two near-identical
674    /// functions — one with the optimized shift-by-32, one with a generic
675    /// shift-by-7 — and asserts the optimized form is meaningfully smaller.
676    #[test]
677    fn test_issue94_hi32_extract_is_smaller_than_generic_shift() {
678        let backend = ArmBackend::new();
679        let config = CompileConfig {
680            target: TargetSpec::cortex_m4f(),
681            ..CompileConfig::default()
682        };
683
684        // Optimized path: `(local.get 0) >>> 32; wrap_i64`
685        let ops_hi32 = vec![
686            WasmOp::LocalGet(0), // i64 param in R0:R1
687            WasmOp::I64Const(32),
688            WasmOp::I64ShrU,
689            WasmOp::I32WrapI64,
690        ];
691        let func_hi32 = backend
692            .compile_function("hi32_extract", &ops_hi32, &config)
693            .unwrap();
694
695        // Generic path: `(local.get 0) >>> 7; wrap_i64` — same shape, but the
696        // shift amount is not a multiple of 32, so it falls through to the
697        // 38-byte runtime shift.
698        let ops_generic = vec![
699            WasmOp::LocalGet(0),
700            WasmOp::I64Const(7),
701            WasmOp::I64ShrU,
702            WasmOp::I32WrapI64,
703        ];
704        let func_generic = backend
705            .compile_function("generic_shr", &ops_generic, &config)
706            .unwrap();
707
708        let bytes_hi32 = func_hi32.code.len();
709        let bytes_generic = func_generic.code.len();
710        println!(
711            "\n[issue #94] hi32 extract: {} bytes (vs generic shift: {} bytes; saved {})",
712            bytes_hi32,
713            bytes_generic,
714            bytes_generic.saturating_sub(bytes_hi32)
715        );
716        let hex: String = func_hi32
717            .code
718            .iter()
719            .map(|b| format!("{:02x}", b))
720            .collect::<Vec<_>>()
721            .join(" ");
722        println!("[issue #94] hi32 bytes: {}", hex);
723        // We expect the optimized form to be at least 30 bytes smaller than
724        // the generic 64-bit shift sequence. (Empirically: 14 vs 50 bytes.)
725        assert!(
726            bytes_hi32 + 30 <= bytes_generic,
727            "issue #94: hi32 extract = {} bytes, generic shift = {} bytes; \
728             expected optimized form to be at least 30 bytes smaller",
729            bytes_hi32,
730            bytes_generic,
731        );
732    }
733}