blob: 6f12972200b3ec7ce1ef64addf6b14b8d3faa528 [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;
Gregory P. Smithdd96db62008-06-09 04:58:54 +0000642 if (classname != NULL && PyString_Check(classname))
643 cname = PyString_AsString(classname);
Jeremy Hyltonc5007aa2000-06-30 05:02:53 +0000644 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();
Gregory P. Smith2fe77062008-07-06 03:35:58 +0000739 (void)PyInt_ClearFreeList();
740 (void)PyFloat_ClearFreeList();
Christian Heimes3b718a72008-02-14 12:47:33 +0000741}
742
Jeremy Hyltonc5007aa2000-06-30 05:02:53 +0000743/* This is the main function. Read this to understand how the
744 * collection process works. */
Neal Norwitz7b216c52006-03-04 20:01:53 +0000745static Py_ssize_t
Neil Schemenauer2880ae52002-05-04 05:35:20 +0000746collect(int generation)
Jeremy Hyltonc5007aa2000-06-30 05:02:53 +0000747{
Neil Schemenauer2880ae52002-05-04 05:35:20 +0000748 int i;
Neal Norwitz7b216c52006-03-04 20:01:53 +0000749 Py_ssize_t m = 0; /* # objects collected */
750 Py_ssize_t n = 0; /* # unreachable objects that couldn't be collected */
Neil Schemenauer2880ae52002-05-04 05:35:20 +0000751 PyGC_Head *young; /* the generation we are examining */
752 PyGC_Head *old; /* next older generation */
Tim Peters403a2032003-11-20 21:21:46 +0000753 PyGC_Head unreachable; /* non-problematic unreachable trash */
754 PyGC_Head finalizers; /* objects with, & reachable from, __del__ */
Jeremy Hyltonc5007aa2000-06-30 05:02:53 +0000755 PyGC_Head *gc;
Skip Montanaroc34b9312006-04-21 01:33:40 +0000756 double t1 = 0.0;
Jeremy Hyltonc5007aa2000-06-30 05:02:53 +0000757
Tim Peters93ad66d2003-04-05 17:15:44 +0000758 if (delstr == NULL) {
Gregory P. Smithdd96db62008-06-09 04:58:54 +0000759 delstr = PyString_InternFromString("__del__");
Tim Peters93ad66d2003-04-05 17:15:44 +0000760 if (delstr == NULL)
761 Py_FatalError("gc couldn't allocate \"__del__\"");
762 }
763
Jeremy Hyltonc5007aa2000-06-30 05:02:53 +0000764 if (debug & DEBUG_STATS) {
Skip Montanaroc34b9312006-04-21 01:33:40 +0000765 if (tmod != NULL) {
766 PyObject *f = PyObject_CallMethod(tmod, "time", NULL);
767 if (f == NULL) {
768 PyErr_Clear();
769 }
770 else {
771 t1 = PyFloat_AsDouble(f);
772 Py_DECREF(f);
773 }
774 }
Neil Schemenauer2880ae52002-05-04 05:35:20 +0000775 PySys_WriteStderr("gc: collecting generation %d...\n",
776 generation);
777 PySys_WriteStderr("gc: objects in each generation:");
Tim Peters62e97f02006-03-28 21:44:32 +0000778 for (i = 0; i < NUM_GENERATIONS; i++)
779 PySys_WriteStderr(" %" PY_FORMAT_SIZE_T "d",
780 gc_list_size(GEN_HEAD(i)));
Neil Schemenauer2880ae52002-05-04 05:35:20 +0000781 PySys_WriteStderr("\n");
782 }
783
784 /* update collection and allocation counters */
785 if (generation+1 < NUM_GENERATIONS)
786 generations[generation+1].count += 1;
787 for (i = 0; i <= generation; i++)
Neil Schemenauerc9051642002-06-28 19:16:04 +0000788 generations[i].count = 0;
Neil Schemenauer2880ae52002-05-04 05:35:20 +0000789
790 /* merge younger generations with one we are currently collecting */
791 for (i = 0; i < generation; i++) {
792 gc_list_merge(GEN_HEAD(i), GEN_HEAD(generation));
793 }
794
795 /* handy references */
796 young = GEN_HEAD(generation);
Tim Peters19b74c72002-07-01 03:52:19 +0000797 if (generation < NUM_GENERATIONS-1)
Neil Schemenauer2880ae52002-05-04 05:35:20 +0000798 old = GEN_HEAD(generation+1);
Tim Peters19b74c72002-07-01 03:52:19 +0000799 else
800 old = young;
Jeremy Hyltonc5007aa2000-06-30 05:02:53 +0000801
802 /* Using ob_refcnt and gc_refs, calculate which objects in the
Tim Petersead8b7a2004-10-30 23:09:22 +0000803 * container set are reachable from outside the set (i.e., have a
Jeremy Hyltonc5007aa2000-06-30 05:02:53 +0000804 * refcount greater than 0 when all the references within the
Tim Petersead8b7a2004-10-30 23:09:22 +0000805 * set are taken into account).
Tim Peters19b74c72002-07-01 03:52:19 +0000806 */
Jeremy Hyltonc5007aa2000-06-30 05:02:53 +0000807 update_refs(young);
808 subtract_refs(young);
809
Tim Peters19b74c72002-07-01 03:52:19 +0000810 /* Leave everything reachable from outside young in young, and move
811 * everything else (in young) to unreachable.
812 * NOTE: This used to move the reachable objects into a reachable
813 * set instead. But most things usually turn out to be reachable,
814 * so it's more efficient to move the unreachable things.
815 */
Jeremy Hyltonc5007aa2000-06-30 05:02:53 +0000816 gc_list_init(&unreachable);
Tim Peters19b74c72002-07-01 03:52:19 +0000817 move_unreachable(young, &unreachable);
Jeremy Hyltonc5007aa2000-06-30 05:02:53 +0000818
Tim Peters19b74c72002-07-01 03:52:19 +0000819 /* Move reachable objects to next generation. */
820 if (young != old)
821 gc_list_merge(young, old);
Jeremy Hyltonc5007aa2000-06-30 05:02:53 +0000822
Tim Peters19b74c72002-07-01 03:52:19 +0000823 /* All objects in unreachable are trash, but objects reachable from
824 * finalizers can't safely be deleted. Python programmers should take
825 * care not to create such things. For Python, finalizers means
Tim Peters403a2032003-11-20 21:21:46 +0000826 * instance objects with __del__ methods. Weakrefs with callbacks
Tim Petersead8b7a2004-10-30 23:09:22 +0000827 * can also call arbitrary Python code but they will be dealt with by
828 * handle_weakrefs().
Tim Petersf6b80452003-04-07 19:21:15 +0000829 */
Jeremy Hyltonc5007aa2000-06-30 05:02:53 +0000830 gc_list_init(&finalizers);
Tim Petersead8b7a2004-10-30 23:09:22 +0000831 move_finalizers(&unreachable, &finalizers);
Tim Petersbf384c22003-04-06 00:11:39 +0000832 /* finalizers contains the unreachable objects with a finalizer;
Tim Peters403a2032003-11-20 21:21:46 +0000833 * unreachable objects reachable *from* those are also uncollectable,
834 * and we move those into the finalizers list too.
Tim Petersbf384c22003-04-06 00:11:39 +0000835 */
Tim Petersf6b80452003-04-07 19:21:15 +0000836 move_finalizer_reachable(&finalizers);
Jeremy Hyltonc5007aa2000-06-30 05:02:53 +0000837
838 /* Collect statistics on collectable objects found and print
Tim Peters403a2032003-11-20 21:21:46 +0000839 * debugging information.
840 */
Tim Petersf6b80452003-04-07 19:21:15 +0000841 for (gc = unreachable.gc.gc_next; gc != &unreachable;
Tim Peters9e4ca102001-10-11 18:31:31 +0000842 gc = gc->gc.gc_next) {
Jeremy Hyltonc5007aa2000-06-30 05:02:53 +0000843 m++;
Jeremy Hylton06257772000-08-31 15:10:24 +0000844 if (debug & DEBUG_COLLECTABLE) {
Neil Schemenauer43411b52001-08-30 00:05:51 +0000845 debug_cycle("collectable", FROM_GC(gc));
Jeremy Hyltonc5007aa2000-06-30 05:02:53 +0000846 }
Skip Montanaroc34b9312006-04-21 01:33:40 +0000847 if (tmod != NULL && (debug & DEBUG_STATS)) {
848 PyObject *f = PyObject_CallMethod(tmod, "time", NULL);
849 if (f == NULL) {
850 PyErr_Clear();
851 }
852 else {
853 t1 = PyFloat_AsDouble(f)-t1;
854 Py_DECREF(f);
855 PySys_WriteStderr("gc: %.4fs elapsed.\n", t1);
856 }
857 }
Jeremy Hyltonc5007aa2000-06-30 05:02:53 +0000858 }
Tim Petersead8b7a2004-10-30 23:09:22 +0000859
860 /* Clear weakrefs and invoke callbacks as necessary. */
861 m += handle_weakrefs(&unreachable, old);
862
Tim Petersfb2ab4d2003-04-07 22:41:24 +0000863 /* Call tp_clear on objects in the unreachable set. This will cause
864 * the reference cycles to be broken. It may also cause some objects
865 * in finalizers to be freed.
866 */
Tim Petersf6b80452003-04-07 19:21:15 +0000867 delete_garbage(&unreachable, old);
Jeremy Hyltonc5007aa2000-06-30 05:02:53 +0000868
869 /* Collect statistics on uncollectable objects found and print
870 * debugging information. */
Tim Peters50c61d52003-04-06 01:50:50 +0000871 for (gc = finalizers.gc.gc_next;
Tim Petersbf384c22003-04-06 00:11:39 +0000872 gc != &finalizers;
873 gc = gc->gc.gc_next) {
Jeremy Hyltonc5007aa2000-06-30 05:02:53 +0000874 n++;
Tim Petersbf384c22003-04-06 00:11:39 +0000875 if (debug & DEBUG_UNCOLLECTABLE)
Neil Schemenauer43411b52001-08-30 00:05:51 +0000876 debug_cycle("uncollectable", FROM_GC(gc));
Tim Petersbf384c22003-04-06 00:11:39 +0000877 }
Jeremy Hylton06257772000-08-31 15:10:24 +0000878 if (debug & DEBUG_STATS) {
Tim Peters62e97f02006-03-28 21:44:32 +0000879 if (m == 0 && n == 0)
Jeremy Hylton06257772000-08-31 15:10:24 +0000880 PySys_WriteStderr("gc: done.\n");
Tim Peters62e97f02006-03-28 21:44:32 +0000881 else
Neal Norwitze22373d2006-03-06 23:31:56 +0000882 PySys_WriteStderr(
Tim Peters62e97f02006-03-28 21:44:32 +0000883 "gc: done, "
884 "%" PY_FORMAT_SIZE_T "d unreachable, "
885 "%" PY_FORMAT_SIZE_T "d uncollectable.\n",
Neal Norwitze22373d2006-03-06 23:31:56 +0000886 n+m, n);
Jeremy Hyltonc5007aa2000-06-30 05:02:53 +0000887 }
888
889 /* Append instances in the uncollectable set to a Python
890 * reachable list of garbage. The programmer has to deal with
Tim Petersbf384c22003-04-06 00:11:39 +0000891 * this if they insist on creating this type of structure.
892 */
Tim Petersf6b80452003-04-07 19:21:15 +0000893 (void)handle_finalizers(&finalizers, old);
Jeremy Hyltonc5007aa2000-06-30 05:02:53 +0000894
Christian Heimes3b718a72008-02-14 12:47:33 +0000895 /* Clear free list only during the collection of the higest
896 * generation */
897 if (generation == NUM_GENERATIONS-1) {
898 clear_freelists();
899 }
900
Jeremy Hyltonb709df32000-09-01 02:47:25 +0000901 if (PyErr_Occurred()) {
Tim Petersf6b80452003-04-07 19:21:15 +0000902 if (gc_str == NULL)
Gregory P. Smithdd96db62008-06-09 04:58:54 +0000903 gc_str = PyString_FromString("garbage collection");
Jeremy Hyltonb709df32000-09-01 02:47:25 +0000904 PyErr_WriteUnraisable(gc_str);
905 Py_FatalError("unexpected exception during garbage collection");
906 }
Jeremy Hyltonc5007aa2000-06-30 05:02:53 +0000907 return n+m;
908}
909
Neal Norwitz7b216c52006-03-04 20:01:53 +0000910static Py_ssize_t
Jeremy Hyltonc5007aa2000-06-30 05:02:53 +0000911collect_generations(void)
912{
Neil Schemenauer2880ae52002-05-04 05:35:20 +0000913 int i;
Neal Norwitz7b216c52006-03-04 20:01:53 +0000914 Py_ssize_t n = 0;
Jeremy Hyltonc5007aa2000-06-30 05:02:53 +0000915
Neil Schemenauer2880ae52002-05-04 05:35:20 +0000916 /* Find the oldest generation (higest numbered) where the count
917 * exceeds the threshold. Objects in the that generation and
918 * generations younger than it will be collected. */
919 for (i = NUM_GENERATIONS-1; i >= 0; i--) {
920 if (generations[i].count > generations[i].threshold) {
921 n = collect(i);
922 break;
Jeremy Hyltonc5007aa2000-06-30 05:02:53 +0000923 }
924 }
925 return n;
926}
927
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +0000928PyDoc_STRVAR(gc_enable__doc__,
Vladimir Marangozovf9d20c32000-08-06 22:45:31 +0000929"enable() -> None\n"
930"\n"
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +0000931"Enable automatic garbage collection.\n");
Jeremy Hyltonc5007aa2000-06-30 05:02:53 +0000932
Vladimir Marangozovf9d20c32000-08-06 22:45:31 +0000933static PyObject *
Tim Peters50c61d52003-04-06 01:50:50 +0000934gc_enable(PyObject *self, PyObject *noargs)
Vladimir Marangozovf9d20c32000-08-06 22:45:31 +0000935{
Vladimir Marangozovf9d20c32000-08-06 22:45:31 +0000936 enabled = 1;
Vladimir Marangozovf9d20c32000-08-06 22:45:31 +0000937 Py_INCREF(Py_None);
938 return Py_None;
939}
940
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +0000941PyDoc_STRVAR(gc_disable__doc__,
Vladimir Marangozovf9d20c32000-08-06 22:45:31 +0000942"disable() -> None\n"
943"\n"
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +0000944"Disable automatic garbage collection.\n");
Vladimir Marangozovf9d20c32000-08-06 22:45:31 +0000945
946static PyObject *
Tim Peters50c61d52003-04-06 01:50:50 +0000947gc_disable(PyObject *self, PyObject *noargs)
Vladimir Marangozovf9d20c32000-08-06 22:45:31 +0000948{
Vladimir Marangozovf9d20c32000-08-06 22:45:31 +0000949 enabled = 0;
Vladimir Marangozovf9d20c32000-08-06 22:45:31 +0000950 Py_INCREF(Py_None);
951 return Py_None;
952}
953
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +0000954PyDoc_STRVAR(gc_isenabled__doc__,
Vladimir Marangozovf9d20c32000-08-06 22:45:31 +0000955"isenabled() -> status\n"
956"\n"
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +0000957"Returns true if automatic garbage collection is enabled.\n");
Vladimir Marangozovf9d20c32000-08-06 22:45:31 +0000958
959static PyObject *
Tim Peters50c61d52003-04-06 01:50:50 +0000960gc_isenabled(PyObject *self, PyObject *noargs)
Vladimir Marangozovf9d20c32000-08-06 22:45:31 +0000961{
Raymond Hettinger674d56b2004-01-04 04:00:13 +0000962 return PyBool_FromLong((long)enabled);
Vladimir Marangozovf9d20c32000-08-06 22:45:31 +0000963}
964
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +0000965PyDoc_STRVAR(gc_collect__doc__,
Barry Warsawd3c38ff2006-03-07 09:46:03 +0000966"collect([generation]) -> n\n"
Jeremy Hyltonc5007aa2000-06-30 05:02:53 +0000967"\n"
Barry Warsawd3c38ff2006-03-07 09:46:03 +0000968"With no arguments, run a full collection. The optional argument\n"
969"may be an integer specifying which generation to collect. A ValueError\n"
970"is raised if the generation number is invalid.\n\n"
971"The number of unreachable objects is returned.\n");
Jeremy Hyltonc5007aa2000-06-30 05:02:53 +0000972
973static PyObject *
Barry Warsawd3c38ff2006-03-07 09:46:03 +0000974gc_collect(PyObject *self, PyObject *args, PyObject *kws)
Jeremy Hyltonc5007aa2000-06-30 05:02:53 +0000975{
Barry Warsawd3c38ff2006-03-07 09:46:03 +0000976 static char *keywords[] = {"generation", NULL};
977 int genarg = NUM_GENERATIONS - 1;
Neal Norwitz7b216c52006-03-04 20:01:53 +0000978 Py_ssize_t n;
Jeremy Hyltonc5007aa2000-06-30 05:02:53 +0000979
Barry Warsawd3c38ff2006-03-07 09:46:03 +0000980 if (!PyArg_ParseTupleAndKeywords(args, kws, "|i", keywords, &genarg))
981 return NULL;
982
983 else if (genarg < 0 || genarg >= NUM_GENERATIONS) {
984 PyErr_SetString(PyExc_ValueError, "invalid generation");
985 return NULL;
986 }
987
Tim Peters50c61d52003-04-06 01:50:50 +0000988 if (collecting)
Neil Schemenauere8c40cb2001-10-31 23:09:35 +0000989 n = 0; /* already collecting, don't do anything */
Neil Schemenauere8c40cb2001-10-31 23:09:35 +0000990 else {
991 collecting = 1;
Barry Warsawd3c38ff2006-03-07 09:46:03 +0000992 n = collect(genarg);
Neil Schemenauere8c40cb2001-10-31 23:09:35 +0000993 collecting = 0;
994 }
Jeremy Hyltonc5007aa2000-06-30 05:02:53 +0000995
Neal Norwitz7b216c52006-03-04 20:01:53 +0000996 return PyInt_FromSsize_t(n);
Jeremy Hyltonc5007aa2000-06-30 05:02:53 +0000997}
998
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +0000999PyDoc_STRVAR(gc_set_debug__doc__,
Jeremy Hyltonc5007aa2000-06-30 05:02:53 +00001000"set_debug(flags) -> None\n"
1001"\n"
1002"Set the garbage collection debugging flags. Debugging information is\n"
1003"written to sys.stderr.\n"
1004"\n"
1005"flags is an integer and can have the following bits turned on:\n"
1006"\n"
1007" DEBUG_STATS - Print statistics during collection.\n"
1008" DEBUG_COLLECTABLE - Print collectable objects found.\n"
1009" DEBUG_UNCOLLECTABLE - Print unreachable but uncollectable objects found.\n"
1010" DEBUG_INSTANCES - Print instance objects.\n"
1011" DEBUG_OBJECTS - Print objects other than instances.\n"
Neil Schemenauer544de1e2000-09-22 15:22:38 +00001012" DEBUG_SAVEALL - Save objects to gc.garbage rather than freeing them.\n"
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00001013" DEBUG_LEAK - Debug leaking programs (everything but STATS).\n");
Jeremy Hyltonc5007aa2000-06-30 05:02:53 +00001014
1015static PyObject *
Vladimir Marangozovf9d20c32000-08-06 22:45:31 +00001016gc_set_debug(PyObject *self, PyObject *args)
Jeremy Hyltonc5007aa2000-06-30 05:02:53 +00001017{
Neil Schemenauer7760cff2000-09-22 22:35:36 +00001018 if (!PyArg_ParseTuple(args, "i:set_debug", &debug))
Jeremy Hyltonc5007aa2000-06-30 05:02:53 +00001019 return NULL;
1020
1021 Py_INCREF(Py_None);
1022 return Py_None;
1023}
1024
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00001025PyDoc_STRVAR(gc_get_debug__doc__,
Jeremy Hyltonc5007aa2000-06-30 05:02:53 +00001026"get_debug() -> flags\n"
1027"\n"
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00001028"Get the garbage collection debugging flags.\n");
Jeremy Hyltonc5007aa2000-06-30 05:02:53 +00001029
1030static PyObject *
Tim Peters50c61d52003-04-06 01:50:50 +00001031gc_get_debug(PyObject *self, PyObject *noargs)
Jeremy Hyltonc5007aa2000-06-30 05:02:53 +00001032{
Jeremy Hyltonc5007aa2000-06-30 05:02:53 +00001033 return Py_BuildValue("i", debug);
1034}
1035
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00001036PyDoc_STRVAR(gc_set_thresh__doc__,
Neal Norwitz2a47c0f2002-01-29 00:53:41 +00001037"set_threshold(threshold0, [threshold1, threshold2]) -> None\n"
Jeremy Hyltonc5007aa2000-06-30 05:02:53 +00001038"\n"
1039"Sets the collection thresholds. Setting threshold0 to zero disables\n"
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00001040"collection.\n");
Jeremy Hyltonc5007aa2000-06-30 05:02:53 +00001041
1042static PyObject *
Vladimir Marangozovf9d20c32000-08-06 22:45:31 +00001043gc_set_thresh(PyObject *self, PyObject *args)
Jeremy Hyltonc5007aa2000-06-30 05:02:53 +00001044{
Neil Schemenauer2880ae52002-05-04 05:35:20 +00001045 int i;
1046 if (!PyArg_ParseTuple(args, "i|ii:set_threshold",
1047 &generations[0].threshold,
1048 &generations[1].threshold,
1049 &generations[2].threshold))
Jeremy Hyltonc5007aa2000-06-30 05:02:53 +00001050 return NULL;
Neil Schemenauer2880ae52002-05-04 05:35:20 +00001051 for (i = 2; i < NUM_GENERATIONS; i++) {
1052 /* generations higher than 2 get the same threshold */
1053 generations[i].threshold = generations[2].threshold;
1054 }
Jeremy Hyltonc5007aa2000-06-30 05:02:53 +00001055
1056 Py_INCREF(Py_None);
1057 return Py_None;
1058}
1059
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00001060PyDoc_STRVAR(gc_get_thresh__doc__,
Jeremy Hyltonc5007aa2000-06-30 05:02:53 +00001061"get_threshold() -> (threshold0, threshold1, threshold2)\n"
1062"\n"
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00001063"Return the current collection thresholds\n");
Jeremy Hyltonc5007aa2000-06-30 05:02:53 +00001064
1065static PyObject *
Tim Peters50c61d52003-04-06 01:50:50 +00001066gc_get_thresh(PyObject *self, PyObject *noargs)
Jeremy Hyltonc5007aa2000-06-30 05:02:53 +00001067{
Neil Schemenauer2880ae52002-05-04 05:35:20 +00001068 return Py_BuildValue("(iii)",
1069 generations[0].threshold,
1070 generations[1].threshold,
1071 generations[2].threshold);
Jeremy Hyltonc5007aa2000-06-30 05:02:53 +00001072}
1073
Barry Warsawd3c38ff2006-03-07 09:46:03 +00001074PyDoc_STRVAR(gc_get_count__doc__,
1075"get_count() -> (count0, count1, count2)\n"
1076"\n"
1077"Return the current collection counts\n");
1078
1079static PyObject *
1080gc_get_count(PyObject *self, PyObject *noargs)
1081{
1082 return Py_BuildValue("(iii)",
1083 generations[0].count,
1084 generations[1].count,
1085 generations[2].count);
1086}
1087
Neil Schemenauer48c70342001-08-09 15:38:31 +00001088static int
Martin v. Löwis560da622001-11-24 09:24:51 +00001089referrersvisit(PyObject* obj, PyObject *objs)
Neil Schemenauer48c70342001-08-09 15:38:31 +00001090{
Neal Norwitzffb0d902006-04-06 08:07:25 +00001091 Py_ssize_t i;
Martin v. Löwisc8fe77b2001-11-29 18:08:31 +00001092 for (i = 0; i < PyTuple_GET_SIZE(objs); i++)
1093 if (PyTuple_GET_ITEM(objs, i) == obj)
1094 return 1;
Neil Schemenauer48c70342001-08-09 15:38:31 +00001095 return 0;
1096}
1097
Neil Schemenauer17e7be62001-08-10 14:46:47 +00001098static int
Martin v. Löwis560da622001-11-24 09:24:51 +00001099gc_referrers_for(PyObject *objs, PyGC_Head *list, PyObject *resultlist)
Neil Schemenauer48c70342001-08-09 15:38:31 +00001100{
1101 PyGC_Head *gc;
1102 PyObject *obj;
1103 traverseproc traverse;
Tim Peters9e4ca102001-10-11 18:31:31 +00001104 for (gc = list->gc.gc_next; gc != list; gc = gc->gc.gc_next) {
Neil Schemenauer43411b52001-08-30 00:05:51 +00001105 obj = FROM_GC(gc);
Christian Heimese93237d2007-12-19 02:37:44 +00001106 traverse = Py_TYPE(obj)->tp_traverse;
Neil Schemenauer48c70342001-08-09 15:38:31 +00001107 if (obj == objs || obj == resultlist)
1108 continue;
Martin v. Löwis560da622001-11-24 09:24:51 +00001109 if (traverse(obj, (visitproc)referrersvisit, objs)) {
Neil Schemenauer17e7be62001-08-10 14:46:47 +00001110 if (PyList_Append(resultlist, obj) < 0)
1111 return 0; /* error */
Neil Schemenauer48c70342001-08-09 15:38:31 +00001112 }
1113 }
Neil Schemenauer17e7be62001-08-10 14:46:47 +00001114 return 1; /* no error */
Neil Schemenauer48c70342001-08-09 15:38:31 +00001115}
1116
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00001117PyDoc_STRVAR(gc_get_referrers__doc__,
Martin v. Löwis560da622001-11-24 09:24:51 +00001118"get_referrers(*objs) -> list\n\
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00001119Return the list of objects that directly refer to any of objs.");
Neil Schemenauer48c70342001-08-09 15:38:31 +00001120
Neil Schemenauer17e7be62001-08-10 14:46:47 +00001121static PyObject *
Martin v. Löwis560da622001-11-24 09:24:51 +00001122gc_get_referrers(PyObject *self, PyObject *args)
Neil Schemenauer48c70342001-08-09 15:38:31 +00001123{
Neil Schemenauer2880ae52002-05-04 05:35:20 +00001124 int i;
Neil Schemenauer48c70342001-08-09 15:38:31 +00001125 PyObject *result = PyList_New(0);
Georg Brandl5c170fd2006-03-17 19:03:25 +00001126 if (!result) return NULL;
1127
Neil Schemenauer2880ae52002-05-04 05:35:20 +00001128 for (i = 0; i < NUM_GENERATIONS; i++) {
1129 if (!(gc_referrers_for(args, GEN_HEAD(i), result))) {
1130 Py_DECREF(result);
1131 return NULL;
1132 }
Neil Schemenauer17e7be62001-08-10 14:46:47 +00001133 }
Neil Schemenauer48c70342001-08-09 15:38:31 +00001134 return result;
1135}
Jeremy Hyltonc5007aa2000-06-30 05:02:53 +00001136
Tim Peters0f81ab62003-04-08 16:39:48 +00001137/* Append obj to list; return true if error (out of memory), false if OK. */
Jeremy Hylton5bd378b2003-04-03 16:28:38 +00001138static int
Tim Peters730f5532003-04-08 17:17:17 +00001139referentsvisit(PyObject *obj, PyObject *list)
Jeremy Hylton5bd378b2003-04-03 16:28:38 +00001140{
Tim Peters0f81ab62003-04-08 16:39:48 +00001141 return PyList_Append(list, obj) < 0;
Jeremy Hylton5bd378b2003-04-03 16:28:38 +00001142}
1143
Tim Peters730f5532003-04-08 17:17:17 +00001144PyDoc_STRVAR(gc_get_referents__doc__,
1145"get_referents(*objs) -> list\n\
Jeremy Hylton059b0942003-04-03 16:29:13 +00001146Return the list of objects that are directly referred to by objs.");
Jeremy Hylton5bd378b2003-04-03 16:28:38 +00001147
1148static PyObject *
Tim Peters730f5532003-04-08 17:17:17 +00001149gc_get_referents(PyObject *self, PyObject *args)
Jeremy Hylton5bd378b2003-04-03 16:28:38 +00001150{
Neal Norwitzffb0d902006-04-06 08:07:25 +00001151 Py_ssize_t i;
Jeremy Hylton5bd378b2003-04-03 16:28:38 +00001152 PyObject *result = PyList_New(0);
Tim Peters0f81ab62003-04-08 16:39:48 +00001153
1154 if (result == NULL)
1155 return NULL;
1156
Jeremy Hylton5bd378b2003-04-03 16:28:38 +00001157 for (i = 0; i < PyTuple_GET_SIZE(args); i++) {
Tim Peters0f81ab62003-04-08 16:39:48 +00001158 traverseproc traverse;
Tim Peters93ad66d2003-04-05 17:15:44 +00001159 PyObject *obj = PyTuple_GET_ITEM(args, i);
Tim Peters0f81ab62003-04-08 16:39:48 +00001160
1161 if (! PyObject_IS_GC(obj))
Jeremy Hylton5bd378b2003-04-03 16:28:38 +00001162 continue;
Christian Heimese93237d2007-12-19 02:37:44 +00001163 traverse = Py_TYPE(obj)->tp_traverse;
Tim Peters0f81ab62003-04-08 16:39:48 +00001164 if (! traverse)
1165 continue;
Tim Peters730f5532003-04-08 17:17:17 +00001166 if (traverse(obj, (visitproc)referentsvisit, result)) {
Tim Peters0f81ab62003-04-08 16:39:48 +00001167 Py_DECREF(result);
Jeremy Hylton5bd378b2003-04-03 16:28:38 +00001168 return NULL;
Tim Peters0f81ab62003-04-08 16:39:48 +00001169 }
Jeremy Hylton5bd378b2003-04-03 16:28:38 +00001170 }
1171 return result;
1172}
1173
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00001174PyDoc_STRVAR(gc_get_objects__doc__,
Neil Schemenauerc7c8d8e2001-08-09 15:58:59 +00001175"get_objects() -> [...]\n"
1176"\n"
1177"Return a list of objects tracked by the collector (excluding the list\n"
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00001178"returned).\n");
Neil Schemenauerc7c8d8e2001-08-09 15:58:59 +00001179
Neil Schemenauerc7c8d8e2001-08-09 15:58:59 +00001180static PyObject *
Tim Peters50c61d52003-04-06 01:50:50 +00001181gc_get_objects(PyObject *self, PyObject *noargs)
Neil Schemenauerc7c8d8e2001-08-09 15:58:59 +00001182{
Neil Schemenauer2880ae52002-05-04 05:35:20 +00001183 int i;
Neil Schemenauerc7c8d8e2001-08-09 15:58:59 +00001184 PyObject* result;
1185
Neil Schemenauerc7c8d8e2001-08-09 15:58:59 +00001186 result = PyList_New(0);
Tim Peters50c61d52003-04-06 01:50:50 +00001187 if (result == NULL)
Martin v. Löwisf8a6f242001-12-02 18:31:02 +00001188 return NULL;
Neil Schemenauer2880ae52002-05-04 05:35:20 +00001189 for (i = 0; i < NUM_GENERATIONS; i++) {
1190 if (append_objects(result, GEN_HEAD(i))) {
1191 Py_DECREF(result);
1192 return NULL;
1193 }
Martin v. Löwis155aad12001-12-02 12:21:34 +00001194 }
Neil Schemenauerc7c8d8e2001-08-09 15:58:59 +00001195 return result;
1196}
1197
1198
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00001199PyDoc_STRVAR(gc__doc__,
Jeremy Hyltonc5007aa2000-06-30 05:02:53 +00001200"This module provides access to the garbage collector for reference cycles.\n"
1201"\n"
Vladimir Marangozovf9d20c32000-08-06 22:45:31 +00001202"enable() -- Enable automatic garbage collection.\n"
1203"disable() -- Disable automatic garbage collection.\n"
1204"isenabled() -- Returns true if automatic collection is enabled.\n"
Jeremy Hyltonc5007aa2000-06-30 05:02:53 +00001205"collect() -- Do a full collection right now.\n"
Barry Warsawe5ec6132006-10-09 19:43:24 +00001206"get_count() -- Return the current collection counts.\n"
Jeremy Hyltonc5007aa2000-06-30 05:02:53 +00001207"set_debug() -- Set debugging flags.\n"
1208"get_debug() -- Get debugging flags.\n"
1209"set_threshold() -- Set the collection thresholds.\n"
1210"get_threshold() -- Return the current the collection thresholds.\n"
Neil Schemenauerc7c8d8e2001-08-09 15:58:59 +00001211"get_objects() -- Return a list of all objects tracked by the collector.\n"
Jeremy Hylton5bd378b2003-04-03 16:28:38 +00001212"get_referrers() -- Return the list of objects that refer to an object.\n"
Tim Peters730f5532003-04-08 17:17:17 +00001213"get_referents() -- Return the list of objects that an object refers to.\n");
Jeremy Hyltonc5007aa2000-06-30 05:02:53 +00001214
1215static PyMethodDef GcMethods[] = {
Tim Peters50c61d52003-04-06 01:50:50 +00001216 {"enable", gc_enable, METH_NOARGS, gc_enable__doc__},
1217 {"disable", gc_disable, METH_NOARGS, gc_disable__doc__},
1218 {"isenabled", gc_isenabled, METH_NOARGS, gc_isenabled__doc__},
Vladimir Marangozovf9d20c32000-08-06 22:45:31 +00001219 {"set_debug", gc_set_debug, METH_VARARGS, gc_set_debug__doc__},
Tim Peters50c61d52003-04-06 01:50:50 +00001220 {"get_debug", gc_get_debug, METH_NOARGS, gc_get_debug__doc__},
Barry Warsawd3c38ff2006-03-07 09:46:03 +00001221 {"get_count", gc_get_count, METH_NOARGS, gc_get_count__doc__},
Vladimir Marangozovf9d20c32000-08-06 22:45:31 +00001222 {"set_threshold", gc_set_thresh, METH_VARARGS, gc_set_thresh__doc__},
Tim Peters50c61d52003-04-06 01:50:50 +00001223 {"get_threshold", gc_get_thresh, METH_NOARGS, gc_get_thresh__doc__},
Barry Warsawd3c38ff2006-03-07 09:46:03 +00001224 {"collect", (PyCFunction)gc_collect,
1225 METH_VARARGS | METH_KEYWORDS, gc_collect__doc__},
Tim Peters50c61d52003-04-06 01:50:50 +00001226 {"get_objects", gc_get_objects,METH_NOARGS, gc_get_objects__doc__},
Martin v. Löwis560da622001-11-24 09:24:51 +00001227 {"get_referrers", gc_get_referrers, METH_VARARGS,
1228 gc_get_referrers__doc__},
Tim Peters730f5532003-04-08 17:17:17 +00001229 {"get_referents", gc_get_referents, METH_VARARGS,
1230 gc_get_referents__doc__},
Jeremy Hyltonc5007aa2000-06-30 05:02:53 +00001231 {NULL, NULL} /* Sentinel */
1232};
1233
Jason Tishler6bc06ec2003-09-04 11:59:50 +00001234PyMODINIT_FUNC
Jeremy Hyltonc5007aa2000-06-30 05:02:53 +00001235initgc(void)
1236{
1237 PyObject *m;
Jeremy Hyltonc5007aa2000-06-30 05:02:53 +00001238
1239 m = Py_InitModule4("gc",
1240 GcMethods,
1241 gc__doc__,
1242 NULL,
1243 PYTHON_API_VERSION);
Neal Norwitz1ac754f2006-01-19 06:09:39 +00001244 if (m == NULL)
1245 return;
Tim Peters11558872003-04-06 23:30:52 +00001246
Jeremy Hyltonc5007aa2000-06-30 05:02:53 +00001247 if (garbage == NULL) {
1248 garbage = PyList_New(0);
Tim Peters11558872003-04-06 23:30:52 +00001249 if (garbage == NULL)
1250 return;
Jeremy Hyltonc5007aa2000-06-30 05:02:53 +00001251 }
Neil Schemenauer3b1cbf92005-06-18 17:37:06 +00001252 Py_INCREF(garbage);
Tim Peters11558872003-04-06 23:30:52 +00001253 if (PyModule_AddObject(m, "garbage", garbage) < 0)
1254 return;
Neal Norwitz57a03612006-04-26 05:34:03 +00001255
1256 /* Importing can't be done in collect() because collect()
1257 * can be called via PyGC_Collect() in Py_Finalize().
1258 * This wouldn't be a problem, except that <initialized> is
1259 * reset to 0 before calling collect which trips up
1260 * the import and triggers an assertion.
1261 */
1262 if (tmod == NULL) {
Christian Heimes000a0742008-01-03 22:16:32 +00001263 tmod = PyImport_ImportModuleNoBlock("time");
Neal Norwitz57a03612006-04-26 05:34:03 +00001264 if (tmod == NULL)
1265 PyErr_Clear();
1266 }
1267
Tim Peters11558872003-04-06 23:30:52 +00001268#define ADD_INT(NAME) if (PyModule_AddIntConstant(m, #NAME, NAME) < 0) return
1269 ADD_INT(DEBUG_STATS);
1270 ADD_INT(DEBUG_COLLECTABLE);
1271 ADD_INT(DEBUG_UNCOLLECTABLE);
1272 ADD_INT(DEBUG_INSTANCES);
1273 ADD_INT(DEBUG_OBJECTS);
1274 ADD_INT(DEBUG_SAVEALL);
1275 ADD_INT(DEBUG_LEAK);
1276#undef ADD_INT
Jeremy Hyltonc5007aa2000-06-30 05:02:53 +00001277}
1278
Guido van Rossume13ddc92003-04-17 17:29:22 +00001279/* API to invoke gc.collect() from C */
Neal Norwitz7b216c52006-03-04 20:01:53 +00001280Py_ssize_t
Guido van Rossume13ddc92003-04-17 17:29:22 +00001281PyGC_Collect(void)
1282{
Neal Norwitz7b216c52006-03-04 20:01:53 +00001283 Py_ssize_t n;
Guido van Rossume13ddc92003-04-17 17:29:22 +00001284
1285 if (collecting)
1286 n = 0; /* already collecting, don't do anything */
1287 else {
1288 collecting = 1;
1289 n = collect(NUM_GENERATIONS - 1);
1290 collecting = 0;
1291 }
1292
1293 return n;
1294}
1295
Neil Schemenauer43411b52001-08-30 00:05:51 +00001296/* for debugging */
Guido van Rossume13ddc92003-04-17 17:29:22 +00001297void
1298_PyGC_Dump(PyGC_Head *g)
Neil Schemenauer43411b52001-08-30 00:05:51 +00001299{
1300 _PyObject_Dump(FROM_GC(g));
1301}
1302
Neil Schemenauer43411b52001-08-30 00:05:51 +00001303/* extension modules might be compiled with GC support so these
1304 functions must always be available */
1305
Neil Schemenauerfec4eb12002-04-12 02:41:03 +00001306#undef PyObject_GC_Track
1307#undef PyObject_GC_UnTrack
1308#undef PyObject_GC_Del
1309#undef _PyObject_GC_Malloc
1310
Neil Schemenauer43411b52001-08-30 00:05:51 +00001311void
Neil Schemenauerfec4eb12002-04-12 02:41:03 +00001312PyObject_GC_Track(void *op)
Neil Schemenauer43411b52001-08-30 00:05:51 +00001313{
1314 _PyObject_GC_TRACK(op);
1315}
1316
Neil Schemenauerfec4eb12002-04-12 02:41:03 +00001317/* for binary compatibility with 2.2 */
Neil Schemenauer43411b52001-08-30 00:05:51 +00001318void
Neil Schemenauerfec4eb12002-04-12 02:41:03 +00001319_PyObject_GC_Track(PyObject *op)
1320{
1321 PyObject_GC_Track(op);
1322}
1323
1324void
1325PyObject_GC_UnTrack(void *op)
Neil Schemenauer43411b52001-08-30 00:05:51 +00001326{
Tim Peters803526b2002-07-07 05:13:56 +00001327 /* Obscure: the Py_TRASHCAN mechanism requires that we be able to
1328 * call PyObject_GC_UnTrack twice on an object.
1329 */
Neil Schemenauera2b11ec2002-05-21 15:53:24 +00001330 if (IS_TRACKED(op))
Guido van Rossumff413af2002-03-28 20:34:59 +00001331 _PyObject_GC_UNTRACK(op);
Neil Schemenauer43411b52001-08-30 00:05:51 +00001332}
1333
Neil Schemenauerfec4eb12002-04-12 02:41:03 +00001334/* for binary compatibility with 2.2 */
1335void
1336_PyObject_GC_UnTrack(PyObject *op)
1337{
1338 PyObject_GC_UnTrack(op);
1339}
1340
Neil Schemenauer43411b52001-08-30 00:05:51 +00001341PyObject *
Neil Schemenauerfec4eb12002-04-12 02:41:03 +00001342_PyObject_GC_Malloc(size_t basicsize)
Neil Schemenauer43411b52001-08-30 00:05:51 +00001343{
1344 PyObject *op;
Neal Norwitze7d8be82008-07-31 17:17:14 +00001345 PyGC_Head *g;
1346 if (basicsize > PY_SSIZE_T_MAX - sizeof(PyGC_Head))
1347 return PyErr_NoMemory();
1348 g = (PyGC_Head *)PyObject_MALLOC(
Anthony Baxter64182fe2006-04-11 12:14:09 +00001349 sizeof(PyGC_Head) + basicsize);
Neil Schemenauer43411b52001-08-30 00:05:51 +00001350 if (g == NULL)
Jeremy Hylton8a135182002-06-06 23:23:55 +00001351 return PyErr_NoMemory();
Tim Petersea405632002-07-02 00:52:30 +00001352 g->gc.gc_refs = GC_UNTRACKED;
Neil Schemenauer2880ae52002-05-04 05:35:20 +00001353 generations[0].count++; /* number of allocated GC objects */
1354 if (generations[0].count > generations[0].threshold &&
Neil Schemenauer43411b52001-08-30 00:05:51 +00001355 enabled &&
Neil Schemenauer2880ae52002-05-04 05:35:20 +00001356 generations[0].threshold &&
Neil Schemenauer43411b52001-08-30 00:05:51 +00001357 !collecting &&
1358 !PyErr_Occurred()) {
1359 collecting = 1;
Neil Schemenauer2880ae52002-05-04 05:35:20 +00001360 collect_generations();
Neil Schemenauer43411b52001-08-30 00:05:51 +00001361 collecting = 0;
1362 }
1363 op = FROM_GC(g);
Neil Schemenauer43411b52001-08-30 00:05:51 +00001364 return op;
1365}
1366
1367PyObject *
1368_PyObject_GC_New(PyTypeObject *tp)
1369{
Neil Schemenauerfec4eb12002-04-12 02:41:03 +00001370 PyObject *op = _PyObject_GC_Malloc(_PyObject_SIZE(tp));
Tim Petersfa8efab2002-04-28 01:57:25 +00001371 if (op != NULL)
1372 op = PyObject_INIT(op, tp);
1373 return op;
Neil Schemenauer43411b52001-08-30 00:05:51 +00001374}
1375
1376PyVarObject *
Martin v. Löwis18e16552006-02-15 17:27:45 +00001377_PyObject_GC_NewVar(PyTypeObject *tp, Py_ssize_t nitems)
Neil Schemenauer43411b52001-08-30 00:05:51 +00001378{
Neil Schemenauerfec4eb12002-04-12 02:41:03 +00001379 const size_t size = _PyObject_VAR_SIZE(tp, nitems);
1380 PyVarObject *op = (PyVarObject *) _PyObject_GC_Malloc(size);
Tim Petersfa8efab2002-04-28 01:57:25 +00001381 if (op != NULL)
1382 op = PyObject_INIT_VAR(op, tp, nitems);
1383 return op;
Neil Schemenauer43411b52001-08-30 00:05:51 +00001384}
1385
1386PyVarObject *
Martin v. Löwis41290682006-02-16 14:56:14 +00001387_PyObject_GC_Resize(PyVarObject *op, Py_ssize_t nitems)
Neil Schemenauer43411b52001-08-30 00:05:51 +00001388{
Christian Heimese93237d2007-12-19 02:37:44 +00001389 const size_t basicsize = _PyObject_VAR_SIZE(Py_TYPE(op), nitems);
Neil Schemenauer43411b52001-08-30 00:05:51 +00001390 PyGC_Head *g = AS_GC(op);
Neal Norwitze7d8be82008-07-31 17:17:14 +00001391 if (basicsize > PY_SSIZE_T_MAX - sizeof(PyGC_Head))
1392 return (PyVarObject *)PyErr_NoMemory();
Anthony Baxter64182fe2006-04-11 12:14:09 +00001393 g = (PyGC_Head *)PyObject_REALLOC(g, sizeof(PyGC_Head) + basicsize);
Neil Schemenauer43411b52001-08-30 00:05:51 +00001394 if (g == NULL)
1395 return (PyVarObject *)PyErr_NoMemory();
1396 op = (PyVarObject *) FROM_GC(g);
Christian Heimese93237d2007-12-19 02:37:44 +00001397 Py_SIZE(op) = nitems;
Neil Schemenauer43411b52001-08-30 00:05:51 +00001398 return op;
1399}
1400
1401void
Neil Schemenauerfec4eb12002-04-12 02:41:03 +00001402PyObject_GC_Del(void *op)
Neil Schemenauer43411b52001-08-30 00:05:51 +00001403{
Neil Schemenauer43411b52001-08-30 00:05:51 +00001404 PyGC_Head *g = AS_GC(op);
Neil Schemenauera2b11ec2002-05-21 15:53:24 +00001405 if (IS_TRACKED(op))
Neil Schemenauer43411b52001-08-30 00:05:51 +00001406 gc_list_remove(g);
Neil Schemenauer2880ae52002-05-04 05:35:20 +00001407 if (generations[0].count > 0) {
1408 generations[0].count--;
Neil Schemenauer43411b52001-08-30 00:05:51 +00001409 }
Neil Schemenauerfec4eb12002-04-12 02:41:03 +00001410 PyObject_FREE(g);
Neil Schemenauer43411b52001-08-30 00:05:51 +00001411}
1412
Neil Schemenauerfec4eb12002-04-12 02:41:03 +00001413/* for binary compatibility with 2.2 */
1414#undef _PyObject_GC_Del
1415void
1416_PyObject_GC_Del(PyObject *op)
1417{
1418 PyObject_GC_Del(op);
1419}