From 673cd824ba009ecbb0723192867c27c8e27c0037 Mon Sep 17 00:00:00 2001 From: Neal Norwitz Date: Fri, 18 Oct 2002 16:33:13 +0000 Subject: [PATCH] Fix SF # 624982, Potential AV in slot_sq_item, by Greg Chapman Don't crash when getting value of a property raises an exception --- Lib/test/test_descr.py | 12 ++++++++++++ Objects/typeobject.c | 6 +++++- 2 files changed, 17 insertions(+), 1 deletion(-) diff --git a/Lib/test/test_descr.py b/Lib/test/test_descr.py index c9bd1cd7d43..7573bfd9a0f 100644 --- a/Lib/test/test_descr.py +++ b/Lib/test/test_descr.py @@ -1814,6 +1814,18 @@ def delx(self): raise TestFailed("expected TypeError from trying to set " "readonly %r attr on a property" % attr) + class D(object): + __getitem__ = property(lambda s: 1/0) + + d = D() + try: + for i in d: + str(i) + except ZeroDivisionError: + pass + else: + raise TestFailed, "expected ZeroDivisionError from bad property" + def supers(): if verbose: print "Testing super..." diff --git a/Objects/typeobject.c b/Objects/typeobject.c index ed5b82951ae..e624ec4da33 100644 --- a/Objects/typeobject.c +++ b/Objects/typeobject.c @@ -3145,8 +3145,12 @@ slot_sq_item(PyObject *self, int i) if (func != NULL) { if ((f = func->ob_type->tp_descr_get) == NULL) Py_INCREF(func); - else + else { func = f(func, self, (PyObject *)(self->ob_type)); + if (func == NULL) { + return NULL; + } + } ival = PyInt_FromLong(i); if (ival != NULL) { args = PyTuple_New(1);