diff options
| author | Serhiy Storchaka <storchaka@gmail.com> | 2015-01-26 08:37:01 (GMT) |
|---|---|---|
| committer | Serhiy Storchaka <storchaka@gmail.com> | 2015-01-26 08:37:01 (GMT) |
| commit | 21d7533c4c13489b4b3baae59f9e25cd038fb16b (patch) | |
| tree | 76bae1e02636081cb504fa39fb5a33629f39c380 /Lib | |
| parent | 5106d044ecd6d7a8124dd4b526b77588747e0167 (diff) | |
| download | cpython-21d7533c4c13489b4b3baae59f9e25cd038fb16b.zip cpython-21d7533c4c13489b4b3baae59f9e25cd038fb16b.tar.gz cpython-21d7533c4c13489b4b3baae59f9e25cd038fb16b.tar.bz2 | |
Issue #23094: Fixed readline with frames in Python implementation of pickle.
Diffstat (limited to 'Lib')
| -rw-r--r-- | Lib/pickle.py | 2 | ||||
| -rw-r--r-- | Lib/test/pickletester.py | 8 |
2 files changed, 9 insertions, 1 deletions
diff --git a/Lib/pickle.py b/Lib/pickle.py index 663773f..e38ecac 100644 --- a/Lib/pickle.py +++ b/Lib/pickle.py @@ -242,7 +242,7 @@ class _Unframer: if not data: self.current_frame = None return self.file_readline() - if data[-1] != b'\n': + if data[-1] != b'\n'[0]: raise UnpicklingError( "pickle exhausted before end of frame") return data diff --git a/Lib/test/pickletester.py b/Lib/test/pickletester.py index 5963175..bdc7bad 100644 --- a/Lib/test/pickletester.py +++ b/Lib/test/pickletester.py @@ -1538,6 +1538,14 @@ class AbstractPickleTests(unittest.TestCase): count_opcode(pickle.FRAME, pickled)) self.assertEqual(obj, self.loads(some_frames_pickle)) + def test_frame_readline(self): + pickled = b'\x80\x04\x95\x05\x00\x00\x00\x00\x00\x00\x00I42\n.' + # 0: \x80 PROTO 4 + # 2: \x95 FRAME 5 + # 11: I INT 42 + # 15: . STOP + self.assertEqual(self.loads(pickled), 42) + def test_nested_names(self): global Nested class Nested: |