feat: support max literal (#4678)

Author: mattsse

config/src/fuzz.rs

@@ -30,9 +30,11 @@ pub struct FuzzConfig {
     /// Once the fuzzer exceeds this limit, it will start evicting random entries
     ///
     /// This limit is put in place to prevent memory blowup.
+    #[serde(deserialize_with = "crate::deserialize_usize_or_max")]
     pub max_fuzz_dictionary_addresses: usize,
     /// How many values to record at most.
     /// Once the fuzzer exceeds this limit, it will start evicting random entries
+    #[serde(deserialize_with = "crate::deserialize_usize_or_max")]
     pub max_fuzz_dictionary_values: usize,
 }
 

config/src/utils.rs

@@ -4,7 +4,7 @@ use crate::Config;
 use ethers_core::types::{serde_helpers::Numeric, U256};
 use ethers_solc::remappings::{Remapping, RemappingError};
 use figment::value::Value;
-use serde::{Deserialize, Deserializer};
+use serde::{de::Error, Deserialize, Deserializer};
 use std::{
     path::{Path, PathBuf},
     str::FromStr,
@@ -182,3 +182,30 @@ where
         Err(serde::de::Error::custom("percent must be lte 100"))
     }
 }
+
+/// Deserialize an usize or
+pub(crate) fn deserialize_usize_or_max<'de, D>(deserializer: D) -> Result<usize, D::Error>
+where
+    D: Deserializer<'de>,
+{
+    #[derive(Deserialize)]
+    #[serde(untagged)]
+    enum Val {
+        Number(usize),
+        Text(String),
+    }
+
+    let num = match Val::deserialize(deserializer)? {
+        Val::Number(num) => num,
+        Val::Text(s) => {
+            match s.as_str() {
+                "max" | "MAX" | "Max" => {
+                    // toml limitation
+                    i64::MAX as usize
+                }
+                s => s.parse::<usize>().map_err(D::Error::custom).unwrap(),
+            }
+        }
+    };
+    Ok(num)
+}

forge/src/runner.rs

@@ -287,7 +287,7 @@ impl<'a> ContractRunner<'a> {
                                 *should_fail,
                                 test_options.fuzzer(),
                                 setup.clone(),
-                                test_options.fuzz.clone(),
+                                test_options.fuzz,
                             )
                         } else {
                             self.clone().run_test(func, *should_fail, setup.clone())