blob: 1876e93884dda3b2c3499927d01fbcb63739d16b [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/
Neil Schemenauera7024e92008-07-15 19:24:01 +000012
13 The following mailing list threads provide a historical perspective on
14 the design of this module. Note that a fair amount of refinement has
15 occurred since those discussions.
16
17 http://mail.python.org/pipermail/python-dev/2000-March/002385.html
18 http://mail.python.org/pipermail/python-dev/2000-March/002434.html
19 http://mail.python.org/pipermail/python-dev/2000-March/002497.html
Jeremy Hyltonc5007aa2000-06-30 05:02:53 +000020
21 For a highlevel view of the collection process, read the collect
22 function.
23
Jeremy Hyltonc5007aa2000-06-30 05:02:53 +000024*/
25
Jeremy Hyltonc5007aa2000-06-30 05:02:53 +000026#include "Python.h"
Antoine Pitrouf95a1b32010-05-09 15:52:27 +000027#include "frameobject.h" /* for PyFrame_ClearFreeList */
Jeremy Hyltonc5007aa2000-06-30 05:02:53 +000028
Neil Schemenauer43411b52001-08-30 00:05:51 +000029/* Get an object's GC head */
30#define AS_GC(o) ((PyGC_Head *)(o)-1)
31
32/* Get the object given the GC head */
33#define FROM_GC(g) ((PyObject *)(((PyGC_Head *)g)+1))
34
Jeremy Hyltonc5007aa2000-06-30 05:02:53 +000035/*** Global GC state ***/
36
Neil Schemenauer2880ae52002-05-04 05:35:20 +000037struct gc_generation {
Antoine Pitrouf95a1b32010-05-09 15:52:27 +000038 PyGC_Head head;
39 int threshold; /* collection threshold */
40 int count; /* count of allocations or collections of younger
41 generations */
Neil Schemenauer2880ae52002-05-04 05:35:20 +000042};
43
44#define NUM_GENERATIONS 3
45#define GEN_HEAD(n) (&generations[n].head)
46
Jeremy Hyltonc5007aa2000-06-30 05:02:53 +000047/* linked lists of container objects */
Neil Schemenauer2880ae52002-05-04 05:35:20 +000048static struct gc_generation generations[NUM_GENERATIONS] = {
Antoine Pitrouf95a1b32010-05-09 15:52:27 +000049 /* PyGC_Head, threshold, count */
50 {{{GEN_HEAD(0), GEN_HEAD(0), 0}}, 700, 0},
51 {{{GEN_HEAD(1), GEN_HEAD(1), 0}}, 10, 0},
52 {{{GEN_HEAD(2), GEN_HEAD(2), 0}}, 10, 0},
Neil Schemenauer2880ae52002-05-04 05:35:20 +000053};
Jeremy Hyltonc5007aa2000-06-30 05:02:53 +000054
Neil Schemenauer2880ae52002-05-04 05:35:20 +000055PyGC_Head *_PyGC_generation0 = GEN_HEAD(0);
56
Vladimir Marangozovf9d20c32000-08-06 22:45:31 +000057static int enabled = 1; /* automatic collection enabled? */
Jeremy Hyltonc5007aa2000-06-30 05:02:53 +000058
Neil Schemenauer43411b52001-08-30 00:05:51 +000059/* true if we are currently running the collector */
Tim Petersbf384c22003-04-06 00:11:39 +000060static int collecting = 0;
Neil Schemenauer43411b52001-08-30 00:05:51 +000061
Tim Peters6fc13d92002-07-02 18:12:35 +000062/* list of uncollectable objects */
Tim Petersbf384c22003-04-06 00:11:39 +000063static PyObject *garbage = NULL;
Tim Peters6fc13d92002-07-02 18:12:35 +000064
65/* Python string to use if unhandled exception occurs */
Tim Petersbf384c22003-04-06 00:11:39 +000066static PyObject *gc_str = NULL;
Tim Peters6fc13d92002-07-02 18:12:35 +000067
Tim Peters93ad66d2003-04-05 17:15:44 +000068/* Python string used to look for __del__ attribute. */
69static PyObject *delstr = NULL;
Jeremy Hyltonce136e92003-04-04 19:59:06 +000070
Antoine Pitrou14b78f52009-01-09 22:27:08 +000071/* This is the number of objects who survived the last full collection. It
72 approximates the number of long lived objects tracked by the GC.
73
74 (by "full collection", we mean a collection of the oldest generation).
75*/
76static Py_ssize_t long_lived_total = 0;
77
78/* This is the number of objects who survived all "non-full" collections,
79 and are awaiting to undergo a full collection for the first time.
80
81*/
82static Py_ssize_t long_lived_pending = 0;
83
84/*
85 NOTE: about the counting of long-lived objects.
86
87 To limit the cost of garbage collection, there are two strategies;
88 - make each collection faster, e.g. by scanning fewer objects
89 - do less collections
90 This heuristic is about the latter strategy.
91
92 In addition to the various configurable thresholds, we only trigger a
93 full collection if the ratio
Antoine Pitrouf95a1b32010-05-09 15:52:27 +000094 long_lived_pending / long_lived_total
Antoine Pitrou14b78f52009-01-09 22:27:08 +000095 is above a given value (hardwired to 25%).
96
97 The reason is that, while "non-full" collections (i.e., collections of
98 the young and middle generations) will always examine roughly the same
99 number of objects -- determined by the aforementioned thresholds --,
100 the cost of a full collection is proportional to the total number of
101 long-lived objects, which is virtually unbounded.
102
103 Indeed, it has been remarked that doing a full collection every
104 <constant number> of object creations entails a dramatic performance
105 degradation in workloads which consist in creating and storing lots of
106 long-lived objects (e.g. building a large list of GC-tracked objects would
107 show quadratic performance, instead of linear as expected: see issue #4074).
108
109 Using the above ratio, instead, yields amortized linear performance in
110 the total number of objects (the effect of which can be summarized
111 thusly: "each full garbage collection is more and more costly as the
112 number of objects grows, but we do fewer and fewer of them").
113
114 This heuristic was suggested by Martin von Löwis on python-dev in
115 June 2008. His original analysis and proposal can be found at:
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000116 http://mail.python.org/pipermail/python-dev/2008-June/080579.html
Antoine Pitrou14b78f52009-01-09 22:27:08 +0000117*/
118
119
Jeremy Hyltonc5007aa2000-06-30 05:02:53 +0000120/* set for debugging information */
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000121#define DEBUG_STATS (1<<0) /* print collection statistics */
122#define DEBUG_COLLECTABLE (1<<1) /* print collectable objects */
123#define DEBUG_UNCOLLECTABLE (1<<2) /* print uncollectable objects */
124#define DEBUG_SAVEALL (1<<5) /* save all garbage in gc.garbage */
125#define DEBUG_LEAK DEBUG_COLLECTABLE | \
126 DEBUG_UNCOLLECTABLE | \
127 DEBUG_SAVEALL
Jeremy Hyltonb709df32000-09-01 02:47:25 +0000128static int debug;
Thomas Wouters477c8d52006-05-27 19:21:47 +0000129static PyObject *tmod = NULL;
Jeremy Hyltonc5007aa2000-06-30 05:02:53 +0000130
Tim Peters6fc13d92002-07-02 18:12:35 +0000131/*--------------------------------------------------------------------------
132gc_refs values.
Neil Schemenauer43411b52001-08-30 00:05:51 +0000133
Tim Peters6fc13d92002-07-02 18:12:35 +0000134Between collections, every gc'ed object has one of two gc_refs values:
135
136GC_UNTRACKED
137 The initial state; objects returned by PyObject_GC_Malloc are in this
138 state. The object doesn't live in any generation list, and its
139 tp_traverse slot must not be called.
140
141GC_REACHABLE
142 The object lives in some generation list, and its tp_traverse is safe to
143 call. An object transitions to GC_REACHABLE when PyObject_GC_Track
144 is called.
145
146During a collection, gc_refs can temporarily take on other states:
147
148>= 0
149 At the start of a collection, update_refs() copies the true refcount
150 to gc_refs, for each object in the generation being collected.
151 subtract_refs() then adjusts gc_refs so that it equals the number of
152 times an object is referenced directly from outside the generation
153 being collected.
Martin v. Löwis774348c2002-11-09 19:54:06 +0000154 gc_refs remains >= 0 throughout these steps.
Tim Peters6fc13d92002-07-02 18:12:35 +0000155
156GC_TENTATIVELY_UNREACHABLE
157 move_unreachable() then moves objects not reachable (whether directly or
158 indirectly) from outside the generation into an "unreachable" set.
159 Objects that are found to be reachable have gc_refs set to GC_REACHABLE
160 again. Objects that are found to be unreachable have gc_refs set to
161 GC_TENTATIVELY_UNREACHABLE. It's "tentatively" because the pass doing
162 this can't be sure until it ends, and GC_TENTATIVELY_UNREACHABLE may
163 transition back to GC_REACHABLE.
164
165 Only objects with GC_TENTATIVELY_UNREACHABLE still set are candidates
166 for collection. If it's decided not to collect such an object (e.g.,
167 it has a __del__ method), its gc_refs is restored to GC_REACHABLE again.
168----------------------------------------------------------------------------
169*/
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000170#define GC_UNTRACKED _PyGC_REFS_UNTRACKED
171#define GC_REACHABLE _PyGC_REFS_REACHABLE
172#define GC_TENTATIVELY_UNREACHABLE _PyGC_REFS_TENTATIVELY_UNREACHABLE
Tim Peters19b74c72002-07-01 03:52:19 +0000173
Tim Peters6fc13d92002-07-02 18:12:35 +0000174#define IS_TRACKED(o) ((AS_GC(o))->gc.gc_refs != GC_UNTRACKED)
Tim Peters19b74c72002-07-01 03:52:19 +0000175#define IS_REACHABLE(o) ((AS_GC(o))->gc.gc_refs == GC_REACHABLE)
176#define IS_TENTATIVELY_UNREACHABLE(o) ( \
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000177 (AS_GC(o))->gc.gc_refs == GC_TENTATIVELY_UNREACHABLE)
Neil Schemenauera2b11ec2002-05-21 15:53:24 +0000178
Jeremy Hyltonc5007aa2000-06-30 05:02:53 +0000179/*** list functions ***/
180
181static void
182gc_list_init(PyGC_Head *list)
183{
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000184 list->gc.gc_prev = list;
185 list->gc.gc_next = list;
Jeremy Hyltonc5007aa2000-06-30 05:02:53 +0000186}
187
Neil Schemenauer2880ae52002-05-04 05:35:20 +0000188static int
189gc_list_is_empty(PyGC_Head *list)
190{
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000191 return (list->gc.gc_next == list);
Neil Schemenauer2880ae52002-05-04 05:35:20 +0000192}
193
Tim Peterse2d59182004-11-01 01:39:08 +0000194#if 0
195/* This became unused after gc_list_move() was introduced. */
196/* Append `node` to `list`. */
Jeremy Hyltonc5007aa2000-06-30 05:02:53 +0000197static void
198gc_list_append(PyGC_Head *node, PyGC_Head *list)
199{
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000200 node->gc.gc_next = list;
201 node->gc.gc_prev = list->gc.gc_prev;
202 node->gc.gc_prev->gc.gc_next = node;
203 list->gc.gc_prev = node;
Jeremy Hyltonc5007aa2000-06-30 05:02:53 +0000204}
Tim Peterse2d59182004-11-01 01:39:08 +0000205#endif
Jeremy Hyltonc5007aa2000-06-30 05:02:53 +0000206
Tim Peterse2d59182004-11-01 01:39:08 +0000207/* Remove `node` from the gc list it's currently in. */
Jeremy Hyltonc5007aa2000-06-30 05:02:53 +0000208static void
209gc_list_remove(PyGC_Head *node)
210{
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000211 node->gc.gc_prev->gc.gc_next = node->gc.gc_next;
212 node->gc.gc_next->gc.gc_prev = node->gc.gc_prev;
213 node->gc.gc_next = NULL; /* object is not currently tracked */
Jeremy Hyltonc5007aa2000-06-30 05:02:53 +0000214}
215
Tim Peterse2d59182004-11-01 01:39:08 +0000216/* Move `node` from the gc list it's currently in (which is not explicitly
217 * named here) to the end of `list`. This is semantically the same as
218 * gc_list_remove(node) followed by gc_list_append(node, list).
219 */
220static void
221gc_list_move(PyGC_Head *node, PyGC_Head *list)
222{
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000223 PyGC_Head *new_prev;
224 PyGC_Head *current_prev = node->gc.gc_prev;
225 PyGC_Head *current_next = node->gc.gc_next;
226 /* Unlink from current list. */
227 current_prev->gc.gc_next = current_next;
228 current_next->gc.gc_prev = current_prev;
229 /* Relink at end of new list. */
230 new_prev = node->gc.gc_prev = list->gc.gc_prev;
231 new_prev->gc.gc_next = list->gc.gc_prev = node;
232 node->gc.gc_next = list;
Tim Peterse2d59182004-11-01 01:39:08 +0000233}
234
235/* append list `from` onto list `to`; `from` becomes an empty list */
Jeremy Hyltonc5007aa2000-06-30 05:02:53 +0000236static void
237gc_list_merge(PyGC_Head *from, PyGC_Head *to)
238{
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000239 PyGC_Head *tail;
240 assert(from != to);
241 if (!gc_list_is_empty(from)) {
242 tail = to->gc.gc_prev;
243 tail->gc.gc_next = from->gc.gc_next;
244 tail->gc.gc_next->gc.gc_prev = tail;
245 to->gc.gc_prev = from->gc.gc_prev;
246 to->gc.gc_prev->gc.gc_next = to;
247 }
248 gc_list_init(from);
Jeremy Hyltonc5007aa2000-06-30 05:02:53 +0000249}
250
Neal Norwitz7b216c52006-03-04 20:01:53 +0000251static Py_ssize_t
Jeremy Hyltonc5007aa2000-06-30 05:02:53 +0000252gc_list_size(PyGC_Head *list)
253{
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000254 PyGC_Head *gc;
255 Py_ssize_t n = 0;
256 for (gc = list->gc.gc_next; gc != list; gc = gc->gc.gc_next) {
257 n++;
258 }
259 return n;
Jeremy Hyltonc5007aa2000-06-30 05:02:53 +0000260}
261
Tim Peters259272b2003-04-06 19:41:39 +0000262/* Append objects in a GC list to a Python list.
263 * Return 0 if all OK, < 0 if error (out of memory for list).
264 */
265static int
266append_objects(PyObject *py_list, PyGC_Head *gc_list)
267{
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000268 PyGC_Head *gc;
269 for (gc = gc_list->gc.gc_next; gc != gc_list; gc = gc->gc.gc_next) {
270 PyObject *op = FROM_GC(gc);
271 if (op != py_list) {
272 if (PyList_Append(py_list, op)) {
273 return -1; /* exception */
274 }
275 }
276 }
277 return 0;
Tim Peters259272b2003-04-06 19:41:39 +0000278}
279
Jeremy Hyltonc5007aa2000-06-30 05:02:53 +0000280/*** end of list stuff ***/
281
282
Tim Peters19b74c72002-07-01 03:52:19 +0000283/* Set all gc_refs = ob_refcnt. After this, gc_refs is > 0 for all objects
284 * in containers, and is GC_REACHABLE for all tracked gc objects not in
285 * containers.
Tim Peters88396172002-06-30 17:56:40 +0000286 */
Jeremy Hyltonc5007aa2000-06-30 05:02:53 +0000287static void
288update_refs(PyGC_Head *containers)
289{
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000290 PyGC_Head *gc = containers->gc.gc_next;
291 for (; gc != containers; gc = gc->gc.gc_next) {
292 assert(gc->gc.gc_refs == GC_REACHABLE);
293 gc->gc.gc_refs = Py_REFCNT(FROM_GC(gc));
294 /* Python's cyclic gc should never see an incoming refcount
295 * of 0: if something decref'ed to 0, it should have been
296 * deallocated immediately at that time.
297 * Possible cause (if the assert triggers): a tp_dealloc
298 * routine left a gc-aware object tracked during its teardown
299 * phase, and did something-- or allowed something to happen --
300 * that called back into Python. gc can trigger then, and may
301 * see the still-tracked dying object. Before this assert
302 * was added, such mistakes went on to allow gc to try to
303 * delete the object again. In a debug build, that caused
304 * a mysterious segfault, when _Py_ForgetReference tried
305 * to remove the object from the doubly-linked list of all
306 * objects a second time. In a release build, an actual
307 * double deallocation occurred, which leads to corruption
308 * of the allocator's internal bookkeeping pointers. That's
309 * so serious that maybe this should be a release-build
310 * check instead of an assert?
311 */
312 assert(gc->gc.gc_refs != 0);
313 }
Jeremy Hyltonc5007aa2000-06-30 05:02:53 +0000314}
315
Tim Peters19b74c72002-07-01 03:52:19 +0000316/* A traversal callback for subtract_refs. */
Jeremy Hyltonc5007aa2000-06-30 05:02:53 +0000317static int
318visit_decref(PyObject *op, void *data)
319{
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000320 assert(op != NULL);
321 if (PyObject_IS_GC(op)) {
322 PyGC_Head *gc = AS_GC(op);
323 /* We're only interested in gc_refs for objects in the
324 * generation being collected, which can be recognized
325 * because only they have positive gc_refs.
326 */
327 assert(gc->gc.gc_refs != 0); /* else refcount was too small */
328 if (gc->gc.gc_refs > 0)
329 gc->gc.gc_refs--;
330 }
331 return 0;
Jeremy Hyltonc5007aa2000-06-30 05:02:53 +0000332}
333
Tim Peters19b74c72002-07-01 03:52:19 +0000334/* Subtract internal references from gc_refs. After this, gc_refs is >= 0
335 * for all objects in containers, and is GC_REACHABLE for all tracked gc
336 * objects not in containers. The ones with gc_refs > 0 are directly
337 * reachable from outside containers, and so can't be collected.
338 */
Jeremy Hyltonc5007aa2000-06-30 05:02:53 +0000339static void
340subtract_refs(PyGC_Head *containers)
341{
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000342 traverseproc traverse;
343 PyGC_Head *gc = containers->gc.gc_next;
344 for (; gc != containers; gc=gc->gc.gc_next) {
345 traverse = Py_TYPE(FROM_GC(gc))->tp_traverse;
346 (void) traverse(FROM_GC(gc),
347 (visitproc)visit_decref,
348 NULL);
349 }
Jeremy Hyltonc5007aa2000-06-30 05:02:53 +0000350}
351
Tim Peters19b74c72002-07-01 03:52:19 +0000352/* A traversal callback for move_unreachable. */
Jeremy Hyltonc5007aa2000-06-30 05:02:53 +0000353static int
Tim Peters19b74c72002-07-01 03:52:19 +0000354visit_reachable(PyObject *op, PyGC_Head *reachable)
Jeremy Hyltonc5007aa2000-06-30 05:02:53 +0000355{
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000356 if (PyObject_IS_GC(op)) {
357 PyGC_Head *gc = AS_GC(op);
358 const Py_ssize_t gc_refs = gc->gc.gc_refs;
Tim Peters19b74c72002-07-01 03:52:19 +0000359
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000360 if (gc_refs == 0) {
361 /* This is in move_unreachable's 'young' list, but
362 * the traversal hasn't yet gotten to it. All
363 * we need to do is tell move_unreachable that it's
364 * reachable.
365 */
366 gc->gc.gc_refs = 1;
367 }
368 else if (gc_refs == GC_TENTATIVELY_UNREACHABLE) {
369 /* This had gc_refs = 0 when move_unreachable got
370 * to it, but turns out it's reachable after all.
371 * Move it back to move_unreachable's 'young' list,
372 * and move_unreachable will eventually get to it
373 * again.
374 */
375 gc_list_move(gc, reachable);
376 gc->gc.gc_refs = 1;
377 }
378 /* Else there's nothing to do.
379 * If gc_refs > 0, it must be in move_unreachable's 'young'
380 * list, and move_unreachable will eventually get to it.
381 * If gc_refs == GC_REACHABLE, it's either in some other
382 * generation so we don't care about it, or move_unreachable
383 * already dealt with it.
384 * If gc_refs == GC_UNTRACKED, it must be ignored.
385 */
386 else {
387 assert(gc_refs > 0
388 || gc_refs == GC_REACHABLE
389 || gc_refs == GC_UNTRACKED);
390 }
391 }
392 return 0;
Jeremy Hyltonc5007aa2000-06-30 05:02:53 +0000393}
394
Tim Peters19b74c72002-07-01 03:52:19 +0000395/* Move the unreachable objects from young to unreachable. After this,
396 * all objects in young have gc_refs = GC_REACHABLE, and all objects in
397 * unreachable have gc_refs = GC_TENTATIVELY_UNREACHABLE. All tracked
398 * gc objects not in young or unreachable still have gc_refs = GC_REACHABLE.
399 * All objects in young after this are directly or indirectly reachable
400 * from outside the original young; and all objects in unreachable are
401 * not.
Tim Peters88396172002-06-30 17:56:40 +0000402 */
Jeremy Hyltonc5007aa2000-06-30 05:02:53 +0000403static void
Tim Peters19b74c72002-07-01 03:52:19 +0000404move_unreachable(PyGC_Head *young, PyGC_Head *unreachable)
Jeremy Hyltonc5007aa2000-06-30 05:02:53 +0000405{
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000406 PyGC_Head *gc = young->gc.gc_next;
Tim Peters19b74c72002-07-01 03:52:19 +0000407
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000408 /* Invariants: all objects "to the left" of us in young have gc_refs
409 * = GC_REACHABLE, and are indeed reachable (directly or indirectly)
410 * from outside the young list as it was at entry. All other objects
411 * from the original young "to the left" of us are in unreachable now,
412 * and have gc_refs = GC_TENTATIVELY_UNREACHABLE. All objects to the
413 * left of us in 'young' now have been scanned, and no objects here
414 * or to the right have been scanned yet.
415 */
Tim Peters19b74c72002-07-01 03:52:19 +0000416
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000417 while (gc != young) {
418 PyGC_Head *next;
Tim Peters19b74c72002-07-01 03:52:19 +0000419
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000420 if (gc->gc.gc_refs) {
421 /* gc is definitely reachable from outside the
422 * original 'young'. Mark it as such, and traverse
423 * its pointers to find any other objects that may
424 * be directly reachable from it. Note that the
425 * call to tp_traverse may append objects to young,
426 * so we have to wait until it returns to determine
427 * the next object to visit.
428 */
429 PyObject *op = FROM_GC(gc);
430 traverseproc traverse = Py_TYPE(op)->tp_traverse;
431 assert(gc->gc.gc_refs > 0);
432 gc->gc.gc_refs = GC_REACHABLE;
433 (void) traverse(op,
434 (visitproc)visit_reachable,
435 (void *)young);
436 next = gc->gc.gc_next;
437 if (PyTuple_CheckExact(op)) {
438 _PyTuple_MaybeUntrack(op);
439 }
440 else if (PyDict_CheckExact(op)) {
441 _PyDict_MaybeUntrack(op);
442 }
443 }
444 else {
445 /* This *may* be unreachable. To make progress,
446 * assume it is. gc isn't directly reachable from
447 * any object we've already traversed, but may be
448 * reachable from an object we haven't gotten to yet.
449 * visit_reachable will eventually move gc back into
450 * young if that's so, and we'll see it again.
451 */
452 next = gc->gc.gc_next;
453 gc_list_move(gc, unreachable);
454 gc->gc.gc_refs = GC_TENTATIVELY_UNREACHABLE;
455 }
456 gc = next;
457 }
Jeremy Hyltonc5007aa2000-06-30 05:02:53 +0000458}
459
Amaury Forgeot d'Arcad8dcd52007-12-10 23:58:35 +0000460/* Return true if object has a finalization method. */
Neil Schemenauera765c122001-11-01 17:35:23 +0000461static int
462has_finalizer(PyObject *op)
Jeremy Hyltonc5007aa2000-06-30 05:02:53 +0000463{
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000464 if (PyGen_CheckExact(op))
465 return PyGen_NeedsFinalizing((PyGenObject *)op);
466 else
467 return op->ob_type->tp_del != NULL;
Neil Schemenauera765c122001-11-01 17:35:23 +0000468}
469
Tim Petersead8b7a2004-10-30 23:09:22 +0000470/* Move the objects in unreachable with __del__ methods into `finalizers`.
471 * Objects moved into `finalizers` have gc_refs set to GC_REACHABLE; the
472 * objects remaining in unreachable are left at GC_TENTATIVELY_UNREACHABLE.
Jeremy Hyltonce136e92003-04-04 19:59:06 +0000473 */
Neil Schemenauera765c122001-11-01 17:35:23 +0000474static void
Tim Petersead8b7a2004-10-30 23:09:22 +0000475move_finalizers(PyGC_Head *unreachable, PyGC_Head *finalizers)
Neil Schemenauera765c122001-11-01 17:35:23 +0000476{
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000477 PyGC_Head *gc;
478 PyGC_Head *next;
Tim Petersf6b80452003-04-07 19:21:15 +0000479
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000480 /* March over unreachable. Move objects with finalizers into
481 * `finalizers`.
482 */
483 for (gc = unreachable->gc.gc_next; gc != unreachable; gc = next) {
484 PyObject *op = FROM_GC(gc);
Jeremy Hyltonce136e92003-04-04 19:59:06 +0000485
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000486 assert(IS_TENTATIVELY_UNREACHABLE(op));
487 next = gc->gc.gc_next;
Tim Petersf6ae7a42003-04-05 18:40:50 +0000488
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000489 if (has_finalizer(op)) {
490 gc_list_move(gc, finalizers);
491 gc->gc.gc_refs = GC_REACHABLE;
492 }
493 }
Jeremy Hyltonc5007aa2000-06-30 05:02:53 +0000494}
495
Tim Peters19b74c72002-07-01 03:52:19 +0000496/* A traversal callback for move_finalizer_reachable. */
497static int
498visit_move(PyObject *op, PyGC_Head *tolist)
499{
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000500 if (PyObject_IS_GC(op)) {
501 if (IS_TENTATIVELY_UNREACHABLE(op)) {
502 PyGC_Head *gc = AS_GC(op);
503 gc_list_move(gc, tolist);
504 gc->gc.gc_refs = GC_REACHABLE;
505 }
506 }
507 return 0;
Tim Peters19b74c72002-07-01 03:52:19 +0000508}
509
510/* Move objects that are reachable from finalizers, from the unreachable set
Tim Petersf6b80452003-04-07 19:21:15 +0000511 * into finalizers set.
Tim Peters19b74c72002-07-01 03:52:19 +0000512 */
Jeremy Hyltonc5007aa2000-06-30 05:02:53 +0000513static void
Tim Petersf6b80452003-04-07 19:21:15 +0000514move_finalizer_reachable(PyGC_Head *finalizers)
Jeremy Hyltonc5007aa2000-06-30 05:02:53 +0000515{
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000516 traverseproc traverse;
517 PyGC_Head *gc = finalizers->gc.gc_next;
518 for (; gc != finalizers; gc = gc->gc.gc_next) {
519 /* Note that the finalizers list may grow during this. */
520 traverse = Py_TYPE(FROM_GC(gc))->tp_traverse;
521 (void) traverse(FROM_GC(gc),
522 (visitproc)visit_move,
523 (void *)finalizers);
524 }
Jeremy Hyltonc5007aa2000-06-30 05:02:53 +0000525}
526
Tim Petersead8b7a2004-10-30 23:09:22 +0000527/* Clear all weakrefs to unreachable objects, and if such a weakref has a
528 * callback, invoke it if necessary. Note that it's possible for such
529 * weakrefs to be outside the unreachable set -- indeed, those are precisely
530 * the weakrefs whose callbacks must be invoked. See gc_weakref.txt for
531 * overview & some details. Some weakrefs with callbacks may be reclaimed
532 * directly by this routine; the number reclaimed is the return value. Other
533 * weakrefs with callbacks may be moved into the `old` generation. Objects
534 * moved into `old` have gc_refs set to GC_REACHABLE; the objects remaining in
535 * unreachable are left at GC_TENTATIVELY_UNREACHABLE. When this returns,
536 * no object in `unreachable` is weakly referenced anymore.
Tim Peters403a2032003-11-20 21:21:46 +0000537 */
538static int
Tim Petersead8b7a2004-10-30 23:09:22 +0000539handle_weakrefs(PyGC_Head *unreachable, PyGC_Head *old)
Tim Peters403a2032003-11-20 21:21:46 +0000540{
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000541 PyGC_Head *gc;
542 PyObject *op; /* generally FROM_GC(gc) */
543 PyWeakReference *wr; /* generally a cast of op */
544 PyGC_Head wrcb_to_call; /* weakrefs with callbacks to call */
545 PyGC_Head *next;
546 int num_freed = 0;
Tim Peters403a2032003-11-20 21:21:46 +0000547
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000548 gc_list_init(&wrcb_to_call);
Tim Peters403a2032003-11-20 21:21:46 +0000549
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000550 /* Clear all weakrefs to the objects in unreachable. If such a weakref
551 * also has a callback, move it into `wrcb_to_call` if the callback
552 * needs to be invoked. Note that we cannot invoke any callbacks until
553 * all weakrefs to unreachable objects are cleared, lest the callback
554 * resurrect an unreachable object via a still-active weakref. We
555 * make another pass over wrcb_to_call, invoking callbacks, after this
556 * pass completes.
557 */
558 for (gc = unreachable->gc.gc_next; gc != unreachable; gc = next) {
559 PyWeakReference **wrlist;
Tim Petersead8b7a2004-10-30 23:09:22 +0000560
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000561 op = FROM_GC(gc);
562 assert(IS_TENTATIVELY_UNREACHABLE(op));
563 next = gc->gc.gc_next;
Tim Petersead8b7a2004-10-30 23:09:22 +0000564
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000565 if (! PyType_SUPPORTS_WEAKREFS(Py_TYPE(op)))
566 continue;
Tim Petersead8b7a2004-10-30 23:09:22 +0000567
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000568 /* It supports weakrefs. Does it have any? */
569 wrlist = (PyWeakReference **)
570 PyObject_GET_WEAKREFS_LISTPTR(op);
Tim Petersead8b7a2004-10-30 23:09:22 +0000571
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000572 /* `op` may have some weakrefs. March over the list, clear
573 * all the weakrefs, and move the weakrefs with callbacks
574 * that must be called into wrcb_to_call.
575 */
576 for (wr = *wrlist; wr != NULL; wr = *wrlist) {
577 PyGC_Head *wrasgc; /* AS_GC(wr) */
Tim Petersead8b7a2004-10-30 23:09:22 +0000578
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000579 /* _PyWeakref_ClearRef clears the weakref but leaves
580 * the callback pointer intact. Obscure: it also
581 * changes *wrlist.
582 */
583 assert(wr->wr_object == op);
584 _PyWeakref_ClearRef(wr);
585 assert(wr->wr_object == Py_None);
586 if (wr->wr_callback == NULL)
587 continue; /* no callback */
Tim Petersead8b7a2004-10-30 23:09:22 +0000588
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000589 /* Headache time. `op` is going away, and is weakly referenced by
590 * `wr`, which has a callback. Should the callback be invoked? If wr
591 * is also trash, no:
592 *
593 * 1. There's no need to call it. The object and the weakref are
594 * both going away, so it's legitimate to pretend the weakref is
595 * going away first. The user has to ensure a weakref outlives its
596 * referent if they want a guarantee that the wr callback will get
597 * invoked.
598 *
599 * 2. It may be catastrophic to call it. If the callback is also in
600 * cyclic trash (CT), then although the CT is unreachable from
601 * outside the current generation, CT may be reachable from the
602 * callback. Then the callback could resurrect insane objects.
603 *
604 * Since the callback is never needed and may be unsafe in this case,
605 * wr is simply left in the unreachable set. Note that because we
606 * already called _PyWeakref_ClearRef(wr), its callback will never
607 * trigger.
608 *
609 * OTOH, if wr isn't part of CT, we should invoke the callback: the
610 * weakref outlived the trash. Note that since wr isn't CT in this
611 * case, its callback can't be CT either -- wr acted as an external
612 * root to this generation, and therefore its callback did too. So
613 * nothing in CT is reachable from the callback either, so it's hard
614 * to imagine how calling it later could create a problem for us. wr
615 * is moved to wrcb_to_call in this case.
616 */
617 if (IS_TENTATIVELY_UNREACHABLE(wr))
618 continue;
619 assert(IS_REACHABLE(wr));
Tim Peterscc2a8662004-10-31 22:12:43 +0000620
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000621 /* Create a new reference so that wr can't go away
622 * before we can process it again.
623 */
624 Py_INCREF(wr);
Tim Petersead8b7a2004-10-30 23:09:22 +0000625
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000626 /* Move wr to wrcb_to_call, for the next pass. */
627 wrasgc = AS_GC(wr);
628 assert(wrasgc != next); /* wrasgc is reachable, but
629 next isn't, so they can't
630 be the same */
631 gc_list_move(wrasgc, &wrcb_to_call);
632 }
633 }
Tim Petersead8b7a2004-10-30 23:09:22 +0000634
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000635 /* Invoke the callbacks we decided to honor. It's safe to invoke them
636 * because they can't reference unreachable objects.
637 */
638 while (! gc_list_is_empty(&wrcb_to_call)) {
639 PyObject *temp;
640 PyObject *callback;
Tim Petersead8b7a2004-10-30 23:09:22 +0000641
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000642 gc = wrcb_to_call.gc.gc_next;
643 op = FROM_GC(gc);
644 assert(IS_REACHABLE(op));
645 assert(PyWeakref_Check(op));
646 wr = (PyWeakReference *)op;
647 callback = wr->wr_callback;
648 assert(callback != NULL);
Tim Petersead8b7a2004-10-30 23:09:22 +0000649
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000650 /* copy-paste of weakrefobject.c's handle_callback() */
651 temp = PyObject_CallFunctionObjArgs(callback, wr, NULL);
652 if (temp == NULL)
653 PyErr_WriteUnraisable(callback);
654 else
655 Py_DECREF(temp);
Tim Petersead8b7a2004-10-30 23:09:22 +0000656
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000657 /* Give up the reference we created in the first pass. When
658 * op's refcount hits 0 (which it may or may not do right now),
659 * op's tp_dealloc will decref op->wr_callback too. Note
660 * that the refcount probably will hit 0 now, and because this
661 * weakref was reachable to begin with, gc didn't already
662 * add it to its count of freed objects. Example: a reachable
663 * weak value dict maps some key to this reachable weakref.
664 * The callback removes this key->weakref mapping from the
665 * dict, leaving no other references to the weakref (excepting
666 * ours).
667 */
668 Py_DECREF(op);
669 if (wrcb_to_call.gc.gc_next == gc) {
670 /* object is still alive -- move it */
671 gc_list_move(gc, old);
672 }
673 else
674 ++num_freed;
675 }
Tim Petersead8b7a2004-10-30 23:09:22 +0000676
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000677 return num_freed;
Tim Peters403a2032003-11-20 21:21:46 +0000678}
679
Jeremy Hyltonc5007aa2000-06-30 05:02:53 +0000680static void
Jeremy Hylton06257772000-08-31 15:10:24 +0000681debug_cycle(char *msg, PyObject *op)
Jeremy Hyltonc5007aa2000-06-30 05:02:53 +0000682{
Victor Stinner499dfcf2011-03-21 13:26:24 +0100683 PySys_FormatStderr("gc: %s <%s %p>\n",
684 msg, Py_TYPE(op)->tp_name, op);
Jeremy Hyltonc5007aa2000-06-30 05:02:53 +0000685}
686
Tim Petersbf384c22003-04-06 00:11:39 +0000687/* Handle uncollectable garbage (cycles with finalizers, and stuff reachable
688 * only from such cycles).
Tim Petersf6b80452003-04-07 19:21:15 +0000689 * If DEBUG_SAVEALL, all objects in finalizers are appended to the module
690 * garbage list (a Python list), else only the objects in finalizers with
691 * __del__ methods are appended to garbage. All objects in finalizers are
692 * merged into the old list regardless.
Tim Peters259272b2003-04-06 19:41:39 +0000693 * Returns 0 if all OK, <0 on error (out of memory to grow the garbage list).
694 * The finalizers list is made empty on a successful return.
Tim Petersbf384c22003-04-06 00:11:39 +0000695 */
Tim Peters259272b2003-04-06 19:41:39 +0000696static int
Tim Petersf6b80452003-04-07 19:21:15 +0000697handle_finalizers(PyGC_Head *finalizers, PyGC_Head *old)
Jeremy Hyltonc5007aa2000-06-30 05:02:53 +0000698{
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000699 PyGC_Head *gc = finalizers->gc.gc_next;
Tim Petersf6b80452003-04-07 19:21:15 +0000700
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000701 if (garbage == NULL) {
702 garbage = PyList_New(0);
703 if (garbage == NULL)
704 Py_FatalError("gc couldn't create gc.garbage list");
705 }
706 for (; gc != finalizers; gc = gc->gc.gc_next) {
707 PyObject *op = FROM_GC(gc);
Tim Petersf6b80452003-04-07 19:21:15 +0000708
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000709 if ((debug & DEBUG_SAVEALL) || has_finalizer(op)) {
710 if (PyList_Append(garbage, op) < 0)
711 return -1;
712 }
713 }
Tim Petersf6b80452003-04-07 19:21:15 +0000714
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000715 gc_list_merge(finalizers, old);
716 return 0;
Jeremy Hyltonc5007aa2000-06-30 05:02:53 +0000717}
718
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000719/* Break reference cycles by clearing the containers involved. This is
Jeremy Hyltonc5007aa2000-06-30 05:02:53 +0000720 * tricky business as the lists can be changing and we don't know which
Tim Peters19b74c72002-07-01 03:52:19 +0000721 * objects may be freed. It is possible I screwed something up here.
722 */
Jeremy Hyltonc5007aa2000-06-30 05:02:53 +0000723static void
Jeremy Hyltonce136e92003-04-04 19:59:06 +0000724delete_garbage(PyGC_Head *collectable, PyGC_Head *old)
Jeremy Hyltonc5007aa2000-06-30 05:02:53 +0000725{
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000726 inquiry clear;
Jeremy Hyltonc5007aa2000-06-30 05:02:53 +0000727
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000728 while (!gc_list_is_empty(collectable)) {
729 PyGC_Head *gc = collectable->gc.gc_next;
730 PyObject *op = FROM_GC(gc);
Tim Peters88396172002-06-30 17:56:40 +0000731
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000732 assert(IS_TENTATIVELY_UNREACHABLE(op));
733 if (debug & DEBUG_SAVEALL) {
734 PyList_Append(garbage, op);
735 }
736 else {
737 if ((clear = Py_TYPE(op)->tp_clear) != NULL) {
738 Py_INCREF(op);
739 clear(op);
740 Py_DECREF(op);
741 }
742 }
743 if (collectable->gc.gc_next == gc) {
744 /* object is still alive, move it, it may die later */
745 gc_list_move(gc, old);
746 gc->gc.gc_refs = GC_REACHABLE;
747 }
748 }
Jeremy Hyltonc5007aa2000-06-30 05:02:53 +0000749}
750
Christian Heimesa156e092008-02-16 07:38:31 +0000751/* Clear all free lists
752 * All free lists are cleared during the collection of the highest generation.
753 * Allocated items in the free list may keep a pymalloc arena occupied.
754 * Clearing the free lists may give back memory to the OS earlier.
755 */
756static void
757clear_freelists(void)
758{
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000759 (void)PyMethod_ClearFreeList();
760 (void)PyFrame_ClearFreeList();
761 (void)PyCFunction_ClearFreeList();
762 (void)PyTuple_ClearFreeList();
763 (void)PyUnicode_ClearFreeList();
764 (void)PyFloat_ClearFreeList();
Antoine Pitrou9a812cb2011-11-15 00:00:12 +0100765 (void)PyList_ClearFreeList();
766 (void)PyDict_ClearFreeList();
Antoine Pitrou093ce9c2011-12-16 11:24:27 +0100767 (void)PySet_ClearFreeList();
Christian Heimesa156e092008-02-16 07:38:31 +0000768}
769
Antoine Pitrou621601a2008-12-17 23:18:19 +0000770static double
771get_time(void)
772{
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000773 double result = 0;
774 if (tmod != NULL) {
Martin v. Löwisbd928fe2011-10-14 10:20:37 +0200775 _Py_IDENTIFIER(time);
Martin v. Löwisafe55bb2011-10-09 10:38:36 +0200776
777 PyObject *f = _PyObject_CallMethodId(tmod, &PyId_time, NULL);
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000778 if (f == NULL) {
779 PyErr_Clear();
780 }
781 else {
782 if (PyFloat_Check(f))
783 result = PyFloat_AsDouble(f);
784 Py_DECREF(f);
785 }
786 }
787 return result;
Antoine Pitrou621601a2008-12-17 23:18:19 +0000788}
789
Jeremy Hyltonc5007aa2000-06-30 05:02:53 +0000790/* This is the main function. Read this to understand how the
791 * collection process works. */
Neal Norwitz7b216c52006-03-04 20:01:53 +0000792static Py_ssize_t
Neil Schemenauer2880ae52002-05-04 05:35:20 +0000793collect(int generation)
Jeremy Hyltonc5007aa2000-06-30 05:02:53 +0000794{
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000795 int i;
796 Py_ssize_t m = 0; /* # objects collected */
797 Py_ssize_t n = 0; /* # unreachable objects that couldn't be collected */
798 PyGC_Head *young; /* the generation we are examining */
799 PyGC_Head *old; /* next older generation */
800 PyGC_Head unreachable; /* non-problematic unreachable trash */
801 PyGC_Head finalizers; /* objects with, & reachable from, __del__ */
802 PyGC_Head *gc;
803 double t1 = 0.0;
Jeremy Hyltonc5007aa2000-06-30 05:02:53 +0000804
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000805 if (delstr == NULL) {
806 delstr = PyUnicode_InternFromString("__del__");
807 if (delstr == NULL)
808 Py_FatalError("gc couldn't allocate \"__del__\"");
809 }
Tim Peters93ad66d2003-04-05 17:15:44 +0000810
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000811 if (debug & DEBUG_STATS) {
812 PySys_WriteStderr("gc: collecting generation %d...\n",
813 generation);
814 PySys_WriteStderr("gc: objects in each generation:");
815 for (i = 0; i < NUM_GENERATIONS; i++)
816 PySys_WriteStderr(" %" PY_FORMAT_SIZE_T "d",
817 gc_list_size(GEN_HEAD(i)));
818 t1 = get_time();
819 PySys_WriteStderr("\n");
820 }
Neil Schemenauer2880ae52002-05-04 05:35:20 +0000821
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000822 /* update collection and allocation counters */
823 if (generation+1 < NUM_GENERATIONS)
824 generations[generation+1].count += 1;
825 for (i = 0; i <= generation; i++)
826 generations[i].count = 0;
Neil Schemenauer2880ae52002-05-04 05:35:20 +0000827
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000828 /* merge younger generations with one we are currently collecting */
829 for (i = 0; i < generation; i++) {
830 gc_list_merge(GEN_HEAD(i), GEN_HEAD(generation));
831 }
Neil Schemenauer2880ae52002-05-04 05:35:20 +0000832
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000833 /* handy references */
834 young = GEN_HEAD(generation);
835 if (generation < NUM_GENERATIONS-1)
836 old = GEN_HEAD(generation+1);
837 else
838 old = young;
Jeremy Hyltonc5007aa2000-06-30 05:02:53 +0000839
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000840 /* Using ob_refcnt and gc_refs, calculate which objects in the
841 * container set are reachable from outside the set (i.e., have a
842 * refcount greater than 0 when all the references within the
843 * set are taken into account).
844 */
845 update_refs(young);
846 subtract_refs(young);
Jeremy Hyltonc5007aa2000-06-30 05:02:53 +0000847
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000848 /* Leave everything reachable from outside young in young, and move
849 * everything else (in young) to unreachable.
850 * NOTE: This used to move the reachable objects into a reachable
851 * set instead. But most things usually turn out to be reachable,
852 * so it's more efficient to move the unreachable things.
853 */
854 gc_list_init(&unreachable);
855 move_unreachable(young, &unreachable);
Jeremy Hyltonc5007aa2000-06-30 05:02:53 +0000856
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000857 /* Move reachable objects to next generation. */
858 if (young != old) {
859 if (generation == NUM_GENERATIONS - 2) {
860 long_lived_pending += gc_list_size(young);
861 }
862 gc_list_merge(young, old);
863 }
864 else {
865 long_lived_pending = 0;
866 long_lived_total = gc_list_size(young);
867 }
Jeremy Hyltonc5007aa2000-06-30 05:02:53 +0000868
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000869 /* All objects in unreachable are trash, but objects reachable from
870 * finalizers can't safely be deleted. Python programmers should take
871 * care not to create such things. For Python, finalizers means
872 * instance objects with __del__ methods. Weakrefs with callbacks
873 * can also call arbitrary Python code but they will be dealt with by
874 * handle_weakrefs().
875 */
876 gc_list_init(&finalizers);
877 move_finalizers(&unreachable, &finalizers);
878 /* finalizers contains the unreachable objects with a finalizer;
879 * unreachable objects reachable *from* those are also uncollectable,
880 * and we move those into the finalizers list too.
881 */
882 move_finalizer_reachable(&finalizers);
Jeremy Hyltonc5007aa2000-06-30 05:02:53 +0000883
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000884 /* Collect statistics on collectable objects found and print
885 * debugging information.
886 */
887 for (gc = unreachable.gc.gc_next; gc != &unreachable;
888 gc = gc->gc.gc_next) {
889 m++;
890 if (debug & DEBUG_COLLECTABLE) {
891 debug_cycle("collectable", FROM_GC(gc));
892 }
893 }
Tim Petersead8b7a2004-10-30 23:09:22 +0000894
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000895 /* Clear weakrefs and invoke callbacks as necessary. */
896 m += handle_weakrefs(&unreachable, old);
Tim Petersead8b7a2004-10-30 23:09:22 +0000897
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000898 /* Call tp_clear on objects in the unreachable set. This will cause
899 * the reference cycles to be broken. It may also cause some objects
900 * in finalizers to be freed.
901 */
902 delete_garbage(&unreachable, old);
Jeremy Hyltonc5007aa2000-06-30 05:02:53 +0000903
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000904 /* Collect statistics on uncollectable objects found and print
905 * debugging information. */
906 for (gc = finalizers.gc.gc_next;
907 gc != &finalizers;
908 gc = gc->gc.gc_next) {
909 n++;
910 if (debug & DEBUG_UNCOLLECTABLE)
911 debug_cycle("uncollectable", FROM_GC(gc));
912 }
913 if (debug & DEBUG_STATS) {
914 double t2 = get_time();
915 if (m == 0 && n == 0)
916 PySys_WriteStderr("gc: done");
917 else
918 PySys_WriteStderr(
919 "gc: done, "
920 "%" PY_FORMAT_SIZE_T "d unreachable, "
921 "%" PY_FORMAT_SIZE_T "d uncollectable",
922 n+m, n);
923 if (t1 && t2) {
924 PySys_WriteStderr(", %.4fs elapsed", t2-t1);
925 }
926 PySys_WriteStderr(".\n");
927 }
Jeremy Hyltonc5007aa2000-06-30 05:02:53 +0000928
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000929 /* Append instances in the uncollectable set to a Python
930 * reachable list of garbage. The programmer has to deal with
931 * this if they insist on creating this type of structure.
932 */
933 (void)handle_finalizers(&finalizers, old);
Jeremy Hyltonc5007aa2000-06-30 05:02:53 +0000934
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000935 /* Clear free list only during the collection of the highest
936 * generation */
937 if (generation == NUM_GENERATIONS-1) {
938 clear_freelists();
939 }
Christian Heimesa156e092008-02-16 07:38:31 +0000940
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000941 if (PyErr_Occurred()) {
942 if (gc_str == NULL)
943 gc_str = PyUnicode_FromString("garbage collection");
944 PyErr_WriteUnraisable(gc_str);
945 Py_FatalError("unexpected exception during garbage collection");
946 }
947 return n+m;
Jeremy Hyltonc5007aa2000-06-30 05:02:53 +0000948}
949
Neal Norwitz7b216c52006-03-04 20:01:53 +0000950static Py_ssize_t
Jeremy Hyltonc5007aa2000-06-30 05:02:53 +0000951collect_generations(void)
952{
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000953 int i;
954 Py_ssize_t n = 0;
Jeremy Hyltonc5007aa2000-06-30 05:02:53 +0000955
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000956 /* Find the oldest generation (highest numbered) where the count
957 * exceeds the threshold. Objects in the that generation and
958 * generations younger than it will be collected. */
959 for (i = NUM_GENERATIONS-1; i >= 0; i--) {
960 if (generations[i].count > generations[i].threshold) {
961 /* Avoid quadratic performance degradation in number
962 of tracked objects. See comments at the beginning
963 of this file, and issue #4074.
964 */
965 if (i == NUM_GENERATIONS - 1
966 && long_lived_pending < long_lived_total / 4)
967 continue;
968 n = collect(i);
969 break;
970 }
971 }
972 return n;
Jeremy Hyltonc5007aa2000-06-30 05:02:53 +0000973}
974
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +0000975PyDoc_STRVAR(gc_enable__doc__,
Vladimir Marangozovf9d20c32000-08-06 22:45:31 +0000976"enable() -> None\n"
977"\n"
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +0000978"Enable automatic garbage collection.\n");
Jeremy Hyltonc5007aa2000-06-30 05:02:53 +0000979
Vladimir Marangozovf9d20c32000-08-06 22:45:31 +0000980static PyObject *
Tim Peters50c61d52003-04-06 01:50:50 +0000981gc_enable(PyObject *self, PyObject *noargs)
Vladimir Marangozovf9d20c32000-08-06 22:45:31 +0000982{
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000983 enabled = 1;
984 Py_INCREF(Py_None);
985 return Py_None;
Vladimir Marangozovf9d20c32000-08-06 22:45:31 +0000986}
987
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +0000988PyDoc_STRVAR(gc_disable__doc__,
Vladimir Marangozovf9d20c32000-08-06 22:45:31 +0000989"disable() -> None\n"
990"\n"
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +0000991"Disable automatic garbage collection.\n");
Vladimir Marangozovf9d20c32000-08-06 22:45:31 +0000992
993static PyObject *
Tim Peters50c61d52003-04-06 01:50:50 +0000994gc_disable(PyObject *self, PyObject *noargs)
Vladimir Marangozovf9d20c32000-08-06 22:45:31 +0000995{
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000996 enabled = 0;
997 Py_INCREF(Py_None);
998 return Py_None;
Vladimir Marangozovf9d20c32000-08-06 22:45:31 +0000999}
1000
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00001001PyDoc_STRVAR(gc_isenabled__doc__,
Vladimir Marangozovf9d20c32000-08-06 22:45:31 +00001002"isenabled() -> status\n"
1003"\n"
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00001004"Returns true if automatic garbage collection is enabled.\n");
Vladimir Marangozovf9d20c32000-08-06 22:45:31 +00001005
1006static PyObject *
Tim Peters50c61d52003-04-06 01:50:50 +00001007gc_isenabled(PyObject *self, PyObject *noargs)
Vladimir Marangozovf9d20c32000-08-06 22:45:31 +00001008{
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001009 return PyBool_FromLong((long)enabled);
Vladimir Marangozovf9d20c32000-08-06 22:45:31 +00001010}
1011
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00001012PyDoc_STRVAR(gc_collect__doc__,
Barry Warsawd3c38ff2006-03-07 09:46:03 +00001013"collect([generation]) -> n\n"
Jeremy Hyltonc5007aa2000-06-30 05:02:53 +00001014"\n"
Barry Warsawd3c38ff2006-03-07 09:46:03 +00001015"With no arguments, run a full collection. The optional argument\n"
1016"may be an integer specifying which generation to collect. A ValueError\n"
1017"is raised if the generation number is invalid.\n\n"
1018"The number of unreachable objects is returned.\n");
Jeremy Hyltonc5007aa2000-06-30 05:02:53 +00001019
1020static PyObject *
Barry Warsawd3c38ff2006-03-07 09:46:03 +00001021gc_collect(PyObject *self, PyObject *args, PyObject *kws)
Jeremy Hyltonc5007aa2000-06-30 05:02:53 +00001022{
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001023 static char *keywords[] = {"generation", NULL};
1024 int genarg = NUM_GENERATIONS - 1;
1025 Py_ssize_t n;
Jeremy Hyltonc5007aa2000-06-30 05:02:53 +00001026
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001027 if (!PyArg_ParseTupleAndKeywords(args, kws, "|i", keywords, &genarg))
1028 return NULL;
Barry Warsawd3c38ff2006-03-07 09:46:03 +00001029
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001030 else if (genarg < 0 || genarg >= NUM_GENERATIONS) {
1031 PyErr_SetString(PyExc_ValueError, "invalid generation");
1032 return NULL;
1033 }
Barry Warsawd3c38ff2006-03-07 09:46:03 +00001034
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001035 if (collecting)
1036 n = 0; /* already collecting, don't do anything */
1037 else {
1038 collecting = 1;
1039 n = collect(genarg);
1040 collecting = 0;
1041 }
Jeremy Hyltonc5007aa2000-06-30 05:02:53 +00001042
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001043 return PyLong_FromSsize_t(n);
Jeremy Hyltonc5007aa2000-06-30 05:02:53 +00001044}
1045
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00001046PyDoc_STRVAR(gc_set_debug__doc__,
Jeremy Hyltonc5007aa2000-06-30 05:02:53 +00001047"set_debug(flags) -> None\n"
1048"\n"
1049"Set the garbage collection debugging flags. Debugging information is\n"
1050"written to sys.stderr.\n"
1051"\n"
1052"flags is an integer and can have the following bits turned on:\n"
1053"\n"
1054" DEBUG_STATS - Print statistics during collection.\n"
1055" DEBUG_COLLECTABLE - Print collectable objects found.\n"
1056" DEBUG_UNCOLLECTABLE - Print unreachable but uncollectable objects found.\n"
Neil Schemenauer544de1e2000-09-22 15:22:38 +00001057" DEBUG_SAVEALL - Save objects to gc.garbage rather than freeing them.\n"
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00001058" DEBUG_LEAK - Debug leaking programs (everything but STATS).\n");
Jeremy Hyltonc5007aa2000-06-30 05:02:53 +00001059
1060static PyObject *
Vladimir Marangozovf9d20c32000-08-06 22:45:31 +00001061gc_set_debug(PyObject *self, PyObject *args)
Jeremy Hyltonc5007aa2000-06-30 05:02:53 +00001062{
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001063 if (!PyArg_ParseTuple(args, "i:set_debug", &debug))
1064 return NULL;
Jeremy Hyltonc5007aa2000-06-30 05:02:53 +00001065
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001066 Py_INCREF(Py_None);
1067 return Py_None;
Jeremy Hyltonc5007aa2000-06-30 05:02:53 +00001068}
1069
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00001070PyDoc_STRVAR(gc_get_debug__doc__,
Jeremy Hyltonc5007aa2000-06-30 05:02:53 +00001071"get_debug() -> flags\n"
1072"\n"
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00001073"Get the garbage collection debugging flags.\n");
Jeremy Hyltonc5007aa2000-06-30 05:02:53 +00001074
1075static PyObject *
Tim Peters50c61d52003-04-06 01:50:50 +00001076gc_get_debug(PyObject *self, PyObject *noargs)
Jeremy Hyltonc5007aa2000-06-30 05:02:53 +00001077{
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001078 return Py_BuildValue("i", debug);
Jeremy Hyltonc5007aa2000-06-30 05:02:53 +00001079}
1080
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00001081PyDoc_STRVAR(gc_set_thresh__doc__,
Neal Norwitz2a47c0f2002-01-29 00:53:41 +00001082"set_threshold(threshold0, [threshold1, threshold2]) -> None\n"
Jeremy Hyltonc5007aa2000-06-30 05:02:53 +00001083"\n"
1084"Sets the collection thresholds. Setting threshold0 to zero disables\n"
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00001085"collection.\n");
Jeremy Hyltonc5007aa2000-06-30 05:02:53 +00001086
1087static PyObject *
Vladimir Marangozovf9d20c32000-08-06 22:45:31 +00001088gc_set_thresh(PyObject *self, PyObject *args)
Jeremy Hyltonc5007aa2000-06-30 05:02:53 +00001089{
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001090 int i;
1091 if (!PyArg_ParseTuple(args, "i|ii:set_threshold",
1092 &generations[0].threshold,
1093 &generations[1].threshold,
1094 &generations[2].threshold))
1095 return NULL;
1096 for (i = 2; i < NUM_GENERATIONS; i++) {
1097 /* generations higher than 2 get the same threshold */
1098 generations[i].threshold = generations[2].threshold;
1099 }
Jeremy Hyltonc5007aa2000-06-30 05:02:53 +00001100
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001101 Py_INCREF(Py_None);
1102 return Py_None;
Jeremy Hyltonc5007aa2000-06-30 05:02:53 +00001103}
1104
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00001105PyDoc_STRVAR(gc_get_thresh__doc__,
Jeremy Hyltonc5007aa2000-06-30 05:02:53 +00001106"get_threshold() -> (threshold0, threshold1, threshold2)\n"
1107"\n"
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00001108"Return the current collection thresholds\n");
Jeremy Hyltonc5007aa2000-06-30 05:02:53 +00001109
1110static PyObject *
Tim Peters50c61d52003-04-06 01:50:50 +00001111gc_get_thresh(PyObject *self, PyObject *noargs)
Jeremy Hyltonc5007aa2000-06-30 05:02:53 +00001112{
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001113 return Py_BuildValue("(iii)",
1114 generations[0].threshold,
1115 generations[1].threshold,
1116 generations[2].threshold);
Jeremy Hyltonc5007aa2000-06-30 05:02:53 +00001117}
1118
Barry Warsawd3c38ff2006-03-07 09:46:03 +00001119PyDoc_STRVAR(gc_get_count__doc__,
1120"get_count() -> (count0, count1, count2)\n"
1121"\n"
1122"Return the current collection counts\n");
1123
1124static PyObject *
1125gc_get_count(PyObject *self, PyObject *noargs)
1126{
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001127 return Py_BuildValue("(iii)",
1128 generations[0].count,
1129 generations[1].count,
1130 generations[2].count);
Barry Warsawd3c38ff2006-03-07 09:46:03 +00001131}
1132
Neil Schemenauer48c70342001-08-09 15:38:31 +00001133static int
Martin v. Löwis560da622001-11-24 09:24:51 +00001134referrersvisit(PyObject* obj, PyObject *objs)
Neil Schemenauer48c70342001-08-09 15:38:31 +00001135{
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001136 Py_ssize_t i;
1137 for (i = 0; i < PyTuple_GET_SIZE(objs); i++)
1138 if (PyTuple_GET_ITEM(objs, i) == obj)
1139 return 1;
1140 return 0;
Neil Schemenauer48c70342001-08-09 15:38:31 +00001141}
1142
Neil Schemenauer17e7be62001-08-10 14:46:47 +00001143static int
Martin v. Löwis560da622001-11-24 09:24:51 +00001144gc_referrers_for(PyObject *objs, PyGC_Head *list, PyObject *resultlist)
Neil Schemenauer48c70342001-08-09 15:38:31 +00001145{
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001146 PyGC_Head *gc;
1147 PyObject *obj;
1148 traverseproc traverse;
1149 for (gc = list->gc.gc_next; gc != list; gc = gc->gc.gc_next) {
1150 obj = FROM_GC(gc);
1151 traverse = Py_TYPE(obj)->tp_traverse;
1152 if (obj == objs || obj == resultlist)
1153 continue;
1154 if (traverse(obj, (visitproc)referrersvisit, objs)) {
1155 if (PyList_Append(resultlist, obj) < 0)
1156 return 0; /* error */
1157 }
1158 }
1159 return 1; /* no error */
Neil Schemenauer48c70342001-08-09 15:38:31 +00001160}
1161
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00001162PyDoc_STRVAR(gc_get_referrers__doc__,
Martin v. Löwis560da622001-11-24 09:24:51 +00001163"get_referrers(*objs) -> list\n\
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00001164Return the list of objects that directly refer to any of objs.");
Neil Schemenauer48c70342001-08-09 15:38:31 +00001165
Neil Schemenauer17e7be62001-08-10 14:46:47 +00001166static PyObject *
Martin v. Löwis560da622001-11-24 09:24:51 +00001167gc_get_referrers(PyObject *self, PyObject *args)
Neil Schemenauer48c70342001-08-09 15:38:31 +00001168{
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001169 int i;
1170 PyObject *result = PyList_New(0);
1171 if (!result) return NULL;
Thomas Wouters49fd7fa2006-04-21 10:40:58 +00001172
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001173 for (i = 0; i < NUM_GENERATIONS; i++) {
1174 if (!(gc_referrers_for(args, GEN_HEAD(i), result))) {
1175 Py_DECREF(result);
1176 return NULL;
1177 }
1178 }
1179 return result;
Neil Schemenauer48c70342001-08-09 15:38:31 +00001180}
Jeremy Hyltonc5007aa2000-06-30 05:02:53 +00001181
Tim Peters0f81ab62003-04-08 16:39:48 +00001182/* Append obj to list; return true if error (out of memory), false if OK. */
Jeremy Hylton5bd378b2003-04-03 16:28:38 +00001183static int
Tim Peters730f5532003-04-08 17:17:17 +00001184referentsvisit(PyObject *obj, PyObject *list)
Jeremy Hylton5bd378b2003-04-03 16:28:38 +00001185{
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001186 return PyList_Append(list, obj) < 0;
Jeremy Hylton5bd378b2003-04-03 16:28:38 +00001187}
1188
Tim Peters730f5532003-04-08 17:17:17 +00001189PyDoc_STRVAR(gc_get_referents__doc__,
1190"get_referents(*objs) -> list\n\
Jeremy Hylton059b0942003-04-03 16:29:13 +00001191Return the list of objects that are directly referred to by objs.");
Jeremy Hylton5bd378b2003-04-03 16:28:38 +00001192
1193static PyObject *
Tim Peters730f5532003-04-08 17:17:17 +00001194gc_get_referents(PyObject *self, PyObject *args)
Jeremy Hylton5bd378b2003-04-03 16:28:38 +00001195{
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001196 Py_ssize_t i;
1197 PyObject *result = PyList_New(0);
Tim Peters0f81ab62003-04-08 16:39:48 +00001198
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001199 if (result == NULL)
1200 return NULL;
Tim Peters0f81ab62003-04-08 16:39:48 +00001201
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001202 for (i = 0; i < PyTuple_GET_SIZE(args); i++) {
1203 traverseproc traverse;
1204 PyObject *obj = PyTuple_GET_ITEM(args, i);
Tim Peters0f81ab62003-04-08 16:39:48 +00001205
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001206 if (! PyObject_IS_GC(obj))
1207 continue;
1208 traverse = Py_TYPE(obj)->tp_traverse;
1209 if (! traverse)
1210 continue;
1211 if (traverse(obj, (visitproc)referentsvisit, result)) {
1212 Py_DECREF(result);
1213 return NULL;
1214 }
1215 }
1216 return result;
Jeremy Hylton5bd378b2003-04-03 16:28:38 +00001217}
1218
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00001219PyDoc_STRVAR(gc_get_objects__doc__,
Neil Schemenauerc7c8d8e2001-08-09 15:58:59 +00001220"get_objects() -> [...]\n"
1221"\n"
1222"Return a list of objects tracked by the collector (excluding the list\n"
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00001223"returned).\n");
Neil Schemenauerc7c8d8e2001-08-09 15:58:59 +00001224
Neil Schemenauerc7c8d8e2001-08-09 15:58:59 +00001225static PyObject *
Tim Peters50c61d52003-04-06 01:50:50 +00001226gc_get_objects(PyObject *self, PyObject *noargs)
Neil Schemenauerc7c8d8e2001-08-09 15:58:59 +00001227{
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001228 int i;
1229 PyObject* result;
Neil Schemenauerc7c8d8e2001-08-09 15:58:59 +00001230
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001231 result = PyList_New(0);
1232 if (result == NULL)
1233 return NULL;
1234 for (i = 0; i < NUM_GENERATIONS; i++) {
1235 if (append_objects(result, GEN_HEAD(i))) {
1236 Py_DECREF(result);
1237 return NULL;
1238 }
1239 }
1240 return result;
Neil Schemenauerc7c8d8e2001-08-09 15:58:59 +00001241}
1242
Antoine Pitrou3a652b12009-03-23 18:52:06 +00001243PyDoc_STRVAR(gc_is_tracked__doc__,
1244"is_tracked(obj) -> bool\n"
1245"\n"
1246"Returns true if the object is tracked by the garbage collector.\n"
1247"Simple atomic objects will return false.\n"
1248);
1249
1250static PyObject *
1251gc_is_tracked(PyObject *self, PyObject *obj)
1252{
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001253 PyObject *result;
1254
1255 if (PyObject_IS_GC(obj) && IS_TRACKED(obj))
1256 result = Py_True;
1257 else
1258 result = Py_False;
1259 Py_INCREF(result);
1260 return result;
Antoine Pitrou3a652b12009-03-23 18:52:06 +00001261}
1262
Neil Schemenauerc7c8d8e2001-08-09 15:58:59 +00001263
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00001264PyDoc_STRVAR(gc__doc__,
Jeremy Hyltonc5007aa2000-06-30 05:02:53 +00001265"This module provides access to the garbage collector for reference cycles.\n"
1266"\n"
Vladimir Marangozovf9d20c32000-08-06 22:45:31 +00001267"enable() -- Enable automatic garbage collection.\n"
1268"disable() -- Disable automatic garbage collection.\n"
1269"isenabled() -- Returns true if automatic collection is enabled.\n"
Jeremy Hyltonc5007aa2000-06-30 05:02:53 +00001270"collect() -- Do a full collection right now.\n"
Thomas Wouters89f507f2006-12-13 04:49:30 +00001271"get_count() -- Return the current collection counts.\n"
Jeremy Hyltonc5007aa2000-06-30 05:02:53 +00001272"set_debug() -- Set debugging flags.\n"
1273"get_debug() -- Get debugging flags.\n"
1274"set_threshold() -- Set the collection thresholds.\n"
1275"get_threshold() -- Return the current the collection thresholds.\n"
Neil Schemenauerc7c8d8e2001-08-09 15:58:59 +00001276"get_objects() -- Return a list of all objects tracked by the collector.\n"
Antoine Pitrou3a652b12009-03-23 18:52:06 +00001277"is_tracked() -- Returns true if a given object is tracked.\n"
Jeremy Hylton5bd378b2003-04-03 16:28:38 +00001278"get_referrers() -- Return the list of objects that refer to an object.\n"
Tim Peters730f5532003-04-08 17:17:17 +00001279"get_referents() -- Return the list of objects that an object refers to.\n");
Jeremy Hyltonc5007aa2000-06-30 05:02:53 +00001280
1281static PyMethodDef GcMethods[] = {
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001282 {"enable", gc_enable, METH_NOARGS, gc_enable__doc__},
1283 {"disable", gc_disable, METH_NOARGS, gc_disable__doc__},
1284 {"isenabled", gc_isenabled, METH_NOARGS, gc_isenabled__doc__},
1285 {"set_debug", gc_set_debug, METH_VARARGS, gc_set_debug__doc__},
1286 {"get_debug", gc_get_debug, METH_NOARGS, gc_get_debug__doc__},
1287 {"get_count", gc_get_count, METH_NOARGS, gc_get_count__doc__},
1288 {"set_threshold", gc_set_thresh, METH_VARARGS, gc_set_thresh__doc__},
1289 {"get_threshold", gc_get_thresh, METH_NOARGS, gc_get_thresh__doc__},
1290 {"collect", (PyCFunction)gc_collect,
1291 METH_VARARGS | METH_KEYWORDS, gc_collect__doc__},
1292 {"get_objects", gc_get_objects,METH_NOARGS, gc_get_objects__doc__},
1293 {"is_tracked", gc_is_tracked, METH_O, gc_is_tracked__doc__},
1294 {"get_referrers", gc_get_referrers, METH_VARARGS,
1295 gc_get_referrers__doc__},
1296 {"get_referents", gc_get_referents, METH_VARARGS,
1297 gc_get_referents__doc__},
1298 {NULL, NULL} /* Sentinel */
Jeremy Hyltonc5007aa2000-06-30 05:02:53 +00001299};
1300
Martin v. Löwis1a214512008-06-11 05:26:20 +00001301static struct PyModuleDef gcmodule = {
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001302 PyModuleDef_HEAD_INIT,
Antoine Pitrou696e0352010-08-08 22:18:46 +00001303 "gc", /* m_name */
1304 gc__doc__, /* m_doc */
1305 -1, /* m_size */
1306 GcMethods, /* m_methods */
1307 NULL, /* m_reload */
1308 NULL, /* m_traverse */
1309 NULL, /* m_clear */
1310 NULL /* m_free */
Martin v. Löwis1a214512008-06-11 05:26:20 +00001311};
1312
Jason Tishler6bc06ec2003-09-04 11:59:50 +00001313PyMODINIT_FUNC
Martin v. Löwis1a214512008-06-11 05:26:20 +00001314PyInit_gc(void)
Jeremy Hyltonc5007aa2000-06-30 05:02:53 +00001315{
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001316 PyObject *m;
Jeremy Hyltonc5007aa2000-06-30 05:02:53 +00001317
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001318 m = PyModule_Create(&gcmodule);
Martin v. Löwis1a214512008-06-11 05:26:20 +00001319
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001320 if (m == NULL)
1321 return NULL;
Tim Peters11558872003-04-06 23:30:52 +00001322
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001323 if (garbage == NULL) {
1324 garbage = PyList_New(0);
1325 if (garbage == NULL)
1326 return NULL;
1327 }
1328 Py_INCREF(garbage);
1329 if (PyModule_AddObject(m, "garbage", garbage) < 0)
1330 return NULL;
Thomas Wouters477c8d52006-05-27 19:21:47 +00001331
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001332 /* Importing can't be done in collect() because collect()
1333 * can be called via PyGC_Collect() in Py_Finalize().
1334 * This wouldn't be a problem, except that <initialized> is
1335 * reset to 0 before calling collect which trips up
1336 * the import and triggers an assertion.
1337 */
1338 if (tmod == NULL) {
1339 tmod = PyImport_ImportModuleNoBlock("time");
1340 if (tmod == NULL)
1341 PyErr_Clear();
1342 }
Thomas Wouters477c8d52006-05-27 19:21:47 +00001343
Martin v. Löwis1a214512008-06-11 05:26:20 +00001344#define ADD_INT(NAME) if (PyModule_AddIntConstant(m, #NAME, NAME) < 0) return NULL
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001345 ADD_INT(DEBUG_STATS);
1346 ADD_INT(DEBUG_COLLECTABLE);
1347 ADD_INT(DEBUG_UNCOLLECTABLE);
1348 ADD_INT(DEBUG_SAVEALL);
1349 ADD_INT(DEBUG_LEAK);
Tim Peters11558872003-04-06 23:30:52 +00001350#undef ADD_INT
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001351 return m;
Jeremy Hyltonc5007aa2000-06-30 05:02:53 +00001352}
1353
Guido van Rossume13ddc92003-04-17 17:29:22 +00001354/* API to invoke gc.collect() from C */
Neal Norwitz7b216c52006-03-04 20:01:53 +00001355Py_ssize_t
Guido van Rossume13ddc92003-04-17 17:29:22 +00001356PyGC_Collect(void)
1357{
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001358 Py_ssize_t n;
Guido van Rossume13ddc92003-04-17 17:29:22 +00001359
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001360 if (collecting)
1361 n = 0; /* already collecting, don't do anything */
1362 else {
1363 collecting = 1;
1364 n = collect(NUM_GENERATIONS - 1);
1365 collecting = 0;
1366 }
Guido van Rossume13ddc92003-04-17 17:29:22 +00001367
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001368 return n;
Guido van Rossume13ddc92003-04-17 17:29:22 +00001369}
1370
Antoine Pitrou696e0352010-08-08 22:18:46 +00001371void
1372_PyGC_Fini(void)
1373{
Antoine Pitrou2ed94eb2010-09-14 09:48:39 +00001374 if (!(debug & DEBUG_SAVEALL)
1375 && garbage != NULL && PyList_GET_SIZE(garbage) > 0) {
Georg Brandl08be72d2010-10-24 15:11:22 +00001376 char *message;
1377 if (debug & DEBUG_UNCOLLECTABLE)
Antoine Pitroub5d82042010-11-05 00:05:25 +00001378 message = "gc: %zd uncollectable objects at " \
Georg Brandl08be72d2010-10-24 15:11:22 +00001379 "shutdown";
1380 else
Antoine Pitroub5d82042010-11-05 00:05:25 +00001381 message = "gc: %zd uncollectable objects at " \
Georg Brandl08be72d2010-10-24 15:11:22 +00001382 "shutdown; use gc.set_debug(gc.DEBUG_UNCOLLECTABLE) to list them";
1383 if (PyErr_WarnFormat(PyExc_ResourceWarning, 0, message,
1384 PyList_GET_SIZE(garbage)) < 0)
1385 PyErr_WriteUnraisable(NULL);
Antoine Pitrou696e0352010-08-08 22:18:46 +00001386 if (debug & DEBUG_UNCOLLECTABLE) {
1387 PyObject *repr = NULL, *bytes = NULL;
1388 repr = PyObject_Repr(garbage);
1389 if (!repr || !(bytes = PyUnicode_EncodeFSDefault(repr)))
1390 PyErr_WriteUnraisable(garbage);
1391 else {
1392 PySys_WriteStderr(
1393 " %s\n",
1394 PyBytes_AS_STRING(bytes)
1395 );
1396 }
1397 Py_XDECREF(repr);
1398 Py_XDECREF(bytes);
1399 }
Antoine Pitrou696e0352010-08-08 22:18:46 +00001400 }
1401}
1402
Neil Schemenauer43411b52001-08-30 00:05:51 +00001403/* for debugging */
Guido van Rossume13ddc92003-04-17 17:29:22 +00001404void
1405_PyGC_Dump(PyGC_Head *g)
Neil Schemenauer43411b52001-08-30 00:05:51 +00001406{
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001407 _PyObject_Dump(FROM_GC(g));
Neil Schemenauer43411b52001-08-30 00:05:51 +00001408}
1409
Neil Schemenauer43411b52001-08-30 00:05:51 +00001410/* extension modules might be compiled with GC support so these
1411 functions must always be available */
1412
Neil Schemenauerfec4eb12002-04-12 02:41:03 +00001413#undef PyObject_GC_Track
1414#undef PyObject_GC_UnTrack
1415#undef PyObject_GC_Del
1416#undef _PyObject_GC_Malloc
1417
Neil Schemenauer43411b52001-08-30 00:05:51 +00001418void
Neil Schemenauerfec4eb12002-04-12 02:41:03 +00001419PyObject_GC_Track(void *op)
Neil Schemenauer43411b52001-08-30 00:05:51 +00001420{
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001421 _PyObject_GC_TRACK(op);
Neil Schemenauer43411b52001-08-30 00:05:51 +00001422}
1423
Neil Schemenauerfec4eb12002-04-12 02:41:03 +00001424/* for binary compatibility with 2.2 */
Neil Schemenauer43411b52001-08-30 00:05:51 +00001425void
Neil Schemenauerfec4eb12002-04-12 02:41:03 +00001426_PyObject_GC_Track(PyObject *op)
1427{
1428 PyObject_GC_Track(op);
1429}
1430
1431void
1432PyObject_GC_UnTrack(void *op)
Neil Schemenauer43411b52001-08-30 00:05:51 +00001433{
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001434 /* Obscure: the Py_TRASHCAN mechanism requires that we be able to
1435 * call PyObject_GC_UnTrack twice on an object.
1436 */
1437 if (IS_TRACKED(op))
1438 _PyObject_GC_UNTRACK(op);
Neil Schemenauer43411b52001-08-30 00:05:51 +00001439}
1440
Neil Schemenauerfec4eb12002-04-12 02:41:03 +00001441/* for binary compatibility with 2.2 */
1442void
1443_PyObject_GC_UnTrack(PyObject *op)
1444{
1445 PyObject_GC_UnTrack(op);
1446}
1447
Neil Schemenauer43411b52001-08-30 00:05:51 +00001448PyObject *
Neil Schemenauerfec4eb12002-04-12 02:41:03 +00001449_PyObject_GC_Malloc(size_t basicsize)
Neil Schemenauer43411b52001-08-30 00:05:51 +00001450{
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001451 PyObject *op;
1452 PyGC_Head *g;
1453 if (basicsize > PY_SSIZE_T_MAX - sizeof(PyGC_Head))
1454 return PyErr_NoMemory();
1455 g = (PyGC_Head *)PyObject_MALLOC(
1456 sizeof(PyGC_Head) + basicsize);
1457 if (g == NULL)
1458 return PyErr_NoMemory();
1459 g->gc.gc_refs = GC_UNTRACKED;
1460 generations[0].count++; /* number of allocated GC objects */
1461 if (generations[0].count > generations[0].threshold &&
1462 enabled &&
1463 generations[0].threshold &&
1464 !collecting &&
1465 !PyErr_Occurred()) {
1466 collecting = 1;
1467 collect_generations();
1468 collecting = 0;
1469 }
1470 op = FROM_GC(g);
1471 return op;
Neil Schemenauer43411b52001-08-30 00:05:51 +00001472}
1473
1474PyObject *
1475_PyObject_GC_New(PyTypeObject *tp)
1476{
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001477 PyObject *op = _PyObject_GC_Malloc(_PyObject_SIZE(tp));
1478 if (op != NULL)
1479 op = PyObject_INIT(op, tp);
1480 return op;
Neil Schemenauer43411b52001-08-30 00:05:51 +00001481}
1482
1483PyVarObject *
Martin v. Löwis18e16552006-02-15 17:27:45 +00001484_PyObject_GC_NewVar(PyTypeObject *tp, Py_ssize_t nitems)
Neil Schemenauer43411b52001-08-30 00:05:51 +00001485{
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001486 const size_t size = _PyObject_VAR_SIZE(tp, nitems);
1487 PyVarObject *op = (PyVarObject *) _PyObject_GC_Malloc(size);
1488 if (op != NULL)
1489 op = PyObject_INIT_VAR(op, tp, nitems);
1490 return op;
Neil Schemenauer43411b52001-08-30 00:05:51 +00001491}
1492
1493PyVarObject *
Martin v. Löwis41290682006-02-16 14:56:14 +00001494_PyObject_GC_Resize(PyVarObject *op, Py_ssize_t nitems)
Neil Schemenauer43411b52001-08-30 00:05:51 +00001495{
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001496 const size_t basicsize = _PyObject_VAR_SIZE(Py_TYPE(op), nitems);
1497 PyGC_Head *g = AS_GC(op);
1498 if (basicsize > PY_SSIZE_T_MAX - sizeof(PyGC_Head))
1499 return (PyVarObject *)PyErr_NoMemory();
1500 g = (PyGC_Head *)PyObject_REALLOC(g, sizeof(PyGC_Head) + basicsize);
1501 if (g == NULL)
1502 return (PyVarObject *)PyErr_NoMemory();
1503 op = (PyVarObject *) FROM_GC(g);
1504 Py_SIZE(op) = nitems;
1505 return op;
Neil Schemenauer43411b52001-08-30 00:05:51 +00001506}
1507
1508void
Neil Schemenauerfec4eb12002-04-12 02:41:03 +00001509PyObject_GC_Del(void *op)
Neil Schemenauer43411b52001-08-30 00:05:51 +00001510{
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001511 PyGC_Head *g = AS_GC(op);
1512 if (IS_TRACKED(op))
1513 gc_list_remove(g);
1514 if (generations[0].count > 0) {
1515 generations[0].count--;
1516 }
1517 PyObject_FREE(g);
Neil Schemenauer43411b52001-08-30 00:05:51 +00001518}