blob: 7976b40058d5d4475cbb3570486c664123dbb812 [file] [log] [blame]
Jeremy Hyltonc5007aa2000-06-30 05:02:53 +00001/*
Tim Peters88396172002-06-30 17:56:40 +00002
Jeremy Hyltonc5007aa2000-06-30 05:02:53 +00003 Reference Cycle Garbage Collection
4 ==================================
5
Neil Schemenauerb2c2c9e2000-10-04 16:34:09 +00006 Neil Schemenauer <nas@arctrix.com>
Jeremy Hyltonc5007aa2000-06-30 05:02:53 +00007
8 Based on a post on the python-dev list. Ideas from Guido van Rossum,
9 Eric Tiedemann, and various others.
10
Neil Schemenauer43411b52001-08-30 00:05:51 +000011 http://www.arctrix.com/nas/python/gc/
Jeremy Hyltonc5007aa2000-06-30 05:02:53 +000012 http://www.python.org/pipermail/python-dev/2000-March/003869.html
13 http://www.python.org/pipermail/python-dev/2000-March/004010.html
14 http://www.python.org/pipermail/python-dev/2000-March/004022.html
15
16 For a highlevel view of the collection process, read the collect
17 function.
18
Jeremy Hyltonc5007aa2000-06-30 05:02:53 +000019*/
20
Jeremy Hyltonc5007aa2000-06-30 05:02:53 +000021#include "Python.h"
22
Neil Schemenauer43411b52001-08-30 00:05:51 +000023/* Get an object's GC head */
24#define AS_GC(o) ((PyGC_Head *)(o)-1)
25
26/* Get the object given the GC head */
27#define FROM_GC(g) ((PyObject *)(((PyGC_Head *)g)+1))
28
Jeremy Hyltonc5007aa2000-06-30 05:02:53 +000029/*** Global GC state ***/
30
Neil Schemenauer2880ae52002-05-04 05:35:20 +000031struct gc_generation {
32 PyGC_Head head;
33 int threshold; /* collection threshold */
34 int count; /* count of allocations or collections of younger
35 generations */
36};
37
38#define NUM_GENERATIONS 3
39#define GEN_HEAD(n) (&generations[n].head)
40
Jeremy Hyltonc5007aa2000-06-30 05:02:53 +000041/* linked lists of container objects */
Neil Schemenauer2880ae52002-05-04 05:35:20 +000042static struct gc_generation generations[NUM_GENERATIONS] = {
43 /* PyGC_Head, threshold, count */
44 {{{GEN_HEAD(0), GEN_HEAD(0), 0}}, 700, 0},
45 {{{GEN_HEAD(1), GEN_HEAD(1), 0}}, 10, 0},
46 {{{GEN_HEAD(2), GEN_HEAD(2), 0}}, 10, 0},
47};
Jeremy Hyltonc5007aa2000-06-30 05:02:53 +000048
Neil Schemenauer2880ae52002-05-04 05:35:20 +000049PyGC_Head *_PyGC_generation0 = GEN_HEAD(0);
50
Vladimir Marangozovf9d20c32000-08-06 22:45:31 +000051static int enabled = 1; /* automatic collection enabled? */
Jeremy Hyltonc5007aa2000-06-30 05:02:53 +000052
Neil Schemenauer43411b52001-08-30 00:05:51 +000053/* true if we are currently running the collector */
Tim Petersbf384c22003-04-06 00:11:39 +000054static int collecting = 0;
Neil Schemenauer43411b52001-08-30 00:05:51 +000055
Tim Peters6fc13d92002-07-02 18:12:35 +000056/* list of uncollectable objects */
Tim Petersbf384c22003-04-06 00:11:39 +000057static PyObject *garbage = NULL;
Tim Peters6fc13d92002-07-02 18:12:35 +000058
59/* Python string to use if unhandled exception occurs */
Tim Petersbf384c22003-04-06 00:11:39 +000060static PyObject *gc_str = NULL;
Tim Peters6fc13d92002-07-02 18:12:35 +000061
Tim Peters93ad66d2003-04-05 17:15:44 +000062/* Python string used to look for __del__ attribute. */
63static PyObject *delstr = NULL;
Jeremy Hyltonce136e92003-04-04 19:59:06 +000064
Jeremy Hyltonc5007aa2000-06-30 05:02:53 +000065/* set for debugging information */
66#define DEBUG_STATS (1<<0) /* print collection statistics */
67#define DEBUG_COLLECTABLE (1<<1) /* print collectable objects */
68#define DEBUG_UNCOLLECTABLE (1<<2) /* print uncollectable objects */
69#define DEBUG_INSTANCES (1<<3) /* print instances */
70#define DEBUG_OBJECTS (1<<4) /* print other objects */
Neil Schemenauer544de1e2000-09-22 15:22:38 +000071#define DEBUG_SAVEALL (1<<5) /* save all garbage in gc.garbage */
Jeremy Hyltonc5007aa2000-06-30 05:02:53 +000072#define DEBUG_LEAK DEBUG_COLLECTABLE | \
73 DEBUG_UNCOLLECTABLE | \
74 DEBUG_INSTANCES | \
Neil Schemenauer544de1e2000-09-22 15:22:38 +000075 DEBUG_OBJECTS | \
76 DEBUG_SAVEALL
Jeremy Hyltonb709df32000-09-01 02:47:25 +000077static int debug;
Jeremy Hyltonc5007aa2000-06-30 05:02:53 +000078
Tim Peters6fc13d92002-07-02 18:12:35 +000079/*--------------------------------------------------------------------------
80gc_refs values.
Neil Schemenauer43411b52001-08-30 00:05:51 +000081
Tim Peters6fc13d92002-07-02 18:12:35 +000082Between collections, every gc'ed object has one of two gc_refs values:
83
84GC_UNTRACKED
85 The initial state; objects returned by PyObject_GC_Malloc are in this
86 state. The object doesn't live in any generation list, and its
87 tp_traverse slot must not be called.
88
89GC_REACHABLE
90 The object lives in some generation list, and its tp_traverse is safe to
91 call. An object transitions to GC_REACHABLE when PyObject_GC_Track
92 is called.
93
94During a collection, gc_refs can temporarily take on other states:
95
96>= 0
97 At the start of a collection, update_refs() copies the true refcount
98 to gc_refs, for each object in the generation being collected.
99 subtract_refs() then adjusts gc_refs so that it equals the number of
100 times an object is referenced directly from outside the generation
101 being collected.
Martin v. Löwis774348c2002-11-09 19:54:06 +0000102 gc_refs remains >= 0 throughout these steps.
Tim Peters6fc13d92002-07-02 18:12:35 +0000103
104GC_TENTATIVELY_UNREACHABLE
105 move_unreachable() then moves objects not reachable (whether directly or
106 indirectly) from outside the generation into an "unreachable" set.
107 Objects that are found to be reachable have gc_refs set to GC_REACHABLE
108 again. Objects that are found to be unreachable have gc_refs set to
109 GC_TENTATIVELY_UNREACHABLE. It's "tentatively" because the pass doing
110 this can't be sure until it ends, and GC_TENTATIVELY_UNREACHABLE may
111 transition back to GC_REACHABLE.
112
113 Only objects with GC_TENTATIVELY_UNREACHABLE still set are candidates
114 for collection. If it's decided not to collect such an object (e.g.,
115 it has a __del__ method), its gc_refs is restored to GC_REACHABLE again.
116----------------------------------------------------------------------------
117*/
Tim Petersea405632002-07-02 00:52:30 +0000118#define GC_UNTRACKED _PyGC_REFS_UNTRACKED
119#define GC_REACHABLE _PyGC_REFS_REACHABLE
120#define GC_TENTATIVELY_UNREACHABLE _PyGC_REFS_TENTATIVELY_UNREACHABLE
Tim Peters19b74c72002-07-01 03:52:19 +0000121
Tim Peters6fc13d92002-07-02 18:12:35 +0000122#define IS_TRACKED(o) ((AS_GC(o))->gc.gc_refs != GC_UNTRACKED)
Tim Peters19b74c72002-07-01 03:52:19 +0000123#define IS_REACHABLE(o) ((AS_GC(o))->gc.gc_refs == GC_REACHABLE)
124#define IS_TENTATIVELY_UNREACHABLE(o) ( \
125 (AS_GC(o))->gc.gc_refs == GC_TENTATIVELY_UNREACHABLE)
Neil Schemenauera2b11ec2002-05-21 15:53:24 +0000126
Jeremy Hyltonc5007aa2000-06-30 05:02:53 +0000127/*** list functions ***/
128
129static void
130gc_list_init(PyGC_Head *list)
131{
Tim Peters9e4ca102001-10-11 18:31:31 +0000132 list->gc.gc_prev = list;
133 list->gc.gc_next = list;
Jeremy Hyltonc5007aa2000-06-30 05:02:53 +0000134}
135
Neil Schemenauer2880ae52002-05-04 05:35:20 +0000136static int
137gc_list_is_empty(PyGC_Head *list)
138{
139 return (list->gc.gc_next == list);
140}
141
Jeremy Hyltonc5007aa2000-06-30 05:02:53 +0000142static void
143gc_list_append(PyGC_Head *node, PyGC_Head *list)
144{
Tim Peters9e4ca102001-10-11 18:31:31 +0000145 node->gc.gc_next = list;
146 node->gc.gc_prev = list->gc.gc_prev;
147 node->gc.gc_prev->gc.gc_next = node;
148 list->gc.gc_prev = node;
Jeremy Hyltonc5007aa2000-06-30 05:02:53 +0000149}
150
151static void
152gc_list_remove(PyGC_Head *node)
153{
Tim Peters9e4ca102001-10-11 18:31:31 +0000154 node->gc.gc_prev->gc.gc_next = node->gc.gc_next;
155 node->gc.gc_next->gc.gc_prev = node->gc.gc_prev;
156 node->gc.gc_next = NULL; /* object is not currently tracked */
Jeremy Hyltonc5007aa2000-06-30 05:02:53 +0000157}
158
Jeremy Hyltonc5007aa2000-06-30 05:02:53 +0000159/* append a list onto another list, from becomes an empty list */
160static void
161gc_list_merge(PyGC_Head *from, PyGC_Head *to)
162{
163 PyGC_Head *tail;
Neil Schemenauer2880ae52002-05-04 05:35:20 +0000164 if (!gc_list_is_empty(from)) {
Tim Peters9e4ca102001-10-11 18:31:31 +0000165 tail = to->gc.gc_prev;
166 tail->gc.gc_next = from->gc.gc_next;
167 tail->gc.gc_next->gc.gc_prev = tail;
168 to->gc.gc_prev = from->gc.gc_prev;
169 to->gc.gc_prev->gc.gc_next = to;
Jeremy Hyltonc5007aa2000-06-30 05:02:53 +0000170 }
171 gc_list_init(from);
172}
173
174static long
175gc_list_size(PyGC_Head *list)
176{
177 PyGC_Head *gc;
178 long n = 0;
Tim Peters9e4ca102001-10-11 18:31:31 +0000179 for (gc = list->gc.gc_next; gc != list; gc = gc->gc.gc_next) {
Jeremy Hyltonc5007aa2000-06-30 05:02:53 +0000180 n++;
181 }
182 return n;
183}
184
Tim Peters259272b2003-04-06 19:41:39 +0000185/* Append objects in a GC list to a Python list.
186 * Return 0 if all OK, < 0 if error (out of memory for list).
187 */
188static int
189append_objects(PyObject *py_list, PyGC_Head *gc_list)
190{
191 PyGC_Head *gc;
192 for (gc = gc_list->gc.gc_next; gc != gc_list; gc = gc->gc.gc_next) {
193 PyObject *op = FROM_GC(gc);
194 if (op != py_list) {
195 if (PyList_Append(py_list, op)) {
196 return -1; /* exception */
197 }
198 }
199 }
200 return 0;
201}
202
Jeremy Hyltonc5007aa2000-06-30 05:02:53 +0000203/*** end of list stuff ***/
204
205
Tim Peters19b74c72002-07-01 03:52:19 +0000206/* Set all gc_refs = ob_refcnt. After this, gc_refs is > 0 for all objects
207 * in containers, and is GC_REACHABLE for all tracked gc objects not in
208 * containers.
Tim Peters88396172002-06-30 17:56:40 +0000209 */
Jeremy Hyltonc5007aa2000-06-30 05:02:53 +0000210static void
211update_refs(PyGC_Head *containers)
212{
Tim Peters9e4ca102001-10-11 18:31:31 +0000213 PyGC_Head *gc = containers->gc.gc_next;
Tim Petersea405632002-07-02 00:52:30 +0000214 for (; gc != containers; gc = gc->gc.gc_next) {
215 assert(gc->gc.gc_refs == GC_REACHABLE);
Tim Peters9e4ca102001-10-11 18:31:31 +0000216 gc->gc.gc_refs = FROM_GC(gc)->ob_refcnt;
Tim Peters780c4972003-11-14 00:01:17 +0000217 /* Python's cyclic gc should never see an incoming refcount
218 * of 0: if something decref'ed to 0, it should have been
219 * deallocated immediately at that time.
220 * Possible cause (if the assert triggers): a tp_dealloc
221 * routine left a gc-aware object tracked during its teardown
222 * phase, and did something-- or allowed something to happen --
223 * that called back into Python. gc can trigger then, and may
224 * see the still-tracked dying object. Before this assert
225 * was added, such mistakes went on to allow gc to try to
226 * delete the object again. In a debug build, that caused
227 * a mysterious segfault, when _Py_ForgetReference tried
228 * to remove the object from the doubly-linked list of all
229 * objects a second time. In a release build, an actual
230 * double deallocation occurred, which leads to corruption
231 * of the allocator's internal bookkeeping pointers. That's
232 * so serious that maybe this should be a release-build
233 * check instead of an assert?
234 */
235 assert(gc->gc.gc_refs != 0);
Tim Petersea405632002-07-02 00:52:30 +0000236 }
Jeremy Hyltonc5007aa2000-06-30 05:02:53 +0000237}
238
Tim Peters19b74c72002-07-01 03:52:19 +0000239/* A traversal callback for subtract_refs. */
Jeremy Hyltonc5007aa2000-06-30 05:02:53 +0000240static int
241visit_decref(PyObject *op, void *data)
242{
Tim Peters93cd83e2002-06-30 21:31:03 +0000243 assert(op != NULL);
Tim Peters19b74c72002-07-01 03:52:19 +0000244 if (PyObject_IS_GC(op)) {
245 PyGC_Head *gc = AS_GC(op);
246 /* We're only interested in gc_refs for objects in the
247 * generation being collected, which can be recognized
248 * because only they have positive gc_refs.
249 */
Tim Petersaab713b2002-07-02 22:15:28 +0000250 assert(gc->gc.gc_refs != 0); /* else refcount was too small */
Tim Peters19b74c72002-07-01 03:52:19 +0000251 if (gc->gc.gc_refs > 0)
252 gc->gc.gc_refs--;
253 }
Jeremy Hyltonc5007aa2000-06-30 05:02:53 +0000254 return 0;
255}
256
Tim Peters19b74c72002-07-01 03:52:19 +0000257/* Subtract internal references from gc_refs. After this, gc_refs is >= 0
258 * for all objects in containers, and is GC_REACHABLE for all tracked gc
259 * objects not in containers. The ones with gc_refs > 0 are directly
260 * reachable from outside containers, and so can't be collected.
261 */
Jeremy Hyltonc5007aa2000-06-30 05:02:53 +0000262static void
263subtract_refs(PyGC_Head *containers)
264{
265 traverseproc traverse;
Tim Peters9e4ca102001-10-11 18:31:31 +0000266 PyGC_Head *gc = containers->gc.gc_next;
267 for (; gc != containers; gc=gc->gc.gc_next) {
Neil Schemenauer43411b52001-08-30 00:05:51 +0000268 traverse = FROM_GC(gc)->ob_type->tp_traverse;
269 (void) traverse(FROM_GC(gc),
Jeremy Hyltonc5007aa2000-06-30 05:02:53 +0000270 (visitproc)visit_decref,
271 NULL);
272 }
273}
274
Tim Peters19b74c72002-07-01 03:52:19 +0000275/* A traversal callback for move_unreachable. */
Jeremy Hyltonc5007aa2000-06-30 05:02:53 +0000276static int
Tim Peters19b74c72002-07-01 03:52:19 +0000277visit_reachable(PyObject *op, PyGC_Head *reachable)
Jeremy Hyltonc5007aa2000-06-30 05:02:53 +0000278{
Tim Petersea405632002-07-02 00:52:30 +0000279 if (PyObject_IS_GC(op)) {
Tim Peters19b74c72002-07-01 03:52:19 +0000280 PyGC_Head *gc = AS_GC(op);
281 const int gc_refs = gc->gc.gc_refs;
282
283 if (gc_refs == 0) {
284 /* This is in move_unreachable's 'young' list, but
285 * the traversal hasn't yet gotten to it. All
286 * we need to do is tell move_unreachable that it's
287 * reachable.
288 */
289 gc->gc.gc_refs = 1;
Jeremy Hyltonc5007aa2000-06-30 05:02:53 +0000290 }
Tim Peters19b74c72002-07-01 03:52:19 +0000291 else if (gc_refs == GC_TENTATIVELY_UNREACHABLE) {
292 /* This had gc_refs = 0 when move_unreachable got
293 * to it, but turns out it's reachable after all.
294 * Move it back to move_unreachable's 'young' list,
295 * and move_unreachable will eventually get to it
296 * again.
297 */
298 gc_list_remove(gc);
299 gc_list_append(gc, reachable);
300 gc->gc.gc_refs = 1;
301 }
302 /* Else there's nothing to do.
303 * If gc_refs > 0, it must be in move_unreachable's 'young'
304 * list, and move_unreachable will eventually get to it.
305 * If gc_refs == GC_REACHABLE, it's either in some other
306 * generation so we don't care about it, or move_unreachable
Tim Peters6fc13d92002-07-02 18:12:35 +0000307 * already dealt with it.
Tim Petersea405632002-07-02 00:52:30 +0000308 * If gc_refs == GC_UNTRACKED, it must be ignored.
Tim Peters19b74c72002-07-01 03:52:19 +0000309 */
Tim Petersea405632002-07-02 00:52:30 +0000310 else {
311 assert(gc_refs > 0
312 || gc_refs == GC_REACHABLE
313 || gc_refs == GC_UNTRACKED);
314 }
Jeremy Hyltonc5007aa2000-06-30 05:02:53 +0000315 }
316 return 0;
317}
318
Tim Peters19b74c72002-07-01 03:52:19 +0000319/* Move the unreachable objects from young to unreachable. After this,
320 * all objects in young have gc_refs = GC_REACHABLE, and all objects in
321 * unreachable have gc_refs = GC_TENTATIVELY_UNREACHABLE. All tracked
322 * gc objects not in young or unreachable still have gc_refs = GC_REACHABLE.
323 * All objects in young after this are directly or indirectly reachable
324 * from outside the original young; and all objects in unreachable are
325 * not.
Tim Peters88396172002-06-30 17:56:40 +0000326 */
Jeremy Hyltonc5007aa2000-06-30 05:02:53 +0000327static void
Tim Peters19b74c72002-07-01 03:52:19 +0000328move_unreachable(PyGC_Head *young, PyGC_Head *unreachable)
Jeremy Hyltonc5007aa2000-06-30 05:02:53 +0000329{
Tim Peters19b74c72002-07-01 03:52:19 +0000330 PyGC_Head *gc = young->gc.gc_next;
331
332 /* Invariants: all objects "to the left" of us in young have gc_refs
333 * = GC_REACHABLE, and are indeed reachable (directly or indirectly)
334 * from outside the young list as it was at entry. All other objects
335 * from the original young "to the left" of us are in unreachable now,
336 * and have gc_refs = GC_TENTATIVELY_UNREACHABLE. All objects to the
337 * left of us in 'young' now have been scanned, and no objects here
338 * or to the right have been scanned yet.
339 */
340
341 while (gc != young) {
342 PyGC_Head *next;
343
Tim Peters6fc13d92002-07-02 18:12:35 +0000344 if (gc->gc.gc_refs) {
345 /* gc is definitely reachable from outside the
346 * original 'young'. Mark it as such, and traverse
347 * its pointers to find any other objects that may
348 * be directly reachable from it. Note that the
349 * call to tp_traverse may append objects to young,
350 * so we have to wait until it returns to determine
351 * the next object to visit.
352 */
353 PyObject *op = FROM_GC(gc);
354 traverseproc traverse = op->ob_type->tp_traverse;
355 assert(gc->gc.gc_refs > 0);
356 gc->gc.gc_refs = GC_REACHABLE;
357 (void) traverse(op,
358 (visitproc)visit_reachable,
359 (void *)young);
360 next = gc->gc.gc_next;
361 }
362 else {
Tim Peters19b74c72002-07-01 03:52:19 +0000363 /* This *may* be unreachable. To make progress,
364 * assume it is. gc isn't directly reachable from
365 * any object we've already traversed, but may be
366 * reachable from an object we haven't gotten to yet.
367 * visit_reachable will eventually move gc back into
368 * young if that's so, and we'll see it again.
369 */
370 next = gc->gc.gc_next;
371 gc_list_remove(gc);
372 gc_list_append(gc, unreachable);
373 gc->gc.gc_refs = GC_TENTATIVELY_UNREACHABLE;
374 }
Tim Peters19b74c72002-07-01 03:52:19 +0000375 gc = next;
Jeremy Hyltonc5007aa2000-06-30 05:02:53 +0000376 }
377}
378
Tim Peters86b993b2003-04-05 17:35:54 +0000379/* Return true if object has a finalization method.
380 * CAUTION: An instance of an old-style class has to be checked for a
Tim Petersf6b80452003-04-07 19:21:15 +0000381 *__del__ method, and earlier versions of this used to call PyObject_HasAttr,
382 * which in turn could call the class's __getattr__ hook (if any). That
383 * could invoke arbitrary Python code, mutating the object graph in arbitrary
384 * ways, and that was the source of some excruciatingly subtle bugs.
Tim Peters86b993b2003-04-05 17:35:54 +0000385 */
Neil Schemenauera765c122001-11-01 17:35:23 +0000386static int
387has_finalizer(PyObject *op)
Jeremy Hyltonc5007aa2000-06-30 05:02:53 +0000388{
Tim Peters86b993b2003-04-05 17:35:54 +0000389 if (PyInstance_Check(op)) {
Tim Peters86b993b2003-04-05 17:35:54 +0000390 assert(delstr != NULL);
Tim Petersf6b80452003-04-07 19:21:15 +0000391 return _PyInstance_Lookup(op, delstr) != NULL;
Tim Peters86b993b2003-04-05 17:35:54 +0000392 }
393 else if (PyType_HasFeature(op->ob_type, Py_TPFLAGS_HEAPTYPE))
394 return op->ob_type->tp_del != NULL;
395 else
396 return 0;
Neil Schemenauera765c122001-11-01 17:35:23 +0000397}
398
Tim Peters403a2032003-11-20 21:21:46 +0000399/* Move the objects in unreachable with __del__ methods into finalizers,
400 * and weakrefs with callbacks into wr_callbacks.
401 * The objects remaining in unreachable do not have __del__ methods, and are
402 * not weakrefs with callbacks.
403 * The objects moved have gc_refs changed to GC_REACHABLE; the objects
404 * remaining in unreachable are left at GC_TENTATIVELY_UNREACHABLE.
Jeremy Hyltonce136e92003-04-04 19:59:06 +0000405 */
Neil Schemenauera765c122001-11-01 17:35:23 +0000406static void
Tim Peters403a2032003-11-20 21:21:46 +0000407move_troublemakers(PyGC_Head *unreachable,
408 PyGC_Head *finalizers,
409 PyGC_Head *wr_callbacks)
Neil Schemenauera765c122001-11-01 17:35:23 +0000410{
Tim Petersf6b80452003-04-07 19:21:15 +0000411 PyGC_Head *gc = unreachable->gc.gc_next;
412
413 while (gc != unreachable) {
Neil Schemenauer43411b52001-08-30 00:05:51 +0000414 PyObject *op = FROM_GC(gc);
Tim Petersf6b80452003-04-07 19:21:15 +0000415 PyGC_Head *next = gc->gc.gc_next;
Jeremy Hyltonce136e92003-04-04 19:59:06 +0000416
Tim Petersf6ae7a42003-04-05 18:40:50 +0000417 assert(IS_TENTATIVELY_UNREACHABLE(op));
418
Tim Petersf6b80452003-04-07 19:21:15 +0000419 if (has_finalizer(op)) {
Tim Petersf6ae7a42003-04-05 18:40:50 +0000420 gc_list_remove(gc);
Tim Petersf6b80452003-04-07 19:21:15 +0000421 gc_list_append(gc, finalizers);
422 gc->gc.gc_refs = GC_REACHABLE;
Jeremy Hyltonce136e92003-04-04 19:59:06 +0000423 }
Tim Peters403a2032003-11-20 21:21:46 +0000424 else if (PyWeakref_Check(op) &&
425 ((PyWeakReference *)op)->wr_callback) {
426 gc_list_remove(gc);
427 gc_list_append(gc, wr_callbacks);
428 gc->gc.gc_refs = GC_REACHABLE;
429 }
Tim Petersf6b80452003-04-07 19:21:15 +0000430 gc = next;
Jeremy Hyltonc5007aa2000-06-30 05:02:53 +0000431 }
432}
433
Tim Peters19b74c72002-07-01 03:52:19 +0000434/* A traversal callback for move_finalizer_reachable. */
435static int
436visit_move(PyObject *op, PyGC_Head *tolist)
437{
438 if (PyObject_IS_GC(op)) {
Tim Petersea405632002-07-02 00:52:30 +0000439 if (IS_TENTATIVELY_UNREACHABLE(op)) {
Tim Peters19b74c72002-07-01 03:52:19 +0000440 PyGC_Head *gc = AS_GC(op);
441 gc_list_remove(gc);
442 gc_list_append(gc, tolist);
443 gc->gc.gc_refs = GC_REACHABLE;
444 }
445 }
446 return 0;
447}
448
449/* Move objects that are reachable from finalizers, from the unreachable set
Tim Petersf6b80452003-04-07 19:21:15 +0000450 * into finalizers set.
Tim Peters19b74c72002-07-01 03:52:19 +0000451 */
Jeremy Hyltonc5007aa2000-06-30 05:02:53 +0000452static void
Tim Petersf6b80452003-04-07 19:21:15 +0000453move_finalizer_reachable(PyGC_Head *finalizers)
Jeremy Hyltonc5007aa2000-06-30 05:02:53 +0000454{
455 traverseproc traverse;
Tim Peters9e4ca102001-10-11 18:31:31 +0000456 PyGC_Head *gc = finalizers->gc.gc_next;
Tim Petersbf384c22003-04-06 00:11:39 +0000457 for (; gc != finalizers; gc = gc->gc.gc_next) {
458 /* Note that the finalizers list may grow during this. */
Neil Schemenauer43411b52001-08-30 00:05:51 +0000459 traverse = FROM_GC(gc)->ob_type->tp_traverse;
Tim Peters88396172002-06-30 17:56:40 +0000460 (void) traverse(FROM_GC(gc),
Tim Petersbf384c22003-04-06 00:11:39 +0000461 (visitproc)visit_move,
Tim Petersf6b80452003-04-07 19:21:15 +0000462 (void *)finalizers);
Jeremy Hyltonc5007aa2000-06-30 05:02:53 +0000463 }
464}
465
Tim Peters403a2032003-11-20 21:21:46 +0000466/* Clear all trash weakrefs with callbacks. This clears weakrefs first,
467 * which has the happy result of disabling the callbacks without executing
468 * them. A nasty technical complication: a weakref callback can itself be
469 * the target of a weakref, in which case decrefing the callback can cause
470 * another callback to trigger. But we can't allow arbitrary Python code to
471 * get executed at this point (the callback on the callback may try to muck
472 * with other cyclic trash we're trying to collect, even resurrecting it
473 * while we're in the middle of doing tp_clear() on the trash).
474 *
475 * The private _PyWeakref_ClearRef() function exists so that we can clear
476 * the reference in a weakref without triggering a callback on the callback.
477 *
478 * We have to save the callback objects and decref them later. But we can't
479 * allocate new memory to save them (if we can't get new memory, we're dead).
480 * So we grab a new reference on the clear'ed weakref, which prevents the
481 * rest of gc from reclaiming it. _PyWeakref_ClearRef() leaves the
482 * weakref's wr_callback member intact.
483 *
484 * In the end, then, wr_callbacks consists of cleared weakrefs that are
485 * immune from collection. Near the end of gc, after collecting all the
486 * cyclic trash, we call release_weakrefs(). That releases our references
487 * to the cleared weakrefs, which in turn may trigger callbacks on their
488 * callbacks.
489 */
490static void
491clear_weakrefs(PyGC_Head *wr_callbacks)
492{
493 PyGC_Head *gc = wr_callbacks->gc.gc_next;
494
495 for (; gc != wr_callbacks; gc = gc->gc.gc_next) {
496 PyObject *op = FROM_GC(gc);
497 PyWeakReference *wr;
498
499 assert(IS_REACHABLE(op));
500 assert(PyWeakref_Check(op));
501 wr = (PyWeakReference *)op;
502 assert(wr->wr_callback != NULL);
503 Py_INCREF(op);
504 _PyWeakref_ClearRef(wr);
505 }
506}
507
508/* Called near the end of gc. This gives up the references we own to
509 * cleared weakrefs, allowing them to get collected, and in turn decref'ing
510 * their callbacks.
511 *
512 * If a callback object is itself the target of a weakref callback,
513 * decref'ing the callback object may trigger that other callback. If
514 * that other callback was part of the cyclic trash in this generation,
515 * that won't happen, since we cleared *all* trash-weakref callbacks near
516 * the start of gc. If that other callback was not part of the cyclic trash
517 * in this generation, then it acted like an external root to this round
518 * of gc, so all the objects reachable from that callback are still alive.
519 *
520 * Giving up the references to the weakref objects will probably make
521 * them go away too. However, if a weakref is reachable from finalizers,
522 * it won't go away. We move it to the old generation then. Since a
523 * weakref object doesn't have a finalizer, that's the right thing to do (it
524 * doesn't belong in gc.garbage).
525 *
526 * We return the number of weakref objects freed (those not appended to old).
527 */
528static int
529release_weakrefs(PyGC_Head *wr_callbacks, PyGC_Head *old)
530{
531 int num_freed = 0;
532
533 while (! gc_list_is_empty(wr_callbacks)) {
534 PyGC_Head *gc = wr_callbacks->gc.gc_next;
535 PyObject *op = FROM_GC(gc);
536 PyWeakReference *wr = (PyWeakReference *)op;
537
538 assert(IS_REACHABLE(op));
539 assert(PyWeakref_Check(op));
540 assert(wr->wr_callback != NULL);
541 Py_DECREF(op);
542 if (wr_callbacks->gc.gc_next == gc) {
543 /* object is still alive -- move it */
544 gc_list_remove(gc);
545 gc_list_append(gc, old);
546 }
547 else
548 ++num_freed;
549 }
550 return num_freed;
551}
552
Jeremy Hyltonc5007aa2000-06-30 05:02:53 +0000553static void
Jeremy Hylton06257772000-08-31 15:10:24 +0000554debug_instance(char *msg, PyInstanceObject *inst)
Jeremy Hyltonc5007aa2000-06-30 05:02:53 +0000555{
Jeremy Hyltonc5007aa2000-06-30 05:02:53 +0000556 char *cname;
Neil Schemenauera765c122001-11-01 17:35:23 +0000557 /* simple version of instance_repr */
Jeremy Hyltonc5007aa2000-06-30 05:02:53 +0000558 PyObject *classname = inst->in_class->cl_name;
559 if (classname != NULL && PyString_Check(classname))
560 cname = PyString_AsString(classname);
561 else
562 cname = "?";
Jeremy Hylton06257772000-08-31 15:10:24 +0000563 PySys_WriteStderr("gc: %.100s <%.100s instance at %p>\n",
564 msg, cname, inst);
Jeremy Hyltonc5007aa2000-06-30 05:02:53 +0000565}
566
567static void
Jeremy Hylton06257772000-08-31 15:10:24 +0000568debug_cycle(char *msg, PyObject *op)
Jeremy Hyltonc5007aa2000-06-30 05:02:53 +0000569{
570 if ((debug & DEBUG_INSTANCES) && PyInstance_Check(op)) {
Jeremy Hylton06257772000-08-31 15:10:24 +0000571 debug_instance(msg, (PyInstanceObject *)op);
Neil Schemenauer544de1e2000-09-22 15:22:38 +0000572 }
573 else if (debug & DEBUG_OBJECTS) {
Jeremy Hylton06257772000-08-31 15:10:24 +0000574 PySys_WriteStderr("gc: %.100s <%.100s %p>\n",
575 msg, op->ob_type->tp_name, op);
Jeremy Hyltonc5007aa2000-06-30 05:02:53 +0000576 }
577}
578
Tim Petersbf384c22003-04-06 00:11:39 +0000579/* Handle uncollectable garbage (cycles with finalizers, and stuff reachable
580 * only from such cycles).
Tim Petersf6b80452003-04-07 19:21:15 +0000581 * If DEBUG_SAVEALL, all objects in finalizers are appended to the module
582 * garbage list (a Python list), else only the objects in finalizers with
583 * __del__ methods are appended to garbage. All objects in finalizers are
584 * merged into the old list regardless.
Tim Peters259272b2003-04-06 19:41:39 +0000585 * Returns 0 if all OK, <0 on error (out of memory to grow the garbage list).
586 * The finalizers list is made empty on a successful return.
Tim Petersbf384c22003-04-06 00:11:39 +0000587 */
Tim Peters259272b2003-04-06 19:41:39 +0000588static int
Tim Petersf6b80452003-04-07 19:21:15 +0000589handle_finalizers(PyGC_Head *finalizers, PyGC_Head *old)
Jeremy Hyltonc5007aa2000-06-30 05:02:53 +0000590{
Tim Petersf6b80452003-04-07 19:21:15 +0000591 PyGC_Head *gc = finalizers->gc.gc_next;
592
Jeremy Hyltonc5007aa2000-06-30 05:02:53 +0000593 if (garbage == NULL) {
594 garbage = PyList_New(0);
Tim Petersbf384c22003-04-06 00:11:39 +0000595 if (garbage == NULL)
596 Py_FatalError("gc couldn't create gc.garbage list");
Jeremy Hyltonc5007aa2000-06-30 05:02:53 +0000597 }
Tim Petersf6b80452003-04-07 19:21:15 +0000598 for (; gc != finalizers; gc = gc->gc.gc_next) {
599 PyObject *op = FROM_GC(gc);
600
601 if ((debug & DEBUG_SAVEALL) || has_finalizer(op)) {
602 if (PyList_Append(garbage, op) < 0)
603 return -1;
604 }
Jeremy Hyltonc5007aa2000-06-30 05:02:53 +0000605 }
Tim Petersf6b80452003-04-07 19:21:15 +0000606
Tim Peters259272b2003-04-06 19:41:39 +0000607 gc_list_merge(finalizers, old);
608 return 0;
Jeremy Hyltonc5007aa2000-06-30 05:02:53 +0000609}
610
Neil Schemenauer544de1e2000-09-22 15:22:38 +0000611/* Break reference cycles by clearing the containers involved. This is
Jeremy Hyltonc5007aa2000-06-30 05:02:53 +0000612 * tricky business as the lists can be changing and we don't know which
Tim Peters19b74c72002-07-01 03:52:19 +0000613 * objects may be freed. It is possible I screwed something up here.
614 */
Jeremy Hyltonc5007aa2000-06-30 05:02:53 +0000615static void
Jeremy Hyltonce136e92003-04-04 19:59:06 +0000616delete_garbage(PyGC_Head *collectable, PyGC_Head *old)
Jeremy Hyltonc5007aa2000-06-30 05:02:53 +0000617{
618 inquiry clear;
619
Jeremy Hyltonce136e92003-04-04 19:59:06 +0000620 while (!gc_list_is_empty(collectable)) {
621 PyGC_Head *gc = collectable->gc.gc_next;
Neil Schemenauer43411b52001-08-30 00:05:51 +0000622 PyObject *op = FROM_GC(gc);
Tim Peters88396172002-06-30 17:56:40 +0000623
Tim Peters19b74c72002-07-01 03:52:19 +0000624 assert(IS_TENTATIVELY_UNREACHABLE(op));
Neil Schemenauer544de1e2000-09-22 15:22:38 +0000625 if (debug & DEBUG_SAVEALL) {
626 PyList_Append(garbage, op);
Jeremy Hyltonc5007aa2000-06-30 05:02:53 +0000627 }
Neil Schemenauer544de1e2000-09-22 15:22:38 +0000628 else {
629 if ((clear = op->ob_type->tp_clear) != NULL) {
630 Py_INCREF(op);
Jeremy Hylton8a135182002-06-06 23:23:55 +0000631 clear(op);
Neil Schemenauer544de1e2000-09-22 15:22:38 +0000632 Py_DECREF(op);
633 }
634 }
Jeremy Hyltonce136e92003-04-04 19:59:06 +0000635 if (collectable->gc.gc_next == gc) {
Neil Schemenauer544de1e2000-09-22 15:22:38 +0000636 /* object is still alive, move it, it may die later */
Jeremy Hyltonc5007aa2000-06-30 05:02:53 +0000637 gc_list_remove(gc);
638 gc_list_append(gc, old);
Tim Peters19b74c72002-07-01 03:52:19 +0000639 gc->gc.gc_refs = GC_REACHABLE;
Jeremy Hyltonc5007aa2000-06-30 05:02:53 +0000640 }
641 }
642}
643
644/* This is the main function. Read this to understand how the
645 * collection process works. */
646static long
Neil Schemenauer2880ae52002-05-04 05:35:20 +0000647collect(int generation)
Jeremy Hyltonc5007aa2000-06-30 05:02:53 +0000648{
Neil Schemenauer2880ae52002-05-04 05:35:20 +0000649 int i;
Tim Peters19b74c72002-07-01 03:52:19 +0000650 long m = 0; /* # objects collected */
651 long n = 0; /* # unreachable objects that couldn't be collected */
Neil Schemenauer2880ae52002-05-04 05:35:20 +0000652 PyGC_Head *young; /* the generation we are examining */
653 PyGC_Head *old; /* next older generation */
Tim Peters403a2032003-11-20 21:21:46 +0000654 PyGC_Head unreachable; /* non-problematic unreachable trash */
655 PyGC_Head finalizers; /* objects with, & reachable from, __del__ */
656 PyGC_Head wr_callbacks; /* weakrefs with callbacks */
Jeremy Hyltonc5007aa2000-06-30 05:02:53 +0000657 PyGC_Head *gc;
Jeremy Hyltonc5007aa2000-06-30 05:02:53 +0000658
Tim Peters93ad66d2003-04-05 17:15:44 +0000659 if (delstr == NULL) {
660 delstr = PyString_InternFromString("__del__");
661 if (delstr == NULL)
662 Py_FatalError("gc couldn't allocate \"__del__\"");
663 }
664
Jeremy Hyltonc5007aa2000-06-30 05:02:53 +0000665 if (debug & DEBUG_STATS) {
Neil Schemenauer2880ae52002-05-04 05:35:20 +0000666 PySys_WriteStderr("gc: collecting generation %d...\n",
667 generation);
668 PySys_WriteStderr("gc: objects in each generation:");
669 for (i = 0; i < NUM_GENERATIONS; i++) {
670 PySys_WriteStderr(" %ld", gc_list_size(GEN_HEAD(i)));
671 }
672 PySys_WriteStderr("\n");
673 }
674
675 /* update collection and allocation counters */
676 if (generation+1 < NUM_GENERATIONS)
677 generations[generation+1].count += 1;
678 for (i = 0; i <= generation; i++)
Neil Schemenauerc9051642002-06-28 19:16:04 +0000679 generations[i].count = 0;
Neil Schemenauer2880ae52002-05-04 05:35:20 +0000680
681 /* merge younger generations with one we are currently collecting */
682 for (i = 0; i < generation; i++) {
683 gc_list_merge(GEN_HEAD(i), GEN_HEAD(generation));
684 }
685
686 /* handy references */
687 young = GEN_HEAD(generation);
Tim Peters19b74c72002-07-01 03:52:19 +0000688 if (generation < NUM_GENERATIONS-1)
Neil Schemenauer2880ae52002-05-04 05:35:20 +0000689 old = GEN_HEAD(generation+1);
Tim Peters19b74c72002-07-01 03:52:19 +0000690 else
691 old = young;
Jeremy Hyltonc5007aa2000-06-30 05:02:53 +0000692
693 /* Using ob_refcnt and gc_refs, calculate which objects in the
694 * container set are reachable from outside the set (ie. have a
695 * refcount greater than 0 when all the references within the
Tim Peters19b74c72002-07-01 03:52:19 +0000696 * set are taken into account
697 */
Jeremy Hyltonc5007aa2000-06-30 05:02:53 +0000698 update_refs(young);
699 subtract_refs(young);
700
Tim Peters19b74c72002-07-01 03:52:19 +0000701 /* Leave everything reachable from outside young in young, and move
702 * everything else (in young) to unreachable.
703 * NOTE: This used to move the reachable objects into a reachable
704 * set instead. But most things usually turn out to be reachable,
705 * so it's more efficient to move the unreachable things.
706 */
Jeremy Hyltonc5007aa2000-06-30 05:02:53 +0000707 gc_list_init(&unreachable);
Tim Peters19b74c72002-07-01 03:52:19 +0000708 move_unreachable(young, &unreachable);
Jeremy Hyltonc5007aa2000-06-30 05:02:53 +0000709
Tim Peters19b74c72002-07-01 03:52:19 +0000710 /* Move reachable objects to next generation. */
711 if (young != old)
712 gc_list_merge(young, old);
Jeremy Hyltonc5007aa2000-06-30 05:02:53 +0000713
Tim Peters19b74c72002-07-01 03:52:19 +0000714 /* All objects in unreachable are trash, but objects reachable from
715 * finalizers can't safely be deleted. Python programmers should take
716 * care not to create such things. For Python, finalizers means
Tim Peters403a2032003-11-20 21:21:46 +0000717 * instance objects with __del__ methods. Weakrefs with callbacks
718 * can call arbitrary Python code, so those are special-cased too.
Tim Peters93ad66d2003-04-05 17:15:44 +0000719 *
Tim Peters403a2032003-11-20 21:21:46 +0000720 * Move unreachable objects with finalizers, and weakrefs with
721 * callbacks, into different lists.
Tim Petersf6b80452003-04-07 19:21:15 +0000722 */
Jeremy Hyltonc5007aa2000-06-30 05:02:53 +0000723 gc_list_init(&finalizers);
Tim Peters403a2032003-11-20 21:21:46 +0000724 gc_list_init(&wr_callbacks);
725 move_troublemakers(&unreachable, &finalizers, &wr_callbacks);
726 /* Clear the trash weakrefs with callbacks. This prevents their
727 * callbacks from getting invoked (when a weakref goes away, so does
728 * its callback).
729 * We do this even if the weakrefs are reachable from finalizers.
730 * If we didn't, breaking cycles in unreachable later could trigger
731 * deallocation of objects in finalizers, which could in turn
732 * cause callbacks to trigger. This may not be ideal behavior.
733 */
734 clear_weakrefs(&wr_callbacks);
Tim Petersbf384c22003-04-06 00:11:39 +0000735 /* finalizers contains the unreachable objects with a finalizer;
Tim Peters403a2032003-11-20 21:21:46 +0000736 * unreachable objects reachable *from* those are also uncollectable,
737 * and we move those into the finalizers list too.
Tim Petersbf384c22003-04-06 00:11:39 +0000738 */
Tim Petersf6b80452003-04-07 19:21:15 +0000739 move_finalizer_reachable(&finalizers);
Jeremy Hyltonc5007aa2000-06-30 05:02:53 +0000740
741 /* Collect statistics on collectable objects found and print
Tim Peters403a2032003-11-20 21:21:46 +0000742 * debugging information.
743 */
Tim Petersf6b80452003-04-07 19:21:15 +0000744 for (gc = unreachable.gc.gc_next; gc != &unreachable;
Tim Peters9e4ca102001-10-11 18:31:31 +0000745 gc = gc->gc.gc_next) {
Jeremy Hyltonc5007aa2000-06-30 05:02:53 +0000746 m++;
Jeremy Hylton06257772000-08-31 15:10:24 +0000747 if (debug & DEBUG_COLLECTABLE) {
Neil Schemenauer43411b52001-08-30 00:05:51 +0000748 debug_cycle("collectable", FROM_GC(gc));
Jeremy Hyltonc5007aa2000-06-30 05:02:53 +0000749 }
750 }
Tim Petersfb2ab4d2003-04-07 22:41:24 +0000751 /* Call tp_clear on objects in the unreachable set. This will cause
752 * the reference cycles to be broken. It may also cause some objects
753 * in finalizers to be freed.
754 */
Tim Petersf6b80452003-04-07 19:21:15 +0000755 delete_garbage(&unreachable, old);
Jeremy Hyltonc5007aa2000-06-30 05:02:53 +0000756
Tim Peters403a2032003-11-20 21:21:46 +0000757 /* Now that we're done analyzing stuff and breaking cycles, let
758 * delayed weakref callbacks run.
759 */
760 m += release_weakrefs(&wr_callbacks, old);
761
Jeremy Hyltonc5007aa2000-06-30 05:02:53 +0000762 /* Collect statistics on uncollectable objects found and print
763 * debugging information. */
Tim Peters50c61d52003-04-06 01:50:50 +0000764 for (gc = finalizers.gc.gc_next;
Tim Petersbf384c22003-04-06 00:11:39 +0000765 gc != &finalizers;
766 gc = gc->gc.gc_next) {
Jeremy Hyltonc5007aa2000-06-30 05:02:53 +0000767 n++;
Tim Petersbf384c22003-04-06 00:11:39 +0000768 if (debug & DEBUG_UNCOLLECTABLE)
Neil Schemenauer43411b52001-08-30 00:05:51 +0000769 debug_cycle("uncollectable", FROM_GC(gc));
Tim Petersbf384c22003-04-06 00:11:39 +0000770 }
Jeremy Hylton06257772000-08-31 15:10:24 +0000771 if (debug & DEBUG_STATS) {
Jeremy Hyltonc5007aa2000-06-30 05:02:53 +0000772 if (m == 0 && n == 0) {
Jeremy Hylton06257772000-08-31 15:10:24 +0000773 PySys_WriteStderr("gc: done.\n");
Neil Schemenauer544de1e2000-09-22 15:22:38 +0000774 }
775 else {
Jeremy Hylton06257772000-08-31 15:10:24 +0000776 PySys_WriteStderr(
777 "gc: done, %ld unreachable, %ld uncollectable.\n",
778 n+m, n);
Jeremy Hyltonc5007aa2000-06-30 05:02:53 +0000779 }
780 }
781
782 /* Append instances in the uncollectable set to a Python
783 * reachable list of garbage. The programmer has to deal with
Tim Petersbf384c22003-04-06 00:11:39 +0000784 * this if they insist on creating this type of structure.
785 */
Tim Petersf6b80452003-04-07 19:21:15 +0000786 (void)handle_finalizers(&finalizers, old);
Jeremy Hyltonc5007aa2000-06-30 05:02:53 +0000787
Jeremy Hyltonb709df32000-09-01 02:47:25 +0000788 if (PyErr_Occurred()) {
Tim Petersf6b80452003-04-07 19:21:15 +0000789 if (gc_str == NULL)
Tim Petersfb2ab4d2003-04-07 22:41:24 +0000790 gc_str = PyString_FromString("garbage collection");
Jeremy Hyltonb709df32000-09-01 02:47:25 +0000791 PyErr_WriteUnraisable(gc_str);
792 Py_FatalError("unexpected exception during garbage collection");
793 }
Jeremy Hyltonc5007aa2000-06-30 05:02:53 +0000794 return n+m;
795}
796
797static long
798collect_generations(void)
799{
Neil Schemenauer2880ae52002-05-04 05:35:20 +0000800 int i;
Vladimir Marangozovb16714b2000-07-10 05:37:39 +0000801 long n = 0;
Jeremy Hyltonc5007aa2000-06-30 05:02:53 +0000802
Neil Schemenauer2880ae52002-05-04 05:35:20 +0000803 /* Find the oldest generation (higest numbered) where the count
804 * exceeds the threshold. Objects in the that generation and
805 * generations younger than it will be collected. */
806 for (i = NUM_GENERATIONS-1; i >= 0; i--) {
807 if (generations[i].count > generations[i].threshold) {
808 n = collect(i);
809 break;
Jeremy Hyltonc5007aa2000-06-30 05:02:53 +0000810 }
811 }
812 return n;
813}
814
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +0000815PyDoc_STRVAR(gc_enable__doc__,
Vladimir Marangozovf9d20c32000-08-06 22:45:31 +0000816"enable() -> None\n"
817"\n"
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +0000818"Enable automatic garbage collection.\n");
Jeremy Hyltonc5007aa2000-06-30 05:02:53 +0000819
Vladimir Marangozovf9d20c32000-08-06 22:45:31 +0000820static PyObject *
Tim Peters50c61d52003-04-06 01:50:50 +0000821gc_enable(PyObject *self, PyObject *noargs)
Vladimir Marangozovf9d20c32000-08-06 22:45:31 +0000822{
Vladimir Marangozovf9d20c32000-08-06 22:45:31 +0000823 enabled = 1;
Vladimir Marangozovf9d20c32000-08-06 22:45:31 +0000824 Py_INCREF(Py_None);
825 return Py_None;
826}
827
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +0000828PyDoc_STRVAR(gc_disable__doc__,
Vladimir Marangozovf9d20c32000-08-06 22:45:31 +0000829"disable() -> None\n"
830"\n"
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +0000831"Disable automatic garbage collection.\n");
Vladimir Marangozovf9d20c32000-08-06 22:45:31 +0000832
833static PyObject *
Tim Peters50c61d52003-04-06 01:50:50 +0000834gc_disable(PyObject *self, PyObject *noargs)
Vladimir Marangozovf9d20c32000-08-06 22:45:31 +0000835{
Vladimir Marangozovf9d20c32000-08-06 22:45:31 +0000836 enabled = 0;
Vladimir Marangozovf9d20c32000-08-06 22:45:31 +0000837 Py_INCREF(Py_None);
838 return Py_None;
839}
840
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +0000841PyDoc_STRVAR(gc_isenabled__doc__,
Vladimir Marangozovf9d20c32000-08-06 22:45:31 +0000842"isenabled() -> status\n"
843"\n"
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +0000844"Returns true if automatic garbage collection is enabled.\n");
Vladimir Marangozovf9d20c32000-08-06 22:45:31 +0000845
846static PyObject *
Tim Peters50c61d52003-04-06 01:50:50 +0000847gc_isenabled(PyObject *self, PyObject *noargs)
Vladimir Marangozovf9d20c32000-08-06 22:45:31 +0000848{
Vladimir Marangozovf9d20c32000-08-06 22:45:31 +0000849 return Py_BuildValue("i", enabled);
850}
851
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +0000852PyDoc_STRVAR(gc_collect__doc__,
Jeremy Hyltonc5007aa2000-06-30 05:02:53 +0000853"collect() -> n\n"
854"\n"
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +0000855"Run a full collection. The number of unreachable objects is returned.\n");
Jeremy Hyltonc5007aa2000-06-30 05:02:53 +0000856
857static PyObject *
Tim Peters50c61d52003-04-06 01:50:50 +0000858gc_collect(PyObject *self, PyObject *noargs)
Jeremy Hyltonc5007aa2000-06-30 05:02:53 +0000859{
860 long n;
861
Tim Peters50c61d52003-04-06 01:50:50 +0000862 if (collecting)
Neil Schemenauere8c40cb2001-10-31 23:09:35 +0000863 n = 0; /* already collecting, don't do anything */
Neil Schemenauere8c40cb2001-10-31 23:09:35 +0000864 else {
865 collecting = 1;
Neil Schemenauer2880ae52002-05-04 05:35:20 +0000866 n = collect(NUM_GENERATIONS - 1);
Neil Schemenauere8c40cb2001-10-31 23:09:35 +0000867 collecting = 0;
868 }
Jeremy Hyltonc5007aa2000-06-30 05:02:53 +0000869
Neil Schemenauer7760cff2000-09-22 22:35:36 +0000870 return Py_BuildValue("l", n);
Jeremy Hyltonc5007aa2000-06-30 05:02:53 +0000871}
872
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +0000873PyDoc_STRVAR(gc_set_debug__doc__,
Jeremy Hyltonc5007aa2000-06-30 05:02:53 +0000874"set_debug(flags) -> None\n"
875"\n"
876"Set the garbage collection debugging flags. Debugging information is\n"
877"written to sys.stderr.\n"
878"\n"
879"flags is an integer and can have the following bits turned on:\n"
880"\n"
881" DEBUG_STATS - Print statistics during collection.\n"
882" DEBUG_COLLECTABLE - Print collectable objects found.\n"
883" DEBUG_UNCOLLECTABLE - Print unreachable but uncollectable objects found.\n"
884" DEBUG_INSTANCES - Print instance objects.\n"
885" DEBUG_OBJECTS - Print objects other than instances.\n"
Neil Schemenauer544de1e2000-09-22 15:22:38 +0000886" DEBUG_SAVEALL - Save objects to gc.garbage rather than freeing them.\n"
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +0000887" DEBUG_LEAK - Debug leaking programs (everything but STATS).\n");
Jeremy Hyltonc5007aa2000-06-30 05:02:53 +0000888
889static PyObject *
Vladimir Marangozovf9d20c32000-08-06 22:45:31 +0000890gc_set_debug(PyObject *self, PyObject *args)
Jeremy Hyltonc5007aa2000-06-30 05:02:53 +0000891{
Neil Schemenauer7760cff2000-09-22 22:35:36 +0000892 if (!PyArg_ParseTuple(args, "i:set_debug", &debug))
Jeremy Hyltonc5007aa2000-06-30 05:02:53 +0000893 return NULL;
894
895 Py_INCREF(Py_None);
896 return Py_None;
897}
898
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +0000899PyDoc_STRVAR(gc_get_debug__doc__,
Jeremy Hyltonc5007aa2000-06-30 05:02:53 +0000900"get_debug() -> flags\n"
901"\n"
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +0000902"Get the garbage collection debugging flags.\n");
Jeremy Hyltonc5007aa2000-06-30 05:02:53 +0000903
904static PyObject *
Tim Peters50c61d52003-04-06 01:50:50 +0000905gc_get_debug(PyObject *self, PyObject *noargs)
Jeremy Hyltonc5007aa2000-06-30 05:02:53 +0000906{
Jeremy Hyltonc5007aa2000-06-30 05:02:53 +0000907 return Py_BuildValue("i", debug);
908}
909
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +0000910PyDoc_STRVAR(gc_set_thresh__doc__,
Neal Norwitz2a47c0f2002-01-29 00:53:41 +0000911"set_threshold(threshold0, [threshold1, threshold2]) -> None\n"
Jeremy Hyltonc5007aa2000-06-30 05:02:53 +0000912"\n"
913"Sets the collection thresholds. Setting threshold0 to zero disables\n"
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +0000914"collection.\n");
Jeremy Hyltonc5007aa2000-06-30 05:02:53 +0000915
916static PyObject *
Vladimir Marangozovf9d20c32000-08-06 22:45:31 +0000917gc_set_thresh(PyObject *self, PyObject *args)
Jeremy Hyltonc5007aa2000-06-30 05:02:53 +0000918{
Neil Schemenauer2880ae52002-05-04 05:35:20 +0000919 int i;
920 if (!PyArg_ParseTuple(args, "i|ii:set_threshold",
921 &generations[0].threshold,
922 &generations[1].threshold,
923 &generations[2].threshold))
Jeremy Hyltonc5007aa2000-06-30 05:02:53 +0000924 return NULL;
Neil Schemenauer2880ae52002-05-04 05:35:20 +0000925 for (i = 2; i < NUM_GENERATIONS; i++) {
926 /* generations higher than 2 get the same threshold */
927 generations[i].threshold = generations[2].threshold;
928 }
Jeremy Hyltonc5007aa2000-06-30 05:02:53 +0000929
930 Py_INCREF(Py_None);
931 return Py_None;
932}
933
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +0000934PyDoc_STRVAR(gc_get_thresh__doc__,
Jeremy Hyltonc5007aa2000-06-30 05:02:53 +0000935"get_threshold() -> (threshold0, threshold1, threshold2)\n"
936"\n"
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +0000937"Return the current collection thresholds\n");
Jeremy Hyltonc5007aa2000-06-30 05:02:53 +0000938
939static PyObject *
Tim Peters50c61d52003-04-06 01:50:50 +0000940gc_get_thresh(PyObject *self, PyObject *noargs)
Jeremy Hyltonc5007aa2000-06-30 05:02:53 +0000941{
Neil Schemenauer2880ae52002-05-04 05:35:20 +0000942 return Py_BuildValue("(iii)",
943 generations[0].threshold,
944 generations[1].threshold,
945 generations[2].threshold);
Jeremy Hyltonc5007aa2000-06-30 05:02:53 +0000946}
947
Neil Schemenauer48c70342001-08-09 15:38:31 +0000948static int
Martin v. Löwis560da622001-11-24 09:24:51 +0000949referrersvisit(PyObject* obj, PyObject *objs)
Neil Schemenauer48c70342001-08-09 15:38:31 +0000950{
Martin v. Löwisc8fe77b2001-11-29 18:08:31 +0000951 int i;
952 for (i = 0; i < PyTuple_GET_SIZE(objs); i++)
953 if (PyTuple_GET_ITEM(objs, i) == obj)
954 return 1;
Neil Schemenauer48c70342001-08-09 15:38:31 +0000955 return 0;
956}
957
Neil Schemenauer17e7be62001-08-10 14:46:47 +0000958static int
Martin v. Löwis560da622001-11-24 09:24:51 +0000959gc_referrers_for(PyObject *objs, PyGC_Head *list, PyObject *resultlist)
Neil Schemenauer48c70342001-08-09 15:38:31 +0000960{
961 PyGC_Head *gc;
962 PyObject *obj;
963 traverseproc traverse;
Tim Peters9e4ca102001-10-11 18:31:31 +0000964 for (gc = list->gc.gc_next; gc != list; gc = gc->gc.gc_next) {
Neil Schemenauer43411b52001-08-30 00:05:51 +0000965 obj = FROM_GC(gc);
Neil Schemenauer48c70342001-08-09 15:38:31 +0000966 traverse = obj->ob_type->tp_traverse;
967 if (obj == objs || obj == resultlist)
968 continue;
Martin v. Löwis560da622001-11-24 09:24:51 +0000969 if (traverse(obj, (visitproc)referrersvisit, objs)) {
Neil Schemenauer17e7be62001-08-10 14:46:47 +0000970 if (PyList_Append(resultlist, obj) < 0)
971 return 0; /* error */
Neil Schemenauer48c70342001-08-09 15:38:31 +0000972 }
973 }
Neil Schemenauer17e7be62001-08-10 14:46:47 +0000974 return 1; /* no error */
Neil Schemenauer48c70342001-08-09 15:38:31 +0000975}
976
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +0000977PyDoc_STRVAR(gc_get_referrers__doc__,
Martin v. Löwis560da622001-11-24 09:24:51 +0000978"get_referrers(*objs) -> list\n\
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +0000979Return the list of objects that directly refer to any of objs.");
Neil Schemenauer48c70342001-08-09 15:38:31 +0000980
Neil Schemenauer17e7be62001-08-10 14:46:47 +0000981static PyObject *
Martin v. Löwis560da622001-11-24 09:24:51 +0000982gc_get_referrers(PyObject *self, PyObject *args)
Neil Schemenauer48c70342001-08-09 15:38:31 +0000983{
Neil Schemenauer2880ae52002-05-04 05:35:20 +0000984 int i;
Neil Schemenauer48c70342001-08-09 15:38:31 +0000985 PyObject *result = PyList_New(0);
Neil Schemenauer2880ae52002-05-04 05:35:20 +0000986 for (i = 0; i < NUM_GENERATIONS; i++) {
987 if (!(gc_referrers_for(args, GEN_HEAD(i), result))) {
988 Py_DECREF(result);
989 return NULL;
990 }
Neil Schemenauer17e7be62001-08-10 14:46:47 +0000991 }
Neil Schemenauer48c70342001-08-09 15:38:31 +0000992 return result;
993}
Jeremy Hyltonc5007aa2000-06-30 05:02:53 +0000994
Tim Peters0f81ab62003-04-08 16:39:48 +0000995/* Append obj to list; return true if error (out of memory), false if OK. */
Jeremy Hylton5bd378b2003-04-03 16:28:38 +0000996static int
Tim Peters730f5532003-04-08 17:17:17 +0000997referentsvisit(PyObject *obj, PyObject *list)
Jeremy Hylton5bd378b2003-04-03 16:28:38 +0000998{
Tim Peters0f81ab62003-04-08 16:39:48 +0000999 return PyList_Append(list, obj) < 0;
Jeremy Hylton5bd378b2003-04-03 16:28:38 +00001000}
1001
Tim Peters730f5532003-04-08 17:17:17 +00001002PyDoc_STRVAR(gc_get_referents__doc__,
1003"get_referents(*objs) -> list\n\
Jeremy Hylton059b0942003-04-03 16:29:13 +00001004Return the list of objects that are directly referred to by objs.");
Jeremy Hylton5bd378b2003-04-03 16:28:38 +00001005
1006static PyObject *
Tim Peters730f5532003-04-08 17:17:17 +00001007gc_get_referents(PyObject *self, PyObject *args)
Jeremy Hylton5bd378b2003-04-03 16:28:38 +00001008{
1009 int i;
1010 PyObject *result = PyList_New(0);
Tim Peters0f81ab62003-04-08 16:39:48 +00001011
1012 if (result == NULL)
1013 return NULL;
1014
Jeremy Hylton5bd378b2003-04-03 16:28:38 +00001015 for (i = 0; i < PyTuple_GET_SIZE(args); i++) {
Tim Peters0f81ab62003-04-08 16:39:48 +00001016 traverseproc traverse;
Tim Peters93ad66d2003-04-05 17:15:44 +00001017 PyObject *obj = PyTuple_GET_ITEM(args, i);
Tim Peters0f81ab62003-04-08 16:39:48 +00001018
1019 if (! PyObject_IS_GC(obj))
Jeremy Hylton5bd378b2003-04-03 16:28:38 +00001020 continue;
Tim Peters0f81ab62003-04-08 16:39:48 +00001021 traverse = obj->ob_type->tp_traverse;
1022 if (! traverse)
1023 continue;
Tim Peters730f5532003-04-08 17:17:17 +00001024 if (traverse(obj, (visitproc)referentsvisit, result)) {
Tim Peters0f81ab62003-04-08 16:39:48 +00001025 Py_DECREF(result);
Jeremy Hylton5bd378b2003-04-03 16:28:38 +00001026 return NULL;
Tim Peters0f81ab62003-04-08 16:39:48 +00001027 }
Jeremy Hylton5bd378b2003-04-03 16:28:38 +00001028 }
1029 return result;
1030}
1031
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00001032PyDoc_STRVAR(gc_get_objects__doc__,
Neil Schemenauerc7c8d8e2001-08-09 15:58:59 +00001033"get_objects() -> [...]\n"
1034"\n"
1035"Return a list of objects tracked by the collector (excluding the list\n"
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00001036"returned).\n");
Neil Schemenauerc7c8d8e2001-08-09 15:58:59 +00001037
Neil Schemenauerc7c8d8e2001-08-09 15:58:59 +00001038static PyObject *
Tim Peters50c61d52003-04-06 01:50:50 +00001039gc_get_objects(PyObject *self, PyObject *noargs)
Neil Schemenauerc7c8d8e2001-08-09 15:58:59 +00001040{
Neil Schemenauer2880ae52002-05-04 05:35:20 +00001041 int i;
Neil Schemenauerc7c8d8e2001-08-09 15:58:59 +00001042 PyObject* result;
1043
Neil Schemenauerc7c8d8e2001-08-09 15:58:59 +00001044 result = PyList_New(0);
Tim Peters50c61d52003-04-06 01:50:50 +00001045 if (result == NULL)
Martin v. Löwisf8a6f242001-12-02 18:31:02 +00001046 return NULL;
Neil Schemenauer2880ae52002-05-04 05:35:20 +00001047 for (i = 0; i < NUM_GENERATIONS; i++) {
1048 if (append_objects(result, GEN_HEAD(i))) {
1049 Py_DECREF(result);
1050 return NULL;
1051 }
Martin v. Löwis155aad12001-12-02 12:21:34 +00001052 }
Neil Schemenauerc7c8d8e2001-08-09 15:58:59 +00001053 return result;
1054}
1055
1056
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00001057PyDoc_STRVAR(gc__doc__,
Jeremy Hyltonc5007aa2000-06-30 05:02:53 +00001058"This module provides access to the garbage collector for reference cycles.\n"
1059"\n"
Vladimir Marangozovf9d20c32000-08-06 22:45:31 +00001060"enable() -- Enable automatic garbage collection.\n"
1061"disable() -- Disable automatic garbage collection.\n"
1062"isenabled() -- Returns true if automatic collection is enabled.\n"
Jeremy Hyltonc5007aa2000-06-30 05:02:53 +00001063"collect() -- Do a full collection right now.\n"
1064"set_debug() -- Set debugging flags.\n"
1065"get_debug() -- Get debugging flags.\n"
1066"set_threshold() -- Set the collection thresholds.\n"
1067"get_threshold() -- Return the current the collection thresholds.\n"
Neil Schemenauerc7c8d8e2001-08-09 15:58:59 +00001068"get_objects() -- Return a list of all objects tracked by the collector.\n"
Jeremy Hylton5bd378b2003-04-03 16:28:38 +00001069"get_referrers() -- Return the list of objects that refer to an object.\n"
Tim Peters730f5532003-04-08 17:17:17 +00001070"get_referents() -- Return the list of objects that an object refers to.\n");
Jeremy Hyltonc5007aa2000-06-30 05:02:53 +00001071
1072static PyMethodDef GcMethods[] = {
Tim Peters50c61d52003-04-06 01:50:50 +00001073 {"enable", gc_enable, METH_NOARGS, gc_enable__doc__},
1074 {"disable", gc_disable, METH_NOARGS, gc_disable__doc__},
1075 {"isenabled", gc_isenabled, METH_NOARGS, gc_isenabled__doc__},
Vladimir Marangozovf9d20c32000-08-06 22:45:31 +00001076 {"set_debug", gc_set_debug, METH_VARARGS, gc_set_debug__doc__},
Tim Peters50c61d52003-04-06 01:50:50 +00001077 {"get_debug", gc_get_debug, METH_NOARGS, gc_get_debug__doc__},
Vladimir Marangozovf9d20c32000-08-06 22:45:31 +00001078 {"set_threshold", gc_set_thresh, METH_VARARGS, gc_set_thresh__doc__},
Tim Peters50c61d52003-04-06 01:50:50 +00001079 {"get_threshold", gc_get_thresh, METH_NOARGS, gc_get_thresh__doc__},
1080 {"collect", gc_collect, METH_NOARGS, gc_collect__doc__},
1081 {"get_objects", gc_get_objects,METH_NOARGS, gc_get_objects__doc__},
Martin v. Löwis560da622001-11-24 09:24:51 +00001082 {"get_referrers", gc_get_referrers, METH_VARARGS,
1083 gc_get_referrers__doc__},
Tim Peters730f5532003-04-08 17:17:17 +00001084 {"get_referents", gc_get_referents, METH_VARARGS,
1085 gc_get_referents__doc__},
Jeremy Hyltonc5007aa2000-06-30 05:02:53 +00001086 {NULL, NULL} /* Sentinel */
1087};
1088
Jason Tishler6bc06ec2003-09-04 11:59:50 +00001089PyMODINIT_FUNC
Jeremy Hyltonc5007aa2000-06-30 05:02:53 +00001090initgc(void)
1091{
1092 PyObject *m;
Jeremy Hyltonc5007aa2000-06-30 05:02:53 +00001093
1094 m = Py_InitModule4("gc",
1095 GcMethods,
1096 gc__doc__,
1097 NULL,
1098 PYTHON_API_VERSION);
Tim Peters11558872003-04-06 23:30:52 +00001099
Jeremy Hyltonc5007aa2000-06-30 05:02:53 +00001100 if (garbage == NULL) {
1101 garbage = PyList_New(0);
Tim Peters11558872003-04-06 23:30:52 +00001102 if (garbage == NULL)
1103 return;
Jeremy Hyltonc5007aa2000-06-30 05:02:53 +00001104 }
Tim Peters11558872003-04-06 23:30:52 +00001105 if (PyModule_AddObject(m, "garbage", garbage) < 0)
1106 return;
1107#define ADD_INT(NAME) if (PyModule_AddIntConstant(m, #NAME, NAME) < 0) return
1108 ADD_INT(DEBUG_STATS);
1109 ADD_INT(DEBUG_COLLECTABLE);
1110 ADD_INT(DEBUG_UNCOLLECTABLE);
1111 ADD_INT(DEBUG_INSTANCES);
1112 ADD_INT(DEBUG_OBJECTS);
1113 ADD_INT(DEBUG_SAVEALL);
1114 ADD_INT(DEBUG_LEAK);
1115#undef ADD_INT
Jeremy Hyltonc5007aa2000-06-30 05:02:53 +00001116}
1117
Guido van Rossume13ddc92003-04-17 17:29:22 +00001118/* API to invoke gc.collect() from C */
1119long
1120PyGC_Collect(void)
1121{
1122 long n;
1123
1124 if (collecting)
1125 n = 0; /* already collecting, don't do anything */
1126 else {
1127 collecting = 1;
1128 n = collect(NUM_GENERATIONS - 1);
1129 collecting = 0;
1130 }
1131
1132 return n;
1133}
1134
Neil Schemenauer43411b52001-08-30 00:05:51 +00001135/* for debugging */
Guido van Rossume13ddc92003-04-17 17:29:22 +00001136void
1137_PyGC_Dump(PyGC_Head *g)
Neil Schemenauer43411b52001-08-30 00:05:51 +00001138{
1139 _PyObject_Dump(FROM_GC(g));
1140}
1141
Neil Schemenauer43411b52001-08-30 00:05:51 +00001142/* extension modules might be compiled with GC support so these
1143 functions must always be available */
1144
Neil Schemenauerfec4eb12002-04-12 02:41:03 +00001145#undef PyObject_GC_Track
1146#undef PyObject_GC_UnTrack
1147#undef PyObject_GC_Del
1148#undef _PyObject_GC_Malloc
1149
Neil Schemenauer43411b52001-08-30 00:05:51 +00001150void
Neil Schemenauerfec4eb12002-04-12 02:41:03 +00001151PyObject_GC_Track(void *op)
Neil Schemenauer43411b52001-08-30 00:05:51 +00001152{
1153 _PyObject_GC_TRACK(op);
1154}
1155
Neil Schemenauerfec4eb12002-04-12 02:41:03 +00001156/* for binary compatibility with 2.2 */
Neil Schemenauer43411b52001-08-30 00:05:51 +00001157void
Neil Schemenauerfec4eb12002-04-12 02:41:03 +00001158_PyObject_GC_Track(PyObject *op)
1159{
1160 PyObject_GC_Track(op);
1161}
1162
1163void
1164PyObject_GC_UnTrack(void *op)
Neil Schemenauer43411b52001-08-30 00:05:51 +00001165{
Tim Peters803526b2002-07-07 05:13:56 +00001166 /* Obscure: the Py_TRASHCAN mechanism requires that we be able to
1167 * call PyObject_GC_UnTrack twice on an object.
1168 */
Neil Schemenauera2b11ec2002-05-21 15:53:24 +00001169 if (IS_TRACKED(op))
Guido van Rossumff413af2002-03-28 20:34:59 +00001170 _PyObject_GC_UNTRACK(op);
Neil Schemenauer43411b52001-08-30 00:05:51 +00001171}
1172
Neil Schemenauerfec4eb12002-04-12 02:41:03 +00001173/* for binary compatibility with 2.2 */
1174void
1175_PyObject_GC_UnTrack(PyObject *op)
1176{
1177 PyObject_GC_UnTrack(op);
1178}
1179
Neil Schemenauer43411b52001-08-30 00:05:51 +00001180PyObject *
Neil Schemenauerfec4eb12002-04-12 02:41:03 +00001181_PyObject_GC_Malloc(size_t basicsize)
Neil Schemenauer43411b52001-08-30 00:05:51 +00001182{
1183 PyObject *op;
Neil Schemenauerfec4eb12002-04-12 02:41:03 +00001184 PyGC_Head *g = PyObject_MALLOC(sizeof(PyGC_Head) + basicsize);
Neil Schemenauer43411b52001-08-30 00:05:51 +00001185 if (g == NULL)
Jeremy Hylton8a135182002-06-06 23:23:55 +00001186 return PyErr_NoMemory();
Tim Petersea405632002-07-02 00:52:30 +00001187 g->gc.gc_refs = GC_UNTRACKED;
Neil Schemenauer2880ae52002-05-04 05:35:20 +00001188 generations[0].count++; /* number of allocated GC objects */
1189 if (generations[0].count > generations[0].threshold &&
Neil Schemenauer43411b52001-08-30 00:05:51 +00001190 enabled &&
Neil Schemenauer2880ae52002-05-04 05:35:20 +00001191 generations[0].threshold &&
Neil Schemenauer43411b52001-08-30 00:05:51 +00001192 !collecting &&
1193 !PyErr_Occurred()) {
1194 collecting = 1;
Neil Schemenauer2880ae52002-05-04 05:35:20 +00001195 collect_generations();
Neil Schemenauer43411b52001-08-30 00:05:51 +00001196 collecting = 0;
1197 }
1198 op = FROM_GC(g);
Neil Schemenauer43411b52001-08-30 00:05:51 +00001199 return op;
1200}
1201
1202PyObject *
1203_PyObject_GC_New(PyTypeObject *tp)
1204{
Neil Schemenauerfec4eb12002-04-12 02:41:03 +00001205 PyObject *op = _PyObject_GC_Malloc(_PyObject_SIZE(tp));
Tim Petersfa8efab2002-04-28 01:57:25 +00001206 if (op != NULL)
1207 op = PyObject_INIT(op, tp);
1208 return op;
Neil Schemenauer43411b52001-08-30 00:05:51 +00001209}
1210
1211PyVarObject *
Tim Peters6d483d32001-10-06 21:27:34 +00001212_PyObject_GC_NewVar(PyTypeObject *tp, int nitems)
Neil Schemenauer43411b52001-08-30 00:05:51 +00001213{
Neil Schemenauerfec4eb12002-04-12 02:41:03 +00001214 const size_t size = _PyObject_VAR_SIZE(tp, nitems);
1215 PyVarObject *op = (PyVarObject *) _PyObject_GC_Malloc(size);
Tim Petersfa8efab2002-04-28 01:57:25 +00001216 if (op != NULL)
1217 op = PyObject_INIT_VAR(op, tp, nitems);
1218 return op;
Neil Schemenauer43411b52001-08-30 00:05:51 +00001219}
1220
1221PyVarObject *
Tim Peters6d483d32001-10-06 21:27:34 +00001222_PyObject_GC_Resize(PyVarObject *op, int nitems)
Neil Schemenauer43411b52001-08-30 00:05:51 +00001223{
Tim Petersf2a67da2001-10-07 03:54:51 +00001224 const size_t basicsize = _PyObject_VAR_SIZE(op->ob_type, nitems);
Neil Schemenauer43411b52001-08-30 00:05:51 +00001225 PyGC_Head *g = AS_GC(op);
Neil Schemenauerfec4eb12002-04-12 02:41:03 +00001226 g = PyObject_REALLOC(g, sizeof(PyGC_Head) + basicsize);
Neil Schemenauer43411b52001-08-30 00:05:51 +00001227 if (g == NULL)
1228 return (PyVarObject *)PyErr_NoMemory();
1229 op = (PyVarObject *) FROM_GC(g);
Tim Peters6d483d32001-10-06 21:27:34 +00001230 op->ob_size = nitems;
Neil Schemenauer43411b52001-08-30 00:05:51 +00001231 return op;
1232}
1233
1234void
Neil Schemenauerfec4eb12002-04-12 02:41:03 +00001235PyObject_GC_Del(void *op)
Neil Schemenauer43411b52001-08-30 00:05:51 +00001236{
Neil Schemenauer43411b52001-08-30 00:05:51 +00001237 PyGC_Head *g = AS_GC(op);
Neil Schemenauera2b11ec2002-05-21 15:53:24 +00001238 if (IS_TRACKED(op))
Neil Schemenauer43411b52001-08-30 00:05:51 +00001239 gc_list_remove(g);
Neil Schemenauer2880ae52002-05-04 05:35:20 +00001240 if (generations[0].count > 0) {
1241 generations[0].count--;
Neil Schemenauer43411b52001-08-30 00:05:51 +00001242 }
Neil Schemenauerfec4eb12002-04-12 02:41:03 +00001243 PyObject_FREE(g);
Neil Schemenauer43411b52001-08-30 00:05:51 +00001244}
1245
Neil Schemenauerfec4eb12002-04-12 02:41:03 +00001246/* for binary compatibility with 2.2 */
1247#undef _PyObject_GC_Del
1248void
1249_PyObject_GC_Del(PyObject *op)
1250{
1251 PyObject_GC_Del(op);
1252}