blob: 4f8c85ffd9785aef380eecf5916fb0c5be1d4e3e [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"
Christian Heimes3b718a72008-02-14 12:47:33 +000022#include "frameobject.h" /* for PyFrame_ClearFreeList */
Jeremy Hyltonc5007aa2000-06-30 05:02:53 +000023
Neil Schemenauer43411b52001-08-30 00:05:51 +000024/* Get an object's GC head */
25#define AS_GC(o) ((PyGC_Head *)(o)-1)
26
27/* Get the object given the GC head */
28#define FROM_GC(g) ((PyObject *)(((PyGC_Head *)g)+1))
29
Jeremy Hyltonc5007aa2000-06-30 05:02:53 +000030/*** Global GC state ***/
31
Neil Schemenauer2880ae52002-05-04 05:35:20 +000032struct gc_generation {
33 PyGC_Head head;
34 int threshold; /* collection threshold */
35 int count; /* count of allocations or collections of younger
36 generations */
37};
38
39#define NUM_GENERATIONS 3
40#define GEN_HEAD(n) (&generations[n].head)
41
Jeremy Hyltonc5007aa2000-06-30 05:02:53 +000042/* linked lists of container objects */
Neil Schemenauer2880ae52002-05-04 05:35:20 +000043static struct gc_generation generations[NUM_GENERATIONS] = {
44 /* PyGC_Head, threshold, count */
45 {{{GEN_HEAD(0), GEN_HEAD(0), 0}}, 700, 0},
46 {{{GEN_HEAD(1), GEN_HEAD(1), 0}}, 10, 0},
47 {{{GEN_HEAD(2), GEN_HEAD(2), 0}}, 10, 0},
48};
Jeremy Hyltonc5007aa2000-06-30 05:02:53 +000049
Neil Schemenauer2880ae52002-05-04 05:35:20 +000050PyGC_Head *_PyGC_generation0 = GEN_HEAD(0);
51
Vladimir Marangozovf9d20c32000-08-06 22:45:31 +000052static int enabled = 1; /* automatic collection enabled? */
Jeremy Hyltonc5007aa2000-06-30 05:02:53 +000053
Neil Schemenauer43411b52001-08-30 00:05:51 +000054/* true if we are currently running the collector */
Tim Petersbf384c22003-04-06 00:11:39 +000055static int collecting = 0;
Neil Schemenauer43411b52001-08-30 00:05:51 +000056
Tim Peters6fc13d92002-07-02 18:12:35 +000057/* list of uncollectable objects */
Tim Petersbf384c22003-04-06 00:11:39 +000058static PyObject *garbage = NULL;
Tim Peters6fc13d92002-07-02 18:12:35 +000059
60/* Python string to use if unhandled exception occurs */
Tim Petersbf384c22003-04-06 00:11:39 +000061static PyObject *gc_str = NULL;
Tim Peters6fc13d92002-07-02 18:12:35 +000062
Tim Peters93ad66d2003-04-05 17:15:44 +000063/* Python string used to look for __del__ attribute. */
64static PyObject *delstr = NULL;
Jeremy Hyltonce136e92003-04-04 19:59:06 +000065
Jeremy Hyltonc5007aa2000-06-30 05:02:53 +000066/* set for debugging information */
67#define DEBUG_STATS (1<<0) /* print collection statistics */
68#define DEBUG_COLLECTABLE (1<<1) /* print collectable objects */
69#define DEBUG_UNCOLLECTABLE (1<<2) /* print uncollectable objects */
70#define DEBUG_INSTANCES (1<<3) /* print instances */
71#define DEBUG_OBJECTS (1<<4) /* print other objects */
Neil Schemenauer544de1e2000-09-22 15:22:38 +000072#define DEBUG_SAVEALL (1<<5) /* save all garbage in gc.garbage */
Jeremy Hyltonc5007aa2000-06-30 05:02:53 +000073#define DEBUG_LEAK DEBUG_COLLECTABLE | \
74 DEBUG_UNCOLLECTABLE | \
75 DEBUG_INSTANCES | \
Neil Schemenauer544de1e2000-09-22 15:22:38 +000076 DEBUG_OBJECTS | \
77 DEBUG_SAVEALL
Jeremy Hyltonb709df32000-09-01 02:47:25 +000078static int debug;
Neal Norwitz57a03612006-04-26 05:34:03 +000079static PyObject *tmod = NULL;
Jeremy Hyltonc5007aa2000-06-30 05:02:53 +000080
Tim Peters6fc13d92002-07-02 18:12:35 +000081/*--------------------------------------------------------------------------
82gc_refs values.
Neil Schemenauer43411b52001-08-30 00:05:51 +000083
Tim Peters6fc13d92002-07-02 18:12:35 +000084Between collections, every gc'ed object has one of two gc_refs values:
85
86GC_UNTRACKED
87 The initial state; objects returned by PyObject_GC_Malloc are in this
88 state. The object doesn't live in any generation list, and its
89 tp_traverse slot must not be called.
90
91GC_REACHABLE
92 The object lives in some generation list, and its tp_traverse is safe to
93 call. An object transitions to GC_REACHABLE when PyObject_GC_Track
94 is called.
95
96During a collection, gc_refs can temporarily take on other states:
97
98>= 0
99 At the start of a collection, update_refs() copies the true refcount
100 to gc_refs, for each object in the generation being collected.
101 subtract_refs() then adjusts gc_refs so that it equals the number of
102 times an object is referenced directly from outside the generation
103 being collected.
Martin v. Löwis774348c2002-11-09 19:54:06 +0000104 gc_refs remains >= 0 throughout these steps.
Tim Peters6fc13d92002-07-02 18:12:35 +0000105
106GC_TENTATIVELY_UNREACHABLE
107 move_unreachable() then moves objects not reachable (whether directly or
108 indirectly) from outside the generation into an "unreachable" set.
109 Objects that are found to be reachable have gc_refs set to GC_REACHABLE
110 again. Objects that are found to be unreachable have gc_refs set to
111 GC_TENTATIVELY_UNREACHABLE. It's "tentatively" because the pass doing
112 this can't be sure until it ends, and GC_TENTATIVELY_UNREACHABLE may
113 transition back to GC_REACHABLE.
114
115 Only objects with GC_TENTATIVELY_UNREACHABLE still set are candidates
116 for collection. If it's decided not to collect such an object (e.g.,
117 it has a __del__ method), its gc_refs is restored to GC_REACHABLE again.
118----------------------------------------------------------------------------
119*/
Tim Petersea405632002-07-02 00:52:30 +0000120#define GC_UNTRACKED _PyGC_REFS_UNTRACKED
121#define GC_REACHABLE _PyGC_REFS_REACHABLE
122#define GC_TENTATIVELY_UNREACHABLE _PyGC_REFS_TENTATIVELY_UNREACHABLE
Tim Peters19b74c72002-07-01 03:52:19 +0000123
Tim Peters6fc13d92002-07-02 18:12:35 +0000124#define IS_TRACKED(o) ((AS_GC(o))->gc.gc_refs != GC_UNTRACKED)
Tim Peters19b74c72002-07-01 03:52:19 +0000125#define IS_REACHABLE(o) ((AS_GC(o))->gc.gc_refs == GC_REACHABLE)
126#define IS_TENTATIVELY_UNREACHABLE(o) ( \
127 (AS_GC(o))->gc.gc_refs == GC_TENTATIVELY_UNREACHABLE)
Neil Schemenauera2b11ec2002-05-21 15:53:24 +0000128
Jeremy Hyltonc5007aa2000-06-30 05:02:53 +0000129/*** list functions ***/
130
131static void
132gc_list_init(PyGC_Head *list)
133{
Tim Peters9e4ca102001-10-11 18:31:31 +0000134 list->gc.gc_prev = list;
135 list->gc.gc_next = list;
Jeremy Hyltonc5007aa2000-06-30 05:02:53 +0000136}
137
Neil Schemenauer2880ae52002-05-04 05:35:20 +0000138static int
139gc_list_is_empty(PyGC_Head *list)
140{
141 return (list->gc.gc_next == list);
142}
143
Tim Peterse2d59182004-11-01 01:39:08 +0000144#if 0
145/* This became unused after gc_list_move() was introduced. */
146/* Append `node` to `list`. */
Jeremy Hyltonc5007aa2000-06-30 05:02:53 +0000147static void
148gc_list_append(PyGC_Head *node, PyGC_Head *list)
149{
Tim Peters9e4ca102001-10-11 18:31:31 +0000150 node->gc.gc_next = list;
151 node->gc.gc_prev = list->gc.gc_prev;
152 node->gc.gc_prev->gc.gc_next = node;
153 list->gc.gc_prev = node;
Jeremy Hyltonc5007aa2000-06-30 05:02:53 +0000154}
Tim Peterse2d59182004-11-01 01:39:08 +0000155#endif
Jeremy Hyltonc5007aa2000-06-30 05:02:53 +0000156
Tim Peterse2d59182004-11-01 01:39:08 +0000157/* Remove `node` from the gc list it's currently in. */
Jeremy Hyltonc5007aa2000-06-30 05:02:53 +0000158static void
159gc_list_remove(PyGC_Head *node)
160{
Tim Peters9e4ca102001-10-11 18:31:31 +0000161 node->gc.gc_prev->gc.gc_next = node->gc.gc_next;
162 node->gc.gc_next->gc.gc_prev = node->gc.gc_prev;
163 node->gc.gc_next = NULL; /* object is not currently tracked */
Jeremy Hyltonc5007aa2000-06-30 05:02:53 +0000164}
165
Tim Peterse2d59182004-11-01 01:39:08 +0000166/* Move `node` from the gc list it's currently in (which is not explicitly
167 * named here) to the end of `list`. This is semantically the same as
168 * gc_list_remove(node) followed by gc_list_append(node, list).
169 */
170static void
171gc_list_move(PyGC_Head *node, PyGC_Head *list)
172{
Tim Petersbc1d1b82004-11-01 16:39:57 +0000173 PyGC_Head *new_prev;
Tim Peterse2d59182004-11-01 01:39:08 +0000174 PyGC_Head *current_prev = node->gc.gc_prev;
175 PyGC_Head *current_next = node->gc.gc_next;
Tim Petersbc1d1b82004-11-01 16:39:57 +0000176 /* Unlink from current list. */
Tim Peterse2d59182004-11-01 01:39:08 +0000177 current_prev->gc.gc_next = current_next;
178 current_next->gc.gc_prev = current_prev;
Tim Petersbc1d1b82004-11-01 16:39:57 +0000179 /* Relink at end of new list. */
180 new_prev = node->gc.gc_prev = list->gc.gc_prev;
Tim Peterse2d59182004-11-01 01:39:08 +0000181 new_prev->gc.gc_next = list->gc.gc_prev = node;
Tim Petersbc1d1b82004-11-01 16:39:57 +0000182 node->gc.gc_next = list;
Tim Peterse2d59182004-11-01 01:39:08 +0000183}
184
185/* append list `from` onto list `to`; `from` becomes an empty list */
Jeremy Hyltonc5007aa2000-06-30 05:02:53 +0000186static void
187gc_list_merge(PyGC_Head *from, PyGC_Head *to)
188{
189 PyGC_Head *tail;
Tim Peterse2d59182004-11-01 01:39:08 +0000190 assert(from != to);
Neil Schemenauer2880ae52002-05-04 05:35:20 +0000191 if (!gc_list_is_empty(from)) {
Tim Peters9e4ca102001-10-11 18:31:31 +0000192 tail = to->gc.gc_prev;
193 tail->gc.gc_next = from->gc.gc_next;
194 tail->gc.gc_next->gc.gc_prev = tail;
195 to->gc.gc_prev = from->gc.gc_prev;
196 to->gc.gc_prev->gc.gc_next = to;
Jeremy Hyltonc5007aa2000-06-30 05:02:53 +0000197 }
198 gc_list_init(from);
199}
200
Neal Norwitz7b216c52006-03-04 20:01:53 +0000201static Py_ssize_t
Jeremy Hyltonc5007aa2000-06-30 05:02:53 +0000202gc_list_size(PyGC_Head *list)
203{
204 PyGC_Head *gc;
Neal Norwitz7b216c52006-03-04 20:01:53 +0000205 Py_ssize_t n = 0;
Tim Peters9e4ca102001-10-11 18:31:31 +0000206 for (gc = list->gc.gc_next; gc != list; gc = gc->gc.gc_next) {
Jeremy Hyltonc5007aa2000-06-30 05:02:53 +0000207 n++;
208 }
209 return n;
210}
211
Tim Peters259272b2003-04-06 19:41:39 +0000212/* Append objects in a GC list to a Python list.
213 * Return 0 if all OK, < 0 if error (out of memory for list).
214 */
215static int
216append_objects(PyObject *py_list, PyGC_Head *gc_list)
217{
218 PyGC_Head *gc;
219 for (gc = gc_list->gc.gc_next; gc != gc_list; gc = gc->gc.gc_next) {
220 PyObject *op = FROM_GC(gc);
221 if (op != py_list) {
222 if (PyList_Append(py_list, op)) {
223 return -1; /* exception */
224 }
225 }
226 }
227 return 0;
228}
229
Jeremy Hyltonc5007aa2000-06-30 05:02:53 +0000230/*** end of list stuff ***/
231
232
Tim Peters19b74c72002-07-01 03:52:19 +0000233/* Set all gc_refs = ob_refcnt. After this, gc_refs is > 0 for all objects
234 * in containers, and is GC_REACHABLE for all tracked gc objects not in
235 * containers.
Tim Peters88396172002-06-30 17:56:40 +0000236 */
Jeremy Hyltonc5007aa2000-06-30 05:02:53 +0000237static void
238update_refs(PyGC_Head *containers)
239{
Tim Peters9e4ca102001-10-11 18:31:31 +0000240 PyGC_Head *gc = containers->gc.gc_next;
Tim Petersea405632002-07-02 00:52:30 +0000241 for (; gc != containers; gc = gc->gc.gc_next) {
242 assert(gc->gc.gc_refs == GC_REACHABLE);
Christian Heimese93237d2007-12-19 02:37:44 +0000243 gc->gc.gc_refs = Py_REFCNT(FROM_GC(gc));
Tim Peters780c4972003-11-14 00:01:17 +0000244 /* Python's cyclic gc should never see an incoming refcount
245 * of 0: if something decref'ed to 0, it should have been
246 * deallocated immediately at that time.
247 * Possible cause (if the assert triggers): a tp_dealloc
248 * routine left a gc-aware object tracked during its teardown
249 * phase, and did something-- or allowed something to happen --
250 * that called back into Python. gc can trigger then, and may
251 * see the still-tracked dying object. Before this assert
252 * was added, such mistakes went on to allow gc to try to
253 * delete the object again. In a debug build, that caused
254 * a mysterious segfault, when _Py_ForgetReference tried
255 * to remove the object from the doubly-linked list of all
256 * objects a second time. In a release build, an actual
257 * double deallocation occurred, which leads to corruption
258 * of the allocator's internal bookkeeping pointers. That's
259 * so serious that maybe this should be a release-build
260 * check instead of an assert?
261 */
262 assert(gc->gc.gc_refs != 0);
Tim Petersea405632002-07-02 00:52:30 +0000263 }
Jeremy Hyltonc5007aa2000-06-30 05:02:53 +0000264}
265
Tim Peters19b74c72002-07-01 03:52:19 +0000266/* A traversal callback for subtract_refs. */
Jeremy Hyltonc5007aa2000-06-30 05:02:53 +0000267static int
268visit_decref(PyObject *op, void *data)
269{
Tim Peters93cd83e2002-06-30 21:31:03 +0000270 assert(op != NULL);
Tim Peters19b74c72002-07-01 03:52:19 +0000271 if (PyObject_IS_GC(op)) {
272 PyGC_Head *gc = AS_GC(op);
273 /* We're only interested in gc_refs for objects in the
274 * generation being collected, which can be recognized
275 * because only they have positive gc_refs.
276 */
Tim Petersaab713b2002-07-02 22:15:28 +0000277 assert(gc->gc.gc_refs != 0); /* else refcount was too small */
Tim Peters19b74c72002-07-01 03:52:19 +0000278 if (gc->gc.gc_refs > 0)
279 gc->gc.gc_refs--;
280 }
Jeremy Hyltonc5007aa2000-06-30 05:02:53 +0000281 return 0;
282}
283
Tim Peters19b74c72002-07-01 03:52:19 +0000284/* Subtract internal references from gc_refs. After this, gc_refs is >= 0
285 * for all objects in containers, and is GC_REACHABLE for all tracked gc
286 * objects not in containers. The ones with gc_refs > 0 are directly
287 * reachable from outside containers, and so can't be collected.
288 */
Jeremy Hyltonc5007aa2000-06-30 05:02:53 +0000289static void
290subtract_refs(PyGC_Head *containers)
291{
292 traverseproc traverse;
Tim Peters9e4ca102001-10-11 18:31:31 +0000293 PyGC_Head *gc = containers->gc.gc_next;
294 for (; gc != containers; gc=gc->gc.gc_next) {
Christian Heimese93237d2007-12-19 02:37:44 +0000295 traverse = Py_TYPE(FROM_GC(gc))->tp_traverse;
Neil Schemenauer43411b52001-08-30 00:05:51 +0000296 (void) traverse(FROM_GC(gc),
Jeremy Hyltonc5007aa2000-06-30 05:02:53 +0000297 (visitproc)visit_decref,
298 NULL);
299 }
300}
301
Tim Peters19b74c72002-07-01 03:52:19 +0000302/* A traversal callback for move_unreachable. */
Jeremy Hyltonc5007aa2000-06-30 05:02:53 +0000303static int
Tim Peters19b74c72002-07-01 03:52:19 +0000304visit_reachable(PyObject *op, PyGC_Head *reachable)
Jeremy Hyltonc5007aa2000-06-30 05:02:53 +0000305{
Tim Petersea405632002-07-02 00:52:30 +0000306 if (PyObject_IS_GC(op)) {
Tim Peters19b74c72002-07-01 03:52:19 +0000307 PyGC_Head *gc = AS_GC(op);
Martin v. Löwis6db0e002006-03-01 16:56:25 +0000308 const Py_ssize_t gc_refs = gc->gc.gc_refs;
Tim Peters19b74c72002-07-01 03:52:19 +0000309
310 if (gc_refs == 0) {
311 /* This is in move_unreachable's 'young' list, but
312 * the traversal hasn't yet gotten to it. All
313 * we need to do is tell move_unreachable that it's
314 * reachable.
315 */
316 gc->gc.gc_refs = 1;
Jeremy Hyltonc5007aa2000-06-30 05:02:53 +0000317 }
Tim Peters19b74c72002-07-01 03:52:19 +0000318 else if (gc_refs == GC_TENTATIVELY_UNREACHABLE) {
319 /* This had gc_refs = 0 when move_unreachable got
320 * to it, but turns out it's reachable after all.
321 * Move it back to move_unreachable's 'young' list,
322 * and move_unreachable will eventually get to it
323 * again.
324 */
Tim Peterse2d59182004-11-01 01:39:08 +0000325 gc_list_move(gc, reachable);
Tim Peters19b74c72002-07-01 03:52:19 +0000326 gc->gc.gc_refs = 1;
327 }
328 /* Else there's nothing to do.
329 * If gc_refs > 0, it must be in move_unreachable's 'young'
330 * list, and move_unreachable will eventually get to it.
331 * If gc_refs == GC_REACHABLE, it's either in some other
332 * generation so we don't care about it, or move_unreachable
Tim Peters6fc13d92002-07-02 18:12:35 +0000333 * already dealt with it.
Tim Petersea405632002-07-02 00:52:30 +0000334 * If gc_refs == GC_UNTRACKED, it must be ignored.
Tim Peters19b74c72002-07-01 03:52:19 +0000335 */
Tim Petersea405632002-07-02 00:52:30 +0000336 else {
337 assert(gc_refs > 0
338 || gc_refs == GC_REACHABLE
339 || gc_refs == GC_UNTRACKED);
340 }
Jeremy Hyltonc5007aa2000-06-30 05:02:53 +0000341 }
342 return 0;
343}
344
Tim Peters19b74c72002-07-01 03:52:19 +0000345/* Move the unreachable objects from young to unreachable. After this,
346 * all objects in young have gc_refs = GC_REACHABLE, and all objects in
347 * unreachable have gc_refs = GC_TENTATIVELY_UNREACHABLE. All tracked
348 * gc objects not in young or unreachable still have gc_refs = GC_REACHABLE.
349 * All objects in young after this are directly or indirectly reachable
350 * from outside the original young; and all objects in unreachable are
351 * not.
Tim Peters88396172002-06-30 17:56:40 +0000352 */
Jeremy Hyltonc5007aa2000-06-30 05:02:53 +0000353static void
Tim Peters19b74c72002-07-01 03:52:19 +0000354move_unreachable(PyGC_Head *young, PyGC_Head *unreachable)
Jeremy Hyltonc5007aa2000-06-30 05:02:53 +0000355{
Tim Peters19b74c72002-07-01 03:52:19 +0000356 PyGC_Head *gc = young->gc.gc_next;
357
358 /* Invariants: all objects "to the left" of us in young have gc_refs
359 * = GC_REACHABLE, and are indeed reachable (directly or indirectly)
360 * from outside the young list as it was at entry. All other objects
361 * from the original young "to the left" of us are in unreachable now,
362 * and have gc_refs = GC_TENTATIVELY_UNREACHABLE. All objects to the
363 * left of us in 'young' now have been scanned, and no objects here
364 * or to the right have been scanned yet.
365 */
366
367 while (gc != young) {
368 PyGC_Head *next;
369
Tim Peters6fc13d92002-07-02 18:12:35 +0000370 if (gc->gc.gc_refs) {
371 /* gc is definitely reachable from outside the
372 * original 'young'. Mark it as such, and traverse
373 * its pointers to find any other objects that may
374 * be directly reachable from it. Note that the
375 * call to tp_traverse may append objects to young,
376 * so we have to wait until it returns to determine
377 * the next object to visit.
378 */
379 PyObject *op = FROM_GC(gc);
Christian Heimese93237d2007-12-19 02:37:44 +0000380 traverseproc traverse = Py_TYPE(op)->tp_traverse;
Tim Peters6fc13d92002-07-02 18:12:35 +0000381 assert(gc->gc.gc_refs > 0);
382 gc->gc.gc_refs = GC_REACHABLE;
383 (void) traverse(op,
384 (visitproc)visit_reachable,
385 (void *)young);
386 next = gc->gc.gc_next;
387 }
388 else {
Tim Peters19b74c72002-07-01 03:52:19 +0000389 /* This *may* be unreachable. To make progress,
390 * assume it is. gc isn't directly reachable from
391 * any object we've already traversed, but may be
392 * reachable from an object we haven't gotten to yet.
393 * visit_reachable will eventually move gc back into
394 * young if that's so, and we'll see it again.
395 */
396 next = gc->gc.gc_next;
Tim Peterse2d59182004-11-01 01:39:08 +0000397 gc_list_move(gc, unreachable);
Tim Peters19b74c72002-07-01 03:52:19 +0000398 gc->gc.gc_refs = GC_TENTATIVELY_UNREACHABLE;
399 }
Tim Peters19b74c72002-07-01 03:52:19 +0000400 gc = next;
Jeremy Hyltonc5007aa2000-06-30 05:02:53 +0000401 }
402}
403
Tim Peters86b993b2003-04-05 17:35:54 +0000404/* Return true if object has a finalization method.
405 * CAUTION: An instance of an old-style class has to be checked for a
Tim Petersf6b80452003-04-07 19:21:15 +0000406 *__del__ method, and earlier versions of this used to call PyObject_HasAttr,
407 * which in turn could call the class's __getattr__ hook (if any). That
408 * could invoke arbitrary Python code, mutating the object graph in arbitrary
409 * ways, and that was the source of some excruciatingly subtle bugs.
Tim Peters86b993b2003-04-05 17:35:54 +0000410 */
Neil Schemenauera765c122001-11-01 17:35:23 +0000411static int
412has_finalizer(PyObject *op)
Jeremy Hyltonc5007aa2000-06-30 05:02:53 +0000413{
Tim Peters86b993b2003-04-05 17:35:54 +0000414 if (PyInstance_Check(op)) {
Tim Peters86b993b2003-04-05 17:35:54 +0000415 assert(delstr != NULL);
Tim Petersf6b80452003-04-07 19:21:15 +0000416 return _PyInstance_Lookup(op, delstr) != NULL;
Tim Peters86b993b2003-04-05 17:35:54 +0000417 }
Phillip J. Eby2ba96612006-04-10 17:51:05 +0000418 else if (PyType_HasFeature(op->ob_type, Py_TPFLAGS_HEAPTYPE))
Tim Peters86b993b2003-04-05 17:35:54 +0000419 return op->ob_type->tp_del != NULL;
Phillip J. Eby2ba96612006-04-10 17:51:05 +0000420 else if (PyGen_CheckExact(op))
421 return PyGen_NeedsFinalizing((PyGenObject *)op);
422 else
423 return 0;
Neil Schemenauera765c122001-11-01 17:35:23 +0000424}
425
Tim Petersead8b7a2004-10-30 23:09:22 +0000426/* Move the objects in unreachable with __del__ methods into `finalizers`.
427 * Objects moved into `finalizers` have gc_refs set to GC_REACHABLE; the
428 * objects remaining in unreachable are left at GC_TENTATIVELY_UNREACHABLE.
Jeremy Hyltonce136e92003-04-04 19:59:06 +0000429 */
Neil Schemenauera765c122001-11-01 17:35:23 +0000430static void
Tim Petersead8b7a2004-10-30 23:09:22 +0000431move_finalizers(PyGC_Head *unreachable, PyGC_Head *finalizers)
Neil Schemenauera765c122001-11-01 17:35:23 +0000432{
Tim Petersead8b7a2004-10-30 23:09:22 +0000433 PyGC_Head *gc;
434 PyGC_Head *next;
Tim Petersf6b80452003-04-07 19:21:15 +0000435
Tim Petersead8b7a2004-10-30 23:09:22 +0000436 /* March over unreachable. Move objects with finalizers into
437 * `finalizers`.
438 */
439 for (gc = unreachable->gc.gc_next; gc != unreachable; gc = next) {
Neil Schemenauer43411b52001-08-30 00:05:51 +0000440 PyObject *op = FROM_GC(gc);
Jeremy Hyltonce136e92003-04-04 19:59:06 +0000441
Tim Petersf6ae7a42003-04-05 18:40:50 +0000442 assert(IS_TENTATIVELY_UNREACHABLE(op));
Tim Petersead8b7a2004-10-30 23:09:22 +0000443 next = gc->gc.gc_next;
Tim Petersf6ae7a42003-04-05 18:40:50 +0000444
Tim Petersf6b80452003-04-07 19:21:15 +0000445 if (has_finalizer(op)) {
Tim Peterse2d59182004-11-01 01:39:08 +0000446 gc_list_move(gc, finalizers);
Tim Petersf6b80452003-04-07 19:21:15 +0000447 gc->gc.gc_refs = GC_REACHABLE;
Jeremy Hyltonce136e92003-04-04 19:59:06 +0000448 }
Jeremy Hyltonc5007aa2000-06-30 05:02:53 +0000449 }
450}
451
Tim Peters19b74c72002-07-01 03:52:19 +0000452/* A traversal callback for move_finalizer_reachable. */
453static int
454visit_move(PyObject *op, PyGC_Head *tolist)
455{
456 if (PyObject_IS_GC(op)) {
Tim Petersea405632002-07-02 00:52:30 +0000457 if (IS_TENTATIVELY_UNREACHABLE(op)) {
Tim Peters19b74c72002-07-01 03:52:19 +0000458 PyGC_Head *gc = AS_GC(op);
Tim Peterse2d59182004-11-01 01:39:08 +0000459 gc_list_move(gc, tolist);
Tim Peters19b74c72002-07-01 03:52:19 +0000460 gc->gc.gc_refs = GC_REACHABLE;
461 }
462 }
463 return 0;
464}
465
466/* Move objects that are reachable from finalizers, from the unreachable set
Tim Petersf6b80452003-04-07 19:21:15 +0000467 * into finalizers set.
Tim Peters19b74c72002-07-01 03:52:19 +0000468 */
Jeremy Hyltonc5007aa2000-06-30 05:02:53 +0000469static void
Tim Petersf6b80452003-04-07 19:21:15 +0000470move_finalizer_reachable(PyGC_Head *finalizers)
Jeremy Hyltonc5007aa2000-06-30 05:02:53 +0000471{
472 traverseproc traverse;
Tim Peters9e4ca102001-10-11 18:31:31 +0000473 PyGC_Head *gc = finalizers->gc.gc_next;
Tim Petersbf384c22003-04-06 00:11:39 +0000474 for (; gc != finalizers; gc = gc->gc.gc_next) {
475 /* Note that the finalizers list may grow during this. */
Christian Heimese93237d2007-12-19 02:37:44 +0000476 traverse = Py_TYPE(FROM_GC(gc))->tp_traverse;
Tim Peters88396172002-06-30 17:56:40 +0000477 (void) traverse(FROM_GC(gc),
Tim Petersbf384c22003-04-06 00:11:39 +0000478 (visitproc)visit_move,
Tim Petersf6b80452003-04-07 19:21:15 +0000479 (void *)finalizers);
Jeremy Hyltonc5007aa2000-06-30 05:02:53 +0000480 }
481}
482
Tim Petersead8b7a2004-10-30 23:09:22 +0000483/* Clear all weakrefs to unreachable objects, and if such a weakref has a
484 * callback, invoke it if necessary. Note that it's possible for such
485 * weakrefs to be outside the unreachable set -- indeed, those are precisely
486 * the weakrefs whose callbacks must be invoked. See gc_weakref.txt for
487 * overview & some details. Some weakrefs with callbacks may be reclaimed
488 * directly by this routine; the number reclaimed is the return value. Other
489 * weakrefs with callbacks may be moved into the `old` generation. Objects
490 * moved into `old` have gc_refs set to GC_REACHABLE; the objects remaining in
491 * unreachable are left at GC_TENTATIVELY_UNREACHABLE. When this returns,
492 * no object in `unreachable` is weakly referenced anymore.
Tim Peters403a2032003-11-20 21:21:46 +0000493 */
494static int
Tim Petersead8b7a2004-10-30 23:09:22 +0000495handle_weakrefs(PyGC_Head *unreachable, PyGC_Head *old)
Tim Peters403a2032003-11-20 21:21:46 +0000496{
Tim Petersead8b7a2004-10-30 23:09:22 +0000497 PyGC_Head *gc;
498 PyObject *op; /* generally FROM_GC(gc) */
499 PyWeakReference *wr; /* generally a cast of op */
Tim Petersead8b7a2004-10-30 23:09:22 +0000500 PyGC_Head wrcb_to_call; /* weakrefs with callbacks to call */
Tim Petersead8b7a2004-10-30 23:09:22 +0000501 PyGC_Head *next;
Tim Peters403a2032003-11-20 21:21:46 +0000502 int num_freed = 0;
503
Tim Petersead8b7a2004-10-30 23:09:22 +0000504 gc_list_init(&wrcb_to_call);
Tim Peters403a2032003-11-20 21:21:46 +0000505
Tim Petersead8b7a2004-10-30 23:09:22 +0000506 /* Clear all weakrefs to the objects in unreachable. If such a weakref
507 * also has a callback, move it into `wrcb_to_call` if the callback
Tim Peterscc2a8662004-10-31 22:12:43 +0000508 * needs to be invoked. Note that we cannot invoke any callbacks until
509 * all weakrefs to unreachable objects are cleared, lest the callback
510 * resurrect an unreachable object via a still-active weakref. We
511 * make another pass over wrcb_to_call, invoking callbacks, after this
512 * pass completes.
Tim Petersead8b7a2004-10-30 23:09:22 +0000513 */
514 for (gc = unreachable->gc.gc_next; gc != unreachable; gc = next) {
515 PyWeakReference **wrlist;
516
517 op = FROM_GC(gc);
518 assert(IS_TENTATIVELY_UNREACHABLE(op));
519 next = gc->gc.gc_next;
520
Christian Heimese93237d2007-12-19 02:37:44 +0000521 if (! PyType_SUPPORTS_WEAKREFS(Py_TYPE(op)))
Tim Petersead8b7a2004-10-30 23:09:22 +0000522 continue;
523
524 /* It supports weakrefs. Does it have any? */
525 wrlist = (PyWeakReference **)
526 PyObject_GET_WEAKREFS_LISTPTR(op);
527
528 /* `op` may have some weakrefs. March over the list, clear
529 * all the weakrefs, and move the weakrefs with callbacks
Tim Peterscc2a8662004-10-31 22:12:43 +0000530 * that must be called into wrcb_to_call.
Tim Petersead8b7a2004-10-30 23:09:22 +0000531 */
532 for (wr = *wrlist; wr != NULL; wr = *wrlist) {
533 PyGC_Head *wrasgc; /* AS_GC(wr) */
534
535 /* _PyWeakref_ClearRef clears the weakref but leaves
536 * the callback pointer intact. Obscure: it also
537 * changes *wrlist.
538 */
539 assert(wr->wr_object == op);
540 _PyWeakref_ClearRef(wr);
541 assert(wr->wr_object == Py_None);
542 if (wr->wr_callback == NULL)
543 continue; /* no callback */
544
545 /* Headache time. `op` is going away, and is weakly referenced by
546 * `wr`, which has a callback. Should the callback be invoked? If wr
547 * is also trash, no:
548 *
549 * 1. There's no need to call it. The object and the weakref are
550 * both going away, so it's legitimate to pretend the weakref is
551 * going away first. The user has to ensure a weakref outlives its
552 * referent if they want a guarantee that the wr callback will get
553 * invoked.
554 *
555 * 2. It may be catastrophic to call it. If the callback is also in
556 * cyclic trash (CT), then although the CT is unreachable from
557 * outside the current generation, CT may be reachable from the
558 * callback. Then the callback could resurrect insane objects.
559 *
560 * Since the callback is never needed and may be unsafe in this case,
Tim Peterscc2a8662004-10-31 22:12:43 +0000561 * wr is simply left in the unreachable set. Note that because we
562 * already called _PyWeakref_ClearRef(wr), its callback will never
563 * trigger.
Tim Petersead8b7a2004-10-30 23:09:22 +0000564 *
565 * OTOH, if wr isn't part of CT, we should invoke the callback: the
566 * weakref outlived the trash. Note that since wr isn't CT in this
567 * case, its callback can't be CT either -- wr acted as an external
568 * root to this generation, and therefore its callback did too. So
569 * nothing in CT is reachable from the callback either, so it's hard
570 * to imagine how calling it later could create a problem for us. wr
571 * is moved to wrcb_to_call in this case.
Tim Petersead8b7a2004-10-30 23:09:22 +0000572 */
Tim Peterscc2a8662004-10-31 22:12:43 +0000573 if (IS_TENTATIVELY_UNREACHABLE(wr))
574 continue;
575 assert(IS_REACHABLE(wr));
576
Tim Petersead8b7a2004-10-30 23:09:22 +0000577 /* Create a new reference so that wr can't go away
578 * before we can process it again.
579 */
580 Py_INCREF(wr);
581
Tim Peterscc2a8662004-10-31 22:12:43 +0000582 /* Move wr to wrcb_to_call, for the next pass. */
Tim Petersead8b7a2004-10-30 23:09:22 +0000583 wrasgc = AS_GC(wr);
Tim Peterscc2a8662004-10-31 22:12:43 +0000584 assert(wrasgc != next); /* wrasgc is reachable, but
585 next isn't, so they can't
586 be the same */
Tim Peterse2d59182004-11-01 01:39:08 +0000587 gc_list_move(wrasgc, &wrcb_to_call);
Tim Petersead8b7a2004-10-30 23:09:22 +0000588 }
589 }
590
Tim Peterscc2a8662004-10-31 22:12:43 +0000591 /* Invoke the callbacks we decided to honor. It's safe to invoke them
592 * because they can't reference unreachable objects.
Tim Petersead8b7a2004-10-30 23:09:22 +0000593 */
594 while (! gc_list_is_empty(&wrcb_to_call)) {
595 PyObject *temp;
596 PyObject *callback;
597
598 gc = wrcb_to_call.gc.gc_next;
599 op = FROM_GC(gc);
600 assert(IS_REACHABLE(op));
601 assert(PyWeakref_Check(op));
602 wr = (PyWeakReference *)op;
603 callback = wr->wr_callback;
604 assert(callback != NULL);
605
606 /* copy-paste of weakrefobject.c's handle_callback() */
Georg Brandl684fd0c2006-05-25 19:15:31 +0000607 temp = PyObject_CallFunctionObjArgs(callback, wr, NULL);
Tim Petersead8b7a2004-10-30 23:09:22 +0000608 if (temp == NULL)
609 PyErr_WriteUnraisable(callback);
610 else
611 Py_DECREF(temp);
612
613 /* Give up the reference we created in the first pass. When
614 * op's refcount hits 0 (which it may or may not do right now),
Tim Peterscc2a8662004-10-31 22:12:43 +0000615 * op's tp_dealloc will decref op->wr_callback too. Note
616 * that the refcount probably will hit 0 now, and because this
617 * weakref was reachable to begin with, gc didn't already
618 * add it to its count of freed objects. Example: a reachable
619 * weak value dict maps some key to this reachable weakref.
620 * The callback removes this key->weakref mapping from the
621 * dict, leaving no other references to the weakref (excepting
622 * ours).
Tim Petersead8b7a2004-10-30 23:09:22 +0000623 */
624 Py_DECREF(op);
625 if (wrcb_to_call.gc.gc_next == gc) {
626 /* object is still alive -- move it */
Tim Peterse2d59182004-11-01 01:39:08 +0000627 gc_list_move(gc, old);
Tim Petersead8b7a2004-10-30 23:09:22 +0000628 }
629 else
630 ++num_freed;
631 }
632
Tim Peters403a2032003-11-20 21:21:46 +0000633 return num_freed;
634}
635
Jeremy Hyltonc5007aa2000-06-30 05:02:53 +0000636static void
Jeremy Hylton06257772000-08-31 15:10:24 +0000637debug_instance(char *msg, PyInstanceObject *inst)
Jeremy Hyltonc5007aa2000-06-30 05:02:53 +0000638{
Jeremy Hyltonc5007aa2000-06-30 05:02:53 +0000639 char *cname;
Neil Schemenauera765c122001-11-01 17:35:23 +0000640 /* simple version of instance_repr */
Jeremy Hyltonc5007aa2000-06-30 05:02:53 +0000641 PyObject *classname = inst->in_class->cl_name;
642 if (classname != NULL && PyString_Check(classname))
643 cname = PyString_AsString(classname);
644 else
645 cname = "?";
Jeremy Hylton06257772000-08-31 15:10:24 +0000646 PySys_WriteStderr("gc: %.100s <%.100s instance at %p>\n",
647 msg, cname, inst);
Jeremy Hyltonc5007aa2000-06-30 05:02:53 +0000648}
649
650static void
Jeremy Hylton06257772000-08-31 15:10:24 +0000651debug_cycle(char *msg, PyObject *op)
Jeremy Hyltonc5007aa2000-06-30 05:02:53 +0000652{
653 if ((debug & DEBUG_INSTANCES) && PyInstance_Check(op)) {
Jeremy Hylton06257772000-08-31 15:10:24 +0000654 debug_instance(msg, (PyInstanceObject *)op);
Neil Schemenauer544de1e2000-09-22 15:22:38 +0000655 }
656 else if (debug & DEBUG_OBJECTS) {
Jeremy Hylton06257772000-08-31 15:10:24 +0000657 PySys_WriteStderr("gc: %.100s <%.100s %p>\n",
Christian Heimese93237d2007-12-19 02:37:44 +0000658 msg, Py_TYPE(op)->tp_name, op);
Jeremy Hyltonc5007aa2000-06-30 05:02:53 +0000659 }
660}
661
Tim Petersbf384c22003-04-06 00:11:39 +0000662/* Handle uncollectable garbage (cycles with finalizers, and stuff reachable
663 * only from such cycles).
Tim Petersf6b80452003-04-07 19:21:15 +0000664 * If DEBUG_SAVEALL, all objects in finalizers are appended to the module
665 * garbage list (a Python list), else only the objects in finalizers with
666 * __del__ methods are appended to garbage. All objects in finalizers are
667 * merged into the old list regardless.
Tim Peters259272b2003-04-06 19:41:39 +0000668 * Returns 0 if all OK, <0 on error (out of memory to grow the garbage list).
669 * The finalizers list is made empty on a successful return.
Tim Petersbf384c22003-04-06 00:11:39 +0000670 */
Tim Peters259272b2003-04-06 19:41:39 +0000671static int
Tim Petersf6b80452003-04-07 19:21:15 +0000672handle_finalizers(PyGC_Head *finalizers, PyGC_Head *old)
Jeremy Hyltonc5007aa2000-06-30 05:02:53 +0000673{
Tim Petersf6b80452003-04-07 19:21:15 +0000674 PyGC_Head *gc = finalizers->gc.gc_next;
675
Jeremy Hyltonc5007aa2000-06-30 05:02:53 +0000676 if (garbage == NULL) {
677 garbage = PyList_New(0);
Tim Petersbf384c22003-04-06 00:11:39 +0000678 if (garbage == NULL)
679 Py_FatalError("gc couldn't create gc.garbage list");
Jeremy Hyltonc5007aa2000-06-30 05:02:53 +0000680 }
Tim Petersf6b80452003-04-07 19:21:15 +0000681 for (; gc != finalizers; gc = gc->gc.gc_next) {
682 PyObject *op = FROM_GC(gc);
683
684 if ((debug & DEBUG_SAVEALL) || has_finalizer(op)) {
685 if (PyList_Append(garbage, op) < 0)
686 return -1;
687 }
Jeremy Hyltonc5007aa2000-06-30 05:02:53 +0000688 }
Tim Petersf6b80452003-04-07 19:21:15 +0000689
Tim Peters259272b2003-04-06 19:41:39 +0000690 gc_list_merge(finalizers, old);
691 return 0;
Jeremy Hyltonc5007aa2000-06-30 05:02:53 +0000692}
693
Neil Schemenauer544de1e2000-09-22 15:22:38 +0000694/* Break reference cycles by clearing the containers involved. This is
Jeremy Hyltonc5007aa2000-06-30 05:02:53 +0000695 * tricky business as the lists can be changing and we don't know which
Tim Peters19b74c72002-07-01 03:52:19 +0000696 * objects may be freed. It is possible I screwed something up here.
697 */
Jeremy Hyltonc5007aa2000-06-30 05:02:53 +0000698static void
Jeremy Hyltonce136e92003-04-04 19:59:06 +0000699delete_garbage(PyGC_Head *collectable, PyGC_Head *old)
Jeremy Hyltonc5007aa2000-06-30 05:02:53 +0000700{
701 inquiry clear;
702
Jeremy Hyltonce136e92003-04-04 19:59:06 +0000703 while (!gc_list_is_empty(collectable)) {
704 PyGC_Head *gc = collectable->gc.gc_next;
Neil Schemenauer43411b52001-08-30 00:05:51 +0000705 PyObject *op = FROM_GC(gc);
Tim Peters88396172002-06-30 17:56:40 +0000706
Tim Peters19b74c72002-07-01 03:52:19 +0000707 assert(IS_TENTATIVELY_UNREACHABLE(op));
Neil Schemenauer544de1e2000-09-22 15:22:38 +0000708 if (debug & DEBUG_SAVEALL) {
709 PyList_Append(garbage, op);
Jeremy Hyltonc5007aa2000-06-30 05:02:53 +0000710 }
Neil Schemenauer544de1e2000-09-22 15:22:38 +0000711 else {
Christian Heimese93237d2007-12-19 02:37:44 +0000712 if ((clear = Py_TYPE(op)->tp_clear) != NULL) {
Neil Schemenauer544de1e2000-09-22 15:22:38 +0000713 Py_INCREF(op);
Jeremy Hylton8a135182002-06-06 23:23:55 +0000714 clear(op);
Neil Schemenauer544de1e2000-09-22 15:22:38 +0000715 Py_DECREF(op);
716 }
717 }
Jeremy Hyltonce136e92003-04-04 19:59:06 +0000718 if (collectable->gc.gc_next == gc) {
Neil Schemenauer544de1e2000-09-22 15:22:38 +0000719 /* object is still alive, move it, it may die later */
Tim Peterse2d59182004-11-01 01:39:08 +0000720 gc_list_move(gc, old);
Tim Peters19b74c72002-07-01 03:52:19 +0000721 gc->gc.gc_refs = GC_REACHABLE;
Jeremy Hyltonc5007aa2000-06-30 05:02:53 +0000722 }
723 }
724}
725
Christian Heimes3b718a72008-02-14 12:47:33 +0000726/* Clear all free lists
727 * All free lists are cleared during the collection of the highest generation.
728 * Allocated items in the free list may keep a pymalloc arena occupied.
729 * Clearing the free lists may give back memory to the OS earlier.
730 */
731static void
732clear_freelists(void)
733{
734 (void)PyMethod_ClearFreeList();
735 (void)PyFrame_ClearFreeList();
736 (void)PyCFunction_ClearFreeList();
737 (void)PyTuple_ClearFreeList();
738 (void)PyUnicode_ClearFreeList();
739}
740
Jeremy Hyltonc5007aa2000-06-30 05:02:53 +0000741/* This is the main function. Read this to understand how the
742 * collection process works. */
Neal Norwitz7b216c52006-03-04 20:01:53 +0000743static Py_ssize_t
Neil Schemenauer2880ae52002-05-04 05:35:20 +0000744collect(int generation)
Jeremy Hyltonc5007aa2000-06-30 05:02:53 +0000745{
Neil Schemenauer2880ae52002-05-04 05:35:20 +0000746 int i;
Neal Norwitz7b216c52006-03-04 20:01:53 +0000747 Py_ssize_t m = 0; /* # objects collected */
748 Py_ssize_t n = 0; /* # unreachable objects that couldn't be collected */
Neil Schemenauer2880ae52002-05-04 05:35:20 +0000749 PyGC_Head *young; /* the generation we are examining */
750 PyGC_Head *old; /* next older generation */
Tim Peters403a2032003-11-20 21:21:46 +0000751 PyGC_Head unreachable; /* non-problematic unreachable trash */
752 PyGC_Head finalizers; /* objects with, & reachable from, __del__ */
Jeremy Hyltonc5007aa2000-06-30 05:02:53 +0000753 PyGC_Head *gc;
Skip Montanaroc34b9312006-04-21 01:33:40 +0000754 double t1 = 0.0;
Jeremy Hyltonc5007aa2000-06-30 05:02:53 +0000755
Tim Peters93ad66d2003-04-05 17:15:44 +0000756 if (delstr == NULL) {
757 delstr = PyString_InternFromString("__del__");
758 if (delstr == NULL)
759 Py_FatalError("gc couldn't allocate \"__del__\"");
760 }
761
Jeremy Hyltonc5007aa2000-06-30 05:02:53 +0000762 if (debug & DEBUG_STATS) {
Skip Montanaroc34b9312006-04-21 01:33:40 +0000763 if (tmod != NULL) {
764 PyObject *f = PyObject_CallMethod(tmod, "time", NULL);
765 if (f == NULL) {
766 PyErr_Clear();
767 }
768 else {
769 t1 = PyFloat_AsDouble(f);
770 Py_DECREF(f);
771 }
772 }
Neil Schemenauer2880ae52002-05-04 05:35:20 +0000773 PySys_WriteStderr("gc: collecting generation %d...\n",
774 generation);
775 PySys_WriteStderr("gc: objects in each generation:");
Tim Peters62e97f02006-03-28 21:44:32 +0000776 for (i = 0; i < NUM_GENERATIONS; i++)
777 PySys_WriteStderr(" %" PY_FORMAT_SIZE_T "d",
778 gc_list_size(GEN_HEAD(i)));
Neil Schemenauer2880ae52002-05-04 05:35:20 +0000779 PySys_WriteStderr("\n");
780 }
781
782 /* update collection and allocation counters */
783 if (generation+1 < NUM_GENERATIONS)
784 generations[generation+1].count += 1;
785 for (i = 0; i <= generation; i++)
Neil Schemenauerc9051642002-06-28 19:16:04 +0000786 generations[i].count = 0;
Neil Schemenauer2880ae52002-05-04 05:35:20 +0000787
788 /* merge younger generations with one we are currently collecting */
789 for (i = 0; i < generation; i++) {
790 gc_list_merge(GEN_HEAD(i), GEN_HEAD(generation));
791 }
792
793 /* handy references */
794 young = GEN_HEAD(generation);
Tim Peters19b74c72002-07-01 03:52:19 +0000795 if (generation < NUM_GENERATIONS-1)
Neil Schemenauer2880ae52002-05-04 05:35:20 +0000796 old = GEN_HEAD(generation+1);
Tim Peters19b74c72002-07-01 03:52:19 +0000797 else
798 old = young;
Jeremy Hyltonc5007aa2000-06-30 05:02:53 +0000799
800 /* Using ob_refcnt and gc_refs, calculate which objects in the
Tim Petersead8b7a2004-10-30 23:09:22 +0000801 * container set are reachable from outside the set (i.e., have a
Jeremy Hyltonc5007aa2000-06-30 05:02:53 +0000802 * refcount greater than 0 when all the references within the
Tim Petersead8b7a2004-10-30 23:09:22 +0000803 * set are taken into account).
Tim Peters19b74c72002-07-01 03:52:19 +0000804 */
Jeremy Hyltonc5007aa2000-06-30 05:02:53 +0000805 update_refs(young);
806 subtract_refs(young);
807
Tim Peters19b74c72002-07-01 03:52:19 +0000808 /* Leave everything reachable from outside young in young, and move
809 * everything else (in young) to unreachable.
810 * NOTE: This used to move the reachable objects into a reachable
811 * set instead. But most things usually turn out to be reachable,
812 * so it's more efficient to move the unreachable things.
813 */
Jeremy Hyltonc5007aa2000-06-30 05:02:53 +0000814 gc_list_init(&unreachable);
Tim Peters19b74c72002-07-01 03:52:19 +0000815 move_unreachable(young, &unreachable);
Jeremy Hyltonc5007aa2000-06-30 05:02:53 +0000816
Tim Peters19b74c72002-07-01 03:52:19 +0000817 /* Move reachable objects to next generation. */
818 if (young != old)
819 gc_list_merge(young, old);
Jeremy Hyltonc5007aa2000-06-30 05:02:53 +0000820
Tim Peters19b74c72002-07-01 03:52:19 +0000821 /* All objects in unreachable are trash, but objects reachable from
822 * finalizers can't safely be deleted. Python programmers should take
823 * care not to create such things. For Python, finalizers means
Tim Peters403a2032003-11-20 21:21:46 +0000824 * instance objects with __del__ methods. Weakrefs with callbacks
Tim Petersead8b7a2004-10-30 23:09:22 +0000825 * can also call arbitrary Python code but they will be dealt with by
826 * handle_weakrefs().
Tim Petersf6b80452003-04-07 19:21:15 +0000827 */
Jeremy Hyltonc5007aa2000-06-30 05:02:53 +0000828 gc_list_init(&finalizers);
Tim Petersead8b7a2004-10-30 23:09:22 +0000829 move_finalizers(&unreachable, &finalizers);
Tim Petersbf384c22003-04-06 00:11:39 +0000830 /* finalizers contains the unreachable objects with a finalizer;
Tim Peters403a2032003-11-20 21:21:46 +0000831 * unreachable objects reachable *from* those are also uncollectable,
832 * and we move those into the finalizers list too.
Tim Petersbf384c22003-04-06 00:11:39 +0000833 */
Tim Petersf6b80452003-04-07 19:21:15 +0000834 move_finalizer_reachable(&finalizers);
Jeremy Hyltonc5007aa2000-06-30 05:02:53 +0000835
836 /* Collect statistics on collectable objects found and print
Tim Peters403a2032003-11-20 21:21:46 +0000837 * debugging information.
838 */
Tim Petersf6b80452003-04-07 19:21:15 +0000839 for (gc = unreachable.gc.gc_next; gc != &unreachable;
Tim Peters9e4ca102001-10-11 18:31:31 +0000840 gc = gc->gc.gc_next) {
Jeremy Hyltonc5007aa2000-06-30 05:02:53 +0000841 m++;
Jeremy Hylton06257772000-08-31 15:10:24 +0000842 if (debug & DEBUG_COLLECTABLE) {
Neil Schemenauer43411b52001-08-30 00:05:51 +0000843 debug_cycle("collectable", FROM_GC(gc));
Jeremy Hyltonc5007aa2000-06-30 05:02:53 +0000844 }
Skip Montanaroc34b9312006-04-21 01:33:40 +0000845 if (tmod != NULL && (debug & DEBUG_STATS)) {
846 PyObject *f = PyObject_CallMethod(tmod, "time", NULL);
847 if (f == NULL) {
848 PyErr_Clear();
849 }
850 else {
851 t1 = PyFloat_AsDouble(f)-t1;
852 Py_DECREF(f);
853 PySys_WriteStderr("gc: %.4fs elapsed.\n", t1);
854 }
855 }
Jeremy Hyltonc5007aa2000-06-30 05:02:53 +0000856 }
Tim Petersead8b7a2004-10-30 23:09:22 +0000857
858 /* Clear weakrefs and invoke callbacks as necessary. */
859 m += handle_weakrefs(&unreachable, old);
860
Tim Petersfb2ab4d2003-04-07 22:41:24 +0000861 /* Call tp_clear on objects in the unreachable set. This will cause
862 * the reference cycles to be broken. It may also cause some objects
863 * in finalizers to be freed.
864 */
Tim Petersf6b80452003-04-07 19:21:15 +0000865 delete_garbage(&unreachable, old);
Jeremy Hyltonc5007aa2000-06-30 05:02:53 +0000866
867 /* Collect statistics on uncollectable objects found and print
868 * debugging information. */
Tim Peters50c61d52003-04-06 01:50:50 +0000869 for (gc = finalizers.gc.gc_next;
Tim Petersbf384c22003-04-06 00:11:39 +0000870 gc != &finalizers;
871 gc = gc->gc.gc_next) {
Jeremy Hyltonc5007aa2000-06-30 05:02:53 +0000872 n++;
Tim Petersbf384c22003-04-06 00:11:39 +0000873 if (debug & DEBUG_UNCOLLECTABLE)
Neil Schemenauer43411b52001-08-30 00:05:51 +0000874 debug_cycle("uncollectable", FROM_GC(gc));
Tim Petersbf384c22003-04-06 00:11:39 +0000875 }
Jeremy Hylton06257772000-08-31 15:10:24 +0000876 if (debug & DEBUG_STATS) {
Tim Peters62e97f02006-03-28 21:44:32 +0000877 if (m == 0 && n == 0)
Jeremy Hylton06257772000-08-31 15:10:24 +0000878 PySys_WriteStderr("gc: done.\n");
Tim Peters62e97f02006-03-28 21:44:32 +0000879 else
Neal Norwitze22373d2006-03-06 23:31:56 +0000880 PySys_WriteStderr(
Tim Peters62e97f02006-03-28 21:44:32 +0000881 "gc: done, "
882 "%" PY_FORMAT_SIZE_T "d unreachable, "
883 "%" PY_FORMAT_SIZE_T "d uncollectable.\n",
Neal Norwitze22373d2006-03-06 23:31:56 +0000884 n+m, n);
Jeremy Hyltonc5007aa2000-06-30 05:02:53 +0000885 }
886
887 /* Append instances in the uncollectable set to a Python
888 * reachable list of garbage. The programmer has to deal with
Tim Petersbf384c22003-04-06 00:11:39 +0000889 * this if they insist on creating this type of structure.
890 */
Tim Petersf6b80452003-04-07 19:21:15 +0000891 (void)handle_finalizers(&finalizers, old);
Jeremy Hyltonc5007aa2000-06-30 05:02:53 +0000892
Christian Heimes3b718a72008-02-14 12:47:33 +0000893 /* Clear free list only during the collection of the higest
894 * generation */
895 if (generation == NUM_GENERATIONS-1) {
896 clear_freelists();
897 }
898
Jeremy Hyltonb709df32000-09-01 02:47:25 +0000899 if (PyErr_Occurred()) {
Tim Petersf6b80452003-04-07 19:21:15 +0000900 if (gc_str == NULL)
Tim Petersfb2ab4d2003-04-07 22:41:24 +0000901 gc_str = PyString_FromString("garbage collection");
Jeremy Hyltonb709df32000-09-01 02:47:25 +0000902 PyErr_WriteUnraisable(gc_str);
903 Py_FatalError("unexpected exception during garbage collection");
904 }
Jeremy Hyltonc5007aa2000-06-30 05:02:53 +0000905 return n+m;
906}
907
Neal Norwitz7b216c52006-03-04 20:01:53 +0000908static Py_ssize_t
Jeremy Hyltonc5007aa2000-06-30 05:02:53 +0000909collect_generations(void)
910{
Neil Schemenauer2880ae52002-05-04 05:35:20 +0000911 int i;
Neal Norwitz7b216c52006-03-04 20:01:53 +0000912 Py_ssize_t n = 0;
Jeremy Hyltonc5007aa2000-06-30 05:02:53 +0000913
Neil Schemenauer2880ae52002-05-04 05:35:20 +0000914 /* Find the oldest generation (higest numbered) where the count
915 * exceeds the threshold. Objects in the that generation and
916 * generations younger than it will be collected. */
917 for (i = NUM_GENERATIONS-1; i >= 0; i--) {
918 if (generations[i].count > generations[i].threshold) {
919 n = collect(i);
920 break;
Jeremy Hyltonc5007aa2000-06-30 05:02:53 +0000921 }
922 }
923 return n;
924}
925
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +0000926PyDoc_STRVAR(gc_enable__doc__,
Vladimir Marangozovf9d20c32000-08-06 22:45:31 +0000927"enable() -> None\n"
928"\n"
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +0000929"Enable automatic garbage collection.\n");
Jeremy Hyltonc5007aa2000-06-30 05:02:53 +0000930
Vladimir Marangozovf9d20c32000-08-06 22:45:31 +0000931static PyObject *
Tim Peters50c61d52003-04-06 01:50:50 +0000932gc_enable(PyObject *self, PyObject *noargs)
Vladimir Marangozovf9d20c32000-08-06 22:45:31 +0000933{
Vladimir Marangozovf9d20c32000-08-06 22:45:31 +0000934 enabled = 1;
Vladimir Marangozovf9d20c32000-08-06 22:45:31 +0000935 Py_INCREF(Py_None);
936 return Py_None;
937}
938
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +0000939PyDoc_STRVAR(gc_disable__doc__,
Vladimir Marangozovf9d20c32000-08-06 22:45:31 +0000940"disable() -> None\n"
941"\n"
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +0000942"Disable automatic garbage collection.\n");
Vladimir Marangozovf9d20c32000-08-06 22:45:31 +0000943
944static PyObject *
Tim Peters50c61d52003-04-06 01:50:50 +0000945gc_disable(PyObject *self, PyObject *noargs)
Vladimir Marangozovf9d20c32000-08-06 22:45:31 +0000946{
Vladimir Marangozovf9d20c32000-08-06 22:45:31 +0000947 enabled = 0;
Vladimir Marangozovf9d20c32000-08-06 22:45:31 +0000948 Py_INCREF(Py_None);
949 return Py_None;
950}
951
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +0000952PyDoc_STRVAR(gc_isenabled__doc__,
Vladimir Marangozovf9d20c32000-08-06 22:45:31 +0000953"isenabled() -> status\n"
954"\n"
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +0000955"Returns true if automatic garbage collection is enabled.\n");
Vladimir Marangozovf9d20c32000-08-06 22:45:31 +0000956
957static PyObject *
Tim Peters50c61d52003-04-06 01:50:50 +0000958gc_isenabled(PyObject *self, PyObject *noargs)
Vladimir Marangozovf9d20c32000-08-06 22:45:31 +0000959{
Raymond Hettinger674d56b2004-01-04 04:00:13 +0000960 return PyBool_FromLong((long)enabled);
Vladimir Marangozovf9d20c32000-08-06 22:45:31 +0000961}
962
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +0000963PyDoc_STRVAR(gc_collect__doc__,
Barry Warsawd3c38ff2006-03-07 09:46:03 +0000964"collect([generation]) -> n\n"
Jeremy Hyltonc5007aa2000-06-30 05:02:53 +0000965"\n"
Barry Warsawd3c38ff2006-03-07 09:46:03 +0000966"With no arguments, run a full collection. The optional argument\n"
967"may be an integer specifying which generation to collect. A ValueError\n"
968"is raised if the generation number is invalid.\n\n"
969"The number of unreachable objects is returned.\n");
Jeremy Hyltonc5007aa2000-06-30 05:02:53 +0000970
971static PyObject *
Barry Warsawd3c38ff2006-03-07 09:46:03 +0000972gc_collect(PyObject *self, PyObject *args, PyObject *kws)
Jeremy Hyltonc5007aa2000-06-30 05:02:53 +0000973{
Barry Warsawd3c38ff2006-03-07 09:46:03 +0000974 static char *keywords[] = {"generation", NULL};
975 int genarg = NUM_GENERATIONS - 1;
Neal Norwitz7b216c52006-03-04 20:01:53 +0000976 Py_ssize_t n;
Jeremy Hyltonc5007aa2000-06-30 05:02:53 +0000977
Barry Warsawd3c38ff2006-03-07 09:46:03 +0000978 if (!PyArg_ParseTupleAndKeywords(args, kws, "|i", keywords, &genarg))
979 return NULL;
980
981 else if (genarg < 0 || genarg >= NUM_GENERATIONS) {
982 PyErr_SetString(PyExc_ValueError, "invalid generation");
983 return NULL;
984 }
985
Tim Peters50c61d52003-04-06 01:50:50 +0000986 if (collecting)
Neil Schemenauere8c40cb2001-10-31 23:09:35 +0000987 n = 0; /* already collecting, don't do anything */
Neil Schemenauere8c40cb2001-10-31 23:09:35 +0000988 else {
989 collecting = 1;
Barry Warsawd3c38ff2006-03-07 09:46:03 +0000990 n = collect(genarg);
Neil Schemenauere8c40cb2001-10-31 23:09:35 +0000991 collecting = 0;
992 }
Jeremy Hyltonc5007aa2000-06-30 05:02:53 +0000993
Neal Norwitz7b216c52006-03-04 20:01:53 +0000994 return PyInt_FromSsize_t(n);
Jeremy Hyltonc5007aa2000-06-30 05:02:53 +0000995}
996
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +0000997PyDoc_STRVAR(gc_set_debug__doc__,
Jeremy Hyltonc5007aa2000-06-30 05:02:53 +0000998"set_debug(flags) -> None\n"
999"\n"
1000"Set the garbage collection debugging flags. Debugging information is\n"
1001"written to sys.stderr.\n"
1002"\n"
1003"flags is an integer and can have the following bits turned on:\n"
1004"\n"
1005" DEBUG_STATS - Print statistics during collection.\n"
1006" DEBUG_COLLECTABLE - Print collectable objects found.\n"
1007" DEBUG_UNCOLLECTABLE - Print unreachable but uncollectable objects found.\n"
1008" DEBUG_INSTANCES - Print instance objects.\n"
1009" DEBUG_OBJECTS - Print objects other than instances.\n"
Neil Schemenauer544de1e2000-09-22 15:22:38 +00001010" DEBUG_SAVEALL - Save objects to gc.garbage rather than freeing them.\n"
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00001011" DEBUG_LEAK - Debug leaking programs (everything but STATS).\n");
Jeremy Hyltonc5007aa2000-06-30 05:02:53 +00001012
1013static PyObject *
Vladimir Marangozovf9d20c32000-08-06 22:45:31 +00001014gc_set_debug(PyObject *self, PyObject *args)
Jeremy Hyltonc5007aa2000-06-30 05:02:53 +00001015{
Neil Schemenauer7760cff2000-09-22 22:35:36 +00001016 if (!PyArg_ParseTuple(args, "i:set_debug", &debug))
Jeremy Hyltonc5007aa2000-06-30 05:02:53 +00001017 return NULL;
1018
1019 Py_INCREF(Py_None);
1020 return Py_None;
1021}
1022
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00001023PyDoc_STRVAR(gc_get_debug__doc__,
Jeremy Hyltonc5007aa2000-06-30 05:02:53 +00001024"get_debug() -> flags\n"
1025"\n"
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00001026"Get the garbage collection debugging flags.\n");
Jeremy Hyltonc5007aa2000-06-30 05:02:53 +00001027
1028static PyObject *
Tim Peters50c61d52003-04-06 01:50:50 +00001029gc_get_debug(PyObject *self, PyObject *noargs)
Jeremy Hyltonc5007aa2000-06-30 05:02:53 +00001030{
Jeremy Hyltonc5007aa2000-06-30 05:02:53 +00001031 return Py_BuildValue("i", debug);
1032}
1033
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00001034PyDoc_STRVAR(gc_set_thresh__doc__,
Neal Norwitz2a47c0f2002-01-29 00:53:41 +00001035"set_threshold(threshold0, [threshold1, threshold2]) -> None\n"
Jeremy Hyltonc5007aa2000-06-30 05:02:53 +00001036"\n"
1037"Sets the collection thresholds. Setting threshold0 to zero disables\n"
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00001038"collection.\n");
Jeremy Hyltonc5007aa2000-06-30 05:02:53 +00001039
1040static PyObject *
Vladimir Marangozovf9d20c32000-08-06 22:45:31 +00001041gc_set_thresh(PyObject *self, PyObject *args)
Jeremy Hyltonc5007aa2000-06-30 05:02:53 +00001042{
Neil Schemenauer2880ae52002-05-04 05:35:20 +00001043 int i;
1044 if (!PyArg_ParseTuple(args, "i|ii:set_threshold",
1045 &generations[0].threshold,
1046 &generations[1].threshold,
1047 &generations[2].threshold))
Jeremy Hyltonc5007aa2000-06-30 05:02:53 +00001048 return NULL;
Neil Schemenauer2880ae52002-05-04 05:35:20 +00001049 for (i = 2; i < NUM_GENERATIONS; i++) {
1050 /* generations higher than 2 get the same threshold */
1051 generations[i].threshold = generations[2].threshold;
1052 }
Jeremy Hyltonc5007aa2000-06-30 05:02:53 +00001053
1054 Py_INCREF(Py_None);
1055 return Py_None;
1056}
1057
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00001058PyDoc_STRVAR(gc_get_thresh__doc__,
Jeremy Hyltonc5007aa2000-06-30 05:02:53 +00001059"get_threshold() -> (threshold0, threshold1, threshold2)\n"
1060"\n"
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00001061"Return the current collection thresholds\n");
Jeremy Hyltonc5007aa2000-06-30 05:02:53 +00001062
1063static PyObject *
Tim Peters50c61d52003-04-06 01:50:50 +00001064gc_get_thresh(PyObject *self, PyObject *noargs)
Jeremy Hyltonc5007aa2000-06-30 05:02:53 +00001065{
Neil Schemenauer2880ae52002-05-04 05:35:20 +00001066 return Py_BuildValue("(iii)",
1067 generations[0].threshold,
1068 generations[1].threshold,
1069 generations[2].threshold);
Jeremy Hyltonc5007aa2000-06-30 05:02:53 +00001070}
1071
Barry Warsawd3c38ff2006-03-07 09:46:03 +00001072PyDoc_STRVAR(gc_get_count__doc__,
1073"get_count() -> (count0, count1, count2)\n"
1074"\n"
1075"Return the current collection counts\n");
1076
1077static PyObject *
1078gc_get_count(PyObject *self, PyObject *noargs)
1079{
1080 return Py_BuildValue("(iii)",
1081 generations[0].count,
1082 generations[1].count,
1083 generations[2].count);
1084}
1085
Neil Schemenauer48c70342001-08-09 15:38:31 +00001086static int
Martin v. Löwis560da622001-11-24 09:24:51 +00001087referrersvisit(PyObject* obj, PyObject *objs)
Neil Schemenauer48c70342001-08-09 15:38:31 +00001088{
Neal Norwitzffb0d902006-04-06 08:07:25 +00001089 Py_ssize_t i;
Martin v. Löwisc8fe77b2001-11-29 18:08:31 +00001090 for (i = 0; i < PyTuple_GET_SIZE(objs); i++)
1091 if (PyTuple_GET_ITEM(objs, i) == obj)
1092 return 1;
Neil Schemenauer48c70342001-08-09 15:38:31 +00001093 return 0;
1094}
1095
Neil Schemenauer17e7be62001-08-10 14:46:47 +00001096static int
Martin v. Löwis560da622001-11-24 09:24:51 +00001097gc_referrers_for(PyObject *objs, PyGC_Head *list, PyObject *resultlist)
Neil Schemenauer48c70342001-08-09 15:38:31 +00001098{
1099 PyGC_Head *gc;
1100 PyObject *obj;
1101 traverseproc traverse;
Tim Peters9e4ca102001-10-11 18:31:31 +00001102 for (gc = list->gc.gc_next; gc != list; gc = gc->gc.gc_next) {
Neil Schemenauer43411b52001-08-30 00:05:51 +00001103 obj = FROM_GC(gc);
Christian Heimese93237d2007-12-19 02:37:44 +00001104 traverse = Py_TYPE(obj)->tp_traverse;
Neil Schemenauer48c70342001-08-09 15:38:31 +00001105 if (obj == objs || obj == resultlist)
1106 continue;
Martin v. Löwis560da622001-11-24 09:24:51 +00001107 if (traverse(obj, (visitproc)referrersvisit, objs)) {
Neil Schemenauer17e7be62001-08-10 14:46:47 +00001108 if (PyList_Append(resultlist, obj) < 0)
1109 return 0; /* error */
Neil Schemenauer48c70342001-08-09 15:38:31 +00001110 }
1111 }
Neil Schemenauer17e7be62001-08-10 14:46:47 +00001112 return 1; /* no error */
Neil Schemenauer48c70342001-08-09 15:38:31 +00001113}
1114
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00001115PyDoc_STRVAR(gc_get_referrers__doc__,
Martin v. Löwis560da622001-11-24 09:24:51 +00001116"get_referrers(*objs) -> list\n\
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00001117Return the list of objects that directly refer to any of objs.");
Neil Schemenauer48c70342001-08-09 15:38:31 +00001118
Neil Schemenauer17e7be62001-08-10 14:46:47 +00001119static PyObject *
Martin v. Löwis560da622001-11-24 09:24:51 +00001120gc_get_referrers(PyObject *self, PyObject *args)
Neil Schemenauer48c70342001-08-09 15:38:31 +00001121{
Neil Schemenauer2880ae52002-05-04 05:35:20 +00001122 int i;
Neil Schemenauer48c70342001-08-09 15:38:31 +00001123 PyObject *result = PyList_New(0);
Georg Brandl5c170fd2006-03-17 19:03:25 +00001124 if (!result) return NULL;
1125
Neil Schemenauer2880ae52002-05-04 05:35:20 +00001126 for (i = 0; i < NUM_GENERATIONS; i++) {
1127 if (!(gc_referrers_for(args, GEN_HEAD(i), result))) {
1128 Py_DECREF(result);
1129 return NULL;
1130 }
Neil Schemenauer17e7be62001-08-10 14:46:47 +00001131 }
Neil Schemenauer48c70342001-08-09 15:38:31 +00001132 return result;
1133}
Jeremy Hyltonc5007aa2000-06-30 05:02:53 +00001134
Tim Peters0f81ab62003-04-08 16:39:48 +00001135/* Append obj to list; return true if error (out of memory), false if OK. */
Jeremy Hylton5bd378b2003-04-03 16:28:38 +00001136static int
Tim Peters730f5532003-04-08 17:17:17 +00001137referentsvisit(PyObject *obj, PyObject *list)
Jeremy Hylton5bd378b2003-04-03 16:28:38 +00001138{
Tim Peters0f81ab62003-04-08 16:39:48 +00001139 return PyList_Append(list, obj) < 0;
Jeremy Hylton5bd378b2003-04-03 16:28:38 +00001140}
1141
Tim Peters730f5532003-04-08 17:17:17 +00001142PyDoc_STRVAR(gc_get_referents__doc__,
1143"get_referents(*objs) -> list\n\
Jeremy Hylton059b0942003-04-03 16:29:13 +00001144Return the list of objects that are directly referred to by objs.");
Jeremy Hylton5bd378b2003-04-03 16:28:38 +00001145
1146static PyObject *
Tim Peters730f5532003-04-08 17:17:17 +00001147gc_get_referents(PyObject *self, PyObject *args)
Jeremy Hylton5bd378b2003-04-03 16:28:38 +00001148{
Neal Norwitzffb0d902006-04-06 08:07:25 +00001149 Py_ssize_t i;
Jeremy Hylton5bd378b2003-04-03 16:28:38 +00001150 PyObject *result = PyList_New(0);
Tim Peters0f81ab62003-04-08 16:39:48 +00001151
1152 if (result == NULL)
1153 return NULL;
1154
Jeremy Hylton5bd378b2003-04-03 16:28:38 +00001155 for (i = 0; i < PyTuple_GET_SIZE(args); i++) {
Tim Peters0f81ab62003-04-08 16:39:48 +00001156 traverseproc traverse;
Tim Peters93ad66d2003-04-05 17:15:44 +00001157 PyObject *obj = PyTuple_GET_ITEM(args, i);
Tim Peters0f81ab62003-04-08 16:39:48 +00001158
1159 if (! PyObject_IS_GC(obj))
Jeremy Hylton5bd378b2003-04-03 16:28:38 +00001160 continue;
Christian Heimese93237d2007-12-19 02:37:44 +00001161 traverse = Py_TYPE(obj)->tp_traverse;
Tim Peters0f81ab62003-04-08 16:39:48 +00001162 if (! traverse)
1163 continue;
Tim Peters730f5532003-04-08 17:17:17 +00001164 if (traverse(obj, (visitproc)referentsvisit, result)) {
Tim Peters0f81ab62003-04-08 16:39:48 +00001165 Py_DECREF(result);
Jeremy Hylton5bd378b2003-04-03 16:28:38 +00001166 return NULL;
Tim Peters0f81ab62003-04-08 16:39:48 +00001167 }
Jeremy Hylton5bd378b2003-04-03 16:28:38 +00001168 }
1169 return result;
1170}
1171
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00001172PyDoc_STRVAR(gc_get_objects__doc__,
Neil Schemenauerc7c8d8e2001-08-09 15:58:59 +00001173"get_objects() -> [...]\n"
1174"\n"
1175"Return a list of objects tracked by the collector (excluding the list\n"
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00001176"returned).\n");
Neil Schemenauerc7c8d8e2001-08-09 15:58:59 +00001177
Neil Schemenauerc7c8d8e2001-08-09 15:58:59 +00001178static PyObject *
Tim Peters50c61d52003-04-06 01:50:50 +00001179gc_get_objects(PyObject *self, PyObject *noargs)
Neil Schemenauerc7c8d8e2001-08-09 15:58:59 +00001180{
Neil Schemenauer2880ae52002-05-04 05:35:20 +00001181 int i;
Neil Schemenauerc7c8d8e2001-08-09 15:58:59 +00001182 PyObject* result;
1183
Neil Schemenauerc7c8d8e2001-08-09 15:58:59 +00001184 result = PyList_New(0);
Tim Peters50c61d52003-04-06 01:50:50 +00001185 if (result == NULL)
Martin v. Löwisf8a6f242001-12-02 18:31:02 +00001186 return NULL;
Neil Schemenauer2880ae52002-05-04 05:35:20 +00001187 for (i = 0; i < NUM_GENERATIONS; i++) {
1188 if (append_objects(result, GEN_HEAD(i))) {
1189 Py_DECREF(result);
1190 return NULL;
1191 }
Martin v. Löwis155aad12001-12-02 12:21:34 +00001192 }
Neil Schemenauerc7c8d8e2001-08-09 15:58:59 +00001193 return result;
1194}
1195
1196
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00001197PyDoc_STRVAR(gc__doc__,
Jeremy Hyltonc5007aa2000-06-30 05:02:53 +00001198"This module provides access to the garbage collector for reference cycles.\n"
1199"\n"
Vladimir Marangozovf9d20c32000-08-06 22:45:31 +00001200"enable() -- Enable automatic garbage collection.\n"
1201"disable() -- Disable automatic garbage collection.\n"
1202"isenabled() -- Returns true if automatic collection is enabled.\n"
Jeremy Hyltonc5007aa2000-06-30 05:02:53 +00001203"collect() -- Do a full collection right now.\n"
Barry Warsawe5ec6132006-10-09 19:43:24 +00001204"get_count() -- Return the current collection counts.\n"
Jeremy Hyltonc5007aa2000-06-30 05:02:53 +00001205"set_debug() -- Set debugging flags.\n"
1206"get_debug() -- Get debugging flags.\n"
1207"set_threshold() -- Set the collection thresholds.\n"
1208"get_threshold() -- Return the current the collection thresholds.\n"
Neil Schemenauerc7c8d8e2001-08-09 15:58:59 +00001209"get_objects() -- Return a list of all objects tracked by the collector.\n"
Jeremy Hylton5bd378b2003-04-03 16:28:38 +00001210"get_referrers() -- Return the list of objects that refer to an object.\n"
Tim Peters730f5532003-04-08 17:17:17 +00001211"get_referents() -- Return the list of objects that an object refers to.\n");
Jeremy Hyltonc5007aa2000-06-30 05:02:53 +00001212
1213static PyMethodDef GcMethods[] = {
Tim Peters50c61d52003-04-06 01:50:50 +00001214 {"enable", gc_enable, METH_NOARGS, gc_enable__doc__},
1215 {"disable", gc_disable, METH_NOARGS, gc_disable__doc__},
1216 {"isenabled", gc_isenabled, METH_NOARGS, gc_isenabled__doc__},
Vladimir Marangozovf9d20c32000-08-06 22:45:31 +00001217 {"set_debug", gc_set_debug, METH_VARARGS, gc_set_debug__doc__},
Tim Peters50c61d52003-04-06 01:50:50 +00001218 {"get_debug", gc_get_debug, METH_NOARGS, gc_get_debug__doc__},
Barry Warsawd3c38ff2006-03-07 09:46:03 +00001219 {"get_count", gc_get_count, METH_NOARGS, gc_get_count__doc__},
Vladimir Marangozovf9d20c32000-08-06 22:45:31 +00001220 {"set_threshold", gc_set_thresh, METH_VARARGS, gc_set_thresh__doc__},
Tim Peters50c61d52003-04-06 01:50:50 +00001221 {"get_threshold", gc_get_thresh, METH_NOARGS, gc_get_thresh__doc__},
Barry Warsawd3c38ff2006-03-07 09:46:03 +00001222 {"collect", (PyCFunction)gc_collect,
1223 METH_VARARGS | METH_KEYWORDS, gc_collect__doc__},
Tim Peters50c61d52003-04-06 01:50:50 +00001224 {"get_objects", gc_get_objects,METH_NOARGS, gc_get_objects__doc__},
Martin v. Löwis560da622001-11-24 09:24:51 +00001225 {"get_referrers", gc_get_referrers, METH_VARARGS,
1226 gc_get_referrers__doc__},
Tim Peters730f5532003-04-08 17:17:17 +00001227 {"get_referents", gc_get_referents, METH_VARARGS,
1228 gc_get_referents__doc__},
Jeremy Hyltonc5007aa2000-06-30 05:02:53 +00001229 {NULL, NULL} /* Sentinel */
1230};
1231
Jason Tishler6bc06ec2003-09-04 11:59:50 +00001232PyMODINIT_FUNC
Jeremy Hyltonc5007aa2000-06-30 05:02:53 +00001233initgc(void)
1234{
1235 PyObject *m;
Jeremy Hyltonc5007aa2000-06-30 05:02:53 +00001236
1237 m = Py_InitModule4("gc",
1238 GcMethods,
1239 gc__doc__,
1240 NULL,
1241 PYTHON_API_VERSION);
Neal Norwitz1ac754f2006-01-19 06:09:39 +00001242 if (m == NULL)
1243 return;
Tim Peters11558872003-04-06 23:30:52 +00001244
Jeremy Hyltonc5007aa2000-06-30 05:02:53 +00001245 if (garbage == NULL) {
1246 garbage = PyList_New(0);
Tim Peters11558872003-04-06 23:30:52 +00001247 if (garbage == NULL)
1248 return;
Jeremy Hyltonc5007aa2000-06-30 05:02:53 +00001249 }
Neil Schemenauer3b1cbf92005-06-18 17:37:06 +00001250 Py_INCREF(garbage);
Tim Peters11558872003-04-06 23:30:52 +00001251 if (PyModule_AddObject(m, "garbage", garbage) < 0)
1252 return;
Neal Norwitz57a03612006-04-26 05:34:03 +00001253
1254 /* Importing can't be done in collect() because collect()
1255 * can be called via PyGC_Collect() in Py_Finalize().
1256 * This wouldn't be a problem, except that <initialized> is
1257 * reset to 0 before calling collect which trips up
1258 * the import and triggers an assertion.
1259 */
1260 if (tmod == NULL) {
Christian Heimes000a0742008-01-03 22:16:32 +00001261 tmod = PyImport_ImportModuleNoBlock("time");
Neal Norwitz57a03612006-04-26 05:34:03 +00001262 if (tmod == NULL)
1263 PyErr_Clear();
1264 }
1265
Tim Peters11558872003-04-06 23:30:52 +00001266#define ADD_INT(NAME) if (PyModule_AddIntConstant(m, #NAME, NAME) < 0) return
1267 ADD_INT(DEBUG_STATS);
1268 ADD_INT(DEBUG_COLLECTABLE);
1269 ADD_INT(DEBUG_UNCOLLECTABLE);
1270 ADD_INT(DEBUG_INSTANCES);
1271 ADD_INT(DEBUG_OBJECTS);
1272 ADD_INT(DEBUG_SAVEALL);
1273 ADD_INT(DEBUG_LEAK);
1274#undef ADD_INT
Jeremy Hyltonc5007aa2000-06-30 05:02:53 +00001275}
1276
Guido van Rossume13ddc92003-04-17 17:29:22 +00001277/* API to invoke gc.collect() from C */
Neal Norwitz7b216c52006-03-04 20:01:53 +00001278Py_ssize_t
Guido van Rossume13ddc92003-04-17 17:29:22 +00001279PyGC_Collect(void)
1280{
Neal Norwitz7b216c52006-03-04 20:01:53 +00001281 Py_ssize_t n;
Guido van Rossume13ddc92003-04-17 17:29:22 +00001282
1283 if (collecting)
1284 n = 0; /* already collecting, don't do anything */
1285 else {
1286 collecting = 1;
1287 n = collect(NUM_GENERATIONS - 1);
1288 collecting = 0;
1289 }
1290
1291 return n;
1292}
1293
Neil Schemenauer43411b52001-08-30 00:05:51 +00001294/* for debugging */
Guido van Rossume13ddc92003-04-17 17:29:22 +00001295void
1296_PyGC_Dump(PyGC_Head *g)
Neil Schemenauer43411b52001-08-30 00:05:51 +00001297{
1298 _PyObject_Dump(FROM_GC(g));
1299}
1300
Neil Schemenauer43411b52001-08-30 00:05:51 +00001301/* extension modules might be compiled with GC support so these
1302 functions must always be available */
1303
Neil Schemenauerfec4eb12002-04-12 02:41:03 +00001304#undef PyObject_GC_Track
1305#undef PyObject_GC_UnTrack
1306#undef PyObject_GC_Del
1307#undef _PyObject_GC_Malloc
1308
Neil Schemenauer43411b52001-08-30 00:05:51 +00001309void
Neil Schemenauerfec4eb12002-04-12 02:41:03 +00001310PyObject_GC_Track(void *op)
Neil Schemenauer43411b52001-08-30 00:05:51 +00001311{
1312 _PyObject_GC_TRACK(op);
1313}
1314
Neil Schemenauerfec4eb12002-04-12 02:41:03 +00001315/* for binary compatibility with 2.2 */
Neil Schemenauer43411b52001-08-30 00:05:51 +00001316void
Neil Schemenauerfec4eb12002-04-12 02:41:03 +00001317_PyObject_GC_Track(PyObject *op)
1318{
1319 PyObject_GC_Track(op);
1320}
1321
1322void
1323PyObject_GC_UnTrack(void *op)
Neil Schemenauer43411b52001-08-30 00:05:51 +00001324{
Tim Peters803526b2002-07-07 05:13:56 +00001325 /* Obscure: the Py_TRASHCAN mechanism requires that we be able to
1326 * call PyObject_GC_UnTrack twice on an object.
1327 */
Neil Schemenauera2b11ec2002-05-21 15:53:24 +00001328 if (IS_TRACKED(op))
Guido van Rossumff413af2002-03-28 20:34:59 +00001329 _PyObject_GC_UNTRACK(op);
Neil Schemenauer43411b52001-08-30 00:05:51 +00001330}
1331
Neil Schemenauerfec4eb12002-04-12 02:41:03 +00001332/* for binary compatibility with 2.2 */
1333void
1334_PyObject_GC_UnTrack(PyObject *op)
1335{
1336 PyObject_GC_UnTrack(op);
1337}
1338
Neil Schemenauer43411b52001-08-30 00:05:51 +00001339PyObject *
Neil Schemenauerfec4eb12002-04-12 02:41:03 +00001340_PyObject_GC_Malloc(size_t basicsize)
Neil Schemenauer43411b52001-08-30 00:05:51 +00001341{
1342 PyObject *op;
Anthony Baxter64182fe2006-04-11 12:14:09 +00001343 PyGC_Head *g = (PyGC_Head *)PyObject_MALLOC(
1344 sizeof(PyGC_Head) + basicsize);
Neil Schemenauer43411b52001-08-30 00:05:51 +00001345 if (g == NULL)
Jeremy Hylton8a135182002-06-06 23:23:55 +00001346 return PyErr_NoMemory();
Tim Petersea405632002-07-02 00:52:30 +00001347 g->gc.gc_refs = GC_UNTRACKED;
Neil Schemenauer2880ae52002-05-04 05:35:20 +00001348 generations[0].count++; /* number of allocated GC objects */
1349 if (generations[0].count > generations[0].threshold &&
Neil Schemenauer43411b52001-08-30 00:05:51 +00001350 enabled &&
Neil Schemenauer2880ae52002-05-04 05:35:20 +00001351 generations[0].threshold &&
Neil Schemenauer43411b52001-08-30 00:05:51 +00001352 !collecting &&
1353 !PyErr_Occurred()) {
1354 collecting = 1;
Neil Schemenauer2880ae52002-05-04 05:35:20 +00001355 collect_generations();
Neil Schemenauer43411b52001-08-30 00:05:51 +00001356 collecting = 0;
1357 }
1358 op = FROM_GC(g);
Neil Schemenauer43411b52001-08-30 00:05:51 +00001359 return op;
1360}
1361
1362PyObject *
1363_PyObject_GC_New(PyTypeObject *tp)
1364{
Neil Schemenauerfec4eb12002-04-12 02:41:03 +00001365 PyObject *op = _PyObject_GC_Malloc(_PyObject_SIZE(tp));
Tim Petersfa8efab2002-04-28 01:57:25 +00001366 if (op != NULL)
1367 op = PyObject_INIT(op, tp);
1368 return op;
Neil Schemenauer43411b52001-08-30 00:05:51 +00001369}
1370
1371PyVarObject *
Martin v. Löwis18e16552006-02-15 17:27:45 +00001372_PyObject_GC_NewVar(PyTypeObject *tp, Py_ssize_t nitems)
Neil Schemenauer43411b52001-08-30 00:05:51 +00001373{
Neil Schemenauerfec4eb12002-04-12 02:41:03 +00001374 const size_t size = _PyObject_VAR_SIZE(tp, nitems);
1375 PyVarObject *op = (PyVarObject *) _PyObject_GC_Malloc(size);
Tim Petersfa8efab2002-04-28 01:57:25 +00001376 if (op != NULL)
1377 op = PyObject_INIT_VAR(op, tp, nitems);
1378 return op;
Neil Schemenauer43411b52001-08-30 00:05:51 +00001379}
1380
1381PyVarObject *
Martin v. Löwis41290682006-02-16 14:56:14 +00001382_PyObject_GC_Resize(PyVarObject *op, Py_ssize_t nitems)
Neil Schemenauer43411b52001-08-30 00:05:51 +00001383{
Christian Heimese93237d2007-12-19 02:37:44 +00001384 const size_t basicsize = _PyObject_VAR_SIZE(Py_TYPE(op), nitems);
Neil Schemenauer43411b52001-08-30 00:05:51 +00001385 PyGC_Head *g = AS_GC(op);
Anthony Baxter64182fe2006-04-11 12:14:09 +00001386 g = (PyGC_Head *)PyObject_REALLOC(g, sizeof(PyGC_Head) + basicsize);
Neil Schemenauer43411b52001-08-30 00:05:51 +00001387 if (g == NULL)
1388 return (PyVarObject *)PyErr_NoMemory();
1389 op = (PyVarObject *) FROM_GC(g);
Christian Heimese93237d2007-12-19 02:37:44 +00001390 Py_SIZE(op) = nitems;
Neil Schemenauer43411b52001-08-30 00:05:51 +00001391 return op;
1392}
1393
1394void
Neil Schemenauerfec4eb12002-04-12 02:41:03 +00001395PyObject_GC_Del(void *op)
Neil Schemenauer43411b52001-08-30 00:05:51 +00001396{
Neil Schemenauer43411b52001-08-30 00:05:51 +00001397 PyGC_Head *g = AS_GC(op);
Neil Schemenauera2b11ec2002-05-21 15:53:24 +00001398 if (IS_TRACKED(op))
Neil Schemenauer43411b52001-08-30 00:05:51 +00001399 gc_list_remove(g);
Neil Schemenauer2880ae52002-05-04 05:35:20 +00001400 if (generations[0].count > 0) {
1401 generations[0].count--;
Neil Schemenauer43411b52001-08-30 00:05:51 +00001402 }
Neil Schemenauerfec4eb12002-04-12 02:41:03 +00001403 PyObject_FREE(g);
Neil Schemenauer43411b52001-08-30 00:05:51 +00001404}
1405
Neil Schemenauerfec4eb12002-04-12 02:41:03 +00001406/* for binary compatibility with 2.2 */
1407#undef _PyObject_GC_Del
1408void
1409_PyObject_GC_Del(PyObject *op)
1410{
1411 PyObject_GC_Del(op);
1412}