topdown: Fix panic in rand.intn function

This commit just puts an error check in place to the panic from the stdlib when rand.intn is called with an input < 1. Signed-off-by: Torin Sandall <[email protected]>
mj520 · Jul 12, 2021 · a1ac763 · a1ac763
1 parent abb8d0c
commit a1ac763
Showing 1 changed file with 2 additions and 0 deletions.
diff --git a/topdown/numbers.go b/topdown/numbers.go
@@ -66,6 +66,8 @@ func builtinRandIntn(bctx BuiltinContext, args []*ast.Term, iter func(*ast.Term)
 	n, err := builtins.IntOperand(args[1].Value, 2)
 	if err != nil {
 		return err
+	} else if n <= 0 {
+		return fmt.Errorf("invalid argument (n must be > 0)")
 	}
 
 	var key = uuidCachingKey(fmt.Sprintf("%s-%d", strOp, n))