Modifying extraction of global statements to omit local function decl…

…arations. Signed-off-by: Adam Storek <[email protected]>
mandiant · Jul 5, 2022 · 2fdcac2 · 2fdcac2
1 parent 726d68e
commit 2fdcac2
Showing 1 changed file with 3 additions and 1 deletion.
diff --git a/capa/features/extractors/ts/query.py b/capa/features/extractors/ts/query.py
@@ -35,6 +35,8 @@ def __init__(self, language: str):
             self.namespace = self.language.query(
                 "(using_directive [(identifier) @namespace (qualified_name) @namespace])"
             )
-            self.global_statement = self.language.query("(global_statement) @global-statement")
+            self.global_statement = self.language.query(
+                "(global_statement [(expression_statement) @global-statement (local_declaration_statement) @global-statement])"
+            )
         else:
             raise NotImplementedError(f"Tree-sitter queries for {language} are not implemented.")