From ea2b490f3dad99cfd677be9fd0c730622a723f38 Mon Sep 17 00:00:00 2001 From: Serhiy Storchaka Date: Tue, 28 May 2013 15:27:29 +0300 Subject: Issue #18011: base64.b32decode() now raises a binascii.Error if there are non-alphabet characters present in the input string to conform a docstring. Updated the module documentation. --- Doc/library/base64.rst | 2 +- Lib/base64.py | 2 +- Lib/test/test_base64.py | 8 +++++--- Misc/NEWS | 4 ++++ 4 files changed, 11 insertions(+), 5 deletions(-) diff --git a/Doc/library/base64.rst b/Doc/library/base64.rst index ade0f07..3b23e79 100644 --- a/Doc/library/base64.rst +++ b/Doc/library/base64.rst @@ -103,7 +103,7 @@ The modern interface provides: digit 0 is always mapped to the letter O). For security purposes the default is ``None``, so that 0 and 1 are not allowed in the input. - The decoded byte string is returned. A :exc:`TypeError` is raised if *s* were + The decoded byte string is returned. A :exc:`binascii.Error` is raised if *s* were incorrectly padded or if there are non-alphabet characters present in the string. diff --git a/Lib/base64.py b/Lib/base64.py index 6bcdff6..b6e82b6 100755 --- a/Lib/base64.py +++ b/Lib/base64.py @@ -245,7 +245,7 @@ def b32decode(s, casefold=False, map01=None): for c in s: val = _b32rev.get(c) if val is None: - raise TypeError('Non-base32 digit found') + raise binascii.Error('Non-base32 digit found') acc += _b32rev[c] << shift shift -= 5 if shift < 0: diff --git a/Lib/test/test_base64.py b/Lib/test/test_base64.py index abba5bf..13695de 100644 --- a/Lib/test/test_base64.py +++ b/Lib/test/test_base64.py @@ -244,8 +244,8 @@ class BaseXYTestCase(unittest.TestCase): eq(base64.b32decode(data, True), res) eq(base64.b32decode(data.decode('ascii'), True), res) - self.assertRaises(TypeError, base64.b32decode, b'me======') - self.assertRaises(TypeError, base64.b32decode, 'me======') + self.assertRaises(binascii.Error, base64.b32decode, b'me======') + self.assertRaises(binascii.Error, base64.b32decode, 'me======') # Mapping zero and one eq(base64.b32decode(b'MLO23456'), b'b\xdd\xad\xf3\xbe') @@ -262,9 +262,11 @@ class BaseXYTestCase(unittest.TestCase): eq(base64.b32decode(data_str, map01=map01), res) eq(base64.b32decode(data, map01=map01_str), res) eq(base64.b32decode(data_str, map01=map01_str), res) + self.assertRaises(binascii.Error, base64.b32decode, data) + self.assertRaises(binascii.Error, base64.b32decode, data_str) def test_b32decode_error(self): - for data in [b'abc', b'ABCDEF==']: + for data in [b'abc', b'ABCDEF==', b'==ABCDEF']: with self.assertRaises(binascii.Error): base64.b32decode(data) with self.assertRaises(binascii.Error): diff --git a/Misc/NEWS b/Misc/NEWS index 3090d9f..41bc856 100644 --- a/Misc/NEWS +++ b/Misc/NEWS @@ -24,6 +24,10 @@ Core and Builtins Library ------- +- Issue #18011: base64.b32decode() now raises a binascii.Error if there are + non-alphabet characters present in the input string to conform a docstring. + Updated the module documentation. + - Issue #13772: Restored directory detection of targets in ``os.symlink`` on Windows, which was temporarily removed in Python 3.2.3 due to an incomplete implementation. The implementation now works even if the symlink is created -- cgit v0.12