Coverage for /pythoncovmergedfiles/medio/medio/src/pcap_fuzzer.py: 35%
Shortcuts on this page
r m x toggle line displays
j k next/prev highlighted chunk
0 (zero) top of page
1 (one) first highlighted chunk
Shortcuts on this page
r m x toggle line displays
j k next/prev highlighted chunk
0 (zero) top of page
1 (one) first highlighted chunk
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 2022 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 sys
29import atheris
31with atheris.instrument_imports():
32 import io
33 from nfstream import NFStreamer
36def TestOneInput(input_bytes):
37 with open('fuzz_one_input.pcap', 'wb') as w:
38 # Save it as binary file with .pcap extension
39 w.write(io.BytesIO(input_bytes).read())
40 try:
41 for _ in NFStreamer(source="fuzz_one_input.pcap"):
42 pass
43 except (ValueError, TypeError):
44 pass
47def main():
48 atheris.Setup(sys.argv, TestOneInput)
49 atheris.Fuzz()
52if __name__ == "__main__":
53 main()