commit 3e4e72f66f4e9d379d7734b5d0de92fc0b4d9596
author Amaury Forgeot d'Arc <amauryfa@gmail.com> Tue Nov 11 20:05:06 2008 +0000
committer Amaury Forgeot d'Arc <amauryfa@gmail.com> Tue Nov 11 20:05:06 2008 +0000
tree 4eee1f9789483a66f079269f0c3c2f194b5910be
parent 3bae65bacdc7c03e665fabb2065c740358ac362f

#4298: pickle.load() can segfault on invalid or truncated input.

Patch and test by Hirokazu Yamamoto.

Lib/test/pickletester.py

diff --git a/Lib/test/pickletester.py b/Lib/test/pickletester.py
index 8519fb5..c7c89d1 100644
--- a/Lib/test/pickletester.py
+++ b/Lib/test/pickletester.py
@@ -1032,6 +1032,11 @@
         self.assertRaises(pickle.PicklingError, BadPickler().dump, 0)
         self.assertRaises(pickle.UnpicklingError, BadUnpickler().load)
 
+    def test_bad_input(self):
+        # Test issue4298
+        s = bytes([0x58, 0, 0, 0, 0x54])
+        self.assertRaises(EOFError, pickle.loads, s)
+
 
 class AbstractPersistentPicklerTests(unittest.TestCase):