implement FuncTranslator::register_locals

Author: Robbepop

crates/wasmi/src/engine/translator/translator2/layout/mod.rs

@@ -2,7 +2,12 @@ mod consts;
 
 use self::consts::ConstRegistry;
 use super::{LocalIdx, OperandIdx, Reset};
-use crate::{core::UntypedVal, engine::TranslationError, ir::Reg, Error};
+use crate::{
+    core::{UntypedVal, ValType},
+    engine::TranslationError,
+    ir::Reg,
+    Error,
+};
 
 #[cfg(doc)]
 use super::Stack;
@@ -24,6 +29,16 @@ impl Reset for StackLayout {
 }
 
 impl StackLayout {
+    /// Register `amount` local variables of common type `ty`.
+    ///
+    /// # Errors
+    ///
+    /// If too many local variables are being registered.
+    pub fn register_locals(&mut self, amount: u32, ty: ValType) -> Result<(), Error> {
+        self.len_locals += amount as usize;
+        Ok(())
+    }
+
     /// Returns the [`StackSpace`] of the [`Reg`].
     ///
     /// Returns `None` if the [`Reg`] is unknown to the [`Stack`].

crates/wasmi/src/engine/translator/translator2/mod.rs

@@ -15,7 +15,7 @@ use self::{
 use crate::{
     core::FuelCostsProvider,
     engine::{translator::WasmTranslator, CompiledFuncEntity},
-    module::{FuncIdx, ModuleHeader},
+    module::{FuncIdx, ModuleHeader, WasmiValueType},
     Engine,
     Error,
 };
@@ -79,10 +79,13 @@ impl WasmTranslator<'_> for FuncTranslator {
 
     fn translate_locals(
         &mut self,
-        _amount: u32,
-        _value_type: wasmparser::ValType,
+        amount: u32,
+        value_type: wasmparser::ValType,
     ) -> Result<(), Error> {
-        todo!()
+        let ty = WasmiValueType::from(value_type).into_inner();
+        self.stack.register_locals(amount, ty)?;
+        self.layout.register_locals(amount, ty)?;
+        Ok(())
     }
 
     fn finish_translate_locals(&mut self) -> Result<(), Error> {