blob: 5c2f3816e1f8cdb0e632396e1373bae97d7df474 [file] [log] [blame]
Jeremy Hyltonc5007aa2000-06-30 05:02:53 +00001/*
Tim Peters88396172002-06-30 17:56:40 +00002
Jeremy Hyltonc5007aa2000-06-30 05:02:53 +00003 Reference Cycle Garbage Collection
4 ==================================
5
Neil Schemenauerb2c2c9e2000-10-04 16:34:09 +00006 Neil Schemenauer <nas@arctrix.com>
Jeremy Hyltonc5007aa2000-06-30 05:02:53 +00007
8 Based on a post on the python-dev list. Ideas from Guido van Rossum,
9 Eric Tiedemann, and various others.
10
Neil Schemenauer43411b52001-08-30 00:05:51 +000011 http://www.arctrix.com/nas/python/gc/
Jeremy Hyltonc5007aa2000-06-30 05:02:53 +000012 http://www.python.org/pipermail/python-dev/2000-March/003869.html
13 http://www.python.org/pipermail/python-dev/2000-March/004010.html
14 http://www.python.org/pipermail/python-dev/2000-March/004022.html
15
16 For a highlevel view of the collection process, read the collect
17 function.
18
Jeremy Hyltonc5007aa2000-06-30 05:02:53 +000019*/
20
Jeremy Hyltonc5007aa2000-06-30 05:02:53 +000021#include "Python.h"
22
Neil Schemenauer43411b52001-08-30 00:05:51 +000023/* Get an object's GC head */
24#define AS_GC(o) ((PyGC_Head *)(o)-1)
25
26/* Get the object given the GC head */
27#define FROM_GC(g) ((PyObject *)(((PyGC_Head *)g)+1))
28
Jeremy Hyltonc5007aa2000-06-30 05:02:53 +000029/*** Global GC state ***/
30
Neil Schemenauer2880ae52002-05-04 05:35:20 +000031struct gc_generation {
32 PyGC_Head head;
33 int threshold; /* collection threshold */
34 int count; /* count of allocations or collections of younger
35 generations */
36};
37
38#define NUM_GENERATIONS 3
39#define GEN_HEAD(n) (&generations[n].head)
40
Jeremy Hyltonc5007aa2000-06-30 05:02:53 +000041/* linked lists of container objects */
Neil Schemenauer2880ae52002-05-04 05:35:20 +000042static struct gc_generation generations[NUM_GENERATIONS] = {
43 /* PyGC_Head, threshold, count */
44 {{{GEN_HEAD(0), GEN_HEAD(0), 0}}, 700, 0},
45 {{{GEN_HEAD(1), GEN_HEAD(1), 0}}, 10, 0},
46 {{{GEN_HEAD(2), GEN_HEAD(2), 0}}, 10, 0},
47};
Jeremy Hyltonc5007aa2000-06-30 05:02:53 +000048
Neil Schemenauer2880ae52002-05-04 05:35:20 +000049PyGC_Head *_PyGC_generation0 = GEN_HEAD(0);
50
Vladimir Marangozovf9d20c32000-08-06 22:45:31 +000051static int enabled = 1; /* automatic collection enabled? */
Jeremy Hyltonc5007aa2000-06-30 05:02:53 +000052
Neil Schemenauer43411b52001-08-30 00:05:51 +000053/* true if we are currently running the collector */
Tim Petersbf384c22003-04-06 00:11:39 +000054static int collecting = 0;
Neil Schemenauer43411b52001-08-30 00:05:51 +000055
Tim Peters6fc13d92002-07-02 18:12:35 +000056/* list of uncollectable objects */
Tim Petersbf384c22003-04-06 00:11:39 +000057static PyObject *garbage = NULL;
Tim Peters6fc13d92002-07-02 18:12:35 +000058
59/* Python string to use if unhandled exception occurs */
Tim Petersbf384c22003-04-06 00:11:39 +000060static PyObject *gc_str = NULL;
Tim Peters6fc13d92002-07-02 18:12:35 +000061
Tim Peters93ad66d2003-04-05 17:15:44 +000062/* Python string used to look for __del__ attribute. */
63static PyObject *delstr = NULL;
Jeremy Hyltonce136e92003-04-04 19:59:06 +000064
Jeremy Hyltonc5007aa2000-06-30 05:02:53 +000065/* set for debugging information */
66#define DEBUG_STATS (1<<0) /* print collection statistics */
67#define DEBUG_COLLECTABLE (1<<1) /* print collectable objects */
68#define DEBUG_UNCOLLECTABLE (1<<2) /* print uncollectable objects */
Jeremy Hyltonc5007aa2000-06-30 05:02:53 +000069#define DEBUG_OBJECTS (1<<4) /* print other objects */
Neil Schemenauer544de1e2000-09-22 15:22:38 +000070#define DEBUG_SAVEALL (1<<5) /* save all garbage in gc.garbage */
Jeremy Hyltonc5007aa2000-06-30 05:02:53 +000071#define DEBUG_LEAK DEBUG_COLLECTABLE | \
72 DEBUG_UNCOLLECTABLE | \
Neil Schemenauer544de1e2000-09-22 15:22:38 +000073 DEBUG_OBJECTS | \
74 DEBUG_SAVEALL
Jeremy Hyltonb709df32000-09-01 02:47:25 +000075static int debug;
Thomas Wouters477c8d52006-05-27 19:21:47 +000076static PyObject *tmod = NULL;
Jeremy Hyltonc5007aa2000-06-30 05:02:53 +000077
Tim Peters6fc13d92002-07-02 18:12:35 +000078/*--------------------------------------------------------------------------
79gc_refs values.
Neil Schemenauer43411b52001-08-30 00:05:51 +000080
Tim Peters6fc13d92002-07-02 18:12:35 +000081Between collections, every gc'ed object has one of two gc_refs values:
82
83GC_UNTRACKED
84 The initial state; objects returned by PyObject_GC_Malloc are in this
85 state. The object doesn't live in any generation list, and its
86 tp_traverse slot must not be called.
87
88GC_REACHABLE
89 The object lives in some generation list, and its tp_traverse is safe to
90 call. An object transitions to GC_REACHABLE when PyObject_GC_Track
91 is called.
92
93During a collection, gc_refs can temporarily take on other states:
94
95>= 0
96 At the start of a collection, update_refs() copies the true refcount
97 to gc_refs, for each object in the generation being collected.
98 subtract_refs() then adjusts gc_refs so that it equals the number of
99 times an object is referenced directly from outside the generation
100 being collected.
Martin v. Löwis774348c2002-11-09 19:54:06 +0000101 gc_refs remains >= 0 throughout these steps.
Tim Peters6fc13d92002-07-02 18:12:35 +0000102
103GC_TENTATIVELY_UNREACHABLE
104 move_unreachable() then moves objects not reachable (whether directly or
105 indirectly) from outside the generation into an "unreachable" set.
106 Objects that are found to be reachable have gc_refs set to GC_REACHABLE
107 again. Objects that are found to be unreachable have gc_refs set to
108 GC_TENTATIVELY_UNREACHABLE. It's "tentatively" because the pass doing
109 this can't be sure until it ends, and GC_TENTATIVELY_UNREACHABLE may
110 transition back to GC_REACHABLE.
111
112 Only objects with GC_TENTATIVELY_UNREACHABLE still set are candidates
113 for collection. If it's decided not to collect such an object (e.g.,
114 it has a __del__ method), its gc_refs is restored to GC_REACHABLE again.
115----------------------------------------------------------------------------
116*/
Tim Petersea405632002-07-02 00:52:30 +0000117#define GC_UNTRACKED _PyGC_REFS_UNTRACKED
118#define GC_REACHABLE _PyGC_REFS_REACHABLE
119#define GC_TENTATIVELY_UNREACHABLE _PyGC_REFS_TENTATIVELY_UNREACHABLE
Tim Peters19b74c72002-07-01 03:52:19 +0000120
Tim Peters6fc13d92002-07-02 18:12:35 +0000121#define IS_TRACKED(o) ((AS_GC(o))->gc.gc_refs != GC_UNTRACKED)
Tim Peters19b74c72002-07-01 03:52:19 +0000122#define IS_REACHABLE(o) ((AS_GC(o))->gc.gc_refs == GC_REACHABLE)
123#define IS_TENTATIVELY_UNREACHABLE(o) ( \
124 (AS_GC(o))->gc.gc_refs == GC_TENTATIVELY_UNREACHABLE)
Neil Schemenauera2b11ec2002-05-21 15:53:24 +0000125
Jeremy Hyltonc5007aa2000-06-30 05:02:53 +0000126/*** list functions ***/
127
128static void
129gc_list_init(PyGC_Head *list)
130{
Tim Peters9e4ca102001-10-11 18:31:31 +0000131 list->gc.gc_prev = list;
132 list->gc.gc_next = list;
Jeremy Hyltonc5007aa2000-06-30 05:02:53 +0000133}
134
Neil Schemenauer2880ae52002-05-04 05:35:20 +0000135static int
136gc_list_is_empty(PyGC_Head *list)
137{
138 return (list->gc.gc_next == list);
139}
140
Tim Peterse2d59182004-11-01 01:39:08 +0000141#if 0
142/* This became unused after gc_list_move() was introduced. */
143/* Append `node` to `list`. */
Jeremy Hyltonc5007aa2000-06-30 05:02:53 +0000144static void
145gc_list_append(PyGC_Head *node, PyGC_Head *list)
146{
Tim Peters9e4ca102001-10-11 18:31:31 +0000147 node->gc.gc_next = list;
148 node->gc.gc_prev = list->gc.gc_prev;
149 node->gc.gc_prev->gc.gc_next = node;
150 list->gc.gc_prev = node;
Jeremy Hyltonc5007aa2000-06-30 05:02:53 +0000151}
Tim Peterse2d59182004-11-01 01:39:08 +0000152#endif
Jeremy Hyltonc5007aa2000-06-30 05:02:53 +0000153
Tim Peterse2d59182004-11-01 01:39:08 +0000154/* Remove `node` from the gc list it's currently in. */
Jeremy Hyltonc5007aa2000-06-30 05:02:53 +0000155static void
156gc_list_remove(PyGC_Head *node)
157{
Tim Peters9e4ca102001-10-11 18:31:31 +0000158 node->gc.gc_prev->gc.gc_next = node->gc.gc_next;
159 node->gc.gc_next->gc.gc_prev = node->gc.gc_prev;
160 node->gc.gc_next = NULL; /* object is not currently tracked */
Jeremy Hyltonc5007aa2000-06-30 05:02:53 +0000161}
162
Tim Peterse2d59182004-11-01 01:39:08 +0000163/* Move `node` from the gc list it's currently in (which is not explicitly
164 * named here) to the end of `list`. This is semantically the same as
165 * gc_list_remove(node) followed by gc_list_append(node, list).
166 */
167static void
168gc_list_move(PyGC_Head *node, PyGC_Head *list)
169{
Tim Petersbc1d1b82004-11-01 16:39:57 +0000170 PyGC_Head *new_prev;
Tim Peterse2d59182004-11-01 01:39:08 +0000171 PyGC_Head *current_prev = node->gc.gc_prev;
172 PyGC_Head *current_next = node->gc.gc_next;
Tim Petersbc1d1b82004-11-01 16:39:57 +0000173 /* Unlink from current list. */
Tim Peterse2d59182004-11-01 01:39:08 +0000174 current_prev->gc.gc_next = current_next;
175 current_next->gc.gc_prev = current_prev;
Tim Petersbc1d1b82004-11-01 16:39:57 +0000176 /* Relink at end of new list. */
177 new_prev = node->gc.gc_prev = list->gc.gc_prev;
Tim Peterse2d59182004-11-01 01:39:08 +0000178 new_prev->gc.gc_next = list->gc.gc_prev = node;
Tim Petersbc1d1b82004-11-01 16:39:57 +0000179 node->gc.gc_next = list;
Tim Peterse2d59182004-11-01 01:39:08 +0000180}
181
182/* append list `from` onto list `to`; `from` becomes an empty list */
Jeremy Hyltonc5007aa2000-06-30 05:02:53 +0000183static void
184gc_list_merge(PyGC_Head *from, PyGC_Head *to)
185{
186 PyGC_Head *tail;
Tim Peterse2d59182004-11-01 01:39:08 +0000187 assert(from != to);
Neil Schemenauer2880ae52002-05-04 05:35:20 +0000188 if (!gc_list_is_empty(from)) {
Tim Peters9e4ca102001-10-11 18:31:31 +0000189 tail = to->gc.gc_prev;
190 tail->gc.gc_next = from->gc.gc_next;
191 tail->gc.gc_next->gc.gc_prev = tail;
192 to->gc.gc_prev = from->gc.gc_prev;
193 to->gc.gc_prev->gc.gc_next = to;
Jeremy Hyltonc5007aa2000-06-30 05:02:53 +0000194 }
195 gc_list_init(from);
196}
197
Neal Norwitz7b216c52006-03-04 20:01:53 +0000198static Py_ssize_t
Jeremy Hyltonc5007aa2000-06-30 05:02:53 +0000199gc_list_size(PyGC_Head *list)
200{
201 PyGC_Head *gc;
Neal Norwitz7b216c52006-03-04 20:01:53 +0000202 Py_ssize_t n = 0;
Tim Peters9e4ca102001-10-11 18:31:31 +0000203 for (gc = list->gc.gc_next; gc != list; gc = gc->gc.gc_next) {
Jeremy Hyltonc5007aa2000-06-30 05:02:53 +0000204 n++;
205 }
206 return n;
207}
208
Tim Peters259272b2003-04-06 19:41:39 +0000209/* Append objects in a GC list to a Python list.
210 * Return 0 if all OK, < 0 if error (out of memory for list).
211 */
212static int
213append_objects(PyObject *py_list, PyGC_Head *gc_list)
214{
215 PyGC_Head *gc;
216 for (gc = gc_list->gc.gc_next; gc != gc_list; gc = gc->gc.gc_next) {
217 PyObject *op = FROM_GC(gc);
218 if (op != py_list) {
219 if (PyList_Append(py_list, op)) {
220 return -1; /* exception */
221 }
222 }
223 }
224 return 0;
225}
226
Jeremy Hyltonc5007aa2000-06-30 05:02:53 +0000227/*** end of list stuff ***/
228
229
Tim Peters19b74c72002-07-01 03:52:19 +0000230/* Set all gc_refs = ob_refcnt. After this, gc_refs is > 0 for all objects
231 * in containers, and is GC_REACHABLE for all tracked gc objects not in
232 * containers.
Tim Peters88396172002-06-30 17:56:40 +0000233 */
Jeremy Hyltonc5007aa2000-06-30 05:02:53 +0000234static void
235update_refs(PyGC_Head *containers)
236{
Tim Peters9e4ca102001-10-11 18:31:31 +0000237 PyGC_Head *gc = containers->gc.gc_next;
Tim Petersea405632002-07-02 00:52:30 +0000238 for (; gc != containers; gc = gc->gc.gc_next) {
239 assert(gc->gc.gc_refs == GC_REACHABLE);
Tim Peters9e4ca102001-10-11 18:31:31 +0000240 gc->gc.gc_refs = FROM_GC(gc)->ob_refcnt;
Tim Peters780c4972003-11-14 00:01:17 +0000241 /* Python's cyclic gc should never see an incoming refcount
242 * of 0: if something decref'ed to 0, it should have been
243 * deallocated immediately at that time.
244 * Possible cause (if the assert triggers): a tp_dealloc
245 * routine left a gc-aware object tracked during its teardown
246 * phase, and did something-- or allowed something to happen --
247 * that called back into Python. gc can trigger then, and may
248 * see the still-tracked dying object. Before this assert
249 * was added, such mistakes went on to allow gc to try to
250 * delete the object again. In a debug build, that caused
251 * a mysterious segfault, when _Py_ForgetReference tried
252 * to remove the object from the doubly-linked list of all
253 * objects a second time. In a release build, an actual
254 * double deallocation occurred, which leads to corruption
255 * of the allocator's internal bookkeeping pointers. That's
256 * so serious that maybe this should be a release-build
257 * check instead of an assert?
258 */
259 assert(gc->gc.gc_refs != 0);
Tim Petersea405632002-07-02 00:52:30 +0000260 }
Jeremy Hyltonc5007aa2000-06-30 05:02:53 +0000261}
262
Tim Peters19b74c72002-07-01 03:52:19 +0000263/* A traversal callback for subtract_refs. */
Jeremy Hyltonc5007aa2000-06-30 05:02:53 +0000264static int
265visit_decref(PyObject *op, void *data)
266{
Tim Peters93cd83e2002-06-30 21:31:03 +0000267 assert(op != NULL);
Tim Peters19b74c72002-07-01 03:52:19 +0000268 if (PyObject_IS_GC(op)) {
269 PyGC_Head *gc = AS_GC(op);
270 /* We're only interested in gc_refs for objects in the
271 * generation being collected, which can be recognized
272 * because only they have positive gc_refs.
273 */
Tim Petersaab713b2002-07-02 22:15:28 +0000274 assert(gc->gc.gc_refs != 0); /* else refcount was too small */
Tim Peters19b74c72002-07-01 03:52:19 +0000275 if (gc->gc.gc_refs > 0)
276 gc->gc.gc_refs--;
277 }
Jeremy Hyltonc5007aa2000-06-30 05:02:53 +0000278 return 0;
279}
280
Tim Peters19b74c72002-07-01 03:52:19 +0000281/* Subtract internal references from gc_refs. After this, gc_refs is >= 0
282 * for all objects in containers, and is GC_REACHABLE for all tracked gc
283 * objects not in containers. The ones with gc_refs > 0 are directly
284 * reachable from outside containers, and so can't be collected.
285 */
Jeremy Hyltonc5007aa2000-06-30 05:02:53 +0000286static void
287subtract_refs(PyGC_Head *containers)
288{
289 traverseproc traverse;
Tim Peters9e4ca102001-10-11 18:31:31 +0000290 PyGC_Head *gc = containers->gc.gc_next;
291 for (; gc != containers; gc=gc->gc.gc_next) {
Neil Schemenauer43411b52001-08-30 00:05:51 +0000292 traverse = FROM_GC(gc)->ob_type->tp_traverse;
293 (void) traverse(FROM_GC(gc),
Jeremy Hyltonc5007aa2000-06-30 05:02:53 +0000294 (visitproc)visit_decref,
295 NULL);
296 }
297}
298
Tim Peters19b74c72002-07-01 03:52:19 +0000299/* A traversal callback for move_unreachable. */
Jeremy Hyltonc5007aa2000-06-30 05:02:53 +0000300static int
Tim Peters19b74c72002-07-01 03:52:19 +0000301visit_reachable(PyObject *op, PyGC_Head *reachable)
Jeremy Hyltonc5007aa2000-06-30 05:02:53 +0000302{
Tim Petersea405632002-07-02 00:52:30 +0000303 if (PyObject_IS_GC(op)) {
Tim Peters19b74c72002-07-01 03:52:19 +0000304 PyGC_Head *gc = AS_GC(op);
Martin v. Löwis6db0e002006-03-01 16:56:25 +0000305 const Py_ssize_t gc_refs = gc->gc.gc_refs;
Tim Peters19b74c72002-07-01 03:52:19 +0000306
307 if (gc_refs == 0) {
308 /* This is in move_unreachable's 'young' list, but
309 * the traversal hasn't yet gotten to it. All
310 * we need to do is tell move_unreachable that it's
311 * reachable.
312 */
313 gc->gc.gc_refs = 1;
Jeremy Hyltonc5007aa2000-06-30 05:02:53 +0000314 }
Tim Peters19b74c72002-07-01 03:52:19 +0000315 else if (gc_refs == GC_TENTATIVELY_UNREACHABLE) {
316 /* This had gc_refs = 0 when move_unreachable got
317 * to it, but turns out it's reachable after all.
318 * Move it back to move_unreachable's 'young' list,
319 * and move_unreachable will eventually get to it
320 * again.
321 */
Tim Peterse2d59182004-11-01 01:39:08 +0000322 gc_list_move(gc, reachable);
Tim Peters19b74c72002-07-01 03:52:19 +0000323 gc->gc.gc_refs = 1;
324 }
325 /* Else there's nothing to do.
326 * If gc_refs > 0, it must be in move_unreachable's 'young'
327 * list, and move_unreachable will eventually get to it.
328 * If gc_refs == GC_REACHABLE, it's either in some other
329 * generation so we don't care about it, or move_unreachable
Tim Peters6fc13d92002-07-02 18:12:35 +0000330 * already dealt with it.
Tim Petersea405632002-07-02 00:52:30 +0000331 * If gc_refs == GC_UNTRACKED, it must be ignored.
Tim Peters19b74c72002-07-01 03:52:19 +0000332 */
Tim Petersea405632002-07-02 00:52:30 +0000333 else {
334 assert(gc_refs > 0
335 || gc_refs == GC_REACHABLE
336 || gc_refs == GC_UNTRACKED);
337 }
Jeremy Hyltonc5007aa2000-06-30 05:02:53 +0000338 }
339 return 0;
340}
341
Tim Peters19b74c72002-07-01 03:52:19 +0000342/* Move the unreachable objects from young to unreachable. After this,
343 * all objects in young have gc_refs = GC_REACHABLE, and all objects in
344 * unreachable have gc_refs = GC_TENTATIVELY_UNREACHABLE. All tracked
345 * gc objects not in young or unreachable still have gc_refs = GC_REACHABLE.
346 * All objects in young after this are directly or indirectly reachable
347 * from outside the original young; and all objects in unreachable are
348 * not.
Tim Peters88396172002-06-30 17:56:40 +0000349 */
Jeremy Hyltonc5007aa2000-06-30 05:02:53 +0000350static void
Tim Peters19b74c72002-07-01 03:52:19 +0000351move_unreachable(PyGC_Head *young, PyGC_Head *unreachable)
Jeremy Hyltonc5007aa2000-06-30 05:02:53 +0000352{
Tim Peters19b74c72002-07-01 03:52:19 +0000353 PyGC_Head *gc = young->gc.gc_next;
354
355 /* Invariants: all objects "to the left" of us in young have gc_refs
356 * = GC_REACHABLE, and are indeed reachable (directly or indirectly)
357 * from outside the young list as it was at entry. All other objects
358 * from the original young "to the left" of us are in unreachable now,
359 * and have gc_refs = GC_TENTATIVELY_UNREACHABLE. All objects to the
360 * left of us in 'young' now have been scanned, and no objects here
361 * or to the right have been scanned yet.
362 */
363
364 while (gc != young) {
365 PyGC_Head *next;
366
Tim Peters6fc13d92002-07-02 18:12:35 +0000367 if (gc->gc.gc_refs) {
368 /* gc is definitely reachable from outside the
369 * original 'young'. Mark it as such, and traverse
370 * its pointers to find any other objects that may
371 * be directly reachable from it. Note that the
372 * call to tp_traverse may append objects to young,
373 * so we have to wait until it returns to determine
374 * the next object to visit.
375 */
376 PyObject *op = FROM_GC(gc);
377 traverseproc traverse = op->ob_type->tp_traverse;
378 assert(gc->gc.gc_refs > 0);
379 gc->gc.gc_refs = GC_REACHABLE;
380 (void) traverse(op,
381 (visitproc)visit_reachable,
382 (void *)young);
383 next = gc->gc.gc_next;
384 }
385 else {
Tim Peters19b74c72002-07-01 03:52:19 +0000386 /* This *may* be unreachable. To make progress,
387 * assume it is. gc isn't directly reachable from
388 * any object we've already traversed, but may be
389 * reachable from an object we haven't gotten to yet.
390 * visit_reachable will eventually move gc back into
391 * young if that's so, and we'll see it again.
392 */
393 next = gc->gc.gc_next;
Tim Peterse2d59182004-11-01 01:39:08 +0000394 gc_list_move(gc, unreachable);
Tim Peters19b74c72002-07-01 03:52:19 +0000395 gc->gc.gc_refs = GC_TENTATIVELY_UNREACHABLE;
396 }
Tim Peters19b74c72002-07-01 03:52:19 +0000397 gc = next;
Jeremy Hyltonc5007aa2000-06-30 05:02:53 +0000398 }
399}
400
Tim Peters86b993b2003-04-05 17:35:54 +0000401/* Return true if object has a finalization method.
402 * CAUTION: An instance of an old-style class has to be checked for a
Tim Petersf6b80452003-04-07 19:21:15 +0000403 *__del__ method, and earlier versions of this used to call PyObject_HasAttr,
404 * which in turn could call the class's __getattr__ hook (if any). That
405 * could invoke arbitrary Python code, mutating the object graph in arbitrary
406 * ways, and that was the source of some excruciatingly subtle bugs.
Tim Peters86b993b2003-04-05 17:35:54 +0000407 */
Neil Schemenauera765c122001-11-01 17:35:23 +0000408static int
409has_finalizer(PyObject *op)
Jeremy Hyltonc5007aa2000-06-30 05:02:53 +0000410{
Guido van Rossum50e9fb92006-08-17 05:42:55 +0000411 if (PyGen_CheckExact(op))
Thomas Wouters49fd7fa2006-04-21 10:40:58 +0000412 return PyGen_NeedsFinalizing((PyGenObject *)op);
413 else
414 return 0;
Neil Schemenauera765c122001-11-01 17:35:23 +0000415}
416
Tim Petersead8b7a2004-10-30 23:09:22 +0000417/* Move the objects in unreachable with __del__ methods into `finalizers`.
418 * Objects moved into `finalizers` have gc_refs set to GC_REACHABLE; the
419 * objects remaining in unreachable are left at GC_TENTATIVELY_UNREACHABLE.
Jeremy Hyltonce136e92003-04-04 19:59:06 +0000420 */
Neil Schemenauera765c122001-11-01 17:35:23 +0000421static void
Tim Petersead8b7a2004-10-30 23:09:22 +0000422move_finalizers(PyGC_Head *unreachable, PyGC_Head *finalizers)
Neil Schemenauera765c122001-11-01 17:35:23 +0000423{
Tim Petersead8b7a2004-10-30 23:09:22 +0000424 PyGC_Head *gc;
425 PyGC_Head *next;
Tim Petersf6b80452003-04-07 19:21:15 +0000426
Tim Petersead8b7a2004-10-30 23:09:22 +0000427 /* March over unreachable. Move objects with finalizers into
428 * `finalizers`.
429 */
430 for (gc = unreachable->gc.gc_next; gc != unreachable; gc = next) {
Neil Schemenauer43411b52001-08-30 00:05:51 +0000431 PyObject *op = FROM_GC(gc);
Jeremy Hyltonce136e92003-04-04 19:59:06 +0000432
Tim Petersf6ae7a42003-04-05 18:40:50 +0000433 assert(IS_TENTATIVELY_UNREACHABLE(op));
Tim Petersead8b7a2004-10-30 23:09:22 +0000434 next = gc->gc.gc_next;
Tim Petersf6ae7a42003-04-05 18:40:50 +0000435
Tim Petersf6b80452003-04-07 19:21:15 +0000436 if (has_finalizer(op)) {
Tim Peterse2d59182004-11-01 01:39:08 +0000437 gc_list_move(gc, finalizers);
Tim Petersf6b80452003-04-07 19:21:15 +0000438 gc->gc.gc_refs = GC_REACHABLE;
Jeremy Hyltonce136e92003-04-04 19:59:06 +0000439 }
Jeremy Hyltonc5007aa2000-06-30 05:02:53 +0000440 }
441}
442
Tim Peters19b74c72002-07-01 03:52:19 +0000443/* A traversal callback for move_finalizer_reachable. */
444static int
445visit_move(PyObject *op, PyGC_Head *tolist)
446{
447 if (PyObject_IS_GC(op)) {
Tim Petersea405632002-07-02 00:52:30 +0000448 if (IS_TENTATIVELY_UNREACHABLE(op)) {
Tim Peters19b74c72002-07-01 03:52:19 +0000449 PyGC_Head *gc = AS_GC(op);
Tim Peterse2d59182004-11-01 01:39:08 +0000450 gc_list_move(gc, tolist);
Tim Peters19b74c72002-07-01 03:52:19 +0000451 gc->gc.gc_refs = GC_REACHABLE;
452 }
453 }
454 return 0;
455}
456
457/* Move objects that are reachable from finalizers, from the unreachable set
Tim Petersf6b80452003-04-07 19:21:15 +0000458 * into finalizers set.
Tim Peters19b74c72002-07-01 03:52:19 +0000459 */
Jeremy Hyltonc5007aa2000-06-30 05:02:53 +0000460static void
Tim Petersf6b80452003-04-07 19:21:15 +0000461move_finalizer_reachable(PyGC_Head *finalizers)
Jeremy Hyltonc5007aa2000-06-30 05:02:53 +0000462{
463 traverseproc traverse;
Tim Peters9e4ca102001-10-11 18:31:31 +0000464 PyGC_Head *gc = finalizers->gc.gc_next;
Tim Petersbf384c22003-04-06 00:11:39 +0000465 for (; gc != finalizers; gc = gc->gc.gc_next) {
466 /* Note that the finalizers list may grow during this. */
Neil Schemenauer43411b52001-08-30 00:05:51 +0000467 traverse = FROM_GC(gc)->ob_type->tp_traverse;
Tim Peters88396172002-06-30 17:56:40 +0000468 (void) traverse(FROM_GC(gc),
Tim Petersbf384c22003-04-06 00:11:39 +0000469 (visitproc)visit_move,
Tim Petersf6b80452003-04-07 19:21:15 +0000470 (void *)finalizers);
Jeremy Hyltonc5007aa2000-06-30 05:02:53 +0000471 }
472}
473
Tim Petersead8b7a2004-10-30 23:09:22 +0000474/* Clear all weakrefs to unreachable objects, and if such a weakref has a
475 * callback, invoke it if necessary. Note that it's possible for such
476 * weakrefs to be outside the unreachable set -- indeed, those are precisely
477 * the weakrefs whose callbacks must be invoked. See gc_weakref.txt for
478 * overview & some details. Some weakrefs with callbacks may be reclaimed
479 * directly by this routine; the number reclaimed is the return value. Other
480 * weakrefs with callbacks may be moved into the `old` generation. Objects
481 * moved into `old` have gc_refs set to GC_REACHABLE; the objects remaining in
482 * unreachable are left at GC_TENTATIVELY_UNREACHABLE. When this returns,
483 * no object in `unreachable` is weakly referenced anymore.
Tim Peters403a2032003-11-20 21:21:46 +0000484 */
485static int
Tim Petersead8b7a2004-10-30 23:09:22 +0000486handle_weakrefs(PyGC_Head *unreachable, PyGC_Head *old)
Tim Peters403a2032003-11-20 21:21:46 +0000487{
Tim Petersead8b7a2004-10-30 23:09:22 +0000488 PyGC_Head *gc;
489 PyObject *op; /* generally FROM_GC(gc) */
490 PyWeakReference *wr; /* generally a cast of op */
Tim Petersead8b7a2004-10-30 23:09:22 +0000491 PyGC_Head wrcb_to_call; /* weakrefs with callbacks to call */
Tim Petersead8b7a2004-10-30 23:09:22 +0000492 PyGC_Head *next;
Tim Peters403a2032003-11-20 21:21:46 +0000493 int num_freed = 0;
494
Tim Petersead8b7a2004-10-30 23:09:22 +0000495 gc_list_init(&wrcb_to_call);
Tim Peters403a2032003-11-20 21:21:46 +0000496
Tim Petersead8b7a2004-10-30 23:09:22 +0000497 /* Clear all weakrefs to the objects in unreachable. If such a weakref
498 * also has a callback, move it into `wrcb_to_call` if the callback
Tim Peterscc2a8662004-10-31 22:12:43 +0000499 * needs to be invoked. Note that we cannot invoke any callbacks until
500 * all weakrefs to unreachable objects are cleared, lest the callback
501 * resurrect an unreachable object via a still-active weakref. We
502 * make another pass over wrcb_to_call, invoking callbacks, after this
503 * pass completes.
Tim Petersead8b7a2004-10-30 23:09:22 +0000504 */
505 for (gc = unreachable->gc.gc_next; gc != unreachable; gc = next) {
506 PyWeakReference **wrlist;
507
508 op = FROM_GC(gc);
509 assert(IS_TENTATIVELY_UNREACHABLE(op));
510 next = gc->gc.gc_next;
511
512 if (! PyType_SUPPORTS_WEAKREFS(op->ob_type))
513 continue;
514
515 /* It supports weakrefs. Does it have any? */
516 wrlist = (PyWeakReference **)
517 PyObject_GET_WEAKREFS_LISTPTR(op);
518
519 /* `op` may have some weakrefs. March over the list, clear
520 * all the weakrefs, and move the weakrefs with callbacks
Tim Peterscc2a8662004-10-31 22:12:43 +0000521 * that must be called into wrcb_to_call.
Tim Petersead8b7a2004-10-30 23:09:22 +0000522 */
523 for (wr = *wrlist; wr != NULL; wr = *wrlist) {
524 PyGC_Head *wrasgc; /* AS_GC(wr) */
525
526 /* _PyWeakref_ClearRef clears the weakref but leaves
527 * the callback pointer intact. Obscure: it also
528 * changes *wrlist.
529 */
530 assert(wr->wr_object == op);
531 _PyWeakref_ClearRef(wr);
532 assert(wr->wr_object == Py_None);
533 if (wr->wr_callback == NULL)
534 continue; /* no callback */
535
536 /* Headache time. `op` is going away, and is weakly referenced by
537 * `wr`, which has a callback. Should the callback be invoked? If wr
538 * is also trash, no:
539 *
540 * 1. There's no need to call it. The object and the weakref are
541 * both going away, so it's legitimate to pretend the weakref is
542 * going away first. The user has to ensure a weakref outlives its
543 * referent if they want a guarantee that the wr callback will get
544 * invoked.
545 *
546 * 2. It may be catastrophic to call it. If the callback is also in
547 * cyclic trash (CT), then although the CT is unreachable from
548 * outside the current generation, CT may be reachable from the
549 * callback. Then the callback could resurrect insane objects.
550 *
551 * Since the callback is never needed and may be unsafe in this case,
Tim Peterscc2a8662004-10-31 22:12:43 +0000552 * wr is simply left in the unreachable set. Note that because we
553 * already called _PyWeakref_ClearRef(wr), its callback will never
554 * trigger.
Tim Petersead8b7a2004-10-30 23:09:22 +0000555 *
556 * OTOH, if wr isn't part of CT, we should invoke the callback: the
557 * weakref outlived the trash. Note that since wr isn't CT in this
558 * case, its callback can't be CT either -- wr acted as an external
559 * root to this generation, and therefore its callback did too. So
560 * nothing in CT is reachable from the callback either, so it's hard
561 * to imagine how calling it later could create a problem for us. wr
562 * is moved to wrcb_to_call in this case.
Tim Petersead8b7a2004-10-30 23:09:22 +0000563 */
Tim Peterscc2a8662004-10-31 22:12:43 +0000564 if (IS_TENTATIVELY_UNREACHABLE(wr))
565 continue;
566 assert(IS_REACHABLE(wr));
567
Tim Petersead8b7a2004-10-30 23:09:22 +0000568 /* Create a new reference so that wr can't go away
569 * before we can process it again.
570 */
571 Py_INCREF(wr);
572
Tim Peterscc2a8662004-10-31 22:12:43 +0000573 /* Move wr to wrcb_to_call, for the next pass. */
Tim Petersead8b7a2004-10-30 23:09:22 +0000574 wrasgc = AS_GC(wr);
Tim Peterscc2a8662004-10-31 22:12:43 +0000575 assert(wrasgc != next); /* wrasgc is reachable, but
576 next isn't, so they can't
577 be the same */
Tim Peterse2d59182004-11-01 01:39:08 +0000578 gc_list_move(wrasgc, &wrcb_to_call);
Tim Petersead8b7a2004-10-30 23:09:22 +0000579 }
580 }
581
Tim Peterscc2a8662004-10-31 22:12:43 +0000582 /* Invoke the callbacks we decided to honor. It's safe to invoke them
583 * because they can't reference unreachable objects.
Tim Petersead8b7a2004-10-30 23:09:22 +0000584 */
585 while (! gc_list_is_empty(&wrcb_to_call)) {
586 PyObject *temp;
587 PyObject *callback;
588
589 gc = wrcb_to_call.gc.gc_next;
590 op = FROM_GC(gc);
591 assert(IS_REACHABLE(op));
592 assert(PyWeakref_Check(op));
593 wr = (PyWeakReference *)op;
594 callback = wr->wr_callback;
595 assert(callback != NULL);
596
597 /* copy-paste of weakrefobject.c's handle_callback() */
Thomas Wouters477c8d52006-05-27 19:21:47 +0000598 temp = PyObject_CallFunctionObjArgs(callback, wr, NULL);
Tim Petersead8b7a2004-10-30 23:09:22 +0000599 if (temp == NULL)
600 PyErr_WriteUnraisable(callback);
601 else
602 Py_DECREF(temp);
603
604 /* Give up the reference we created in the first pass. When
605 * op's refcount hits 0 (which it may or may not do right now),
Tim Peterscc2a8662004-10-31 22:12:43 +0000606 * op's tp_dealloc will decref op->wr_callback too. Note
607 * that the refcount probably will hit 0 now, and because this
608 * weakref was reachable to begin with, gc didn't already
609 * add it to its count of freed objects. Example: a reachable
610 * weak value dict maps some key to this reachable weakref.
611 * The callback removes this key->weakref mapping from the
612 * dict, leaving no other references to the weakref (excepting
613 * ours).
Tim Petersead8b7a2004-10-30 23:09:22 +0000614 */
615 Py_DECREF(op);
616 if (wrcb_to_call.gc.gc_next == gc) {
617 /* object is still alive -- move it */
Tim Peterse2d59182004-11-01 01:39:08 +0000618 gc_list_move(gc, old);
Tim Petersead8b7a2004-10-30 23:09:22 +0000619 }
620 else
621 ++num_freed;
622 }
623
Tim Peters403a2032003-11-20 21:21:46 +0000624 return num_freed;
625}
626
Jeremy Hyltonc5007aa2000-06-30 05:02:53 +0000627static void
Jeremy Hylton06257772000-08-31 15:10:24 +0000628debug_cycle(char *msg, PyObject *op)
Jeremy Hyltonc5007aa2000-06-30 05:02:53 +0000629{
Guido van Rossum50e9fb92006-08-17 05:42:55 +0000630 if (debug & DEBUG_OBJECTS) {
Jeremy Hylton06257772000-08-31 15:10:24 +0000631 PySys_WriteStderr("gc: %.100s <%.100s %p>\n",
632 msg, op->ob_type->tp_name, op);
Jeremy Hyltonc5007aa2000-06-30 05:02:53 +0000633 }
634}
635
Tim Petersbf384c22003-04-06 00:11:39 +0000636/* Handle uncollectable garbage (cycles with finalizers, and stuff reachable
637 * only from such cycles).
Tim Petersf6b80452003-04-07 19:21:15 +0000638 * If DEBUG_SAVEALL, all objects in finalizers are appended to the module
639 * garbage list (a Python list), else only the objects in finalizers with
640 * __del__ methods are appended to garbage. All objects in finalizers are
641 * merged into the old list regardless.
Tim Peters259272b2003-04-06 19:41:39 +0000642 * Returns 0 if all OK, <0 on error (out of memory to grow the garbage list).
643 * The finalizers list is made empty on a successful return.
Tim Petersbf384c22003-04-06 00:11:39 +0000644 */
Tim Peters259272b2003-04-06 19:41:39 +0000645static int
Tim Petersf6b80452003-04-07 19:21:15 +0000646handle_finalizers(PyGC_Head *finalizers, PyGC_Head *old)
Jeremy Hyltonc5007aa2000-06-30 05:02:53 +0000647{
Tim Petersf6b80452003-04-07 19:21:15 +0000648 PyGC_Head *gc = finalizers->gc.gc_next;
649
Jeremy Hyltonc5007aa2000-06-30 05:02:53 +0000650 if (garbage == NULL) {
651 garbage = PyList_New(0);
Tim Petersbf384c22003-04-06 00:11:39 +0000652 if (garbage == NULL)
653 Py_FatalError("gc couldn't create gc.garbage list");
Jeremy Hyltonc5007aa2000-06-30 05:02:53 +0000654 }
Tim Petersf6b80452003-04-07 19:21:15 +0000655 for (; gc != finalizers; gc = gc->gc.gc_next) {
656 PyObject *op = FROM_GC(gc);
657
658 if ((debug & DEBUG_SAVEALL) || has_finalizer(op)) {
659 if (PyList_Append(garbage, op) < 0)
660 return -1;
661 }
Jeremy Hyltonc5007aa2000-06-30 05:02:53 +0000662 }
Tim Petersf6b80452003-04-07 19:21:15 +0000663
Tim Peters259272b2003-04-06 19:41:39 +0000664 gc_list_merge(finalizers, old);
665 return 0;
Jeremy Hyltonc5007aa2000-06-30 05:02:53 +0000666}
667
Neil Schemenauer544de1e2000-09-22 15:22:38 +0000668/* Break reference cycles by clearing the containers involved. This is
Jeremy Hyltonc5007aa2000-06-30 05:02:53 +0000669 * tricky business as the lists can be changing and we don't know which
Tim Peters19b74c72002-07-01 03:52:19 +0000670 * objects may be freed. It is possible I screwed something up here.
671 */
Jeremy Hyltonc5007aa2000-06-30 05:02:53 +0000672static void
Jeremy Hyltonce136e92003-04-04 19:59:06 +0000673delete_garbage(PyGC_Head *collectable, PyGC_Head *old)
Jeremy Hyltonc5007aa2000-06-30 05:02:53 +0000674{
675 inquiry clear;
676
Jeremy Hyltonce136e92003-04-04 19:59:06 +0000677 while (!gc_list_is_empty(collectable)) {
678 PyGC_Head *gc = collectable->gc.gc_next;
Neil Schemenauer43411b52001-08-30 00:05:51 +0000679 PyObject *op = FROM_GC(gc);
Tim Peters88396172002-06-30 17:56:40 +0000680
Tim Peters19b74c72002-07-01 03:52:19 +0000681 assert(IS_TENTATIVELY_UNREACHABLE(op));
Neil Schemenauer544de1e2000-09-22 15:22:38 +0000682 if (debug & DEBUG_SAVEALL) {
683 PyList_Append(garbage, op);
Jeremy Hyltonc5007aa2000-06-30 05:02:53 +0000684 }
Neil Schemenauer544de1e2000-09-22 15:22:38 +0000685 else {
686 if ((clear = op->ob_type->tp_clear) != NULL) {
687 Py_INCREF(op);
Jeremy Hylton8a135182002-06-06 23:23:55 +0000688 clear(op);
Neil Schemenauer544de1e2000-09-22 15:22:38 +0000689 Py_DECREF(op);
690 }
691 }
Jeremy Hyltonce136e92003-04-04 19:59:06 +0000692 if (collectable->gc.gc_next == gc) {
Neil Schemenauer544de1e2000-09-22 15:22:38 +0000693 /* object is still alive, move it, it may die later */
Tim Peterse2d59182004-11-01 01:39:08 +0000694 gc_list_move(gc, old);
Tim Peters19b74c72002-07-01 03:52:19 +0000695 gc->gc.gc_refs = GC_REACHABLE;
Jeremy Hyltonc5007aa2000-06-30 05:02:53 +0000696 }
697 }
698}
699
700/* This is the main function. Read this to understand how the
701 * collection process works. */
Neal Norwitz7b216c52006-03-04 20:01:53 +0000702static Py_ssize_t
Neil Schemenauer2880ae52002-05-04 05:35:20 +0000703collect(int generation)
Jeremy Hyltonc5007aa2000-06-30 05:02:53 +0000704{
Neil Schemenauer2880ae52002-05-04 05:35:20 +0000705 int i;
Neal Norwitz7b216c52006-03-04 20:01:53 +0000706 Py_ssize_t m = 0; /* # objects collected */
707 Py_ssize_t n = 0; /* # unreachable objects that couldn't be collected */
Neil Schemenauer2880ae52002-05-04 05:35:20 +0000708 PyGC_Head *young; /* the generation we are examining */
709 PyGC_Head *old; /* next older generation */
Tim Peters403a2032003-11-20 21:21:46 +0000710 PyGC_Head unreachable; /* non-problematic unreachable trash */
711 PyGC_Head finalizers; /* objects with, & reachable from, __del__ */
Jeremy Hyltonc5007aa2000-06-30 05:02:53 +0000712 PyGC_Head *gc;
Thomas Wouters49fd7fa2006-04-21 10:40:58 +0000713 double t1 = 0.0;
Jeremy Hyltonc5007aa2000-06-30 05:02:53 +0000714
Tim Peters93ad66d2003-04-05 17:15:44 +0000715 if (delstr == NULL) {
716 delstr = PyString_InternFromString("__del__");
717 if (delstr == NULL)
718 Py_FatalError("gc couldn't allocate \"__del__\"");
719 }
720
Jeremy Hyltonc5007aa2000-06-30 05:02:53 +0000721 if (debug & DEBUG_STATS) {
Thomas Wouters49fd7fa2006-04-21 10:40:58 +0000722 if (tmod != NULL) {
723 PyObject *f = PyObject_CallMethod(tmod, "time", NULL);
724 if (f == NULL) {
725 PyErr_Clear();
726 }
727 else {
728 t1 = PyFloat_AsDouble(f);
729 Py_DECREF(f);
730 }
731 }
Neil Schemenauer2880ae52002-05-04 05:35:20 +0000732 PySys_WriteStderr("gc: collecting generation %d...\n",
733 generation);
734 PySys_WriteStderr("gc: objects in each generation:");
Thomas Wouters49fd7fa2006-04-21 10:40:58 +0000735 for (i = 0; i < NUM_GENERATIONS; i++)
736 PySys_WriteStderr(" %" PY_FORMAT_SIZE_T "d",
737 gc_list_size(GEN_HEAD(i)));
Neil Schemenauer2880ae52002-05-04 05:35:20 +0000738 PySys_WriteStderr("\n");
739 }
740
741 /* update collection and allocation counters */
742 if (generation+1 < NUM_GENERATIONS)
743 generations[generation+1].count += 1;
744 for (i = 0; i <= generation; i++)
Neil Schemenauerc9051642002-06-28 19:16:04 +0000745 generations[i].count = 0;
Neil Schemenauer2880ae52002-05-04 05:35:20 +0000746
747 /* merge younger generations with one we are currently collecting */
748 for (i = 0; i < generation; i++) {
749 gc_list_merge(GEN_HEAD(i), GEN_HEAD(generation));
750 }
751
752 /* handy references */
753 young = GEN_HEAD(generation);
Tim Peters19b74c72002-07-01 03:52:19 +0000754 if (generation < NUM_GENERATIONS-1)
Neil Schemenauer2880ae52002-05-04 05:35:20 +0000755 old = GEN_HEAD(generation+1);
Tim Peters19b74c72002-07-01 03:52:19 +0000756 else
757 old = young;
Jeremy Hyltonc5007aa2000-06-30 05:02:53 +0000758
759 /* Using ob_refcnt and gc_refs, calculate which objects in the
Tim Petersead8b7a2004-10-30 23:09:22 +0000760 * container set are reachable from outside the set (i.e., have a
Jeremy Hyltonc5007aa2000-06-30 05:02:53 +0000761 * refcount greater than 0 when all the references within the
Tim Petersead8b7a2004-10-30 23:09:22 +0000762 * set are taken into account).
Tim Peters19b74c72002-07-01 03:52:19 +0000763 */
Jeremy Hyltonc5007aa2000-06-30 05:02:53 +0000764 update_refs(young);
765 subtract_refs(young);
766
Tim Peters19b74c72002-07-01 03:52:19 +0000767 /* Leave everything reachable from outside young in young, and move
768 * everything else (in young) to unreachable.
769 * NOTE: This used to move the reachable objects into a reachable
770 * set instead. But most things usually turn out to be reachable,
771 * so it's more efficient to move the unreachable things.
772 */
Jeremy Hyltonc5007aa2000-06-30 05:02:53 +0000773 gc_list_init(&unreachable);
Tim Peters19b74c72002-07-01 03:52:19 +0000774 move_unreachable(young, &unreachable);
Jeremy Hyltonc5007aa2000-06-30 05:02:53 +0000775
Tim Peters19b74c72002-07-01 03:52:19 +0000776 /* Move reachable objects to next generation. */
777 if (young != old)
778 gc_list_merge(young, old);
Jeremy Hyltonc5007aa2000-06-30 05:02:53 +0000779
Tim Peters19b74c72002-07-01 03:52:19 +0000780 /* All objects in unreachable are trash, but objects reachable from
781 * finalizers can't safely be deleted. Python programmers should take
782 * care not to create such things. For Python, finalizers means
Tim Peters403a2032003-11-20 21:21:46 +0000783 * instance objects with __del__ methods. Weakrefs with callbacks
Tim Petersead8b7a2004-10-30 23:09:22 +0000784 * can also call arbitrary Python code but they will be dealt with by
785 * handle_weakrefs().
Tim Petersf6b80452003-04-07 19:21:15 +0000786 */
Jeremy Hyltonc5007aa2000-06-30 05:02:53 +0000787 gc_list_init(&finalizers);
Tim Petersead8b7a2004-10-30 23:09:22 +0000788 move_finalizers(&unreachable, &finalizers);
Tim Petersbf384c22003-04-06 00:11:39 +0000789 /* finalizers contains the unreachable objects with a finalizer;
Tim Peters403a2032003-11-20 21:21:46 +0000790 * unreachable objects reachable *from* those are also uncollectable,
791 * and we move those into the finalizers list too.
Tim Petersbf384c22003-04-06 00:11:39 +0000792 */
Tim Petersf6b80452003-04-07 19:21:15 +0000793 move_finalizer_reachable(&finalizers);
Jeremy Hyltonc5007aa2000-06-30 05:02:53 +0000794
795 /* Collect statistics on collectable objects found and print
Tim Peters403a2032003-11-20 21:21:46 +0000796 * debugging information.
797 */
Tim Petersf6b80452003-04-07 19:21:15 +0000798 for (gc = unreachable.gc.gc_next; gc != &unreachable;
Tim Peters9e4ca102001-10-11 18:31:31 +0000799 gc = gc->gc.gc_next) {
Jeremy Hyltonc5007aa2000-06-30 05:02:53 +0000800 m++;
Jeremy Hylton06257772000-08-31 15:10:24 +0000801 if (debug & DEBUG_COLLECTABLE) {
Neil Schemenauer43411b52001-08-30 00:05:51 +0000802 debug_cycle("collectable", FROM_GC(gc));
Jeremy Hyltonc5007aa2000-06-30 05:02:53 +0000803 }
Thomas Wouters49fd7fa2006-04-21 10:40:58 +0000804 if (tmod != NULL && (debug & DEBUG_STATS)) {
805 PyObject *f = PyObject_CallMethod(tmod, "time", NULL);
806 if (f == NULL) {
807 PyErr_Clear();
808 }
809 else {
810 t1 = PyFloat_AsDouble(f)-t1;
811 Py_DECREF(f);
812 PySys_WriteStderr("gc: %.4fs elapsed.\n", t1);
813 }
814 }
Jeremy Hyltonc5007aa2000-06-30 05:02:53 +0000815 }
Tim Petersead8b7a2004-10-30 23:09:22 +0000816
817 /* Clear weakrefs and invoke callbacks as necessary. */
818 m += handle_weakrefs(&unreachable, old);
819
Tim Petersfb2ab4d2003-04-07 22:41:24 +0000820 /* Call tp_clear on objects in the unreachable set. This will cause
821 * the reference cycles to be broken. It may also cause some objects
822 * in finalizers to be freed.
823 */
Tim Petersf6b80452003-04-07 19:21:15 +0000824 delete_garbage(&unreachable, old);
Jeremy Hyltonc5007aa2000-06-30 05:02:53 +0000825
826 /* Collect statistics on uncollectable objects found and print
827 * debugging information. */
Tim Peters50c61d52003-04-06 01:50:50 +0000828 for (gc = finalizers.gc.gc_next;
Tim Petersbf384c22003-04-06 00:11:39 +0000829 gc != &finalizers;
830 gc = gc->gc.gc_next) {
Jeremy Hyltonc5007aa2000-06-30 05:02:53 +0000831 n++;
Tim Petersbf384c22003-04-06 00:11:39 +0000832 if (debug & DEBUG_UNCOLLECTABLE)
Neil Schemenauer43411b52001-08-30 00:05:51 +0000833 debug_cycle("uncollectable", FROM_GC(gc));
Tim Petersbf384c22003-04-06 00:11:39 +0000834 }
Jeremy Hylton06257772000-08-31 15:10:24 +0000835 if (debug & DEBUG_STATS) {
Thomas Wouters49fd7fa2006-04-21 10:40:58 +0000836 if (m == 0 && n == 0)
Jeremy Hylton06257772000-08-31 15:10:24 +0000837 PySys_WriteStderr("gc: done.\n");
Thomas Wouters49fd7fa2006-04-21 10:40:58 +0000838 else
Neal Norwitze22373d2006-03-06 23:31:56 +0000839 PySys_WriteStderr(
Thomas Wouters49fd7fa2006-04-21 10:40:58 +0000840 "gc: done, "
841 "%" PY_FORMAT_SIZE_T "d unreachable, "
842 "%" PY_FORMAT_SIZE_T "d uncollectable.\n",
Neal Norwitze22373d2006-03-06 23:31:56 +0000843 n+m, n);
Jeremy Hyltonc5007aa2000-06-30 05:02:53 +0000844 }
845
846 /* Append instances in the uncollectable set to a Python
847 * reachable list of garbage. The programmer has to deal with
Tim Petersbf384c22003-04-06 00:11:39 +0000848 * this if they insist on creating this type of structure.
849 */
Tim Petersf6b80452003-04-07 19:21:15 +0000850 (void)handle_finalizers(&finalizers, old);
Jeremy Hyltonc5007aa2000-06-30 05:02:53 +0000851
Jeremy Hyltonb709df32000-09-01 02:47:25 +0000852 if (PyErr_Occurred()) {
Tim Petersf6b80452003-04-07 19:21:15 +0000853 if (gc_str == NULL)
Tim Petersfb2ab4d2003-04-07 22:41:24 +0000854 gc_str = PyString_FromString("garbage collection");
Jeremy Hyltonb709df32000-09-01 02:47:25 +0000855 PyErr_WriteUnraisable(gc_str);
856 Py_FatalError("unexpected exception during garbage collection");
857 }
Jeremy Hyltonc5007aa2000-06-30 05:02:53 +0000858 return n+m;
859}
860
Neal Norwitz7b216c52006-03-04 20:01:53 +0000861static Py_ssize_t
Jeremy Hyltonc5007aa2000-06-30 05:02:53 +0000862collect_generations(void)
863{
Neil Schemenauer2880ae52002-05-04 05:35:20 +0000864 int i;
Neal Norwitz7b216c52006-03-04 20:01:53 +0000865 Py_ssize_t n = 0;
Jeremy Hyltonc5007aa2000-06-30 05:02:53 +0000866
Neil Schemenauer2880ae52002-05-04 05:35:20 +0000867 /* Find the oldest generation (higest numbered) where the count
868 * exceeds the threshold. Objects in the that generation and
869 * generations younger than it will be collected. */
870 for (i = NUM_GENERATIONS-1; i >= 0; i--) {
871 if (generations[i].count > generations[i].threshold) {
872 n = collect(i);
873 break;
Jeremy Hyltonc5007aa2000-06-30 05:02:53 +0000874 }
875 }
876 return n;
877}
878
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +0000879PyDoc_STRVAR(gc_enable__doc__,
Vladimir Marangozovf9d20c32000-08-06 22:45:31 +0000880"enable() -> None\n"
881"\n"
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +0000882"Enable automatic garbage collection.\n");
Jeremy Hyltonc5007aa2000-06-30 05:02:53 +0000883
Vladimir Marangozovf9d20c32000-08-06 22:45:31 +0000884static PyObject *
Tim Peters50c61d52003-04-06 01:50:50 +0000885gc_enable(PyObject *self, PyObject *noargs)
Vladimir Marangozovf9d20c32000-08-06 22:45:31 +0000886{
Vladimir Marangozovf9d20c32000-08-06 22:45:31 +0000887 enabled = 1;
Vladimir Marangozovf9d20c32000-08-06 22:45:31 +0000888 Py_INCREF(Py_None);
889 return Py_None;
890}
891
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +0000892PyDoc_STRVAR(gc_disable__doc__,
Vladimir Marangozovf9d20c32000-08-06 22:45:31 +0000893"disable() -> None\n"
894"\n"
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +0000895"Disable automatic garbage collection.\n");
Vladimir Marangozovf9d20c32000-08-06 22:45:31 +0000896
897static PyObject *
Tim Peters50c61d52003-04-06 01:50:50 +0000898gc_disable(PyObject *self, PyObject *noargs)
Vladimir Marangozovf9d20c32000-08-06 22:45:31 +0000899{
Vladimir Marangozovf9d20c32000-08-06 22:45:31 +0000900 enabled = 0;
Vladimir Marangozovf9d20c32000-08-06 22:45:31 +0000901 Py_INCREF(Py_None);
902 return Py_None;
903}
904
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +0000905PyDoc_STRVAR(gc_isenabled__doc__,
Vladimir Marangozovf9d20c32000-08-06 22:45:31 +0000906"isenabled() -> status\n"
907"\n"
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +0000908"Returns true if automatic garbage collection is enabled.\n");
Vladimir Marangozovf9d20c32000-08-06 22:45:31 +0000909
910static PyObject *
Tim Peters50c61d52003-04-06 01:50:50 +0000911gc_isenabled(PyObject *self, PyObject *noargs)
Vladimir Marangozovf9d20c32000-08-06 22:45:31 +0000912{
Raymond Hettinger674d56b2004-01-04 04:00:13 +0000913 return PyBool_FromLong((long)enabled);
Vladimir Marangozovf9d20c32000-08-06 22:45:31 +0000914}
915
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +0000916PyDoc_STRVAR(gc_collect__doc__,
Barry Warsawd3c38ff2006-03-07 09:46:03 +0000917"collect([generation]) -> n\n"
Jeremy Hyltonc5007aa2000-06-30 05:02:53 +0000918"\n"
Barry Warsawd3c38ff2006-03-07 09:46:03 +0000919"With no arguments, run a full collection. The optional argument\n"
920"may be an integer specifying which generation to collect. A ValueError\n"
921"is raised if the generation number is invalid.\n\n"
922"The number of unreachable objects is returned.\n");
Jeremy Hyltonc5007aa2000-06-30 05:02:53 +0000923
924static PyObject *
Barry Warsawd3c38ff2006-03-07 09:46:03 +0000925gc_collect(PyObject *self, PyObject *args, PyObject *kws)
Jeremy Hyltonc5007aa2000-06-30 05:02:53 +0000926{
Barry Warsawd3c38ff2006-03-07 09:46:03 +0000927 static char *keywords[] = {"generation", NULL};
928 int genarg = NUM_GENERATIONS - 1;
Neal Norwitz7b216c52006-03-04 20:01:53 +0000929 Py_ssize_t n;
Jeremy Hyltonc5007aa2000-06-30 05:02:53 +0000930
Barry Warsawd3c38ff2006-03-07 09:46:03 +0000931 if (!PyArg_ParseTupleAndKeywords(args, kws, "|i", keywords, &genarg))
932 return NULL;
933
934 else if (genarg < 0 || genarg >= NUM_GENERATIONS) {
935 PyErr_SetString(PyExc_ValueError, "invalid generation");
936 return NULL;
937 }
938
Tim Peters50c61d52003-04-06 01:50:50 +0000939 if (collecting)
Neil Schemenauere8c40cb2001-10-31 23:09:35 +0000940 n = 0; /* already collecting, don't do anything */
Neil Schemenauere8c40cb2001-10-31 23:09:35 +0000941 else {
942 collecting = 1;
Barry Warsawd3c38ff2006-03-07 09:46:03 +0000943 n = collect(genarg);
Neil Schemenauere8c40cb2001-10-31 23:09:35 +0000944 collecting = 0;
945 }
Jeremy Hyltonc5007aa2000-06-30 05:02:53 +0000946
Neal Norwitz7b216c52006-03-04 20:01:53 +0000947 return PyInt_FromSsize_t(n);
Jeremy Hyltonc5007aa2000-06-30 05:02:53 +0000948}
949
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +0000950PyDoc_STRVAR(gc_set_debug__doc__,
Jeremy Hyltonc5007aa2000-06-30 05:02:53 +0000951"set_debug(flags) -> None\n"
952"\n"
953"Set the garbage collection debugging flags. Debugging information is\n"
954"written to sys.stderr.\n"
955"\n"
956"flags is an integer and can have the following bits turned on:\n"
957"\n"
958" DEBUG_STATS - Print statistics during collection.\n"
959" DEBUG_COLLECTABLE - Print collectable objects found.\n"
960" DEBUG_UNCOLLECTABLE - Print unreachable but uncollectable objects found.\n"
Jeremy Hyltonc5007aa2000-06-30 05:02:53 +0000961" DEBUG_OBJECTS - Print objects other than instances.\n"
Neil Schemenauer544de1e2000-09-22 15:22:38 +0000962" DEBUG_SAVEALL - Save objects to gc.garbage rather than freeing them.\n"
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +0000963" DEBUG_LEAK - Debug leaking programs (everything but STATS).\n");
Jeremy Hyltonc5007aa2000-06-30 05:02:53 +0000964
965static PyObject *
Vladimir Marangozovf9d20c32000-08-06 22:45:31 +0000966gc_set_debug(PyObject *self, PyObject *args)
Jeremy Hyltonc5007aa2000-06-30 05:02:53 +0000967{
Neil Schemenauer7760cff2000-09-22 22:35:36 +0000968 if (!PyArg_ParseTuple(args, "i:set_debug", &debug))
Jeremy Hyltonc5007aa2000-06-30 05:02:53 +0000969 return NULL;
970
971 Py_INCREF(Py_None);
972 return Py_None;
973}
974
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +0000975PyDoc_STRVAR(gc_get_debug__doc__,
Jeremy Hyltonc5007aa2000-06-30 05:02:53 +0000976"get_debug() -> flags\n"
977"\n"
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +0000978"Get the garbage collection debugging flags.\n");
Jeremy Hyltonc5007aa2000-06-30 05:02:53 +0000979
980static PyObject *
Tim Peters50c61d52003-04-06 01:50:50 +0000981gc_get_debug(PyObject *self, PyObject *noargs)
Jeremy Hyltonc5007aa2000-06-30 05:02:53 +0000982{
Jeremy Hyltonc5007aa2000-06-30 05:02:53 +0000983 return Py_BuildValue("i", debug);
984}
985
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +0000986PyDoc_STRVAR(gc_set_thresh__doc__,
Neal Norwitz2a47c0f2002-01-29 00:53:41 +0000987"set_threshold(threshold0, [threshold1, threshold2]) -> None\n"
Jeremy Hyltonc5007aa2000-06-30 05:02:53 +0000988"\n"
989"Sets the collection thresholds. Setting threshold0 to zero disables\n"
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +0000990"collection.\n");
Jeremy Hyltonc5007aa2000-06-30 05:02:53 +0000991
992static PyObject *
Vladimir Marangozovf9d20c32000-08-06 22:45:31 +0000993gc_set_thresh(PyObject *self, PyObject *args)
Jeremy Hyltonc5007aa2000-06-30 05:02:53 +0000994{
Neil Schemenauer2880ae52002-05-04 05:35:20 +0000995 int i;
996 if (!PyArg_ParseTuple(args, "i|ii:set_threshold",
997 &generations[0].threshold,
998 &generations[1].threshold,
999 &generations[2].threshold))
Jeremy Hyltonc5007aa2000-06-30 05:02:53 +00001000 return NULL;
Neil Schemenauer2880ae52002-05-04 05:35:20 +00001001 for (i = 2; i < NUM_GENERATIONS; i++) {
1002 /* generations higher than 2 get the same threshold */
1003 generations[i].threshold = generations[2].threshold;
1004 }
Jeremy Hyltonc5007aa2000-06-30 05:02:53 +00001005
1006 Py_INCREF(Py_None);
1007 return Py_None;
1008}
1009
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00001010PyDoc_STRVAR(gc_get_thresh__doc__,
Jeremy Hyltonc5007aa2000-06-30 05:02:53 +00001011"get_threshold() -> (threshold0, threshold1, threshold2)\n"
1012"\n"
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00001013"Return the current collection thresholds\n");
Jeremy Hyltonc5007aa2000-06-30 05:02:53 +00001014
1015static PyObject *
Tim Peters50c61d52003-04-06 01:50:50 +00001016gc_get_thresh(PyObject *self, PyObject *noargs)
Jeremy Hyltonc5007aa2000-06-30 05:02:53 +00001017{
Neil Schemenauer2880ae52002-05-04 05:35:20 +00001018 return Py_BuildValue("(iii)",
1019 generations[0].threshold,
1020 generations[1].threshold,
1021 generations[2].threshold);
Jeremy Hyltonc5007aa2000-06-30 05:02:53 +00001022}
1023
Barry Warsawd3c38ff2006-03-07 09:46:03 +00001024PyDoc_STRVAR(gc_get_count__doc__,
1025"get_count() -> (count0, count1, count2)\n"
1026"\n"
1027"Return the current collection counts\n");
1028
1029static PyObject *
1030gc_get_count(PyObject *self, PyObject *noargs)
1031{
1032 return Py_BuildValue("(iii)",
1033 generations[0].count,
1034 generations[1].count,
1035 generations[2].count);
1036}
1037
Neil Schemenauer48c70342001-08-09 15:38:31 +00001038static int
Martin v. Löwis560da622001-11-24 09:24:51 +00001039referrersvisit(PyObject* obj, PyObject *objs)
Neil Schemenauer48c70342001-08-09 15:38:31 +00001040{
Thomas Wouters49fd7fa2006-04-21 10:40:58 +00001041 Py_ssize_t i;
Martin v. Löwisc8fe77b2001-11-29 18:08:31 +00001042 for (i = 0; i < PyTuple_GET_SIZE(objs); i++)
1043 if (PyTuple_GET_ITEM(objs, i) == obj)
1044 return 1;
Neil Schemenauer48c70342001-08-09 15:38:31 +00001045 return 0;
1046}
1047
Neil Schemenauer17e7be62001-08-10 14:46:47 +00001048static int
Martin v. Löwis560da622001-11-24 09:24:51 +00001049gc_referrers_for(PyObject *objs, PyGC_Head *list, PyObject *resultlist)
Neil Schemenauer48c70342001-08-09 15:38:31 +00001050{
1051 PyGC_Head *gc;
1052 PyObject *obj;
1053 traverseproc traverse;
Tim Peters9e4ca102001-10-11 18:31:31 +00001054 for (gc = list->gc.gc_next; gc != list; gc = gc->gc.gc_next) {
Neil Schemenauer43411b52001-08-30 00:05:51 +00001055 obj = FROM_GC(gc);
Neil Schemenauer48c70342001-08-09 15:38:31 +00001056 traverse = obj->ob_type->tp_traverse;
1057 if (obj == objs || obj == resultlist)
1058 continue;
Martin v. Löwis560da622001-11-24 09:24:51 +00001059 if (traverse(obj, (visitproc)referrersvisit, objs)) {
Neil Schemenauer17e7be62001-08-10 14:46:47 +00001060 if (PyList_Append(resultlist, obj) < 0)
1061 return 0; /* error */
Neil Schemenauer48c70342001-08-09 15:38:31 +00001062 }
1063 }
Neil Schemenauer17e7be62001-08-10 14:46:47 +00001064 return 1; /* no error */
Neil Schemenauer48c70342001-08-09 15:38:31 +00001065}
1066
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00001067PyDoc_STRVAR(gc_get_referrers__doc__,
Martin v. Löwis560da622001-11-24 09:24:51 +00001068"get_referrers(*objs) -> list\n\
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00001069Return the list of objects that directly refer to any of objs.");
Neil Schemenauer48c70342001-08-09 15:38:31 +00001070
Neil Schemenauer17e7be62001-08-10 14:46:47 +00001071static PyObject *
Martin v. Löwis560da622001-11-24 09:24:51 +00001072gc_get_referrers(PyObject *self, PyObject *args)
Neil Schemenauer48c70342001-08-09 15:38:31 +00001073{
Neil Schemenauer2880ae52002-05-04 05:35:20 +00001074 int i;
Neil Schemenauer48c70342001-08-09 15:38:31 +00001075 PyObject *result = PyList_New(0);
Thomas Wouters49fd7fa2006-04-21 10:40:58 +00001076 if (!result) return NULL;
1077
Neil Schemenauer2880ae52002-05-04 05:35:20 +00001078 for (i = 0; i < NUM_GENERATIONS; i++) {
1079 if (!(gc_referrers_for(args, GEN_HEAD(i), result))) {
1080 Py_DECREF(result);
1081 return NULL;
1082 }
Neil Schemenauer17e7be62001-08-10 14:46:47 +00001083 }
Neil Schemenauer48c70342001-08-09 15:38:31 +00001084 return result;
1085}
Jeremy Hyltonc5007aa2000-06-30 05:02:53 +00001086
Tim Peters0f81ab62003-04-08 16:39:48 +00001087/* Append obj to list; return true if error (out of memory), false if OK. */
Jeremy Hylton5bd378b2003-04-03 16:28:38 +00001088static int
Tim Peters730f5532003-04-08 17:17:17 +00001089referentsvisit(PyObject *obj, PyObject *list)
Jeremy Hylton5bd378b2003-04-03 16:28:38 +00001090{
Tim Peters0f81ab62003-04-08 16:39:48 +00001091 return PyList_Append(list, obj) < 0;
Jeremy Hylton5bd378b2003-04-03 16:28:38 +00001092}
1093
Tim Peters730f5532003-04-08 17:17:17 +00001094PyDoc_STRVAR(gc_get_referents__doc__,
1095"get_referents(*objs) -> list\n\
Jeremy Hylton059b0942003-04-03 16:29:13 +00001096Return the list of objects that are directly referred to by objs.");
Jeremy Hylton5bd378b2003-04-03 16:28:38 +00001097
1098static PyObject *
Tim Peters730f5532003-04-08 17:17:17 +00001099gc_get_referents(PyObject *self, PyObject *args)
Jeremy Hylton5bd378b2003-04-03 16:28:38 +00001100{
Thomas Wouters49fd7fa2006-04-21 10:40:58 +00001101 Py_ssize_t i;
Jeremy Hylton5bd378b2003-04-03 16:28:38 +00001102 PyObject *result = PyList_New(0);
Tim Peters0f81ab62003-04-08 16:39:48 +00001103
1104 if (result == NULL)
1105 return NULL;
1106
Jeremy Hylton5bd378b2003-04-03 16:28:38 +00001107 for (i = 0; i < PyTuple_GET_SIZE(args); i++) {
Tim Peters0f81ab62003-04-08 16:39:48 +00001108 traverseproc traverse;
Tim Peters93ad66d2003-04-05 17:15:44 +00001109 PyObject *obj = PyTuple_GET_ITEM(args, i);
Tim Peters0f81ab62003-04-08 16:39:48 +00001110
1111 if (! PyObject_IS_GC(obj))
Jeremy Hylton5bd378b2003-04-03 16:28:38 +00001112 continue;
Tim Peters0f81ab62003-04-08 16:39:48 +00001113 traverse = obj->ob_type->tp_traverse;
1114 if (! traverse)
1115 continue;
Tim Peters730f5532003-04-08 17:17:17 +00001116 if (traverse(obj, (visitproc)referentsvisit, result)) {
Tim Peters0f81ab62003-04-08 16:39:48 +00001117 Py_DECREF(result);
Jeremy Hylton5bd378b2003-04-03 16:28:38 +00001118 return NULL;
Tim Peters0f81ab62003-04-08 16:39:48 +00001119 }
Jeremy Hylton5bd378b2003-04-03 16:28:38 +00001120 }
1121 return result;
1122}
1123
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00001124PyDoc_STRVAR(gc_get_objects__doc__,
Neil Schemenauerc7c8d8e2001-08-09 15:58:59 +00001125"get_objects() -> [...]\n"
1126"\n"
1127"Return a list of objects tracked by the collector (excluding the list\n"
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00001128"returned).\n");
Neil Schemenauerc7c8d8e2001-08-09 15:58:59 +00001129
Neil Schemenauerc7c8d8e2001-08-09 15:58:59 +00001130static PyObject *
Tim Peters50c61d52003-04-06 01:50:50 +00001131gc_get_objects(PyObject *self, PyObject *noargs)
Neil Schemenauerc7c8d8e2001-08-09 15:58:59 +00001132{
Neil Schemenauer2880ae52002-05-04 05:35:20 +00001133 int i;
Neil Schemenauerc7c8d8e2001-08-09 15:58:59 +00001134 PyObject* result;
1135
Neil Schemenauerc7c8d8e2001-08-09 15:58:59 +00001136 result = PyList_New(0);
Tim Peters50c61d52003-04-06 01:50:50 +00001137 if (result == NULL)
Martin v. Löwisf8a6f242001-12-02 18:31:02 +00001138 return NULL;
Neil Schemenauer2880ae52002-05-04 05:35:20 +00001139 for (i = 0; i < NUM_GENERATIONS; i++) {
1140 if (append_objects(result, GEN_HEAD(i))) {
1141 Py_DECREF(result);
1142 return NULL;
1143 }
Martin v. Löwis155aad12001-12-02 12:21:34 +00001144 }
Neil Schemenauerc7c8d8e2001-08-09 15:58:59 +00001145 return result;
1146}
1147
1148
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00001149PyDoc_STRVAR(gc__doc__,
Jeremy Hyltonc5007aa2000-06-30 05:02:53 +00001150"This module provides access to the garbage collector for reference cycles.\n"
1151"\n"
Vladimir Marangozovf9d20c32000-08-06 22:45:31 +00001152"enable() -- Enable automatic garbage collection.\n"
1153"disable() -- Disable automatic garbage collection.\n"
1154"isenabled() -- Returns true if automatic collection is enabled.\n"
Jeremy Hyltonc5007aa2000-06-30 05:02:53 +00001155"collect() -- Do a full collection right now.\n"
1156"set_debug() -- Set debugging flags.\n"
1157"get_debug() -- Get debugging flags.\n"
1158"set_threshold() -- Set the collection thresholds.\n"
1159"get_threshold() -- Return the current the collection thresholds.\n"
Neil Schemenauerc7c8d8e2001-08-09 15:58:59 +00001160"get_objects() -- Return a list of all objects tracked by the collector.\n"
Jeremy Hylton5bd378b2003-04-03 16:28:38 +00001161"get_referrers() -- Return the list of objects that refer to an object.\n"
Tim Peters730f5532003-04-08 17:17:17 +00001162"get_referents() -- Return the list of objects that an object refers to.\n");
Jeremy Hyltonc5007aa2000-06-30 05:02:53 +00001163
1164static PyMethodDef GcMethods[] = {
Tim Peters50c61d52003-04-06 01:50:50 +00001165 {"enable", gc_enable, METH_NOARGS, gc_enable__doc__},
1166 {"disable", gc_disable, METH_NOARGS, gc_disable__doc__},
1167 {"isenabled", gc_isenabled, METH_NOARGS, gc_isenabled__doc__},
Vladimir Marangozovf9d20c32000-08-06 22:45:31 +00001168 {"set_debug", gc_set_debug, METH_VARARGS, gc_set_debug__doc__},
Tim Peters50c61d52003-04-06 01:50:50 +00001169 {"get_debug", gc_get_debug, METH_NOARGS, gc_get_debug__doc__},
Barry Warsawd3c38ff2006-03-07 09:46:03 +00001170 {"get_count", gc_get_count, METH_NOARGS, gc_get_count__doc__},
Vladimir Marangozovf9d20c32000-08-06 22:45:31 +00001171 {"set_threshold", gc_set_thresh, METH_VARARGS, gc_set_thresh__doc__},
Tim Peters50c61d52003-04-06 01:50:50 +00001172 {"get_threshold", gc_get_thresh, METH_NOARGS, gc_get_thresh__doc__},
Barry Warsawd3c38ff2006-03-07 09:46:03 +00001173 {"collect", (PyCFunction)gc_collect,
1174 METH_VARARGS | METH_KEYWORDS, gc_collect__doc__},
Tim Peters50c61d52003-04-06 01:50:50 +00001175 {"get_objects", gc_get_objects,METH_NOARGS, gc_get_objects__doc__},
Martin v. Löwis560da622001-11-24 09:24:51 +00001176 {"get_referrers", gc_get_referrers, METH_VARARGS,
1177 gc_get_referrers__doc__},
Tim Peters730f5532003-04-08 17:17:17 +00001178 {"get_referents", gc_get_referents, METH_VARARGS,
1179 gc_get_referents__doc__},
Jeremy Hyltonc5007aa2000-06-30 05:02:53 +00001180 {NULL, NULL} /* Sentinel */
1181};
1182
Jason Tishler6bc06ec2003-09-04 11:59:50 +00001183PyMODINIT_FUNC
Jeremy Hyltonc5007aa2000-06-30 05:02:53 +00001184initgc(void)
1185{
1186 PyObject *m;
Jeremy Hyltonc5007aa2000-06-30 05:02:53 +00001187
1188 m = Py_InitModule4("gc",
1189 GcMethods,
1190 gc__doc__,
1191 NULL,
1192 PYTHON_API_VERSION);
Neal Norwitz1ac754f2006-01-19 06:09:39 +00001193 if (m == NULL)
1194 return;
Tim Peters11558872003-04-06 23:30:52 +00001195
Jeremy Hyltonc5007aa2000-06-30 05:02:53 +00001196 if (garbage == NULL) {
1197 garbage = PyList_New(0);
Tim Peters11558872003-04-06 23:30:52 +00001198 if (garbage == NULL)
1199 return;
Jeremy Hyltonc5007aa2000-06-30 05:02:53 +00001200 }
Neil Schemenauer3b1cbf92005-06-18 17:37:06 +00001201 Py_INCREF(garbage);
Tim Peters11558872003-04-06 23:30:52 +00001202 if (PyModule_AddObject(m, "garbage", garbage) < 0)
1203 return;
Thomas Wouters477c8d52006-05-27 19:21:47 +00001204
1205 /* Importing can't be done in collect() because collect()
1206 * can be called via PyGC_Collect() in Py_Finalize().
1207 * This wouldn't be a problem, except that <initialized> is
1208 * reset to 0 before calling collect which trips up
1209 * the import and triggers an assertion.
1210 */
1211 if (tmod == NULL) {
1212 tmod = PyImport_ImportModule("time");
1213 if (tmod == NULL)
1214 PyErr_Clear();
1215 }
1216
Tim Peters11558872003-04-06 23:30:52 +00001217#define ADD_INT(NAME) if (PyModule_AddIntConstant(m, #NAME, NAME) < 0) return
1218 ADD_INT(DEBUG_STATS);
1219 ADD_INT(DEBUG_COLLECTABLE);
1220 ADD_INT(DEBUG_UNCOLLECTABLE);
Tim Peters11558872003-04-06 23:30:52 +00001221 ADD_INT(DEBUG_OBJECTS);
1222 ADD_INT(DEBUG_SAVEALL);
1223 ADD_INT(DEBUG_LEAK);
1224#undef ADD_INT
Jeremy Hyltonc5007aa2000-06-30 05:02:53 +00001225}
1226
Guido van Rossume13ddc92003-04-17 17:29:22 +00001227/* API to invoke gc.collect() from C */
Neal Norwitz7b216c52006-03-04 20:01:53 +00001228Py_ssize_t
Guido van Rossume13ddc92003-04-17 17:29:22 +00001229PyGC_Collect(void)
1230{
Neal Norwitz7b216c52006-03-04 20:01:53 +00001231 Py_ssize_t n;
Guido van Rossume13ddc92003-04-17 17:29:22 +00001232
1233 if (collecting)
1234 n = 0; /* already collecting, don't do anything */
1235 else {
1236 collecting = 1;
1237 n = collect(NUM_GENERATIONS - 1);
1238 collecting = 0;
1239 }
1240
1241 return n;
1242}
1243
Neil Schemenauer43411b52001-08-30 00:05:51 +00001244/* for debugging */
Guido van Rossume13ddc92003-04-17 17:29:22 +00001245void
1246_PyGC_Dump(PyGC_Head *g)
Neil Schemenauer43411b52001-08-30 00:05:51 +00001247{
1248 _PyObject_Dump(FROM_GC(g));
1249}
1250
Neil Schemenauer43411b52001-08-30 00:05:51 +00001251/* extension modules might be compiled with GC support so these
1252 functions must always be available */
1253
Neil Schemenauerfec4eb12002-04-12 02:41:03 +00001254#undef PyObject_GC_Track
1255#undef PyObject_GC_UnTrack
1256#undef PyObject_GC_Del
1257#undef _PyObject_GC_Malloc
1258
Neil Schemenauer43411b52001-08-30 00:05:51 +00001259void
Neil Schemenauerfec4eb12002-04-12 02:41:03 +00001260PyObject_GC_Track(void *op)
Neil Schemenauer43411b52001-08-30 00:05:51 +00001261{
1262 _PyObject_GC_TRACK(op);
1263}
1264
Neil Schemenauerfec4eb12002-04-12 02:41:03 +00001265/* for binary compatibility with 2.2 */
Neil Schemenauer43411b52001-08-30 00:05:51 +00001266void
Neil Schemenauerfec4eb12002-04-12 02:41:03 +00001267_PyObject_GC_Track(PyObject *op)
1268{
1269 PyObject_GC_Track(op);
1270}
1271
1272void
1273PyObject_GC_UnTrack(void *op)
Neil Schemenauer43411b52001-08-30 00:05:51 +00001274{
Tim Peters803526b2002-07-07 05:13:56 +00001275 /* Obscure: the Py_TRASHCAN mechanism requires that we be able to
1276 * call PyObject_GC_UnTrack twice on an object.
1277 */
Neil Schemenauera2b11ec2002-05-21 15:53:24 +00001278 if (IS_TRACKED(op))
Guido van Rossumff413af2002-03-28 20:34:59 +00001279 _PyObject_GC_UNTRACK(op);
Neil Schemenauer43411b52001-08-30 00:05:51 +00001280}
1281
Neil Schemenauerfec4eb12002-04-12 02:41:03 +00001282/* for binary compatibility with 2.2 */
1283void
1284_PyObject_GC_UnTrack(PyObject *op)
1285{
1286 PyObject_GC_UnTrack(op);
1287}
1288
Neil Schemenauer43411b52001-08-30 00:05:51 +00001289PyObject *
Neil Schemenauerfec4eb12002-04-12 02:41:03 +00001290_PyObject_GC_Malloc(size_t basicsize)
Neil Schemenauer43411b52001-08-30 00:05:51 +00001291{
1292 PyObject *op;
Thomas Wouters49fd7fa2006-04-21 10:40:58 +00001293 PyGC_Head *g = (PyGC_Head *)PyObject_MALLOC(
1294 sizeof(PyGC_Head) + basicsize);
Neil Schemenauer43411b52001-08-30 00:05:51 +00001295 if (g == NULL)
Jeremy Hylton8a135182002-06-06 23:23:55 +00001296 return PyErr_NoMemory();
Tim Petersea405632002-07-02 00:52:30 +00001297 g->gc.gc_refs = GC_UNTRACKED;
Neil Schemenauer2880ae52002-05-04 05:35:20 +00001298 generations[0].count++; /* number of allocated GC objects */
1299 if (generations[0].count > generations[0].threshold &&
Neil Schemenauer43411b52001-08-30 00:05:51 +00001300 enabled &&
Neil Schemenauer2880ae52002-05-04 05:35:20 +00001301 generations[0].threshold &&
Neil Schemenauer43411b52001-08-30 00:05:51 +00001302 !collecting &&
1303 !PyErr_Occurred()) {
1304 collecting = 1;
Neil Schemenauer2880ae52002-05-04 05:35:20 +00001305 collect_generations();
Neil Schemenauer43411b52001-08-30 00:05:51 +00001306 collecting = 0;
1307 }
1308 op = FROM_GC(g);
Neil Schemenauer43411b52001-08-30 00:05:51 +00001309 return op;
1310}
1311
1312PyObject *
1313_PyObject_GC_New(PyTypeObject *tp)
1314{
Neil Schemenauerfec4eb12002-04-12 02:41:03 +00001315 PyObject *op = _PyObject_GC_Malloc(_PyObject_SIZE(tp));
Tim Petersfa8efab2002-04-28 01:57:25 +00001316 if (op != NULL)
1317 op = PyObject_INIT(op, tp);
1318 return op;
Neil Schemenauer43411b52001-08-30 00:05:51 +00001319}
1320
1321PyVarObject *
Martin v. Löwis18e16552006-02-15 17:27:45 +00001322_PyObject_GC_NewVar(PyTypeObject *tp, Py_ssize_t nitems)
Neil Schemenauer43411b52001-08-30 00:05:51 +00001323{
Neil Schemenauerfec4eb12002-04-12 02:41:03 +00001324 const size_t size = _PyObject_VAR_SIZE(tp, nitems);
1325 PyVarObject *op = (PyVarObject *) _PyObject_GC_Malloc(size);
Tim Petersfa8efab2002-04-28 01:57:25 +00001326 if (op != NULL)
1327 op = PyObject_INIT_VAR(op, tp, nitems);
1328 return op;
Neil Schemenauer43411b52001-08-30 00:05:51 +00001329}
1330
1331PyVarObject *
Martin v. Löwis41290682006-02-16 14:56:14 +00001332_PyObject_GC_Resize(PyVarObject *op, Py_ssize_t nitems)
Neil Schemenauer43411b52001-08-30 00:05:51 +00001333{
Tim Petersf2a67da2001-10-07 03:54:51 +00001334 const size_t basicsize = _PyObject_VAR_SIZE(op->ob_type, nitems);
Neil Schemenauer43411b52001-08-30 00:05:51 +00001335 PyGC_Head *g = AS_GC(op);
Thomas Wouters49fd7fa2006-04-21 10:40:58 +00001336 g = (PyGC_Head *)PyObject_REALLOC(g, sizeof(PyGC_Head) + basicsize);
Neil Schemenauer43411b52001-08-30 00:05:51 +00001337 if (g == NULL)
1338 return (PyVarObject *)PyErr_NoMemory();
1339 op = (PyVarObject *) FROM_GC(g);
Tim Peters6d483d32001-10-06 21:27:34 +00001340 op->ob_size = nitems;
Neil Schemenauer43411b52001-08-30 00:05:51 +00001341 return op;
1342}
1343
1344void
Neil Schemenauerfec4eb12002-04-12 02:41:03 +00001345PyObject_GC_Del(void *op)
Neil Schemenauer43411b52001-08-30 00:05:51 +00001346{
Neil Schemenauer43411b52001-08-30 00:05:51 +00001347 PyGC_Head *g = AS_GC(op);
Neil Schemenauera2b11ec2002-05-21 15:53:24 +00001348 if (IS_TRACKED(op))
Neil Schemenauer43411b52001-08-30 00:05:51 +00001349 gc_list_remove(g);
Neil Schemenauer2880ae52002-05-04 05:35:20 +00001350 if (generations[0].count > 0) {
1351 generations[0].count--;
Neil Schemenauer43411b52001-08-30 00:05:51 +00001352 }
Neil Schemenauerfec4eb12002-04-12 02:41:03 +00001353 PyObject_FREE(g);
Neil Schemenauer43411b52001-08-30 00:05:51 +00001354}
1355
Neil Schemenauerfec4eb12002-04-12 02:41:03 +00001356/* for binary compatibility with 2.2 */
1357#undef _PyObject_GC_Del
1358void
1359_PyObject_GC_Del(PyObject *op)
1360{
1361 PyObject_GC_Del(op);
1362}