Commit 4b4986f8 authored by Jeremy Hylton's avatar Jeremy Hylton

Fix ExtensionClass to work with Python debug builds.

When an object is initialized with null bytes, do not overwrite the
PyObject_HEAD or PyObject_VAR_HEAD, which is initialized by Python
core function, e.g. PyObject_NEW().

New function EC_NewObject() is called for all object creations --
replacing four calls to either PyObject_NEW() or PyObject_NEW_VAR().
The new function initializes all the bytes that ExtensionClass is
resposible for and INCREFs the base class.

In ALLOC_FREE() macro, call _Py_NewReference() for objects allocated
from the free list.
parent 54ad7f67
...@@ -33,7 +33,7 @@ ...@@ -33,7 +33,7 @@
USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH
DAMAGE. DAMAGE.
$Id: ExtensionClass.c,v 1.51 2002/03/08 18:34:23 jeremy Exp $ $Id: ExtensionClass.c,v 1.52 2002/06/10 22:30:47 jeremy Exp $
If you have questions regarding this software, If you have questions regarding this software,
contact: contact:
...@@ -54,7 +54,7 @@ static char ExtensionClass_module_documentation[] = ...@@ -54,7 +54,7 @@ static char ExtensionClass_module_documentation[] =
" - They provide access to unbound methods,\n" " - They provide access to unbound methods,\n"
" - They can be called to create instances.\n" " - They can be called to create instances.\n"
"\n" "\n"
"$Id: ExtensionClass.c,v 1.51 2002/03/08 18:34:23 jeremy Exp $\n" "$Id: ExtensionClass.c,v 1.52 2002/06/10 22:30:47 jeremy Exp $\n"
; ;
#include <stdio.h> #include <stdio.h>
...@@ -95,7 +95,8 @@ staticforward PyExtensionClass ECType; ...@@ -95,7 +95,8 @@ staticforward PyExtensionClass ECType;
if (free ## T) { \ if (free ## T) { \
self=free ## T; \ self=free ## T; \
free ## T=(T*)self->self; \ free ## T=(T*)self->self; \
self->ob_refcnt=1; \ _Py_NewReference((PyObject *)self); \
assert(self->ob_refcnt == 1); \
} \ } \
else UNLESS(self = PyObject_NEW(T, & T ## Type)) return NULL; else UNLESS(self = PyObject_NEW(T, & T ## Type)) return NULL;
...@@ -336,7 +337,29 @@ JimString_Build(va_alist) va_dcl ...@@ -336,7 +337,29 @@ JimString_Build(va_alist) va_dcl
return retval; return retval;
} }
static PyObject *
EC_NewObject(PyTypeObject *type, int size)
{
PyObject *inst;
int len;
if (type->tp_itemsize) {
inst = PyObject_NEW_VAR(PyObject, type, size);
if (inst == NULL)
return NULL;
((PyVarObject *)inst)->ob_size = size;
}
else {
assert(size == 0);
inst = PyObject_NEW(PyObject, type);
if (inst == NULL)
return NULL;
}
Py_INCREF(type);
len = (type->tp_basicsize + type->tp_itemsize * size) - sizeof(PyObject);
memset(((char *)inst) + sizeof(PyObject), 0, len);
return inst;
}
static int static int
CMethod_issubclass(PyExtensionClass *sub, PyExtensionClass *type) CMethod_issubclass(PyExtensionClass *sub, PyExtensionClass *type)
...@@ -1414,7 +1437,7 @@ static PyObject * ...@@ -1414,7 +1437,7 @@ static PyObject *
basicnew(PyExtensionClass *self, PyObject *args) basicnew(PyExtensionClass *self, PyObject *args)
{ {
PyObject *inst=0; PyObject *inst=0;
typedef struct { PyObject_VAR_HEAD } PyVarObject__; int size = 0;
if (! self->tp_dealloc) if (! self->tp_dealloc)
{ {
...@@ -1430,26 +1453,14 @@ basicnew(PyExtensionClass *self, PyObject *args) ...@@ -1430,26 +1453,14 @@ basicnew(PyExtensionClass *self, PyObject *args)
{ {
/* We have a variable-sized object, we need to get it's size */ /* We have a variable-sized object, we need to get it's size */
PyObject *var_size; PyObject *var_size;
int size;
UNLESS(var_size=CCL_getattr(self, py__var_size__, 0)) return NULL; UNLESS(var_size=CCL_getattr(self, py__var_size__, 0)) return NULL;
UNLESS_ASSIGN(var_size,PyObject_CallObject(var_size,NULL)) return NULL; UNLESS_ASSIGN(var_size,PyObject_CallObject(var_size,NULL)) return NULL;
size=PyInt_AsLong(var_size); size=PyInt_AsLong(var_size);
if (PyErr_Occurred()) return NULL; if (PyErr_Occurred()) return NULL;
UNLESS(inst=PyObject_NEW_VAR(PyObject,(PyTypeObject *)self, size))
return NULL;
memset(inst,0,self->tp_basicsize+self->tp_itemsize*size);
((PyVarObject__*)inst)->ob_size=size;
}
else
{
UNLESS(inst=PyObject_NEW(PyObject,(PyTypeObject *)self)) return NULL;
memset(inst,0,self->tp_basicsize);
} }
UNLESS(inst=EC_NewObject((PyTypeObject *)self, size))
inst->ob_refcnt=1; return NULL;
inst->ob_type=(PyTypeObject *)self;
Py_INCREF(self);
if (ClassHasInstDict(self)) if (ClassHasInstDict(self))
UNLESS(INSTANCE_DICT(inst)=PyDict_New()) goto err; UNLESS(INSTANCE_DICT(inst)=PyDict_New()) goto err;
...@@ -1990,7 +2001,7 @@ static PyObject * ...@@ -1990,7 +2001,7 @@ static PyObject *
CCL_call(PyExtensionClass *self, PyObject *arg, PyObject *kw) CCL_call(PyExtensionClass *self, PyObject *arg, PyObject *kw)
{ {
PyObject *inst=0, *init=0, *args=0; PyObject *inst=0, *init=0, *args=0;
typedef struct { PyObject_VAR_HEAD } PyVarObject__; int size = 0;
if (! self->tp_dealloc) if (! self->tp_dealloc)
{ {
...@@ -2003,7 +2014,6 @@ CCL_call(PyExtensionClass *self, PyObject *arg, PyObject *kw) ...@@ -2003,7 +2014,6 @@ CCL_call(PyExtensionClass *self, PyObject *arg, PyObject *kw)
{ {
/* We have a variable-sized object, we need to get it's size */ /* We have a variable-sized object, we need to get it's size */
PyObject *var_size; PyObject *var_size;
int size;
if ((var_size=CCL_getattr(self,py__var_size__, 0))) if ((var_size=CCL_getattr(self,py__var_size__, 0)))
{ {
...@@ -2032,20 +2042,8 @@ CCL_call(PyExtensionClass *self, PyObject *arg, PyObject *kw) ...@@ -2032,20 +2042,8 @@ CCL_call(PyExtensionClass *self, PyObject *arg, PyObject *kw)
return NULL; return NULL;
} }
} }
UNLESS(inst=PyObject_NEW_VAR(PyObject,(PyTypeObject *)self, size))
return NULL;
memset(inst,0,self->tp_basicsize+self->tp_itemsize*size);
((PyVarObject__*)inst)->ob_size=size;
}
else
{
UNLESS(inst=PyObject_NEW(PyObject,(PyTypeObject *)self)) return NULL;
memset(inst,0,self->tp_basicsize);
} }
UNLESS(inst=EC_NewObject((PyTypeObject *)self, size)) return NULL;
inst->ob_refcnt=1;
inst->ob_type=(PyTypeObject *)self;
Py_INCREF(self);
if (ClassHasInstDict(self)) if (ClassHasInstDict(self))
UNLESS(INSTANCE_DICT(inst)=PyDict_New()) goto err; UNLESS(INSTANCE_DICT(inst)=PyDict_New()) goto err;
......
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