This commit is contained in:
Stan Ulbrych 2025-12-08 06:11:21 +02:00 committed by GitHub
commit 83f5353598
No known key found for this signature in database
GPG key ID: B5690EEEBB952194
8 changed files with 76 additions and 20 deletions

View file

@ -8,14 +8,20 @@
import codecs
import base64
### Codec Helpers
def _assert_strict(errors):
if errors != 'strict':
raise ValueError(f'Unsupported error handling mode: "{errors}" - must be "strict"')
### Codec APIs
def base64_encode(input, errors='strict'):
assert errors == 'strict'
_assert_strict(errors)
return (base64.encodebytes(input), len(input))
def base64_decode(input, errors='strict'):
assert errors == 'strict'
_assert_strict(errors)
return (base64.decodebytes(input), len(input))
class Codec(codecs.Codec):
@ -26,12 +32,12 @@ def decode(self, input, errors='strict'):
class IncrementalEncoder(codecs.IncrementalEncoder):
def encode(self, input, final=False):
assert self.errors == 'strict'
_assert_strict(self.errors)
return base64.encodebytes(input)
class IncrementalDecoder(codecs.IncrementalDecoder):
def decode(self, input, final=False):
assert self.errors == 'strict'
_assert_strict(self.errors)
return base64.decodebytes(input)
class StreamWriter(Codec, codecs.StreamWriter):

View file

@ -10,14 +10,20 @@
import codecs
import bz2 # this codec needs the optional bz2 module !
### Codec Helpers
def _assert_strict(errors):
if errors != 'strict':
raise ValueError(f'Unsupported error handling mode: "{errors}" - must be "strict"')
### Codec APIs
def bz2_encode(input, errors='strict'):
assert errors == 'strict'
_assert_strict(errors)
return (bz2.compress(input), len(input))
def bz2_decode(input, errors='strict'):
assert errors == 'strict'
_assert_strict(errors)
return (bz2.decompress(input), len(input))
class Codec(codecs.Codec):
@ -28,7 +34,7 @@ def decode(self, input, errors='strict'):
class IncrementalEncoder(codecs.IncrementalEncoder):
def __init__(self, errors='strict'):
assert errors == 'strict'
_assert_strict(errors)
self.errors = errors
self.compressobj = bz2.BZ2Compressor()
@ -44,7 +50,7 @@ def reset(self):
class IncrementalDecoder(codecs.IncrementalDecoder):
def __init__(self, errors='strict'):
assert errors == 'strict'
_assert_strict(errors)
self.errors = errors
self.decompressobj = bz2.BZ2Decompressor()

View file

@ -8,14 +8,20 @@
import codecs
import binascii
### Codec Helpers
def _assert_strict(errors):
if errors != 'strict':
raise ValueError(f'Unsupported error handling mode: "{errors}" - must be "strict"')
### Codec APIs
def hex_encode(input, errors='strict'):
assert errors == 'strict'
_assert_strict(errors)
return (binascii.b2a_hex(input), len(input))
def hex_decode(input, errors='strict'):
assert errors == 'strict'
_assert_strict(errors)
return (binascii.a2b_hex(input), len(input))
class Codec(codecs.Codec):
@ -26,12 +32,12 @@ def decode(self, input, errors='strict'):
class IncrementalEncoder(codecs.IncrementalEncoder):
def encode(self, input, final=False):
assert self.errors == 'strict'
_assert_strict(self.errors)
return binascii.b2a_hex(input)
class IncrementalDecoder(codecs.IncrementalDecoder):
def decode(self, input, final=False):
assert self.errors == 'strict'
_assert_strict(self.errors)
return binascii.a2b_hex(input)
class StreamWriter(Codec, codecs.StreamWriter):

View file

@ -7,15 +7,23 @@
import quopri
from io import BytesIO
### Codec Helpers
def _assert_strict(errors):
if errors != 'strict':
raise ValueError(f'Unsupported error handling mode: "{errors}" - must be "strict"')
### Codec APIs
def quopri_encode(input, errors='strict'):
assert errors == 'strict'
_assert_strict(errors)
f = BytesIO(input)
g = BytesIO()
quopri.encode(f, g, quotetabs=True)
return (g.getvalue(), len(input))
def quopri_decode(input, errors='strict'):
assert errors == 'strict'
_assert_strict(errors)
f = BytesIO(input)
g = BytesIO()
quopri.decode(f, g)

View file

@ -11,10 +11,16 @@
import binascii
from io import BytesIO
### Codec Helpers
def _assert_strict(errors):
if errors != 'strict':
raise ValueError(f'Unsupported error handling mode: "{errors}" - must be "strict"')
### Codec APIs
def uu_encode(input, errors='strict', filename='<data>', mode=0o666):
assert errors == 'strict'
_assert_strict(errors)
infile = BytesIO(input)
outfile = BytesIO()
read = infile.read
@ -35,7 +41,7 @@ def uu_encode(input, errors='strict', filename='<data>', mode=0o666):
return (outfile.getvalue(), len(input))
def uu_decode(input, errors='strict'):
assert errors == 'strict'
_assert_strict(errors)
infile = BytesIO(input)
outfile = BytesIO()
readline = infile.readline

View file

@ -8,14 +8,20 @@
import codecs
import zlib # this codec needs the optional zlib module !
### Codec Helpers
def _assert_strict(errors):
if errors != 'strict':
raise ValueError(f'Unsupported error handling mode: "{errors}" - must be "strict"')
### Codec APIs
def zlib_encode(input, errors='strict'):
assert errors == 'strict'
_assert_strict(errors)
return (zlib.compress(input), len(input))
def zlib_decode(input, errors='strict'):
assert errors == 'strict'
_assert_strict(errors)
return (zlib.decompress(input), len(input))
class Codec(codecs.Codec):
@ -26,7 +32,7 @@ def decode(self, input, errors='strict'):
class IncrementalEncoder(codecs.IncrementalEncoder):
def __init__(self, errors='strict'):
assert errors == 'strict'
_assert_strict(errors)
self.errors = errors
self.compressobj = zlib.compressobj()
@ -42,7 +48,7 @@ def reset(self):
class IncrementalDecoder(codecs.IncrementalDecoder):
def __init__(self, errors='strict'):
assert errors == 'strict'
_assert_strict(errors)
self.errors = errors
self.decompressobj = zlib.decompressobj()

View file

@ -3129,6 +3129,20 @@ def test_uu_invalid(self):
# Missing "begin" line
self.assertRaises(ValueError, codecs.decode, b"", "uu-codec")
def test_invalid_error_input(self):
for encoding in bytes_transform_encodings:
with self.subTest(encoding=encoding):
encoder = codecs.getencoder(encoding)
decoder = codecs.getdecoder(encoding)
self.assertRaises(ValueError, encoder, 'in', errors='notstrict')
self.assertRaises(ValueError, decoder, 'in', errors='notstrict')
incdev = codecs.getincrementaldecoder(encoding)
if encoding not in ('base64_codec', 'uu_codec', 'quopri_codec', 'hex_codec'):
self.assertRaises(ValueError, incdev, errors='notstrict')
# The codec system tries to add notes to exceptions in order to ensure
# the error mentions the operation being performed and the codec involved.

View file

@ -0,0 +1,4 @@
The ``base64_codec``, ``uu_codec``, ``quopri_codec``, ``hex_codec``,
``zlib_codec`` and ``bz2_codec`` now raise a :exc:`ValueError` when their
decoder/encoder is provided an *errors* parameter that is not equal to
``'strict'``.