def test_base64_excess_data(self):
# Test excess data exceptions
- def assertExcessData(data, non_strict_expected,
- ignore_padchar_expected=None):
+ def assertExcessData(data, expected):
assert_regex = r'(?i)Excess data'
data = self.type2test(data)
with self.assertRaisesRegex(binascii.Error, assert_regex):
binascii.a2b_base64(data, strict_mode=True)
self.assertEqual(binascii.a2b_base64(data, strict_mode=False),
- non_strict_expected)
- if ignore_padchar_expected is not None:
- self.assertEqual(binascii.a2b_base64(data, strict_mode=True,
- ignorechars=b'='),
- ignore_padchar_expected)
- self.assertEqual(binascii.a2b_base64(data), non_strict_expected)
-
- assertExcessData(b'ab==c', b'i')
- assertExcessData(b'ab==cd', b'i', b'i\xb7\x1d')
- assertExcessData(b'abc=d', b'i\xb7', b'i\xb7\x1d')
+ expected)
+ self.assertEqual(binascii.a2b_base64(data, strict_mode=True,
+ ignorechars=b'='),
+ expected)
+ self.assertEqual(binascii.a2b_base64(data), expected)
+
+ assertExcessData(b'ab==c=', b'i\xb7')
+ assertExcessData(b'ab==cd', b'i\xb7\x1d')
+ assertExcessData(b'abc=d', b'i\xb7\x1d')
def test_base64errors(self):
# Test base64 with invalid padding
*/
if (this_ch == BASE64_PAD) {
pads++;
-
- if (strict_mode) {
- if (quad_pos >= 2 && quad_pos + pads <= 4) {
- continue;
- }
- if (ignorechar(BASE64_PAD, ignorechars, ignorecache)) {
- continue;
- }
- if (quad_pos == 1) {
- /* Set an error below. */
- break;
- }
- state = get_binascii_state(module);
- if (state) {
- PyErr_SetString(state->Error,
- (quad_pos == 0 && ascii_data == data->buf)
- ? "Leading padding not allowed"
- : "Excess padding not allowed");
- }
- goto error_end;
+ if (quad_pos >= 2 && quad_pos + pads <= 4) {
+ continue;
}
- else {
- if (quad_pos >= 2 && quad_pos + pads >= 4) {
- /* A pad sequence means we should not parse more input.
- ** We've already interpreted the data from the quad at this point.
- */
- goto done;
- }
+ // See RFC 4648, section-3.3: "specifications MAY ignore the
+ // pad character, "=", treating it as non-alphabet data, if
+ // it is present before the end of the encoded data" and
+ // "the excess pad characters MAY also be ignored."
+ if (!strict_mode || ignorechar(BASE64_PAD, ignorechars, ignorecache)) {
continue;
}
+ if (quad_pos == 1) {
+ /* Set an error below. */
+ break;
+ }
+ state = get_binascii_state(module);
+ if (state) {
+ PyErr_SetString(state->Error,
+ (quad_pos == 0 && ascii_data == data->buf)
+ ? "Leading padding not allowed"
+ : "Excess padding not allowed");
+ }
+ goto error_end;
}
unsigned char v = table_a2b[this_ch];
if (v >= 64) {
+ // See RFC 4648, section-3.3.
if (strict_mode && !ignorechar(this_ch, ignorechars, ignorecache)) {
state = get_binascii_state(module);
if (state) {
continue;
}
- // Characters that are not '=', in the middle of the padding, are not allowed
+ // Characters that are not '=', in the middle of the padding, are
+ // not allowed (except when they are). See RFC 4648, section-3.3.
if (pads && strict_mode &&
!ignorechar(BASE64_PAD, ignorechars, ignorecache))
{
goto error_end;
}
-done:
Py_XDECREF(table_obj);
return PyBytesWriter_FinishWithPointer(writer, bin_data);