Skip to content

Z
ZODB
Commit de5d6448 authored by Jeremy Hylton's avatar Jeremy Hylton
Browse files
Options

Fix several ref count bugs. 

Remove object_from_oid() which appeared to be mis-used about half the
time.  Instead use std PyDict_GetItem() and incref only when
necessary.

Make sure an object's reference to its cache is decref'd when the
object is deallocated.  This change seems to eliminate leaking pickle
caches.

Remaining mystery: Why do objects stay in the cache even when there
are no references to them?
parent b701349c
Show whitespace changes
Inline Side-by-side
Showing with 72 additions and 96 deletions
src/Persistence/cPickleCache.c
View file @ de5d6448
...@@ -88,7 +88,7 @@ process must skip such objects, rather than deactivating them. ...@@ -88,7 +88,7 @@ process must skip such objects, rather than deactivating them.
static char cPickleCache_doc_string[] = static char cPickleCache_doc_string[] =
"Defines the PickleCache used by ZODB Connection objects.\n" "Defines the PickleCache used by ZODB Connection objects.\n"
"\n" "\n"
"$Id: cPickleCache.c,v 1.71 2003/02/11 19:16:27 bwarsaw Exp $\n"; "$Id: cPickleCache.c,v 1.72 2003/03/31 22:39:33 jeremy Exp $\n";
#define ASSIGN(V,E) {PyObject *__e; __e=(E); Py_XDECREF(V); (V)=__e;} #define ASSIGN(V,E) {PyObject *__e; __e=(E); Py_XDECREF(V); (V)=__e;}
#define UNLESS(E) if(!(E)) #define UNLESS(E) if(!(E))
...@@ -147,18 +147,6 @@ static int cc_ass_sub(ccobject *self, PyObject *key, PyObject *v); ...@@ -147,18 +147,6 @@ static int cc_ass_sub(ccobject *self, PyObject *key, PyObject *v);
/* ---------------------------------------------------------------- */ /* ---------------------------------------------------------------- */
/* somewhat of a replacement for PyDict_GetItem(self->data....
however this returns a *new* reference */
static PyObject *
object_from_oid(ccobject *self, PyObject *key)
{
PyObject *v = PyDict_GetItem(self->data, key);
if (!v)
return NULL;
Py_INCREF(v);
return v;
}
#define OBJECT_FROM_RING(SELF, HERE, CTX) \ #define OBJECT_FROM_RING(SELF, HERE, CTX) \
((cPersistentObject *)(((char *)here) - offsetof(cPersistentObject, ring))) ((cPersistentObject *)(((char *)here) - offsetof(cPersistentObject, ring)))
...@@ -308,7 +296,7 @@ cc_minimize(ccobject *self, PyObject *args) ...@@ -308,7 +296,7 @@ cc_minimize(ccobject *self, PyObject *args)
static void static void
_invalidate(ccobject *self, PyObject *key) _invalidate(ccobject *self, PyObject *key)
{ {
PyObject *v = object_from_oid(self, key); PyObject *v = PyDict_GetItem(self->data, key);
if (!v) if (!v)
return; return;
...@@ -329,7 +317,6 @@ _invalidate(ccobject *self, PyObject *key) ...@@ -329,7 +317,6 @@ _invalidate(ccobject *self, PyObject *key)
if (PyObject_DelAttr(v, py__p_changed) < 0) if (PyObject_DelAttr(v, py__p_changed) < 0)
PyErr_Clear(); PyErr_Clear();
} }
Py_XDECREF(v);
} }
static PyObject * static PyObject *
...@@ -374,21 +361,21 @@ cc_invalidate(ccobject *self, PyObject *args) ...@@ -374,21 +361,21 @@ cc_invalidate(ccobject *self, PyObject *args)
static PyObject * static PyObject *
cc_get(ccobject *self, PyObject *args) cc_get(ccobject *self, PyObject *args)
{ {
PyObject *r, *key, *d = NULL; PyObject *r, *key, *def = NULL;
if (!PyArg_ParseTuple(args, "O|O:get", &key, &d)) if (!PyArg_ParseTuple(args, "O|O:get", &key, &def))
return NULL; return NULL;
r = (PyObject *)object_from_oid(self, key); r = PyDict_GetItem(self->data, key);
if (!r) { if (!r) {
if (d) { if (def)
r = d; r = def;
Py_INCREF(r); else {
} else {
PyErr_SetObject(PyExc_KeyError, key); PyErr_SetObject(PyExc_KeyError, key);
return NULL; return NULL;
} }
} }
Py_INCREF(r);
return r; return r;
} }
...@@ -501,6 +488,13 @@ cc_oid_unreferenced(ccobject *self, PyObject *oid) ...@@ -501,6 +488,13 @@ cc_oid_unreferenced(ccobject *self, PyObject *oid)
interpreter has untracked the reference. Track it again. interpreter has untracked the reference. Track it again.
*/ */
_Py_NewReference(v); _Py_NewReference(v);
/* Don't increment total refcount as a result of the
shenanigans played in this function. The _Py_NewReference()
call above and the Py_INCREF() below both create artificial
references to v.
*/
_Py_RefTotal -= 2;
/* XXX it may be a problem that v->ob_type is still NULL? /* XXX it may be a problem that v->ob_type is still NULL?
I don't understand what this comment means. --jeremy */ I don't understand what this comment means. --jeremy */
assert(v->ob_type); assert(v->ob_type);
...@@ -517,6 +511,7 @@ cc_oid_unreferenced(ccobject *self, PyObject *oid) ...@@ -517,6 +511,7 @@ cc_oid_unreferenced(ccobject *self, PyObject *oid)
/* XXX Should we call _Py_ForgetReference() on error exit? */ /* XXX Should we call _Py_ForgetReference() on error exit? */
if (PyDict_DelItem(self->data, oid) < 0) if (PyDict_DelItem(self->data, oid) < 0)
return -1; return -1;
Py_DECREF((ccobject *)((cPersistentObject *)v)->cache);
if (v->ob_refcnt != 1) { if (v->ob_refcnt != 1) {
PyErr_SetString(PyExc_ValueError, PyErr_SetString(PyExc_ValueError,
...@@ -664,11 +659,12 @@ cc_subscript(ccobject *self, PyObject *key) ...@@ -664,11 +659,12 @@ cc_subscript(ccobject *self, PyObject *key)
{ {
PyObject *r; PyObject *r;
r = (PyObject *)object_from_oid(self, key); r = PyDict_GetItem(self->data, key);
if (r == NULL) { if (r == NULL) {
PyErr_SetObject(PyExc_KeyError, key); PyErr_SetObject(PyExc_KeyError, key);
return NULL; return NULL;
} }
Py_INCREF(r);
return r; return r;
} }
...@@ -686,7 +682,7 @@ cc_add_item(ccobject *self, PyObject *key, PyObject *v) ...@@ -686,7 +682,7 @@ cc_add_item(ccobject *self, PyObject *key, PyObject *v)
else if( PyExtensionInstance_Check(v) && else if( PyExtensionInstance_Check(v) &&
(((PyExtensionClass*)(v->ob_type))->class_flags & PERSISTENT_TYPE_FLAG) && (((PyExtensionClass*)(v->ob_type))->class_flags & PERSISTENT_TYPE_FLAG) &&
(v->ob_type->tp_basicsize >= sizeof(cPersistentObject)) ) { (v->ob_type->tp_basicsize >= sizeof(cPersistentObject)) ) {
/* Its and instance of a persistent class, (ie Python classeses that /* Its an instance of a persistent class, (ie Python classes that
derive from Persistence.Persistent, BTrees, etc). Thats ok. */ derive from Persistence.Persistent, BTrees, etc). Thats ok. */
} }
else { else {
...@@ -725,7 +721,7 @@ cc_add_item(ccobject *self, PyObject *key, PyObject *v) ...@@ -725,7 +721,7 @@ cc_add_item(ccobject *self, PyObject *key, PyObject *v)
jar = PyObject_GetAttr(v, py__p_jar); jar = PyObject_GetAttr(v, py__p_jar);
if (jar == NULL) if (jar == NULL)
return -1; return -1;
if (jar==Py_None) { if (jar == Py_None) {
Py_DECREF(jar); Py_DECREF(jar);
PyErr_SetString(PyExc_ValueError, PyErr_SetString(PyExc_ValueError,
"Cached object jar missing"); "Cached object jar missing");
...@@ -733,16 +729,14 @@ cc_add_item(ccobject *self, PyObject *key, PyObject *v) ...@@ -733,16 +729,14 @@ cc_add_item(ccobject *self, PyObject *key, PyObject *v)
} }
Py_DECREF(jar); Py_DECREF(jar);
object_again = object_from_oid(self, key); object_again = PyDict_GetItem(self->data, key);
if (object_again) { if (object_again) {
if (object_again != v) { if (object_again != v) {
Py_DECREF(object_again);
PyErr_SetString(PyExc_ValueError, PyErr_SetString(PyExc_ValueError,
"Can not re-register object under a different oid"); "Can not re-register object under a different oid");
return -1; return -1;
} else { } else {
/* re-register under the same oid - no work needed */ /* re-register under the same oid - no work needed */
Py_DECREF(object_again);
return 0; return 0;
} }
} }
...@@ -799,7 +793,7 @@ cc_del_item(ccobject *self, PyObject *key) ...@@ -799,7 +793,7 @@ cc_del_item(ccobject *self, PyObject *key)
cPersistentObject *p; cPersistentObject *p;
/* unlink this item from the ring */ /* unlink this item from the ring */
v = (PyObject *)object_from_oid(self, key); v = PyDict_GetItem(self->data, key);
if (v == NULL) if (v == NULL)
return -1; return -1;
...@@ -826,8 +820,6 @@ cc_del_item(ccobject *self, PyObject *key) ...@@ -826,8 +820,6 @@ cc_del_item(ccobject *self, PyObject *key)
p->cache = NULL; p->cache = NULL;
} }
Py_DECREF(v);
if (PyDict_DelItem(self->data, key) < 0) { if (PyDict_DelItem(self->data, key) < 0) {
PyErr_SetString(PyExc_RuntimeError, PyErr_SetString(PyExc_RuntimeError,
"unexpectedly couldn't remove key in cc_ass_sub"); "unexpectedly couldn't remove key in cc_ass_sub");
......
src/ZODB/cPickleCache.c
View file @ de5d6448
...@@ -88,7 +88,7 @@ process must skip such objects, rather than deactivating them. ...@@ -88,7 +88,7 @@ process must skip such objects, rather than deactivating them.
static char cPickleCache_doc_string[] = static char cPickleCache_doc_string[] =
"Defines the PickleCache used by ZODB Connection objects.\n" "Defines the PickleCache used by ZODB Connection objects.\n"
"\n" "\n"
"$Id: cPickleCache.c,v 1.71 2003/02/11 19:16:27 bwarsaw Exp $\n"; "$Id: cPickleCache.c,v 1.72 2003/03/31 22:39:33 jeremy Exp $\n";
#define ASSIGN(V,E) {PyObject *__e; __e=(E); Py_XDECREF(V); (V)=__e;} #define ASSIGN(V,E) {PyObject *__e; __e=(E); Py_XDECREF(V); (V)=__e;}
#define UNLESS(E) if(!(E)) #define UNLESS(E) if(!(E))
...@@ -147,18 +147,6 @@ static int cc_ass_sub(ccobject *self, PyObject *key, PyObject *v); ...@@ -147,18 +147,6 @@ static int cc_ass_sub(ccobject *self, PyObject *key, PyObject *v);
/* ---------------------------------------------------------------- */ /* ---------------------------------------------------------------- */
/* somewhat of a replacement for PyDict_GetItem(self->data....
however this returns a *new* reference */
static PyObject *
object_from_oid(ccobject *self, PyObject *key)
{
PyObject *v = PyDict_GetItem(self->data, key);
if (!v)
return NULL;
Py_INCREF(v);
return v;
}
#define OBJECT_FROM_RING(SELF, HERE, CTX) \ #define OBJECT_FROM_RING(SELF, HERE, CTX) \
((cPersistentObject *)(((char *)here) - offsetof(cPersistentObject, ring))) ((cPersistentObject *)(((char *)here) - offsetof(cPersistentObject, ring)))
...@@ -308,7 +296,7 @@ cc_minimize(ccobject *self, PyObject *args) ...@@ -308,7 +296,7 @@ cc_minimize(ccobject *self, PyObject *args)
static void static void
_invalidate(ccobject *self, PyObject *key) _invalidate(ccobject *self, PyObject *key)
{ {
PyObject *v = object_from_oid(self, key); PyObject *v = PyDict_GetItem(self->data, key);
if (!v) if (!v)
return; return;
...@@ -329,7 +317,6 @@ _invalidate(ccobject *self, PyObject *key) ...@@ -329,7 +317,6 @@ _invalidate(ccobject *self, PyObject *key)
if (PyObject_DelAttr(v, py__p_changed) < 0) if (PyObject_DelAttr(v, py__p_changed) < 0)
PyErr_Clear(); PyErr_Clear();
} }
Py_XDECREF(v);
} }
static PyObject * static PyObject *
...@@ -374,21 +361,21 @@ cc_invalidate(ccobject *self, PyObject *args) ...@@ -374,21 +361,21 @@ cc_invalidate(ccobject *self, PyObject *args)
static PyObject * static PyObject *
cc_get(ccobject *self, PyObject *args) cc_get(ccobject *self, PyObject *args)
{ {
PyObject *r, *key, *d = NULL; PyObject *r, *key, *def = NULL;
if (!PyArg_ParseTuple(args, "O|O:get", &key, &d)) if (!PyArg_ParseTuple(args, "O|O:get", &key, &def))
return NULL; return NULL;
r = (PyObject *)object_from_oid(self, key); r = PyDict_GetItem(self->data, key);
if (!r) { if (!r) {
if (d) { if (def)
r = d; r = def;
Py_INCREF(r); else {
} else {
PyErr_SetObject(PyExc_KeyError, key); PyErr_SetObject(PyExc_KeyError, key);
return NULL; return NULL;
} }
} }
Py_INCREF(r);
return r; return r;
} }
...@@ -501,6 +488,13 @@ cc_oid_unreferenced(ccobject *self, PyObject *oid) ...@@ -501,6 +488,13 @@ cc_oid_unreferenced(ccobject *self, PyObject *oid)
interpreter has untracked the reference. Track it again. interpreter has untracked the reference. Track it again.
*/ */
_Py_NewReference(v); _Py_NewReference(v);
/* Don't increment total refcount as a result of the
shenanigans played in this function. The _Py_NewReference()
call above and the Py_INCREF() below both create artificial
references to v.
*/
_Py_RefTotal -= 2;
/* XXX it may be a problem that v->ob_type is still NULL? /* XXX it may be a problem that v->ob_type is still NULL?
I don't understand what this comment means. --jeremy */ I don't understand what this comment means. --jeremy */
assert(v->ob_type); assert(v->ob_type);
...@@ -517,6 +511,7 @@ cc_oid_unreferenced(ccobject *self, PyObject *oid) ...@@ -517,6 +511,7 @@ cc_oid_unreferenced(ccobject *self, PyObject *oid)
/* XXX Should we call _Py_ForgetReference() on error exit? */ /* XXX Should we call _Py_ForgetReference() on error exit? */
if (PyDict_DelItem(self->data, oid) < 0) if (PyDict_DelItem(self->data, oid) < 0)
return -1; return -1;
Py_DECREF((ccobject *)((cPersistentObject *)v)->cache);
if (v->ob_refcnt != 1) { if (v->ob_refcnt != 1) {
PyErr_SetString(PyExc_ValueError, PyErr_SetString(PyExc_ValueError,
...@@ -664,11 +659,12 @@ cc_subscript(ccobject *self, PyObject *key) ...@@ -664,11 +659,12 @@ cc_subscript(ccobject *self, PyObject *key)
{ {
PyObject *r; PyObject *r;
r = (PyObject *)object_from_oid(self, key); r = PyDict_GetItem(self->data, key);
if (r == NULL) { if (r == NULL) {
PyErr_SetObject(PyExc_KeyError, key); PyErr_SetObject(PyExc_KeyError, key);
return NULL; return NULL;
} }
Py_INCREF(r);
return r; return r;
} }
...@@ -686,7 +682,7 @@ cc_add_item(ccobject *self, PyObject *key, PyObject *v) ...@@ -686,7 +682,7 @@ cc_add_item(ccobject *self, PyObject *key, PyObject *v)
else if( PyExtensionInstance_Check(v) && else if( PyExtensionInstance_Check(v) &&
(((PyExtensionClass*)(v->ob_type))->class_flags & PERSISTENT_TYPE_FLAG) && (((PyExtensionClass*)(v->ob_type))->class_flags & PERSISTENT_TYPE_FLAG) &&
(v->ob_type->tp_basicsize >= sizeof(cPersistentObject)) ) { (v->ob_type->tp_basicsize >= sizeof(cPersistentObject)) ) {
/* Its and instance of a persistent class, (ie Python classeses that /* Its an instance of a persistent class, (ie Python classes that
derive from Persistence.Persistent, BTrees, etc). Thats ok. */ derive from Persistence.Persistent, BTrees, etc). Thats ok. */
} }
else { else {
...@@ -725,7 +721,7 @@ cc_add_item(ccobject *self, PyObject *key, PyObject *v) ...@@ -725,7 +721,7 @@ cc_add_item(ccobject *self, PyObject *key, PyObject *v)
jar = PyObject_GetAttr(v, py__p_jar); jar = PyObject_GetAttr(v, py__p_jar);
if (jar == NULL) if (jar == NULL)
return -1; return -1;
if (jar==Py_None) { if (jar == Py_None) {
Py_DECREF(jar); Py_DECREF(jar);
PyErr_SetString(PyExc_ValueError, PyErr_SetString(PyExc_ValueError,
"Cached object jar missing"); "Cached object jar missing");
...@@ -733,16 +729,14 @@ cc_add_item(ccobject *self, PyObject *key, PyObject *v) ...@@ -733,16 +729,14 @@ cc_add_item(ccobject *self, PyObject *key, PyObject *v)
} }
Py_DECREF(jar); Py_DECREF(jar);
object_again = object_from_oid(self, key); object_again = PyDict_GetItem(self->data, key);
if (object_again) { if (object_again) {
if (object_again != v) { if (object_again != v) {
Py_DECREF(object_again);
PyErr_SetString(PyExc_ValueError, PyErr_SetString(PyExc_ValueError,
"Can not re-register object under a different oid"); "Can not re-register object under a different oid");
return -1; return -1;
} else { } else {
/* re-register under the same oid - no work needed */ /* re-register under the same oid - no work needed */
Py_DECREF(object_again);
return 0; return 0;
} }
} }
...@@ -799,7 +793,7 @@ cc_del_item(ccobject *self, PyObject *key) ...@@ -799,7 +793,7 @@ cc_del_item(ccobject *self, PyObject *key)
cPersistentObject *p; cPersistentObject *p;
/* unlink this item from the ring */ /* unlink this item from the ring */
v = (PyObject *)object_from_oid(self, key); v = PyDict_GetItem(self->data, key);
if (v == NULL) if (v == NULL)
return -1; return -1;
...@@ -826,8 +820,6 @@ cc_del_item(ccobject *self, PyObject *key) ...@@ -826,8 +820,6 @@ cc_del_item(ccobject *self, PyObject *key)
p->cache = NULL; p->cache = NULL;
} }
Py_DECREF(v);
if (PyDict_DelItem(self->data, key) < 0) { if (PyDict_DelItem(self->data, key) < 0) {
PyErr_SetString(PyExc_RuntimeError, PyErr_SetString(PyExc_RuntimeError,
"unexpectedly couldn't remove key in cc_ass_sub"); "unexpectedly couldn't remove key in cc_ass_sub");
......
src/persistent/cPickleCache.c
View file @ de5d6448
...@@ -88,7 +88,7 @@ process must skip such objects, rather than deactivating them. ...@@ -88,7 +88,7 @@ process must skip such objects, rather than deactivating them.
static char cPickleCache_doc_string[] = static char cPickleCache_doc_string[] =
"Defines the PickleCache used by ZODB Connection objects.\n" "Defines the PickleCache used by ZODB Connection objects.\n"
"\n" "\n"
"$Id: cPickleCache.c,v 1.71 2003/02/11 19:16:27 bwarsaw Exp $\n"; "$Id: cPickleCache.c,v 1.72 2003/03/31 22:39:33 jeremy Exp $\n";
#define ASSIGN(V,E) {PyObject *__e; __e=(E); Py_XDECREF(V); (V)=__e;} #define ASSIGN(V,E) {PyObject *__e; __e=(E); Py_XDECREF(V); (V)=__e;}
#define UNLESS(E) if(!(E)) #define UNLESS(E) if(!(E))
...@@ -147,18 +147,6 @@ static int cc_ass_sub(ccobject *self, PyObject *key, PyObject *v); ...@@ -147,18 +147,6 @@ static int cc_ass_sub(ccobject *self, PyObject *key, PyObject *v);
/* ---------------------------------------------------------------- */ /* ---------------------------------------------------------------- */
/* somewhat of a replacement for PyDict_GetItem(self->data....
however this returns a *new* reference */
static PyObject *
object_from_oid(ccobject *self, PyObject *key)
{
PyObject *v = PyDict_GetItem(self->data, key);
if (!v)
return NULL;
Py_INCREF(v);
return v;
}
#define OBJECT_FROM_RING(SELF, HERE, CTX) \ #define OBJECT_FROM_RING(SELF, HERE, CTX) \
((cPersistentObject *)(((char *)here) - offsetof(cPersistentObject, ring))) ((cPersistentObject *)(((char *)here) - offsetof(cPersistentObject, ring)))
...@@ -308,7 +296,7 @@ cc_minimize(ccobject *self, PyObject *args) ...@@ -308,7 +296,7 @@ cc_minimize(ccobject *self, PyObject *args)
static void static void
_invalidate(ccobject *self, PyObject *key) _invalidate(ccobject *self, PyObject *key)
{ {
PyObject *v = object_from_oid(self, key); PyObject *v = PyDict_GetItem(self->data, key);
if (!v) if (!v)
return; return;
...@@ -329,7 +317,6 @@ _invalidate(ccobject *self, PyObject *key) ...@@ -329,7 +317,6 @@ _invalidate(ccobject *self, PyObject *key)
if (PyObject_DelAttr(v, py__p_changed) < 0) if (PyObject_DelAttr(v, py__p_changed) < 0)
PyErr_Clear(); PyErr_Clear();
} }
Py_XDECREF(v);
} }
static PyObject * static PyObject *
...@@ -374,21 +361,21 @@ cc_invalidate(ccobject *self, PyObject *args) ...@@ -374,21 +361,21 @@ cc_invalidate(ccobject *self, PyObject *args)
static PyObject * static PyObject *
cc_get(ccobject *self, PyObject *args) cc_get(ccobject *self, PyObject *args)
{ {
PyObject *r, *key, *d = NULL; PyObject *r, *key, *def = NULL;
if (!PyArg_ParseTuple(args, "O|O:get", &key, &d)) if (!PyArg_ParseTuple(args, "O|O:get", &key, &def))
return NULL; return NULL;
r = (PyObject *)object_from_oid(self, key); r = PyDict_GetItem(self->data, key);
if (!r) { if (!r) {
if (d) { if (def)
r = d; r = def;
Py_INCREF(r); else {
} else {
PyErr_SetObject(PyExc_KeyError, key); PyErr_SetObject(PyExc_KeyError, key);
return NULL; return NULL;
} }
} }
Py_INCREF(r);
return r; return r;
} }
...@@ -501,6 +488,13 @@ cc_oid_unreferenced(ccobject *self, PyObject *oid) ...@@ -501,6 +488,13 @@ cc_oid_unreferenced(ccobject *self, PyObject *oid)
interpreter has untracked the reference. Track it again. interpreter has untracked the reference. Track it again.
*/ */
_Py_NewReference(v); _Py_NewReference(v);
/* Don't increment total refcount as a result of the
shenanigans played in this function. The _Py_NewReference()
call above and the Py_INCREF() below both create artificial
references to v.
*/
_Py_RefTotal -= 2;
/* XXX it may be a problem that v->ob_type is still NULL? /* XXX it may be a problem that v->ob_type is still NULL?
I don't understand what this comment means. --jeremy */ I don't understand what this comment means. --jeremy */
assert(v->ob_type); assert(v->ob_type);
...@@ -517,6 +511,7 @@ cc_oid_unreferenced(ccobject *self, PyObject *oid) ...@@ -517,6 +511,7 @@ cc_oid_unreferenced(ccobject *self, PyObject *oid)
/* XXX Should we call _Py_ForgetReference() on error exit? */ /* XXX Should we call _Py_ForgetReference() on error exit? */
if (PyDict_DelItem(self->data, oid) < 0) if (PyDict_DelItem(self->data, oid) < 0)
return -1; return -1;
Py_DECREF((ccobject *)((cPersistentObject *)v)->cache);
if (v->ob_refcnt != 1) { if (v->ob_refcnt != 1) {
PyErr_SetString(PyExc_ValueError, PyErr_SetString(PyExc_ValueError,
...@@ -664,11 +659,12 @@ cc_subscript(ccobject *self, PyObject *key) ...@@ -664,11 +659,12 @@ cc_subscript(ccobject *self, PyObject *key)
{ {
PyObject *r; PyObject *r;
r = (PyObject *)object_from_oid(self, key); r = PyDict_GetItem(self->data, key);
if (r == NULL) { if (r == NULL) {
PyErr_SetObject(PyExc_KeyError, key); PyErr_SetObject(PyExc_KeyError, key);
return NULL; return NULL;
} }
Py_INCREF(r);
return r; return r;
} }
...@@ -686,7 +682,7 @@ cc_add_item(ccobject *self, PyObject *key, PyObject *v) ...@@ -686,7 +682,7 @@ cc_add_item(ccobject *self, PyObject *key, PyObject *v)
else if( PyExtensionInstance_Check(v) && else if( PyExtensionInstance_Check(v) &&
(((PyExtensionClass*)(v->ob_type))->class_flags & PERSISTENT_TYPE_FLAG) && (((PyExtensionClass*)(v->ob_type))->class_flags & PERSISTENT_TYPE_FLAG) &&
(v->ob_type->tp_basicsize >= sizeof(cPersistentObject)) ) { (v->ob_type->tp_basicsize >= sizeof(cPersistentObject)) ) {
/* Its and instance of a persistent class, (ie Python classeses that /* Its an instance of a persistent class, (ie Python classes that
derive from Persistence.Persistent, BTrees, etc). Thats ok. */ derive from Persistence.Persistent, BTrees, etc). Thats ok. */
} }
else { else {
...@@ -725,7 +721,7 @@ cc_add_item(ccobject *self, PyObject *key, PyObject *v) ...@@ -725,7 +721,7 @@ cc_add_item(ccobject *self, PyObject *key, PyObject *v)
jar = PyObject_GetAttr(v, py__p_jar); jar = PyObject_GetAttr(v, py__p_jar);
if (jar == NULL) if (jar == NULL)
return -1; return -1;
if (jar==Py_None) { if (jar == Py_None) {
Py_DECREF(jar); Py_DECREF(jar);
PyErr_SetString(PyExc_ValueError, PyErr_SetString(PyExc_ValueError,
"Cached object jar missing"); "Cached object jar missing");
...@@ -733,16 +729,14 @@ cc_add_item(ccobject *self, PyObject *key, PyObject *v) ...@@ -733,16 +729,14 @@ cc_add_item(ccobject *self, PyObject *key, PyObject *v)
} }
Py_DECREF(jar); Py_DECREF(jar);
object_again = object_from_oid(self, key); object_again = PyDict_GetItem(self->data, key);
if (object_again) { if (object_again) {
if (object_again != v) { if (object_again != v) {
Py_DECREF(object_again);
PyErr_SetString(PyExc_ValueError, PyErr_SetString(PyExc_ValueError,
"Can not re-register object under a different oid"); "Can not re-register object under a different oid");
return -1; return -1;
} else { } else {
/* re-register under the same oid - no work needed */ /* re-register under the same oid - no work needed */
Py_DECREF(object_again);
return 0; return 0;
} }
} }
...@@ -799,7 +793,7 @@ cc_del_item(ccobject *self, PyObject *key) ...@@ -799,7 +793,7 @@ cc_del_item(ccobject *self, PyObject *key)
cPersistentObject *p; cPersistentObject *p;
/* unlink this item from the ring */ /* unlink this item from the ring */
v = (PyObject *)object_from_oid(self, key); v = PyDict_GetItem(self->data, key);
if (v == NULL) if (v == NULL)
return -1; return -1;
...@@ -826,8 +820,6 @@ cc_del_item(ccobject *self, PyObject *key) ...@@ -826,8 +820,6 @@ cc_del_item(ccobject *self, PyObject *key)
p->cache = NULL; p->cache = NULL;
} }
Py_DECREF(v);
if (PyDict_DelItem(self->data, key) < 0) { if (PyDict_DelItem(self->data, key) < 0) {
PyErr_SetString(PyExc_RuntimeError, PyErr_SetString(PyExc_RuntimeError,
"unexpectedly couldn't remove key in cc_ass_sub"); "unexpectedly couldn't remove key in cc_ass_sub");
......
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
GitLab Nexedi Edition | About GitLab | About Nexedi | 沪ICP备2021021310号-2 | 沪ICP备2021021310号-7