add table.size instruction execution

wasmi-labs · Aug 22, 2023 · 304680d · 304680d
1 parent 40fa003
commit 304680d
Show file tree

Hide file tree

Showing 2 changed files with 10 additions and 1 deletion.
diff --git a/crates/wasmi/src/engine/regmach/executor/instrs.rs b/crates/wasmi/src/engine/regmach/executor/instrs.rs
@@ -351,7 +351,7 @@ impl<'ctx, 'engine> Executor<'ctx, 'engine> {
                 Instr::TableGetImm { result, index } => {
                     self.execute_table_get_imm(result, index)?
                 }
-                Instr::TableSize { result, table } => todo!(),
+                Instr::TableSize { result, table } => self.execute_table_size(result, table),
                 Instr::TableSet { index, value } => todo!(),
                 Instr::TableSetAt { index, value } => todo!(),
                 Instr::TableCopy { dst, src, len } => todo!(),

diff --git a/crates/wasmi/src/engine/regmach/executor/instrs/table.rs b/crates/wasmi/src/engine/regmach/executor/instrs/table.rs
@@ -48,4 +48,13 @@ impl<'engine, 'ctx> Executor<'engine, 'ctx> {
         self.set_register(result, value);
         self.try_next_instr()
     }
+
+    /// Executes an [`Instruction::TableSize`].
+    #[inline(always)]
+    pub fn execute_table_size(&mut self, result: Register, table_index: TableIdx) {
+        let table = self.cache.get_table(self.ctx, table_index);
+        let size = self.ctx.resolve_table(&table).size();
+        self.set_register(result, size);
+        self.next_instr();
+    }
 }