cpython: 45cc298d40eb (original) (raw)

Mercurial > cpython

changeset 68875:45cc298d40eb 3.1

#11590: fix quoprimime decode handling of empty strings and line endings. [#11590]

line wrap: on

line diff

--- a/Lib/email/quoprimime.py +++ b/Lib/email/quoprimime.py @@ -135,9 +135,9 @@ def header_encode(header_bytes, charset= charset names the character set to use in the RFC 2046 header. It defaults to iso-8859-1. """

• Return empty headers unchanged

• Return empty headers as an empty string.

  if not header_bytes:

•    return str(header_bytes)[](#l1.10)

•    return ''[](#l1.11)

  Iterate over every byte, encoding if necessary.

  encoded = [] for octet in header_bytes: @@ -268,7 +268,7 @@ def decode(encoded, eol=NL): if i == n: decoded += eol

  Special case if original string did not end with eol

• if not encoded.endswith(eol) and decoded.endswith(eol):

• if encoded[-1] not in '\r\n' and decoded.endswith(eol): decoded = decoded[:-1] return decoded

--- a/Lib/email/test/test_email.py +++ b/Lib/email/test/test_email.py @@ -2890,6 +2890,9 @@ class TestQuopri(unittest.TestCase): encoded_header = quoprimime.header_encode(header, charset) self.assertEqual(encoded_header, expected_encoded_header)

• def test_header_encode_null(self):

•    self._test_header_encode(b'', '')[](#l2.8)

+ def test_header_encode_one_word(self): self._test_header_encode(b'hello', '=?iso-8859-1?q?hello?=') @@ -2946,6 +2949,15 @@ class TestQuopri(unittest.TestCase): def test_decode_one_line_lf(self): self._test_decode('hello\n', 'hello\n')

• def test_decode_one_line_cr(self):

•    self._test_decode('hello\r', 'hello\n')[](#l2.18)

+

• def test_decode_one_line_nl(self):

•    self._test_decode('hello\n', 'helloX', eol='X')[](#l2.21)

+

• def test_decode_one_line_crnl(self):

•    self._test_decode('hello\r\n', 'helloX', eol='X')[](#l2.24)

+ def test_decode_one_line_one_word(self): self._test_decode('hello\r\nworld', 'hello\nworld') @@ -2955,6 +2967,9 @@ class TestQuopri(unittest.TestCase): def test_decode_two_lines(self): self._test_decode('hello\r\nworld\r\n', 'hello\nworld\n')

• def test_decode_two_lines_eol(self):

•    self._test_decode('hello\r\nworld\r\n', 'helloXworldX', eol='X')[](#l2.34)

+ def test_decode_one_long_line(self): self._test_decode('Spam' * 250, 'Spam' * 250)