From a508baae2032fe58e9b3034231f36809a271f4da Mon Sep 17 00:00:00 2001
From: pca006132 <john.lck40@gmail.com>
Date: Wed, 22 Sep 2021 15:28:17 +0800
Subject: [PATCH] added syscall annotation

and temporarily disabled the keyword check for top-level functions
---
 nac3core/src/toplevel/composer.rs | 18 +++++++++++++-----
 1 file changed, 13 insertions(+), 5 deletions(-)

diff --git a/nac3core/src/toplevel/composer.rs b/nac3core/src/toplevel/composer.rs
index b961ccd8..1e915a33 100644
--- a/nac3core/src/toplevel/composer.rs
+++ b/nac3core/src/toplevel/composer.rs
@@ -244,9 +244,9 @@ impl TopLevelComposer {
             }
 
             ast::StmtKind::FunctionDef { name, .. } => {
-                if self.keyword_list.contains(name) {
-                    return Err("cannot use keyword as a top level function name".into());
-                }
+                // if self.keyword_list.contains(name) {
+                //     return Err("cannot use keyword as a top level function name".into());
+                // }
                 let fun_name = name.to_string();
                 let global_fun_name = {
                     let mut n = mod_path;
@@ -608,6 +608,7 @@ impl TopLevelComposer {
             let function_ast = if let Some(x) = function_ast.as_ref() {
                 x
             } else {
+                // if let TopLevelDef::Function { name, .. } = ``
                 continue;
             };
 
@@ -1223,8 +1224,9 @@ impl TopLevelComposer {
             let mut function_def = def.write();
             if let TopLevelDef::Function {
                 instance_to_stmt,
+                instance_to_symbol,
                 name,
-                simple_name: _,
+                simple_name,
                 signature,
                 resolver,
                 ..
@@ -1342,9 +1344,15 @@ impl TopLevelComposer {
                             calls: &mut calls,
                         };
 
-                        let fun_body = if let ast::StmtKind::FunctionDef { body, .. } =
+                        let fun_body = if let ast::StmtKind::FunctionDef { body, decorator_list, .. } =
                             ast.clone().unwrap().node
                         {
+                            if !decorator_list.is_empty() &&
+                                    matches!(&decorator_list[0].node,
+                                        ast::ExprKind::Name{ id, .. } if id == "syscall") {
+                                instance_to_symbol.insert("".to_string(), simple_name.clone());
+                                continue
+                            }
                             body
                         } else {
                             unreachable!("must be function def ast")