From 477ba919c158ba6e7a4b85d15f2e2b8820272cd9 Mon Sep 17 00:00:00 2001 From: Benjamin Peterson Date: Wed, 12 Jan 2011 15:34:01 +0000 Subject: [PATCH] don't segfault on deleting __abstractmethods__ #10892 --- Lib/test/test_descr.py | 6 +++++- Misc/NEWS | 3 +++ Objects/typeobject.c | 13 +++++++++++-- 3 files changed, 19 insertions(+), 3 deletions(-) diff --git a/Lib/test/test_descr.py b/Lib/test/test_descr.py index a81c2183761..914b249f7a4 100644 --- a/Lib/test/test_descr.py +++ b/Lib/test/test_descr.py @@ -4224,12 +4224,16 @@ order (MRO) for bases """ self.assertRaises(AttributeError, getattr, EvilGetattribute(), "attr") - def test_type_has_no_abstractmethods(self): + def test_abstractmethods(self): # type pretends not to have __abstractmethods__. self.assertRaises(AttributeError, getattr, type, "__abstractmethods__") class meta(type): pass self.assertRaises(AttributeError, getattr, meta, "__abstractmethods__") + class X(object): + pass + with self.assertRaises(AttributeError): + del X.__abstractmethods__ class DictProxyTests(unittest.TestCase): diff --git a/Misc/NEWS b/Misc/NEWS index 262f9828536..ee69f507db3 100644 --- a/Misc/NEWS +++ b/Misc/NEWS @@ -8,6 +8,9 @@ What's New in Python 3.2 Release Candidate 1 Core and Builtins ----------------- +- Issue #10892: Don't segfault when trying to delete __abstractmethods__ from a + class. + - Issue #8020: Avoid a crash where the small objects allocator would read non-Python managed memory while it is being modified by another thread. Patch by Matt Bandy. diff --git a/Objects/typeobject.c b/Objects/typeobject.c index 1fefe8414fe..930297a7bb6 100644 --- a/Objects/typeobject.c +++ b/Objects/typeobject.c @@ -340,8 +340,17 @@ type_set_abstractmethods(PyTypeObject *type, PyObject *value, void *context) abc.ABCMeta.__new__, so this function doesn't do anything special to update subclasses. */ - int res = PyDict_SetItemString(type->tp_dict, - "__abstractmethods__", value); + int res; + if (value != NULL) { + res = PyDict_SetItemString(type->tp_dict, "__abstractmethods__", value); + } + else { + res = PyDict_DelItemString(type->tp_dict, "__abstractmethods__"); + if (res && PyErr_ExceptionMatches(PyExc_KeyError)) { + PyErr_Format(PyExc_AttributeError, "__abstractmethods__", value); + return -1; + } + } if (res == 0) { PyType_Modified(type); if (value && PyObject_IsTrue(value)) {