From e2ab0ee51acf97a8b7127798fda7fd390d10ee12 Mon Sep 17 00:00:00 2001 From: DavidKorczynski Date: Wed, 28 Dec 2022 16:44:54 +0000 Subject: [PATCH] pymysql: initial integration (#9295) Signed-off-by: David Korczynski Signed-off-by: David Korczynski --- projects/pymysql/Dockerfile | 19 ++++++ projects/pymysql/build.sh | 22 +++++++ projects/pymysql/fuzz_connection.py | 96 +++++++++++++++++++++++++++++ projects/pymysql/project.yaml | 10 +++ 4 files changed, 147 insertions(+) create mode 100644 projects/pymysql/Dockerfile create mode 100644 projects/pymysql/build.sh create mode 100644 projects/pymysql/fuzz_connection.py create mode 100644 projects/pymysql/project.yaml diff --git a/projects/pymysql/Dockerfile b/projects/pymysql/Dockerfile new file mode 100644 index 000000000..53eb3c7b1 --- /dev/null +++ b/projects/pymysql/Dockerfile @@ -0,0 +1,19 @@ +#!/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. +FROM gcr.io/oss-fuzz-base/base-builder-python +RUN pip3 install --upgrade pip mock +RUN git clone https://github.com/PyMySQL/PyMySQL pymysql +COPY *.sh *py $SRC/ +WORKDIR $SRC/pymysql diff --git a/projects/pymysql/build.sh b/projects/pymysql/build.sh new file mode 100644 index 000000000..338e12e52 --- /dev/null +++ b/projects/pymysql/build.sh @@ -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 . + +# Build fuzzers in $OUT. +for fuzzer in $(find $SRC -name 'fuzz_*.py'); do + compile_python_fuzzer $fuzzer +done diff --git a/projects/pymysql/fuzz_connection.py b/projects/pymysql/fuzz_connection.py new file mode 100644 index 000000000..07f2e4e38 --- /dev/null +++ b/projects/pymysql/fuzz_connection.py @@ -0,0 +1,96 @@ +#!/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 sys +import atheris +import threading +import socket +import time + +import configparser +import pymysql +from pymysql import connections + + +fuzzed_input = b"" + +# somehow ugly as fuzzing cannot be run in parallel +def SetFuzzedInput(input_bytes): + global fuzzed_input + fuzzed_input = input_bytes + +class ServerThread(threading.Thread): + def __init__(self): + self.s = socket.socket(socket.AF_INET, socket.SOCK_STREAM) + self.s.setsockopt(socket.SOL_SOCKET, socket.SO_REUSEADDR, 1) + self.s.bind(("127.0.0.1", 8001)) + self.s.listen(1) + + threading.Thread.__init__(self) + + def run(self): + global fuzzed_input + conn, addr = self.s.accept() + conn.settimeout(0.3) + try: + conn.recv(1024) + except: + pass + conn.send(fuzzed_input) + time.sleep(0.005) + conn.close() + self.s.shutdown(1) + self.s.close() + time.sleep(0.01) + + +def TestOneInput(data): + fdp = atheris.FuzzedDataProvider(data) + local_config_file = "/tmp/my.cnf" + with open(local_config_file, "w") as cfg: + cfg.write(fdp.ConsumeUnicodeNoSurrogates(1024)) + + # Catch config parser errors + try: + c = connections.Connection( + db=None, + read_default_file=local_config_file, + ssl_disabled=True, + defer_connect=True + ) + except configparser.Error: + return + c.port = 8001 + c.host = "127.0.0.1" + + SetFuzzedInput(fdp.ConsumeBytes(512)) + t1 = ServerThread() + t1.start() + + try: + c.connect(sock=None) + except pymysql.err.MySQLError: + pass + t1.join() + + +def main(): + atheris.instrument_all() + atheris.Setup(sys.argv, TestOneInput, enable_python_coverage=True) + atheris.Fuzz() + + +if __name__ == "__main__": + main() diff --git a/projects/pymysql/project.yaml b/projects/pymysql/project.yaml new file mode 100644 index 000000000..ccfd339fc --- /dev/null +++ b/projects/pymysql/project.yaml @@ -0,0 +1,10 @@ +fuzzing_engines: +- libfuzzer +homepage: https://github.com/PyMySQL/PyMySQL +language: python +main_repo: https://github.com/PyMySQL/PyMySQL +sanitizers: +- address +- undefined +vendor_ccs: +- david@adalogics.com