Rename Fuzz Harness

mayhemheroes · Sep 30, 2024 · 537c644 · 537c644
1 parent 33b3f3c
commit 537c644
Show file tree

Hide file tree

Showing 3 changed files with 5 additions and 7 deletions.
diff --git a/Dockerfile b/Dockerfile
@@ -3,4 +3,4 @@ RUN pip3 install atheris
 
 COPY . /symspellpy
 WORKDIR /symspellpy
-RUN python3 -m pip install . && chmod +x fuzz/fuzz.py
+RUN python3 -m pip install . && chmod +x fuzz/fuzz_spellcheck.py
diff --git a/Mayhemfile b/Mayhemfile
@@ -1,6 +1,6 @@
 project: symspellpy
-target:  fuzz
+target: fuzz_spellcheck
 
 cmds:
-    - cmd: fuzz/fuzz.py
+    - cmd: fuzz/fuzz_spellcheck.py
       libfuzzer: true
diff --git a/fuzz/fuzz.py → fuzz/fuzz_spellcheck.py b/fuzz/fuzz.py → fuzz/fuzz_spellcheck.py
@@ -19,10 +19,8 @@
 
 @atheris.instrument_func
 def TestOneInput(data):
-    try:
-        input_term = data.decode("utf-8")
-    except UnicodeDecodeError:
-        return
+    fdp = atheris.FuzzedDataProvider(data)
+    input_term = fdp.ConsumeUnicodeNoSurrogates(len(data))
 
     sym_spell.lookup_compound(input_term, max_edit_distance=2)