Issue #22193: Added private function _PySys_GetSizeOf() needed to implement

some __sizeof__() methods.
This commit is contained in:
Serhiy Storchaka 2014-08-14 22:21:18 +03:00
parent 143fe05da1
commit 547d3bc3a6
2 changed files with 44 additions and 32 deletions

View File

@ -33,6 +33,10 @@ PyAPI_FUNC(int) PySys_HasWarnOptions(void);
PyAPI_FUNC(void) PySys_AddXOption(const wchar_t *); PyAPI_FUNC(void) PySys_AddXOption(const wchar_t *);
PyAPI_FUNC(PyObject *) PySys_GetXOptions(void); PyAPI_FUNC(PyObject *) PySys_GetXOptions(void);
#ifndef Py_LIMITED_API
PyAPI_DATA(size_t) _PySys_GetSizeOf(PyObject *);
#endif
#ifdef __cplusplus #ifdef __cplusplus
} }
#endif #endif

View File

@ -863,29 +863,16 @@ sys_mdebug(PyObject *self, PyObject *args)
} }
#endif /* USE_MALLOPT */ #endif /* USE_MALLOPT */
static PyObject * size_t
sys_getsizeof(PyObject *self, PyObject *args, PyObject *kwds) _PySys_GetSizeOf(PyObject *o)
{ {
PyObject *res = NULL; PyObject *res = NULL;
static PyObject *gc_head_size = NULL;
static char *kwlist[] = {"object", "default", 0};
PyObject *o, *dflt = NULL;
PyObject *method; PyObject *method;
size_t size;
if (!PyArg_ParseTupleAndKeywords(args, kwds, "O|O:getsizeof",
kwlist, &o, &dflt))
return NULL;
/* Initialize static variable for GC head size */
if (gc_head_size == NULL) {
gc_head_size = PyLong_FromSsize_t(sizeof(PyGC_Head));
if (gc_head_size == NULL)
return NULL;
}
/* Make sure the type is initialized. float gets initialized late */ /* Make sure the type is initialized. float gets initialized late */
if (PyType_Ready(Py_TYPE(o)) < 0) if (PyType_Ready(Py_TYPE(o)) < 0)
return NULL; return (size_t)-1;
method = _PyObject_LookupSpecial(o, &PyId___sizeof__); method = _PyObject_LookupSpecial(o, &PyId___sizeof__);
if (method == NULL) { if (method == NULL) {
@ -899,24 +886,45 @@ sys_getsizeof(PyObject *self, PyObject *args, PyObject *kwds)
Py_DECREF(method); Py_DECREF(method);
} }
/* Has a default value been given */ if (res == NULL)
if ((res == NULL) && (dflt != NULL) && return (size_t)-1;
PyErr_ExceptionMatches(PyExc_TypeError))
size = PyLong_AsSize_t(res);
Py_DECREF(res);
if (size == (size_t)-1 && PyErr_Occurred())
return (size_t)-1;
/* add gc_head size */
if (PyObject_IS_GC(o))
size += sizeof(PyGC_Head);
return size;
}
static PyObject *
sys_getsizeof(PyObject *self, PyObject *args, PyObject *kwds)
{ {
static char *kwlist[] = {"object", "default", 0};
size_t size;
PyObject *o, *dflt = NULL;
if (!PyArg_ParseTupleAndKeywords(args, kwds, "O|O:getsizeof",
kwlist, &o, &dflt))
return NULL;
size = _PySys_GetSizeOf(o);
if (size == (size_t)-1 && PyErr_Occurred()) {
/* Has a default value been given */
if (dflt != NULL && PyErr_ExceptionMatches(PyExc_TypeError)) {
PyErr_Clear(); PyErr_Clear();
Py_INCREF(dflt); Py_INCREF(dflt);
return dflt; return dflt;
} }
else if (res == NULL) else
return res; return NULL;
/* add gc_head size */
if (PyObject_IS_GC(o)) {
PyObject *tmp = res;
res = PyNumber_Add(tmp, gc_head_size);
Py_DECREF(tmp);
} }
return res;
return PyLong_FromSize_t(size);
} }
PyDoc_STRVAR(getsizeof_doc, PyDoc_STRVAR(getsizeof_doc,