Kaydet (Commit) 14e10a19 authored tarafından Serhiy Storchaka's avatar Serhiy Storchaka

Issue #24102: Fixed exception type checking in standard error handlers.

üst 51dbc9a4
...@@ -836,6 +836,26 @@ class CodecCallbackTest(unittest.TestCase): ...@@ -836,6 +836,26 @@ class CodecCallbackTest(unittest.TestCase):
text = u'abc<def>ghi'*n text = u'abc<def>ghi'*n
text.translate(charmap) text.translate(charmap)
def test_fake_error_class(self):
handlers = [
codecs.strict_errors,
codecs.ignore_errors,
codecs.replace_errors,
codecs.backslashreplace_errors,
codecs.xmlcharrefreplace_errors,
]
for cls in UnicodeEncodeError, UnicodeDecodeError, UnicodeTranslateError:
class FakeUnicodeError(str):
__class__ = cls
for handler in handlers:
self.assertRaises(TypeError, handler, FakeUnicodeError())
class FakeUnicodeError(Exception):
__class__ = cls
for handler in handlers:
with self.assertRaises((TypeError, FakeUnicodeError)):
handler(FakeUnicodeError())
def test_main(): def test_main():
test.test_support.run_unittest(CodecCallbackTest) test.test_support.run_unittest(CodecCallbackTest)
......
...@@ -10,6 +10,8 @@ What's New in Python 2.7.11? ...@@ -10,6 +10,8 @@ What's New in Python 2.7.11?
Core and Builtins Core and Builtins
----------------- -----------------
- Issue #24102: Fixed exception type checking in standard error handlers.
Library Library
------- -------
......
...@@ -472,15 +472,16 @@ PyObject *PyCodec_StrictErrors(PyObject *exc) ...@@ -472,15 +472,16 @@ PyObject *PyCodec_StrictErrors(PyObject *exc)
PyObject *PyCodec_IgnoreErrors(PyObject *exc) PyObject *PyCodec_IgnoreErrors(PyObject *exc)
{ {
Py_ssize_t end; Py_ssize_t end;
if (PyObject_IsInstance(exc, PyExc_UnicodeEncodeError)) {
if (PyObject_TypeCheck(exc, (PyTypeObject *)PyExc_UnicodeEncodeError)) {
if (PyUnicodeEncodeError_GetEnd(exc, &end)) if (PyUnicodeEncodeError_GetEnd(exc, &end))
return NULL; return NULL;
} }
else if (PyObject_IsInstance(exc, PyExc_UnicodeDecodeError)) { else if (PyObject_TypeCheck(exc, (PyTypeObject *)PyExc_UnicodeDecodeError)) {
if (PyUnicodeDecodeError_GetEnd(exc, &end)) if (PyUnicodeDecodeError_GetEnd(exc, &end))
return NULL; return NULL;
} }
else if (PyObject_IsInstance(exc, PyExc_UnicodeTranslateError)) { else if (PyObject_TypeCheck(exc, (PyTypeObject *)PyExc_UnicodeTranslateError)) {
if (PyUnicodeTranslateError_GetEnd(exc, &end)) if (PyUnicodeTranslateError_GetEnd(exc, &end))
return NULL; return NULL;
} }
...@@ -500,7 +501,7 @@ PyObject *PyCodec_ReplaceErrors(PyObject *exc) ...@@ -500,7 +501,7 @@ PyObject *PyCodec_ReplaceErrors(PyObject *exc)
Py_ssize_t end; Py_ssize_t end;
Py_ssize_t i; Py_ssize_t i;
if (PyObject_IsInstance(exc, PyExc_UnicodeEncodeError)) { if (PyObject_TypeCheck(exc, (PyTypeObject *)PyExc_UnicodeEncodeError)) {
PyObject *res; PyObject *res;
Py_UNICODE *p; Py_UNICODE *p;
if (PyUnicodeEncodeError_GetStart(exc, &start)) if (PyUnicodeEncodeError_GetStart(exc, &start))
...@@ -517,13 +518,13 @@ PyObject *PyCodec_ReplaceErrors(PyObject *exc) ...@@ -517,13 +518,13 @@ PyObject *PyCodec_ReplaceErrors(PyObject *exc)
Py_DECREF(res); Py_DECREF(res);
return restuple; return restuple;
} }
else if (PyObject_IsInstance(exc, PyExc_UnicodeDecodeError)) { else if (PyObject_TypeCheck(exc, (PyTypeObject *)PyExc_UnicodeDecodeError)) {
Py_UNICODE res = Py_UNICODE_REPLACEMENT_CHARACTER; Py_UNICODE res = Py_UNICODE_REPLACEMENT_CHARACTER;
if (PyUnicodeDecodeError_GetEnd(exc, &end)) if (PyUnicodeDecodeError_GetEnd(exc, &end))
return NULL; return NULL;
return Py_BuildValue("(u#n)", &res, (Py_ssize_t)1, end); return Py_BuildValue("(u#n)", &res, (Py_ssize_t)1, end);
} }
else if (PyObject_IsInstance(exc, PyExc_UnicodeTranslateError)) { else if (PyObject_TypeCheck(exc, (PyTypeObject *)PyExc_UnicodeTranslateError)) {
PyObject *res; PyObject *res;
Py_UNICODE *p; Py_UNICODE *p;
if (PyUnicodeTranslateError_GetStart(exc, &start)) if (PyUnicodeTranslateError_GetStart(exc, &start))
...@@ -548,7 +549,7 @@ PyObject *PyCodec_ReplaceErrors(PyObject *exc) ...@@ -548,7 +549,7 @@ PyObject *PyCodec_ReplaceErrors(PyObject *exc)
PyObject *PyCodec_XMLCharRefReplaceErrors(PyObject *exc) PyObject *PyCodec_XMLCharRefReplaceErrors(PyObject *exc)
{ {
if (PyObject_IsInstance(exc, PyExc_UnicodeEncodeError)) { if (PyObject_TypeCheck(exc, (PyTypeObject *)PyExc_UnicodeEncodeError)) {
PyObject *restuple; PyObject *restuple;
PyObject *object; PyObject *object;
Py_ssize_t start; Py_ssize_t start;
...@@ -673,7 +674,7 @@ static Py_UNICODE hexdigits[] = { ...@@ -673,7 +674,7 @@ static Py_UNICODE hexdigits[] = {
PyObject *PyCodec_BackslashReplaceErrors(PyObject *exc) PyObject *PyCodec_BackslashReplaceErrors(PyObject *exc)
{ {
if (PyObject_IsInstance(exc, PyExc_UnicodeEncodeError)) { if (PyObject_TypeCheck(exc, (PyTypeObject *)PyExc_UnicodeEncodeError)) {
PyObject *restuple; PyObject *restuple;
PyObject *object; PyObject *object;
Py_ssize_t start; Py_ssize_t start;
......
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