1#!/usr/bin/python3 2 3# Copyright 2020 Google LLC 4# 5# Licensed under the Apache License, Version 2.0 (the "License"); 6# you may not use this file except in compliance with the License. 7# You may obtain a copy of the License at 8# 9# http://www.apache.org/licenses/LICENSE-2.0 10# 11# Unless required by applicable law or agreed to in writing, software 12# distributed under the License is distributed on an "AS IS" BASIS, 13# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. 14# See the License for the specific language governing permissions and 15# limitations under the License. 16 17import sys 18import atheris 19import pygments 20import pygments.formatters 21import pygments.lexers 22 23 24def TestOneInput(input_bytes): 25 fdp = atheris.FuzzedDataProvider(input_bytes) 26 data = fdp.ConsumeUnicode(atheris.ALL_REMAINING) 27 28 try: 29 lexer = pygments.lexers.guess_lexer(data) 30 except ValueError: 31 return 32 pygments.highlight(data, lexer, pygments.formatters.HtmlFormatter()) 33 34 35def main(): 36 atheris.Setup(sys.argv, TestOneInput, enable_python_coverage=True) 37 atheris.Fuzz() 38 39 40if __name__ == "__main__": 41 main() 42