fix: IOOB exception when arg count is invalid in SetDirectiveHandler

AndroidIDEOfficial · Mar 17, 2024 · fb827ce · fb827ce
1 parent 92c1b11
commit fb827ce
Showing 1 changed file with 4 additions and 0 deletions.
diff --git a/...-sitter/src/main/java/com/itsaky/androidide/treesitter/predicate/SetDirectiveHandler.java b/...-sitter/src/main/java/com/itsaky/androidide/treesitter/predicate/SetDirectiveHandler.java
@@ -54,6 +54,10 @@ public String[] getSupportedPredicates() {
   @Override
   public Result handle(TSQuery query, TSQueryMatch match, List<PredicateStep> args
   ) {
+    if (args.size() < PARAMETERS_1.length || args.size() > PARAMETERS_2.length) {
+      return Result.UNHANDLED;
+    }
+
     if (!DIRECTIVE.equals(args.get(0).value)) {
       return Result.UNHANDLED;
     }