Commit feac6cd7 authored by Abhilash Raj's avatar Abhilash Raj Committed by R. David Murray

bpo-33524: Fix the folding of email header when max_line_length is 0 or None (#13391)

and there are non-ascii characters in the header.
parent cbe72d84
...@@ -68,6 +68,7 @@ XXX: provide complete list of token types. ...@@ -68,6 +68,7 @@ XXX: provide complete list of token types.
""" """
import re import re
import sys
import urllib # For urllib.parse.unquote import urllib # For urllib.parse.unquote
from string import hexdigits from string import hexdigits
from operator import itemgetter from operator import itemgetter
...@@ -2590,7 +2591,7 @@ def _refold_parse_tree(parse_tree, *, policy): ...@@ -2590,7 +2591,7 @@ def _refold_parse_tree(parse_tree, *, policy):
""" """
# max_line_length 0/None means no limit, ie: infinitely long. # max_line_length 0/None means no limit, ie: infinitely long.
maxlen = policy.max_line_length or float("+inf") maxlen = policy.max_line_length or sys.maxsize
encoding = 'utf-8' if policy.utf8 else 'us-ascii' encoding = 'utf-8' if policy.utf8 else 'us-ascii'
lines = [''] lines = ['']
last_ew = None last_ew = None
......
...@@ -3,6 +3,7 @@ code that adds all the email6 features. ...@@ -3,6 +3,7 @@ code that adds all the email6 features.
""" """
import re import re
import sys
from email._policybase import Policy, Compat32, compat32, _extend_docstrings from email._policybase import Policy, Compat32, compat32, _extend_docstrings
from email.utils import _has_surrogates from email.utils import _has_surrogates
from email.headerregistry import HeaderRegistry as HeaderRegistry from email.headerregistry import HeaderRegistry as HeaderRegistry
...@@ -203,7 +204,7 @@ class EmailPolicy(Policy): ...@@ -203,7 +204,7 @@ class EmailPolicy(Policy):
def _fold(self, name, value, refold_binary=False): def _fold(self, name, value, refold_binary=False):
if hasattr(value, 'name'): if hasattr(value, 'name'):
return value.fold(policy=self) return value.fold(policy=self)
maxlen = self.max_line_length if self.max_line_length else float('inf') maxlen = self.max_line_length if self.max_line_length else sys.maxsize
lines = value.splitlines() lines = value.splitlines()
refold = (self.refold_source == 'all' or refold = (self.refold_source == 'all' or
self.refold_source == 'long' and self.refold_source == 'long' and
......
import io import io
import sys
import types import types
import textwrap import textwrap
import unittest import unittest
...@@ -134,6 +135,18 @@ class PolicyAPITests(unittest.TestCase): ...@@ -134,6 +135,18 @@ class PolicyAPITests(unittest.TestCase):
for attr, value in expected.items(): for attr, value in expected.items():
self.assertEqual(getattr(added, attr), value) self.assertEqual(getattr(added, attr), value)
def test_fold_zero_max_line_length(self):
expected = 'Subject: =?utf-8?q?=C3=A1?=\n'
msg = email.message.EmailMessage()
msg['Subject'] = 'á'
p1 = email.policy.default.clone(max_line_length=0)
p2 = email.policy.default.clone(max_line_length=None)
self.assertEqual(p1.fold('Subject', msg['Subject']), expected)
self.assertEqual(p2.fold('Subject', msg['Subject']), expected)
def test_register_defect(self): def test_register_defect(self):
class Dummy: class Dummy:
def __init__(self): def __init__(self):
......
Fix the folding of email header when the max_line_length is 0 or None and the
header contains non-ascii characters. Contributed by Licht Takeuchi
(@Licht-T).
Markdown is supported
0%
or
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment