fix a potential crash in primes

uiua-lang · Jun 10, 2024 · f1c88fd · f1c88fd
1 parent 3994c1d
commit f1c88fd
Showing 1 changed file with 7 additions and 1 deletion.
diff --git a/src/algorithm/monadic.rs b/src/algorithm/monadic.rs
@@ -1432,10 +1432,16 @@ impl Array<f64> {
  }
  max = max.max(n as usize);
  }
+ if max.overflowing_mul(max).1 {
+ return Err(env.error(format!(
+ "Cannot get primes of {} because it is too large",
+ max
+ )));
+ }
  validate_size::<usize>([max, 2], env)?;
 
  // Sieve of Eratosthenes
- let mut spf = vec![0; max + 1];
+ let mut spf = vec![1; max + 1];
  for i in 2..=max {
  spf[i] = i;
  }