websocket-client: initial integration (#8243)

* websocket-client: initial integration

* nit
This commit is contained in:
DavidKorczynski 2022-08-19 14:36:46 +01:00 committed by GitHub
parent 8b50e8f07b
commit 6754771bec
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
5 changed files with 166 additions and 0 deletions

View File

@ -0,0 +1,21 @@
# Copyright 2022 Google LLC
#
# Licensed under the Apache License, Version 2.0 (the "License");
# you may not use this file except in compliance with the License.
# You may obtain a copy of the License at
#
# http://www.apache.org/licenses/LICENSE-2.0
#
# Unless required by applicable law or agreed to in writing, software
# distributed under the License is distributed on an "AS IS" BASIS,
# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
# See the License for the specific language governing permissions and
# limitations under the License.
#
################################################################################
FROM gcr.io/oss-fuzz-base/base-builder-python
RUN apt-get update && apt-get install -y make autoconf automake libtool
RUN git clone --depth 1 https://github.com/websocket-client/websocket-client websocket-client
WORKDIR websocket-client
COPY build.sh *.py $SRC/

View File

@ -0,0 +1,22 @@
#!/bin/bash -eu
# Copyright 2022 Google LLC
#
# Licensed under the Apache License, Version 2.0 (the "License");
# you may not use this file except in compliance with the License.
# You may obtain a copy of the License at
#
# http://www.apache.org/licenses/LICENSE-2.0
#
# Unless required by applicable law or agreed to in writing, software
# distributed under the License is distributed on an "AS IS" BASIS,
# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
# See the License for the specific language governing permissions and
# limitations under the License.
#
################################################################################
pip3 install .
for fuzzer in $(find $SRC -name 'fuzz_*.py'); do
compile_python_fuzzer $fuzzer
done

View File

@ -0,0 +1,74 @@
#!/usr/bin/python3
# Copyright 2022 Google LLC
#
# Licensed under the Apache License, Version 2.0 (the "License");
# you may not use this file except in compliance with the License.
# You may obtain a copy of the License at
#
# http://www.apache.org/licenses/LICENSE-2.0
#
# Unless required by applicable law or agreed to in writing, software
# distributed under the License is distributed on an "AS IS" BASIS,
# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
# See the License for the specific language governing permissions and
# limitations under the License.
import os
import sys
import atheris
from websocket._exceptions import WebSocketException
from websocket._http import read_headers, _tunnel
class FuzzMock:
"""FuzzMock class inspired by SockMock"""
def __init__(self, packet):
self.data = [packet]
self.sent = []
def gettimeout(self):
return None
def recv(self, bufsize):
if self.data:
e = self.data.pop(0)
if isinstance(e, Exception):
raise e
if len(e) > bufsize:
self.data.insert(0, e[bufsize:])
return e[:bufsize]
def send(self, data):
self.sent.append(data)
return len(data)
def TestOneInput(data):
try:
fdp = atheris.FuzzedDataProvider(data)
# Create mock socket with fuzzer-derived data.
mock_sock = FuzzMock(fdp.ConsumeBytes(512))
_tunnel(
mock_sock,
fdp.ConsumeUnicodeNoSurrogates(200),
fdp.ConsumeIntInRange(1, 1000),
(
fdp.ConsumeUnicodeNoSurrogates(100),
fdp.ConsumeUnicodeNoSurrogates(100)
)
)
except WebSocketException:
pass
except UnicodeDecodeError:
pass
def main():
atheris.instrument_all()
atheris.Setup(sys.argv, TestOneInput, enable_python_coverage=True)
atheris.Fuzz()
if __name__ == "__main__":
main()

View File

@ -0,0 +1,39 @@
#!/usr/bin/python3
# Copyright 2022 Google LLC
#
# Licensed under the Apache License, Version 2.0 (the "License");
# you may not use this file except in compliance with the License.
# You may obtain a copy of the License at
#
# http://www.apache.org/licenses/LICENSE-2.0
#
# Unless required by applicable law or agreed to in writing, software
# distributed under the License is distributed on an "AS IS" BASIS,
# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
# See the License for the specific language governing permissions and
# limitations under the License.
import os
import sys
import atheris
from websocket._url import parse_url
def TestOneInput(data):
fdp = atheris.FuzzedDataProvider(data)
try:
parse_url(fdp.ConsumeUnicodeNoSurrogates(20))
except ValueError as e:
# Catch all ValueError exceptions. This is also
# because urllib will throw a set of them which
# is not visible from the websocket-client code.
return
def main():
atheris.instrument_all()
atheris.Setup(sys.argv, TestOneInput, enable_python_coverage=True)
atheris.Fuzz()
if __name__ == "__main__":
main()

View File

@ -0,0 +1,10 @@
homepage: "https://github.com/websocket-client/websocket-client"
language: python
main_repo: "https://github.com/websocket-client/websocket-client"
fuzzing_engines:
- libfuzzer
sanitizers:
- address
- undefined
vendor_ccs:
- david@adalogics.com