Add missing instructions to assembler

Author: maximecb

vm/src/asm.rs

@@ -957,6 +957,22 @@ impl Assembler
                 self.code.push_u8(idx);
             }
 
+            "get_argc" => {
+                self.code.push_op(Op::get_argc);
+            }
+
+            "thread_get" => {
+                let idx: u8 = self.parse_int_arg(input)?;
+                self.code.push_op(Op::thread_get);
+                self.code.push_u8(idx);
+            }
+
+            "thread_set" => {
+                let idx: u8 = self.parse_int_arg(input)?;
+                self.code.push_op(Op::thread_set);
+                self.code.push_u8(idx);
+            }
+
             "push_0" => {
                 self.code.push_op(Op::push_0);
             }
@@ -1274,6 +1290,13 @@ mod tests
         parse_ok(".code;\npush_u32 0xFFFFFFFF;");
         parse_ok(".code; push_u32 1_000_000;");
         parse_ok(".code; push_i8 55; push_i8 -1;");
+
+        // Frame argument count and thread-local variable access
+        parse_ok(".code; get_argc;");
+        parse_ok(".code; thread_get 0;");
+        parse_ok(".code; push_i8 7; thread_set 3;");
+        parse_fails(".code; thread_get;");
+        parse_fails(".code; thread_set;");
     }
 
     #[test]

vm/src/vm.rs

@@ -838,6 +838,11 @@ impl Thread
                     self.push(b);
                 }
 
+                Op::get_argc => {
+                    let argc = self.frames[self.frames.len() - 1].argc;
+                    self.push(Value::from(argc as u64));
+                }
+
                 Op::get_arg => {
                     let idx = self.code.read_pc::<u8>(&mut pc) as usize;
 
@@ -1928,6 +1933,17 @@ mod tests
         eval_i64(".code; push 0; push 77; set_local 0; get_local 0; ret;", 77);
     }
 
+    #[test]
+    fn test_get_argc()
+    {
+        // The top-level frame is entered with no arguments
+        eval_i64("get_argc; ret;", 0);
+
+        // get_argc reports the argument count of the current frame
+        eval_i64("push 42; call FN, 1; ret; FN: get_argc; ret;", 1);
+        eval_i64("push 10; push 20; push 30; call FN, 3; ret; FN: get_argc; ret;", 3);
+    }
+
     #[test]
     fn test_floats()
     {