python-protobuf: extend fuzzer (#7543)

1 files changed, 11 insertions, 13 deletions

diff --git a/projects/protobuf-python/fuzz_protobuf.py b/projects/protobuf-python/fuzz_protobuf.py
index 02c24ce5..b614df3a 100644
--- a/projects/protobuf-python/fuzz_protobuf.py
+++ b/projects/protobuf-python/fuzz_protobuf.py
@@ -19,23 +19,21 @@ import atheris
 
 with atheris.instrument_imports():
   import test_full_pb2
-  from google.protobuf.message import DecodeError
+  from google.protobuf.message import DecodeError, EncodeError
 
 @atheris.instrument_func
 def TestOneInput(input_bytes):
-  # We need to make the file an absolute path
-  testfile_path = os.path.join(os.getcwd(), "serialized.bin")
-  with open(testfile_path, "wb") as f:
-    f.write(input_bytes)
-
+  """Test ParseFromString with bytes string"""
   pbmsg = test_full_pb2.TestMessSubMess()
-  with open(testfile_path, "rb") as fd:
-    try:
-      pbmsg.ParseFromString(fd.read())
-    except DecodeError:
-      None
-
-  os.remove(testfile_path)
+  try:
+    pbmsg.ParseFromString(input_bytes)
+  except DecodeError:
+    None
+
+  try:
+    pbmsg.SerializeToString()
+  except EncodeError:
+    None
 
 
 def main():