Fix test67.py from issue #1303614.

This commit is contained in:
Guido van Rossum 2008-01-24 17:58:05 +00:00
parent 4e3f12486f
commit 37edeab778
3 changed files with 29 additions and 21 deletions

View File

@ -1,21 +0,0 @@
# http://python.org/sf/1303614
class Strange(object):
def __hash__(self):
return hash('hello')
def __eq__(self, other):
x.__dict__ = {} # the old x.__dict__ is deallocated
return False
class X(object):
pass
if __name__ == '__main__':
v = 123
x = X()
x.__dict__ = {Strange(): 42,
'hello': v+456}
x.hello # segfault: the above dict is accessed after it's deallocated

View File

@ -4504,6 +4504,29 @@ def test_borrowed_ref_4_segfault():
finally:
__builtin__.__import__ = orig_import
def test_losing_dict_ref_segfault():
# This used to segfault;
# derived from issue #1303614, test67.py
if verbose:
print "Testing losing dict ref segfault..."
class Strange(object):
def __hash__(self):
return hash('hello')
def __eq__(self, other):
x.__dict__ = {} # the old x.__dict__ is deallocated
return False
class X(object):
pass
v = 123
x = X()
x.__dict__ = {Strange(): 42, 'hello': v+456}
x.hello
def test_main():
weakref_segfault() # Must be first, somehow
wrapper_segfault()
@ -4606,6 +4629,7 @@ def test_main():
test_weakref_in_del_segfault()
test_borrowed_ref_3_segfault()
test_borrowed_ref_4_segfault()
test_losing_dict_ref_segfault()
if verbose: print "All OK"

View File

@ -1349,12 +1349,15 @@ PyObject_GenericGetAttr(PyObject *obj, PyObject *name)
dictptr = (PyObject **) ((char *)obj + dictoffset);
dict = *dictptr;
if (dict != NULL) {
Py_INCREF(dict);
res = PyDict_GetItem(dict, name);
if (res != NULL) {
Py_INCREF(res);
Py_XDECREF(descr);
Py_DECREF(dict);
goto done;
}
Py_DECREF(dict);
}
}
@ -1435,12 +1438,14 @@ PyObject_GenericSetAttr(PyObject *obj, PyObject *name, PyObject *value)
*dictptr = dict;
}
if (dict != NULL) {
Py_INCREF(dict);
if (value == NULL)
res = PyDict_DelItem(dict, name);
else
res = PyDict_SetItem(dict, name, value);
if (res < 0 && PyErr_ExceptionMatches(PyExc_KeyError))
PyErr_SetObject(PyExc_AttributeError, name);
Py_DECREF(dict);
goto done;
}
}