blob: 6906720a49d3a83b92b3386ebe57aed2f2e488bb (
plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
|
import sys
import atheris
with atheris.instrument_imports():
from websockets.exceptions import SecurityError
from websockets.http11 import Response
from websockets.streams import StreamReader
def test_one_input(data):
reader = StreamReader()
reader.feed_data(data)
reader.feed_eof()
parser = Response.parse(
reader.read_line,
reader.read_exact,
reader.read_to_eof,
)
try:
next(parser)
except StopIteration as exc:
assert isinstance(exc.value, Response)
return # input accepted
except (
EOFError, # connection is closed without a full HTTP response
SecurityError, # response exceeds a security limit
LookupError, # response isn't well formatted
ValueError, # response isn't well formatted
):
return # input rejected with a documented exception
raise RuntimeError("parsing didn't complete")
def main():
atheris.Setup(sys.argv, test_one_input)
atheris.Fuzz()
if __name__ == "__main__":
main()
|
