when an argument is a cell, set the local copy to NULL (see #17927)

This commit is contained in:
Benjamin Peterson 2013-05-12 18:16:06 -05:00
parent 3bfc5f5d83
commit 159ae41da6
3 changed files with 28 additions and 11 deletions

View File

@ -130,6 +130,19 @@ class TestSuper(unittest.TestCase):
super()
self.assertRaises(RuntimeError, X().f)
def test_cell_as_self(self):
class X:
def meth(self):
super()
def f():
k = X()
def g():
return k
return g
c = f().__closure__[0]
self.assertRaises(TypeError, X.meth, c)
def test_main():
support.run_unittest(TestSuper)

View File

@ -6510,9 +6510,17 @@ super_init(PyObject *self, PyObject *args, PyObject *kwds)
return -1;
}
obj = f->f_localsplus[0];
if (obj != NULL && PyCell_Check(obj)) {
/* It might be a cell. See cell var initialization in ceval.c. */
obj = PyCell_GET(obj);
if (obj == NULL && co->co_cell2arg) {
/* The first argument might be a cell. */
n = PyTuple_GET_SIZE(co->co_cellvars);
for (i = 0; i < n; i++) {
if (co->co_cell2arg[i] == 0) {
PyObject *cell = f->f_localsplus[co->co_nlocals + i];
assert(PyCell_Check(cell));
obj = PyCell_GET(cell);
break;
}
}
}
if (obj == NULL) {
PyErr_SetString(PyExc_RuntimeError,

View File

@ -3521,18 +3521,14 @@ PyEval_EvalCodeEx(PyObject *_co, PyObject *globals, PyObject *locals,
if (co->co_cell2arg != NULL &&
(arg = co->co_cell2arg[i]) != CO_CELL_NOT_AN_ARG) {
c = PyCell_New(GETLOCAL(arg));
if (c == NULL)
goto fail;
/* Reference the cell from the argument slot, for super().
See typeobject.c. */
Py_INCREF(c);
SETLOCAL(arg, c);
/* Clear the local copy. */
SETLOCAL(arg, NULL);
}
else {
c = PyCell_New(NULL);
}
if (c == NULL)
goto fail;
}
SETLOCAL(co->co_nlocals + i, c);
}
for (i = 0; i < PyTuple_GET_SIZE(co->co_freevars); ++i) {