blob: 9bb3666e1ad0c22cf256ba9facc17d941f9b4d41 [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
Kristján Valur Jónsson69c63522012-04-15 11:41:32 +000068/* a list of callbacks to be invoked when collection is performed */
69static PyObject *callbacks = NULL;
70
71/* This is the number of objects that survived the last full collection. It
Antoine Pitrou14b78f52009-01-09 22:27:08 +000072 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
Kristján Valur Jónsson69c63522012-04-15 11:41:32 +000078/* This is the number of objects that survived all "non-full" collections,
Antoine Pitrou14b78f52009-01-09 22:27:08 +000079 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
Antoine Pitroue1ad3da2012-05-28 22:22:34 +0200119/*
120 NOTE: about untracking of mutable objects.
Victor Stinnerc1eb26c2013-07-08 22:15:05 +0200121
Antoine Pitroue1ad3da2012-05-28 22:22:34 +0200122 Certain types of container cannot participate in a reference cycle, and
123 so do not need to be tracked by the garbage collector. Untracking these
124 objects reduces the cost of garbage collections. However, determining
125 which objects may be untracked is not free, and the costs must be
126 weighed against the benefits for garbage collection.
127
128 There are two possible strategies for when to untrack a container:
129
130 i) When the container is created.
131 ii) When the container is examined by the garbage collector.
132
133 Tuples containing only immutable objects (integers, strings etc, and
134 recursively, tuples of immutable objects) do not need to be tracked.
135 The interpreter creates a large number of tuples, many of which will
136 not survive until garbage collection. It is therefore not worthwhile
137 to untrack eligible tuples at creation time.
138
Victor Stinnerc1eb26c2013-07-08 22:15:05 +0200139 Instead, all tuples except the empty tuple are tracked when created.
140 During garbage collection it is determined whether any surviving tuples
141 can be untracked. A tuple can be untracked if all of its contents are
142 already not tracked. Tuples are examined for untracking in all garbage
Antoine Pitroue1ad3da2012-05-28 22:22:34 +0200143 collection cycles. It may take more than one cycle to untrack a tuple.
144
145 Dictionaries containing only immutable objects also do not need to be
146 tracked. Dictionaries are untracked when created. If a tracked item is
147 inserted into a dictionary (either as a key or value), the dictionary
148 becomes tracked. During a full garbage collection (all generations),
149 the collector will untrack any dictionaries whose contents are not
150 tracked.
151
152 The module provides the python function is_tracked(obj), which returns
153 the CURRENT tracking status of the object. Subsequent garbage
154 collections may change the tracking status of the object.
Victor Stinnerc1eb26c2013-07-08 22:15:05 +0200155
156 Untracking of certain containers was introduced in issue #4688, and
Antoine Pitroue1ad3da2012-05-28 22:22:34 +0200157 the algorithm was refined in response to issue #14775.
158*/
Antoine Pitrou14b78f52009-01-09 22:27:08 +0000159
Jeremy Hyltonc5007aa2000-06-30 05:02:53 +0000160/* set for debugging information */
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000161#define DEBUG_STATS (1<<0) /* print collection statistics */
162#define DEBUG_COLLECTABLE (1<<1) /* print collectable objects */
163#define DEBUG_UNCOLLECTABLE (1<<2) /* print uncollectable objects */
164#define DEBUG_SAVEALL (1<<5) /* save all garbage in gc.garbage */
165#define DEBUG_LEAK DEBUG_COLLECTABLE | \
166 DEBUG_UNCOLLECTABLE | \
167 DEBUG_SAVEALL
Jeremy Hyltonb709df32000-09-01 02:47:25 +0000168static int debug;
Thomas Wouters477c8d52006-05-27 19:21:47 +0000169static PyObject *tmod = NULL;
Jeremy Hyltonc5007aa2000-06-30 05:02:53 +0000170
Antoine Pitroud4156c12012-10-30 22:43:19 +0100171/* Running stats per generation */
172struct gc_generation_stats {
173 /* total number of collections */
174 Py_ssize_t collections;
175 /* total number of collected objects */
176 Py_ssize_t collected;
177 /* total number of uncollectable objects (put into gc.garbage) */
178 Py_ssize_t uncollectable;
179};
180
181static struct gc_generation_stats generation_stats[NUM_GENERATIONS];
182
Tim Peters6fc13d92002-07-02 18:12:35 +0000183/*--------------------------------------------------------------------------
184gc_refs values.
Neil Schemenauer43411b52001-08-30 00:05:51 +0000185
Tim Peters6fc13d92002-07-02 18:12:35 +0000186Between collections, every gc'ed object has one of two gc_refs values:
187
188GC_UNTRACKED
189 The initial state; objects returned by PyObject_GC_Malloc are in this
190 state. The object doesn't live in any generation list, and its
191 tp_traverse slot must not be called.
192
193GC_REACHABLE
194 The object lives in some generation list, and its tp_traverse is safe to
195 call. An object transitions to GC_REACHABLE when PyObject_GC_Track
196 is called.
197
198During a collection, gc_refs can temporarily take on other states:
199
200>= 0
201 At the start of a collection, update_refs() copies the true refcount
202 to gc_refs, for each object in the generation being collected.
203 subtract_refs() then adjusts gc_refs so that it equals the number of
204 times an object is referenced directly from outside the generation
205 being collected.
Martin v. Löwis774348c2002-11-09 19:54:06 +0000206 gc_refs remains >= 0 throughout these steps.
Tim Peters6fc13d92002-07-02 18:12:35 +0000207
208GC_TENTATIVELY_UNREACHABLE
209 move_unreachable() then moves objects not reachable (whether directly or
210 indirectly) from outside the generation into an "unreachable" set.
211 Objects that are found to be reachable have gc_refs set to GC_REACHABLE
212 again. Objects that are found to be unreachable have gc_refs set to
213 GC_TENTATIVELY_UNREACHABLE. It's "tentatively" because the pass doing
214 this can't be sure until it ends, and GC_TENTATIVELY_UNREACHABLE may
215 transition back to GC_REACHABLE.
216
217 Only objects with GC_TENTATIVELY_UNREACHABLE still set are candidates
218 for collection. If it's decided not to collect such an object (e.g.,
219 it has a __del__ method), its gc_refs is restored to GC_REACHABLE again.
220----------------------------------------------------------------------------
221*/
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000222#define GC_UNTRACKED _PyGC_REFS_UNTRACKED
223#define GC_REACHABLE _PyGC_REFS_REACHABLE
224#define GC_TENTATIVELY_UNREACHABLE _PyGC_REFS_TENTATIVELY_UNREACHABLE
Tim Peters19b74c72002-07-01 03:52:19 +0000225
Antoine Pitrou796564c2013-07-30 19:59:21 +0200226#define IS_TRACKED(o) (_PyGC_REFS(o) != GC_UNTRACKED)
227#define IS_REACHABLE(o) (_PyGC_REFS(o) == GC_REACHABLE)
Tim Peters19b74c72002-07-01 03:52:19 +0000228#define IS_TENTATIVELY_UNREACHABLE(o) ( \
Antoine Pitrou796564c2013-07-30 19:59:21 +0200229 _PyGC_REFS(o) == GC_TENTATIVELY_UNREACHABLE)
Neil Schemenauera2b11ec2002-05-21 15:53:24 +0000230
Jeremy Hyltonc5007aa2000-06-30 05:02:53 +0000231/*** list functions ***/
232
233static void
234gc_list_init(PyGC_Head *list)
235{
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000236 list->gc.gc_prev = list;
237 list->gc.gc_next = list;
Jeremy Hyltonc5007aa2000-06-30 05:02:53 +0000238}
239
Neil Schemenauer2880ae52002-05-04 05:35:20 +0000240static int
241gc_list_is_empty(PyGC_Head *list)
242{
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000243 return (list->gc.gc_next == list);
Neil Schemenauer2880ae52002-05-04 05:35:20 +0000244}
245
Tim Peterse2d59182004-11-01 01:39:08 +0000246#if 0
247/* This became unused after gc_list_move() was introduced. */
248/* Append `node` to `list`. */
Jeremy Hyltonc5007aa2000-06-30 05:02:53 +0000249static void
250gc_list_append(PyGC_Head *node, PyGC_Head *list)
251{
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000252 node->gc.gc_next = list;
253 node->gc.gc_prev = list->gc.gc_prev;
254 node->gc.gc_prev->gc.gc_next = node;
255 list->gc.gc_prev = node;
Jeremy Hyltonc5007aa2000-06-30 05:02:53 +0000256}
Tim Peterse2d59182004-11-01 01:39:08 +0000257#endif
Jeremy Hyltonc5007aa2000-06-30 05:02:53 +0000258
Tim Peterse2d59182004-11-01 01:39:08 +0000259/* Remove `node` from the gc list it's currently in. */
Jeremy Hyltonc5007aa2000-06-30 05:02:53 +0000260static void
261gc_list_remove(PyGC_Head *node)
262{
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000263 node->gc.gc_prev->gc.gc_next = node->gc.gc_next;
264 node->gc.gc_next->gc.gc_prev = node->gc.gc_prev;
265 node->gc.gc_next = NULL; /* object is not currently tracked */
Jeremy Hyltonc5007aa2000-06-30 05:02:53 +0000266}
267
Tim Peterse2d59182004-11-01 01:39:08 +0000268/* Move `node` from the gc list it's currently in (which is not explicitly
269 * named here) to the end of `list`. This is semantically the same as
270 * gc_list_remove(node) followed by gc_list_append(node, list).
271 */
272static void
273gc_list_move(PyGC_Head *node, PyGC_Head *list)
274{
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000275 PyGC_Head *new_prev;
276 PyGC_Head *current_prev = node->gc.gc_prev;
277 PyGC_Head *current_next = node->gc.gc_next;
278 /* Unlink from current list. */
279 current_prev->gc.gc_next = current_next;
280 current_next->gc.gc_prev = current_prev;
281 /* Relink at end of new list. */
282 new_prev = node->gc.gc_prev = list->gc.gc_prev;
283 new_prev->gc.gc_next = list->gc.gc_prev = node;
284 node->gc.gc_next = list;
Tim Peterse2d59182004-11-01 01:39:08 +0000285}
286
287/* append list `from` onto list `to`; `from` becomes an empty list */
Jeremy Hyltonc5007aa2000-06-30 05:02:53 +0000288static void
289gc_list_merge(PyGC_Head *from, PyGC_Head *to)
290{
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000291 PyGC_Head *tail;
292 assert(from != to);
293 if (!gc_list_is_empty(from)) {
294 tail = to->gc.gc_prev;
295 tail->gc.gc_next = from->gc.gc_next;
296 tail->gc.gc_next->gc.gc_prev = tail;
297 to->gc.gc_prev = from->gc.gc_prev;
298 to->gc.gc_prev->gc.gc_next = to;
299 }
300 gc_list_init(from);
Jeremy Hyltonc5007aa2000-06-30 05:02:53 +0000301}
302
Neal Norwitz7b216c52006-03-04 20:01:53 +0000303static Py_ssize_t
Jeremy Hyltonc5007aa2000-06-30 05:02:53 +0000304gc_list_size(PyGC_Head *list)
305{
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000306 PyGC_Head *gc;
307 Py_ssize_t n = 0;
308 for (gc = list->gc.gc_next; gc != list; gc = gc->gc.gc_next) {
309 n++;
310 }
311 return n;
Jeremy Hyltonc5007aa2000-06-30 05:02:53 +0000312}
313
Tim Peters259272b2003-04-06 19:41:39 +0000314/* Append objects in a GC list to a Python list.
315 * Return 0 if all OK, < 0 if error (out of memory for list).
316 */
317static int
318append_objects(PyObject *py_list, PyGC_Head *gc_list)
319{
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000320 PyGC_Head *gc;
321 for (gc = gc_list->gc.gc_next; gc != gc_list; gc = gc->gc.gc_next) {
322 PyObject *op = FROM_GC(gc);
323 if (op != py_list) {
324 if (PyList_Append(py_list, op)) {
325 return -1; /* exception */
326 }
327 }
328 }
329 return 0;
Tim Peters259272b2003-04-06 19:41:39 +0000330}
331
Jeremy Hyltonc5007aa2000-06-30 05:02:53 +0000332/*** end of list stuff ***/
333
334
Tim Peters19b74c72002-07-01 03:52:19 +0000335/* Set all gc_refs = ob_refcnt. After this, gc_refs is > 0 for all objects
336 * in containers, and is GC_REACHABLE for all tracked gc objects not in
337 * containers.
Tim Peters88396172002-06-30 17:56:40 +0000338 */
Jeremy Hyltonc5007aa2000-06-30 05:02:53 +0000339static void
340update_refs(PyGC_Head *containers)
341{
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000342 PyGC_Head *gc = containers->gc.gc_next;
343 for (; gc != containers; gc = gc->gc.gc_next) {
Antoine Pitrou796564c2013-07-30 19:59:21 +0200344 assert(_PyGCHead_REFS(gc) == GC_REACHABLE);
345 _PyGCHead_SET_REFS(gc, Py_REFCNT(FROM_GC(gc)));
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000346 /* Python's cyclic gc should never see an incoming refcount
347 * of 0: if something decref'ed to 0, it should have been
348 * deallocated immediately at that time.
349 * Possible cause (if the assert triggers): a tp_dealloc
350 * routine left a gc-aware object tracked during its teardown
351 * phase, and did something-- or allowed something to happen --
352 * that called back into Python. gc can trigger then, and may
353 * see the still-tracked dying object. Before this assert
354 * was added, such mistakes went on to allow gc to try to
355 * delete the object again. In a debug build, that caused
356 * a mysterious segfault, when _Py_ForgetReference tried
357 * to remove the object from the doubly-linked list of all
358 * objects a second time. In a release build, an actual
359 * double deallocation occurred, which leads to corruption
360 * of the allocator's internal bookkeeping pointers. That's
361 * so serious that maybe this should be a release-build
362 * check instead of an assert?
363 */
Antoine Pitrou796564c2013-07-30 19:59:21 +0200364 assert(_PyGCHead_REFS(gc) != 0);
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000365 }
Jeremy Hyltonc5007aa2000-06-30 05:02:53 +0000366}
367
Tim Peters19b74c72002-07-01 03:52:19 +0000368/* A traversal callback for subtract_refs. */
Jeremy Hyltonc5007aa2000-06-30 05:02:53 +0000369static int
370visit_decref(PyObject *op, void *data)
371{
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000372 assert(op != NULL);
373 if (PyObject_IS_GC(op)) {
374 PyGC_Head *gc = AS_GC(op);
375 /* We're only interested in gc_refs for objects in the
376 * generation being collected, which can be recognized
377 * because only they have positive gc_refs.
378 */
Antoine Pitrou796564c2013-07-30 19:59:21 +0200379 assert(_PyGCHead_REFS(gc) != 0); /* else refcount was too small */
380 if (_PyGCHead_REFS(gc) > 0)
381 _PyGCHead_DECREF(gc);
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000382 }
383 return 0;
Jeremy Hyltonc5007aa2000-06-30 05:02:53 +0000384}
385
Tim Peters19b74c72002-07-01 03:52:19 +0000386/* Subtract internal references from gc_refs. After this, gc_refs is >= 0
387 * for all objects in containers, and is GC_REACHABLE for all tracked gc
388 * objects not in containers. The ones with gc_refs > 0 are directly
389 * reachable from outside containers, and so can't be collected.
390 */
Jeremy Hyltonc5007aa2000-06-30 05:02:53 +0000391static void
392subtract_refs(PyGC_Head *containers)
393{
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000394 traverseproc traverse;
395 PyGC_Head *gc = containers->gc.gc_next;
396 for (; gc != containers; gc=gc->gc.gc_next) {
397 traverse = Py_TYPE(FROM_GC(gc))->tp_traverse;
398 (void) traverse(FROM_GC(gc),
399 (visitproc)visit_decref,
400 NULL);
401 }
Jeremy Hyltonc5007aa2000-06-30 05:02:53 +0000402}
403
Tim Peters19b74c72002-07-01 03:52:19 +0000404/* A traversal callback for move_unreachable. */
Jeremy Hyltonc5007aa2000-06-30 05:02:53 +0000405static int
Tim Peters19b74c72002-07-01 03:52:19 +0000406visit_reachable(PyObject *op, PyGC_Head *reachable)
Jeremy Hyltonc5007aa2000-06-30 05:02:53 +0000407{
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000408 if (PyObject_IS_GC(op)) {
409 PyGC_Head *gc = AS_GC(op);
Antoine Pitrou796564c2013-07-30 19:59:21 +0200410 const Py_ssize_t gc_refs = _PyGCHead_REFS(gc);
Tim Peters19b74c72002-07-01 03:52:19 +0000411
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000412 if (gc_refs == 0) {
413 /* This is in move_unreachable's 'young' list, but
414 * the traversal hasn't yet gotten to it. All
415 * we need to do is tell move_unreachable that it's
416 * reachable.
417 */
Antoine Pitrou796564c2013-07-30 19:59:21 +0200418 _PyGCHead_SET_REFS(gc, 1);
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000419 }
420 else if (gc_refs == GC_TENTATIVELY_UNREACHABLE) {
421 /* This had gc_refs = 0 when move_unreachable got
422 * to it, but turns out it's reachable after all.
423 * Move it back to move_unreachable's 'young' list,
424 * and move_unreachable will eventually get to it
425 * again.
426 */
427 gc_list_move(gc, reachable);
Antoine Pitrou796564c2013-07-30 19:59:21 +0200428 _PyGCHead_SET_REFS(gc, 1);
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000429 }
430 /* Else there's nothing to do.
431 * If gc_refs > 0, it must be in move_unreachable's 'young'
432 * list, and move_unreachable will eventually get to it.
433 * If gc_refs == GC_REACHABLE, it's either in some other
434 * generation so we don't care about it, or move_unreachable
435 * already dealt with it.
436 * If gc_refs == GC_UNTRACKED, it must be ignored.
437 */
438 else {
439 assert(gc_refs > 0
440 || gc_refs == GC_REACHABLE
441 || gc_refs == GC_UNTRACKED);
442 }
443 }
444 return 0;
Jeremy Hyltonc5007aa2000-06-30 05:02:53 +0000445}
446
Tim Peters19b74c72002-07-01 03:52:19 +0000447/* Move the unreachable objects from young to unreachable. After this,
448 * all objects in young have gc_refs = GC_REACHABLE, and all objects in
449 * unreachable have gc_refs = GC_TENTATIVELY_UNREACHABLE. All tracked
450 * gc objects not in young or unreachable still have gc_refs = GC_REACHABLE.
451 * All objects in young after this are directly or indirectly reachable
452 * from outside the original young; and all objects in unreachable are
453 * not.
Tim Peters88396172002-06-30 17:56:40 +0000454 */
Jeremy Hyltonc5007aa2000-06-30 05:02:53 +0000455static void
Tim Peters19b74c72002-07-01 03:52:19 +0000456move_unreachable(PyGC_Head *young, PyGC_Head *unreachable)
Jeremy Hyltonc5007aa2000-06-30 05:02:53 +0000457{
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000458 PyGC_Head *gc = young->gc.gc_next;
Tim Peters19b74c72002-07-01 03:52:19 +0000459
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000460 /* Invariants: all objects "to the left" of us in young have gc_refs
461 * = GC_REACHABLE, and are indeed reachable (directly or indirectly)
462 * from outside the young list as it was at entry. All other objects
463 * from the original young "to the left" of us are in unreachable now,
464 * and have gc_refs = GC_TENTATIVELY_UNREACHABLE. All objects to the
465 * left of us in 'young' now have been scanned, and no objects here
466 * or to the right have been scanned yet.
467 */
Tim Peters19b74c72002-07-01 03:52:19 +0000468
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000469 while (gc != young) {
470 PyGC_Head *next;
Tim Peters19b74c72002-07-01 03:52:19 +0000471
Antoine Pitrou796564c2013-07-30 19:59:21 +0200472 if (_PyGCHead_REFS(gc)) {
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000473 /* gc is definitely reachable from outside the
474 * original 'young'. Mark it as such, and traverse
475 * its pointers to find any other objects that may
476 * be directly reachable from it. Note that the
477 * call to tp_traverse may append objects to young,
478 * so we have to wait until it returns to determine
479 * the next object to visit.
480 */
481 PyObject *op = FROM_GC(gc);
482 traverseproc traverse = Py_TYPE(op)->tp_traverse;
Antoine Pitrou796564c2013-07-30 19:59:21 +0200483 assert(_PyGCHead_REFS(gc) > 0);
484 _PyGCHead_SET_REFS(gc, GC_REACHABLE);
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000485 (void) traverse(op,
486 (visitproc)visit_reachable,
487 (void *)young);
488 next = gc->gc.gc_next;
489 if (PyTuple_CheckExact(op)) {
490 _PyTuple_MaybeUntrack(op);
491 }
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000492 }
493 else {
494 /* This *may* be unreachable. To make progress,
495 * assume it is. gc isn't directly reachable from
496 * any object we've already traversed, but may be
497 * reachable from an object we haven't gotten to yet.
498 * visit_reachable will eventually move gc back into
499 * young if that's so, and we'll see it again.
500 */
501 next = gc->gc.gc_next;
502 gc_list_move(gc, unreachable);
Antoine Pitrou796564c2013-07-30 19:59:21 +0200503 _PyGCHead_SET_REFS(gc, GC_TENTATIVELY_UNREACHABLE);
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000504 }
505 gc = next;
506 }
Jeremy Hyltonc5007aa2000-06-30 05:02:53 +0000507}
508
Antoine Pitroue1ad3da2012-05-28 22:22:34 +0200509/* Try to untrack all currently tracked dictionaries */
510static void
511untrack_dicts(PyGC_Head *head)
512{
513 PyGC_Head *next, *gc = head->gc.gc_next;
514 while (gc != head) {
515 PyObject *op = FROM_GC(gc);
516 next = gc->gc.gc_next;
517 if (PyDict_CheckExact(op))
518 _PyDict_MaybeUntrack(op);
519 gc = next;
520 }
521}
522
Antoine Pitrou796564c2013-07-30 19:59:21 +0200523/* Return true if object has a pre-PEP 442 finalization method. */
Neil Schemenauera765c122001-11-01 17:35:23 +0000524static int
Antoine Pitrou796564c2013-07-30 19:59:21 +0200525has_legacy_finalizer(PyObject *op)
Jeremy Hyltonc5007aa2000-06-30 05:02:53 +0000526{
Antoine Pitrou796564c2013-07-30 19:59:21 +0200527 return op->ob_type->tp_del != NULL;
Neil Schemenauera765c122001-11-01 17:35:23 +0000528}
529
Antoine Pitrou796564c2013-07-30 19:59:21 +0200530/* Move the objects in unreachable with tp_del slots into `finalizers`.
Tim Petersead8b7a2004-10-30 23:09:22 +0000531 * Objects moved into `finalizers` have gc_refs set to GC_REACHABLE; the
532 * objects remaining in unreachable are left at GC_TENTATIVELY_UNREACHABLE.
Jeremy Hyltonce136e92003-04-04 19:59:06 +0000533 */
Neil Schemenauera765c122001-11-01 17:35:23 +0000534static void
Antoine Pitrou796564c2013-07-30 19:59:21 +0200535move_legacy_finalizers(PyGC_Head *unreachable, PyGC_Head *finalizers)
Neil Schemenauera765c122001-11-01 17:35:23 +0000536{
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000537 PyGC_Head *gc;
538 PyGC_Head *next;
Tim Petersf6b80452003-04-07 19:21:15 +0000539
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000540 /* March over unreachable. Move objects with finalizers into
541 * `finalizers`.
542 */
543 for (gc = unreachable->gc.gc_next; gc != unreachable; gc = next) {
544 PyObject *op = FROM_GC(gc);
Jeremy Hyltonce136e92003-04-04 19:59:06 +0000545
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000546 assert(IS_TENTATIVELY_UNREACHABLE(op));
547 next = gc->gc.gc_next;
Tim Petersf6ae7a42003-04-05 18:40:50 +0000548
Antoine Pitrou796564c2013-07-30 19:59:21 +0200549 if (has_legacy_finalizer(op)) {
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000550 gc_list_move(gc, finalizers);
Antoine Pitrou796564c2013-07-30 19:59:21 +0200551 _PyGCHead_SET_REFS(gc, GC_REACHABLE);
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000552 }
553 }
Jeremy Hyltonc5007aa2000-06-30 05:02:53 +0000554}
555
Antoine Pitrou796564c2013-07-30 19:59:21 +0200556/* A traversal callback for move_legacy_finalizer_reachable. */
Tim Peters19b74c72002-07-01 03:52:19 +0000557static int
558visit_move(PyObject *op, PyGC_Head *tolist)
559{
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000560 if (PyObject_IS_GC(op)) {
561 if (IS_TENTATIVELY_UNREACHABLE(op)) {
562 PyGC_Head *gc = AS_GC(op);
563 gc_list_move(gc, tolist);
Antoine Pitrou796564c2013-07-30 19:59:21 +0200564 _PyGCHead_SET_REFS(gc, GC_REACHABLE);
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000565 }
566 }
567 return 0;
Tim Peters19b74c72002-07-01 03:52:19 +0000568}
569
570/* Move objects that are reachable from finalizers, from the unreachable set
Tim Petersf6b80452003-04-07 19:21:15 +0000571 * into finalizers set.
Tim Peters19b74c72002-07-01 03:52:19 +0000572 */
Jeremy Hyltonc5007aa2000-06-30 05:02:53 +0000573static void
Antoine Pitrou796564c2013-07-30 19:59:21 +0200574move_legacy_finalizer_reachable(PyGC_Head *finalizers)
Jeremy Hyltonc5007aa2000-06-30 05:02:53 +0000575{
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000576 traverseproc traverse;
577 PyGC_Head *gc = finalizers->gc.gc_next;
578 for (; gc != finalizers; gc = gc->gc.gc_next) {
579 /* Note that the finalizers list may grow during this. */
580 traverse = Py_TYPE(FROM_GC(gc))->tp_traverse;
581 (void) traverse(FROM_GC(gc),
582 (visitproc)visit_move,
583 (void *)finalizers);
584 }
Jeremy Hyltonc5007aa2000-06-30 05:02:53 +0000585}
586
Tim Petersead8b7a2004-10-30 23:09:22 +0000587/* Clear all weakrefs to unreachable objects, and if such a weakref has a
588 * callback, invoke it if necessary. Note that it's possible for such
589 * weakrefs to be outside the unreachable set -- indeed, those are precisely
590 * the weakrefs whose callbacks must be invoked. See gc_weakref.txt for
591 * overview & some details. Some weakrefs with callbacks may be reclaimed
592 * directly by this routine; the number reclaimed is the return value. Other
593 * weakrefs with callbacks may be moved into the `old` generation. Objects
594 * moved into `old` have gc_refs set to GC_REACHABLE; the objects remaining in
595 * unreachable are left at GC_TENTATIVELY_UNREACHABLE. When this returns,
596 * no object in `unreachable` is weakly referenced anymore.
Tim Peters403a2032003-11-20 21:21:46 +0000597 */
598static int
Tim Petersead8b7a2004-10-30 23:09:22 +0000599handle_weakrefs(PyGC_Head *unreachable, PyGC_Head *old)
Tim Peters403a2032003-11-20 21:21:46 +0000600{
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000601 PyGC_Head *gc;
602 PyObject *op; /* generally FROM_GC(gc) */
603 PyWeakReference *wr; /* generally a cast of op */
604 PyGC_Head wrcb_to_call; /* weakrefs with callbacks to call */
605 PyGC_Head *next;
606 int num_freed = 0;
Tim Peters403a2032003-11-20 21:21:46 +0000607
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000608 gc_list_init(&wrcb_to_call);
Tim Peters403a2032003-11-20 21:21:46 +0000609
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000610 /* Clear all weakrefs to the objects in unreachable. If such a weakref
611 * also has a callback, move it into `wrcb_to_call` if the callback
612 * needs to be invoked. Note that we cannot invoke any callbacks until
613 * all weakrefs to unreachable objects are cleared, lest the callback
614 * resurrect an unreachable object via a still-active weakref. We
615 * make another pass over wrcb_to_call, invoking callbacks, after this
616 * pass completes.
617 */
618 for (gc = unreachable->gc.gc_next; gc != unreachable; gc = next) {
619 PyWeakReference **wrlist;
Tim Petersead8b7a2004-10-30 23:09:22 +0000620
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000621 op = FROM_GC(gc);
622 assert(IS_TENTATIVELY_UNREACHABLE(op));
623 next = gc->gc.gc_next;
Tim Petersead8b7a2004-10-30 23:09:22 +0000624
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000625 if (! PyType_SUPPORTS_WEAKREFS(Py_TYPE(op)))
626 continue;
Tim Petersead8b7a2004-10-30 23:09:22 +0000627
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000628 /* It supports weakrefs. Does it have any? */
629 wrlist = (PyWeakReference **)
630 PyObject_GET_WEAKREFS_LISTPTR(op);
Tim Petersead8b7a2004-10-30 23:09:22 +0000631
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000632 /* `op` may have some weakrefs. March over the list, clear
633 * all the weakrefs, and move the weakrefs with callbacks
634 * that must be called into wrcb_to_call.
635 */
636 for (wr = *wrlist; wr != NULL; wr = *wrlist) {
637 PyGC_Head *wrasgc; /* AS_GC(wr) */
Tim Petersead8b7a2004-10-30 23:09:22 +0000638
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000639 /* _PyWeakref_ClearRef clears the weakref but leaves
640 * the callback pointer intact. Obscure: it also
641 * changes *wrlist.
642 */
643 assert(wr->wr_object == op);
644 _PyWeakref_ClearRef(wr);
645 assert(wr->wr_object == Py_None);
646 if (wr->wr_callback == NULL)
647 continue; /* no callback */
Tim Petersead8b7a2004-10-30 23:09:22 +0000648
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000649 /* Headache time. `op` is going away, and is weakly referenced by
650 * `wr`, which has a callback. Should the callback be invoked? If wr
651 * is also trash, no:
652 *
653 * 1. There's no need to call it. The object and the weakref are
654 * both going away, so it's legitimate to pretend the weakref is
655 * going away first. The user has to ensure a weakref outlives its
656 * referent if they want a guarantee that the wr callback will get
657 * invoked.
658 *
659 * 2. It may be catastrophic to call it. If the callback is also in
660 * cyclic trash (CT), then although the CT is unreachable from
661 * outside the current generation, CT may be reachable from the
662 * callback. Then the callback could resurrect insane objects.
663 *
664 * Since the callback is never needed and may be unsafe in this case,
665 * wr is simply left in the unreachable set. Note that because we
666 * already called _PyWeakref_ClearRef(wr), its callback will never
667 * trigger.
668 *
669 * OTOH, if wr isn't part of CT, we should invoke the callback: the
670 * weakref outlived the trash. Note that since wr isn't CT in this
671 * case, its callback can't be CT either -- wr acted as an external
672 * root to this generation, and therefore its callback did too. So
673 * nothing in CT is reachable from the callback either, so it's hard
674 * to imagine how calling it later could create a problem for us. wr
675 * is moved to wrcb_to_call in this case.
676 */
677 if (IS_TENTATIVELY_UNREACHABLE(wr))
678 continue;
679 assert(IS_REACHABLE(wr));
Tim Peterscc2a8662004-10-31 22:12:43 +0000680
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000681 /* Create a new reference so that wr can't go away
682 * before we can process it again.
683 */
684 Py_INCREF(wr);
Tim Petersead8b7a2004-10-30 23:09:22 +0000685
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000686 /* Move wr to wrcb_to_call, for the next pass. */
687 wrasgc = AS_GC(wr);
688 assert(wrasgc != next); /* wrasgc is reachable, but
689 next isn't, so they can't
690 be the same */
691 gc_list_move(wrasgc, &wrcb_to_call);
692 }
693 }
Tim Petersead8b7a2004-10-30 23:09:22 +0000694
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000695 /* Invoke the callbacks we decided to honor. It's safe to invoke them
696 * because they can't reference unreachable objects.
697 */
698 while (! gc_list_is_empty(&wrcb_to_call)) {
699 PyObject *temp;
700 PyObject *callback;
Tim Petersead8b7a2004-10-30 23:09:22 +0000701
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000702 gc = wrcb_to_call.gc.gc_next;
703 op = FROM_GC(gc);
704 assert(IS_REACHABLE(op));
705 assert(PyWeakref_Check(op));
706 wr = (PyWeakReference *)op;
707 callback = wr->wr_callback;
708 assert(callback != NULL);
Tim Petersead8b7a2004-10-30 23:09:22 +0000709
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000710 /* copy-paste of weakrefobject.c's handle_callback() */
711 temp = PyObject_CallFunctionObjArgs(callback, wr, NULL);
712 if (temp == NULL)
713 PyErr_WriteUnraisable(callback);
714 else
715 Py_DECREF(temp);
Tim Petersead8b7a2004-10-30 23:09:22 +0000716
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000717 /* Give up the reference we created in the first pass. When
718 * op's refcount hits 0 (which it may or may not do right now),
719 * op's tp_dealloc will decref op->wr_callback too. Note
720 * that the refcount probably will hit 0 now, and because this
721 * weakref was reachable to begin with, gc didn't already
722 * add it to its count of freed objects. Example: a reachable
723 * weak value dict maps some key to this reachable weakref.
724 * The callback removes this key->weakref mapping from the
725 * dict, leaving no other references to the weakref (excepting
726 * ours).
727 */
728 Py_DECREF(op);
729 if (wrcb_to_call.gc.gc_next == gc) {
730 /* object is still alive -- move it */
731 gc_list_move(gc, old);
732 }
733 else
734 ++num_freed;
735 }
Tim Petersead8b7a2004-10-30 23:09:22 +0000736
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000737 return num_freed;
Tim Peters403a2032003-11-20 21:21:46 +0000738}
739
Jeremy Hyltonc5007aa2000-06-30 05:02:53 +0000740static void
Jeremy Hylton06257772000-08-31 15:10:24 +0000741debug_cycle(char *msg, PyObject *op)
Jeremy Hyltonc5007aa2000-06-30 05:02:53 +0000742{
Victor Stinner499dfcf2011-03-21 13:26:24 +0100743 PySys_FormatStderr("gc: %s <%s %p>\n",
744 msg, Py_TYPE(op)->tp_name, op);
Jeremy Hyltonc5007aa2000-06-30 05:02:53 +0000745}
746
Antoine Pitrou796564c2013-07-30 19:59:21 +0200747/* Handle uncollectable garbage (cycles with tp_del slots, and stuff reachable
Tim Petersbf384c22003-04-06 00:11:39 +0000748 * only from such cycles).
Tim Petersf6b80452003-04-07 19:21:15 +0000749 * If DEBUG_SAVEALL, all objects in finalizers are appended to the module
750 * garbage list (a Python list), else only the objects in finalizers with
751 * __del__ methods are appended to garbage. All objects in finalizers are
752 * merged into the old list regardless.
Tim Peters259272b2003-04-06 19:41:39 +0000753 * Returns 0 if all OK, <0 on error (out of memory to grow the garbage list).
754 * The finalizers list is made empty on a successful return.
Tim Petersbf384c22003-04-06 00:11:39 +0000755 */
Tim Peters259272b2003-04-06 19:41:39 +0000756static int
Antoine Pitrou796564c2013-07-30 19:59:21 +0200757handle_legacy_finalizers(PyGC_Head *finalizers, PyGC_Head *old)
Jeremy Hyltonc5007aa2000-06-30 05:02:53 +0000758{
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000759 PyGC_Head *gc = finalizers->gc.gc_next;
Tim Petersf6b80452003-04-07 19:21:15 +0000760
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000761 if (garbage == NULL) {
762 garbage = PyList_New(0);
763 if (garbage == NULL)
764 Py_FatalError("gc couldn't create gc.garbage list");
765 }
766 for (; gc != finalizers; gc = gc->gc.gc_next) {
767 PyObject *op = FROM_GC(gc);
Tim Petersf6b80452003-04-07 19:21:15 +0000768
Antoine Pitrou796564c2013-07-30 19:59:21 +0200769 if ((debug & DEBUG_SAVEALL) || has_legacy_finalizer(op)) {
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000770 if (PyList_Append(garbage, op) < 0)
771 return -1;
772 }
773 }
Tim Petersf6b80452003-04-07 19:21:15 +0000774
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000775 gc_list_merge(finalizers, old);
776 return 0;
Jeremy Hyltonc5007aa2000-06-30 05:02:53 +0000777}
778
Tim Peters5fbc7b12014-05-08 17:42:19 -0500779/* Run first-time finalizers (if any) on all the objects in collectable.
780 * Note that this may remove some (or even all) of the objects from the
781 * list, due to refcounts falling to 0.
782 */
Antoine Pitrou796564c2013-07-30 19:59:21 +0200783static void
Tim Peters5fbc7b12014-05-08 17:42:19 -0500784finalize_garbage(PyGC_Head *collectable)
Antoine Pitrou796564c2013-07-30 19:59:21 +0200785{
786 destructor finalize;
Tim Peters5fbc7b12014-05-08 17:42:19 -0500787 PyGC_Head seen;
Antoine Pitrou796564c2013-07-30 19:59:21 +0200788
Tim Peters5fbc7b12014-05-08 17:42:19 -0500789 /* While we're going through the loop, `finalize(op)` may cause op, or
790 * other objects, to be reclaimed via refcounts falling to zero. So
791 * there's little we can rely on about the structure of the input
792 * `collectable` list across iterations. For safety, we always take the
793 * first object in that list and move it to a temporary `seen` list.
794 * If objects vanish from the `collectable` and `seen` lists we don't
795 * care.
796 */
797 gc_list_init(&seen);
798
799 while (!gc_list_is_empty(collectable)) {
800 PyGC_Head *gc = collectable->gc.gc_next;
Antoine Pitrou796564c2013-07-30 19:59:21 +0200801 PyObject *op = FROM_GC(gc);
Tim Peters5fbc7b12014-05-08 17:42:19 -0500802 gc_list_move(gc, &seen);
Antoine Pitrou796564c2013-07-30 19:59:21 +0200803 if (!_PyGCHead_FINALIZED(gc) &&
Tim Peters5fbc7b12014-05-08 17:42:19 -0500804 PyType_HasFeature(Py_TYPE(op), Py_TPFLAGS_HAVE_FINALIZE) &&
805 (finalize = Py_TYPE(op)->tp_finalize) != NULL) {
Antoine Pitrou796564c2013-07-30 19:59:21 +0200806 _PyGCHead_SET_FINALIZED(gc, 1);
807 Py_INCREF(op);
808 finalize(op);
Antoine Pitrou796564c2013-07-30 19:59:21 +0200809 Py_DECREF(op);
810 }
811 }
Tim Peters5fbc7b12014-05-08 17:42:19 -0500812 gc_list_merge(&seen, collectable);
Antoine Pitrou796564c2013-07-30 19:59:21 +0200813}
814
815/* Walk the collectable list and check that they are really unreachable
816 from the outside (some objects could have been resurrected by a
817 finalizer). */
818static int
819check_garbage(PyGC_Head *collectable)
820{
821 PyGC_Head *gc;
822 for (gc = collectable->gc.gc_next; gc != collectable;
823 gc = gc->gc.gc_next) {
824 _PyGCHead_SET_REFS(gc, Py_REFCNT(FROM_GC(gc)));
825 assert(_PyGCHead_REFS(gc) != 0);
826 }
827 subtract_refs(collectable);
828 for (gc = collectable->gc.gc_next; gc != collectable;
829 gc = gc->gc.gc_next) {
830 assert(_PyGCHead_REFS(gc) >= 0);
831 if (_PyGCHead_REFS(gc) != 0)
832 return -1;
833 }
834 return 0;
835}
836
837static void
838revive_garbage(PyGC_Head *collectable)
839{
840 PyGC_Head *gc;
841 for (gc = collectable->gc.gc_next; gc != collectable;
842 gc = gc->gc.gc_next) {
843 _PyGCHead_SET_REFS(gc, GC_REACHABLE);
844 }
845}
846
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000847/* Break reference cycles by clearing the containers involved. This is
Jeremy Hyltonc5007aa2000-06-30 05:02:53 +0000848 * tricky business as the lists can be changing and we don't know which
Tim Peters19b74c72002-07-01 03:52:19 +0000849 * objects may be freed. It is possible I screwed something up here.
850 */
Jeremy Hyltonc5007aa2000-06-30 05:02:53 +0000851static void
Jeremy Hyltonce136e92003-04-04 19:59:06 +0000852delete_garbage(PyGC_Head *collectable, PyGC_Head *old)
Jeremy Hyltonc5007aa2000-06-30 05:02:53 +0000853{
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000854 inquiry clear;
Jeremy Hyltonc5007aa2000-06-30 05:02:53 +0000855
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000856 while (!gc_list_is_empty(collectable)) {
857 PyGC_Head *gc = collectable->gc.gc_next;
858 PyObject *op = FROM_GC(gc);
Tim Peters88396172002-06-30 17:56:40 +0000859
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000860 if (debug & DEBUG_SAVEALL) {
861 PyList_Append(garbage, op);
862 }
863 else {
864 if ((clear = Py_TYPE(op)->tp_clear) != NULL) {
865 Py_INCREF(op);
866 clear(op);
867 Py_DECREF(op);
868 }
869 }
870 if (collectable->gc.gc_next == gc) {
871 /* object is still alive, move it, it may die later */
872 gc_list_move(gc, old);
Antoine Pitrou796564c2013-07-30 19:59:21 +0200873 _PyGCHead_SET_REFS(gc, GC_REACHABLE);
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000874 }
875 }
Jeremy Hyltonc5007aa2000-06-30 05:02:53 +0000876}
877
Christian Heimesa156e092008-02-16 07:38:31 +0000878/* Clear all free lists
879 * All free lists are cleared during the collection of the highest generation.
880 * Allocated items in the free list may keep a pymalloc arena occupied.
881 * Clearing the free lists may give back memory to the OS earlier.
882 */
883static void
884clear_freelists(void)
885{
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000886 (void)PyMethod_ClearFreeList();
887 (void)PyFrame_ClearFreeList();
888 (void)PyCFunction_ClearFreeList();
889 (void)PyTuple_ClearFreeList();
890 (void)PyUnicode_ClearFreeList();
891 (void)PyFloat_ClearFreeList();
Antoine Pitrou9a812cb2011-11-15 00:00:12 +0100892 (void)PyList_ClearFreeList();
893 (void)PyDict_ClearFreeList();
Antoine Pitrou093ce9c2011-12-16 11:24:27 +0100894 (void)PySet_ClearFreeList();
Christian Heimesa156e092008-02-16 07:38:31 +0000895}
896
Antoine Pitrou621601a2008-12-17 23:18:19 +0000897static double
898get_time(void)
899{
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000900 double result = 0;
901 if (tmod != NULL) {
Martin v. Löwisbd928fe2011-10-14 10:20:37 +0200902 _Py_IDENTIFIER(time);
Martin v. Löwisafe55bb2011-10-09 10:38:36 +0200903
904 PyObject *f = _PyObject_CallMethodId(tmod, &PyId_time, NULL);
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000905 if (f == NULL) {
906 PyErr_Clear();
907 }
908 else {
909 if (PyFloat_Check(f))
910 result = PyFloat_AsDouble(f);
911 Py_DECREF(f);
912 }
913 }
914 return result;
Antoine Pitrou621601a2008-12-17 23:18:19 +0000915}
916
Jeremy Hyltonc5007aa2000-06-30 05:02:53 +0000917/* This is the main function. Read this to understand how the
918 * collection process works. */
Neal Norwitz7b216c52006-03-04 20:01:53 +0000919static Py_ssize_t
Antoine Pitroufef34e32013-05-19 01:11:58 +0200920collect(int generation, Py_ssize_t *n_collected, Py_ssize_t *n_uncollectable,
921 int nofail)
Jeremy Hyltonc5007aa2000-06-30 05:02:53 +0000922{
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000923 int i;
924 Py_ssize_t m = 0; /* # objects collected */
925 Py_ssize_t n = 0; /* # unreachable objects that couldn't be collected */
926 PyGC_Head *young; /* the generation we are examining */
927 PyGC_Head *old; /* next older generation */
928 PyGC_Head unreachable; /* non-problematic unreachable trash */
929 PyGC_Head finalizers; /* objects with, & reachable from, __del__ */
930 PyGC_Head *gc;
931 double t1 = 0.0;
Antoine Pitroud4156c12012-10-30 22:43:19 +0100932 struct gc_generation_stats *stats = &generation_stats[generation];
Jeremy Hyltonc5007aa2000-06-30 05:02:53 +0000933
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000934 if (debug & DEBUG_STATS) {
935 PySys_WriteStderr("gc: collecting generation %d...\n",
936 generation);
937 PySys_WriteStderr("gc: objects in each generation:");
938 for (i = 0; i < NUM_GENERATIONS; i++)
939 PySys_WriteStderr(" %" PY_FORMAT_SIZE_T "d",
940 gc_list_size(GEN_HEAD(i)));
941 t1 = get_time();
942 PySys_WriteStderr("\n");
943 }
Neil Schemenauer2880ae52002-05-04 05:35:20 +0000944
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000945 /* update collection and allocation counters */
946 if (generation+1 < NUM_GENERATIONS)
947 generations[generation+1].count += 1;
948 for (i = 0; i <= generation; i++)
949 generations[i].count = 0;
Neil Schemenauer2880ae52002-05-04 05:35:20 +0000950
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000951 /* merge younger generations with one we are currently collecting */
952 for (i = 0; i < generation; i++) {
953 gc_list_merge(GEN_HEAD(i), GEN_HEAD(generation));
954 }
Neil Schemenauer2880ae52002-05-04 05:35:20 +0000955
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000956 /* handy references */
957 young = GEN_HEAD(generation);
958 if (generation < NUM_GENERATIONS-1)
959 old = GEN_HEAD(generation+1);
960 else
961 old = young;
Jeremy Hyltonc5007aa2000-06-30 05:02:53 +0000962
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000963 /* Using ob_refcnt and gc_refs, calculate which objects in the
964 * container set are reachable from outside the set (i.e., have a
965 * refcount greater than 0 when all the references within the
966 * set are taken into account).
967 */
968 update_refs(young);
969 subtract_refs(young);
Jeremy Hyltonc5007aa2000-06-30 05:02:53 +0000970
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000971 /* Leave everything reachable from outside young in young, and move
972 * everything else (in young) to unreachable.
973 * NOTE: This used to move the reachable objects into a reachable
974 * set instead. But most things usually turn out to be reachable,
975 * so it's more efficient to move the unreachable things.
976 */
977 gc_list_init(&unreachable);
978 move_unreachable(young, &unreachable);
Jeremy Hyltonc5007aa2000-06-30 05:02:53 +0000979
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000980 /* Move reachable objects to next generation. */
981 if (young != old) {
982 if (generation == NUM_GENERATIONS - 2) {
983 long_lived_pending += gc_list_size(young);
984 }
985 gc_list_merge(young, old);
986 }
987 else {
Antoine Pitroue1ad3da2012-05-28 22:22:34 +0200988 /* We only untrack dicts in full collections, to avoid quadratic
989 dict build-up. See issue #14775. */
990 untrack_dicts(young);
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000991 long_lived_pending = 0;
992 long_lived_total = gc_list_size(young);
993 }
Jeremy Hyltonc5007aa2000-06-30 05:02:53 +0000994
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000995 /* All objects in unreachable are trash, but objects reachable from
Antoine Pitrou796564c2013-07-30 19:59:21 +0200996 * legacy finalizers (e.g. tp_del) can't safely be deleted.
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000997 */
998 gc_list_init(&finalizers);
Antoine Pitrou796564c2013-07-30 19:59:21 +0200999 move_legacy_finalizers(&unreachable, &finalizers);
1000 /* finalizers contains the unreachable objects with a legacy finalizer;
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001001 * unreachable objects reachable *from* those are also uncollectable,
1002 * and we move those into the finalizers list too.
1003 */
Antoine Pitrou796564c2013-07-30 19:59:21 +02001004 move_legacy_finalizer_reachable(&finalizers);
Jeremy Hyltonc5007aa2000-06-30 05:02:53 +00001005
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001006 /* Collect statistics on collectable objects found and print
1007 * debugging information.
1008 */
1009 for (gc = unreachable.gc.gc_next; gc != &unreachable;
1010 gc = gc->gc.gc_next) {
1011 m++;
1012 if (debug & DEBUG_COLLECTABLE) {
1013 debug_cycle("collectable", FROM_GC(gc));
1014 }
1015 }
Tim Petersead8b7a2004-10-30 23:09:22 +00001016
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001017 /* Clear weakrefs and invoke callbacks as necessary. */
1018 m += handle_weakrefs(&unreachable, old);
Tim Petersead8b7a2004-10-30 23:09:22 +00001019
Antoine Pitrou796564c2013-07-30 19:59:21 +02001020 /* Call tp_finalize on objects which have one. */
Tim Peters5fbc7b12014-05-08 17:42:19 -05001021 finalize_garbage(&unreachable);
Antoine Pitrou796564c2013-07-30 19:59:21 +02001022
1023 if (check_garbage(&unreachable)) {
1024 revive_garbage(&unreachable);
1025 gc_list_merge(&unreachable, old);
1026 }
1027 else {
1028 /* Call tp_clear on objects in the unreachable set. This will cause
1029 * the reference cycles to be broken. It may also cause some objects
1030 * in finalizers to be freed.
1031 */
1032 delete_garbage(&unreachable, old);
1033 }
Jeremy Hyltonc5007aa2000-06-30 05:02:53 +00001034
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001035 /* Collect statistics on uncollectable objects found and print
1036 * debugging information. */
1037 for (gc = finalizers.gc.gc_next;
1038 gc != &finalizers;
1039 gc = gc->gc.gc_next) {
1040 n++;
1041 if (debug & DEBUG_UNCOLLECTABLE)
1042 debug_cycle("uncollectable", FROM_GC(gc));
1043 }
1044 if (debug & DEBUG_STATS) {
1045 double t2 = get_time();
1046 if (m == 0 && n == 0)
1047 PySys_WriteStderr("gc: done");
1048 else
1049 PySys_WriteStderr(
1050 "gc: done, "
1051 "%" PY_FORMAT_SIZE_T "d unreachable, "
1052 "%" PY_FORMAT_SIZE_T "d uncollectable",
1053 n+m, n);
1054 if (t1 && t2) {
1055 PySys_WriteStderr(", %.4fs elapsed", t2-t1);
1056 }
1057 PySys_WriteStderr(".\n");
1058 }
Jeremy Hyltonc5007aa2000-06-30 05:02:53 +00001059
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001060 /* Append instances in the uncollectable set to a Python
1061 * reachable list of garbage. The programmer has to deal with
1062 * this if they insist on creating this type of structure.
1063 */
Antoine Pitrou796564c2013-07-30 19:59:21 +02001064 (void)handle_legacy_finalizers(&finalizers, old);
Jeremy Hyltonc5007aa2000-06-30 05:02:53 +00001065
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001066 /* Clear free list only during the collection of the highest
1067 * generation */
1068 if (generation == NUM_GENERATIONS-1) {
1069 clear_freelists();
1070 }
Christian Heimesa156e092008-02-16 07:38:31 +00001071
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001072 if (PyErr_Occurred()) {
Antoine Pitroufef34e32013-05-19 01:11:58 +02001073 if (nofail) {
1074 PyErr_Clear();
1075 }
1076 else {
1077 if (gc_str == NULL)
1078 gc_str = PyUnicode_FromString("garbage collection");
1079 PyErr_WriteUnraisable(gc_str);
1080 Py_FatalError("unexpected exception during garbage collection");
1081 }
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001082 }
Kristján Valur Jónsson69c63522012-04-15 11:41:32 +00001083
Antoine Pitroud4156c12012-10-30 22:43:19 +01001084 /* Update stats */
Kristján Valur Jónsson69c63522012-04-15 11:41:32 +00001085 if (n_collected)
1086 *n_collected = m;
1087 if (n_uncollectable)
1088 *n_uncollectable = n;
Antoine Pitroud4156c12012-10-30 22:43:19 +01001089 stats->collections++;
1090 stats->collected += m;
1091 stats->uncollectable += n;
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001092 return n+m;
Jeremy Hyltonc5007aa2000-06-30 05:02:53 +00001093}
1094
Kristján Valur Jónsson69c63522012-04-15 11:41:32 +00001095/* Invoke progress callbacks to notify clients that garbage collection
1096 * is starting or stopping
1097 */
1098static void
1099invoke_gc_callback(const char *phase, int generation,
1100 Py_ssize_t collected, Py_ssize_t uncollectable)
1101{
1102 Py_ssize_t i;
1103 PyObject *info = NULL;
1104
1105 /* we may get called very early */
1106 if (callbacks == NULL)
1107 return;
1108 /* The local variable cannot be rebound, check it for sanity */
1109 assert(callbacks != NULL && PyList_CheckExact(callbacks));
1110 if (PyList_GET_SIZE(callbacks) != 0) {
1111 info = Py_BuildValue("{sisnsn}",
1112 "generation", generation,
1113 "collected", collected,
1114 "uncollectable", uncollectable);
1115 if (info == NULL) {
1116 PyErr_WriteUnraisable(NULL);
1117 return;
1118 }
1119 }
1120 for (i=0; i<PyList_GET_SIZE(callbacks); i++) {
1121 PyObject *r, *cb = PyList_GET_ITEM(callbacks, i);
1122 Py_INCREF(cb); /* make sure cb doesn't go away */
1123 r = PyObject_CallFunction(cb, "sO", phase, info);
1124 Py_XDECREF(r);
1125 if (r == NULL)
1126 PyErr_WriteUnraisable(cb);
1127 Py_DECREF(cb);
1128 }
1129 Py_XDECREF(info);
1130}
1131
1132/* Perform garbage collection of a generation and invoke
1133 * progress callbacks.
1134 */
1135static Py_ssize_t
1136collect_with_callback(int generation)
1137{
1138 Py_ssize_t result, collected, uncollectable;
1139 invoke_gc_callback("start", generation, 0, 0);
Antoine Pitroufef34e32013-05-19 01:11:58 +02001140 result = collect(generation, &collected, &uncollectable, 0);
Kristján Valur Jónsson69c63522012-04-15 11:41:32 +00001141 invoke_gc_callback("stop", generation, collected, uncollectable);
1142 return result;
1143}
1144
Neal Norwitz7b216c52006-03-04 20:01:53 +00001145static Py_ssize_t
Jeremy Hyltonc5007aa2000-06-30 05:02:53 +00001146collect_generations(void)
1147{
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001148 int i;
1149 Py_ssize_t n = 0;
Jeremy Hyltonc5007aa2000-06-30 05:02:53 +00001150
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001151 /* Find the oldest generation (highest numbered) where the count
1152 * exceeds the threshold. Objects in the that generation and
1153 * generations younger than it will be collected. */
1154 for (i = NUM_GENERATIONS-1; i >= 0; i--) {
1155 if (generations[i].count > generations[i].threshold) {
1156 /* Avoid quadratic performance degradation in number
1157 of tracked objects. See comments at the beginning
1158 of this file, and issue #4074.
1159 */
1160 if (i == NUM_GENERATIONS - 1
1161 && long_lived_pending < long_lived_total / 4)
1162 continue;
Kristján Valur Jónsson69c63522012-04-15 11:41:32 +00001163 n = collect_with_callback(i);
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001164 break;
1165 }
1166 }
1167 return n;
Jeremy Hyltonc5007aa2000-06-30 05:02:53 +00001168}
1169
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00001170PyDoc_STRVAR(gc_enable__doc__,
Vladimir Marangozovf9d20c32000-08-06 22:45:31 +00001171"enable() -> None\n"
1172"\n"
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00001173"Enable automatic garbage collection.\n");
Jeremy Hyltonc5007aa2000-06-30 05:02:53 +00001174
Vladimir Marangozovf9d20c32000-08-06 22:45:31 +00001175static PyObject *
Tim Peters50c61d52003-04-06 01:50:50 +00001176gc_enable(PyObject *self, PyObject *noargs)
Vladimir Marangozovf9d20c32000-08-06 22:45:31 +00001177{
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001178 enabled = 1;
1179 Py_INCREF(Py_None);
1180 return Py_None;
Vladimir Marangozovf9d20c32000-08-06 22:45:31 +00001181}
1182
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00001183PyDoc_STRVAR(gc_disable__doc__,
Vladimir Marangozovf9d20c32000-08-06 22:45:31 +00001184"disable() -> None\n"
1185"\n"
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00001186"Disable automatic garbage collection.\n");
Vladimir Marangozovf9d20c32000-08-06 22:45:31 +00001187
1188static PyObject *
Tim Peters50c61d52003-04-06 01:50:50 +00001189gc_disable(PyObject *self, PyObject *noargs)
Vladimir Marangozovf9d20c32000-08-06 22:45:31 +00001190{
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001191 enabled = 0;
1192 Py_INCREF(Py_None);
1193 return Py_None;
Vladimir Marangozovf9d20c32000-08-06 22:45:31 +00001194}
1195
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00001196PyDoc_STRVAR(gc_isenabled__doc__,
Vladimir Marangozovf9d20c32000-08-06 22:45:31 +00001197"isenabled() -> status\n"
1198"\n"
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00001199"Returns true if automatic garbage collection is enabled.\n");
Vladimir Marangozovf9d20c32000-08-06 22:45:31 +00001200
1201static PyObject *
Tim Peters50c61d52003-04-06 01:50:50 +00001202gc_isenabled(PyObject *self, PyObject *noargs)
Vladimir Marangozovf9d20c32000-08-06 22:45:31 +00001203{
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001204 return PyBool_FromLong((long)enabled);
Vladimir Marangozovf9d20c32000-08-06 22:45:31 +00001205}
1206
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00001207PyDoc_STRVAR(gc_collect__doc__,
Barry Warsawd3c38ff2006-03-07 09:46:03 +00001208"collect([generation]) -> n\n"
Jeremy Hyltonc5007aa2000-06-30 05:02:53 +00001209"\n"
Barry Warsawd3c38ff2006-03-07 09:46:03 +00001210"With no arguments, run a full collection. The optional argument\n"
1211"may be an integer specifying which generation to collect. A ValueError\n"
1212"is raised if the generation number is invalid.\n\n"
1213"The number of unreachable objects is returned.\n");
Jeremy Hyltonc5007aa2000-06-30 05:02:53 +00001214
1215static PyObject *
Barry Warsawd3c38ff2006-03-07 09:46:03 +00001216gc_collect(PyObject *self, PyObject *args, PyObject *kws)
Jeremy Hyltonc5007aa2000-06-30 05:02:53 +00001217{
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001218 static char *keywords[] = {"generation", NULL};
1219 int genarg = NUM_GENERATIONS - 1;
1220 Py_ssize_t n;
Jeremy Hyltonc5007aa2000-06-30 05:02:53 +00001221
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001222 if (!PyArg_ParseTupleAndKeywords(args, kws, "|i", keywords, &genarg))
1223 return NULL;
Barry Warsawd3c38ff2006-03-07 09:46:03 +00001224
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001225 else if (genarg < 0 || genarg >= NUM_GENERATIONS) {
1226 PyErr_SetString(PyExc_ValueError, "invalid generation");
1227 return NULL;
1228 }
Barry Warsawd3c38ff2006-03-07 09:46:03 +00001229
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001230 if (collecting)
1231 n = 0; /* already collecting, don't do anything */
1232 else {
1233 collecting = 1;
Kristján Valur Jónsson69c63522012-04-15 11:41:32 +00001234 n = collect_with_callback(genarg);
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001235 collecting = 0;
1236 }
Jeremy Hyltonc5007aa2000-06-30 05:02:53 +00001237
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001238 return PyLong_FromSsize_t(n);
Jeremy Hyltonc5007aa2000-06-30 05:02:53 +00001239}
1240
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00001241PyDoc_STRVAR(gc_set_debug__doc__,
Jeremy Hyltonc5007aa2000-06-30 05:02:53 +00001242"set_debug(flags) -> None\n"
1243"\n"
1244"Set the garbage collection debugging flags. Debugging information is\n"
1245"written to sys.stderr.\n"
1246"\n"
1247"flags is an integer and can have the following bits turned on:\n"
1248"\n"
1249" DEBUG_STATS - Print statistics during collection.\n"
1250" DEBUG_COLLECTABLE - Print collectable objects found.\n"
1251" DEBUG_UNCOLLECTABLE - Print unreachable but uncollectable objects found.\n"
Neil Schemenauer544de1e2000-09-22 15:22:38 +00001252" DEBUG_SAVEALL - Save objects to gc.garbage rather than freeing them.\n"
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00001253" DEBUG_LEAK - Debug leaking programs (everything but STATS).\n");
Jeremy Hyltonc5007aa2000-06-30 05:02:53 +00001254
1255static PyObject *
Vladimir Marangozovf9d20c32000-08-06 22:45:31 +00001256gc_set_debug(PyObject *self, PyObject *args)
Jeremy Hyltonc5007aa2000-06-30 05:02:53 +00001257{
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001258 if (!PyArg_ParseTuple(args, "i:set_debug", &debug))
1259 return NULL;
Jeremy Hyltonc5007aa2000-06-30 05:02:53 +00001260
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001261 Py_INCREF(Py_None);
1262 return Py_None;
Jeremy Hyltonc5007aa2000-06-30 05:02:53 +00001263}
1264
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00001265PyDoc_STRVAR(gc_get_debug__doc__,
Jeremy Hyltonc5007aa2000-06-30 05:02:53 +00001266"get_debug() -> flags\n"
1267"\n"
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00001268"Get the garbage collection debugging flags.\n");
Jeremy Hyltonc5007aa2000-06-30 05:02:53 +00001269
1270static PyObject *
Tim Peters50c61d52003-04-06 01:50:50 +00001271gc_get_debug(PyObject *self, PyObject *noargs)
Jeremy Hyltonc5007aa2000-06-30 05:02:53 +00001272{
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001273 return Py_BuildValue("i", debug);
Jeremy Hyltonc5007aa2000-06-30 05:02:53 +00001274}
1275
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00001276PyDoc_STRVAR(gc_set_thresh__doc__,
Neal Norwitz2a47c0f2002-01-29 00:53:41 +00001277"set_threshold(threshold0, [threshold1, threshold2]) -> None\n"
Jeremy Hyltonc5007aa2000-06-30 05:02:53 +00001278"\n"
1279"Sets the collection thresholds. Setting threshold0 to zero disables\n"
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00001280"collection.\n");
Jeremy Hyltonc5007aa2000-06-30 05:02:53 +00001281
1282static PyObject *
Vladimir Marangozovf9d20c32000-08-06 22:45:31 +00001283gc_set_thresh(PyObject *self, PyObject *args)
Jeremy Hyltonc5007aa2000-06-30 05:02:53 +00001284{
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001285 int i;
1286 if (!PyArg_ParseTuple(args, "i|ii:set_threshold",
1287 &generations[0].threshold,
1288 &generations[1].threshold,
1289 &generations[2].threshold))
1290 return NULL;
1291 for (i = 2; i < NUM_GENERATIONS; i++) {
1292 /* generations higher than 2 get the same threshold */
1293 generations[i].threshold = generations[2].threshold;
1294 }
Jeremy Hyltonc5007aa2000-06-30 05:02:53 +00001295
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001296 Py_INCREF(Py_None);
1297 return Py_None;
Jeremy Hyltonc5007aa2000-06-30 05:02:53 +00001298}
1299
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00001300PyDoc_STRVAR(gc_get_thresh__doc__,
Jeremy Hyltonc5007aa2000-06-30 05:02:53 +00001301"get_threshold() -> (threshold0, threshold1, threshold2)\n"
1302"\n"
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00001303"Return the current collection thresholds\n");
Jeremy Hyltonc5007aa2000-06-30 05:02:53 +00001304
1305static PyObject *
Tim Peters50c61d52003-04-06 01:50:50 +00001306gc_get_thresh(PyObject *self, PyObject *noargs)
Jeremy Hyltonc5007aa2000-06-30 05:02:53 +00001307{
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001308 return Py_BuildValue("(iii)",
1309 generations[0].threshold,
1310 generations[1].threshold,
1311 generations[2].threshold);
Jeremy Hyltonc5007aa2000-06-30 05:02:53 +00001312}
1313
Barry Warsawd3c38ff2006-03-07 09:46:03 +00001314PyDoc_STRVAR(gc_get_count__doc__,
1315"get_count() -> (count0, count1, count2)\n"
1316"\n"
1317"Return the current collection counts\n");
1318
1319static PyObject *
1320gc_get_count(PyObject *self, PyObject *noargs)
1321{
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001322 return Py_BuildValue("(iii)",
1323 generations[0].count,
1324 generations[1].count,
1325 generations[2].count);
Barry Warsawd3c38ff2006-03-07 09:46:03 +00001326}
1327
Neil Schemenauer48c70342001-08-09 15:38:31 +00001328static int
Martin v. Löwis560da622001-11-24 09:24:51 +00001329referrersvisit(PyObject* obj, PyObject *objs)
Neil Schemenauer48c70342001-08-09 15:38:31 +00001330{
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001331 Py_ssize_t i;
1332 for (i = 0; i < PyTuple_GET_SIZE(objs); i++)
1333 if (PyTuple_GET_ITEM(objs, i) == obj)
1334 return 1;
1335 return 0;
Neil Schemenauer48c70342001-08-09 15:38:31 +00001336}
1337
Neil Schemenauer17e7be62001-08-10 14:46:47 +00001338static int
Martin v. Löwis560da622001-11-24 09:24:51 +00001339gc_referrers_for(PyObject *objs, PyGC_Head *list, PyObject *resultlist)
Neil Schemenauer48c70342001-08-09 15:38:31 +00001340{
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001341 PyGC_Head *gc;
1342 PyObject *obj;
1343 traverseproc traverse;
1344 for (gc = list->gc.gc_next; gc != list; gc = gc->gc.gc_next) {
1345 obj = FROM_GC(gc);
1346 traverse = Py_TYPE(obj)->tp_traverse;
1347 if (obj == objs || obj == resultlist)
1348 continue;
1349 if (traverse(obj, (visitproc)referrersvisit, objs)) {
1350 if (PyList_Append(resultlist, obj) < 0)
1351 return 0; /* error */
1352 }
1353 }
1354 return 1; /* no error */
Neil Schemenauer48c70342001-08-09 15:38:31 +00001355}
1356
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00001357PyDoc_STRVAR(gc_get_referrers__doc__,
Martin v. Löwis560da622001-11-24 09:24:51 +00001358"get_referrers(*objs) -> list\n\
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00001359Return the list of objects that directly refer to any of objs.");
Neil Schemenauer48c70342001-08-09 15:38:31 +00001360
Neil Schemenauer17e7be62001-08-10 14:46:47 +00001361static PyObject *
Martin v. Löwis560da622001-11-24 09:24:51 +00001362gc_get_referrers(PyObject *self, PyObject *args)
Neil Schemenauer48c70342001-08-09 15:38:31 +00001363{
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001364 int i;
1365 PyObject *result = PyList_New(0);
1366 if (!result) return NULL;
Thomas Wouters49fd7fa2006-04-21 10:40:58 +00001367
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001368 for (i = 0; i < NUM_GENERATIONS; i++) {
1369 if (!(gc_referrers_for(args, GEN_HEAD(i), result))) {
1370 Py_DECREF(result);
1371 return NULL;
1372 }
1373 }
1374 return result;
Neil Schemenauer48c70342001-08-09 15:38:31 +00001375}
Jeremy Hyltonc5007aa2000-06-30 05:02:53 +00001376
Tim Peters0f81ab62003-04-08 16:39:48 +00001377/* Append obj to list; return true if error (out of memory), false if OK. */
Jeremy Hylton5bd378b2003-04-03 16:28:38 +00001378static int
Tim Peters730f5532003-04-08 17:17:17 +00001379referentsvisit(PyObject *obj, PyObject *list)
Jeremy Hylton5bd378b2003-04-03 16:28:38 +00001380{
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001381 return PyList_Append(list, obj) < 0;
Jeremy Hylton5bd378b2003-04-03 16:28:38 +00001382}
1383
Tim Peters730f5532003-04-08 17:17:17 +00001384PyDoc_STRVAR(gc_get_referents__doc__,
1385"get_referents(*objs) -> list\n\
Jeremy Hylton059b0942003-04-03 16:29:13 +00001386Return the list of objects that are directly referred to by objs.");
Jeremy Hylton5bd378b2003-04-03 16:28:38 +00001387
1388static PyObject *
Tim Peters730f5532003-04-08 17:17:17 +00001389gc_get_referents(PyObject *self, PyObject *args)
Jeremy Hylton5bd378b2003-04-03 16:28:38 +00001390{
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001391 Py_ssize_t i;
1392 PyObject *result = PyList_New(0);
Tim Peters0f81ab62003-04-08 16:39:48 +00001393
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001394 if (result == NULL)
1395 return NULL;
Tim Peters0f81ab62003-04-08 16:39:48 +00001396
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001397 for (i = 0; i < PyTuple_GET_SIZE(args); i++) {
1398 traverseproc traverse;
1399 PyObject *obj = PyTuple_GET_ITEM(args, i);
Tim Peters0f81ab62003-04-08 16:39:48 +00001400
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001401 if (! PyObject_IS_GC(obj))
1402 continue;
1403 traverse = Py_TYPE(obj)->tp_traverse;
1404 if (! traverse)
1405 continue;
1406 if (traverse(obj, (visitproc)referentsvisit, result)) {
1407 Py_DECREF(result);
1408 return NULL;
1409 }
1410 }
1411 return result;
Jeremy Hylton5bd378b2003-04-03 16:28:38 +00001412}
1413
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00001414PyDoc_STRVAR(gc_get_objects__doc__,
Neil Schemenauerc7c8d8e2001-08-09 15:58:59 +00001415"get_objects() -> [...]\n"
1416"\n"
1417"Return a list of objects tracked by the collector (excluding the list\n"
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00001418"returned).\n");
Neil Schemenauerc7c8d8e2001-08-09 15:58:59 +00001419
Neil Schemenauerc7c8d8e2001-08-09 15:58:59 +00001420static PyObject *
Tim Peters50c61d52003-04-06 01:50:50 +00001421gc_get_objects(PyObject *self, PyObject *noargs)
Neil Schemenauerc7c8d8e2001-08-09 15:58:59 +00001422{
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001423 int i;
1424 PyObject* result;
Neil Schemenauerc7c8d8e2001-08-09 15:58:59 +00001425
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001426 result = PyList_New(0);
1427 if (result == NULL)
1428 return NULL;
1429 for (i = 0; i < NUM_GENERATIONS; i++) {
1430 if (append_objects(result, GEN_HEAD(i))) {
1431 Py_DECREF(result);
1432 return NULL;
1433 }
1434 }
1435 return result;
Neil Schemenauerc7c8d8e2001-08-09 15:58:59 +00001436}
1437
Antoine Pitroud4156c12012-10-30 22:43:19 +01001438PyDoc_STRVAR(gc_get_stats__doc__,
1439"get_stats() -> [...]\n"
1440"\n"
1441"Return a list of dictionaries containing per-generation statistics.\n");
1442
1443static PyObject *
1444gc_get_stats(PyObject *self, PyObject *noargs)
1445{
1446 int i;
1447 PyObject *result;
1448 struct gc_generation_stats stats[NUM_GENERATIONS], *st;
1449
1450 /* To get consistent values despite allocations while constructing
1451 the result list, we use a snapshot of the running stats. */
1452 for (i = 0; i < NUM_GENERATIONS; i++) {
1453 stats[i] = generation_stats[i];
1454 }
1455
1456 result = PyList_New(0);
1457 if (result == NULL)
1458 return NULL;
1459
1460 for (i = 0; i < NUM_GENERATIONS; i++) {
1461 PyObject *dict;
1462 st = &stats[i];
1463 dict = Py_BuildValue("{snsnsn}",
1464 "collections", st->collections,
1465 "collected", st->collected,
1466 "uncollectable", st->uncollectable
1467 );
1468 if (dict == NULL)
1469 goto error;
1470 if (PyList_Append(result, dict)) {
1471 Py_DECREF(dict);
1472 goto error;
1473 }
1474 Py_DECREF(dict);
1475 }
1476 return result;
1477
1478error:
1479 Py_XDECREF(result);
1480 return NULL;
1481}
1482
1483
Antoine Pitrou3a652b12009-03-23 18:52:06 +00001484PyDoc_STRVAR(gc_is_tracked__doc__,
1485"is_tracked(obj) -> bool\n"
1486"\n"
1487"Returns true if the object is tracked by the garbage collector.\n"
1488"Simple atomic objects will return false.\n"
1489);
1490
1491static PyObject *
1492gc_is_tracked(PyObject *self, PyObject *obj)
1493{
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001494 PyObject *result;
1495
1496 if (PyObject_IS_GC(obj) && IS_TRACKED(obj))
1497 result = Py_True;
1498 else
1499 result = Py_False;
1500 Py_INCREF(result);
1501 return result;
Antoine Pitrou3a652b12009-03-23 18:52:06 +00001502}
1503
Neil Schemenauerc7c8d8e2001-08-09 15:58:59 +00001504
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00001505PyDoc_STRVAR(gc__doc__,
Jeremy Hyltonc5007aa2000-06-30 05:02:53 +00001506"This module provides access to the garbage collector for reference cycles.\n"
1507"\n"
Vladimir Marangozovf9d20c32000-08-06 22:45:31 +00001508"enable() -- Enable automatic garbage collection.\n"
1509"disable() -- Disable automatic garbage collection.\n"
1510"isenabled() -- Returns true if automatic collection is enabled.\n"
Jeremy Hyltonc5007aa2000-06-30 05:02:53 +00001511"collect() -- Do a full collection right now.\n"
Thomas Wouters89f507f2006-12-13 04:49:30 +00001512"get_count() -- Return the current collection counts.\n"
R David Murray0e814632013-12-26 15:11:28 -05001513"get_stats() -- Return list of dictionaries containing per-generation stats.\n"
Jeremy Hyltonc5007aa2000-06-30 05:02:53 +00001514"set_debug() -- Set debugging flags.\n"
1515"get_debug() -- Get debugging flags.\n"
1516"set_threshold() -- Set the collection thresholds.\n"
1517"get_threshold() -- Return the current the collection thresholds.\n"
Neil Schemenauerc7c8d8e2001-08-09 15:58:59 +00001518"get_objects() -- Return a list of all objects tracked by the collector.\n"
Antoine Pitrou3a652b12009-03-23 18:52:06 +00001519"is_tracked() -- Returns true if a given object is tracked.\n"
Jeremy Hylton5bd378b2003-04-03 16:28:38 +00001520"get_referrers() -- Return the list of objects that refer to an object.\n"
Tim Peters730f5532003-04-08 17:17:17 +00001521"get_referents() -- Return the list of objects that an object refers to.\n");
Jeremy Hyltonc5007aa2000-06-30 05:02:53 +00001522
1523static PyMethodDef GcMethods[] = {
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001524 {"enable", gc_enable, METH_NOARGS, gc_enable__doc__},
1525 {"disable", gc_disable, METH_NOARGS, gc_disable__doc__},
1526 {"isenabled", gc_isenabled, METH_NOARGS, gc_isenabled__doc__},
1527 {"set_debug", gc_set_debug, METH_VARARGS, gc_set_debug__doc__},
1528 {"get_debug", gc_get_debug, METH_NOARGS, gc_get_debug__doc__},
1529 {"get_count", gc_get_count, METH_NOARGS, gc_get_count__doc__},
1530 {"set_threshold", gc_set_thresh, METH_VARARGS, gc_set_thresh__doc__},
1531 {"get_threshold", gc_get_thresh, METH_NOARGS, gc_get_thresh__doc__},
1532 {"collect", (PyCFunction)gc_collect,
1533 METH_VARARGS | METH_KEYWORDS, gc_collect__doc__},
1534 {"get_objects", gc_get_objects,METH_NOARGS, gc_get_objects__doc__},
Antoine Pitroud4156c12012-10-30 22:43:19 +01001535 {"get_stats", gc_get_stats, METH_NOARGS, gc_get_stats__doc__},
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001536 {"is_tracked", gc_is_tracked, METH_O, gc_is_tracked__doc__},
1537 {"get_referrers", gc_get_referrers, METH_VARARGS,
1538 gc_get_referrers__doc__},
1539 {"get_referents", gc_get_referents, METH_VARARGS,
1540 gc_get_referents__doc__},
1541 {NULL, NULL} /* Sentinel */
Jeremy Hyltonc5007aa2000-06-30 05:02:53 +00001542};
1543
Martin v. Löwis1a214512008-06-11 05:26:20 +00001544static struct PyModuleDef gcmodule = {
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001545 PyModuleDef_HEAD_INIT,
Antoine Pitrou696e0352010-08-08 22:18:46 +00001546 "gc", /* m_name */
1547 gc__doc__, /* m_doc */
1548 -1, /* m_size */
1549 GcMethods, /* m_methods */
1550 NULL, /* m_reload */
1551 NULL, /* m_traverse */
1552 NULL, /* m_clear */
1553 NULL /* m_free */
Martin v. Löwis1a214512008-06-11 05:26:20 +00001554};
1555
Jason Tishler6bc06ec2003-09-04 11:59:50 +00001556PyMODINIT_FUNC
Martin v. Löwis1a214512008-06-11 05:26:20 +00001557PyInit_gc(void)
Jeremy Hyltonc5007aa2000-06-30 05:02:53 +00001558{
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001559 PyObject *m;
Jeremy Hyltonc5007aa2000-06-30 05:02:53 +00001560
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001561 m = PyModule_Create(&gcmodule);
Martin v. Löwis1a214512008-06-11 05:26:20 +00001562
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001563 if (m == NULL)
1564 return NULL;
Tim Peters11558872003-04-06 23:30:52 +00001565
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001566 if (garbage == NULL) {
1567 garbage = PyList_New(0);
1568 if (garbage == NULL)
1569 return NULL;
1570 }
1571 Py_INCREF(garbage);
1572 if (PyModule_AddObject(m, "garbage", garbage) < 0)
1573 return NULL;
Thomas Wouters477c8d52006-05-27 19:21:47 +00001574
Kristján Valur Jónsson69c63522012-04-15 11:41:32 +00001575 if (callbacks == NULL) {
1576 callbacks = PyList_New(0);
1577 if (callbacks == NULL)
1578 return NULL;
1579 }
1580 Py_INCREF(callbacks);
1581 if (PyModule_AddObject(m, "callbacks", callbacks) < 0)
1582 return NULL;
1583
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001584 /* Importing can't be done in collect() because collect()
1585 * can be called via PyGC_Collect() in Py_Finalize().
1586 * This wouldn't be a problem, except that <initialized> is
1587 * reset to 0 before calling collect which trips up
1588 * the import and triggers an assertion.
1589 */
1590 if (tmod == NULL) {
1591 tmod = PyImport_ImportModuleNoBlock("time");
1592 if (tmod == NULL)
1593 PyErr_Clear();
1594 }
Thomas Wouters477c8d52006-05-27 19:21:47 +00001595
Martin v. Löwis1a214512008-06-11 05:26:20 +00001596#define ADD_INT(NAME) if (PyModule_AddIntConstant(m, #NAME, NAME) < 0) return NULL
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001597 ADD_INT(DEBUG_STATS);
1598 ADD_INT(DEBUG_COLLECTABLE);
1599 ADD_INT(DEBUG_UNCOLLECTABLE);
1600 ADD_INT(DEBUG_SAVEALL);
1601 ADD_INT(DEBUG_LEAK);
Tim Peters11558872003-04-06 23:30:52 +00001602#undef ADD_INT
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001603 return m;
Jeremy Hyltonc5007aa2000-06-30 05:02:53 +00001604}
1605
Guido van Rossume13ddc92003-04-17 17:29:22 +00001606/* API to invoke gc.collect() from C */
Neal Norwitz7b216c52006-03-04 20:01:53 +00001607Py_ssize_t
Guido van Rossume13ddc92003-04-17 17:29:22 +00001608PyGC_Collect(void)
1609{
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001610 Py_ssize_t n;
Guido van Rossume13ddc92003-04-17 17:29:22 +00001611
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001612 if (collecting)
1613 n = 0; /* already collecting, don't do anything */
1614 else {
1615 collecting = 1;
Kristján Valur Jónsson69c63522012-04-15 11:41:32 +00001616 n = collect_with_callback(NUM_GENERATIONS - 1);
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001617 collecting = 0;
1618 }
Guido van Rossume13ddc92003-04-17 17:29:22 +00001619
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001620 return n;
Guido van Rossume13ddc92003-04-17 17:29:22 +00001621}
1622
Antoine Pitroufef34e32013-05-19 01:11:58 +02001623Py_ssize_t
1624_PyGC_CollectNoFail(void)
1625{
1626 Py_ssize_t n;
1627
Antoine Pitrouc69c9bc2013-08-15 20:15:15 +02001628 /* Ideally, this function is only called on interpreter shutdown,
1629 and therefore not recursively. Unfortunately, when there are daemon
1630 threads, a daemon thread can start a cyclic garbage collection
1631 during interpreter shutdown (and then never finish it).
1632 See http://bugs.python.org/issue8713#msg195178 for an example.
1633 */
1634 if (collecting)
1635 n = 0;
1636 else {
1637 collecting = 1;
1638 n = collect(NUM_GENERATIONS - 1, NULL, NULL, 1);
1639 collecting = 0;
1640 }
Antoine Pitroufef34e32013-05-19 01:11:58 +02001641 return n;
1642}
Antoine Pitrou5f454a02013-05-06 21:15:57 +02001643
Antoine Pitrou696e0352010-08-08 22:18:46 +00001644void
Antoine Pitrou5f454a02013-05-06 21:15:57 +02001645_PyGC_DumpShutdownStats(void)
Antoine Pitrou696e0352010-08-08 22:18:46 +00001646{
Antoine Pitrou2ed94eb2010-09-14 09:48:39 +00001647 if (!(debug & DEBUG_SAVEALL)
1648 && garbage != NULL && PyList_GET_SIZE(garbage) > 0) {
Georg Brandl08be72d2010-10-24 15:11:22 +00001649 char *message;
1650 if (debug & DEBUG_UNCOLLECTABLE)
Antoine Pitroub5d82042010-11-05 00:05:25 +00001651 message = "gc: %zd uncollectable objects at " \
Georg Brandl08be72d2010-10-24 15:11:22 +00001652 "shutdown";
1653 else
Antoine Pitroub5d82042010-11-05 00:05:25 +00001654 message = "gc: %zd uncollectable objects at " \
Georg Brandl08be72d2010-10-24 15:11:22 +00001655 "shutdown; use gc.set_debug(gc.DEBUG_UNCOLLECTABLE) to list them";
Antoine Pitrou070cb3c2013-05-08 13:23:25 +02001656 /* PyErr_WarnFormat does too many things and we are at shutdown,
1657 the warnings module's dependencies (e.g. linecache) may be gone
1658 already. */
1659 if (PyErr_WarnExplicitFormat(PyExc_ResourceWarning, "gc", 0,
1660 "gc", NULL, message,
1661 PyList_GET_SIZE(garbage)))
Georg Brandl08be72d2010-10-24 15:11:22 +00001662 PyErr_WriteUnraisable(NULL);
Antoine Pitrou696e0352010-08-08 22:18:46 +00001663 if (debug & DEBUG_UNCOLLECTABLE) {
1664 PyObject *repr = NULL, *bytes = NULL;
1665 repr = PyObject_Repr(garbage);
1666 if (!repr || !(bytes = PyUnicode_EncodeFSDefault(repr)))
1667 PyErr_WriteUnraisable(garbage);
1668 else {
1669 PySys_WriteStderr(
Antoine Pitrou070cb3c2013-05-08 13:23:25 +02001670 " %s\n",
Antoine Pitrou696e0352010-08-08 22:18:46 +00001671 PyBytes_AS_STRING(bytes)
1672 );
1673 }
1674 Py_XDECREF(repr);
1675 Py_XDECREF(bytes);
1676 }
Antoine Pitrou696e0352010-08-08 22:18:46 +00001677 }
Antoine Pitrou5f454a02013-05-06 21:15:57 +02001678}
1679
1680void
1681_PyGC_Fini(void)
1682{
Kristján Valur Jónsson69c63522012-04-15 11:41:32 +00001683 Py_CLEAR(callbacks);
Antoine Pitrou257cf2f2013-08-06 20:50:48 +02001684 Py_CLEAR(tmod);
Antoine Pitrou696e0352010-08-08 22:18:46 +00001685}
1686
Neil Schemenauer43411b52001-08-30 00:05:51 +00001687/* for debugging */
Guido van Rossume13ddc92003-04-17 17:29:22 +00001688void
1689_PyGC_Dump(PyGC_Head *g)
Neil Schemenauer43411b52001-08-30 00:05:51 +00001690{
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001691 _PyObject_Dump(FROM_GC(g));
Neil Schemenauer43411b52001-08-30 00:05:51 +00001692}
1693
Neil Schemenauer43411b52001-08-30 00:05:51 +00001694/* extension modules might be compiled with GC support so these
1695 functions must always be available */
1696
Neil Schemenauerfec4eb12002-04-12 02:41:03 +00001697#undef PyObject_GC_Track
1698#undef PyObject_GC_UnTrack
1699#undef PyObject_GC_Del
1700#undef _PyObject_GC_Malloc
1701
Neil Schemenauer43411b52001-08-30 00:05:51 +00001702void
Neil Schemenauerfec4eb12002-04-12 02:41:03 +00001703PyObject_GC_Track(void *op)
Neil Schemenauer43411b52001-08-30 00:05:51 +00001704{
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001705 _PyObject_GC_TRACK(op);
Neil Schemenauer43411b52001-08-30 00:05:51 +00001706}
1707
Neil Schemenauerfec4eb12002-04-12 02:41:03 +00001708void
1709PyObject_GC_UnTrack(void *op)
Neil Schemenauer43411b52001-08-30 00:05:51 +00001710{
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001711 /* Obscure: the Py_TRASHCAN mechanism requires that we be able to
1712 * call PyObject_GC_UnTrack twice on an object.
1713 */
1714 if (IS_TRACKED(op))
1715 _PyObject_GC_UNTRACK(op);
Neil Schemenauer43411b52001-08-30 00:05:51 +00001716}
1717
1718PyObject *
Neil Schemenauerfec4eb12002-04-12 02:41:03 +00001719_PyObject_GC_Malloc(size_t basicsize)
Neil Schemenauer43411b52001-08-30 00:05:51 +00001720{
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001721 PyObject *op;
1722 PyGC_Head *g;
1723 if (basicsize > PY_SSIZE_T_MAX - sizeof(PyGC_Head))
1724 return PyErr_NoMemory();
1725 g = (PyGC_Head *)PyObject_MALLOC(
1726 sizeof(PyGC_Head) + basicsize);
1727 if (g == NULL)
1728 return PyErr_NoMemory();
Antoine Pitrou796564c2013-07-30 19:59:21 +02001729 g->gc.gc_refs = 0;
1730 _PyGCHead_SET_REFS(g, GC_UNTRACKED);
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001731 generations[0].count++; /* number of allocated GC objects */
1732 if (generations[0].count > generations[0].threshold &&
1733 enabled &&
1734 generations[0].threshold &&
1735 !collecting &&
1736 !PyErr_Occurred()) {
1737 collecting = 1;
1738 collect_generations();
1739 collecting = 0;
1740 }
1741 op = FROM_GC(g);
1742 return op;
Neil Schemenauer43411b52001-08-30 00:05:51 +00001743}
1744
1745PyObject *
1746_PyObject_GC_New(PyTypeObject *tp)
1747{
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001748 PyObject *op = _PyObject_GC_Malloc(_PyObject_SIZE(tp));
1749 if (op != NULL)
1750 op = PyObject_INIT(op, tp);
1751 return op;
Neil Schemenauer43411b52001-08-30 00:05:51 +00001752}
1753
1754PyVarObject *
Martin v. Löwis18e16552006-02-15 17:27:45 +00001755_PyObject_GC_NewVar(PyTypeObject *tp, Py_ssize_t nitems)
Neil Schemenauer43411b52001-08-30 00:05:51 +00001756{
Victor Stinner5d1866c2013-07-08 22:17:52 +02001757 size_t size;
1758 PyVarObject *op;
1759
1760 if (nitems < 0) {
1761 PyErr_BadInternalCall();
1762 return NULL;
1763 }
1764 size = _PyObject_VAR_SIZE(tp, nitems);
1765 op = (PyVarObject *) _PyObject_GC_Malloc(size);
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001766 if (op != NULL)
1767 op = PyObject_INIT_VAR(op, tp, nitems);
1768 return op;
Neil Schemenauer43411b52001-08-30 00:05:51 +00001769}
1770
1771PyVarObject *
Martin v. Löwis41290682006-02-16 14:56:14 +00001772_PyObject_GC_Resize(PyVarObject *op, Py_ssize_t nitems)
Neil Schemenauer43411b52001-08-30 00:05:51 +00001773{
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001774 const size_t basicsize = _PyObject_VAR_SIZE(Py_TYPE(op), nitems);
1775 PyGC_Head *g = AS_GC(op);
1776 if (basicsize > PY_SSIZE_T_MAX - sizeof(PyGC_Head))
1777 return (PyVarObject *)PyErr_NoMemory();
1778 g = (PyGC_Head *)PyObject_REALLOC(g, sizeof(PyGC_Head) + basicsize);
1779 if (g == NULL)
1780 return (PyVarObject *)PyErr_NoMemory();
1781 op = (PyVarObject *) FROM_GC(g);
1782 Py_SIZE(op) = nitems;
1783 return op;
Neil Schemenauer43411b52001-08-30 00:05:51 +00001784}
1785
1786void
Neil Schemenauerfec4eb12002-04-12 02:41:03 +00001787PyObject_GC_Del(void *op)
Neil Schemenauer43411b52001-08-30 00:05:51 +00001788{
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001789 PyGC_Head *g = AS_GC(op);
1790 if (IS_TRACKED(op))
1791 gc_list_remove(g);
1792 if (generations[0].count > 0) {
1793 generations[0].count--;
1794 }
1795 PyObject_FREE(g);
Neil Schemenauer43411b52001-08-30 00:05:51 +00001796}