chore: support address block scope matching

semgrep · Jun 28, 2024 · 7954412 · 7954412
1 parent a7b069b
commit 7954412
Show file tree

Hide file tree

Showing 3 changed files with 29 additions and 2 deletions.
diff --git a/lang/semgrep-grammars/src/semgrep-move-on-aptos/grammar.js b/lang/semgrep-grammars/src/semgrep-move-on-aptos/grammar.js
@@ -79,6 +79,12 @@ module.exports = grammar(base_grammar, {
     _script_func_decl: ($, previous) => choice(previous, $.ellipsis),
     _script_spec_block: ($, previous) => choice(previous, $.ellipsis),
 
+    // Address block members
+    _address_member: ($, previous) => choice(
+      previous,
+      $.ellipsis,
+    ),
+
     // Spec block members
     _spec_block_member: ($, previous) => choice(
       previous,

diff --git a/lang/semgrep-grammars/src/semgrep-move-on-aptos/test/corpus/semgrep.txt b/lang/semgrep-grammars/src/semgrep-move-on-aptos/test/corpus/semgrep.txt
@@ -442,4 +442,25 @@ script {
       (spec_block
         (ellipsis)))
     (declaration
-      (ellipsis))))
+      (ellipsis))))
+
+=======================
+Address Block
+=======================
+
+address $ADDR {
+  ...
+  module $MOD { ... }
+  ...
+}
+---
+
+(source_file
+  (address_block
+    (identifier)
+    (ellipsis)
+    (module
+      (identifier)
+      (declaration
+        (ellipsis)))
+    (ellipsis)))
diff --git a/lang/semgrep-grammars/src/tree-sitter-move-on-aptos b/lang/semgrep-grammars/src/tree-sitter-move-on-aptos