Update afl/src/lib.rs

Co-authored-by: Samuel Moelius <[email protected]>
rust-fuzz · Dec 15, 2023 · a86ef47 · a86ef47
1 parent b5a679b
commit a86ef47
Showing 1 changed file with 7 additions and 4 deletions.
diff --git a/afl/src/lib.rs b/afl/src/lib.rs
@@ -70,10 +70,13 @@ where
 
     let mut input = vec![];
 
-    let loop_count: usize = env::var("AFL_FUZZER_LOOPCOUNT")
-        .unwrap_or_else(|_| u32::MAX.to_string())
-        .parse::<usize>()
-        .expect("Failed to parse environment variable to a number");
+    let loop_count = if let Ok(value) = env::var("AFL_FUZZER_LOOPCOUNT") {
+        value
+            .parse()
+            .expect("Failed to parse environment variable to a number")
+    } else {
+        usize::MAX
+    };      
 
     // initialize forkserver there
     unsafe { __afl_manual_init() };