Coverage for /pythoncovmergedfiles/medio/medio/src/fuzz_lexers.py: 32%
25 statements
« prev ^ index » next coverage.py v7.2.2, created at 2023-03-26 07:45 +0000
« prev ^ index » next coverage.py v7.2.2, created at 2023-03-26 07:45 +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
14# Copyright 2020 Google LLC
15#
16# Licensed under the Apache License, Version 2.0 (the "License");
17# you may not use this file except in compliance with the License.
18# You may obtain a copy of the License at
19#
20# http://www.apache.org/licenses/LICENSE-2.0
21#
22# Unless required by applicable law or agreed to in writing, software
23# distributed under the License is distributed on an "AS IS" BASIS,
24# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
25# See the License for the specific language governing permissions and
26# limitations under the License.
28import atheris
30import sys
31import pygments
32from pygments.formatters import *
33import pygments.lexers
35# pygments.LEXERS.values() is a list of tuples like this, with some of then empty:
36# (textual class name, longname, tuple of aliases, tuple of filename patterns, tuple of mimetypes)
37LEXERS = [l[2][0] for l in pygments.lexers.LEXERS.values() if l[2]]
38FORMATTERS = [BBCodeFormatter(), GroffFormatter(), HtmlFormatter(),
39 IRCFormatter(), LatexFormatter(), NullFormatter(),
40 PangoMarkupFormatter(), RawTokenFormatter(), RtfFormatter(),
41 SvgFormatter(), Terminal256Formatter(), TerminalFormatter(),
42 TerminalTrueColorFormatter()]
45def TestOneInput(data: bytes) -> int:
46 fdp = atheris.FuzzedDataProvider(data)
47 random_lexer = pygments.lexers.get_lexer_by_name(fdp.PickValueInList(LEXERS))
48 formatter = fdp.PickValueInList(FORMATTERS)
49 str_data = fdp.ConsumeUnicode(atheris.ALL_REMAINING)
51 pygments.highlight(str_data, random_lexer, formatter)
52 return 0
55atheris.instrument_all()
56atheris.Setup(sys.argv, TestOneInput)
57atheris.Fuzz()