blob: 134f6d168c9616a86d9f4e6d1e26a00a5a42fac9 [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"
Victor Stinner27e2d1f2018-11-01 00:52:28 +010027#include "pycore_context.h"
Victor Stinnerbcda8f12018-11-21 22:27:47 +010028#include "pycore_object.h"
Victor Stinner621cebe2018-11-12 16:53:38 +010029#include "pycore_pymem.h"
30#include "pycore_pystate.h"
Antoine Pitrouf95a1b32010-05-09 15:52:27 +000031#include "frameobject.h" /* for PyFrame_ClearFreeList */
Łukasz Langaa785c872016-09-09 17:37:37 -070032#include "pydtrace.h"
Victor Stinner7181dec2015-03-27 17:47:53 +010033#include "pytime.h" /* for _PyTime_GetMonotonicClock() */
Jeremy Hyltonc5007aa2000-06-30 05:02:53 +000034
Serhiy Storchaka93260282017-02-04 11:19:59 +020035/*[clinic input]
36module gc
37[clinic start generated code]*/
38/*[clinic end generated code: output=da39a3ee5e6b4b0d input=b5c9690ecc842d79]*/
39
INADA Naoki5ac9e6e2018-07-10 17:19:53 +090040#define GC_DEBUG (0) /* Enable more asserts */
41
42#define GC_NEXT _PyGCHead_NEXT
43#define GC_PREV _PyGCHead_PREV
44
45// update_refs() set this bit for all objects in current generation.
46// subtract_refs() and move_unreachable() uses this to distinguish
47// visited object is in GCing or not.
48//
49// move_unreachable() removes this flag from reachable objects.
50// Only unreachable objects have this flag.
51//
52// No objects in interpreter have this flag after GC ends.
53#define PREV_MASK_COLLECTING _PyGC_PREV_MASK_COLLECTING
54
55// Lowest bit of _gc_next is used for UNREACHABLE flag.
56//
57// This flag represents the object is in unreachable list in move_unreachable()
58//
59// Although this flag is used only in move_unreachable(), move_unreachable()
60// doesn't clear this flag to skip unnecessary iteration.
61// move_legacy_finalizers() removes this flag instead.
62// Between them, unreachable list is not normal list and we can not use
63// most gc_list_* functions for it.
64#define NEXT_MASK_UNREACHABLE (1)
65
Victor Stinner626bff82018-10-25 17:31:10 +020066/* Get an object's GC head */
67#define AS_GC(o) ((PyGC_Head *)(o)-1)
68
69/* Get the object given the GC head */
70#define FROM_GC(g) ((PyObject *)(((PyGC_Head *)g)+1))
71
INADA Naoki5ac9e6e2018-07-10 17:19:53 +090072static inline int
73gc_is_collecting(PyGC_Head *g)
74{
75 return (g->_gc_prev & PREV_MASK_COLLECTING) != 0;
76}
77
78static inline void
79gc_clear_collecting(PyGC_Head *g)
80{
81 g->_gc_prev &= ~PREV_MASK_COLLECTING;
82}
83
84static inline Py_ssize_t
85gc_get_refs(PyGC_Head *g)
86{
87 return (Py_ssize_t)(g->_gc_prev >> _PyGC_PREV_SHIFT);
88}
89
90static inline void
91gc_set_refs(PyGC_Head *g, Py_ssize_t refs)
92{
93 g->_gc_prev = (g->_gc_prev & ~_PyGC_PREV_MASK)
94 | ((uintptr_t)(refs) << _PyGC_PREV_SHIFT);
95}
96
97static inline void
98gc_reset_refs(PyGC_Head *g, Py_ssize_t refs)
99{
100 g->_gc_prev = (g->_gc_prev & _PyGC_PREV_MASK_FINALIZED)
101 | PREV_MASK_COLLECTING
102 | ((uintptr_t)(refs) << _PyGC_PREV_SHIFT);
103}
104
105static inline void
106gc_decref(PyGC_Head *g)
107{
Victor Stinner626bff82018-10-25 17:31:10 +0200108 _PyObject_ASSERT_WITH_MSG(FROM_GC(g),
109 gc_get_refs(g) > 0,
110 "refcount is too small");
INADA Naoki5ac9e6e2018-07-10 17:19:53 +0900111 g->_gc_prev -= 1 << _PyGC_PREV_SHIFT;
112}
113
Tim Peters6fc13d92002-07-02 18:12:35 +0000114/* Python string to use if unhandled exception occurs */
Tim Petersbf384c22003-04-06 00:11:39 +0000115static PyObject *gc_str = NULL;
Tim Peters6fc13d92002-07-02 18:12:35 +0000116
Jeremy Hyltonc5007aa2000-06-30 05:02:53 +0000117/* set for debugging information */
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000118#define DEBUG_STATS (1<<0) /* print collection statistics */
119#define DEBUG_COLLECTABLE (1<<1) /* print collectable objects */
120#define DEBUG_UNCOLLECTABLE (1<<2) /* print uncollectable objects */
121#define DEBUG_SAVEALL (1<<5) /* save all garbage in gc.garbage */
122#define DEBUG_LEAK DEBUG_COLLECTABLE | \
123 DEBUG_UNCOLLECTABLE | \
124 DEBUG_SAVEALL
Jeremy Hyltonc5007aa2000-06-30 05:02:53 +0000125
Victor Stinner9db03242019-04-26 02:32:01 +0200126#define GEN_HEAD(state, n) (&(state)->generations[n].head)
Antoine Pitroud4156c12012-10-30 22:43:19 +0100127
Eric Snow2ebc5ce2017-09-07 23:51:28 -0600128void
129_PyGC_Initialize(struct _gc_runtime_state *state)
130{
131 state->enabled = 1; /* automatic collection enabled? */
132
Victor Stinner9db03242019-04-26 02:32:01 +0200133#define _GEN_HEAD(n) GEN_HEAD(state, n)
Eric Snow2ebc5ce2017-09-07 23:51:28 -0600134 struct gc_generation generations[NUM_GENERATIONS] = {
INADA Naoki5ac9e6e2018-07-10 17:19:53 +0900135 /* PyGC_Head, threshold, count */
136 {{(uintptr_t)_GEN_HEAD(0), (uintptr_t)_GEN_HEAD(0)}, 700, 0},
137 {{(uintptr_t)_GEN_HEAD(1), (uintptr_t)_GEN_HEAD(1)}, 10, 0},
138 {{(uintptr_t)_GEN_HEAD(2), (uintptr_t)_GEN_HEAD(2)}, 10, 0},
Eric Snow2ebc5ce2017-09-07 23:51:28 -0600139 };
140 for (int i = 0; i < NUM_GENERATIONS; i++) {
141 state->generations[i] = generations[i];
142 };
Victor Stinner9db03242019-04-26 02:32:01 +0200143 state->generation0 = GEN_HEAD(state, 0);
brainfvckc75edab2017-10-16 12:49:41 -0700144 struct gc_generation permanent_generation = {
INADA Naoki5ac9e6e2018-07-10 17:19:53 +0900145 {(uintptr_t)&state->permanent_generation.head,
146 (uintptr_t)&state->permanent_generation.head}, 0, 0
brainfvckc75edab2017-10-16 12:49:41 -0700147 };
148 state->permanent_generation = permanent_generation;
Eric Snow2ebc5ce2017-09-07 23:51:28 -0600149}
Antoine Pitroud4156c12012-10-30 22:43:19 +0100150
INADA Naoki5ac9e6e2018-07-10 17:19:53 +0900151/*
152_gc_prev values
153---------------
Neil Schemenauer43411b52001-08-30 00:05:51 +0000154
INADA Naoki5ac9e6e2018-07-10 17:19:53 +0900155Between collections, _gc_prev is used for doubly linked list.
Tim Peters6fc13d92002-07-02 18:12:35 +0000156
INADA Naoki5ac9e6e2018-07-10 17:19:53 +0900157Lowest two bits of _gc_prev are used for flags.
158PREV_MASK_COLLECTING is used only while collecting and cleared before GC ends
159or _PyObject_GC_UNTRACK() is called.
Tim Peters6fc13d92002-07-02 18:12:35 +0000160
INADA Naoki5ac9e6e2018-07-10 17:19:53 +0900161During a collection, _gc_prev is temporary used for gc_refs, and the gc list
162is singly linked until _gc_prev is restored.
Tim Peters6fc13d92002-07-02 18:12:35 +0000163
INADA Naoki5ac9e6e2018-07-10 17:19:53 +0900164gc_refs
Tim Peters6fc13d92002-07-02 18:12:35 +0000165 At the start of a collection, update_refs() copies the true refcount
166 to gc_refs, for each object in the generation being collected.
167 subtract_refs() then adjusts gc_refs so that it equals the number of
168 times an object is referenced directly from outside the generation
169 being collected.
Tim Peters6fc13d92002-07-02 18:12:35 +0000170
INADA Naoki5ac9e6e2018-07-10 17:19:53 +0900171PREV_MASK_COLLECTING
172 Objects in generation being collected are marked PREV_MASK_COLLECTING in
173 update_refs().
174
175
176_gc_next values
177---------------
178
179_gc_next takes these values:
180
1810
182 The object is not tracked
183
184!= 0
185 Pointer to the next object in the GC list.
186 Additionally, lowest bit is used temporary for
187 NEXT_MASK_UNREACHABLE flag described below.
188
189NEXT_MASK_UNREACHABLE
Tim Peters6fc13d92002-07-02 18:12:35 +0000190 move_unreachable() then moves objects not reachable (whether directly or
INADA Naoki5ac9e6e2018-07-10 17:19:53 +0900191 indirectly) from outside the generation into an "unreachable" set and
192 set this flag.
Tim Peters6fc13d92002-07-02 18:12:35 +0000193
INADA Naoki5ac9e6e2018-07-10 17:19:53 +0900194 Objects that are found to be reachable have gc_refs set to 1.
195 When this flag is set for the reachable object, the object must be in
196 "unreachable" set.
197 The flag is unset and the object is moved back to "reachable" set.
198
199 move_legacy_finalizers() will remove this flag from "unreachable" set.
Tim Peters6fc13d92002-07-02 18:12:35 +0000200*/
Neil Schemenauera2b11ec2002-05-21 15:53:24 +0000201
Jeremy Hyltonc5007aa2000-06-30 05:02:53 +0000202/*** list functions ***/
203
INADA Naoki5ac9e6e2018-07-10 17:19:53 +0900204static inline void
Jeremy Hyltonc5007aa2000-06-30 05:02:53 +0000205gc_list_init(PyGC_Head *list)
206{
INADA Naoki5ac9e6e2018-07-10 17:19:53 +0900207 // List header must not have flags.
208 // We can assign pointer by simple cast.
209 list->_gc_prev = (uintptr_t)list;
210 list->_gc_next = (uintptr_t)list;
Jeremy Hyltonc5007aa2000-06-30 05:02:53 +0000211}
212
INADA Naoki5ac9e6e2018-07-10 17:19:53 +0900213static inline int
Neil Schemenauer2880ae52002-05-04 05:35:20 +0000214gc_list_is_empty(PyGC_Head *list)
215{
INADA Naoki5ac9e6e2018-07-10 17:19:53 +0900216 return (list->_gc_next == (uintptr_t)list);
Neil Schemenauer2880ae52002-05-04 05:35:20 +0000217}
218
Tim Peterse2d59182004-11-01 01:39:08 +0000219/* Append `node` to `list`. */
INADA Naoki5ac9e6e2018-07-10 17:19:53 +0900220static inline void
Jeremy Hyltonc5007aa2000-06-30 05:02:53 +0000221gc_list_append(PyGC_Head *node, PyGC_Head *list)
222{
INADA Naoki5ac9e6e2018-07-10 17:19:53 +0900223 PyGC_Head *last = (PyGC_Head *)list->_gc_prev;
224
225 // last <-> node
226 _PyGCHead_SET_PREV(node, last);
227 _PyGCHead_SET_NEXT(last, node);
228
229 // node <-> list
230 _PyGCHead_SET_NEXT(node, list);
231 list->_gc_prev = (uintptr_t)node;
Jeremy Hyltonc5007aa2000-06-30 05:02:53 +0000232}
233
Tim Peterse2d59182004-11-01 01:39:08 +0000234/* Remove `node` from the gc list it's currently in. */
INADA Naoki5ac9e6e2018-07-10 17:19:53 +0900235static inline void
Jeremy Hyltonc5007aa2000-06-30 05:02:53 +0000236gc_list_remove(PyGC_Head *node)
237{
INADA Naoki5ac9e6e2018-07-10 17:19:53 +0900238 PyGC_Head *prev = GC_PREV(node);
239 PyGC_Head *next = GC_NEXT(node);
240
241 _PyGCHead_SET_NEXT(prev, next);
242 _PyGCHead_SET_PREV(next, prev);
243
244 node->_gc_next = 0; /* object is not currently tracked */
Jeremy Hyltonc5007aa2000-06-30 05:02:53 +0000245}
246
Tim Peterse2d59182004-11-01 01:39:08 +0000247/* Move `node` from the gc list it's currently in (which is not explicitly
248 * named here) to the end of `list`. This is semantically the same as
249 * gc_list_remove(node) followed by gc_list_append(node, list).
250 */
251static void
252gc_list_move(PyGC_Head *node, PyGC_Head *list)
253{
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000254 /* Unlink from current list. */
INADA Naoki5ac9e6e2018-07-10 17:19:53 +0900255 PyGC_Head *from_prev = GC_PREV(node);
256 PyGC_Head *from_next = GC_NEXT(node);
257 _PyGCHead_SET_NEXT(from_prev, from_next);
258 _PyGCHead_SET_PREV(from_next, from_prev);
259
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000260 /* Relink at end of new list. */
INADA Naoki5ac9e6e2018-07-10 17:19:53 +0900261 // list must not have flags. So we can skip macros.
262 PyGC_Head *to_prev = (PyGC_Head*)list->_gc_prev;
263 _PyGCHead_SET_PREV(node, to_prev);
264 _PyGCHead_SET_NEXT(to_prev, node);
265 list->_gc_prev = (uintptr_t)node;
266 _PyGCHead_SET_NEXT(node, list);
Tim Peterse2d59182004-11-01 01:39:08 +0000267}
268
269/* append list `from` onto list `to`; `from` becomes an empty list */
Jeremy Hyltonc5007aa2000-06-30 05:02:53 +0000270static void
271gc_list_merge(PyGC_Head *from, PyGC_Head *to)
272{
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000273 assert(from != to);
274 if (!gc_list_is_empty(from)) {
INADA Naoki5ac9e6e2018-07-10 17:19:53 +0900275 PyGC_Head *to_tail = GC_PREV(to);
276 PyGC_Head *from_head = GC_NEXT(from);
277 PyGC_Head *from_tail = GC_PREV(from);
278 assert(from_head != from);
279 assert(from_tail != from);
280
281 _PyGCHead_SET_NEXT(to_tail, from_head);
282 _PyGCHead_SET_PREV(from_head, to_tail);
283
284 _PyGCHead_SET_NEXT(from_tail, to);
285 _PyGCHead_SET_PREV(to, from_tail);
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000286 }
287 gc_list_init(from);
Jeremy Hyltonc5007aa2000-06-30 05:02:53 +0000288}
289
Neal Norwitz7b216c52006-03-04 20:01:53 +0000290static Py_ssize_t
Jeremy Hyltonc5007aa2000-06-30 05:02:53 +0000291gc_list_size(PyGC_Head *list)
292{
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000293 PyGC_Head *gc;
294 Py_ssize_t n = 0;
INADA Naoki5ac9e6e2018-07-10 17:19:53 +0900295 for (gc = GC_NEXT(list); gc != list; gc = GC_NEXT(gc)) {
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000296 n++;
297 }
298 return n;
Jeremy Hyltonc5007aa2000-06-30 05:02:53 +0000299}
300
Tim Peters259272b2003-04-06 19:41:39 +0000301/* Append objects in a GC list to a Python list.
302 * Return 0 if all OK, < 0 if error (out of memory for list).
303 */
304static int
305append_objects(PyObject *py_list, PyGC_Head *gc_list)
306{
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000307 PyGC_Head *gc;
INADA Naoki5ac9e6e2018-07-10 17:19:53 +0900308 for (gc = GC_NEXT(gc_list); gc != gc_list; gc = GC_NEXT(gc)) {
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000309 PyObject *op = FROM_GC(gc);
310 if (op != py_list) {
311 if (PyList_Append(py_list, op)) {
312 return -1; /* exception */
313 }
314 }
315 }
316 return 0;
Tim Peters259272b2003-04-06 19:41:39 +0000317}
318
INADA Naoki5ac9e6e2018-07-10 17:19:53 +0900319#if GC_DEBUG
320// validate_list checks list consistency. And it works as document
321// describing when expected_mask is set / unset.
322static void
323validate_list(PyGC_Head *head, uintptr_t expected_mask)
324{
325 PyGC_Head *prev = head;
326 PyGC_Head *gc = GC_NEXT(head);
327 while (gc != head) {
328 assert(GC_NEXT(gc) != NULL);
329 assert(GC_PREV(gc) == prev);
330 assert((gc->_gc_prev & PREV_MASK_COLLECTING) == expected_mask);
331 prev = gc;
332 gc = GC_NEXT(gc);
333 }
334 assert(prev == GC_PREV(head));
335}
336#else
337#define validate_list(x,y) do{}while(0)
338#endif
339
Jeremy Hyltonc5007aa2000-06-30 05:02:53 +0000340/*** end of list stuff ***/
341
342
INADA Naoki5ac9e6e2018-07-10 17:19:53 +0900343/* Set all gc_refs = ob_refcnt. After this, gc_refs is > 0 and
344 * PREV_MASK_COLLECTING bit is set for all objects in containers.
Tim Peters88396172002-06-30 17:56:40 +0000345 */
Jeremy Hyltonc5007aa2000-06-30 05:02:53 +0000346static void
347update_refs(PyGC_Head *containers)
348{
INADA Naoki5ac9e6e2018-07-10 17:19:53 +0900349 PyGC_Head *gc = GC_NEXT(containers);
350 for (; gc != containers; gc = GC_NEXT(gc)) {
351 gc_reset_refs(gc, Py_REFCNT(FROM_GC(gc)));
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000352 /* Python's cyclic gc should never see an incoming refcount
353 * of 0: if something decref'ed to 0, it should have been
354 * deallocated immediately at that time.
355 * Possible cause (if the assert triggers): a tp_dealloc
356 * routine left a gc-aware object tracked during its teardown
357 * phase, and did something-- or allowed something to happen --
358 * that called back into Python. gc can trigger then, and may
359 * see the still-tracked dying object. Before this assert
360 * was added, such mistakes went on to allow gc to try to
361 * delete the object again. In a debug build, that caused
362 * a mysterious segfault, when _Py_ForgetReference tried
363 * to remove the object from the doubly-linked list of all
364 * objects a second time. In a release build, an actual
365 * double deallocation occurred, which leads to corruption
366 * of the allocator's internal bookkeeping pointers. That's
367 * so serious that maybe this should be a release-build
368 * check instead of an assert?
369 */
Victor Stinnera4b2bc72018-10-26 18:00:13 +0200370 _PyObject_ASSERT(FROM_GC(gc), gc_get_refs(gc) != 0);
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000371 }
Jeremy Hyltonc5007aa2000-06-30 05:02:53 +0000372}
373
Tim Peters19b74c72002-07-01 03:52:19 +0000374/* A traversal callback for subtract_refs. */
Jeremy Hyltonc5007aa2000-06-30 05:02:53 +0000375static int
376visit_decref(PyObject *op, void *data)
377{
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000378 assert(op != NULL);
379 if (PyObject_IS_GC(op)) {
380 PyGC_Head *gc = AS_GC(op);
381 /* We're only interested in gc_refs for objects in the
382 * generation being collected, which can be recognized
383 * because only they have positive gc_refs.
384 */
INADA Naoki5ac9e6e2018-07-10 17:19:53 +0900385 if (gc_is_collecting(gc)) {
386 gc_decref(gc);
387 }
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000388 }
389 return 0;
Jeremy Hyltonc5007aa2000-06-30 05:02:53 +0000390}
391
Tim Peters19b74c72002-07-01 03:52:19 +0000392/* Subtract internal references from gc_refs. After this, gc_refs is >= 0
393 * for all objects in containers, and is GC_REACHABLE for all tracked gc
394 * objects not in containers. The ones with gc_refs > 0 are directly
395 * reachable from outside containers, and so can't be collected.
396 */
Jeremy Hyltonc5007aa2000-06-30 05:02:53 +0000397static void
398subtract_refs(PyGC_Head *containers)
399{
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000400 traverseproc traverse;
INADA Naoki5ac9e6e2018-07-10 17:19:53 +0900401 PyGC_Head *gc = GC_NEXT(containers);
402 for (; gc != containers; gc = GC_NEXT(gc)) {
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000403 traverse = Py_TYPE(FROM_GC(gc))->tp_traverse;
404 (void) traverse(FROM_GC(gc),
405 (visitproc)visit_decref,
406 NULL);
407 }
Jeremy Hyltonc5007aa2000-06-30 05:02:53 +0000408}
409
Tim Peters19b74c72002-07-01 03:52:19 +0000410/* A traversal callback for move_unreachable. */
Jeremy Hyltonc5007aa2000-06-30 05:02:53 +0000411static int
Tim Peters19b74c72002-07-01 03:52:19 +0000412visit_reachable(PyObject *op, PyGC_Head *reachable)
Jeremy Hyltonc5007aa2000-06-30 05:02:53 +0000413{
INADA Naoki5ac9e6e2018-07-10 17:19:53 +0900414 if (!PyObject_IS_GC(op)) {
415 return 0;
416 }
Tim Peters19b74c72002-07-01 03:52:19 +0000417
INADA Naoki5ac9e6e2018-07-10 17:19:53 +0900418 PyGC_Head *gc = AS_GC(op);
419 const Py_ssize_t gc_refs = gc_get_refs(gc);
420
421 // Ignore untracked objects and objects in other generation.
422 if (gc->_gc_next == 0 || !gc_is_collecting(gc)) {
423 return 0;
424 }
425
426 if (gc->_gc_next & NEXT_MASK_UNREACHABLE) {
427 /* This had gc_refs = 0 when move_unreachable got
428 * to it, but turns out it's reachable after all.
429 * Move it back to move_unreachable's 'young' list,
430 * and move_unreachable will eventually get to it
431 * again.
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000432 */
INADA Naoki5ac9e6e2018-07-10 17:19:53 +0900433 // Manually unlink gc from unreachable list because
434 PyGC_Head *prev = GC_PREV(gc);
435 PyGC_Head *next = (PyGC_Head*)(gc->_gc_next & ~NEXT_MASK_UNREACHABLE);
Victor Stinnera4b2bc72018-10-26 18:00:13 +0200436 _PyObject_ASSERT(FROM_GC(prev),
437 prev->_gc_next & NEXT_MASK_UNREACHABLE);
438 _PyObject_ASSERT(FROM_GC(next),
439 next->_gc_next & NEXT_MASK_UNREACHABLE);
INADA Naoki5ac9e6e2018-07-10 17:19:53 +0900440 prev->_gc_next = gc->_gc_next; // copy NEXT_MASK_UNREACHABLE
441 _PyGCHead_SET_PREV(next, prev);
442
443 gc_list_append(gc, reachable);
444 gc_set_refs(gc, 1);
445 }
446 else if (gc_refs == 0) {
447 /* This is in move_unreachable's 'young' list, but
448 * the traversal hasn't yet gotten to it. All
449 * we need to do is tell move_unreachable that it's
450 * reachable.
451 */
452 gc_set_refs(gc, 1);
453 }
454 /* Else there's nothing to do.
455 * If gc_refs > 0, it must be in move_unreachable's 'young'
456 * list, and move_unreachable will eventually get to it.
457 */
458 else {
Victor Stinnera4b2bc72018-10-26 18:00:13 +0200459 _PyObject_ASSERT_WITH_MSG(op, gc_refs > 0, "refcount is too small");
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000460 }
461 return 0;
Jeremy Hyltonc5007aa2000-06-30 05:02:53 +0000462}
463
Tim Peters19b74c72002-07-01 03:52:19 +0000464/* Move the unreachable objects from young to unreachable. After this,
INADA Naoki5ac9e6e2018-07-10 17:19:53 +0900465 * all objects in young don't have PREV_MASK_COLLECTING flag and
466 * unreachable have the flag.
Tim Peters19b74c72002-07-01 03:52:19 +0000467 * All objects in young after this are directly or indirectly reachable
468 * from outside the original young; and all objects in unreachable are
469 * not.
INADA Naoki5ac9e6e2018-07-10 17:19:53 +0900470 *
471 * This function restores _gc_prev pointer. young and unreachable are
472 * doubly linked list after this function.
473 * But _gc_next in unreachable list has NEXT_MASK_UNREACHABLE flag.
474 * So we can not gc_list_* functions for unreachable until we remove the flag.
Tim Peters88396172002-06-30 17:56:40 +0000475 */
Jeremy Hyltonc5007aa2000-06-30 05:02:53 +0000476static void
Tim Peters19b74c72002-07-01 03:52:19 +0000477move_unreachable(PyGC_Head *young, PyGC_Head *unreachable)
Jeremy Hyltonc5007aa2000-06-30 05:02:53 +0000478{
INADA Naoki5ac9e6e2018-07-10 17:19:53 +0900479 // previous elem in the young list, used for restore gc_prev.
480 PyGC_Head *prev = young;
481 PyGC_Head *gc = GC_NEXT(young);
Tim Peters19b74c72002-07-01 03:52:19 +0000482
INADA Naoki5ac9e6e2018-07-10 17:19:53 +0900483 /* Invariants: all objects "to the left" of us in young are reachable
484 * (directly or indirectly) from outside the young list as it was at entry.
485 *
486 * All other objects from the original young "to the left" of us are in
487 * unreachable now, and have NEXT_MASK_UNREACHABLE. All objects to the
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000488 * left of us in 'young' now have been scanned, and no objects here
489 * or to the right have been scanned yet.
490 */
Tim Peters19b74c72002-07-01 03:52:19 +0000491
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000492 while (gc != young) {
INADA Naoki5ac9e6e2018-07-10 17:19:53 +0900493 if (gc_get_refs(gc)) {
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000494 /* gc is definitely reachable from outside the
495 * original 'young'. Mark it as such, and traverse
496 * its pointers to find any other objects that may
497 * be directly reachable from it. Note that the
498 * call to tp_traverse may append objects to young,
499 * so we have to wait until it returns to determine
500 * the next object to visit.
501 */
502 PyObject *op = FROM_GC(gc);
503 traverseproc traverse = Py_TYPE(op)->tp_traverse;
Victor Stinnera4b2bc72018-10-26 18:00:13 +0200504 _PyObject_ASSERT_WITH_MSG(op, gc_get_refs(gc) > 0,
505 "refcount is too small");
INADA Naoki5ac9e6e2018-07-10 17:19:53 +0900506 // NOTE: visit_reachable may change gc->_gc_next when
507 // young->_gc_prev == gc. Don't do gc = GC_NEXT(gc) before!
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000508 (void) traverse(op,
INADA Naoki5ac9e6e2018-07-10 17:19:53 +0900509 (visitproc)visit_reachable,
510 (void *)young);
511 // relink gc_prev to prev element.
512 _PyGCHead_SET_PREV(gc, prev);
Quan Tian3bd0d622018-10-20 05:30:03 +0800513 // gc is not COLLECTING state after here.
INADA Naoki5ac9e6e2018-07-10 17:19:53 +0900514 gc_clear_collecting(gc);
515 prev = gc;
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000516 }
517 else {
518 /* This *may* be unreachable. To make progress,
519 * assume it is. gc isn't directly reachable from
520 * any object we've already traversed, but may be
521 * reachable from an object we haven't gotten to yet.
522 * visit_reachable will eventually move gc back into
523 * young if that's so, and we'll see it again.
524 */
INADA Naoki5ac9e6e2018-07-10 17:19:53 +0900525 // Move gc to unreachable.
526 // No need to gc->next->prev = prev because it is single linked.
527 prev->_gc_next = gc->_gc_next;
528
529 // We can't use gc_list_append() here because we use
530 // NEXT_MASK_UNREACHABLE here.
531 PyGC_Head *last = GC_PREV(unreachable);
532 // NOTE: Since all objects in unreachable set has
533 // NEXT_MASK_UNREACHABLE flag, we set it unconditionally.
534 // But this may set the flat to unreachable too.
535 // move_legacy_finalizers() should care about it.
536 last->_gc_next = (NEXT_MASK_UNREACHABLE | (uintptr_t)gc);
537 _PyGCHead_SET_PREV(gc, last);
538 gc->_gc_next = (NEXT_MASK_UNREACHABLE | (uintptr_t)unreachable);
539 unreachable->_gc_prev = (uintptr_t)gc;
540 }
541 gc = (PyGC_Head*)prev->_gc_next;
542 }
543 // young->_gc_prev must be last element remained in the list.
544 young->_gc_prev = (uintptr_t)prev;
545}
546
547static void
548untrack_tuples(PyGC_Head *head)
549{
550 PyGC_Head *next, *gc = GC_NEXT(head);
551 while (gc != head) {
552 PyObject *op = FROM_GC(gc);
553 next = GC_NEXT(gc);
554 if (PyTuple_CheckExact(op)) {
555 _PyTuple_MaybeUntrack(op);
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000556 }
557 gc = next;
558 }
Jeremy Hyltonc5007aa2000-06-30 05:02:53 +0000559}
560
Antoine Pitroue1ad3da2012-05-28 22:22:34 +0200561/* Try to untrack all currently tracked dictionaries */
562static void
563untrack_dicts(PyGC_Head *head)
564{
INADA Naoki5ac9e6e2018-07-10 17:19:53 +0900565 PyGC_Head *next, *gc = GC_NEXT(head);
Antoine Pitroue1ad3da2012-05-28 22:22:34 +0200566 while (gc != head) {
567 PyObject *op = FROM_GC(gc);
INADA Naoki5ac9e6e2018-07-10 17:19:53 +0900568 next = GC_NEXT(gc);
569 if (PyDict_CheckExact(op)) {
Antoine Pitroue1ad3da2012-05-28 22:22:34 +0200570 _PyDict_MaybeUntrack(op);
INADA Naoki5ac9e6e2018-07-10 17:19:53 +0900571 }
Antoine Pitroue1ad3da2012-05-28 22:22:34 +0200572 gc = next;
573 }
574}
575
Antoine Pitrou796564c2013-07-30 19:59:21 +0200576/* Return true if object has a pre-PEP 442 finalization method. */
Neil Schemenauera765c122001-11-01 17:35:23 +0000577static int
Antoine Pitrou796564c2013-07-30 19:59:21 +0200578has_legacy_finalizer(PyObject *op)
Jeremy Hyltonc5007aa2000-06-30 05:02:53 +0000579{
Antoine Pitrou796564c2013-07-30 19:59:21 +0200580 return op->ob_type->tp_del != NULL;
Neil Schemenauera765c122001-11-01 17:35:23 +0000581}
582
Antoine Pitrou796564c2013-07-30 19:59:21 +0200583/* Move the objects in unreachable with tp_del slots into `finalizers`.
INADA Naoki5ac9e6e2018-07-10 17:19:53 +0900584 *
585 * This function also removes NEXT_MASK_UNREACHABLE flag
586 * from _gc_next in unreachable.
Jeremy Hyltonce136e92003-04-04 19:59:06 +0000587 */
Neil Schemenauera765c122001-11-01 17:35:23 +0000588static void
Antoine Pitrou796564c2013-07-30 19:59:21 +0200589move_legacy_finalizers(PyGC_Head *unreachable, PyGC_Head *finalizers)
Neil Schemenauera765c122001-11-01 17:35:23 +0000590{
INADA Naoki5ac9e6e2018-07-10 17:19:53 +0900591 PyGC_Head *gc, *next;
592 unreachable->_gc_next &= ~NEXT_MASK_UNREACHABLE;
Tim Petersf6b80452003-04-07 19:21:15 +0000593
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000594 /* March over unreachable. Move objects with finalizers into
595 * `finalizers`.
596 */
INADA Naoki5ac9e6e2018-07-10 17:19:53 +0900597 for (gc = GC_NEXT(unreachable); gc != unreachable; gc = next) {
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000598 PyObject *op = FROM_GC(gc);
Jeremy Hyltonce136e92003-04-04 19:59:06 +0000599
Victor Stinnera4b2bc72018-10-26 18:00:13 +0200600 _PyObject_ASSERT(op, gc->_gc_next & NEXT_MASK_UNREACHABLE);
INADA Naoki5ac9e6e2018-07-10 17:19:53 +0900601 gc->_gc_next &= ~NEXT_MASK_UNREACHABLE;
602 next = (PyGC_Head*)gc->_gc_next;
Tim Petersf6ae7a42003-04-05 18:40:50 +0000603
Antoine Pitrou796564c2013-07-30 19:59:21 +0200604 if (has_legacy_finalizer(op)) {
INADA Naoki5ac9e6e2018-07-10 17:19:53 +0900605 gc_clear_collecting(gc);
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000606 gc_list_move(gc, finalizers);
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000607 }
608 }
Jeremy Hyltonc5007aa2000-06-30 05:02:53 +0000609}
610
Antoine Pitrou796564c2013-07-30 19:59:21 +0200611/* A traversal callback for move_legacy_finalizer_reachable. */
Tim Peters19b74c72002-07-01 03:52:19 +0000612static int
613visit_move(PyObject *op, PyGC_Head *tolist)
614{
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000615 if (PyObject_IS_GC(op)) {
INADA Naoki5ac9e6e2018-07-10 17:19:53 +0900616 PyGC_Head *gc = AS_GC(op);
617 if (gc_is_collecting(gc)) {
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000618 gc_list_move(gc, tolist);
INADA Naoki5ac9e6e2018-07-10 17:19:53 +0900619 gc_clear_collecting(gc);
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000620 }
621 }
622 return 0;
Tim Peters19b74c72002-07-01 03:52:19 +0000623}
624
625/* Move objects that are reachable from finalizers, from the unreachable set
Tim Petersf6b80452003-04-07 19:21:15 +0000626 * into finalizers set.
Tim Peters19b74c72002-07-01 03:52:19 +0000627 */
Jeremy Hyltonc5007aa2000-06-30 05:02:53 +0000628static void
Antoine Pitrou796564c2013-07-30 19:59:21 +0200629move_legacy_finalizer_reachable(PyGC_Head *finalizers)
Jeremy Hyltonc5007aa2000-06-30 05:02:53 +0000630{
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000631 traverseproc traverse;
INADA Naoki5ac9e6e2018-07-10 17:19:53 +0900632 PyGC_Head *gc = GC_NEXT(finalizers);
633 for (; gc != finalizers; gc = GC_NEXT(gc)) {
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000634 /* Note that the finalizers list may grow during this. */
635 traverse = Py_TYPE(FROM_GC(gc))->tp_traverse;
636 (void) traverse(FROM_GC(gc),
637 (visitproc)visit_move,
638 (void *)finalizers);
639 }
Jeremy Hyltonc5007aa2000-06-30 05:02:53 +0000640}
641
Tim Petersead8b7a2004-10-30 23:09:22 +0000642/* Clear all weakrefs to unreachable objects, and if such a weakref has a
643 * callback, invoke it if necessary. Note that it's possible for such
644 * weakrefs to be outside the unreachable set -- indeed, those are precisely
645 * the weakrefs whose callbacks must be invoked. See gc_weakref.txt for
646 * overview & some details. Some weakrefs with callbacks may be reclaimed
647 * directly by this routine; the number reclaimed is the return value. Other
648 * weakrefs with callbacks may be moved into the `old` generation. Objects
649 * moved into `old` have gc_refs set to GC_REACHABLE; the objects remaining in
650 * unreachable are left at GC_TENTATIVELY_UNREACHABLE. When this returns,
651 * no object in `unreachable` is weakly referenced anymore.
Tim Peters403a2032003-11-20 21:21:46 +0000652 */
653static int
Tim Petersead8b7a2004-10-30 23:09:22 +0000654handle_weakrefs(PyGC_Head *unreachable, PyGC_Head *old)
Tim Peters403a2032003-11-20 21:21:46 +0000655{
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000656 PyGC_Head *gc;
657 PyObject *op; /* generally FROM_GC(gc) */
658 PyWeakReference *wr; /* generally a cast of op */
659 PyGC_Head wrcb_to_call; /* weakrefs with callbacks to call */
660 PyGC_Head *next;
661 int num_freed = 0;
Tim Peters403a2032003-11-20 21:21:46 +0000662
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000663 gc_list_init(&wrcb_to_call);
Tim Peters403a2032003-11-20 21:21:46 +0000664
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000665 /* Clear all weakrefs to the objects in unreachable. If such a weakref
666 * also has a callback, move it into `wrcb_to_call` if the callback
667 * needs to be invoked. Note that we cannot invoke any callbacks until
668 * all weakrefs to unreachable objects are cleared, lest the callback
669 * resurrect an unreachable object via a still-active weakref. We
670 * make another pass over wrcb_to_call, invoking callbacks, after this
671 * pass completes.
672 */
INADA Naoki5ac9e6e2018-07-10 17:19:53 +0900673 for (gc = GC_NEXT(unreachable); gc != unreachable; gc = next) {
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000674 PyWeakReference **wrlist;
Tim Petersead8b7a2004-10-30 23:09:22 +0000675
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000676 op = FROM_GC(gc);
INADA Naoki5ac9e6e2018-07-10 17:19:53 +0900677 next = GC_NEXT(gc);
Tim Petersead8b7a2004-10-30 23:09:22 +0000678
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000679 if (! PyType_SUPPORTS_WEAKREFS(Py_TYPE(op)))
680 continue;
Tim Petersead8b7a2004-10-30 23:09:22 +0000681
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000682 /* It supports weakrefs. Does it have any? */
683 wrlist = (PyWeakReference **)
684 PyObject_GET_WEAKREFS_LISTPTR(op);
Tim Petersead8b7a2004-10-30 23:09:22 +0000685
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000686 /* `op` may have some weakrefs. March over the list, clear
687 * all the weakrefs, and move the weakrefs with callbacks
688 * that must be called into wrcb_to_call.
689 */
690 for (wr = *wrlist; wr != NULL; wr = *wrlist) {
691 PyGC_Head *wrasgc; /* AS_GC(wr) */
Tim Petersead8b7a2004-10-30 23:09:22 +0000692
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000693 /* _PyWeakref_ClearRef clears the weakref but leaves
694 * the callback pointer intact. Obscure: it also
695 * changes *wrlist.
696 */
Victor Stinnera4b2bc72018-10-26 18:00:13 +0200697 _PyObject_ASSERT((PyObject *)wr, wr->wr_object == op);
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000698 _PyWeakref_ClearRef(wr);
Victor Stinnera4b2bc72018-10-26 18:00:13 +0200699 _PyObject_ASSERT((PyObject *)wr, wr->wr_object == Py_None);
700 if (wr->wr_callback == NULL) {
701 /* no callback */
702 continue;
703 }
Tim Petersead8b7a2004-10-30 23:09:22 +0000704
Victor Stinnera4b2bc72018-10-26 18:00:13 +0200705 /* Headache time. `op` is going away, and is weakly referenced by
706 * `wr`, which has a callback. Should the callback be invoked? If wr
707 * is also trash, no:
708 *
709 * 1. There's no need to call it. The object and the weakref are
710 * both going away, so it's legitimate to pretend the weakref is
711 * going away first. The user has to ensure a weakref outlives its
712 * referent if they want a guarantee that the wr callback will get
713 * invoked.
714 *
715 * 2. It may be catastrophic to call it. If the callback is also in
716 * cyclic trash (CT), then although the CT is unreachable from
717 * outside the current generation, CT may be reachable from the
718 * callback. Then the callback could resurrect insane objects.
719 *
720 * Since the callback is never needed and may be unsafe in this case,
721 * wr is simply left in the unreachable set. Note that because we
722 * already called _PyWeakref_ClearRef(wr), its callback will never
723 * trigger.
724 *
725 * OTOH, if wr isn't part of CT, we should invoke the callback: the
726 * weakref outlived the trash. Note that since wr isn't CT in this
727 * case, its callback can't be CT either -- wr acted as an external
728 * root to this generation, and therefore its callback did too. So
729 * nothing in CT is reachable from the callback either, so it's hard
730 * to imagine how calling it later could create a problem for us. wr
731 * is moved to wrcb_to_call in this case.
732 */
INADA Naoki5ac9e6e2018-07-10 17:19:53 +0900733 if (gc_is_collecting(AS_GC(wr))) {
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000734 continue;
INADA Naoki5ac9e6e2018-07-10 17:19:53 +0900735 }
Tim Peterscc2a8662004-10-31 22:12:43 +0000736
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000737 /* Create a new reference so that wr can't go away
738 * before we can process it again.
739 */
740 Py_INCREF(wr);
Tim Petersead8b7a2004-10-30 23:09:22 +0000741
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000742 /* Move wr to wrcb_to_call, for the next pass. */
743 wrasgc = AS_GC(wr);
744 assert(wrasgc != next); /* wrasgc is reachable, but
745 next isn't, so they can't
746 be the same */
747 gc_list_move(wrasgc, &wrcb_to_call);
748 }
749 }
Tim Petersead8b7a2004-10-30 23:09:22 +0000750
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000751 /* Invoke the callbacks we decided to honor. It's safe to invoke them
752 * because they can't reference unreachable objects.
753 */
754 while (! gc_list_is_empty(&wrcb_to_call)) {
755 PyObject *temp;
756 PyObject *callback;
Tim Petersead8b7a2004-10-30 23:09:22 +0000757
INADA Naoki5ac9e6e2018-07-10 17:19:53 +0900758 gc = (PyGC_Head*)wrcb_to_call._gc_next;
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000759 op = FROM_GC(gc);
Victor Stinnera4b2bc72018-10-26 18:00:13 +0200760 _PyObject_ASSERT(op, PyWeakref_Check(op));
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000761 wr = (PyWeakReference *)op;
762 callback = wr->wr_callback;
Victor Stinnera4b2bc72018-10-26 18:00:13 +0200763 _PyObject_ASSERT(op, callback != NULL);
Tim Petersead8b7a2004-10-30 23:09:22 +0000764
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000765 /* copy-paste of weakrefobject.c's handle_callback() */
Jeroen Demeyer196a5302019-07-04 12:31:34 +0200766 temp = _PyObject_CallOneArg(callback, (PyObject *)wr);
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000767 if (temp == NULL)
768 PyErr_WriteUnraisable(callback);
769 else
770 Py_DECREF(temp);
Tim Petersead8b7a2004-10-30 23:09:22 +0000771
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000772 /* Give up the reference we created in the first pass. When
773 * op's refcount hits 0 (which it may or may not do right now),
774 * op's tp_dealloc will decref op->wr_callback too. Note
775 * that the refcount probably will hit 0 now, and because this
776 * weakref was reachable to begin with, gc didn't already
777 * add it to its count of freed objects. Example: a reachable
778 * weak value dict maps some key to this reachable weakref.
779 * The callback removes this key->weakref mapping from the
780 * dict, leaving no other references to the weakref (excepting
781 * ours).
782 */
783 Py_DECREF(op);
INADA Naoki5ac9e6e2018-07-10 17:19:53 +0900784 if (wrcb_to_call._gc_next == (uintptr_t)gc) {
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000785 /* object is still alive -- move it */
786 gc_list_move(gc, old);
787 }
INADA Naoki5ac9e6e2018-07-10 17:19:53 +0900788 else {
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000789 ++num_freed;
INADA Naoki5ac9e6e2018-07-10 17:19:53 +0900790 }
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000791 }
Tim Petersead8b7a2004-10-30 23:09:22 +0000792
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000793 return num_freed;
Tim Peters403a2032003-11-20 21:21:46 +0000794}
795
Jeremy Hyltonc5007aa2000-06-30 05:02:53 +0000796static void
Serhiy Storchakaef1585e2015-12-25 20:01:53 +0200797debug_cycle(const char *msg, PyObject *op)
Jeremy Hyltonc5007aa2000-06-30 05:02:53 +0000798{
Victor Stinner499dfcf2011-03-21 13:26:24 +0100799 PySys_FormatStderr("gc: %s <%s %p>\n",
800 msg, Py_TYPE(op)->tp_name, op);
Jeremy Hyltonc5007aa2000-06-30 05:02:53 +0000801}
802
Antoine Pitrou796564c2013-07-30 19:59:21 +0200803/* Handle uncollectable garbage (cycles with tp_del slots, and stuff reachable
Tim Petersbf384c22003-04-06 00:11:39 +0000804 * only from such cycles).
Tim Petersf6b80452003-04-07 19:21:15 +0000805 * If DEBUG_SAVEALL, all objects in finalizers are appended to the module
806 * garbage list (a Python list), else only the objects in finalizers with
807 * __del__ methods are appended to garbage. All objects in finalizers are
808 * merged into the old list regardless.
Tim Petersbf384c22003-04-06 00:11:39 +0000809 */
Serhiy Storchaka301e3cc2018-05-24 15:19:29 +0300810static void
Victor Stinner9db03242019-04-26 02:32:01 +0200811handle_legacy_finalizers(struct _gc_runtime_state *state,
812 PyGC_Head *finalizers, PyGC_Head *old)
Jeremy Hyltonc5007aa2000-06-30 05:02:53 +0000813{
Serhiy Storchakac4653c92018-05-29 18:50:10 +0300814 assert(!PyErr_Occurred());
Victor Stinner9db03242019-04-26 02:32:01 +0200815
816 PyGC_Head *gc = GC_NEXT(finalizers);
817 if (state->garbage == NULL) {
818 state->garbage = PyList_New(0);
819 if (state->garbage == NULL)
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000820 Py_FatalError("gc couldn't create gc.garbage list");
821 }
INADA Naoki5ac9e6e2018-07-10 17:19:53 +0900822 for (; gc != finalizers; gc = GC_NEXT(gc)) {
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000823 PyObject *op = FROM_GC(gc);
Tim Petersf6b80452003-04-07 19:21:15 +0000824
Victor Stinner9db03242019-04-26 02:32:01 +0200825 if ((state->debug & DEBUG_SAVEALL) || has_legacy_finalizer(op)) {
826 if (PyList_Append(state->garbage, op) < 0) {
Serhiy Storchakac4653c92018-05-29 18:50:10 +0300827 PyErr_Clear();
Serhiy Storchaka301e3cc2018-05-24 15:19:29 +0300828 break;
Serhiy Storchakac4653c92018-05-29 18:50:10 +0300829 }
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000830 }
831 }
Tim Petersf6b80452003-04-07 19:21:15 +0000832
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000833 gc_list_merge(finalizers, old);
Jeremy Hyltonc5007aa2000-06-30 05:02:53 +0000834}
835
Tim Peters5fbc7b12014-05-08 17:42:19 -0500836/* Run first-time finalizers (if any) on all the objects in collectable.
837 * Note that this may remove some (or even all) of the objects from the
838 * list, due to refcounts falling to 0.
839 */
Antoine Pitrou796564c2013-07-30 19:59:21 +0200840static void
Tim Peters5fbc7b12014-05-08 17:42:19 -0500841finalize_garbage(PyGC_Head *collectable)
Antoine Pitrou796564c2013-07-30 19:59:21 +0200842{
843 destructor finalize;
Tim Peters5fbc7b12014-05-08 17:42:19 -0500844 PyGC_Head seen;
Antoine Pitrou796564c2013-07-30 19:59:21 +0200845
Tim Peters5fbc7b12014-05-08 17:42:19 -0500846 /* While we're going through the loop, `finalize(op)` may cause op, or
847 * other objects, to be reclaimed via refcounts falling to zero. So
848 * there's little we can rely on about the structure of the input
849 * `collectable` list across iterations. For safety, we always take the
850 * first object in that list and move it to a temporary `seen` list.
851 * If objects vanish from the `collectable` and `seen` lists we don't
852 * care.
853 */
854 gc_list_init(&seen);
855
856 while (!gc_list_is_empty(collectable)) {
INADA Naoki5ac9e6e2018-07-10 17:19:53 +0900857 PyGC_Head *gc = GC_NEXT(collectable);
Antoine Pitrou796564c2013-07-30 19:59:21 +0200858 PyObject *op = FROM_GC(gc);
Tim Peters5fbc7b12014-05-08 17:42:19 -0500859 gc_list_move(gc, &seen);
Antoine Pitrou796564c2013-07-30 19:59:21 +0200860 if (!_PyGCHead_FINALIZED(gc) &&
Tim Peters5fbc7b12014-05-08 17:42:19 -0500861 (finalize = Py_TYPE(op)->tp_finalize) != NULL) {
INADA Naoki5ac9e6e2018-07-10 17:19:53 +0900862 _PyGCHead_SET_FINALIZED(gc);
Antoine Pitrou796564c2013-07-30 19:59:21 +0200863 Py_INCREF(op);
864 finalize(op);
Serhiy Storchakac4653c92018-05-29 18:50:10 +0300865 assert(!PyErr_Occurred());
Antoine Pitrou796564c2013-07-30 19:59:21 +0200866 Py_DECREF(op);
867 }
868 }
Tim Peters5fbc7b12014-05-08 17:42:19 -0500869 gc_list_merge(&seen, collectable);
Antoine Pitrou796564c2013-07-30 19:59:21 +0200870}
871
872/* Walk the collectable list and check that they are really unreachable
873 from the outside (some objects could have been resurrected by a
874 finalizer). */
875static int
876check_garbage(PyGC_Head *collectable)
877{
INADA Naoki5ac9e6e2018-07-10 17:19:53 +0900878 int ret = 0;
Antoine Pitrou796564c2013-07-30 19:59:21 +0200879 PyGC_Head *gc;
INADA Naoki5ac9e6e2018-07-10 17:19:53 +0900880 for (gc = GC_NEXT(collectable); gc != collectable; gc = GC_NEXT(gc)) {
881 // Use gc_refs and break gc_prev again.
882 gc_set_refs(gc, Py_REFCNT(FROM_GC(gc)));
Victor Stinnera4b2bc72018-10-26 18:00:13 +0200883 _PyObject_ASSERT(FROM_GC(gc), gc_get_refs(gc) != 0);
Antoine Pitrou796564c2013-07-30 19:59:21 +0200884 }
885 subtract_refs(collectable);
INADA Naoki5ac9e6e2018-07-10 17:19:53 +0900886 PyGC_Head *prev = collectable;
887 for (gc = GC_NEXT(collectable); gc != collectable; gc = GC_NEXT(gc)) {
Victor Stinnera4b2bc72018-10-26 18:00:13 +0200888 _PyObject_ASSERT_WITH_MSG(FROM_GC(gc),
889 gc_get_refs(gc) >= 0,
890 "refcount is too small");
INADA Naoki5ac9e6e2018-07-10 17:19:53 +0900891 if (gc_get_refs(gc) != 0) {
892 ret = -1;
893 }
894 // Restore gc_prev here.
895 _PyGCHead_SET_PREV(gc, prev);
896 gc_clear_collecting(gc);
897 prev = gc;
Antoine Pitrou796564c2013-07-30 19:59:21 +0200898 }
INADA Naoki5ac9e6e2018-07-10 17:19:53 +0900899 return ret;
Antoine Pitrou796564c2013-07-30 19:59:21 +0200900}
901
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000902/* Break reference cycles by clearing the containers involved. This is
Jeremy Hyltonc5007aa2000-06-30 05:02:53 +0000903 * tricky business as the lists can be changing and we don't know which
Tim Peters19b74c72002-07-01 03:52:19 +0000904 * objects may be freed. It is possible I screwed something up here.
905 */
Jeremy Hyltonc5007aa2000-06-30 05:02:53 +0000906static void
Victor Stinner9db03242019-04-26 02:32:01 +0200907delete_garbage(struct _gc_runtime_state *state,
908 PyGC_Head *collectable, PyGC_Head *old)
Jeremy Hyltonc5007aa2000-06-30 05:02:53 +0000909{
Serhiy Storchakac4653c92018-05-29 18:50:10 +0300910 assert(!PyErr_Occurred());
Victor Stinner9db03242019-04-26 02:32:01 +0200911
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000912 while (!gc_list_is_empty(collectable)) {
INADA Naoki5ac9e6e2018-07-10 17:19:53 +0900913 PyGC_Head *gc = GC_NEXT(collectable);
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000914 PyObject *op = FROM_GC(gc);
Tim Peters88396172002-06-30 17:56:40 +0000915
Victor Stinnera4b2bc72018-10-26 18:00:13 +0200916 _PyObject_ASSERT_WITH_MSG(op, Py_REFCNT(op) > 0,
917 "refcount is too small");
INADA Naoki5ac9e6e2018-07-10 17:19:53 +0900918
Victor Stinner9db03242019-04-26 02:32:01 +0200919 if (state->debug & DEBUG_SAVEALL) {
920 assert(state->garbage != NULL);
921 if (PyList_Append(state->garbage, op) < 0) {
Serhiy Storchakac4653c92018-05-29 18:50:10 +0300922 PyErr_Clear();
923 }
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000924 }
925 else {
Victor Stinner9db03242019-04-26 02:32:01 +0200926 inquiry clear;
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000927 if ((clear = Py_TYPE(op)->tp_clear) != NULL) {
928 Py_INCREF(op);
Serhiy Storchakac4653c92018-05-29 18:50:10 +0300929 (void) clear(op);
930 if (PyErr_Occurred()) {
Victor Stinner71c52e32019-05-27 08:57:14 +0200931 _PyErr_WriteUnraisableMsg("in tp_clear of",
932 (PyObject*)Py_TYPE(op));
Serhiy Storchakac4653c92018-05-29 18:50:10 +0300933 }
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000934 Py_DECREF(op);
935 }
936 }
INADA Naoki5ac9e6e2018-07-10 17:19:53 +0900937 if (GC_NEXT(collectable) == gc) {
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000938 /* object is still alive, move it, it may die later */
939 gc_list_move(gc, old);
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000940 }
941 }
Jeremy Hyltonc5007aa2000-06-30 05:02:53 +0000942}
943
Christian Heimesa156e092008-02-16 07:38:31 +0000944/* Clear all free lists
945 * All free lists are cleared during the collection of the highest generation.
946 * Allocated items in the free list may keep a pymalloc arena occupied.
947 * Clearing the free lists may give back memory to the OS earlier.
948 */
949static void
950clear_freelists(void)
951{
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000952 (void)PyMethod_ClearFreeList();
953 (void)PyFrame_ClearFreeList();
954 (void)PyCFunction_ClearFreeList();
955 (void)PyTuple_ClearFreeList();
956 (void)PyUnicode_ClearFreeList();
957 (void)PyFloat_ClearFreeList();
Antoine Pitrou9a812cb2011-11-15 00:00:12 +0100958 (void)PyList_ClearFreeList();
959 (void)PyDict_ClearFreeList();
Antoine Pitrou093ce9c2011-12-16 11:24:27 +0100960 (void)PySet_ClearFreeList();
Yury Selivanoveb636452016-09-08 22:01:51 -0700961 (void)PyAsyncGen_ClearFreeLists();
Yury Selivanovf23746a2018-01-22 19:11:18 -0500962 (void)PyContext_ClearFreeList();
Christian Heimesa156e092008-02-16 07:38:31 +0000963}
964
Inada Naokibf8162c2019-08-02 16:25:29 +0900965// Show stats for objects in each gennerations.
966static void
967show_stats_each_generations(struct _gc_runtime_state *state)
968{
969 char buf[100];
970 size_t pos = 0;
971
972 for (int i = 0; i < NUM_GENERATIONS && pos < sizeof(buf); i++) {
973 pos += PyOS_snprintf(buf+pos, sizeof(buf)-pos,
974 " %"PY_FORMAT_SIZE_T"d",
975 gc_list_size(GEN_HEAD(state, i)));
976 }
977
978 PySys_FormatStderr(
979 "gc: objects in each generation:%s\n"
980 "gc: objects in permanent generation: %zd\n",
981 buf, gc_list_size(&state->permanent_generation.head));
982}
983
Jeremy Hyltonc5007aa2000-06-30 05:02:53 +0000984/* This is the main function. Read this to understand how the
985 * collection process works. */
Neal Norwitz7b216c52006-03-04 20:01:53 +0000986static Py_ssize_t
Victor Stinner9db03242019-04-26 02:32:01 +0200987collect(struct _gc_runtime_state *state, int generation,
988 Py_ssize_t *n_collected, Py_ssize_t *n_uncollectable, int nofail)
Jeremy Hyltonc5007aa2000-06-30 05:02:53 +0000989{
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000990 int i;
991 Py_ssize_t m = 0; /* # objects collected */
992 Py_ssize_t n = 0; /* # unreachable objects that couldn't be collected */
993 PyGC_Head *young; /* the generation we are examining */
994 PyGC_Head *old; /* next older generation */
995 PyGC_Head unreachable; /* non-problematic unreachable trash */
996 PyGC_Head finalizers; /* objects with, & reachable from, __del__ */
997 PyGC_Head *gc;
Victor Stinner7181dec2015-03-27 17:47:53 +0100998 _PyTime_t t1 = 0; /* initialize to prevent a compiler warning */
Antoine Pitrou40f6b122014-05-24 19:21:53 +0200999
Victor Stinner9db03242019-04-26 02:32:01 +02001000 if (state->debug & DEBUG_STATS) {
Inada Naokibf8162c2019-08-02 16:25:29 +09001001 PySys_WriteStderr("gc: collecting generation %d...\n", generation);
1002 show_stats_each_generations(state);
Victor Stinner7181dec2015-03-27 17:47:53 +01001003 t1 = _PyTime_GetMonotonicClock();
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001004 }
Neil Schemenauer2880ae52002-05-04 05:35:20 +00001005
Łukasz Langaa785c872016-09-09 17:37:37 -07001006 if (PyDTrace_GC_START_ENABLED())
1007 PyDTrace_GC_START(generation);
1008
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001009 /* update collection and allocation counters */
1010 if (generation+1 < NUM_GENERATIONS)
Victor Stinner9db03242019-04-26 02:32:01 +02001011 state->generations[generation+1].count += 1;
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001012 for (i = 0; i <= generation; i++)
Victor Stinner9db03242019-04-26 02:32:01 +02001013 state->generations[i].count = 0;
Neil Schemenauer2880ae52002-05-04 05:35:20 +00001014
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001015 /* merge younger generations with one we are currently collecting */
1016 for (i = 0; i < generation; i++) {
Victor Stinner9db03242019-04-26 02:32:01 +02001017 gc_list_merge(GEN_HEAD(state, i), GEN_HEAD(state, generation));
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001018 }
Neil Schemenauer2880ae52002-05-04 05:35:20 +00001019
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001020 /* handy references */
Victor Stinner9db03242019-04-26 02:32:01 +02001021 young = GEN_HEAD(state, generation);
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001022 if (generation < NUM_GENERATIONS-1)
Victor Stinner9db03242019-04-26 02:32:01 +02001023 old = GEN_HEAD(state, generation+1);
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001024 else
1025 old = young;
Jeremy Hyltonc5007aa2000-06-30 05:02:53 +00001026
INADA Naoki5ac9e6e2018-07-10 17:19:53 +09001027 validate_list(young, 0);
1028 validate_list(old, 0);
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001029 /* Using ob_refcnt and gc_refs, calculate which objects in the
1030 * container set are reachable from outside the set (i.e., have a
1031 * refcount greater than 0 when all the references within the
1032 * set are taken into account).
1033 */
INADA Naoki5ac9e6e2018-07-10 17:19:53 +09001034 update_refs(young); // gc_prev is used for gc_refs
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001035 subtract_refs(young);
Jeremy Hyltonc5007aa2000-06-30 05:02:53 +00001036
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001037 /* Leave everything reachable from outside young in young, and move
1038 * everything else (in young) to unreachable.
1039 * NOTE: This used to move the reachable objects into a reachable
1040 * set instead. But most things usually turn out to be reachable,
1041 * so it's more efficient to move the unreachable things.
1042 */
1043 gc_list_init(&unreachable);
INADA Naoki5ac9e6e2018-07-10 17:19:53 +09001044 move_unreachable(young, &unreachable); // gc_prev is pointer again
1045 validate_list(young, 0);
Jeremy Hyltonc5007aa2000-06-30 05:02:53 +00001046
INADA Naoki5ac9e6e2018-07-10 17:19:53 +09001047 untrack_tuples(young);
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001048 /* Move reachable objects to next generation. */
1049 if (young != old) {
1050 if (generation == NUM_GENERATIONS - 2) {
Victor Stinner9db03242019-04-26 02:32:01 +02001051 state->long_lived_pending += gc_list_size(young);
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001052 }
1053 gc_list_merge(young, old);
1054 }
1055 else {
Antoine Pitroue1ad3da2012-05-28 22:22:34 +02001056 /* We only untrack dicts in full collections, to avoid quadratic
1057 dict build-up. See issue #14775. */
1058 untrack_dicts(young);
Victor Stinner9db03242019-04-26 02:32:01 +02001059 state->long_lived_pending = 0;
1060 state->long_lived_total = gc_list_size(young);
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001061 }
Jeremy Hyltonc5007aa2000-06-30 05:02:53 +00001062
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001063 /* All objects in unreachable are trash, but objects reachable from
Antoine Pitrou796564c2013-07-30 19:59:21 +02001064 * legacy finalizers (e.g. tp_del) can't safely be deleted.
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001065 */
1066 gc_list_init(&finalizers);
INADA Naoki5ac9e6e2018-07-10 17:19:53 +09001067 // NEXT_MASK_UNREACHABLE is cleared here.
1068 // After move_legacy_finalizers(), unreachable is normal list.
Antoine Pitrou796564c2013-07-30 19:59:21 +02001069 move_legacy_finalizers(&unreachable, &finalizers);
1070 /* finalizers contains the unreachable objects with a legacy finalizer;
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001071 * unreachable objects reachable *from* those are also uncollectable,
1072 * and we move those into the finalizers list too.
1073 */
Antoine Pitrou796564c2013-07-30 19:59:21 +02001074 move_legacy_finalizer_reachable(&finalizers);
Jeremy Hyltonc5007aa2000-06-30 05:02:53 +00001075
INADA Naoki5ac9e6e2018-07-10 17:19:53 +09001076 validate_list(&finalizers, 0);
1077 validate_list(&unreachable, PREV_MASK_COLLECTING);
1078
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001079 /* Collect statistics on collectable objects found and print
1080 * debugging information.
1081 */
INADA Naoki5ac9e6e2018-07-10 17:19:53 +09001082 for (gc = GC_NEXT(&unreachable); gc != &unreachable; gc = GC_NEXT(gc)) {
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001083 m++;
Victor Stinner9db03242019-04-26 02:32:01 +02001084 if (state->debug & DEBUG_COLLECTABLE) {
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001085 debug_cycle("collectable", FROM_GC(gc));
1086 }
1087 }
Tim Petersead8b7a2004-10-30 23:09:22 +00001088
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001089 /* Clear weakrefs and invoke callbacks as necessary. */
1090 m += handle_weakrefs(&unreachable, old);
Tim Petersead8b7a2004-10-30 23:09:22 +00001091
INADA Naoki5ac9e6e2018-07-10 17:19:53 +09001092 validate_list(old, 0);
1093 validate_list(&unreachable, PREV_MASK_COLLECTING);
1094
Antoine Pitrou796564c2013-07-30 19:59:21 +02001095 /* Call tp_finalize on objects which have one. */
Tim Peters5fbc7b12014-05-08 17:42:19 -05001096 finalize_garbage(&unreachable);
Antoine Pitrou796564c2013-07-30 19:59:21 +02001097
INADA Naoki5ac9e6e2018-07-10 17:19:53 +09001098 if (check_garbage(&unreachable)) { // clear PREV_MASK_COLLECTING here
Antoine Pitrou796564c2013-07-30 19:59:21 +02001099 gc_list_merge(&unreachable, old);
1100 }
1101 else {
1102 /* Call tp_clear on objects in the unreachable set. This will cause
1103 * the reference cycles to be broken. It may also cause some objects
1104 * in finalizers to be freed.
1105 */
Victor Stinner9db03242019-04-26 02:32:01 +02001106 delete_garbage(state, &unreachable, old);
Antoine Pitrou796564c2013-07-30 19:59:21 +02001107 }
Jeremy Hyltonc5007aa2000-06-30 05:02:53 +00001108
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001109 /* Collect statistics on uncollectable objects found and print
1110 * debugging information. */
INADA Naoki5ac9e6e2018-07-10 17:19:53 +09001111 for (gc = GC_NEXT(&finalizers); gc != &finalizers; gc = GC_NEXT(gc)) {
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001112 n++;
Victor Stinner9db03242019-04-26 02:32:01 +02001113 if (state->debug & DEBUG_UNCOLLECTABLE)
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001114 debug_cycle("uncollectable", FROM_GC(gc));
1115 }
Victor Stinner9db03242019-04-26 02:32:01 +02001116 if (state->debug & DEBUG_STATS) {
Inada Naokibf8162c2019-08-02 16:25:29 +09001117 double d = _PyTime_AsSecondsDouble(_PyTime_GetMonotonicClock() - t1);
Inada Naoki013e52f2019-08-31 09:13:42 +09001118 PySys_WriteStderr(
1119 "gc: done, %" PY_FORMAT_SIZE_T "d unreachable, "
1120 "%" PY_FORMAT_SIZE_T "d uncollectable, %.4fs elapsed\n",
Inada Naokibf8162c2019-08-02 16:25:29 +09001121 n+m, n, d);
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001122 }
Jeremy Hyltonc5007aa2000-06-30 05:02:53 +00001123
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001124 /* Append instances in the uncollectable set to a Python
1125 * reachable list of garbage. The programmer has to deal with
1126 * this if they insist on creating this type of structure.
1127 */
Victor Stinner9db03242019-04-26 02:32:01 +02001128 handle_legacy_finalizers(state, &finalizers, old);
INADA Naoki5ac9e6e2018-07-10 17:19:53 +09001129 validate_list(old, 0);
Jeremy Hyltonc5007aa2000-06-30 05:02:53 +00001130
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001131 /* Clear free list only during the collection of the highest
1132 * generation */
1133 if (generation == NUM_GENERATIONS-1) {
1134 clear_freelists();
1135 }
Christian Heimesa156e092008-02-16 07:38:31 +00001136
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001137 if (PyErr_Occurred()) {
Antoine Pitroufef34e32013-05-19 01:11:58 +02001138 if (nofail) {
1139 PyErr_Clear();
1140 }
1141 else {
1142 if (gc_str == NULL)
1143 gc_str = PyUnicode_FromString("garbage collection");
1144 PyErr_WriteUnraisable(gc_str);
1145 Py_FatalError("unexpected exception during garbage collection");
1146 }
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001147 }
Kristján Valur Jónsson69c63522012-04-15 11:41:32 +00001148
Antoine Pitroud4156c12012-10-30 22:43:19 +01001149 /* Update stats */
Victor Stinner9db03242019-04-26 02:32:01 +02001150 if (n_collected) {
Kristján Valur Jónsson69c63522012-04-15 11:41:32 +00001151 *n_collected = m;
Victor Stinner9db03242019-04-26 02:32:01 +02001152 }
1153 if (n_uncollectable) {
Kristján Valur Jónsson69c63522012-04-15 11:41:32 +00001154 *n_uncollectable = n;
Victor Stinner9db03242019-04-26 02:32:01 +02001155 }
1156
1157 struct gc_generation_stats *stats = &state->generation_stats[generation];
Antoine Pitroud4156c12012-10-30 22:43:19 +01001158 stats->collections++;
1159 stats->collected += m;
1160 stats->uncollectable += n;
Łukasz Langaa785c872016-09-09 17:37:37 -07001161
Victor Stinner9db03242019-04-26 02:32:01 +02001162 if (PyDTrace_GC_DONE_ENABLED()) {
Łukasz Langaa785c872016-09-09 17:37:37 -07001163 PyDTrace_GC_DONE(n+m);
Victor Stinner9db03242019-04-26 02:32:01 +02001164 }
Łukasz Langaa785c872016-09-09 17:37:37 -07001165
Serhiy Storchakac4653c92018-05-29 18:50:10 +03001166 assert(!PyErr_Occurred());
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001167 return n+m;
Jeremy Hyltonc5007aa2000-06-30 05:02:53 +00001168}
1169
Kristján Valur Jónsson69c63522012-04-15 11:41:32 +00001170/* Invoke progress callbacks to notify clients that garbage collection
1171 * is starting or stopping
1172 */
1173static void
Victor Stinner9db03242019-04-26 02:32:01 +02001174invoke_gc_callback(struct _gc_runtime_state *state, const char *phase,
1175 int generation, Py_ssize_t collected,
1176 Py_ssize_t uncollectable)
Kristján Valur Jónsson69c63522012-04-15 11:41:32 +00001177{
Serhiy Storchakac4653c92018-05-29 18:50:10 +03001178 assert(!PyErr_Occurred());
Victor Stinner9db03242019-04-26 02:32:01 +02001179
Kristján Valur Jónsson69c63522012-04-15 11:41:32 +00001180 /* we may get called very early */
Victor Stinner9db03242019-04-26 02:32:01 +02001181 if (state->callbacks == NULL) {
Kristján Valur Jónsson69c63522012-04-15 11:41:32 +00001182 return;
Victor Stinner9db03242019-04-26 02:32:01 +02001183 }
1184
Kristján Valur Jónsson69c63522012-04-15 11:41:32 +00001185 /* The local variable cannot be rebound, check it for sanity */
Victor Stinner9db03242019-04-26 02:32:01 +02001186 assert(PyList_CheckExact(state->callbacks));
1187 PyObject *info = NULL;
1188 if (PyList_GET_SIZE(state->callbacks) != 0) {
Kristján Valur Jónsson69c63522012-04-15 11:41:32 +00001189 info = Py_BuildValue("{sisnsn}",
1190 "generation", generation,
1191 "collected", collected,
1192 "uncollectable", uncollectable);
1193 if (info == NULL) {
1194 PyErr_WriteUnraisable(NULL);
1195 return;
1196 }
1197 }
Victor Stinner9db03242019-04-26 02:32:01 +02001198 for (Py_ssize_t i=0; i<PyList_GET_SIZE(state->callbacks); i++) {
1199 PyObject *r, *cb = PyList_GET_ITEM(state->callbacks, i);
Kristján Valur Jónsson69c63522012-04-15 11:41:32 +00001200 Py_INCREF(cb); /* make sure cb doesn't go away */
1201 r = PyObject_CallFunction(cb, "sO", phase, info);
Serhiy Storchaka301e3cc2018-05-24 15:19:29 +03001202 if (r == NULL) {
Kristján Valur Jónsson69c63522012-04-15 11:41:32 +00001203 PyErr_WriteUnraisable(cb);
Serhiy Storchaka301e3cc2018-05-24 15:19:29 +03001204 }
1205 else {
1206 Py_DECREF(r);
1207 }
Kristján Valur Jónsson69c63522012-04-15 11:41:32 +00001208 Py_DECREF(cb);
1209 }
1210 Py_XDECREF(info);
Serhiy Storchakac4653c92018-05-29 18:50:10 +03001211 assert(!PyErr_Occurred());
Kristján Valur Jónsson69c63522012-04-15 11:41:32 +00001212}
1213
1214/* Perform garbage collection of a generation and invoke
1215 * progress callbacks.
1216 */
1217static Py_ssize_t
Victor Stinner9db03242019-04-26 02:32:01 +02001218collect_with_callback(struct _gc_runtime_state *state, int generation)
Kristján Valur Jónsson69c63522012-04-15 11:41:32 +00001219{
Serhiy Storchakac4653c92018-05-29 18:50:10 +03001220 assert(!PyErr_Occurred());
Victor Stinner9db03242019-04-26 02:32:01 +02001221 Py_ssize_t result, collected, uncollectable;
1222 invoke_gc_callback(state, "start", generation, 0, 0);
1223 result = collect(state, generation, &collected, &uncollectable, 0);
1224 invoke_gc_callback(state, "stop", generation, collected, uncollectable);
Serhiy Storchakac4653c92018-05-29 18:50:10 +03001225 assert(!PyErr_Occurred());
Kristján Valur Jónsson69c63522012-04-15 11:41:32 +00001226 return result;
1227}
1228
Neal Norwitz7b216c52006-03-04 20:01:53 +00001229static Py_ssize_t
Victor Stinner9db03242019-04-26 02:32:01 +02001230collect_generations(struct _gc_runtime_state *state)
Jeremy Hyltonc5007aa2000-06-30 05:02:53 +00001231{
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001232 /* Find the oldest generation (highest numbered) where the count
1233 * exceeds the threshold. Objects in the that generation and
1234 * generations younger than it will be collected. */
Victor Stinner9db03242019-04-26 02:32:01 +02001235 Py_ssize_t n = 0;
1236 for (int i = NUM_GENERATIONS-1; i >= 0; i--) {
1237 if (state->generations[i].count > state->generations[i].threshold) {
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001238 /* Avoid quadratic performance degradation in number
1239 of tracked objects. See comments at the beginning
1240 of this file, and issue #4074.
1241 */
1242 if (i == NUM_GENERATIONS - 1
Victor Stinner9db03242019-04-26 02:32:01 +02001243 && state->long_lived_pending < state->long_lived_total / 4)
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001244 continue;
Victor Stinner9db03242019-04-26 02:32:01 +02001245 n = collect_with_callback(state, i);
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001246 break;
1247 }
1248 }
1249 return n;
Jeremy Hyltonc5007aa2000-06-30 05:02:53 +00001250}
1251
Serhiy Storchaka93260282017-02-04 11:19:59 +02001252#include "clinic/gcmodule.c.h"
1253
1254/*[clinic input]
1255gc.enable
1256
1257Enable automatic garbage collection.
1258[clinic start generated code]*/
Jeremy Hyltonc5007aa2000-06-30 05:02:53 +00001259
Vladimir Marangozovf9d20c32000-08-06 22:45:31 +00001260static PyObject *
Serhiy Storchaka93260282017-02-04 11:19:59 +02001261gc_enable_impl(PyObject *module)
1262/*[clinic end generated code: output=45a427e9dce9155c input=81ac4940ca579707]*/
Vladimir Marangozovf9d20c32000-08-06 22:45:31 +00001263{
Eric Snow2ebc5ce2017-09-07 23:51:28 -06001264 _PyRuntime.gc.enabled = 1;
Serhiy Storchaka228b12e2017-01-23 09:47:21 +02001265 Py_RETURN_NONE;
Vladimir Marangozovf9d20c32000-08-06 22:45:31 +00001266}
1267
Serhiy Storchaka93260282017-02-04 11:19:59 +02001268/*[clinic input]
1269gc.disable
1270
1271Disable automatic garbage collection.
1272[clinic start generated code]*/
Vladimir Marangozovf9d20c32000-08-06 22:45:31 +00001273
1274static PyObject *
Serhiy Storchaka93260282017-02-04 11:19:59 +02001275gc_disable_impl(PyObject *module)
1276/*[clinic end generated code: output=97d1030f7aa9d279 input=8c2e5a14e800d83b]*/
Vladimir Marangozovf9d20c32000-08-06 22:45:31 +00001277{
Eric Snow2ebc5ce2017-09-07 23:51:28 -06001278 _PyRuntime.gc.enabled = 0;
Serhiy Storchaka228b12e2017-01-23 09:47:21 +02001279 Py_RETURN_NONE;
Vladimir Marangozovf9d20c32000-08-06 22:45:31 +00001280}
1281
Serhiy Storchaka93260282017-02-04 11:19:59 +02001282/*[clinic input]
1283gc.isenabled -> bool
Vladimir Marangozovf9d20c32000-08-06 22:45:31 +00001284
Serhiy Storchaka93260282017-02-04 11:19:59 +02001285Returns true if automatic garbage collection is enabled.
1286[clinic start generated code]*/
1287
1288static int
1289gc_isenabled_impl(PyObject *module)
1290/*[clinic end generated code: output=1874298331c49130 input=30005e0422373b31]*/
Vladimir Marangozovf9d20c32000-08-06 22:45:31 +00001291{
Eric Snow2ebc5ce2017-09-07 23:51:28 -06001292 return _PyRuntime.gc.enabled;
Vladimir Marangozovf9d20c32000-08-06 22:45:31 +00001293}
1294
Serhiy Storchaka93260282017-02-04 11:19:59 +02001295/*[clinic input]
1296gc.collect -> Py_ssize_t
Jeremy Hyltonc5007aa2000-06-30 05:02:53 +00001297
Serhiy Storchaka93260282017-02-04 11:19:59 +02001298 generation: int(c_default="NUM_GENERATIONS - 1") = 2
1299
1300Run the garbage collector.
1301
1302With no arguments, run a full collection. The optional argument
1303may be an integer specifying which generation to collect. A ValueError
1304is raised if the generation number is invalid.
1305
1306The number of unreachable objects is returned.
1307[clinic start generated code]*/
1308
1309static Py_ssize_t
1310gc_collect_impl(PyObject *module, int generation)
1311/*[clinic end generated code: output=b697e633043233c7 input=40720128b682d879]*/
Jeremy Hyltonc5007aa2000-06-30 05:02:53 +00001312{
Jeremy Hyltonc5007aa2000-06-30 05:02:53 +00001313
Serhiy Storchaka93260282017-02-04 11:19:59 +02001314 if (generation < 0 || generation >= NUM_GENERATIONS) {
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001315 PyErr_SetString(PyExc_ValueError, "invalid generation");
Serhiy Storchaka93260282017-02-04 11:19:59 +02001316 return -1;
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001317 }
Barry Warsawd3c38ff2006-03-07 09:46:03 +00001318
Victor Stinner9db03242019-04-26 02:32:01 +02001319 struct _gc_runtime_state *state = &_PyRuntime.gc;
1320 Py_ssize_t n;
1321 if (state->collecting) {
1322 /* already collecting, don't do anything */
1323 n = 0;
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001324 }
Victor Stinner9db03242019-04-26 02:32:01 +02001325 else {
1326 state->collecting = 1;
1327 n = collect_with_callback(state, generation);
1328 state->collecting = 0;
1329 }
Serhiy Storchaka93260282017-02-04 11:19:59 +02001330 return n;
Jeremy Hyltonc5007aa2000-06-30 05:02:53 +00001331}
1332
Serhiy Storchaka93260282017-02-04 11:19:59 +02001333/*[clinic input]
1334gc.set_debug
1335
1336 flags: int
1337 An integer that can have the following bits turned on:
1338 DEBUG_STATS - Print statistics during collection.
1339 DEBUG_COLLECTABLE - Print collectable objects found.
1340 DEBUG_UNCOLLECTABLE - Print unreachable but uncollectable objects
1341 found.
1342 DEBUG_SAVEALL - Save objects to gc.garbage rather than freeing them.
1343 DEBUG_LEAK - Debug leaking programs (everything but STATS).
1344 /
1345
1346Set the garbage collection debugging flags.
1347
1348Debugging information is written to sys.stderr.
1349[clinic start generated code]*/
Jeremy Hyltonc5007aa2000-06-30 05:02:53 +00001350
1351static PyObject *
Serhiy Storchaka93260282017-02-04 11:19:59 +02001352gc_set_debug_impl(PyObject *module, int flags)
1353/*[clinic end generated code: output=7c8366575486b228 input=5e5ce15e84fbed15]*/
Jeremy Hyltonc5007aa2000-06-30 05:02:53 +00001354{
Eric Snow2ebc5ce2017-09-07 23:51:28 -06001355 _PyRuntime.gc.debug = flags;
Jeremy Hyltonc5007aa2000-06-30 05:02:53 +00001356
Serhiy Storchaka228b12e2017-01-23 09:47:21 +02001357 Py_RETURN_NONE;
Jeremy Hyltonc5007aa2000-06-30 05:02:53 +00001358}
1359
Serhiy Storchaka93260282017-02-04 11:19:59 +02001360/*[clinic input]
1361gc.get_debug -> int
Jeremy Hyltonc5007aa2000-06-30 05:02:53 +00001362
Serhiy Storchaka93260282017-02-04 11:19:59 +02001363Get the garbage collection debugging flags.
1364[clinic start generated code]*/
1365
1366static int
1367gc_get_debug_impl(PyObject *module)
1368/*[clinic end generated code: output=91242f3506cd1e50 input=91a101e1c3b98366]*/
Jeremy Hyltonc5007aa2000-06-30 05:02:53 +00001369{
Eric Snow2ebc5ce2017-09-07 23:51:28 -06001370 return _PyRuntime.gc.debug;
Jeremy Hyltonc5007aa2000-06-30 05:02:53 +00001371}
1372
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00001373PyDoc_STRVAR(gc_set_thresh__doc__,
Neal Norwitz2a47c0f2002-01-29 00:53:41 +00001374"set_threshold(threshold0, [threshold1, threshold2]) -> None\n"
Jeremy Hyltonc5007aa2000-06-30 05:02:53 +00001375"\n"
1376"Sets the collection thresholds. Setting threshold0 to zero disables\n"
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00001377"collection.\n");
Jeremy Hyltonc5007aa2000-06-30 05:02:53 +00001378
1379static PyObject *
Victor Stinner9db03242019-04-26 02:32:01 +02001380gc_set_threshold(PyObject *self, PyObject *args)
Jeremy Hyltonc5007aa2000-06-30 05:02:53 +00001381{
Victor Stinner9db03242019-04-26 02:32:01 +02001382 struct _gc_runtime_state *state = &_PyRuntime.gc;
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001383 if (!PyArg_ParseTuple(args, "i|ii:set_threshold",
Victor Stinner9db03242019-04-26 02:32:01 +02001384 &state->generations[0].threshold,
1385 &state->generations[1].threshold,
1386 &state->generations[2].threshold))
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001387 return NULL;
Victor Stinner9db03242019-04-26 02:32:01 +02001388 for (int i = 3; i < NUM_GENERATIONS; i++) {
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001389 /* generations higher than 2 get the same threshold */
Victor Stinner9db03242019-04-26 02:32:01 +02001390 state->generations[i].threshold = state->generations[2].threshold;
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001391 }
Serhiy Storchaka228b12e2017-01-23 09:47:21 +02001392 Py_RETURN_NONE;
Jeremy Hyltonc5007aa2000-06-30 05:02:53 +00001393}
1394
Serhiy Storchaka93260282017-02-04 11:19:59 +02001395/*[clinic input]
1396gc.get_threshold
1397
1398Return the current collection thresholds.
1399[clinic start generated code]*/
Jeremy Hyltonc5007aa2000-06-30 05:02:53 +00001400
1401static PyObject *
Serhiy Storchaka93260282017-02-04 11:19:59 +02001402gc_get_threshold_impl(PyObject *module)
1403/*[clinic end generated code: output=7902bc9f41ecbbd8 input=286d79918034d6e6]*/
Jeremy Hyltonc5007aa2000-06-30 05:02:53 +00001404{
Victor Stinner9db03242019-04-26 02:32:01 +02001405 struct _gc_runtime_state *state = &_PyRuntime.gc;
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001406 return Py_BuildValue("(iii)",
Victor Stinner9db03242019-04-26 02:32:01 +02001407 state->generations[0].threshold,
1408 state->generations[1].threshold,
1409 state->generations[2].threshold);
Jeremy Hyltonc5007aa2000-06-30 05:02:53 +00001410}
1411
Serhiy Storchaka93260282017-02-04 11:19:59 +02001412/*[clinic input]
1413gc.get_count
1414
1415Return a three-tuple of the current collection counts.
1416[clinic start generated code]*/
Barry Warsawd3c38ff2006-03-07 09:46:03 +00001417
1418static PyObject *
Serhiy Storchaka93260282017-02-04 11:19:59 +02001419gc_get_count_impl(PyObject *module)
1420/*[clinic end generated code: output=354012e67b16398f input=a392794a08251751]*/
Barry Warsawd3c38ff2006-03-07 09:46:03 +00001421{
Victor Stinner9db03242019-04-26 02:32:01 +02001422 struct _gc_runtime_state *state = &_PyRuntime.gc;
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001423 return Py_BuildValue("(iii)",
Victor Stinner9db03242019-04-26 02:32:01 +02001424 state->generations[0].count,
1425 state->generations[1].count,
1426 state->generations[2].count);
Barry Warsawd3c38ff2006-03-07 09:46:03 +00001427}
1428
Neil Schemenauer48c70342001-08-09 15:38:31 +00001429static int
Martin v. Löwis560da622001-11-24 09:24:51 +00001430referrersvisit(PyObject* obj, PyObject *objs)
Neil Schemenauer48c70342001-08-09 15:38:31 +00001431{
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001432 Py_ssize_t i;
1433 for (i = 0; i < PyTuple_GET_SIZE(objs); i++)
1434 if (PyTuple_GET_ITEM(objs, i) == obj)
1435 return 1;
1436 return 0;
Neil Schemenauer48c70342001-08-09 15:38:31 +00001437}
1438
Neil Schemenauer17e7be62001-08-10 14:46:47 +00001439static int
Martin v. Löwis560da622001-11-24 09:24:51 +00001440gc_referrers_for(PyObject *objs, PyGC_Head *list, PyObject *resultlist)
Neil Schemenauer48c70342001-08-09 15:38:31 +00001441{
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001442 PyGC_Head *gc;
1443 PyObject *obj;
1444 traverseproc traverse;
INADA Naoki5ac9e6e2018-07-10 17:19:53 +09001445 for (gc = GC_NEXT(list); gc != list; gc = GC_NEXT(gc)) {
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001446 obj = FROM_GC(gc);
1447 traverse = Py_TYPE(obj)->tp_traverse;
1448 if (obj == objs || obj == resultlist)
1449 continue;
1450 if (traverse(obj, (visitproc)referrersvisit, objs)) {
1451 if (PyList_Append(resultlist, obj) < 0)
1452 return 0; /* error */
1453 }
1454 }
1455 return 1; /* no error */
Neil Schemenauer48c70342001-08-09 15:38:31 +00001456}
1457
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00001458PyDoc_STRVAR(gc_get_referrers__doc__,
Martin v. Löwis560da622001-11-24 09:24:51 +00001459"get_referrers(*objs) -> list\n\
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00001460Return the list of objects that directly refer to any of objs.");
Neil Schemenauer48c70342001-08-09 15:38:31 +00001461
Neil Schemenauer17e7be62001-08-10 14:46:47 +00001462static PyObject *
Martin v. Löwis560da622001-11-24 09:24:51 +00001463gc_get_referrers(PyObject *self, PyObject *args)
Neil Schemenauer48c70342001-08-09 15:38:31 +00001464{
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001465 int i;
1466 PyObject *result = PyList_New(0);
1467 if (!result) return NULL;
Thomas Wouters49fd7fa2006-04-21 10:40:58 +00001468
Victor Stinner9db03242019-04-26 02:32:01 +02001469 struct _gc_runtime_state *state = &_PyRuntime.gc;
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001470 for (i = 0; i < NUM_GENERATIONS; i++) {
Victor Stinner9db03242019-04-26 02:32:01 +02001471 if (!(gc_referrers_for(args, GEN_HEAD(state, i), result))) {
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001472 Py_DECREF(result);
1473 return NULL;
1474 }
1475 }
1476 return result;
Neil Schemenauer48c70342001-08-09 15:38:31 +00001477}
Jeremy Hyltonc5007aa2000-06-30 05:02:53 +00001478
Tim Peters0f81ab62003-04-08 16:39:48 +00001479/* Append obj to list; return true if error (out of memory), false if OK. */
Jeremy Hylton5bd378b2003-04-03 16:28:38 +00001480static int
Tim Peters730f5532003-04-08 17:17:17 +00001481referentsvisit(PyObject *obj, PyObject *list)
Jeremy Hylton5bd378b2003-04-03 16:28:38 +00001482{
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001483 return PyList_Append(list, obj) < 0;
Jeremy Hylton5bd378b2003-04-03 16:28:38 +00001484}
1485
Tim Peters730f5532003-04-08 17:17:17 +00001486PyDoc_STRVAR(gc_get_referents__doc__,
1487"get_referents(*objs) -> list\n\
Jeremy Hylton059b0942003-04-03 16:29:13 +00001488Return the list of objects that are directly referred to by objs.");
Jeremy Hylton5bd378b2003-04-03 16:28:38 +00001489
1490static PyObject *
Tim Peters730f5532003-04-08 17:17:17 +00001491gc_get_referents(PyObject *self, PyObject *args)
Jeremy Hylton5bd378b2003-04-03 16:28:38 +00001492{
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001493 Py_ssize_t i;
1494 PyObject *result = PyList_New(0);
Tim Peters0f81ab62003-04-08 16:39:48 +00001495
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001496 if (result == NULL)
1497 return NULL;
Tim Peters0f81ab62003-04-08 16:39:48 +00001498
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001499 for (i = 0; i < PyTuple_GET_SIZE(args); i++) {
1500 traverseproc traverse;
1501 PyObject *obj = PyTuple_GET_ITEM(args, i);
Tim Peters0f81ab62003-04-08 16:39:48 +00001502
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001503 if (! PyObject_IS_GC(obj))
1504 continue;
1505 traverse = Py_TYPE(obj)->tp_traverse;
1506 if (! traverse)
1507 continue;
1508 if (traverse(obj, (visitproc)referentsvisit, result)) {
1509 Py_DECREF(result);
1510 return NULL;
1511 }
1512 }
1513 return result;
Jeremy Hylton5bd378b2003-04-03 16:28:38 +00001514}
1515
Serhiy Storchaka93260282017-02-04 11:19:59 +02001516/*[clinic input]
1517gc.get_objects
Pablo Galindo175421b2019-02-23 03:02:06 +00001518 generation: Py_ssize_t(accept={int, NoneType}, c_default="-1") = None
1519 Generation to extract the objects from.
Serhiy Storchaka93260282017-02-04 11:19:59 +02001520
1521Return a list of objects tracked by the collector (excluding the list returned).
Pablo Galindo175421b2019-02-23 03:02:06 +00001522
1523If generation is not None, return only the objects tracked by the collector
1524that are in that generation.
Serhiy Storchaka93260282017-02-04 11:19:59 +02001525[clinic start generated code]*/
Neil Schemenauerc7c8d8e2001-08-09 15:58:59 +00001526
Neil Schemenauerc7c8d8e2001-08-09 15:58:59 +00001527static PyObject *
Pablo Galindo175421b2019-02-23 03:02:06 +00001528gc_get_objects_impl(PyObject *module, Py_ssize_t generation)
1529/*[clinic end generated code: output=48b35fea4ba6cb0e input=ef7da9df9806754c]*/
Neil Schemenauerc7c8d8e2001-08-09 15:58:59 +00001530{
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001531 int i;
1532 PyObject* result;
Victor Stinner9db03242019-04-26 02:32:01 +02001533 struct _gc_runtime_state *state = &_PyRuntime.gc;
Neil Schemenauerc7c8d8e2001-08-09 15:58:59 +00001534
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001535 result = PyList_New(0);
Pablo Galindo175421b2019-02-23 03:02:06 +00001536 if (result == NULL) {
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001537 return NULL;
Pablo Galindo175421b2019-02-23 03:02:06 +00001538 }
1539
1540 /* If generation is passed, we extract only that generation */
Victor Stinner0810fa72019-04-15 17:54:09 +02001541 if (generation != -1) {
Pablo Galindo175421b2019-02-23 03:02:06 +00001542 if (generation >= NUM_GENERATIONS) {
1543 PyErr_Format(PyExc_ValueError,
1544 "generation parameter must be less than the number of "
1545 "available generations (%i)",
1546 NUM_GENERATIONS);
1547 goto error;
1548 }
1549
1550 if (generation < 0) {
1551 PyErr_SetString(PyExc_ValueError,
1552 "generation parameter cannot be negative");
1553 goto error;
1554 }
1555
Victor Stinner9db03242019-04-26 02:32:01 +02001556 if (append_objects(result, GEN_HEAD(state, generation))) {
Pablo Galindo175421b2019-02-23 03:02:06 +00001557 goto error;
1558 }
1559
1560 return result;
1561 }
1562
1563 /* If generation is not passed or None, get all objects from all generations */
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001564 for (i = 0; i < NUM_GENERATIONS; i++) {
Victor Stinner9db03242019-04-26 02:32:01 +02001565 if (append_objects(result, GEN_HEAD(state, i))) {
Pablo Galindo175421b2019-02-23 03:02:06 +00001566 goto error;
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001567 }
1568 }
1569 return result;
Pablo Galindo175421b2019-02-23 03:02:06 +00001570
1571error:
1572 Py_DECREF(result);
1573 return NULL;
Neil Schemenauerc7c8d8e2001-08-09 15:58:59 +00001574}
1575
Serhiy Storchaka93260282017-02-04 11:19:59 +02001576/*[clinic input]
1577gc.get_stats
1578
1579Return a list of dictionaries containing per-generation statistics.
1580[clinic start generated code]*/
Antoine Pitroud4156c12012-10-30 22:43:19 +01001581
1582static PyObject *
Serhiy Storchaka93260282017-02-04 11:19:59 +02001583gc_get_stats_impl(PyObject *module)
1584/*[clinic end generated code: output=a8ab1d8a5d26f3ab input=1ef4ed9d17b1a470]*/
Antoine Pitroud4156c12012-10-30 22:43:19 +01001585{
1586 int i;
Antoine Pitroud4156c12012-10-30 22:43:19 +01001587 struct gc_generation_stats stats[NUM_GENERATIONS], *st;
1588
1589 /* To get consistent values despite allocations while constructing
1590 the result list, we use a snapshot of the running stats. */
Victor Stinner9db03242019-04-26 02:32:01 +02001591 struct _gc_runtime_state *state = &_PyRuntime.gc;
Antoine Pitroud4156c12012-10-30 22:43:19 +01001592 for (i = 0; i < NUM_GENERATIONS; i++) {
Victor Stinner9db03242019-04-26 02:32:01 +02001593 stats[i] = state->generation_stats[i];
Antoine Pitroud4156c12012-10-30 22:43:19 +01001594 }
1595
Victor Stinner9db03242019-04-26 02:32:01 +02001596 PyObject *result = PyList_New(0);
Antoine Pitroud4156c12012-10-30 22:43:19 +01001597 if (result == NULL)
1598 return NULL;
1599
1600 for (i = 0; i < NUM_GENERATIONS; i++) {
1601 PyObject *dict;
1602 st = &stats[i];
1603 dict = Py_BuildValue("{snsnsn}",
1604 "collections", st->collections,
1605 "collected", st->collected,
1606 "uncollectable", st->uncollectable
1607 );
1608 if (dict == NULL)
1609 goto error;
1610 if (PyList_Append(result, dict)) {
1611 Py_DECREF(dict);
1612 goto error;
1613 }
1614 Py_DECREF(dict);
1615 }
1616 return result;
1617
1618error:
1619 Py_XDECREF(result);
1620 return NULL;
1621}
1622
1623
Serhiy Storchaka93260282017-02-04 11:19:59 +02001624/*[clinic input]
1625gc.is_tracked
1626
1627 obj: object
1628 /
1629
1630Returns true if the object is tracked by the garbage collector.
1631
1632Simple atomic objects will return false.
1633[clinic start generated code]*/
Antoine Pitrou3a652b12009-03-23 18:52:06 +00001634
1635static PyObject *
Serhiy Storchaka93260282017-02-04 11:19:59 +02001636gc_is_tracked(PyObject *module, PyObject *obj)
1637/*[clinic end generated code: output=14f0103423b28e31 input=d83057f170ea2723]*/
Antoine Pitrou3a652b12009-03-23 18:52:06 +00001638{
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001639 PyObject *result;
1640
INADA Naoki5ac9e6e2018-07-10 17:19:53 +09001641 if (PyObject_IS_GC(obj) && _PyObject_GC_IS_TRACKED(obj))
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001642 result = Py_True;
1643 else
1644 result = Py_False;
1645 Py_INCREF(result);
1646 return result;
Antoine Pitrou3a652b12009-03-23 18:52:06 +00001647}
1648
brainfvckc75edab2017-10-16 12:49:41 -07001649/*[clinic input]
1650gc.freeze
1651
1652Freeze all current tracked objects and ignore them for future collections.
1653
1654This can be used before a POSIX fork() call to make the gc copy-on-write friendly.
1655Note: collection before a POSIX fork() call may free pages for future allocation
1656which can cause copy-on-write.
1657[clinic start generated code]*/
1658
1659static PyObject *
1660gc_freeze_impl(PyObject *module)
1661/*[clinic end generated code: output=502159d9cdc4c139 input=b602b16ac5febbe5]*/
1662{
Victor Stinner9db03242019-04-26 02:32:01 +02001663 struct _gc_runtime_state *state = &_PyRuntime.gc;
brainfvckc75edab2017-10-16 12:49:41 -07001664 for (int i = 0; i < NUM_GENERATIONS; ++i) {
Victor Stinner9db03242019-04-26 02:32:01 +02001665 gc_list_merge(GEN_HEAD(state, i), &state->permanent_generation.head);
1666 state->generations[i].count = 0;
brainfvckc75edab2017-10-16 12:49:41 -07001667 }
1668 Py_RETURN_NONE;
1669}
1670
1671/*[clinic input]
1672gc.unfreeze
1673
1674Unfreeze all objects in the permanent generation.
1675
1676Put all objects in the permanent generation back into oldest generation.
1677[clinic start generated code]*/
1678
1679static PyObject *
1680gc_unfreeze_impl(PyObject *module)
1681/*[clinic end generated code: output=1c15f2043b25e169 input=2dd52b170f4cef6c]*/
1682{
Victor Stinner9db03242019-04-26 02:32:01 +02001683 struct _gc_runtime_state *state = &_PyRuntime.gc;
1684 gc_list_merge(&state->permanent_generation.head, GEN_HEAD(state, NUM_GENERATIONS-1));
brainfvckc75edab2017-10-16 12:49:41 -07001685 Py_RETURN_NONE;
1686}
1687
1688/*[clinic input]
Victor Stinner05d68a82018-01-18 11:15:25 +01001689gc.get_freeze_count -> Py_ssize_t
brainfvckc75edab2017-10-16 12:49:41 -07001690
1691Return the number of objects in the permanent generation.
1692[clinic start generated code]*/
1693
Victor Stinner05d68a82018-01-18 11:15:25 +01001694static Py_ssize_t
brainfvckc75edab2017-10-16 12:49:41 -07001695gc_get_freeze_count_impl(PyObject *module)
Victor Stinner05d68a82018-01-18 11:15:25 +01001696/*[clinic end generated code: output=61cbd9f43aa032e1 input=45ffbc65cfe2a6ed]*/
brainfvckc75edab2017-10-16 12:49:41 -07001697{
1698 return gc_list_size(&_PyRuntime.gc.permanent_generation.head);
1699}
1700
Neil Schemenauerc7c8d8e2001-08-09 15:58:59 +00001701
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00001702PyDoc_STRVAR(gc__doc__,
Jeremy Hyltonc5007aa2000-06-30 05:02:53 +00001703"This module provides access to the garbage collector for reference cycles.\n"
1704"\n"
Vladimir Marangozovf9d20c32000-08-06 22:45:31 +00001705"enable() -- Enable automatic garbage collection.\n"
1706"disable() -- Disable automatic garbage collection.\n"
1707"isenabled() -- Returns true if automatic collection is enabled.\n"
Jeremy Hyltonc5007aa2000-06-30 05:02:53 +00001708"collect() -- Do a full collection right now.\n"
Thomas Wouters89f507f2006-12-13 04:49:30 +00001709"get_count() -- Return the current collection counts.\n"
R David Murray0e814632013-12-26 15:11:28 -05001710"get_stats() -- Return list of dictionaries containing per-generation stats.\n"
Jeremy Hyltonc5007aa2000-06-30 05:02:53 +00001711"set_debug() -- Set debugging flags.\n"
1712"get_debug() -- Get debugging flags.\n"
1713"set_threshold() -- Set the collection thresholds.\n"
1714"get_threshold() -- Return the current the collection thresholds.\n"
Neil Schemenauerc7c8d8e2001-08-09 15:58:59 +00001715"get_objects() -- Return a list of all objects tracked by the collector.\n"
Antoine Pitrou3a652b12009-03-23 18:52:06 +00001716"is_tracked() -- Returns true if a given object is tracked.\n"
Jeremy Hylton5bd378b2003-04-03 16:28:38 +00001717"get_referrers() -- Return the list of objects that refer to an object.\n"
brainfvckc75edab2017-10-16 12:49:41 -07001718"get_referents() -- Return the list of objects that an object refers to.\n"
1719"freeze() -- Freeze all tracked objects and ignore them for future collections.\n"
1720"unfreeze() -- Unfreeze all objects in the permanent generation.\n"
1721"get_freeze_count() -- Return the number of objects in the permanent generation.\n");
Jeremy Hyltonc5007aa2000-06-30 05:02:53 +00001722
1723static PyMethodDef GcMethods[] = {
Serhiy Storchaka93260282017-02-04 11:19:59 +02001724 GC_ENABLE_METHODDEF
1725 GC_DISABLE_METHODDEF
1726 GC_ISENABLED_METHODDEF
1727 GC_SET_DEBUG_METHODDEF
1728 GC_GET_DEBUG_METHODDEF
1729 GC_GET_COUNT_METHODDEF
Victor Stinner9db03242019-04-26 02:32:01 +02001730 {"set_threshold", gc_set_threshold, METH_VARARGS, gc_set_thresh__doc__},
Serhiy Storchaka93260282017-02-04 11:19:59 +02001731 GC_GET_THRESHOLD_METHODDEF
1732 GC_COLLECT_METHODDEF
1733 GC_GET_OBJECTS_METHODDEF
1734 GC_GET_STATS_METHODDEF
1735 GC_IS_TRACKED_METHODDEF
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001736 {"get_referrers", gc_get_referrers, METH_VARARGS,
1737 gc_get_referrers__doc__},
1738 {"get_referents", gc_get_referents, METH_VARARGS,
1739 gc_get_referents__doc__},
brainfvckc75edab2017-10-16 12:49:41 -07001740 GC_FREEZE_METHODDEF
1741 GC_UNFREEZE_METHODDEF
1742 GC_GET_FREEZE_COUNT_METHODDEF
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001743 {NULL, NULL} /* Sentinel */
Jeremy Hyltonc5007aa2000-06-30 05:02:53 +00001744};
1745
Martin v. Löwis1a214512008-06-11 05:26:20 +00001746static struct PyModuleDef gcmodule = {
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001747 PyModuleDef_HEAD_INIT,
Antoine Pitrou696e0352010-08-08 22:18:46 +00001748 "gc", /* m_name */
1749 gc__doc__, /* m_doc */
1750 -1, /* m_size */
1751 GcMethods, /* m_methods */
1752 NULL, /* m_reload */
1753 NULL, /* m_traverse */
1754 NULL, /* m_clear */
1755 NULL /* m_free */
Martin v. Löwis1a214512008-06-11 05:26:20 +00001756};
1757
Jason Tishler6bc06ec2003-09-04 11:59:50 +00001758PyMODINIT_FUNC
Martin v. Löwis1a214512008-06-11 05:26:20 +00001759PyInit_gc(void)
Jeremy Hyltonc5007aa2000-06-30 05:02:53 +00001760{
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001761 PyObject *m;
Jeremy Hyltonc5007aa2000-06-30 05:02:53 +00001762
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001763 m = PyModule_Create(&gcmodule);
Martin v. Löwis1a214512008-06-11 05:26:20 +00001764
Victor Stinner9db03242019-04-26 02:32:01 +02001765 if (m == NULL) {
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001766 return NULL;
Victor Stinner9db03242019-04-26 02:32:01 +02001767 }
Tim Peters11558872003-04-06 23:30:52 +00001768
Victor Stinner9db03242019-04-26 02:32:01 +02001769 struct _gc_runtime_state *state = &_PyRuntime.gc;
1770 if (state->garbage == NULL) {
1771 state->garbage = PyList_New(0);
1772 if (state->garbage == NULL)
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001773 return NULL;
1774 }
Victor Stinner9db03242019-04-26 02:32:01 +02001775 Py_INCREF(state->garbage);
1776 if (PyModule_AddObject(m, "garbage", state->garbage) < 0)
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001777 return NULL;
Thomas Wouters477c8d52006-05-27 19:21:47 +00001778
Victor Stinner9db03242019-04-26 02:32:01 +02001779 if (state->callbacks == NULL) {
1780 state->callbacks = PyList_New(0);
1781 if (state->callbacks == NULL)
Kristján Valur Jónsson69c63522012-04-15 11:41:32 +00001782 return NULL;
1783 }
Victor Stinner9db03242019-04-26 02:32:01 +02001784 Py_INCREF(state->callbacks);
1785 if (PyModule_AddObject(m, "callbacks", state->callbacks) < 0)
Kristján Valur Jónsson69c63522012-04-15 11:41:32 +00001786 return NULL;
1787
Martin v. Löwis1a214512008-06-11 05:26:20 +00001788#define ADD_INT(NAME) if (PyModule_AddIntConstant(m, #NAME, NAME) < 0) return NULL
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001789 ADD_INT(DEBUG_STATS);
1790 ADD_INT(DEBUG_COLLECTABLE);
1791 ADD_INT(DEBUG_UNCOLLECTABLE);
1792 ADD_INT(DEBUG_SAVEALL);
1793 ADD_INT(DEBUG_LEAK);
Tim Peters11558872003-04-06 23:30:52 +00001794#undef ADD_INT
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001795 return m;
Jeremy Hyltonc5007aa2000-06-30 05:02:53 +00001796}
1797
Guido van Rossume13ddc92003-04-17 17:29:22 +00001798/* API to invoke gc.collect() from C */
Neal Norwitz7b216c52006-03-04 20:01:53 +00001799Py_ssize_t
Guido van Rossume13ddc92003-04-17 17:29:22 +00001800PyGC_Collect(void)
1801{
Victor Stinner9db03242019-04-26 02:32:01 +02001802 struct _gc_runtime_state *state = &_PyRuntime.gc;
1803 if (!state->enabled) {
1804 return 0;
1805 }
Guido van Rossume13ddc92003-04-17 17:29:22 +00001806
Victor Stinner9db03242019-04-26 02:32:01 +02001807 Py_ssize_t n;
1808 if (state->collecting) {
1809 /* already collecting, don't do anything */
1810 n = 0;
1811 }
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001812 else {
Serhiy Storchaka301e3cc2018-05-24 15:19:29 +03001813 PyObject *exc, *value, *tb;
Victor Stinner9db03242019-04-26 02:32:01 +02001814 state->collecting = 1;
Serhiy Storchaka301e3cc2018-05-24 15:19:29 +03001815 PyErr_Fetch(&exc, &value, &tb);
Victor Stinner9db03242019-04-26 02:32:01 +02001816 n = collect_with_callback(state, NUM_GENERATIONS - 1);
Serhiy Storchaka301e3cc2018-05-24 15:19:29 +03001817 PyErr_Restore(exc, value, tb);
Victor Stinner9db03242019-04-26 02:32:01 +02001818 state->collecting = 0;
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001819 }
Guido van Rossume13ddc92003-04-17 17:29:22 +00001820
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001821 return n;
Guido van Rossume13ddc92003-04-17 17:29:22 +00001822}
1823
Antoine Pitroufef34e32013-05-19 01:11:58 +02001824Py_ssize_t
Łukasz Langafef7e942016-09-09 21:47:46 -07001825_PyGC_CollectIfEnabled(void)
1826{
Łukasz Langafef7e942016-09-09 21:47:46 -07001827 return PyGC_Collect();
1828}
1829
1830Py_ssize_t
Antoine Pitroufef34e32013-05-19 01:11:58 +02001831_PyGC_CollectNoFail(void)
1832{
Victor Stinner9db03242019-04-26 02:32:01 +02001833 assert(!PyErr_Occurred());
1834
1835 struct _gc_runtime_state *state = &_PyRuntime.gc;
Antoine Pitroufef34e32013-05-19 01:11:58 +02001836 Py_ssize_t n;
1837
Antoine Pitrouc69c9bc2013-08-15 20:15:15 +02001838 /* Ideally, this function is only called on interpreter shutdown,
1839 and therefore not recursively. Unfortunately, when there are daemon
1840 threads, a daemon thread can start a cyclic garbage collection
1841 during interpreter shutdown (and then never finish it).
1842 See http://bugs.python.org/issue8713#msg195178 for an example.
1843 */
Victor Stinner9db03242019-04-26 02:32:01 +02001844 if (state->collecting) {
Antoine Pitrouc69c9bc2013-08-15 20:15:15 +02001845 n = 0;
Victor Stinner9db03242019-04-26 02:32:01 +02001846 }
Antoine Pitrouc69c9bc2013-08-15 20:15:15 +02001847 else {
Victor Stinner9db03242019-04-26 02:32:01 +02001848 state->collecting = 1;
1849 n = collect(state, NUM_GENERATIONS - 1, NULL, NULL, 1);
1850 state->collecting = 0;
Antoine Pitrouc69c9bc2013-08-15 20:15:15 +02001851 }
Antoine Pitroufef34e32013-05-19 01:11:58 +02001852 return n;
1853}
Antoine Pitrou5f454a02013-05-06 21:15:57 +02001854
Antoine Pitrou696e0352010-08-08 22:18:46 +00001855void
Victor Stinner9db03242019-04-26 02:32:01 +02001856_PyGC_DumpShutdownStats(_PyRuntimeState *runtime)
Antoine Pitrou696e0352010-08-08 22:18:46 +00001857{
Victor Stinner9db03242019-04-26 02:32:01 +02001858 struct _gc_runtime_state *state = &runtime->gc;
1859 if (!(state->debug & DEBUG_SAVEALL)
1860 && state->garbage != NULL && PyList_GET_SIZE(state->garbage) > 0) {
Serhiy Storchakae2f92de2017-11-11 13:06:26 +02001861 const char *message;
Victor Stinner9db03242019-04-26 02:32:01 +02001862 if (state->debug & DEBUG_UNCOLLECTABLE)
Antoine Pitroub5d82042010-11-05 00:05:25 +00001863 message = "gc: %zd uncollectable objects at " \
Georg Brandl08be72d2010-10-24 15:11:22 +00001864 "shutdown";
1865 else
Antoine Pitroub5d82042010-11-05 00:05:25 +00001866 message = "gc: %zd uncollectable objects at " \
Georg Brandl08be72d2010-10-24 15:11:22 +00001867 "shutdown; use gc.set_debug(gc.DEBUG_UNCOLLECTABLE) to list them";
Antoine Pitrou070cb3c2013-05-08 13:23:25 +02001868 /* PyErr_WarnFormat does too many things and we are at shutdown,
1869 the warnings module's dependencies (e.g. linecache) may be gone
1870 already. */
1871 if (PyErr_WarnExplicitFormat(PyExc_ResourceWarning, "gc", 0,
1872 "gc", NULL, message,
Victor Stinner9db03242019-04-26 02:32:01 +02001873 PyList_GET_SIZE(state->garbage)))
Georg Brandl08be72d2010-10-24 15:11:22 +00001874 PyErr_WriteUnraisable(NULL);
Victor Stinner9db03242019-04-26 02:32:01 +02001875 if (state->debug & DEBUG_UNCOLLECTABLE) {
Antoine Pitrou696e0352010-08-08 22:18:46 +00001876 PyObject *repr = NULL, *bytes = NULL;
Victor Stinner9db03242019-04-26 02:32:01 +02001877 repr = PyObject_Repr(state->garbage);
Antoine Pitrou696e0352010-08-08 22:18:46 +00001878 if (!repr || !(bytes = PyUnicode_EncodeFSDefault(repr)))
Victor Stinner9db03242019-04-26 02:32:01 +02001879 PyErr_WriteUnraisable(state->garbage);
Antoine Pitrou696e0352010-08-08 22:18:46 +00001880 else {
1881 PySys_WriteStderr(
Antoine Pitrou070cb3c2013-05-08 13:23:25 +02001882 " %s\n",
Antoine Pitrou696e0352010-08-08 22:18:46 +00001883 PyBytes_AS_STRING(bytes)
1884 );
1885 }
1886 Py_XDECREF(repr);
1887 Py_XDECREF(bytes);
1888 }
Antoine Pitrou696e0352010-08-08 22:18:46 +00001889 }
Antoine Pitrou5f454a02013-05-06 21:15:57 +02001890}
1891
1892void
Victor Stinner8e91c242019-04-24 17:24:01 +02001893_PyGC_Fini(_PyRuntimeState *runtime)
Antoine Pitrou5f454a02013-05-06 21:15:57 +02001894{
Victor Stinner9db03242019-04-26 02:32:01 +02001895 struct _gc_runtime_state *state = &runtime->gc;
1896 Py_CLEAR(state->garbage);
1897 Py_CLEAR(state->callbacks);
Antoine Pitrou696e0352010-08-08 22:18:46 +00001898}
1899
Neil Schemenauer43411b52001-08-30 00:05:51 +00001900/* for debugging */
Guido van Rossume13ddc92003-04-17 17:29:22 +00001901void
1902_PyGC_Dump(PyGC_Head *g)
Neil Schemenauer43411b52001-08-30 00:05:51 +00001903{
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001904 _PyObject_Dump(FROM_GC(g));
Neil Schemenauer43411b52001-08-30 00:05:51 +00001905}
1906
Neil Schemenauer43411b52001-08-30 00:05:51 +00001907/* extension modules might be compiled with GC support so these
1908 functions must always be available */
1909
1910void
Victor Stinnera42de742018-11-22 10:25:22 +01001911PyObject_GC_Track(void *op_raw)
Neil Schemenauer43411b52001-08-30 00:05:51 +00001912{
Victor Stinnera42de742018-11-22 10:25:22 +01001913 PyObject *op = _PyObject_CAST(op_raw);
Victor Stinner271753a2018-11-22 01:02:54 +01001914 if (_PyObject_GC_IS_TRACKED(op)) {
1915 _PyObject_ASSERT_FAILED_MSG(op,
1916 "object already tracked "
1917 "by the garbage collector");
1918 }
Victor Stinnera42de742018-11-22 10:25:22 +01001919 _PyObject_GC_TRACK(op);
Neil Schemenauer43411b52001-08-30 00:05:51 +00001920}
1921
Neil Schemenauerfec4eb12002-04-12 02:41:03 +00001922void
Victor Stinnera42de742018-11-22 10:25:22 +01001923PyObject_GC_UnTrack(void *op_raw)
Neil Schemenauer43411b52001-08-30 00:05:51 +00001924{
Victor Stinnera42de742018-11-22 10:25:22 +01001925 PyObject *op = _PyObject_CAST(op_raw);
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001926 /* Obscure: the Py_TRASHCAN mechanism requires that we be able to
1927 * call PyObject_GC_UnTrack twice on an object.
1928 */
INADA Naoki5ac9e6e2018-07-10 17:19:53 +09001929 if (_PyObject_GC_IS_TRACKED(op)) {
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001930 _PyObject_GC_UNTRACK(op);
INADA Naoki5ac9e6e2018-07-10 17:19:53 +09001931 }
Neil Schemenauer43411b52001-08-30 00:05:51 +00001932}
1933
Victor Stinnerdb067af2014-05-02 22:31:14 +02001934static PyObject *
1935_PyObject_GC_Alloc(int use_calloc, size_t basicsize)
Neil Schemenauer43411b52001-08-30 00:05:51 +00001936{
Victor Stinner9db03242019-04-26 02:32:01 +02001937 struct _gc_runtime_state *state = &_PyRuntime.gc;
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001938 PyObject *op;
1939 PyGC_Head *g;
Victor Stinnerdb067af2014-05-02 22:31:14 +02001940 size_t size;
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001941 if (basicsize > PY_SSIZE_T_MAX - sizeof(PyGC_Head))
1942 return PyErr_NoMemory();
Victor Stinnerdb067af2014-05-02 22:31:14 +02001943 size = sizeof(PyGC_Head) + basicsize;
1944 if (use_calloc)
1945 g = (PyGC_Head *)PyObject_Calloc(1, size);
1946 else
1947 g = (PyGC_Head *)PyObject_Malloc(size);
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001948 if (g == NULL)
1949 return PyErr_NoMemory();
INADA Naoki5ac9e6e2018-07-10 17:19:53 +09001950 assert(((uintptr_t)g & 3) == 0); // g must be aligned 4bytes boundary
1951 g->_gc_next = 0;
1952 g->_gc_prev = 0;
Victor Stinner9db03242019-04-26 02:32:01 +02001953 state->generations[0].count++; /* number of allocated GC objects */
1954 if (state->generations[0].count > state->generations[0].threshold &&
1955 state->enabled &&
1956 state->generations[0].threshold &&
1957 !state->collecting &&
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001958 !PyErr_Occurred()) {
Victor Stinner9db03242019-04-26 02:32:01 +02001959 state->collecting = 1;
1960 collect_generations(state);
1961 state->collecting = 0;
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001962 }
1963 op = FROM_GC(g);
1964 return op;
Neil Schemenauer43411b52001-08-30 00:05:51 +00001965}
1966
1967PyObject *
Victor Stinnerdb067af2014-05-02 22:31:14 +02001968_PyObject_GC_Malloc(size_t basicsize)
1969{
1970 return _PyObject_GC_Alloc(0, basicsize);
1971}
1972
1973PyObject *
1974_PyObject_GC_Calloc(size_t basicsize)
1975{
1976 return _PyObject_GC_Alloc(1, basicsize);
1977}
1978
1979PyObject *
Neil Schemenauer43411b52001-08-30 00:05:51 +00001980_PyObject_GC_New(PyTypeObject *tp)
1981{
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001982 PyObject *op = _PyObject_GC_Malloc(_PyObject_SIZE(tp));
1983 if (op != NULL)
1984 op = PyObject_INIT(op, tp);
1985 return op;
Neil Schemenauer43411b52001-08-30 00:05:51 +00001986}
1987
1988PyVarObject *
Martin v. Löwis18e16552006-02-15 17:27:45 +00001989_PyObject_GC_NewVar(PyTypeObject *tp, Py_ssize_t nitems)
Neil Schemenauer43411b52001-08-30 00:05:51 +00001990{
Victor Stinner5d1866c2013-07-08 22:17:52 +02001991 size_t size;
1992 PyVarObject *op;
1993
1994 if (nitems < 0) {
1995 PyErr_BadInternalCall();
1996 return NULL;
1997 }
1998 size = _PyObject_VAR_SIZE(tp, nitems);
1999 op = (PyVarObject *) _PyObject_GC_Malloc(size);
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002000 if (op != NULL)
2001 op = PyObject_INIT_VAR(op, tp, nitems);
2002 return op;
Neil Schemenauer43411b52001-08-30 00:05:51 +00002003}
2004
2005PyVarObject *
Martin v. Löwis41290682006-02-16 14:56:14 +00002006_PyObject_GC_Resize(PyVarObject *op, Py_ssize_t nitems)
Neil Schemenauer43411b52001-08-30 00:05:51 +00002007{
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002008 const size_t basicsize = _PyObject_VAR_SIZE(Py_TYPE(op), nitems);
Victor Stinnera4b2bc72018-10-26 18:00:13 +02002009 _PyObject_ASSERT((PyObject *)op, !_PyObject_GC_IS_TRACKED(op));
2010 if (basicsize > PY_SSIZE_T_MAX - sizeof(PyGC_Head)) {
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002011 return (PyVarObject *)PyErr_NoMemory();
Victor Stinnera4b2bc72018-10-26 18:00:13 +02002012 }
2013
2014 PyGC_Head *g = AS_GC(op);
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002015 g = (PyGC_Head *)PyObject_REALLOC(g, sizeof(PyGC_Head) + basicsize);
2016 if (g == NULL)
2017 return (PyVarObject *)PyErr_NoMemory();
2018 op = (PyVarObject *) FROM_GC(g);
2019 Py_SIZE(op) = nitems;
2020 return op;
Neil Schemenauer43411b52001-08-30 00:05:51 +00002021}
2022
2023void
Neil Schemenauerfec4eb12002-04-12 02:41:03 +00002024PyObject_GC_Del(void *op)
Neil Schemenauer43411b52001-08-30 00:05:51 +00002025{
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002026 PyGC_Head *g = AS_GC(op);
INADA Naoki5ac9e6e2018-07-10 17:19:53 +09002027 if (_PyObject_GC_IS_TRACKED(op)) {
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002028 gc_list_remove(g);
INADA Naoki5ac9e6e2018-07-10 17:19:53 +09002029 }
Victor Stinner9db03242019-04-26 02:32:01 +02002030 struct _gc_runtime_state *state = &_PyRuntime.gc;
2031 if (state->generations[0].count > 0) {
2032 state->generations[0].count--;
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002033 }
2034 PyObject_FREE(g);
Neil Schemenauer43411b52001-08-30 00:05:51 +00002035}