Coverage for /pythoncovmergedfiles/medio/medio/src/fuzz_menu.py: 52%
46 statements
« prev ^ index » next coverage.py v7.2.7, created at 2023-06-07 06:37 +0000
« prev ^ index » next coverage.py v7.2.7, created at 2023-06-07 06:37 +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.
27import os
28import sys
29import atheris
31with atheris.instrument_imports(enable_loader_override=False):
32 from lxml import etree as et
33 import xml.etree.ElementTree as etree
35 #import etree
36 import xdg.Menu
37 from xdg.Exceptions import ParsingError
39@atheris.instrument_func
40def verify_xml(data):
41 try:
42 root = et.XML(data)
43 if root != None:
44 et.indent(root)
45 et.ElementTree(root)
46 return True
47 except et.XMLSyntaxError:
48 return False
50@atheris.instrument_func
51def TestOneInput(input_bytes):
52 # This is a slight hack to verify xml with code
53 # that gets instrumented. For some reason I was
54 # unable to instrument xml itself.
55 if not verify_xml(input_bytes):
56 return
58 # We need to make the file an absolute path
59 testfile_path = os.path.join(os.getcwd(), "test.menu")
60 with open(testfile_path, "wb") as f:
61 f.write(input_bytes)
63 try:
64 xdg.Menu.parse(filename = testfile_path)
65 except ParsingError:
66 None
67 except RecursionError: # Just catch it because it's not that interesting.
68 None
69 os.remove(testfile_path)
71def main():
72 atheris.instrument_all()
73 atheris.Setup(sys.argv, TestOneInput, enable_python_coverage=True)
74 atheris.Fuzz()
77if __name__ == "__main__":
78 main()