Improve the fuzzer wrt. the current atheris version

This commit is contained in:
Julien Voisin 2021-08-25 16:45:39 +02:00
parent d50052a75c
commit 9d48143567
2 changed files with 12 additions and 12 deletions

View File

@ -14,9 +14,10 @@
# See the License for the specific language governing permissions and # See the License for the specific language governing permissions and
# limitations under the License. # limitations under the License.
import sys
import atheris_no_libfuzzer as atheris import atheris
with atheris.instrument_imports():
import sys
import fuzzers import fuzzers
@ -26,13 +27,12 @@ def TestOneInput(data):
except Exception: except Exception:
# We're catching all exceptions because Pillow's exceptions are # We're catching all exceptions because Pillow's exceptions are
# directly inheriting from Exception. # directly inheriting from Exception.
return pass
return
def main(): def main():
fuzzers.enable_decompressionbomb_error() fuzzers.enable_decompressionbomb_error()
atheris.Setup(sys.argv, TestOneInput, enable_python_coverage=True) atheris.Setup(sys.argv, TestOneInput)
atheris.Fuzz() atheris.Fuzz()
fuzzers.disable_decompressionbomb_error() fuzzers.disable_decompressionbomb_error()

View File

@ -14,9 +14,10 @@
# See the License for the specific language governing permissions and # See the License for the specific language governing permissions and
# limitations under the License. # limitations under the License.
import sys
import atheris_no_libfuzzer as atheris import atheris
with atheris.instrument_imports():
import sys
import fuzzers import fuzzers
@ -26,13 +27,12 @@ def TestOneInput(data):
except Exception: except Exception:
# We're catching all exceptions because Pillow's exceptions are # We're catching all exceptions because Pillow's exceptions are
# directly inheriting from Exception. # directly inheriting from Exception.
return pass
return
def main(): def main():
fuzzers.enable_decompressionbomb_error() fuzzers.enable_decompressionbomb_error()
atheris.Setup(sys.argv, TestOneInput, enable_python_coverage=True) atheris.Setup(sys.argv, TestOneInput)
atheris.Fuzz() atheris.Fuzz()
fuzzers.disable_decompressionbomb_error() fuzzers.disable_decompressionbomb_error()