ctor functions

Author: aceinetx

zenlang/src/ast/function.rs

@@ -65,10 +65,12 @@ impl Compile for AstFunction {
         let module = compiler.get_module();
         {
             let name = self.name.to_string();
+            let ctor = self.attrs.contains(&FunctionAttribute::Ctor);
             module.functions.push(ModuleFunction::new(
                 name,
                 module.opcodes.len() as u32,
                 self.args.len() as u64,
+                ctor,
             ));
         }
 

zenlang/src/func_attr.rs

@@ -3,12 +3,15 @@ use alloc::string::*;
 #[derive(PartialEq)]
 pub enum FunctionAttribute {
     Naked,
+    Ctor,
 }
 
 impl FunctionAttribute {
     pub fn map(name: String) -> Option<FunctionAttribute> {
         if name == "naked" {
             return Some(FunctionAttribute::Naked);
+        } else if name == "ctor" {
+            return Some(FunctionAttribute::Ctor);
         }
         return None;
     }

zenlang/src/module.rs

@@ -21,14 +21,17 @@ pub struct ModuleFunction {
     pub addr: u32,
     /// Argument count
     pub args_count: u64,
+    /// Is a constructor function
+    pub ctor: bool,
 }
 
 impl ModuleFunction {
-    pub fn new(name: String, addr: u32, args_count: u64) -> ModuleFunction {
+    pub fn new(name: String, addr: u32, args_count: u64, ctor: bool) -> ModuleFunction {
         return ModuleFunction {
             name: name,
             addr: addr,
             args_count: args_count,
+            ctor: ctor,
         };
     }
 }

zenlang/src/stdlib.rs

@@ -113,6 +113,7 @@ fn number str {
         "_vmcall_ret_unsafe_1".into(),
         module.opcodes.len() as u32,
         1,
+        false,
     ));
     module.opcodes.push(Opcode::Dynvmcall());
     module.opcodes.push(Opcode::Ret());
@@ -121,6 +122,7 @@ fn number str {
         "_vmcall_ret_unsafe_2".into(),
         module.opcodes.len() as u32,
         2,
+        false,
     ));
     module.opcodes.push(Opcode::Dynvmcall());
     module.opcodes.push(Opcode::Ret());
@@ -129,6 +131,7 @@ fn number str {
         "_vmcall_ret_unsafe_3".into(),
         module.opcodes.len() as u32,
         3,
+        false,
     ));
     module.opcodes.push(Opcode::Dynvmcall());
     module.opcodes.push(Opcode::Ret());
@@ -137,6 +140,7 @@ fn number str {
         "_vmcall_ret_unsafe_4".into(),
         module.opcodes.len() as u32,
         4,
+        false,
     ));
     module.opcodes.push(Opcode::Dynvmcall());
     module.opcodes.push(Opcode::Ret());

zenlang/src/vm/vm.rs

@@ -46,8 +46,38 @@ impl VM {
     }
 
     pub fn load_module(&mut self, module: &Module) -> Result<(), String> {
+        // check if already loaded
+        for m in self.modules.iter_mut() {
+            if m.name == module.name {
+                return Ok(());
+            }
+        }
+
         self.modules.push(module.clone());
 
+        for func in module.functions.iter() {
+            if func.ctor {
+                self.check_stack_overflow();
+                self.pc.set_low(func.addr);
+                self.pc.set_high((self.modules.len() - 1) as u32);
+                self.add_scope();
+
+                loop {
+                    if !self.step() {
+                        break;
+                    }
+                }
+
+                if !self.error.is_empty() {
+                    return Err(format!(
+                        "in constructor of module {}: {}",
+                        module.name, self.error
+                    ));
+                }
+                self.halted = false;
+            }
+        }
+
         for var in module.globals.iter() {
             if self.global_scope.get(var).is_some() {
                 return Err(format!(