Add setn

1 files changed, 10 insertions, 0 deletions

diff --git a/zjit/src/ir.rs b/zjit/src/ir.rs
index a62710647a..6a2f3b52ad 100644
--- a/zjit/src/ir.rs
+++ b/zjit/src/ir.rs
@@ -235,6 +235,11 @@ impl FrameState {
         self.stack.pop().expect("Bytecode stack mismatch (underflow)")
     }
 
+    fn setn(&mut self, n: usize, opnd: Opnd) {
+        let idx = self.stack.len() - n - 1;
+        self.stack[idx] = opnd;
+    }
+
     fn setlocal(&mut self, idx: usize, opnd: Opnd) {
         if idx >= self.locals.len() {
             self.locals.resize(idx+1, Opnd::Const(Qnil));
@@ -420,6 +425,11 @@ pub fn iseq_to_ssa(iseq: *const rb_iseq_t) -> Function {
                 state.push(right);
                 state.push(left);
             }
+            YARVINSN_setn => {
+                let n = get_arg(pc, 0).as_usize();
+                let top = state.top();
+                state.setn(n, top);
+            }
 
             YARVINSN_opt_plus => {
                 let v0 = state.pop();