blob: 154f13623e02da9c7eb63f862f59ac904bfe90cc [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();
Christian Heimesa156e092008-02-16 07:38:31 +0000767}
768
Antoine Pitrou621601a2008-12-17 23:18:19 +0000769static double
770get_time(void)
771{
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000772 double result = 0;
773 if (tmod != NULL) {
Martin v. Löwisbd928fe2011-10-14 10:20:37 +0200774 _Py_IDENTIFIER(time);
Martin v. Löwisafe55bb2011-10-09 10:38:36 +0200775
776 PyObject *f = _PyObject_CallMethodId(tmod, &PyId_time, NULL);
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000777 if (f == NULL) {
778 PyErr_Clear();
779 }
780 else {
781 if (PyFloat_Check(f))
782 result = PyFloat_AsDouble(f);
783 Py_DECREF(f);
784 }
785 }
786 return result;
Antoine Pitrou621601a2008-12-17 23:18:19 +0000787}
788
Jeremy Hyltonc5007aa2000-06-30 05:02:53 +0000789/* This is the main function. Read this to understand how the
790 * collection process works. */
Neal Norwitz7b216c52006-03-04 20:01:53 +0000791static Py_ssize_t
Neil Schemenauer2880ae52002-05-04 05:35:20 +0000792collect(int generation)
Jeremy Hyltonc5007aa2000-06-30 05:02:53 +0000793{
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000794 int i;
795 Py_ssize_t m = 0; /* # objects collected */
796 Py_ssize_t n = 0; /* # unreachable objects that couldn't be collected */
797 PyGC_Head *young; /* the generation we are examining */
798 PyGC_Head *old; /* next older generation */
799 PyGC_Head unreachable; /* non-problematic unreachable trash */
800 PyGC_Head finalizers; /* objects with, & reachable from, __del__ */
801 PyGC_Head *gc;
802 double t1 = 0.0;
Jeremy Hyltonc5007aa2000-06-30 05:02:53 +0000803
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000804 if (delstr == NULL) {
805 delstr = PyUnicode_InternFromString("__del__");
806 if (delstr == NULL)
807 Py_FatalError("gc couldn't allocate \"__del__\"");
808 }
Tim Peters93ad66d2003-04-05 17:15:44 +0000809
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000810 if (debug & DEBUG_STATS) {
811 PySys_WriteStderr("gc: collecting generation %d...\n",
812 generation);
813 PySys_WriteStderr("gc: objects in each generation:");
814 for (i = 0; i < NUM_GENERATIONS; i++)
815 PySys_WriteStderr(" %" PY_FORMAT_SIZE_T "d",
816 gc_list_size(GEN_HEAD(i)));
817 t1 = get_time();
818 PySys_WriteStderr("\n");
819 }
Neil Schemenauer2880ae52002-05-04 05:35:20 +0000820
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000821 /* update collection and allocation counters */
822 if (generation+1 < NUM_GENERATIONS)
823 generations[generation+1].count += 1;
824 for (i = 0; i <= generation; i++)
825 generations[i].count = 0;
Neil Schemenauer2880ae52002-05-04 05:35:20 +0000826
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000827 /* merge younger generations with one we are currently collecting */
828 for (i = 0; i < generation; i++) {
829 gc_list_merge(GEN_HEAD(i), GEN_HEAD(generation));
830 }
Neil Schemenauer2880ae52002-05-04 05:35:20 +0000831
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000832 /* handy references */
833 young = GEN_HEAD(generation);
834 if (generation < NUM_GENERATIONS-1)
835 old = GEN_HEAD(generation+1);
836 else
837 old = young;
Jeremy Hyltonc5007aa2000-06-30 05:02:53 +0000838
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000839 /* Using ob_refcnt and gc_refs, calculate which objects in the
840 * container set are reachable from outside the set (i.e., have a
841 * refcount greater than 0 when all the references within the
842 * set are taken into account).
843 */
844 update_refs(young);
845 subtract_refs(young);
Jeremy Hyltonc5007aa2000-06-30 05:02:53 +0000846
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000847 /* Leave everything reachable from outside young in young, and move
848 * everything else (in young) to unreachable.
849 * NOTE: This used to move the reachable objects into a reachable
850 * set instead. But most things usually turn out to be reachable,
851 * so it's more efficient to move the unreachable things.
852 */
853 gc_list_init(&unreachable);
854 move_unreachable(young, &unreachable);
Jeremy Hyltonc5007aa2000-06-30 05:02:53 +0000855
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000856 /* Move reachable objects to next generation. */
857 if (young != old) {
858 if (generation == NUM_GENERATIONS - 2) {
859 long_lived_pending += gc_list_size(young);
860 }
861 gc_list_merge(young, old);
862 }
863 else {
864 long_lived_pending = 0;
865 long_lived_total = gc_list_size(young);
866 }
Jeremy Hyltonc5007aa2000-06-30 05:02:53 +0000867
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000868 /* All objects in unreachable are trash, but objects reachable from
869 * finalizers can't safely be deleted. Python programmers should take
870 * care not to create such things. For Python, finalizers means
871 * instance objects with __del__ methods. Weakrefs with callbacks
872 * can also call arbitrary Python code but they will be dealt with by
873 * handle_weakrefs().
874 */
875 gc_list_init(&finalizers);
876 move_finalizers(&unreachable, &finalizers);
877 /* finalizers contains the unreachable objects with a finalizer;
878 * unreachable objects reachable *from* those are also uncollectable,
879 * and we move those into the finalizers list too.
880 */
881 move_finalizer_reachable(&finalizers);
Jeremy Hyltonc5007aa2000-06-30 05:02:53 +0000882
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000883 /* Collect statistics on collectable objects found and print
884 * debugging information.
885 */
886 for (gc = unreachable.gc.gc_next; gc != &unreachable;
887 gc = gc->gc.gc_next) {
888 m++;
889 if (debug & DEBUG_COLLECTABLE) {
890 debug_cycle("collectable", FROM_GC(gc));
891 }
892 }
Tim Petersead8b7a2004-10-30 23:09:22 +0000893
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000894 /* Clear weakrefs and invoke callbacks as necessary. */
895 m += handle_weakrefs(&unreachable, old);
Tim Petersead8b7a2004-10-30 23:09:22 +0000896
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000897 /* Call tp_clear on objects in the unreachable set. This will cause
898 * the reference cycles to be broken. It may also cause some objects
899 * in finalizers to be freed.
900 */
901 delete_garbage(&unreachable, old);
Jeremy Hyltonc5007aa2000-06-30 05:02:53 +0000902
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000903 /* Collect statistics on uncollectable objects found and print
904 * debugging information. */
905 for (gc = finalizers.gc.gc_next;
906 gc != &finalizers;
907 gc = gc->gc.gc_next) {
908 n++;
909 if (debug & DEBUG_UNCOLLECTABLE)
910 debug_cycle("uncollectable", FROM_GC(gc));
911 }
912 if (debug & DEBUG_STATS) {
913 double t2 = get_time();
914 if (m == 0 && n == 0)
915 PySys_WriteStderr("gc: done");
916 else
917 PySys_WriteStderr(
918 "gc: done, "
919 "%" PY_FORMAT_SIZE_T "d unreachable, "
920 "%" PY_FORMAT_SIZE_T "d uncollectable",
921 n+m, n);
922 if (t1 && t2) {
923 PySys_WriteStderr(", %.4fs elapsed", t2-t1);
924 }
925 PySys_WriteStderr(".\n");
926 }
Jeremy Hyltonc5007aa2000-06-30 05:02:53 +0000927
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000928 /* Append instances in the uncollectable set to a Python
929 * reachable list of garbage. The programmer has to deal with
930 * this if they insist on creating this type of structure.
931 */
932 (void)handle_finalizers(&finalizers, old);
Jeremy Hyltonc5007aa2000-06-30 05:02:53 +0000933
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000934 /* Clear free list only during the collection of the highest
935 * generation */
936 if (generation == NUM_GENERATIONS-1) {
937 clear_freelists();
938 }
Christian Heimesa156e092008-02-16 07:38:31 +0000939
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000940 if (PyErr_Occurred()) {
941 if (gc_str == NULL)
942 gc_str = PyUnicode_FromString("garbage collection");
943 PyErr_WriteUnraisable(gc_str);
944 Py_FatalError("unexpected exception during garbage collection");
945 }
946 return n+m;
Jeremy Hyltonc5007aa2000-06-30 05:02:53 +0000947}
948
Neal Norwitz7b216c52006-03-04 20:01:53 +0000949static Py_ssize_t
Jeremy Hyltonc5007aa2000-06-30 05:02:53 +0000950collect_generations(void)
951{
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000952 int i;
953 Py_ssize_t n = 0;
Jeremy Hyltonc5007aa2000-06-30 05:02:53 +0000954
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000955 /* Find the oldest generation (highest numbered) where the count
956 * exceeds the threshold. Objects in the that generation and
957 * generations younger than it will be collected. */
958 for (i = NUM_GENERATIONS-1; i >= 0; i--) {
959 if (generations[i].count > generations[i].threshold) {
960 /* Avoid quadratic performance degradation in number
961 of tracked objects. See comments at the beginning
962 of this file, and issue #4074.
963 */
964 if (i == NUM_GENERATIONS - 1
965 && long_lived_pending < long_lived_total / 4)
966 continue;
967 n = collect(i);
968 break;
969 }
970 }
971 return n;
Jeremy Hyltonc5007aa2000-06-30 05:02:53 +0000972}
973
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +0000974PyDoc_STRVAR(gc_enable__doc__,
Vladimir Marangozovf9d20c32000-08-06 22:45:31 +0000975"enable() -> None\n"
976"\n"
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +0000977"Enable automatic garbage collection.\n");
Jeremy Hyltonc5007aa2000-06-30 05:02:53 +0000978
Vladimir Marangozovf9d20c32000-08-06 22:45:31 +0000979static PyObject *
Tim Peters50c61d52003-04-06 01:50:50 +0000980gc_enable(PyObject *self, PyObject *noargs)
Vladimir Marangozovf9d20c32000-08-06 22:45:31 +0000981{
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000982 enabled = 1;
983 Py_INCREF(Py_None);
984 return Py_None;
Vladimir Marangozovf9d20c32000-08-06 22:45:31 +0000985}
986
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +0000987PyDoc_STRVAR(gc_disable__doc__,
Vladimir Marangozovf9d20c32000-08-06 22:45:31 +0000988"disable() -> None\n"
989"\n"
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +0000990"Disable automatic garbage collection.\n");
Vladimir Marangozovf9d20c32000-08-06 22:45:31 +0000991
992static PyObject *
Tim Peters50c61d52003-04-06 01:50:50 +0000993gc_disable(PyObject *self, PyObject *noargs)
Vladimir Marangozovf9d20c32000-08-06 22:45:31 +0000994{
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000995 enabled = 0;
996 Py_INCREF(Py_None);
997 return Py_None;
Vladimir Marangozovf9d20c32000-08-06 22:45:31 +0000998}
999
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00001000PyDoc_STRVAR(gc_isenabled__doc__,
Vladimir Marangozovf9d20c32000-08-06 22:45:31 +00001001"isenabled() -> status\n"
1002"\n"
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00001003"Returns true if automatic garbage collection is enabled.\n");
Vladimir Marangozovf9d20c32000-08-06 22:45:31 +00001004
1005static PyObject *
Tim Peters50c61d52003-04-06 01:50:50 +00001006gc_isenabled(PyObject *self, PyObject *noargs)
Vladimir Marangozovf9d20c32000-08-06 22:45:31 +00001007{
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001008 return PyBool_FromLong((long)enabled);
Vladimir Marangozovf9d20c32000-08-06 22:45:31 +00001009}
1010
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00001011PyDoc_STRVAR(gc_collect__doc__,
Barry Warsawd3c38ff2006-03-07 09:46:03 +00001012"collect([generation]) -> n\n"
Jeremy Hyltonc5007aa2000-06-30 05:02:53 +00001013"\n"
Barry Warsawd3c38ff2006-03-07 09:46:03 +00001014"With no arguments, run a full collection. The optional argument\n"
1015"may be an integer specifying which generation to collect. A ValueError\n"
1016"is raised if the generation number is invalid.\n\n"
1017"The number of unreachable objects is returned.\n");
Jeremy Hyltonc5007aa2000-06-30 05:02:53 +00001018
1019static PyObject *
Barry Warsawd3c38ff2006-03-07 09:46:03 +00001020gc_collect(PyObject *self, PyObject *args, PyObject *kws)
Jeremy Hyltonc5007aa2000-06-30 05:02:53 +00001021{
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001022 static char *keywords[] = {"generation", NULL};
1023 int genarg = NUM_GENERATIONS - 1;
1024 Py_ssize_t n;
Jeremy Hyltonc5007aa2000-06-30 05:02:53 +00001025
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001026 if (!PyArg_ParseTupleAndKeywords(args, kws, "|i", keywords, &genarg))
1027 return NULL;
Barry Warsawd3c38ff2006-03-07 09:46:03 +00001028
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001029 else if (genarg < 0 || genarg >= NUM_GENERATIONS) {
1030 PyErr_SetString(PyExc_ValueError, "invalid generation");
1031 return NULL;
1032 }
Barry Warsawd3c38ff2006-03-07 09:46:03 +00001033
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001034 if (collecting)
1035 n = 0; /* already collecting, don't do anything */
1036 else {
1037 collecting = 1;
1038 n = collect(genarg);
1039 collecting = 0;
1040 }
Jeremy Hyltonc5007aa2000-06-30 05:02:53 +00001041
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001042 return PyLong_FromSsize_t(n);
Jeremy Hyltonc5007aa2000-06-30 05:02:53 +00001043}
1044
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00001045PyDoc_STRVAR(gc_set_debug__doc__,
Jeremy Hyltonc5007aa2000-06-30 05:02:53 +00001046"set_debug(flags) -> None\n"
1047"\n"
1048"Set the garbage collection debugging flags. Debugging information is\n"
1049"written to sys.stderr.\n"
1050"\n"
1051"flags is an integer and can have the following bits turned on:\n"
1052"\n"
1053" DEBUG_STATS - Print statistics during collection.\n"
1054" DEBUG_COLLECTABLE - Print collectable objects found.\n"
1055" DEBUG_UNCOLLECTABLE - Print unreachable but uncollectable objects found.\n"
Neil Schemenauer544de1e2000-09-22 15:22:38 +00001056" DEBUG_SAVEALL - Save objects to gc.garbage rather than freeing them.\n"
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00001057" DEBUG_LEAK - Debug leaking programs (everything but STATS).\n");
Jeremy Hyltonc5007aa2000-06-30 05:02:53 +00001058
1059static PyObject *
Vladimir Marangozovf9d20c32000-08-06 22:45:31 +00001060gc_set_debug(PyObject *self, PyObject *args)
Jeremy Hyltonc5007aa2000-06-30 05:02:53 +00001061{
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001062 if (!PyArg_ParseTuple(args, "i:set_debug", &debug))
1063 return NULL;
Jeremy Hyltonc5007aa2000-06-30 05:02:53 +00001064
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001065 Py_INCREF(Py_None);
1066 return Py_None;
Jeremy Hyltonc5007aa2000-06-30 05:02:53 +00001067}
1068
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00001069PyDoc_STRVAR(gc_get_debug__doc__,
Jeremy Hyltonc5007aa2000-06-30 05:02:53 +00001070"get_debug() -> flags\n"
1071"\n"
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00001072"Get the garbage collection debugging flags.\n");
Jeremy Hyltonc5007aa2000-06-30 05:02:53 +00001073
1074static PyObject *
Tim Peters50c61d52003-04-06 01:50:50 +00001075gc_get_debug(PyObject *self, PyObject *noargs)
Jeremy Hyltonc5007aa2000-06-30 05:02:53 +00001076{
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001077 return Py_BuildValue("i", debug);
Jeremy Hyltonc5007aa2000-06-30 05:02:53 +00001078}
1079
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00001080PyDoc_STRVAR(gc_set_thresh__doc__,
Neal Norwitz2a47c0f2002-01-29 00:53:41 +00001081"set_threshold(threshold0, [threshold1, threshold2]) -> None\n"
Jeremy Hyltonc5007aa2000-06-30 05:02:53 +00001082"\n"
1083"Sets the collection thresholds. Setting threshold0 to zero disables\n"
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00001084"collection.\n");
Jeremy Hyltonc5007aa2000-06-30 05:02:53 +00001085
1086static PyObject *
Vladimir Marangozovf9d20c32000-08-06 22:45:31 +00001087gc_set_thresh(PyObject *self, PyObject *args)
Jeremy Hyltonc5007aa2000-06-30 05:02:53 +00001088{
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001089 int i;
1090 if (!PyArg_ParseTuple(args, "i|ii:set_threshold",
1091 &generations[0].threshold,
1092 &generations[1].threshold,
1093 &generations[2].threshold))
1094 return NULL;
1095 for (i = 2; i < NUM_GENERATIONS; i++) {
1096 /* generations higher than 2 get the same threshold */
1097 generations[i].threshold = generations[2].threshold;
1098 }
Jeremy Hyltonc5007aa2000-06-30 05:02:53 +00001099
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001100 Py_INCREF(Py_None);
1101 return Py_None;
Jeremy Hyltonc5007aa2000-06-30 05:02:53 +00001102}
1103
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00001104PyDoc_STRVAR(gc_get_thresh__doc__,
Jeremy Hyltonc5007aa2000-06-30 05:02:53 +00001105"get_threshold() -> (threshold0, threshold1, threshold2)\n"
1106"\n"
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00001107"Return the current collection thresholds\n");
Jeremy Hyltonc5007aa2000-06-30 05:02:53 +00001108
1109static PyObject *
Tim Peters50c61d52003-04-06 01:50:50 +00001110gc_get_thresh(PyObject *self, PyObject *noargs)
Jeremy Hyltonc5007aa2000-06-30 05:02:53 +00001111{
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001112 return Py_BuildValue("(iii)",
1113 generations[0].threshold,
1114 generations[1].threshold,
1115 generations[2].threshold);
Jeremy Hyltonc5007aa2000-06-30 05:02:53 +00001116}
1117
Barry Warsawd3c38ff2006-03-07 09:46:03 +00001118PyDoc_STRVAR(gc_get_count__doc__,
1119"get_count() -> (count0, count1, count2)\n"
1120"\n"
1121"Return the current collection counts\n");
1122
1123static PyObject *
1124gc_get_count(PyObject *self, PyObject *noargs)
1125{
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001126 return Py_BuildValue("(iii)",
1127 generations[0].count,
1128 generations[1].count,
1129 generations[2].count);
Barry Warsawd3c38ff2006-03-07 09:46:03 +00001130}
1131
Neil Schemenauer48c70342001-08-09 15:38:31 +00001132static int
Martin v. Löwis560da622001-11-24 09:24:51 +00001133referrersvisit(PyObject* obj, PyObject *objs)
Neil Schemenauer48c70342001-08-09 15:38:31 +00001134{
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001135 Py_ssize_t i;
1136 for (i = 0; i < PyTuple_GET_SIZE(objs); i++)
1137 if (PyTuple_GET_ITEM(objs, i) == obj)
1138 return 1;
1139 return 0;
Neil Schemenauer48c70342001-08-09 15:38:31 +00001140}
1141
Neil Schemenauer17e7be62001-08-10 14:46:47 +00001142static int
Martin v. Löwis560da622001-11-24 09:24:51 +00001143gc_referrers_for(PyObject *objs, PyGC_Head *list, PyObject *resultlist)
Neil Schemenauer48c70342001-08-09 15:38:31 +00001144{
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001145 PyGC_Head *gc;
1146 PyObject *obj;
1147 traverseproc traverse;
1148 for (gc = list->gc.gc_next; gc != list; gc = gc->gc.gc_next) {
1149 obj = FROM_GC(gc);
1150 traverse = Py_TYPE(obj)->tp_traverse;
1151 if (obj == objs || obj == resultlist)
1152 continue;
1153 if (traverse(obj, (visitproc)referrersvisit, objs)) {
1154 if (PyList_Append(resultlist, obj) < 0)
1155 return 0; /* error */
1156 }
1157 }
1158 return 1; /* no error */
Neil Schemenauer48c70342001-08-09 15:38:31 +00001159}
1160
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00001161PyDoc_STRVAR(gc_get_referrers__doc__,
Martin v. Löwis560da622001-11-24 09:24:51 +00001162"get_referrers(*objs) -> list\n\
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00001163Return the list of objects that directly refer to any of objs.");
Neil Schemenauer48c70342001-08-09 15:38:31 +00001164
Neil Schemenauer17e7be62001-08-10 14:46:47 +00001165static PyObject *
Martin v. Löwis560da622001-11-24 09:24:51 +00001166gc_get_referrers(PyObject *self, PyObject *args)
Neil Schemenauer48c70342001-08-09 15:38:31 +00001167{
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001168 int i;
1169 PyObject *result = PyList_New(0);
1170 if (!result) return NULL;
Thomas Wouters49fd7fa2006-04-21 10:40:58 +00001171
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001172 for (i = 0; i < NUM_GENERATIONS; i++) {
1173 if (!(gc_referrers_for(args, GEN_HEAD(i), result))) {
1174 Py_DECREF(result);
1175 return NULL;
1176 }
1177 }
1178 return result;
Neil Schemenauer48c70342001-08-09 15:38:31 +00001179}
Jeremy Hyltonc5007aa2000-06-30 05:02:53 +00001180
Tim Peters0f81ab62003-04-08 16:39:48 +00001181/* Append obj to list; return true if error (out of memory), false if OK. */
Jeremy Hylton5bd378b2003-04-03 16:28:38 +00001182static int
Tim Peters730f5532003-04-08 17:17:17 +00001183referentsvisit(PyObject *obj, PyObject *list)
Jeremy Hylton5bd378b2003-04-03 16:28:38 +00001184{
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001185 return PyList_Append(list, obj) < 0;
Jeremy Hylton5bd378b2003-04-03 16:28:38 +00001186}
1187
Tim Peters730f5532003-04-08 17:17:17 +00001188PyDoc_STRVAR(gc_get_referents__doc__,
1189"get_referents(*objs) -> list\n\
Jeremy Hylton059b0942003-04-03 16:29:13 +00001190Return the list of objects that are directly referred to by objs.");
Jeremy Hylton5bd378b2003-04-03 16:28:38 +00001191
1192static PyObject *
Tim Peters730f5532003-04-08 17:17:17 +00001193gc_get_referents(PyObject *self, PyObject *args)
Jeremy Hylton5bd378b2003-04-03 16:28:38 +00001194{
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001195 Py_ssize_t i;
1196 PyObject *result = PyList_New(0);
Tim Peters0f81ab62003-04-08 16:39:48 +00001197
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001198 if (result == NULL)
1199 return NULL;
Tim Peters0f81ab62003-04-08 16:39:48 +00001200
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001201 for (i = 0; i < PyTuple_GET_SIZE(args); i++) {
1202 traverseproc traverse;
1203 PyObject *obj = PyTuple_GET_ITEM(args, i);
Tim Peters0f81ab62003-04-08 16:39:48 +00001204
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001205 if (! PyObject_IS_GC(obj))
1206 continue;
1207 traverse = Py_TYPE(obj)->tp_traverse;
1208 if (! traverse)
1209 continue;
1210 if (traverse(obj, (visitproc)referentsvisit, result)) {
1211 Py_DECREF(result);
1212 return NULL;
1213 }
1214 }
1215 return result;
Jeremy Hylton5bd378b2003-04-03 16:28:38 +00001216}
1217
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00001218PyDoc_STRVAR(gc_get_objects__doc__,
Neil Schemenauerc7c8d8e2001-08-09 15:58:59 +00001219"get_objects() -> [...]\n"
1220"\n"
1221"Return a list of objects tracked by the collector (excluding the list\n"
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00001222"returned).\n");
Neil Schemenauerc7c8d8e2001-08-09 15:58:59 +00001223
Neil Schemenauerc7c8d8e2001-08-09 15:58:59 +00001224static PyObject *
Tim Peters50c61d52003-04-06 01:50:50 +00001225gc_get_objects(PyObject *self, PyObject *noargs)
Neil Schemenauerc7c8d8e2001-08-09 15:58:59 +00001226{
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001227 int i;
1228 PyObject* result;
Neil Schemenauerc7c8d8e2001-08-09 15:58:59 +00001229
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001230 result = PyList_New(0);
1231 if (result == NULL)
1232 return NULL;
1233 for (i = 0; i < NUM_GENERATIONS; i++) {
1234 if (append_objects(result, GEN_HEAD(i))) {
1235 Py_DECREF(result);
1236 return NULL;
1237 }
1238 }
1239 return result;
Neil Schemenauerc7c8d8e2001-08-09 15:58:59 +00001240}
1241
Antoine Pitrou3a652b12009-03-23 18:52:06 +00001242PyDoc_STRVAR(gc_is_tracked__doc__,
1243"is_tracked(obj) -> bool\n"
1244"\n"
1245"Returns true if the object is tracked by the garbage collector.\n"
1246"Simple atomic objects will return false.\n"
1247);
1248
1249static PyObject *
1250gc_is_tracked(PyObject *self, PyObject *obj)
1251{
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001252 PyObject *result;
1253
1254 if (PyObject_IS_GC(obj) && IS_TRACKED(obj))
1255 result = Py_True;
1256 else
1257 result = Py_False;
1258 Py_INCREF(result);
1259 return result;
Antoine Pitrou3a652b12009-03-23 18:52:06 +00001260}
1261
Neil Schemenauerc7c8d8e2001-08-09 15:58:59 +00001262
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00001263PyDoc_STRVAR(gc__doc__,
Jeremy Hyltonc5007aa2000-06-30 05:02:53 +00001264"This module provides access to the garbage collector for reference cycles.\n"
1265"\n"
Vladimir Marangozovf9d20c32000-08-06 22:45:31 +00001266"enable() -- Enable automatic garbage collection.\n"
1267"disable() -- Disable automatic garbage collection.\n"
1268"isenabled() -- Returns true if automatic collection is enabled.\n"
Jeremy Hyltonc5007aa2000-06-30 05:02:53 +00001269"collect() -- Do a full collection right now.\n"
Thomas Wouters89f507f2006-12-13 04:49:30 +00001270"get_count() -- Return the current collection counts.\n"
Jeremy Hyltonc5007aa2000-06-30 05:02:53 +00001271"set_debug() -- Set debugging flags.\n"
1272"get_debug() -- Get debugging flags.\n"
1273"set_threshold() -- Set the collection thresholds.\n"
1274"get_threshold() -- Return the current the collection thresholds.\n"
Neil Schemenauerc7c8d8e2001-08-09 15:58:59 +00001275"get_objects() -- Return a list of all objects tracked by the collector.\n"
Antoine Pitrou3a652b12009-03-23 18:52:06 +00001276"is_tracked() -- Returns true if a given object is tracked.\n"
Jeremy Hylton5bd378b2003-04-03 16:28:38 +00001277"get_referrers() -- Return the list of objects that refer to an object.\n"
Tim Peters730f5532003-04-08 17:17:17 +00001278"get_referents() -- Return the list of objects that an object refers to.\n");
Jeremy Hyltonc5007aa2000-06-30 05:02:53 +00001279
1280static PyMethodDef GcMethods[] = {
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001281 {"enable", gc_enable, METH_NOARGS, gc_enable__doc__},
1282 {"disable", gc_disable, METH_NOARGS, gc_disable__doc__},
1283 {"isenabled", gc_isenabled, METH_NOARGS, gc_isenabled__doc__},
1284 {"set_debug", gc_set_debug, METH_VARARGS, gc_set_debug__doc__},
1285 {"get_debug", gc_get_debug, METH_NOARGS, gc_get_debug__doc__},
1286 {"get_count", gc_get_count, METH_NOARGS, gc_get_count__doc__},
1287 {"set_threshold", gc_set_thresh, METH_VARARGS, gc_set_thresh__doc__},
1288 {"get_threshold", gc_get_thresh, METH_NOARGS, gc_get_thresh__doc__},
1289 {"collect", (PyCFunction)gc_collect,
1290 METH_VARARGS | METH_KEYWORDS, gc_collect__doc__},
1291 {"get_objects", gc_get_objects,METH_NOARGS, gc_get_objects__doc__},
1292 {"is_tracked", gc_is_tracked, METH_O, gc_is_tracked__doc__},
1293 {"get_referrers", gc_get_referrers, METH_VARARGS,
1294 gc_get_referrers__doc__},
1295 {"get_referents", gc_get_referents, METH_VARARGS,
1296 gc_get_referents__doc__},
1297 {NULL, NULL} /* Sentinel */
Jeremy Hyltonc5007aa2000-06-30 05:02:53 +00001298};
1299
Martin v. Löwis1a214512008-06-11 05:26:20 +00001300static struct PyModuleDef gcmodule = {
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001301 PyModuleDef_HEAD_INIT,
Antoine Pitrou696e0352010-08-08 22:18:46 +00001302 "gc", /* m_name */
1303 gc__doc__, /* m_doc */
1304 -1, /* m_size */
1305 GcMethods, /* m_methods */
1306 NULL, /* m_reload */
1307 NULL, /* m_traverse */
1308 NULL, /* m_clear */
1309 NULL /* m_free */
Martin v. Löwis1a214512008-06-11 05:26:20 +00001310};
1311
Jason Tishler6bc06ec2003-09-04 11:59:50 +00001312PyMODINIT_FUNC
Martin v. Löwis1a214512008-06-11 05:26:20 +00001313PyInit_gc(void)
Jeremy Hyltonc5007aa2000-06-30 05:02:53 +00001314{
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001315 PyObject *m;
Jeremy Hyltonc5007aa2000-06-30 05:02:53 +00001316
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001317 m = PyModule_Create(&gcmodule);
Martin v. Löwis1a214512008-06-11 05:26:20 +00001318
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001319 if (m == NULL)
1320 return NULL;
Tim Peters11558872003-04-06 23:30:52 +00001321
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001322 if (garbage == NULL) {
1323 garbage = PyList_New(0);
1324 if (garbage == NULL)
1325 return NULL;
1326 }
1327 Py_INCREF(garbage);
1328 if (PyModule_AddObject(m, "garbage", garbage) < 0)
1329 return NULL;
Thomas Wouters477c8d52006-05-27 19:21:47 +00001330
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001331 /* Importing can't be done in collect() because collect()
1332 * can be called via PyGC_Collect() in Py_Finalize().
1333 * This wouldn't be a problem, except that <initialized> is
1334 * reset to 0 before calling collect which trips up
1335 * the import and triggers an assertion.
1336 */
1337 if (tmod == NULL) {
1338 tmod = PyImport_ImportModuleNoBlock("time");
1339 if (tmod == NULL)
1340 PyErr_Clear();
1341 }
Thomas Wouters477c8d52006-05-27 19:21:47 +00001342
Martin v. Löwis1a214512008-06-11 05:26:20 +00001343#define ADD_INT(NAME) if (PyModule_AddIntConstant(m, #NAME, NAME) < 0) return NULL
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001344 ADD_INT(DEBUG_STATS);
1345 ADD_INT(DEBUG_COLLECTABLE);
1346 ADD_INT(DEBUG_UNCOLLECTABLE);
1347 ADD_INT(DEBUG_SAVEALL);
1348 ADD_INT(DEBUG_LEAK);
Tim Peters11558872003-04-06 23:30:52 +00001349#undef ADD_INT
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001350 return m;
Jeremy Hyltonc5007aa2000-06-30 05:02:53 +00001351}
1352
Guido van Rossume13ddc92003-04-17 17:29:22 +00001353/* API to invoke gc.collect() from C */
Neal Norwitz7b216c52006-03-04 20:01:53 +00001354Py_ssize_t
Guido van Rossume13ddc92003-04-17 17:29:22 +00001355PyGC_Collect(void)
1356{
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001357 Py_ssize_t n;
Guido van Rossume13ddc92003-04-17 17:29:22 +00001358
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001359 if (collecting)
1360 n = 0; /* already collecting, don't do anything */
1361 else {
1362 collecting = 1;
1363 n = collect(NUM_GENERATIONS - 1);
1364 collecting = 0;
1365 }
Guido van Rossume13ddc92003-04-17 17:29:22 +00001366
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001367 return n;
Guido van Rossume13ddc92003-04-17 17:29:22 +00001368}
1369
Antoine Pitrou696e0352010-08-08 22:18:46 +00001370void
1371_PyGC_Fini(void)
1372{
Antoine Pitrou2ed94eb2010-09-14 09:48:39 +00001373 if (!(debug & DEBUG_SAVEALL)
1374 && garbage != NULL && PyList_GET_SIZE(garbage) > 0) {
Georg Brandl08be72d2010-10-24 15:11:22 +00001375 char *message;
1376 if (debug & DEBUG_UNCOLLECTABLE)
Antoine Pitroub5d82042010-11-05 00:05:25 +00001377 message = "gc: %zd uncollectable objects at " \
Georg Brandl08be72d2010-10-24 15:11:22 +00001378 "shutdown";
1379 else
Antoine Pitroub5d82042010-11-05 00:05:25 +00001380 message = "gc: %zd uncollectable objects at " \
Georg Brandl08be72d2010-10-24 15:11:22 +00001381 "shutdown; use gc.set_debug(gc.DEBUG_UNCOLLECTABLE) to list them";
1382 if (PyErr_WarnFormat(PyExc_ResourceWarning, 0, message,
1383 PyList_GET_SIZE(garbage)) < 0)
1384 PyErr_WriteUnraisable(NULL);
Antoine Pitrou696e0352010-08-08 22:18:46 +00001385 if (debug & DEBUG_UNCOLLECTABLE) {
1386 PyObject *repr = NULL, *bytes = NULL;
1387 repr = PyObject_Repr(garbage);
1388 if (!repr || !(bytes = PyUnicode_EncodeFSDefault(repr)))
1389 PyErr_WriteUnraisable(garbage);
1390 else {
1391 PySys_WriteStderr(
1392 " %s\n",
1393 PyBytes_AS_STRING(bytes)
1394 );
1395 }
1396 Py_XDECREF(repr);
1397 Py_XDECREF(bytes);
1398 }
Antoine Pitrou696e0352010-08-08 22:18:46 +00001399 }
1400}
1401
Neil Schemenauer43411b52001-08-30 00:05:51 +00001402/* for debugging */
Guido van Rossume13ddc92003-04-17 17:29:22 +00001403void
1404_PyGC_Dump(PyGC_Head *g)
Neil Schemenauer43411b52001-08-30 00:05:51 +00001405{
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001406 _PyObject_Dump(FROM_GC(g));
Neil Schemenauer43411b52001-08-30 00:05:51 +00001407}
1408
Neil Schemenauer43411b52001-08-30 00:05:51 +00001409/* extension modules might be compiled with GC support so these
1410 functions must always be available */
1411
Neil Schemenauerfec4eb12002-04-12 02:41:03 +00001412#undef PyObject_GC_Track
1413#undef PyObject_GC_UnTrack
1414#undef PyObject_GC_Del
1415#undef _PyObject_GC_Malloc
1416
Neil Schemenauer43411b52001-08-30 00:05:51 +00001417void
Neil Schemenauerfec4eb12002-04-12 02:41:03 +00001418PyObject_GC_Track(void *op)
Neil Schemenauer43411b52001-08-30 00:05:51 +00001419{
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001420 _PyObject_GC_TRACK(op);
Neil Schemenauer43411b52001-08-30 00:05:51 +00001421}
1422
Neil Schemenauerfec4eb12002-04-12 02:41:03 +00001423/* for binary compatibility with 2.2 */
Neil Schemenauer43411b52001-08-30 00:05:51 +00001424void
Neil Schemenauerfec4eb12002-04-12 02:41:03 +00001425_PyObject_GC_Track(PyObject *op)
1426{
1427 PyObject_GC_Track(op);
1428}
1429
1430void
1431PyObject_GC_UnTrack(void *op)
Neil Schemenauer43411b52001-08-30 00:05:51 +00001432{
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001433 /* Obscure: the Py_TRASHCAN mechanism requires that we be able to
1434 * call PyObject_GC_UnTrack twice on an object.
1435 */
1436 if (IS_TRACKED(op))
1437 _PyObject_GC_UNTRACK(op);
Neil Schemenauer43411b52001-08-30 00:05:51 +00001438}
1439
Neil Schemenauerfec4eb12002-04-12 02:41:03 +00001440/* for binary compatibility with 2.2 */
1441void
1442_PyObject_GC_UnTrack(PyObject *op)
1443{
1444 PyObject_GC_UnTrack(op);
1445}
1446
Neil Schemenauer43411b52001-08-30 00:05:51 +00001447PyObject *
Neil Schemenauerfec4eb12002-04-12 02:41:03 +00001448_PyObject_GC_Malloc(size_t basicsize)
Neil Schemenauer43411b52001-08-30 00:05:51 +00001449{
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001450 PyObject *op;
1451 PyGC_Head *g;
1452 if (basicsize > PY_SSIZE_T_MAX - sizeof(PyGC_Head))
1453 return PyErr_NoMemory();
1454 g = (PyGC_Head *)PyObject_MALLOC(
1455 sizeof(PyGC_Head) + basicsize);
1456 if (g == NULL)
1457 return PyErr_NoMemory();
1458 g->gc.gc_refs = GC_UNTRACKED;
1459 generations[0].count++; /* number of allocated GC objects */
1460 if (generations[0].count > generations[0].threshold &&
1461 enabled &&
1462 generations[0].threshold &&
1463 !collecting &&
1464 !PyErr_Occurred()) {
1465 collecting = 1;
1466 collect_generations();
1467 collecting = 0;
1468 }
1469 op = FROM_GC(g);
1470 return op;
Neil Schemenauer43411b52001-08-30 00:05:51 +00001471}
1472
1473PyObject *
1474_PyObject_GC_New(PyTypeObject *tp)
1475{
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001476 PyObject *op = _PyObject_GC_Malloc(_PyObject_SIZE(tp));
1477 if (op != NULL)
1478 op = PyObject_INIT(op, tp);
1479 return op;
Neil Schemenauer43411b52001-08-30 00:05:51 +00001480}
1481
1482PyVarObject *
Martin v. Löwis18e16552006-02-15 17:27:45 +00001483_PyObject_GC_NewVar(PyTypeObject *tp, Py_ssize_t nitems)
Neil Schemenauer43411b52001-08-30 00:05:51 +00001484{
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001485 const size_t size = _PyObject_VAR_SIZE(tp, nitems);
1486 PyVarObject *op = (PyVarObject *) _PyObject_GC_Malloc(size);
1487 if (op != NULL)
1488 op = PyObject_INIT_VAR(op, tp, nitems);
1489 return op;
Neil Schemenauer43411b52001-08-30 00:05:51 +00001490}
1491
1492PyVarObject *
Martin v. Löwis41290682006-02-16 14:56:14 +00001493_PyObject_GC_Resize(PyVarObject *op, Py_ssize_t nitems)
Neil Schemenauer43411b52001-08-30 00:05:51 +00001494{
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001495 const size_t basicsize = _PyObject_VAR_SIZE(Py_TYPE(op), nitems);
1496 PyGC_Head *g = AS_GC(op);
1497 if (basicsize > PY_SSIZE_T_MAX - sizeof(PyGC_Head))
1498 return (PyVarObject *)PyErr_NoMemory();
1499 g = (PyGC_Head *)PyObject_REALLOC(g, sizeof(PyGC_Head) + basicsize);
1500 if (g == NULL)
1501 return (PyVarObject *)PyErr_NoMemory();
1502 op = (PyVarObject *) FROM_GC(g);
1503 Py_SIZE(op) = nitems;
1504 return op;
Neil Schemenauer43411b52001-08-30 00:05:51 +00001505}
1506
1507void
Neil Schemenauerfec4eb12002-04-12 02:41:03 +00001508PyObject_GC_Del(void *op)
Neil Schemenauer43411b52001-08-30 00:05:51 +00001509{
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001510 PyGC_Head *g = AS_GC(op);
1511 if (IS_TRACKED(op))
1512 gc_list_remove(g);
1513 if (generations[0].count > 0) {
1514 generations[0].count--;
1515 }
1516 PyObject_FREE(g);
Neil Schemenauer43411b52001-08-30 00:05:51 +00001517}