diff --git a/pythonfuzz/fuzzer.py b/pythonfuzz/fuzzer.py index acbe9d6..8cb8d8b 100644 --- a/pythonfuzz/fuzzer.py +++ b/pythonfuzz/fuzzer.py @@ -19,7 +19,7 @@ def worker(target, child_conn): cov = coverage.Coverage(branch=True, cover_pylib=True) cov.start() while True: - buf = child_conn.recv() + buf = child_conn.recv_bytes() try: target(buf) except Exception as e: @@ -87,9 +87,10 @@ def start(self): self._p = mp.Process(target=worker, args=(self._target, child_conn)) self._p.start() + while True: buf = self._corpus.generate_input() - parent_conn.send(buf) + parent_conn.send_bytes(buf) if not parent_conn.poll(self._timeout): self._p.kill() logging.info("=================================================================")