Coverage for /pythoncovmergedfiles/medio/medio/src/fuzz_curs_exec.py: 59%
46 statements
« prev ^ index » next coverage.py v7.3.1, created at 2023-09-25 06:04 +0000
« prev ^ index » next coverage.py v7.3.1, created at 2023-09-25 06:04 +0000
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.
26"""Fuzzer that targets native code of pyodbc with a fake odbc driver"""
27import os
28import sys
29import atheris
30import pyodbc
32@atheris.instrument_func
33def fuzz_exec(data):
34 fdp = atheris.FuzzedDataProvider(data)
35 connstr = "DRIVER=FUZZ;"
36 s1 = fdp.ConsumeUnicodeNoSurrogates(50)
37 if "DRIVER" in s1:
38 return
39 cstr = connstr + s1
40 connection_obj = pyodbc.connect(cstr)
41 csr = connection_obj.cursor()
42 try:
43 csr.execute(fdp.ConsumeUnicodeNoSurrogates(20))
44 except Exception as e:
45 if "Invalid string or buffer length" in str(e):
46 pass
47 else:
48 raise e
49 return 0
51@atheris.instrument_func
52def TestOneInput(data):
53 try:
54 return fuzz_exec(data)
55 except SystemError:
56 return 0
59def main():
60 # Write the odbcinst.ini file
61 dir_path = os.path.dirname(os.path.realpath(__file__))
62 with open("/tmp/odbcinst.ini", "w") as f:
63 f.write("[FUZZ]\n")
64 f.write("Driver=%s/fuzzodbc.so\n"%(dir_path))
65 os.environ['ODBCSYSINI'] = '/tmp/'
67 atheris.instrument_all()
68 atheris.Setup(sys.argv, TestOneInput, enable_python_coverage=True)
69 atheris.Fuzz()
72if __name__ == "__main__":
73 main()