Coverage for /pythoncovmergedfiles/medio/medio/src/fuzz_fromregex_loader.py: 2%
Shortcuts on this page
r m x toggle line displays
j k next/prev highlighted chunk
0 (zero) top of page
1 (one) first highlighted chunk
Shortcuts on this page
r m x toggle line displays
j k next/prev highlighted chunk
0 (zero) top of page
1 (one) first highlighted chunk
1###### Coverage stub
2import atexit
3import coverage
4cov = coverage.coverage(data_file='.coverage', cover_pylib=True)
5cov.start()
6# Register an exist handler that will print coverage
7def exit_handler():
8 cov.stop()
9 cov.save()
10atexit.register(exit_handler)
11####### End of coverage stub
12#!/usr/bin/python3
13# Copyright 2022 Google LLC
14#
15# Licensed under the Apache License, Version 2.0 (the "License");
16# you may not use this file except in compliance with the License.
17# You may obtain a copy of the License at
18#
19# http://www.apache.org/licenses/LICENSE-2.0
20#
21# Unless required by applicable law or agreed to in writing, software
22# distributed under the License is distributed on an "AS IS" BASIS,
23# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
24# See the License for the specific language governing permissions and
25# limitations under the License.
27import atheris
28import os
29import re
30import sys
31import tempfile
32from tokenize import TokenError
33from zipfile import BadZipFile
35with atheris.instrument_imports():
36 import numpy as np
38def TestOneInput(input_bytes):
39 fdp = atheris.FuzzedDataProvider(input_bytes)
40 input_str = fdp.ConsumeString(20)
41 try:
42 dtype = np.dtype(input_str)
43 except (ValueError, TypeError, SyntaxError):
44 return
46 input_str = fdp.ConsumeString(20)
47 try:
48 reg = re.compile(input_str)
49 except (ValueError, TypeError, re.error,OverflowError):
50 return
52 with tempfile.NamedTemporaryFile(suffix=".txt", delete=False) as fd:
53 fd.write(fdp.ConsumeBytes(5000))
54 tmpname = fd.name
56 try:
57 a = np.fromregex(tmpname, reg, dtype)
58 if a.shape != (0,):
59 print(a.shape)
60 except OSError:
61 return
62 except TypeError:
63 return
64 except ValueError:
65 return
66 except EOFError:
67 return
68 finally:
69 os.remove(tmpname)
71def main():
72 atheris.instrument_all()
73 atheris.Setup(sys.argv, TestOneInput, enable_python_coverage=True)
74 atheris.Fuzz()
76if __name__ == "__main__":
77 main()