Added _Py_ResetReferences(), if tracing references.

In _Py_PrintReferences(), no longer suppress once-referenced string.

Add Py_Malloc and friends and PyMem_Malloc and friends (malloc
wrappers for third parties).

diff --git a/Objects/object.c b/Objects/object.c
index 023477f4d12fe9d05e7c4e66d8a7582290f2c7ff..83829d26c2631a312451a1f3ced5a4ff249f7b06 100644 (file)
--- a/Objects/object.c
+++ b/Objects/object.c
@@ -569,6 +569,13 @@ PyObject _Py_NoneStruct = {
 
 static PyObject refchain = {&refchain, &refchain};
 
+void
+_Py_ResetReferences()
+{
+       refchain._ob_prev = refchain._ob_next = &refchain;
+       _Py_RefTotal = 0;
+}
+
 void
 _Py_NewReference(op)
        PyObject *op;
@@ -625,10 +632,8 @@ _Py_PrintReferences(fp)
        FILE *fp;
 {
        PyObject *op;
-       fprintf(fp, "Remaining objects (except strings referenced once):\n");
+       fprintf(fp, "Remaining objects:\n");
        for (op = refchain._ob_next; op != &refchain; op = op->_ob_next) {
-               if (op->ob_refcnt == 1 && PyString_Check(op))
-                       continue; /* Will be printed elsewhere */
                fprintf(fp, "[%d] ", op->ob_refcnt);
                if (PyObject_Print(op, fp, 0) != 0)
                        PyErr_Clear();
@@ -676,3 +681,79 @@ PyTypeObject *_Py_cobject_hack = &PyCObject_Type;
 
 /* Hack to force loading of abstract.o */
 int (*_Py_abstract_hack) Py_FPROTO((PyObject *)) = &PyObject_Length;
+
+
+/* Malloc wrappers (see mymalloc.h) */
+
+/* The Py_{Malloc,Realloc} wrappers call PyErr_NoMemory() on failure */
+
+ANY *
+Py_Malloc(nbytes)
+       size_t nbytes;
+{
+       ANY *p;
+#if _PyMem_EXTRA > 0
+       if (nbytes == 0)
+               nbytes = _PyMem_EXTRA;
+#endif
+       p = malloc(nbytes);
+       if (p != NULL)
+               return p;
+       else
+               return PyErr_NoMemory();
+}
+
+ANY *
+Py_Realloc(p, nbytes)
+       ANY *p;
+       size_t nbytes;
+{
+#if _PyMem_EXTRA > 0
+       if (nbytes == 0)
+               nbytes = _PyMem_EXTRA;
+#endif
+       p = realloc(p, nbytes);
+       if (p != NULL)
+               return p;
+       else
+               return PyErr_NoMemory();
+}
+
+void
+Py_Free(p)
+       ANY *p;
+{
+       free(p);
+}
+
+/* The PyMem_{Malloc,Realloc} wrappers don't call anything on failure */
+
+ANY *
+PyMem_Malloc(nbytes)
+       size_t nbytes;
+{
+#if _PyMem_EXTRA > 0
+       if (nbytes == 0)
+               nbytes = _PyMem_EXTRA;
+#endif
+       return malloc(nbytes);
+}
+
+ANY *
+PyMem_Realloc(p, nbytes)
+       ANY *p;
+       size_t nbytes;
+{
+#if _PyMem_EXTRA > 0
+       if (nbytes == 0)
+               nbytes = _PyMem_EXTRA;
+#endif
+       return realloc(p, nbytes);
+}
+
+void
+PyMem_Free(p)
+       ANY *p;
+{
+       free(p);
+}