Rust: Add Callable.getBody()

Author: hvitved

rust/ast-generator/src/main.rs

@@ -44,6 +44,8 @@ fn property_name(type_name: &str, field_name: &str) -> String {
         ("StructField", "expr") => "default",
         ("UseTree", "is_star") => "is_glob",
         (_, "ty") => "type_repr",
+        ("Function", "body") => "function_body",
+        ("ClosureExpr", "body") => "closure_body",
         _ if field_name.contains("record") => &field_name.replacen("record", "struct", 1),
         _ => field_name,
     };

rust/extractor/src/generated/.generated.list

@@ -40,13 +40,6 @@ final class CallableScope extends CfgScopeImpl, Callable {
     this instanceof ClosureExpr
   }
 
-  /** Gets the body of this callable. */
-  AstNode getBody() {
-    result = this.(Function).getBody()
-    or
-    result = this.(ClosureExpr).getBody()
-  }
-
   override predicate scopeFirst(AstNode first) {
     first(this.(CallableScopeTree).getFirstChildTree(), first)
   }

rust/extractor/src/generated/top.rs

@@ -29,5 +29,7 @@ module Impl {
    */
   class ClosureExpr extends Generated::ClosureExpr {
     override string toStringImpl() { result = "|...| " + this.getBody().toAbbreviatedString() }
+
+    override Expr getBody() { result = this.getClosureBody() }
   }
 }

rust/extractor/src/translate/generated.rs

@@ -69,5 +69,7 @@ module Impl {
       this.getLocation().getStartLine() <= result.getLocation().getStartLine() and
       result.getLocation().getStartLine() <= this.getName().getLocation().getStartLine()
     }
+
+    override BlockExpr getBody() { result = this.getFunctionBody() }
   }
 }

rust/ql/.generated.list

@@ -309,7 +309,7 @@ module Impl {
   private predicate parameterDeclInScope(Variable v, VariableScope scope) {
     exists(Callable f |
       v.getParameter() = f.getParamList().getAParamBase() and
-      scope = [f.(Function).getBody(), f.(ClosureExpr).getBody()]
+      scope = f.getBody()
     )
   }