Update the base runner constructor

Author: StefanosChaliasos

src/zkregex_fuzzer/harness.py

@@ -46,7 +46,7 @@ def harness(regex: str, primary_runner_cls: Type[Runner], secondary_runner_cls:
     regex = regex
     inp_num = len(inputs)
     try:
-        primary_runner = primary_runner_cls(regex)
+        primary_runner = primary_runner_cls(regex, {})
     except RegexCompileError as e:
         return HarnessResult(regex, inp_num, oracle, [], HarnessStatus.INVALID_SEED)
 

src/zkregex_fuzzer/runner/base_runner.py

@@ -31,7 +31,7 @@ class Runner(ABC):
     Abstract base class for regex runners.
     """
 
-    def __init__(self, regex: str):
+    def __init__(self, regex: str, kwargs: dict):
         self._regex = regex
         self._runner = "Abstract runner"
         self._regex_object = self.compile(regex)

src/zkregex_fuzzer/runner/circom.py

@@ -239,9 +239,9 @@ def __init__(self, regex: str, kwargs: dict):
         self._run_the_prover = kwargs.get("circom_prove", False)
         self._ptau_path = kwargs.get("circom_ptau", None)
         self._link_path = kwargs.get("circom_library", [])
-        self._max_input_size = kwargs.get("max_input_size", 200)
+        self._circom_max_input_size = kwargs.get("circom_max_input_size", 200)
         self._template_name = "TestRegex"
-        super().__init__(regex)
+        super().__init__(regex, kwargs)
         self._runner = "Circom"
 
     def compile(self, regex: str) -> None:
@@ -269,7 +269,7 @@ def compile(self, regex: str) -> None:
         # Append the circom file to include the main function
         with open(circom_file_path, 'a') as f:
             f.write("\n\n")
-            f.write("component main {public [msg]} = " + f"{self._template_name}({self._max_input_size});")
+            f.write("component main {public [msg]} = " + f"{self._template_name}({self._circom_max_input_size});")
 
         # Compile the circom code to wasm
         self._wasm_path, self._r1cs_path = CircomSubprocess.compile(circom_file_path, self._link_path)
@@ -284,15 +284,15 @@ def match(self, input: str) -> bool:
         Match the regex on an input.
         """
         # Convert input to list of decimal ASCII values and pad input with zeroes
-        numeric_input = [ord(c) for c in input] + [0] * (self._max_input_size - len(input))
+        numeric_input = [ord(c) for c in input] + [0] * (self._circom_max_input_size - len(input))
 
         # Write input to a temporary JSON file
         with tempfile.NamedTemporaryFile(suffix=".json", delete=False) as tmp_file:
             tmp_file.write(json.dumps({"msg": numeric_input}).encode())
             input_path = tmp_file.name
 
         # Skip if input is larger than circuit max input size
-        if len(numeric_input) > self._max_input_size:
+        if len(numeric_input) > self._circom_max_input_size:
             logger.info(f"Input too large for input: {input}")
             raise RegexRunError(f"Input too large for input: {input}")
 

src/zkregex_fuzzer/runner/python.py

@@ -11,8 +11,8 @@ class PythonReRunner(Runner):
     Runner that uses the Python re module.
     """
 
-    def __init__(self, regex: str):
-        super().__init__(regex)
+    def __init__(self, regex: str, kwargs: dict):
+        super().__init__(regex, kwargs)
         self._runner = "Python re module"
 
     def compile(self, regex: str) -> None: