Coverage for /pythoncovmergedfiles/medio/medio/src/fuzz_util.py: 46%

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 

27import atheris 

28import sys 

29import binascii 

30with atheris.instrument_imports(): 

31 from adal.util import * 

32 

33def TestInput(data): 

34 fdp = atheris.FuzzedDataProvider(data) 

35 

36 is_http_success(fdp.ConsumeInt(10)) 

37 

38 class DummyClass(object): 

39 _call_context = { 

40 'log_context':{'correlation_id':fdp.ConsumeString(20)}, 

41 'options':{'http':{ 

42 fdp.ConsumeString(10):fdp.ConsumeString(100), 

43 fdp.ConsumeString(10):fdp.ConsumeString(100), 

44 fdp.ConsumeString(10):fdp.ConsumeString(100) 

45 }} 

46 } 

47 

48 def geturl(self): 

49 return fdp.ConsumeString(100) 

50 

51 create_request_options(DummyClass(),{ 

52 fdp.ConsumeString(10):fdp.ConsumeString(100), 

53 fdp.ConsumeString(10):fdp.ConsumeString(100), 

54 fdp.ConsumeString(10):fdp.ConsumeString(100), 

55 'headers':{ 

56 fdp.ConsumeString(10):fdp.ConsumeString(100), 

57 fdp.ConsumeString(10):fdp.ConsumeString(100), 

58 fdp.ConsumeString(10):fdp.ConsumeString(100) 

59 } 

60 }) 

61 

62 try: 

63 copy_url(fdp.ConsumeString(100)) 

64 copy_url(DummyClass()) 

65 except ValueError as e: 

66 if (("contains invalid characters" not in str(e)) and 

67 ("Invalid IPv6 URL" not in str(e))): 

68 raise e 

69 

70 try: 

71 base64_urlsafe_decode(fdp.ConsumeUnicodeNoSurrogates(100)) 

72 except binascii.Error as e: 

73 if "Invalid base64-encoded string" not in str(e): 

74 raise e 

75 

76def main(): 

77 atheris.Setup(sys.argv, TestInput, enable_python_coverage=True) 

78 atheris.Fuzz() 

79 

80if __name__ == "__main__": 

81 main() 

82