blob: 8a3f8831d6e92a5b8db06efe6733a8df4b02a428 [file] [log] [blame]
Guido van Rossumf70e43a1991-02-19 12:39:46 +00001
Benjamin Peterson722954a2011-06-11 16:33:35 -05002/* Generic object operations; and implementation of None */
Guido van Rossum85a5fbb1990-10-14 12:07:46 +00003
Guido van Rossumc0b618a1997-05-02 03:12:38 +00004#include "Python.h"
Eric Snow2ebc5ce2017-09-07 23:51:28 -06005#include "internal/pystate.h"
Yury Selivanovf23746a2018-01-22 19:11:18 -05006#include "internal/context.h"
Benjamin Petersonfd838e62009-04-20 02:09:13 +00007#include "frameobject.h"
Guido van Rossum85a5fbb1990-10-14 12:07:46 +00008
Thomas Wouters49fd7fa2006-04-21 10:40:58 +00009#ifdef __cplusplus
10extern "C" {
11#endif
12
Victor Stinnerbd303c12013-11-07 23:07:29 +010013_Py_IDENTIFIER(Py_Repr);
14_Py_IDENTIFIER(__bytes__);
15_Py_IDENTIFIER(__dir__);
16_Py_IDENTIFIER(__isabstractmethod__);
17_Py_IDENTIFIER(builtins);
18
Tim Peters34592512002-07-11 06:23:50 +000019#ifdef Py_REF_DEBUG
Neal Norwitz84632ee2006-03-04 20:00:59 +000020Py_ssize_t _Py_RefTotal;
Thomas Wouters49fd7fa2006-04-21 10:40:58 +000021
22Py_ssize_t
23_Py_GetRefTotal(void)
24{
Antoine Pitrouf95a1b32010-05-09 15:52:27 +000025 PyObject *o;
26 Py_ssize_t total = _Py_RefTotal;
Antoine Pitrou9d952542013-08-24 21:07:07 +020027 o = _PySet_Dummy;
Antoine Pitrouf95a1b32010-05-09 15:52:27 +000028 if (o != NULL)
29 total -= o->ob_refcnt;
30 return total;
Thomas Wouters49fd7fa2006-04-21 10:40:58 +000031}
Nick Coghland6009512014-11-20 21:39:37 +100032
33void
34_PyDebug_PrintTotalRefs(void) {
Eric Snowdae02762017-09-14 00:35:58 -070035 fprintf(stderr,
36 "[%" PY_FORMAT_SIZE_T "d refs, "
37 "%" PY_FORMAT_SIZE_T "d blocks]\n",
38 _Py_GetRefTotal(), _Py_GetAllocatedBlocks());
Nick Coghland6009512014-11-20 21:39:37 +100039}
Thomas Wouters49fd7fa2006-04-21 10:40:58 +000040#endif /* Py_REF_DEBUG */
Guido van Rossum85a5fbb1990-10-14 12:07:46 +000041
Guido van Rossum3f5da241990-12-20 15:06:42 +000042/* Object allocation routines used by NEWOBJ and NEWVAROBJ macros.
43 These are used by the individual routines for object creation.
44 Do not call them otherwise, they do not initialize the object! */
Guido van Rossum85a5fbb1990-10-14 12:07:46 +000045
Tim Peters78be7992003-03-23 02:51:01 +000046#ifdef Py_TRACE_REFS
Tim Peters7571a0f2003-03-23 17:52:28 +000047/* Head of circular doubly-linked list of all objects. These are linked
48 * together via the _ob_prev and _ob_next members of a PyObject, which
49 * exist only in a Py_TRACE_REFS build.
50 */
Tim Peters78be7992003-03-23 02:51:01 +000051static PyObject refchain = {&refchain, &refchain};
Tim Peters36eb4df2003-03-23 03:33:13 +000052
Tim Peters7571a0f2003-03-23 17:52:28 +000053/* Insert op at the front of the list of all objects. If force is true,
54 * op is added even if _ob_prev and _ob_next are non-NULL already. If
55 * force is false amd _ob_prev or _ob_next are non-NULL, do nothing.
56 * force should be true if and only if op points to freshly allocated,
57 * uninitialized memory, or you've unlinked op from the list and are
Tim Peters51f8d382003-03-23 18:06:08 +000058 * relinking it into the front.
Tim Peters7571a0f2003-03-23 17:52:28 +000059 * Note that objects are normally added to the list via _Py_NewReference,
60 * which is called by PyObject_Init. Not all objects are initialized that
61 * way, though; exceptions include statically allocated type objects, and
62 * statically allocated singletons (like Py_True and Py_None).
63 */
Tim Peters36eb4df2003-03-23 03:33:13 +000064void
Tim Peters7571a0f2003-03-23 17:52:28 +000065_Py_AddToAllObjects(PyObject *op, int force)
Tim Peters36eb4df2003-03-23 03:33:13 +000066{
Tim Peters7571a0f2003-03-23 17:52:28 +000067#ifdef Py_DEBUG
Antoine Pitrouf95a1b32010-05-09 15:52:27 +000068 if (!force) {
69 /* If it's initialized memory, op must be in or out of
70 * the list unambiguously.
71 */
72 assert((op->_ob_prev == NULL) == (op->_ob_next == NULL));
73 }
Tim Peters78be7992003-03-23 02:51:01 +000074#endif
Antoine Pitrouf95a1b32010-05-09 15:52:27 +000075 if (force || op->_ob_prev == NULL) {
76 op->_ob_next = refchain._ob_next;
77 op->_ob_prev = &refchain;
78 refchain._ob_next->_ob_prev = op;
79 refchain._ob_next = op;
80 }
Tim Peters7571a0f2003-03-23 17:52:28 +000081}
Antoine Pitrouf95a1b32010-05-09 15:52:27 +000082#endif /* Py_TRACE_REFS */
Tim Peters78be7992003-03-23 02:51:01 +000083
Sjoerd Mullendera9c3c221993-10-11 12:54:31 +000084#ifdef COUNT_ALLOCS
Guido van Rossumc0b618a1997-05-02 03:12:38 +000085static PyTypeObject *type_list;
Thomas Wouters49fd7fa2006-04-21 10:40:58 +000086/* All types are added to type_list, at least when
87 they get one object created. That makes them
88 immortal, which unfortunately contributes to
89 garbage itself. If unlist_types_without_objects
90 is set, they will be removed from the type_list
91 once the last object is deallocated. */
Benjamin Petersona4a37fe2009-01-11 17:13:55 +000092static int unlist_types_without_objects;
93extern Py_ssize_t tuple_zero_allocs, fast_tuple_allocs;
94extern Py_ssize_t quick_int_allocs, quick_neg_int_allocs;
95extern Py_ssize_t null_strings, one_strings;
Sjoerd Mullendera9c3c221993-10-11 12:54:31 +000096void
Thomas Wouters49fd7fa2006-04-21 10:40:58 +000097dump_counts(FILE* f)
Sjoerd Mullendera9c3c221993-10-11 12:54:31 +000098{
Victor Stinner25420fe2017-11-20 18:12:22 -080099 PyInterpreterState *interp = PyThreadState_GET()->interp;
Miss Islington (bot)bc2e1102018-02-21 21:44:08 -0800100 if (!interp->core_config.show_alloc_count) {
Serhiy Storchaka7e160ce2016-07-03 21:03:53 +0300101 return;
Victor Stinner25420fe2017-11-20 18:12:22 -0800102 }
Sjoerd Mullendera9c3c221993-10-11 12:54:31 +0000103
Miss Islington (bot)bc2e1102018-02-21 21:44:08 -0800104 PyTypeObject *tp;
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000105 for (tp = type_list; tp; tp = tp->tp_next)
106 fprintf(f, "%s alloc'd: %" PY_FORMAT_SIZE_T "d, "
107 "freed: %" PY_FORMAT_SIZE_T "d, "
108 "max in use: %" PY_FORMAT_SIZE_T "d\n",
109 tp->tp_name, tp->tp_allocs, tp->tp_frees,
110 tp->tp_maxalloc);
111 fprintf(f, "fast tuple allocs: %" PY_FORMAT_SIZE_T "d, "
112 "empty: %" PY_FORMAT_SIZE_T "d\n",
113 fast_tuple_allocs, tuple_zero_allocs);
114 fprintf(f, "fast int allocs: pos: %" PY_FORMAT_SIZE_T "d, "
115 "neg: %" PY_FORMAT_SIZE_T "d\n",
116 quick_int_allocs, quick_neg_int_allocs);
117 fprintf(f, "null strings: %" PY_FORMAT_SIZE_T "d, "
118 "1-strings: %" PY_FORMAT_SIZE_T "d\n",
119 null_strings, one_strings);
Sjoerd Mullendera9c3c221993-10-11 12:54:31 +0000120}
121
Sjoerd Mullender6ec3c651995-08-29 09:18:14 +0000122PyObject *
Fred Drake100814d2000-07-09 15:48:49 +0000123get_counts(void)
Sjoerd Mullender6ec3c651995-08-29 09:18:14 +0000124{
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000125 PyTypeObject *tp;
126 PyObject *result;
127 PyObject *v;
Sjoerd Mullender6ec3c651995-08-29 09:18:14 +0000128
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000129 result = PyList_New(0);
130 if (result == NULL)
131 return NULL;
132 for (tp = type_list; tp; tp = tp->tp_next) {
133 v = Py_BuildValue("(snnn)", tp->tp_name, tp->tp_allocs,
134 tp->tp_frees, tp->tp_maxalloc);
135 if (v == NULL) {
136 Py_DECREF(result);
137 return NULL;
138 }
139 if (PyList_Append(result, v) < 0) {
140 Py_DECREF(v);
141 Py_DECREF(result);
142 return NULL;
143 }
144 Py_DECREF(v);
145 }
146 return result;
Sjoerd Mullender6ec3c651995-08-29 09:18:14 +0000147}
148
Sjoerd Mullendera9c3c221993-10-11 12:54:31 +0000149void
Fred Drake100814d2000-07-09 15:48:49 +0000150inc_count(PyTypeObject *tp)
Sjoerd Mullendera9c3c221993-10-11 12:54:31 +0000151{
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000152 if (tp->tp_next == NULL && tp->tp_prev == NULL) {
153 /* first time; insert in linked list */
154 if (tp->tp_next != NULL) /* sanity check */
155 Py_FatalError("XXX inc_count sanity check");
156 if (type_list)
157 type_list->tp_prev = tp;
158 tp->tp_next = type_list;
159 /* Note that as of Python 2.2, heap-allocated type objects
160 * can go away, but this code requires that they stay alive
161 * until program exit. That's why we're careful with
162 * refcounts here. type_list gets a new reference to tp,
163 * while ownership of the reference type_list used to hold
164 * (if any) was transferred to tp->tp_next in the line above.
165 * tp is thus effectively immortal after this.
166 */
167 Py_INCREF(tp);
168 type_list = tp;
Tim Peters3e40c7f2003-03-23 03:04:32 +0000169#ifdef Py_TRACE_REFS
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000170 /* Also insert in the doubly-linked list of all objects,
171 * if not already there.
172 */
173 _Py_AddToAllObjects((PyObject *)tp, 0);
Tim Peters78be7992003-03-23 02:51:01 +0000174#endif
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000175 }
176 tp->tp_allocs++;
177 if (tp->tp_allocs - tp->tp_frees > tp->tp_maxalloc)
178 tp->tp_maxalloc = tp->tp_allocs - tp->tp_frees;
Sjoerd Mullendera9c3c221993-10-11 12:54:31 +0000179}
Thomas Wouters49fd7fa2006-04-21 10:40:58 +0000180
181void dec_count(PyTypeObject *tp)
182{
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000183 tp->tp_frees++;
184 if (unlist_types_without_objects &&
185 tp->tp_allocs == tp->tp_frees) {
186 /* unlink the type from type_list */
187 if (tp->tp_prev)
188 tp->tp_prev->tp_next = tp->tp_next;
189 else
190 type_list = tp->tp_next;
191 if (tp->tp_next)
192 tp->tp_next->tp_prev = tp->tp_prev;
193 tp->tp_next = tp->tp_prev = NULL;
194 Py_DECREF(tp);
195 }
Thomas Wouters49fd7fa2006-04-21 10:40:58 +0000196}
197
Sjoerd Mullendera9c3c221993-10-11 12:54:31 +0000198#endif
199
Tim Peters7c321a82002-07-09 02:57:01 +0000200#ifdef Py_REF_DEBUG
201/* Log a fatal error; doesn't return. */
202void
203_Py_NegativeRefcount(const char *fname, int lineno, PyObject *op)
204{
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000205 char buf[300];
Tim Peters7c321a82002-07-09 02:57:01 +0000206
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000207 PyOS_snprintf(buf, sizeof(buf),
208 "%s:%i object at %p has negative ref count "
209 "%" PY_FORMAT_SIZE_T "d",
210 fname, lineno, op, op->ob_refcnt);
211 Py_FatalError(buf);
Tim Peters7c321a82002-07-09 02:57:01 +0000212}
213
214#endif /* Py_REF_DEBUG */
215
Thomas Heller1328b522004-04-22 17:23:49 +0000216void
217Py_IncRef(PyObject *o)
218{
219 Py_XINCREF(o);
220}
221
222void
223Py_DecRef(PyObject *o)
224{
225 Py_XDECREF(o);
226}
227
Guido van Rossumc0b618a1997-05-02 03:12:38 +0000228PyObject *
Fred Drake100814d2000-07-09 15:48:49 +0000229PyObject_Init(PyObject *op, PyTypeObject *tp)
Guido van Rossum85a5fbb1990-10-14 12:07:46 +0000230{
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000231 if (op == NULL)
232 return PyErr_NoMemory();
233 /* Any changes should be reflected in PyObject_INIT (objimpl.h) */
234 Py_TYPE(op) = tp;
235 _Py_NewReference(op);
236 return op;
Guido van Rossum85a5fbb1990-10-14 12:07:46 +0000237}
238
Guido van Rossumb18618d2000-05-03 23:44:39 +0000239PyVarObject *
Martin v. Löwis18e16552006-02-15 17:27:45 +0000240PyObject_InitVar(PyVarObject *op, PyTypeObject *tp, Py_ssize_t size)
Guido van Rossumb18618d2000-05-03 23:44:39 +0000241{
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000242 if (op == NULL)
243 return (PyVarObject *) PyErr_NoMemory();
244 /* Any changes should be reflected in PyObject_INIT_VAR */
245 op->ob_size = size;
246 Py_TYPE(op) = tp;
247 _Py_NewReference((PyObject *)op);
248 return op;
Guido van Rossumb18618d2000-05-03 23:44:39 +0000249}
250
251PyObject *
Fred Drake100814d2000-07-09 15:48:49 +0000252_PyObject_New(PyTypeObject *tp)
Guido van Rossumb18618d2000-05-03 23:44:39 +0000253{
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000254 PyObject *op;
255 op = (PyObject *) PyObject_MALLOC(_PyObject_SIZE(tp));
256 if (op == NULL)
257 return PyErr_NoMemory();
258 return PyObject_INIT(op, tp);
Guido van Rossumb18618d2000-05-03 23:44:39 +0000259}
260
Guido van Rossumd0c87ee1997-05-15 21:31:03 +0000261PyVarObject *
Martin v. Löwis18e16552006-02-15 17:27:45 +0000262_PyObject_NewVar(PyTypeObject *tp, Py_ssize_t nitems)
Guido van Rossum85a5fbb1990-10-14 12:07:46 +0000263{
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000264 PyVarObject *op;
265 const size_t size = _PyObject_VAR_SIZE(tp, nitems);
266 op = (PyVarObject *) PyObject_MALLOC(size);
267 if (op == NULL)
268 return (PyVarObject *)PyErr_NoMemory();
269 return PyObject_INIT_VAR(op, tp, nitems);
Guido van Rossumb18618d2000-05-03 23:44:39 +0000270}
271
Antoine Pitrou796564c2013-07-30 19:59:21 +0200272void
273PyObject_CallFinalizer(PyObject *self)
274{
275 PyTypeObject *tp = Py_TYPE(self);
276
277 /* The former could happen on heaptypes created from the C API, e.g.
278 PyType_FromSpec(). */
279 if (!PyType_HasFeature(tp, Py_TPFLAGS_HAVE_FINALIZE) ||
280 tp->tp_finalize == NULL)
281 return;
282 /* tp_finalize should only be called once. */
283 if (PyType_IS_GC(tp) && _PyGC_FINALIZED(self))
284 return;
285
286 tp->tp_finalize(self);
287 if (PyType_IS_GC(tp))
288 _PyGC_SET_FINALIZED(self, 1);
289}
290
291int
292PyObject_CallFinalizerFromDealloc(PyObject *self)
293{
294 Py_ssize_t refcnt;
295
296 /* Temporarily resurrect the object. */
297 if (self->ob_refcnt != 0) {
298 Py_FatalError("PyObject_CallFinalizerFromDealloc called on "
299 "object with a non-zero refcount");
300 }
301 self->ob_refcnt = 1;
302
303 PyObject_CallFinalizer(self);
304
305 /* Undo the temporary resurrection; can't use DECREF here, it would
306 * cause a recursive call.
307 */
308 assert(self->ob_refcnt > 0);
309 if (--self->ob_refcnt == 0)
310 return 0; /* this is the normal path out */
311
312 /* tp_finalize resurrected it! Make it look like the original Py_DECREF
313 * never happened.
314 */
315 refcnt = self->ob_refcnt;
316 _Py_NewReference(self);
317 self->ob_refcnt = refcnt;
318
319 if (PyType_IS_GC(Py_TYPE(self))) {
320 assert(_PyGC_REFS(self) != _PyGC_REFS_UNTRACKED);
321 }
322 /* If Py_REF_DEBUG, _Py_NewReference bumped _Py_RefTotal, so
323 * we need to undo that. */
324 _Py_DEC_REFTOTAL;
325 /* If Py_TRACE_REFS, _Py_NewReference re-added self to the object
326 * chain, so no more to do there.
327 * If COUNT_ALLOCS, the original decref bumped tp_frees, and
328 * _Py_NewReference bumped tp_allocs: both of those need to be
329 * undone.
330 */
331#ifdef COUNT_ALLOCS
332 --Py_TYPE(self)->tp_frees;
333 --Py_TYPE(self)->tp_allocs;
334#endif
335 return -1;
336}
337
Antoine Pitrouc47bd4a2010-07-27 22:08:27 +0000338int
339PyObject_Print(PyObject *op, FILE *fp, int flags)
Guido van Rossum85a5fbb1990-10-14 12:07:46 +0000340{
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000341 int ret = 0;
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000342 if (PyErr_CheckSignals())
343 return -1;
Guido van Rossum9b00dfa1998-04-28 16:06:54 +0000344#ifdef USE_STACKCHECK
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000345 if (PyOS_CheckStack()) {
346 PyErr_SetString(PyExc_MemoryError, "stack overflow");
347 return -1;
348 }
Guido van Rossum9b00dfa1998-04-28 16:06:54 +0000349#endif
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000350 clearerr(fp); /* Clear any previous error condition */
351 if (op == NULL) {
352 Py_BEGIN_ALLOW_THREADS
353 fprintf(fp, "<nil>");
354 Py_END_ALLOW_THREADS
355 }
356 else {
357 if (op->ob_refcnt <= 0)
358 /* XXX(twouters) cast refcount to long until %zd is
359 universally available */
360 Py_BEGIN_ALLOW_THREADS
361 fprintf(fp, "<refcnt %ld at %p>",
362 (long)op->ob_refcnt, op);
363 Py_END_ALLOW_THREADS
364 else {
365 PyObject *s;
366 if (flags & Py_PRINT_RAW)
367 s = PyObject_Str(op);
368 else
369 s = PyObject_Repr(op);
370 if (s == NULL)
371 ret = -1;
372 else if (PyBytes_Check(s)) {
373 fwrite(PyBytes_AS_STRING(s), 1,
374 PyBytes_GET_SIZE(s), fp);
375 }
376 else if (PyUnicode_Check(s)) {
377 PyObject *t;
Martin v. Löwisd63a3b82011-09-28 07:41:54 +0200378 t = PyUnicode_AsEncodedString(s, "utf-8", "backslashreplace");
Miss Islington (bot)49fb49d2018-10-06 00:07:12 -0700379 if (t == NULL) {
380 ret = -1;
381 }
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000382 else {
383 fwrite(PyBytes_AS_STRING(t), 1,
384 PyBytes_GET_SIZE(t), fp);
Victor Stinnerba6b4302010-05-17 09:33:42 +0000385 Py_DECREF(t);
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000386 }
387 }
388 else {
389 PyErr_Format(PyExc_TypeError,
390 "str() or repr() returned '%.100s'",
391 s->ob_type->tp_name);
392 ret = -1;
393 }
394 Py_XDECREF(s);
395 }
396 }
397 if (ret == 0) {
398 if (ferror(fp)) {
Serhiy Storchaka55fe1ae2017-04-16 10:46:38 +0300399 PyErr_SetFromErrno(PyExc_OSError);
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000400 clearerr(fp);
401 ret = -1;
402 }
403 }
404 return ret;
Guido van Rossum85a5fbb1990-10-14 12:07:46 +0000405}
406
Guido van Rossum38938152006-08-21 23:36:26 +0000407/* For debugging convenience. Set a breakpoint here and call it from your DLL */
408void
Thomas Woutersb2137042007-02-01 18:02:27 +0000409_Py_BreakPoint(void)
Guido van Rossum38938152006-08-21 23:36:26 +0000410{
411}
412
Neal Norwitz1a997502003-01-13 20:13:12 +0000413
Victor Stinner9e23f0a2019-04-11 22:30:31 +0200414/* Heuristic checking if the object memory is uninitialized or deallocated.
415 Rely on the debug hooks on Python memory allocators:
416 see _PyMem_IsPtrFreed().
Victor Stinner95036ea2018-11-22 17:15:37 +0100417
418 The function can be used to prevent segmentation fault on dereferencing
Victor Stinner9e23f0a2019-04-11 22:30:31 +0200419 pointers like 0xDDDDDDDDDDDDDDDD. */
Victor Stinner95036ea2018-11-22 17:15:37 +0100420int
421_PyObject_IsFreed(PyObject *op)
422{
Victor Stinner9e23f0a2019-04-11 22:30:31 +0200423 if (_PyMem_IsPtrFreed(op) || _PyMem_IsPtrFreed(op->ob_type)) {
Victor Stinner95036ea2018-11-22 17:15:37 +0100424 return 1;
425 }
Victor Stinner9e23f0a2019-04-11 22:30:31 +0200426 /* ignore op->ob_ref: its value can have be modified
Victor Stinner95036ea2018-11-22 17:15:37 +0100427 by Py_INCREF() and Py_DECREF(). */
428#ifdef Py_TRACE_REFS
Victor Stinner9e23f0a2019-04-11 22:30:31 +0200429 if (_PyMem_IsPtrFreed(op->_ob_next) || _PyMem_IsPtrFreed(op->_ob_prev)) {
430 return 1;
431 }
Victor Stinner95036ea2018-11-22 17:15:37 +0100432#endif
Victor Stinner9e23f0a2019-04-11 22:30:31 +0200433 return 0;
Victor Stinner95036ea2018-11-22 17:15:37 +0100434}
435
436
Barry Warsaw9bf16442001-01-23 16:24:35 +0000437/* For debugging convenience. See Misc/gdbinit for some useful gdb hooks */
Guido van Rossum38938152006-08-21 23:36:26 +0000438void
439_PyObject_Dump(PyObject* op)
Barry Warsaw9bf16442001-01-23 16:24:35 +0000440{
Victor Stinner95036ea2018-11-22 17:15:37 +0100441 if (op == NULL) {
442 fprintf(stderr, "<NULL object>\n");
443 fflush(stderr);
444 return;
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000445 }
Victor Stinner95036ea2018-11-22 17:15:37 +0100446
447 if (_PyObject_IsFreed(op)) {
448 /* It seems like the object memory has been freed:
449 don't access it to prevent a segmentation fault. */
Victor Stinner9e23f0a2019-04-11 22:30:31 +0200450 fprintf(stderr, "<Freed object>\n");
Victor Stinner95036ea2018-11-22 17:15:37 +0100451 return;
452 }
453
454 PyGILState_STATE gil;
455 PyObject *error_type, *error_value, *error_traceback;
456
457 fprintf(stderr, "object : ");
458 fflush(stderr);
459 gil = PyGILState_Ensure();
460
461 PyErr_Fetch(&error_type, &error_value, &error_traceback);
462 (void)PyObject_Print(op, stderr, 0);
463 fflush(stderr);
464 PyErr_Restore(error_type, error_value, error_traceback);
465
466 PyGILState_Release(gil);
467 /* XXX(twouters) cast refcount to long until %zd is
468 universally available */
469 fprintf(stderr, "\n"
470 "type : %s\n"
471 "refcount: %ld\n"
472 "address : %p\n",
473 Py_TYPE(op)==NULL ? "NULL" : Py_TYPE(op)->tp_name,
474 (long)op->ob_refcnt,
475 op);
476 fflush(stderr);
Barry Warsaw9bf16442001-01-23 16:24:35 +0000477}
Barry Warsaw903138f2001-01-23 16:33:18 +0000478
Guido van Rossumc0b618a1997-05-02 03:12:38 +0000479PyObject *
Fred Drake100814d2000-07-09 15:48:49 +0000480PyObject_Repr(PyObject *v)
Guido van Rossum85a5fbb1990-10-14 12:07:46 +0000481{
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000482 PyObject *res;
483 if (PyErr_CheckSignals())
484 return NULL;
Guido van Rossum9b00dfa1998-04-28 16:06:54 +0000485#ifdef USE_STACKCHECK
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000486 if (PyOS_CheckStack()) {
487 PyErr_SetString(PyExc_MemoryError, "stack overflow");
488 return NULL;
489 }
Guido van Rossum9b00dfa1998-04-28 16:06:54 +0000490#endif
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000491 if (v == NULL)
492 return PyUnicode_FromString("<NULL>");
493 if (Py_TYPE(v)->tp_repr == NULL)
494 return PyUnicode_FromFormat("<%s object at %p>",
495 v->ob_type->tp_name, v);
Victor Stinner33824f62013-08-26 14:05:19 +0200496
497#ifdef Py_DEBUG
498 /* PyObject_Repr() must not be called with an exception set,
Victor Stinnera8cb5152017-01-18 14:12:51 +0100499 because it can clear it (directly or indirectly) and so the
Martin Panter9955a372015-10-07 10:26:23 +0000500 caller loses its exception */
Victor Stinner33824f62013-08-26 14:05:19 +0200501 assert(!PyErr_Occurred());
502#endif
503
Serhiy Storchaka1fb72d22017-12-03 22:12:11 +0200504 /* It is possible for a type to have a tp_repr representation that loops
505 infinitely. */
506 if (Py_EnterRecursiveCall(" while getting the repr of an object"))
507 return NULL;
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000508 res = (*v->ob_type->tp_repr)(v);
Serhiy Storchaka1fb72d22017-12-03 22:12:11 +0200509 Py_LeaveRecursiveCall();
Victor Stinner0a54cf12011-12-01 03:22:44 +0100510 if (res == NULL)
511 return NULL;
512 if (!PyUnicode_Check(res)) {
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000513 PyErr_Format(PyExc_TypeError,
514 "__repr__ returned non-string (type %.200s)",
515 res->ob_type->tp_name);
516 Py_DECREF(res);
517 return NULL;
518 }
Victor Stinnerdb88ae52011-12-01 02:15:00 +0100519#ifndef Py_DEBUG
520 if (PyUnicode_READY(res) < 0)
521 return NULL;
522#endif
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000523 return res;
Guido van Rossum85a5fbb1990-10-14 12:07:46 +0000524}
525
Guido van Rossumc0b618a1997-05-02 03:12:38 +0000526PyObject *
Guido van Rossum98297ee2007-11-06 21:34:58 +0000527PyObject_Str(PyObject *v)
Guido van Rossumc6004111993-11-05 10:22:19 +0000528{
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000529 PyObject *res;
530 if (PyErr_CheckSignals())
531 return NULL;
Guido van Rossum98297ee2007-11-06 21:34:58 +0000532#ifdef USE_STACKCHECK
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000533 if (PyOS_CheckStack()) {
534 PyErr_SetString(PyExc_MemoryError, "stack overflow");
535 return NULL;
536 }
Guido van Rossum98297ee2007-11-06 21:34:58 +0000537#endif
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000538 if (v == NULL)
539 return PyUnicode_FromString("<NULL>");
540 if (PyUnicode_CheckExact(v)) {
Victor Stinnerdb88ae52011-12-01 02:15:00 +0100541#ifndef Py_DEBUG
Victor Stinner4ead7c72011-11-20 19:48:36 +0100542 if (PyUnicode_READY(v) < 0)
543 return NULL;
Victor Stinnerdb88ae52011-12-01 02:15:00 +0100544#endif
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000545 Py_INCREF(v);
546 return v;
547 }
548 if (Py_TYPE(v)->tp_str == NULL)
549 return PyObject_Repr(v);
Guido van Rossum4f288ab2001-05-01 16:53:37 +0000550
Victor Stinner33824f62013-08-26 14:05:19 +0200551#ifdef Py_DEBUG
552 /* PyObject_Str() must not be called with an exception set,
Victor Stinnera8cb5152017-01-18 14:12:51 +0100553 because it can clear it (directly or indirectly) and so the
Nick Coghland979e432014-02-09 10:43:21 +1000554 caller loses its exception */
Victor Stinner33824f62013-08-26 14:05:19 +0200555 assert(!PyErr_Occurred());
556#endif
557
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000558 /* It is possible for a type to have a tp_str representation that loops
559 infinitely. */
560 if (Py_EnterRecursiveCall(" while getting the str of an object"))
561 return NULL;
562 res = (*Py_TYPE(v)->tp_str)(v);
563 Py_LeaveRecursiveCall();
564 if (res == NULL)
565 return NULL;
566 if (!PyUnicode_Check(res)) {
567 PyErr_Format(PyExc_TypeError,
568 "__str__ returned non-string (type %.200s)",
569 Py_TYPE(res)->tp_name);
570 Py_DECREF(res);
571 return NULL;
572 }
Victor Stinnerdb88ae52011-12-01 02:15:00 +0100573#ifndef Py_DEBUG
Victor Stinner4ead7c72011-11-20 19:48:36 +0100574 if (PyUnicode_READY(res) < 0)
575 return NULL;
Victor Stinnerdb88ae52011-12-01 02:15:00 +0100576#endif
Victor Stinner4ead7c72011-11-20 19:48:36 +0100577 assert(_PyUnicode_CheckConsistency(res, 1));
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000578 return res;
Neil Schemenauercf52c072005-08-12 17:34:58 +0000579}
580
Georg Brandl559e5d72008-06-11 18:37:52 +0000581PyObject *
582PyObject_ASCII(PyObject *v)
583{
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000584 PyObject *repr, *ascii, *res;
Georg Brandl559e5d72008-06-11 18:37:52 +0000585
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000586 repr = PyObject_Repr(v);
587 if (repr == NULL)
588 return NULL;
Georg Brandl559e5d72008-06-11 18:37:52 +0000589
Victor Stinneraf037572013-04-14 18:44:10 +0200590 if (PyUnicode_IS_ASCII(repr))
591 return repr;
592
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000593 /* repr is guaranteed to be a PyUnicode object by PyObject_Repr */
Martin v. Löwisd63a3b82011-09-28 07:41:54 +0200594 ascii = _PyUnicode_AsASCIIString(repr, "backslashreplace");
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000595 Py_DECREF(repr);
596 if (ascii == NULL)
597 return NULL;
Georg Brandl559e5d72008-06-11 18:37:52 +0000598
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000599 res = PyUnicode_DecodeASCII(
600 PyBytes_AS_STRING(ascii),
601 PyBytes_GET_SIZE(ascii),
602 NULL);
603
604 Py_DECREF(ascii);
605 return res;
Georg Brandl559e5d72008-06-11 18:37:52 +0000606}
Guido van Rossuma3af41d2001-01-18 22:07:06 +0000607
Benjamin Petersonc15a0732008-08-26 16:46:47 +0000608PyObject *
609PyObject_Bytes(PyObject *v)
610{
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000611 PyObject *result, *func;
Benjamin Petersonc15a0732008-08-26 16:46:47 +0000612
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000613 if (v == NULL)
614 return PyBytes_FromString("<NULL>");
Benjamin Petersonc15a0732008-08-26 16:46:47 +0000615
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000616 if (PyBytes_CheckExact(v)) {
617 Py_INCREF(v);
618 return v;
619 }
Benjamin Petersonc15a0732008-08-26 16:46:47 +0000620
Benjamin Petersonce798522012-01-22 11:24:29 -0500621 func = _PyObject_LookupSpecial(v, &PyId___bytes__);
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000622 if (func != NULL) {
Victor Stinnerf17c3de2016-12-06 18:46:19 +0100623 result = _PyObject_CallNoArg(func);
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000624 Py_DECREF(func);
625 if (result == NULL)
Benjamin Peterson41ece392010-09-11 16:39:57 +0000626 return NULL;
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000627 if (!PyBytes_Check(result)) {
Benjamin Peterson41ece392010-09-11 16:39:57 +0000628 PyErr_Format(PyExc_TypeError,
629 "__bytes__ returned non-bytes (type %.200s)",
630 Py_TYPE(result)->tp_name);
631 Py_DECREF(result);
632 return NULL;
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000633 }
634 return result;
635 }
636 else if (PyErr_Occurred())
637 return NULL;
638 return PyBytes_FromObject(v);
Benjamin Petersonc15a0732008-08-26 16:46:47 +0000639}
640
Mark Dickinsonc008a172009-02-01 13:59:22 +0000641/* For Python 3.0.1 and later, the old three-way comparison has been
642 completely removed in favour of rich comparisons. PyObject_Compare() and
643 PyObject_Cmp() are gone, and the builtin cmp function no longer exists.
Mark Dickinsone94c6792009-02-02 20:36:42 +0000644 The old tp_compare slot has been renamed to tp_reserved, and should no
Mark Dickinsonc008a172009-02-01 13:59:22 +0000645 longer be used. Use tp_richcompare instead.
Guido van Rossum98297ee2007-11-06 21:34:58 +0000646
Guido van Rossum47b9ff62006-08-24 00:41:19 +0000647 See (*) below for practical amendments.
648
Mark Dickinsonc008a172009-02-01 13:59:22 +0000649 tp_richcompare gets called with a first argument of the appropriate type
650 and a second object of an arbitrary type. We never do any kind of
651 coercion.
Guido van Rossum47b9ff62006-08-24 00:41:19 +0000652
Mark Dickinsonc008a172009-02-01 13:59:22 +0000653 The tp_richcompare slot should return an object, as follows:
Guido van Rossum47b9ff62006-08-24 00:41:19 +0000654
655 NULL if an exception occurred
656 NotImplemented if the requested comparison is not implemented
657 any other false value if the requested comparison is false
658 any other true value if the requested comparison is true
659
660 The PyObject_RichCompare[Bool]() wrappers raise TypeError when they get
661 NotImplemented.
662
663 (*) Practical amendments:
664
665 - If rich comparison returns NotImplemented, == and != are decided by
666 comparing the object pointer (i.e. falling back to the base object
667 implementation).
668
Guido van Rossuma4073002002-05-31 20:03:54 +0000669*/
Guido van Rossum47b9ff62006-08-24 00:41:19 +0000670
Guido van Rossum47b9ff62006-08-24 00:41:19 +0000671/* Map rich comparison operators to their swapped version, e.g. LT <--> GT */
Brett Cannona5ca2e72004-09-25 01:37:24 +0000672int _Py_SwappedOp[] = {Py_GT, Py_GE, Py_EQ, Py_NE, Py_LT, Py_LE};
Neil Schemenauer5ed85ec2001-01-04 01:48:10 +0000673
Serhiy Storchaka2d06e842015-12-25 19:53:18 +0200674static const char * const opstrings[] = {"<", "<=", "==", "!=", ">", ">="};
Guido van Rossum47b9ff62006-08-24 00:41:19 +0000675
676/* Perform a rich comparison, raising TypeError when the requested comparison
677 operator is not supported. */
Guido van Rossume797ec12001-01-17 15:24:28 +0000678static PyObject *
Guido van Rossum47b9ff62006-08-24 00:41:19 +0000679do_richcompare(PyObject *v, PyObject *w, int op)
Guido van Rossume797ec12001-01-17 15:24:28 +0000680{
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000681 richcmpfunc f;
682 PyObject *res;
683 int checked_reverse_op = 0;
Guido van Rossume797ec12001-01-17 15:24:28 +0000684
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000685 if (v->ob_type != w->ob_type &&
686 PyType_IsSubtype(w->ob_type, v->ob_type) &&
687 (f = w->ob_type->tp_richcompare) != NULL) {
688 checked_reverse_op = 1;
689 res = (*f)(w, v, _Py_SwappedOp[op]);
690 if (res != Py_NotImplemented)
691 return res;
692 Py_DECREF(res);
693 }
694 if ((f = v->ob_type->tp_richcompare) != NULL) {
695 res = (*f)(v, w, op);
696 if (res != Py_NotImplemented)
697 return res;
698 Py_DECREF(res);
699 }
700 if (!checked_reverse_op && (f = w->ob_type->tp_richcompare) != NULL) {
701 res = (*f)(w, v, _Py_SwappedOp[op]);
702 if (res != Py_NotImplemented)
703 return res;
704 Py_DECREF(res);
705 }
706 /* If neither object implements it, provide a sensible default
707 for == and !=, but raise an exception for ordering. */
708 switch (op) {
709 case Py_EQ:
710 res = (v == w) ? Py_True : Py_False;
711 break;
712 case Py_NE:
713 res = (v != w) ? Py_True : Py_False;
714 break;
715 default:
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000716 PyErr_Format(PyExc_TypeError,
Victor Stinner91108f02015-10-14 18:25:31 +0200717 "'%s' not supported between instances of '%.100s' and '%.100s'",
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000718 opstrings[op],
Victor Stinner91108f02015-10-14 18:25:31 +0200719 v->ob_type->tp_name,
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000720 w->ob_type->tp_name);
721 return NULL;
722 }
723 Py_INCREF(res);
724 return res;
Guido van Rossume797ec12001-01-17 15:24:28 +0000725}
726
Guido van Rossum47b9ff62006-08-24 00:41:19 +0000727/* Perform a rich comparison with object result. This wraps do_richcompare()
728 with a check for NULL arguments and a recursion check. */
Guido van Rossume797ec12001-01-17 15:24:28 +0000729
Guido van Rossume797ec12001-01-17 15:24:28 +0000730PyObject *
731PyObject_RichCompare(PyObject *v, PyObject *w, int op)
732{
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000733 PyObject *res;
Guido van Rossume797ec12001-01-17 15:24:28 +0000734
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000735 assert(Py_LT <= op && op <= Py_GE);
736 if (v == NULL || w == NULL) {
737 if (!PyErr_Occurred())
738 PyErr_BadInternalCall();
739 return NULL;
740 }
741 if (Py_EnterRecursiveCall(" in comparison"))
742 return NULL;
743 res = do_richcompare(v, w, op);
744 Py_LeaveRecursiveCall();
745 return res;
Guido van Rossume797ec12001-01-17 15:24:28 +0000746}
747
Guido van Rossum47b9ff62006-08-24 00:41:19 +0000748/* Perform a rich comparison with integer result. This wraps
749 PyObject_RichCompare(), returning -1 for error, 0 for false, 1 for true. */
Guido van Rossume797ec12001-01-17 15:24:28 +0000750int
751PyObject_RichCompareBool(PyObject *v, PyObject *w, int op)
752{
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000753 PyObject *res;
754 int ok;
Guido van Rossume797ec12001-01-17 15:24:28 +0000755
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000756 /* Quick result when objects are the same.
757 Guarantees that identity implies equality. */
758 if (v == w) {
759 if (op == Py_EQ)
760 return 1;
761 else if (op == Py_NE)
762 return 0;
763 }
Mark Dickinson4a1f5932008-11-12 23:23:36 +0000764
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000765 res = PyObject_RichCompare(v, w, op);
766 if (res == NULL)
767 return -1;
768 if (PyBool_Check(res))
769 ok = (res == Py_True);
770 else
771 ok = PyObject_IsTrue(res);
772 Py_DECREF(res);
773 return ok;
Guido van Rossume797ec12001-01-17 15:24:28 +0000774}
Fred Drake13634cf2000-06-29 19:17:04 +0000775
Antoine Pitrouce4a9da2011-11-21 20:46:33 +0100776Py_hash_t
Nick Coghland1abd252008-07-15 15:46:38 +0000777PyObject_HashNotImplemented(PyObject *v)
778{
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000779 PyErr_Format(PyExc_TypeError, "unhashable type: '%.200s'",
780 Py_TYPE(v)->tp_name);
781 return -1;
Nick Coghland1abd252008-07-15 15:46:38 +0000782}
Fred Drake13634cf2000-06-29 19:17:04 +0000783
Benjamin Peterson8f67d082010-10-17 20:54:53 +0000784Py_hash_t
Fred Drake100814d2000-07-09 15:48:49 +0000785PyObject_Hash(PyObject *v)
Guido van Rossum9bfef441993-03-29 10:43:31 +0000786{
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000787 PyTypeObject *tp = Py_TYPE(v);
788 if (tp->tp_hash != NULL)
789 return (*tp->tp_hash)(v);
790 /* To keep to the general practice that inheriting
791 * solely from object in C code should work without
792 * an explicit call to PyType_Ready, we implicitly call
793 * PyType_Ready here and then check the tp_hash slot again
794 */
795 if (tp->tp_dict == NULL) {
796 if (PyType_Ready(tp) < 0)
797 return -1;
798 if (tp->tp_hash != NULL)
799 return (*tp->tp_hash)(v);
800 }
801 /* Otherwise, the object can't be hashed */
802 return PyObject_HashNotImplemented(v);
Guido van Rossum9bfef441993-03-29 10:43:31 +0000803}
804
Guido van Rossumc0b618a1997-05-02 03:12:38 +0000805PyObject *
Jeremy Hyltonaf68c872005-12-10 18:50:16 +0000806PyObject_GetAttrString(PyObject *v, const char *name)
Guido van Rossum3f5da241990-12-20 15:06:42 +0000807{
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000808 PyObject *w, *res;
Guido van Rossumd8eb1b31996-08-09 20:52:03 +0000809
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000810 if (Py_TYPE(v)->tp_getattr != NULL)
811 return (*Py_TYPE(v)->tp_getattr)(v, (char*)name);
INADA Naoki3e8d6cb2017-02-21 23:57:25 +0900812 w = PyUnicode_FromString(name);
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000813 if (w == NULL)
814 return NULL;
815 res = PyObject_GetAttr(v, w);
Victor Stinner59af08f2012-03-22 02:09:08 +0100816 Py_DECREF(w);
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000817 return res;
Guido van Rossum3f5da241990-12-20 15:06:42 +0000818}
819
820int
Jeremy Hyltonaf68c872005-12-10 18:50:16 +0000821PyObject_HasAttrString(PyObject *v, const char *name)
Guido van Rossumed18fdc1993-07-11 19:55:34 +0000822{
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000823 PyObject *res = PyObject_GetAttrString(v, name);
824 if (res != NULL) {
825 Py_DECREF(res);
826 return 1;
827 }
828 PyErr_Clear();
829 return 0;
Guido van Rossumed18fdc1993-07-11 19:55:34 +0000830}
831
832int
Jeremy Hyltonaf68c872005-12-10 18:50:16 +0000833PyObject_SetAttrString(PyObject *v, const char *name, PyObject *w)
Guido van Rossum3f5da241990-12-20 15:06:42 +0000834{
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000835 PyObject *s;
836 int res;
Guido van Rossumd8eb1b31996-08-09 20:52:03 +0000837
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000838 if (Py_TYPE(v)->tp_setattr != NULL)
839 return (*Py_TYPE(v)->tp_setattr)(v, (char*)name, w);
840 s = PyUnicode_InternFromString(name);
841 if (s == NULL)
842 return -1;
843 res = PyObject_SetAttr(v, s, w);
844 Py_XDECREF(s);
845 return res;
Guido van Rossum3f5da241990-12-20 15:06:42 +0000846}
847
Benjamin Petersonbfebb7b2011-12-15 15:34:02 -0500848int
849_PyObject_IsAbstract(PyObject *obj)
850{
851 int res;
852 PyObject* isabstract;
Benjamin Petersonbfebb7b2011-12-15 15:34:02 -0500853
854 if (obj == NULL)
855 return 0;
856
Serhiy Storchakaf320be72018-01-25 10:49:40 +0200857 res = _PyObject_LookupAttrId(obj, &PyId___isabstractmethod__, &isabstract);
858 if (res > 0) {
859 res = PyObject_IsTrue(isabstract);
860 Py_DECREF(isabstract);
Benjamin Petersonbfebb7b2011-12-15 15:34:02 -0500861 }
Benjamin Petersonbfebb7b2011-12-15 15:34:02 -0500862 return res;
863}
864
Guido van Rossum98ff96a1997-05-20 18:34:44 +0000865PyObject *
Martin v. Löwisafe55bb2011-10-09 10:38:36 +0200866_PyObject_GetAttrId(PyObject *v, _Py_Identifier *name)
867{
868 PyObject *result;
Martin v. Löwisd10759f2011-11-07 13:00:05 +0100869 PyObject *oname = _PyUnicode_FromId(name); /* borrowed */
Martin v. Löwisafe55bb2011-10-09 10:38:36 +0200870 if (!oname)
871 return NULL;
872 result = PyObject_GetAttr(v, oname);
Martin v. Löwisafe55bb2011-10-09 10:38:36 +0200873 return result;
874}
875
876int
877_PyObject_HasAttrId(PyObject *v, _Py_Identifier *name)
878{
879 int result;
Martin v. Löwisd10759f2011-11-07 13:00:05 +0100880 PyObject *oname = _PyUnicode_FromId(name); /* borrowed */
Martin v. Löwisafe55bb2011-10-09 10:38:36 +0200881 if (!oname)
882 return -1;
883 result = PyObject_HasAttr(v, oname);
Martin v. Löwisafe55bb2011-10-09 10:38:36 +0200884 return result;
885}
886
887int
888_PyObject_SetAttrId(PyObject *v, _Py_Identifier *name, PyObject *w)
889{
890 int result;
Martin v. Löwisd10759f2011-11-07 13:00:05 +0100891 PyObject *oname = _PyUnicode_FromId(name); /* borrowed */
Martin v. Löwisafe55bb2011-10-09 10:38:36 +0200892 if (!oname)
893 return -1;
894 result = PyObject_SetAttr(v, oname, w);
Martin v. Löwisafe55bb2011-10-09 10:38:36 +0200895 return result;
896}
897
898PyObject *
Fred Drake100814d2000-07-09 15:48:49 +0000899PyObject_GetAttr(PyObject *v, PyObject *name)
Guido van Rossum98ff96a1997-05-20 18:34:44 +0000900{
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000901 PyTypeObject *tp = Py_TYPE(v);
Tim Peters6d6c1a32001-08-02 04:15:00 +0000902
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000903 if (!PyUnicode_Check(name)) {
904 PyErr_Format(PyExc_TypeError,
905 "attribute name must be string, not '%.200s'",
906 name->ob_type->tp_name);
907 return NULL;
908 }
909 if (tp->tp_getattro != NULL)
910 return (*tp->tp_getattro)(v, name);
911 if (tp->tp_getattr != NULL) {
Serhiy Storchaka2a404b62017-01-22 23:07:07 +0200912 const char *name_str = PyUnicode_AsUTF8(name);
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000913 if (name_str == NULL)
914 return NULL;
Serhiy Storchaka2a404b62017-01-22 23:07:07 +0200915 return (*tp->tp_getattr)(v, (char *)name_str);
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000916 }
917 PyErr_Format(PyExc_AttributeError,
918 "'%.50s' object has no attribute '%U'",
919 tp->tp_name, name);
920 return NULL;
Guido van Rossum98ff96a1997-05-20 18:34:44 +0000921}
922
Serhiy Storchakaf320be72018-01-25 10:49:40 +0200923int
924_PyObject_LookupAttr(PyObject *v, PyObject *name, PyObject **result)
INADA Naoki378edee2018-01-16 20:52:41 +0900925{
926 PyTypeObject *tp = Py_TYPE(v);
INADA Naoki378edee2018-01-16 20:52:41 +0900927
928 if (!PyUnicode_Check(name)) {
929 PyErr_Format(PyExc_TypeError,
930 "attribute name must be string, not '%.200s'",
931 name->ob_type->tp_name);
Serhiy Storchakaf320be72018-01-25 10:49:40 +0200932 *result = NULL;
933 return -1;
INADA Naoki378edee2018-01-16 20:52:41 +0900934 }
935
936 if (tp->tp_getattro == PyObject_GenericGetAttr) {
Serhiy Storchakaf320be72018-01-25 10:49:40 +0200937 *result = _PyObject_GenericGetAttrWithDict(v, name, NULL, 1);
938 if (*result != NULL) {
939 return 1;
940 }
941 if (PyErr_Occurred()) {
942 return -1;
943 }
944 return 0;
INADA Naoki378edee2018-01-16 20:52:41 +0900945 }
946 if (tp->tp_getattro != NULL) {
Serhiy Storchakaf320be72018-01-25 10:49:40 +0200947 *result = (*tp->tp_getattro)(v, name);
INADA Naoki378edee2018-01-16 20:52:41 +0900948 }
949 else if (tp->tp_getattr != NULL) {
950 const char *name_str = PyUnicode_AsUTF8(name);
Serhiy Storchakaf320be72018-01-25 10:49:40 +0200951 if (name_str == NULL) {
952 *result = NULL;
953 return -1;
954 }
955 *result = (*tp->tp_getattr)(v, (char *)name_str);
INADA Naoki378edee2018-01-16 20:52:41 +0900956 }
INADA Naokie76daeb2018-01-26 16:22:51 +0900957 else {
958 *result = NULL;
959 return 0;
960 }
961
Serhiy Storchakaf320be72018-01-25 10:49:40 +0200962 if (*result != NULL) {
963 return 1;
INADA Naoki378edee2018-01-16 20:52:41 +0900964 }
Serhiy Storchakaf320be72018-01-25 10:49:40 +0200965 if (!PyErr_ExceptionMatches(PyExc_AttributeError)) {
966 return -1;
967 }
968 PyErr_Clear();
969 return 0;
970}
971
972int
973_PyObject_LookupAttrId(PyObject *v, _Py_Identifier *name, PyObject **result)
974{
975 PyObject *oname = _PyUnicode_FromId(name); /* borrowed */
976 if (!oname) {
977 *result = NULL;
978 return -1;
979 }
980 return _PyObject_LookupAttr(v, oname, result);
INADA Naoki378edee2018-01-16 20:52:41 +0900981}
982
Guido van Rossum98ff96a1997-05-20 18:34:44 +0000983int
Fred Drake100814d2000-07-09 15:48:49 +0000984PyObject_HasAttr(PyObject *v, PyObject *name)
Guido van Rossum98ff96a1997-05-20 18:34:44 +0000985{
Serhiy Storchakaf320be72018-01-25 10:49:40 +0200986 PyObject *res;
987 if (_PyObject_LookupAttr(v, name, &res) < 0) {
988 PyErr_Clear();
989 return 0;
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000990 }
Serhiy Storchakaf320be72018-01-25 10:49:40 +0200991 if (res == NULL) {
992 return 0;
993 }
994 Py_DECREF(res);
995 return 1;
Guido van Rossum98ff96a1997-05-20 18:34:44 +0000996}
997
998int
Fred Drake100814d2000-07-09 15:48:49 +0000999PyObject_SetAttr(PyObject *v, PyObject *name, PyObject *value)
Guido van Rossum98ff96a1997-05-20 18:34:44 +00001000{
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001001 PyTypeObject *tp = Py_TYPE(v);
1002 int err;
Marc-André Lemburge44e5072000-09-18 16:20:57 +00001003
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001004 if (!PyUnicode_Check(name)) {
1005 PyErr_Format(PyExc_TypeError,
1006 "attribute name must be string, not '%.200s'",
1007 name->ob_type->tp_name);
1008 return -1;
1009 }
1010 Py_INCREF(name);
Tim Peters6d6c1a32001-08-02 04:15:00 +00001011
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001012 PyUnicode_InternInPlace(&name);
1013 if (tp->tp_setattro != NULL) {
1014 err = (*tp->tp_setattro)(v, name, value);
1015 Py_DECREF(name);
1016 return err;
1017 }
1018 if (tp->tp_setattr != NULL) {
Serhiy Storchaka2a404b62017-01-22 23:07:07 +02001019 const char *name_str = PyUnicode_AsUTF8(name);
Miss Islington (bot)896c6352019-04-28 06:17:40 -07001020 if (name_str == NULL) {
1021 Py_DECREF(name);
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001022 return -1;
Miss Islington (bot)896c6352019-04-28 06:17:40 -07001023 }
Serhiy Storchaka2a404b62017-01-22 23:07:07 +02001024 err = (*tp->tp_setattr)(v, (char *)name_str, value);
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001025 Py_DECREF(name);
1026 return err;
1027 }
1028 Py_DECREF(name);
1029 assert(name->ob_refcnt >= 1);
1030 if (tp->tp_getattr == NULL && tp->tp_getattro == NULL)
1031 PyErr_Format(PyExc_TypeError,
1032 "'%.100s' object has no attributes "
1033 "(%s .%U)",
1034 tp->tp_name,
1035 value==NULL ? "del" : "assign to",
1036 name);
1037 else
1038 PyErr_Format(PyExc_TypeError,
1039 "'%.100s' object has only read-only attributes "
1040 "(%s .%U)",
1041 tp->tp_name,
1042 value==NULL ? "del" : "assign to",
1043 name);
1044 return -1;
Tim Peters6d6c1a32001-08-02 04:15:00 +00001045}
1046
1047/* Helper to get a pointer to an object's __dict__ slot, if any */
1048
1049PyObject **
1050_PyObject_GetDictPtr(PyObject *obj)
1051{
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001052 Py_ssize_t dictoffset;
1053 PyTypeObject *tp = Py_TYPE(obj);
Tim Peters6d6c1a32001-08-02 04:15:00 +00001054
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001055 dictoffset = tp->tp_dictoffset;
1056 if (dictoffset == 0)
1057 return NULL;
1058 if (dictoffset < 0) {
1059 Py_ssize_t tsize;
1060 size_t size;
Guido van Rossum2eb0b872002-03-01 22:24:49 +00001061
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001062 tsize = ((PyVarObject *)obj)->ob_size;
1063 if (tsize < 0)
1064 tsize = -tsize;
1065 size = _PyObject_VAR_SIZE(tp, tsize);
Guido van Rossum2eb0b872002-03-01 22:24:49 +00001066
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001067 dictoffset += (long)size;
1068 assert(dictoffset > 0);
1069 assert(dictoffset % SIZEOF_VOID_P == 0);
1070 }
1071 return (PyObject **) ((char *)obj + dictoffset);
Tim Peters6d6c1a32001-08-02 04:15:00 +00001072}
1073
Tim Peters6d6c1a32001-08-02 04:15:00 +00001074PyObject *
Raymond Hettinger1da1dbf2003-03-17 19:46:11 +00001075PyObject_SelfIter(PyObject *obj)
Raymond Hettinger01538262003-03-17 08:24:35 +00001076{
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001077 Py_INCREF(obj);
1078 return obj;
Raymond Hettinger01538262003-03-17 08:24:35 +00001079}
1080
Antoine Pitroua7013882012-04-05 00:04:20 +02001081/* Convenience function to get a builtin from its name */
1082PyObject *
1083_PyObject_GetBuiltin(const char *name)
1084{
Victor Stinner53e9ec42013-11-07 00:43:05 +01001085 PyObject *mod_name, *mod, *attr;
1086
Victor Stinnerbd303c12013-11-07 23:07:29 +01001087 mod_name = _PyUnicode_FromId(&PyId_builtins); /* borrowed */
Victor Stinner53e9ec42013-11-07 00:43:05 +01001088 if (mod_name == NULL)
1089 return NULL;
1090 mod = PyImport_Import(mod_name);
Antoine Pitroua7013882012-04-05 00:04:20 +02001091 if (mod == NULL)
1092 return NULL;
1093 attr = PyObject_GetAttrString(mod, name);
1094 Py_DECREF(mod);
1095 return attr;
1096}
1097
Amaury Forgeot d'Arcf343e012009-01-12 23:58:21 +00001098/* Helper used when the __next__ method is removed from a type:
1099 tp_iternext is never NULL and can be safely called without checking
1100 on every iteration.
1101 */
1102
1103PyObject *
1104_PyObject_NextNotImplemented(PyObject *self)
1105{
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001106 PyErr_Format(PyExc_TypeError,
1107 "'%.200s' object is not iterable",
1108 Py_TYPE(self)->tp_name);
1109 return NULL;
Amaury Forgeot d'Arcf343e012009-01-12 23:58:21 +00001110}
1111
Yury Selivanovf2392132016-12-13 19:03:51 -05001112
1113/* Specialized version of _PyObject_GenericGetAttrWithDict
1114 specifically for the LOAD_METHOD opcode.
1115
1116 Return 1 if a method is found, 0 if it's a regular attribute
1117 from __dict__ or something returned by using a descriptor
1118 protocol.
1119
1120 `method` will point to the resolved attribute or NULL. In the
1121 latter case, an error will be set.
1122*/
1123int
1124_PyObject_GetMethod(PyObject *obj, PyObject *name, PyObject **method)
1125{
1126 PyTypeObject *tp = Py_TYPE(obj);
1127 PyObject *descr;
1128 descrgetfunc f = NULL;
1129 PyObject **dictptr, *dict;
1130 PyObject *attr;
1131 int meth_found = 0;
1132
1133 assert(*method == NULL);
1134
1135 if (Py_TYPE(obj)->tp_getattro != PyObject_GenericGetAttr
1136 || !PyUnicode_Check(name)) {
1137 *method = PyObject_GetAttr(obj, name);
1138 return 0;
1139 }
1140
1141 if (tp->tp_dict == NULL && PyType_Ready(tp) < 0)
1142 return 0;
1143
1144 descr = _PyType_Lookup(tp, name);
1145 if (descr != NULL) {
1146 Py_INCREF(descr);
INADA Naoki5566bbb2017-02-03 07:43:03 +09001147 if (PyFunction_Check(descr) ||
1148 (Py_TYPE(descr) == &PyMethodDescr_Type)) {
Yury Selivanovf2392132016-12-13 19:03:51 -05001149 meth_found = 1;
1150 } else {
1151 f = descr->ob_type->tp_descr_get;
1152 if (f != NULL && PyDescr_IsData(descr)) {
1153 *method = f(descr, obj, (PyObject *)obj->ob_type);
1154 Py_DECREF(descr);
1155 return 0;
1156 }
1157 }
1158 }
1159
1160 dictptr = _PyObject_GetDictPtr(obj);
1161 if (dictptr != NULL && (dict = *dictptr) != NULL) {
1162 Py_INCREF(dict);
1163 attr = PyDict_GetItem(dict, name);
1164 if (attr != NULL) {
1165 Py_INCREF(attr);
1166 *method = attr;
1167 Py_DECREF(dict);
1168 Py_XDECREF(descr);
1169 return 0;
1170 }
1171 Py_DECREF(dict);
1172 }
1173
1174 if (meth_found) {
1175 *method = descr;
1176 return 1;
1177 }
1178
1179 if (f != NULL) {
1180 *method = f(descr, obj, (PyObject *)Py_TYPE(obj));
1181 Py_DECREF(descr);
1182 return 0;
1183 }
1184
1185 if (descr != NULL) {
1186 *method = descr;
1187 return 0;
1188 }
1189
1190 PyErr_Format(PyExc_AttributeError,
1191 "'%.50s' object has no attribute '%U'",
1192 tp->tp_name, name);
1193 return 0;
1194}
1195
1196/* Generic GetAttr functions - put these in your tp_[gs]etattro slot. */
Michael W. Hudson1593f502004-09-14 17:09:47 +00001197
Raymond Hettinger01538262003-03-17 08:24:35 +00001198PyObject *
INADA Naoki378edee2018-01-16 20:52:41 +09001199_PyObject_GenericGetAttrWithDict(PyObject *obj, PyObject *name,
1200 PyObject *dict, int suppress)
Tim Peters6d6c1a32001-08-02 04:15:00 +00001201{
Yury Selivanovf2392132016-12-13 19:03:51 -05001202 /* Make sure the logic of _PyObject_GetMethod is in sync with
1203 this method.
INADA Naoki378edee2018-01-16 20:52:41 +09001204
1205 When suppress=1, this function suppress AttributeError.
Yury Selivanovf2392132016-12-13 19:03:51 -05001206 */
1207
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001208 PyTypeObject *tp = Py_TYPE(obj);
1209 PyObject *descr = NULL;
1210 PyObject *res = NULL;
1211 descrgetfunc f;
1212 Py_ssize_t dictoffset;
1213 PyObject **dictptr;
Tim Peters6d6c1a32001-08-02 04:15:00 +00001214
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001215 if (!PyUnicode_Check(name)){
1216 PyErr_Format(PyExc_TypeError,
1217 "attribute name must be string, not '%.200s'",
1218 name->ob_type->tp_name);
1219 return NULL;
1220 }
Serhiy Storchaka55c861f2016-04-17 20:31:51 +03001221 Py_INCREF(name);
Guido van Rossumebca9fc2001-12-04 15:54:53 +00001222
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001223 if (tp->tp_dict == NULL) {
1224 if (PyType_Ready(tp) < 0)
1225 goto done;
1226 }
Tim Peters6d6c1a32001-08-02 04:15:00 +00001227
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001228 descr = _PyType_Lookup(tp, name);
Michael W. Hudsonb2c7de42003-08-15 13:07:47 +00001229
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001230 f = NULL;
1231 if (descr != NULL) {
Serhiy Storchaka55c861f2016-04-17 20:31:51 +03001232 Py_INCREF(descr);
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001233 f = descr->ob_type->tp_descr_get;
1234 if (f != NULL && PyDescr_IsData(descr)) {
1235 res = f(descr, obj, (PyObject *)obj->ob_type);
INADA Naoki378edee2018-01-16 20:52:41 +09001236 if (res == NULL && suppress &&
1237 PyErr_ExceptionMatches(PyExc_AttributeError)) {
1238 PyErr_Clear();
1239 }
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001240 goto done;
1241 }
1242 }
Tim Peters6d6c1a32001-08-02 04:15:00 +00001243
Antoine Pitrou1a9a9d52010-08-28 18:17:03 +00001244 if (dict == NULL) {
1245 /* Inline _PyObject_GetDictPtr */
1246 dictoffset = tp->tp_dictoffset;
1247 if (dictoffset != 0) {
1248 if (dictoffset < 0) {
1249 Py_ssize_t tsize;
1250 size_t size;
Guido van Rossumc66ff442002-08-19 16:50:48 +00001251
Antoine Pitrou1a9a9d52010-08-28 18:17:03 +00001252 tsize = ((PyVarObject *)obj)->ob_size;
1253 if (tsize < 0)
1254 tsize = -tsize;
1255 size = _PyObject_VAR_SIZE(tp, tsize);
Serhiy Storchaka55c861f2016-04-17 20:31:51 +03001256 assert(size <= PY_SSIZE_T_MAX);
Guido van Rossumc66ff442002-08-19 16:50:48 +00001257
Serhiy Storchaka55c861f2016-04-17 20:31:51 +03001258 dictoffset += (Py_ssize_t)size;
Antoine Pitrou1a9a9d52010-08-28 18:17:03 +00001259 assert(dictoffset > 0);
1260 assert(dictoffset % SIZEOF_VOID_P == 0);
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001261 }
Antoine Pitrou1a9a9d52010-08-28 18:17:03 +00001262 dictptr = (PyObject **) ((char *)obj + dictoffset);
1263 dict = *dictptr;
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001264 }
1265 }
Antoine Pitrou1a9a9d52010-08-28 18:17:03 +00001266 if (dict != NULL) {
1267 Py_INCREF(dict);
1268 res = PyDict_GetItem(dict, name);
1269 if (res != NULL) {
1270 Py_INCREF(res);
Antoine Pitrou1a9a9d52010-08-28 18:17:03 +00001271 Py_DECREF(dict);
1272 goto done;
1273 }
1274 Py_DECREF(dict);
1275 }
Tim Peters6d6c1a32001-08-02 04:15:00 +00001276
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001277 if (f != NULL) {
1278 res = f(descr, obj, (PyObject *)Py_TYPE(obj));
INADA Naoki378edee2018-01-16 20:52:41 +09001279 if (res == NULL && suppress &&
1280 PyErr_ExceptionMatches(PyExc_AttributeError)) {
1281 PyErr_Clear();
1282 }
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001283 goto done;
1284 }
Tim Peters6d6c1a32001-08-02 04:15:00 +00001285
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001286 if (descr != NULL) {
1287 res = descr;
Victor Stinner2d01dc02012-03-09 00:44:13 +01001288 descr = NULL;
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001289 goto done;
1290 }
Tim Peters6d6c1a32001-08-02 04:15:00 +00001291
INADA Naoki378edee2018-01-16 20:52:41 +09001292 if (!suppress) {
1293 PyErr_Format(PyExc_AttributeError,
1294 "'%.50s' object has no attribute '%U'",
1295 tp->tp_name, name);
1296 }
Guido van Rossumebca9fc2001-12-04 15:54:53 +00001297 done:
Victor Stinner2d01dc02012-03-09 00:44:13 +01001298 Py_XDECREF(descr);
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001299 Py_DECREF(name);
1300 return res;
Tim Peters6d6c1a32001-08-02 04:15:00 +00001301}
1302
Antoine Pitrou1a9a9d52010-08-28 18:17:03 +00001303PyObject *
1304PyObject_GenericGetAttr(PyObject *obj, PyObject *name)
1305{
INADA Naoki378edee2018-01-16 20:52:41 +09001306 return _PyObject_GenericGetAttrWithDict(obj, name, NULL, 0);
Antoine Pitrou1a9a9d52010-08-28 18:17:03 +00001307}
1308
Tim Peters6d6c1a32001-08-02 04:15:00 +00001309int
Antoine Pitrou1a9a9d52010-08-28 18:17:03 +00001310_PyObject_GenericSetAttrWithDict(PyObject *obj, PyObject *name,
1311 PyObject *value, PyObject *dict)
Tim Peters6d6c1a32001-08-02 04:15:00 +00001312{
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001313 PyTypeObject *tp = Py_TYPE(obj);
1314 PyObject *descr;
1315 descrsetfunc f;
1316 PyObject **dictptr;
1317 int res = -1;
Guido van Rossumebca9fc2001-12-04 15:54:53 +00001318
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001319 if (!PyUnicode_Check(name)){
1320 PyErr_Format(PyExc_TypeError,
1321 "attribute name must be string, not '%.200s'",
1322 name->ob_type->tp_name);
1323 return -1;
1324 }
Tim Peters6d6c1a32001-08-02 04:15:00 +00001325
Benjamin Peterson74529ad2012-03-09 07:25:32 -08001326 if (tp->tp_dict == NULL && PyType_Ready(tp) < 0)
1327 return -1;
1328
1329 Py_INCREF(name);
Tim Peters6d6c1a32001-08-02 04:15:00 +00001330
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001331 descr = _PyType_Lookup(tp, name);
Victor Stinner2d01dc02012-03-09 00:44:13 +01001332
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001333 if (descr != NULL) {
Serhiy Storchaka55c861f2016-04-17 20:31:51 +03001334 Py_INCREF(descr);
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001335 f = descr->ob_type->tp_descr_set;
Serhiy Storchaka55c861f2016-04-17 20:31:51 +03001336 if (f != NULL) {
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001337 res = f(descr, obj, value);
1338 goto done;
1339 }
1340 }
Tim Peters6d6c1a32001-08-02 04:15:00 +00001341
Antoine Pitrou1a9a9d52010-08-28 18:17:03 +00001342 if (dict == NULL) {
1343 dictptr = _PyObject_GetDictPtr(obj);
Serhiy Storchaka55c861f2016-04-17 20:31:51 +03001344 if (dictptr == NULL) {
1345 if (descr == NULL) {
1346 PyErr_Format(PyExc_AttributeError,
1347 "'%.100s' object has no attribute '%U'",
1348 tp->tp_name, name);
1349 }
1350 else {
1351 PyErr_Format(PyExc_AttributeError,
1352 "'%.50s' object attribute '%U' is read-only",
1353 tp->tp_name, name);
1354 }
Benjamin Peterson7d95e402012-04-23 11:24:50 -04001355 goto done;
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001356 }
Serhiy Storchaka55c861f2016-04-17 20:31:51 +03001357 res = _PyObjectDict_SetItem(tp, dictptr, name, value);
Antoine Pitrou1a9a9d52010-08-28 18:17:03 +00001358 }
Serhiy Storchaka55c861f2016-04-17 20:31:51 +03001359 else {
Antoine Pitrou1a9a9d52010-08-28 18:17:03 +00001360 Py_INCREF(dict);
1361 if (value == NULL)
1362 res = PyDict_DelItem(dict, name);
1363 else
1364 res = PyDict_SetItem(dict, name, value);
Benjamin Peterson74529ad2012-03-09 07:25:32 -08001365 Py_DECREF(dict);
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001366 }
Serhiy Storchaka55c861f2016-04-17 20:31:51 +03001367 if (res < 0 && PyErr_ExceptionMatches(PyExc_KeyError))
1368 PyErr_SetObject(PyExc_AttributeError, name);
Tim Peters6d6c1a32001-08-02 04:15:00 +00001369
Guido van Rossumebca9fc2001-12-04 15:54:53 +00001370 done:
Victor Stinner2d01dc02012-03-09 00:44:13 +01001371 Py_XDECREF(descr);
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001372 Py_DECREF(name);
1373 return res;
Guido van Rossum98ff96a1997-05-20 18:34:44 +00001374}
1375
Antoine Pitrou1a9a9d52010-08-28 18:17:03 +00001376int
1377PyObject_GenericSetAttr(PyObject *obj, PyObject *name, PyObject *value)
1378{
1379 return _PyObject_GenericSetAttrWithDict(obj, name, value, NULL);
1380}
1381
Benjamin Peterson8eb12692012-02-19 19:59:10 -05001382int
1383PyObject_GenericSetDict(PyObject *obj, PyObject *value, void *context)
1384{
Serhiy Storchaka576f1322016-01-05 21:27:54 +02001385 PyObject **dictptr = _PyObject_GetDictPtr(obj);
Benjamin Peterson8eb12692012-02-19 19:59:10 -05001386 if (dictptr == NULL) {
1387 PyErr_SetString(PyExc_AttributeError,
1388 "This object has no __dict__");
1389 return -1;
1390 }
1391 if (value == NULL) {
1392 PyErr_SetString(PyExc_TypeError, "cannot delete __dict__");
1393 return -1;
1394 }
1395 if (!PyDict_Check(value)) {
1396 PyErr_Format(PyExc_TypeError,
1397 "__dict__ must be set to a dictionary, "
1398 "not a '%.200s'", Py_TYPE(value)->tp_name);
1399 return -1;
1400 }
Serhiy Storchaka576f1322016-01-05 21:27:54 +02001401 Py_INCREF(value);
Serhiy Storchakaec397562016-04-06 09:50:03 +03001402 Py_XSETREF(*dictptr, value);
Benjamin Peterson8eb12692012-02-19 19:59:10 -05001403 return 0;
1404}
1405
Antoine Pitrou1a9a9d52010-08-28 18:17:03 +00001406
Guido van Rossum6ac258d1993-05-12 08:24:20 +00001407/* Test a value used as condition, e.g., in a for or if statement.
1408 Return -1 if an error occurred */
1409
1410int
Fred Drake100814d2000-07-09 15:48:49 +00001411PyObject_IsTrue(PyObject *v)
Guido van Rossum6ac258d1993-05-12 08:24:20 +00001412{
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001413 Py_ssize_t res;
1414 if (v == Py_True)
1415 return 1;
1416 if (v == Py_False)
1417 return 0;
1418 if (v == Py_None)
1419 return 0;
1420 else if (v->ob_type->tp_as_number != NULL &&
1421 v->ob_type->tp_as_number->nb_bool != NULL)
1422 res = (*v->ob_type->tp_as_number->nb_bool)(v);
1423 else if (v->ob_type->tp_as_mapping != NULL &&
1424 v->ob_type->tp_as_mapping->mp_length != NULL)
1425 res = (*v->ob_type->tp_as_mapping->mp_length)(v);
1426 else if (v->ob_type->tp_as_sequence != NULL &&
1427 v->ob_type->tp_as_sequence->sq_length != NULL)
1428 res = (*v->ob_type->tp_as_sequence->sq_length)(v);
1429 else
1430 return 1;
1431 /* if it is negative, it should be either -1 or -2 */
1432 return (res > 0) ? 1 : Py_SAFE_DOWNCAST(res, Py_ssize_t, int);
Guido van Rossum6ac258d1993-05-12 08:24:20 +00001433}
1434
Tim Peters803526b2002-07-07 05:13:56 +00001435/* equivalent of 'not v'
Guido van Rossumc3d3f961998-04-09 17:53:59 +00001436 Return -1 if an error occurred */
1437
1438int
Fred Drake100814d2000-07-09 15:48:49 +00001439PyObject_Not(PyObject *v)
Guido van Rossumc3d3f961998-04-09 17:53:59 +00001440{
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001441 int res;
1442 res = PyObject_IsTrue(v);
1443 if (res < 0)
1444 return res;
1445 return res == 0;
Guido van Rossumc3d3f961998-04-09 17:53:59 +00001446}
1447
Guido van Rossum49b11fe1995-01-26 00:38:22 +00001448/* Test whether an object can be called */
1449
1450int
Fred Drake100814d2000-07-09 15:48:49 +00001451PyCallable_Check(PyObject *x)
Guido van Rossum49b11fe1995-01-26 00:38:22 +00001452{
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001453 if (x == NULL)
1454 return 0;
1455 return x->ob_type->tp_call != NULL;
Guido van Rossum49b11fe1995-01-26 00:38:22 +00001456}
1457
Tim Peters7eea37e2001-09-04 22:08:56 +00001458
Georg Brandle32b4222007-03-10 22:13:27 +00001459/* Helper for PyObject_Dir without arguments: returns the local scope. */
1460static PyObject *
Guido van Rossumad7d8d12007-04-13 01:39:34 +00001461_dir_locals(void)
Tim Peters305b5852001-09-17 02:38:46 +00001462{
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001463 PyObject *names;
Victor Stinner41bb43a2013-10-29 01:19:37 +01001464 PyObject *locals;
Tim Peters305b5852001-09-17 02:38:46 +00001465
Victor Stinner41bb43a2013-10-29 01:19:37 +01001466 locals = PyEval_GetLocals();
1467 if (locals == NULL)
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001468 return NULL;
Tim Peters305b5852001-09-17 02:38:46 +00001469
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001470 names = PyMapping_Keys(locals);
1471 if (!names)
1472 return NULL;
1473 if (!PyList_Check(names)) {
1474 PyErr_Format(PyExc_TypeError,
1475 "dir(): expected keys() of locals to be a list, "
1476 "not '%.200s'", Py_TYPE(names)->tp_name);
1477 Py_DECREF(names);
1478 return NULL;
1479 }
Benjamin Peterson3bbb7222011-06-11 16:12:08 -05001480 if (PyList_Sort(names)) {
1481 Py_DECREF(names);
1482 return NULL;
1483 }
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001484 /* the locals don't need to be DECREF'd */
1485 return names;
Georg Brandle32b4222007-03-10 22:13:27 +00001486}
1487
Benjamin Peterson82b00c12011-05-24 11:09:06 -05001488/* Helper for PyObject_Dir: object introspection. */
Georg Brandle32b4222007-03-10 22:13:27 +00001489static PyObject *
1490_dir_object(PyObject *obj)
1491{
Benjamin Peterson3bbb7222011-06-11 16:12:08 -05001492 PyObject *result, *sorted;
Benjamin Petersonce798522012-01-22 11:24:29 -05001493 PyObject *dirfunc = _PyObject_LookupSpecial(obj, &PyId___dir__);
Georg Brandle32b4222007-03-10 22:13:27 +00001494
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001495 assert(obj);
1496 if (dirfunc == NULL) {
Benjamin Peterson82b00c12011-05-24 11:09:06 -05001497 if (!PyErr_Occurred())
1498 PyErr_SetString(PyExc_TypeError, "object does not provide __dir__");
1499 return NULL;
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001500 }
Benjamin Peterson82b00c12011-05-24 11:09:06 -05001501 /* use __dir__ */
Victor Stinnerf17c3de2016-12-06 18:46:19 +01001502 result = _PyObject_CallNoArg(dirfunc);
Benjamin Peterson82b00c12011-05-24 11:09:06 -05001503 Py_DECREF(dirfunc);
1504 if (result == NULL)
1505 return NULL;
Benjamin Peterson3bbb7222011-06-11 16:12:08 -05001506 /* return sorted(result) */
1507 sorted = PySequence_List(result);
1508 Py_DECREF(result);
1509 if (sorted == NULL)
1510 return NULL;
1511 if (PyList_Sort(sorted)) {
1512 Py_DECREF(sorted);
1513 return NULL;
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001514 }
Benjamin Peterson3bbb7222011-06-11 16:12:08 -05001515 return sorted;
Georg Brandle32b4222007-03-10 22:13:27 +00001516}
1517
1518/* Implementation of dir() -- if obj is NULL, returns the names in the current
1519 (local) scope. Otherwise, performs introspection of the object: returns a
1520 sorted list of attribute names (supposedly) accessible from the object
1521*/
1522PyObject *
1523PyObject_Dir(PyObject *obj)
1524{
Benjamin Peterson3bbb7222011-06-11 16:12:08 -05001525 return (obj == NULL) ? _dir_locals() : _dir_object(obj);
Tim Peters7eea37e2001-09-04 22:08:56 +00001526}
Guido van Rossum49b11fe1995-01-26 00:38:22 +00001527
Guido van Rossum85a5fbb1990-10-14 12:07:46 +00001528/*
Raymond Hettinger66d2be82011-07-28 09:55:13 -07001529None is a non-NULL undefined value.
Guido van Rossum85a5fbb1990-10-14 12:07:46 +00001530There is (and should be!) no way to create other objects of this type,
Guido van Rossum3f5da241990-12-20 15:06:42 +00001531so there is exactly one (which is indestructible, by the way).
Guido van Rossum85a5fbb1990-10-14 12:07:46 +00001532*/
1533
Guido van Rossum0c182a11992-03-27 17:26:13 +00001534/* ARGSUSED */
Guido van Rossumc0b618a1997-05-02 03:12:38 +00001535static PyObject *
Fred Drake100814d2000-07-09 15:48:49 +00001536none_repr(PyObject *op)
Guido van Rossum3f5da241990-12-20 15:06:42 +00001537{
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001538 return PyUnicode_FromString("None");
Guido van Rossum85a5fbb1990-10-14 12:07:46 +00001539}
1540
Barry Warsaw9bf16442001-01-23 16:24:35 +00001541/* ARGUSED */
1542static void
Tim Peters803526b2002-07-07 05:13:56 +00001543none_dealloc(PyObject* ignore)
Barry Warsaw9bf16442001-01-23 16:24:35 +00001544{
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001545 /* This should never get called, but we also don't want to SEGV if
1546 * we accidentally decref None out of existence.
1547 */
1548 Py_FatalError("deallocating None");
Barry Warsaw9bf16442001-01-23 16:24:35 +00001549}
1550
Benjamin Petersonc4607ae2011-07-29 18:19:43 -05001551static PyObject *
1552none_new(PyTypeObject *type, PyObject *args, PyObject *kwargs)
1553{
Serhiy Storchaka5ab81d72016-12-16 16:18:57 +02001554 if (PyTuple_GET_SIZE(args) || (kwargs && PyDict_GET_SIZE(kwargs))) {
Benjamin Petersonc4607ae2011-07-29 18:19:43 -05001555 PyErr_SetString(PyExc_TypeError, "NoneType takes no arguments");
1556 return NULL;
1557 }
1558 Py_RETURN_NONE;
1559}
1560
Raymond Hettinger66d2be82011-07-28 09:55:13 -07001561static int
1562none_bool(PyObject *v)
1563{
1564 return 0;
1565}
1566
1567static PyNumberMethods none_as_number = {
1568 0, /* nb_add */
1569 0, /* nb_subtract */
1570 0, /* nb_multiply */
1571 0, /* nb_remainder */
1572 0, /* nb_divmod */
1573 0, /* nb_power */
1574 0, /* nb_negative */
1575 0, /* nb_positive */
1576 0, /* nb_absolute */
1577 (inquiry)none_bool, /* nb_bool */
1578 0, /* nb_invert */
1579 0, /* nb_lshift */
1580 0, /* nb_rshift */
1581 0, /* nb_and */
1582 0, /* nb_xor */
1583 0, /* nb_or */
1584 0, /* nb_int */
1585 0, /* nb_reserved */
1586 0, /* nb_float */
1587 0, /* nb_inplace_add */
1588 0, /* nb_inplace_subtract */
1589 0, /* nb_inplace_multiply */
1590 0, /* nb_inplace_remainder */
1591 0, /* nb_inplace_power */
1592 0, /* nb_inplace_lshift */
1593 0, /* nb_inplace_rshift */
1594 0, /* nb_inplace_and */
1595 0, /* nb_inplace_xor */
1596 0, /* nb_inplace_or */
1597 0, /* nb_floor_divide */
1598 0, /* nb_true_divide */
1599 0, /* nb_inplace_floor_divide */
1600 0, /* nb_inplace_true_divide */
1601 0, /* nb_index */
1602};
Barry Warsaw9bf16442001-01-23 16:24:35 +00001603
Alexandre Vassalotti65846c62013-11-30 17:55:48 -08001604PyTypeObject _PyNone_Type = {
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001605 PyVarObject_HEAD_INIT(&PyType_Type, 0)
1606 "NoneType",
1607 0,
1608 0,
1609 none_dealloc, /*tp_dealloc*/ /*never called*/
1610 0, /*tp_print*/
1611 0, /*tp_getattr*/
1612 0, /*tp_setattr*/
1613 0, /*tp_reserved*/
1614 none_repr, /*tp_repr*/
Raymond Hettinger66d2be82011-07-28 09:55:13 -07001615 &none_as_number, /*tp_as_number*/
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001616 0, /*tp_as_sequence*/
1617 0, /*tp_as_mapping*/
1618 0, /*tp_hash */
Benjamin Petersonc4607ae2011-07-29 18:19:43 -05001619 0, /*tp_call */
1620 0, /*tp_str */
1621 0, /*tp_getattro */
1622 0, /*tp_setattro */
1623 0, /*tp_as_buffer */
1624 Py_TPFLAGS_DEFAULT, /*tp_flags */
1625 0, /*tp_doc */
1626 0, /*tp_traverse */
1627 0, /*tp_clear */
1628 0, /*tp_richcompare */
1629 0, /*tp_weaklistoffset */
1630 0, /*tp_iter */
1631 0, /*tp_iternext */
1632 0, /*tp_methods */
1633 0, /*tp_members */
1634 0, /*tp_getset */
1635 0, /*tp_base */
1636 0, /*tp_dict */
1637 0, /*tp_descr_get */
1638 0, /*tp_descr_set */
1639 0, /*tp_dictoffset */
1640 0, /*tp_init */
1641 0, /*tp_alloc */
1642 none_new, /*tp_new */
Guido van Rossum85a5fbb1990-10-14 12:07:46 +00001643};
1644
Guido van Rossumc0b618a1997-05-02 03:12:38 +00001645PyObject _Py_NoneStruct = {
Martin v. Löwis9f2e3462007-07-21 17:22:18 +00001646 _PyObject_EXTRA_INIT
Alexandre Vassalotti65846c62013-11-30 17:55:48 -08001647 1, &_PyNone_Type
Guido van Rossum85a5fbb1990-10-14 12:07:46 +00001648};
1649
Neil Schemenauer5ed85ec2001-01-04 01:48:10 +00001650/* NotImplemented is an object that can be used to signal that an
1651 operation is not implemented for the given type combination. */
1652
1653static PyObject *
1654NotImplemented_repr(PyObject *op)
1655{
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001656 return PyUnicode_FromString("NotImplemented");
Neil Schemenauer5ed85ec2001-01-04 01:48:10 +00001657}
1658
Benjamin Peterson18d7d7a2011-07-29 18:27:44 -05001659static PyObject *
Alexandre Vassalottic49477b2013-11-24 02:53:45 -08001660NotImplemented_reduce(PyObject *op)
1661{
1662 return PyUnicode_FromString("NotImplemented");
1663}
1664
1665static PyMethodDef notimplemented_methods[] = {
1666 {"__reduce__", (PyCFunction)NotImplemented_reduce, METH_NOARGS, NULL},
1667 {NULL, NULL}
1668};
1669
1670static PyObject *
Benjamin Peterson18d7d7a2011-07-29 18:27:44 -05001671notimplemented_new(PyTypeObject *type, PyObject *args, PyObject *kwargs)
1672{
Serhiy Storchaka5ab81d72016-12-16 16:18:57 +02001673 if (PyTuple_GET_SIZE(args) || (kwargs && PyDict_GET_SIZE(kwargs))) {
Benjamin Peterson18d7d7a2011-07-29 18:27:44 -05001674 PyErr_SetString(PyExc_TypeError, "NotImplementedType takes no arguments");
1675 return NULL;
1676 }
Brian Curtindfc80e32011-08-10 20:28:54 -05001677 Py_RETURN_NOTIMPLEMENTED;
Benjamin Peterson18d7d7a2011-07-29 18:27:44 -05001678}
1679
Armin Ronacher226b1db2012-10-06 14:28:58 +02001680static void
1681notimplemented_dealloc(PyObject* ignore)
1682{
1683 /* This should never get called, but we also don't want to SEGV if
1684 * we accidentally decref NotImplemented out of existence.
1685 */
1686 Py_FatalError("deallocating NotImplemented");
1687}
1688
Alexandre Vassalotti65846c62013-11-30 17:55:48 -08001689PyTypeObject _PyNotImplemented_Type = {
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001690 PyVarObject_HEAD_INIT(&PyType_Type, 0)
1691 "NotImplementedType",
1692 0,
1693 0,
Armin Ronacher226b1db2012-10-06 14:28:58 +02001694 notimplemented_dealloc, /*tp_dealloc*/ /*never called*/
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001695 0, /*tp_print*/
1696 0, /*tp_getattr*/
1697 0, /*tp_setattr*/
1698 0, /*tp_reserved*/
1699 NotImplemented_repr, /*tp_repr*/
1700 0, /*tp_as_number*/
1701 0, /*tp_as_sequence*/
1702 0, /*tp_as_mapping*/
1703 0, /*tp_hash */
Benjamin Peterson18d7d7a2011-07-29 18:27:44 -05001704 0, /*tp_call */
1705 0, /*tp_str */
1706 0, /*tp_getattro */
1707 0, /*tp_setattro */
1708 0, /*tp_as_buffer */
1709 Py_TPFLAGS_DEFAULT, /*tp_flags */
1710 0, /*tp_doc */
1711 0, /*tp_traverse */
1712 0, /*tp_clear */
1713 0, /*tp_richcompare */
1714 0, /*tp_weaklistoffset */
1715 0, /*tp_iter */
1716 0, /*tp_iternext */
Alexandre Vassalottic49477b2013-11-24 02:53:45 -08001717 notimplemented_methods, /*tp_methods */
Benjamin Peterson18d7d7a2011-07-29 18:27:44 -05001718 0, /*tp_members */
1719 0, /*tp_getset */
1720 0, /*tp_base */
1721 0, /*tp_dict */
1722 0, /*tp_descr_get */
1723 0, /*tp_descr_set */
1724 0, /*tp_dictoffset */
1725 0, /*tp_init */
1726 0, /*tp_alloc */
1727 notimplemented_new, /*tp_new */
Neil Schemenauer5ed85ec2001-01-04 01:48:10 +00001728};
1729
1730PyObject _Py_NotImplementedStruct = {
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001731 _PyObject_EXTRA_INIT
Alexandre Vassalotti65846c62013-11-30 17:55:48 -08001732 1, &_PyNotImplemented_Type
Neil Schemenauer5ed85ec2001-01-04 01:48:10 +00001733};
1734
Guido van Rossumba21a492001-08-16 08:17:26 +00001735void
1736_Py_ReadyTypes(void)
1737{
Victor Stinner5a1bb4e2014-06-02 14:10:59 +02001738 if (PyType_Ready(&PyBaseObject_Type) < 0)
1739 Py_FatalError("Can't initialize object type");
1740
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001741 if (PyType_Ready(&PyType_Type) < 0)
1742 Py_FatalError("Can't initialize type type");
Guido van Rossumba21a492001-08-16 08:17:26 +00001743
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001744 if (PyType_Ready(&_PyWeakref_RefType) < 0)
1745 Py_FatalError("Can't initialize weakref type");
Fred Drake0a4dd392004-07-02 18:57:45 +00001746
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001747 if (PyType_Ready(&_PyWeakref_CallableProxyType) < 0)
1748 Py_FatalError("Can't initialize callable weakref proxy type");
Benjamin Petersonfd838e62009-04-20 02:09:13 +00001749
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001750 if (PyType_Ready(&_PyWeakref_ProxyType) < 0)
1751 Py_FatalError("Can't initialize weakref proxy type");
Benjamin Petersonfd838e62009-04-20 02:09:13 +00001752
Victor Stinner5a1bb4e2014-06-02 14:10:59 +02001753 if (PyType_Ready(&PyLong_Type) < 0)
1754 Py_FatalError("Can't initialize int type");
1755
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001756 if (PyType_Ready(&PyBool_Type) < 0)
1757 Py_FatalError("Can't initialize bool type");
Guido van Rossum77f6a652002-04-03 22:41:51 +00001758
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001759 if (PyType_Ready(&PyByteArray_Type) < 0)
1760 Py_FatalError("Can't initialize bytearray type");
Guido van Rossum4dfe8a12006-04-22 23:28:04 +00001761
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001762 if (PyType_Ready(&PyBytes_Type) < 0)
1763 Py_FatalError("Can't initialize 'str'");
Guido van Rossumcacfc072002-05-24 19:01:59 +00001764
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001765 if (PyType_Ready(&PyList_Type) < 0)
1766 Py_FatalError("Can't initialize list type");
Guido van Rossumba21a492001-08-16 08:17:26 +00001767
Alexandre Vassalotti65846c62013-11-30 17:55:48 -08001768 if (PyType_Ready(&_PyNone_Type) < 0)
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001769 Py_FatalError("Can't initialize None type");
Guido van Rossumba21a492001-08-16 08:17:26 +00001770
Alexandre Vassalotti65846c62013-11-30 17:55:48 -08001771 if (PyType_Ready(&_PyNotImplemented_Type) < 0)
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001772 Py_FatalError("Can't initialize NotImplemented type");
Benjamin Petersonae937c02009-04-18 20:54:08 +00001773
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001774 if (PyType_Ready(&PyTraceBack_Type) < 0)
1775 Py_FatalError("Can't initialize traceback type");
Benjamin Petersonae937c02009-04-18 20:54:08 +00001776
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001777 if (PyType_Ready(&PySuper_Type) < 0)
1778 Py_FatalError("Can't initialize super type");
Benjamin Petersonae937c02009-04-18 20:54:08 +00001779
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001780 if (PyType_Ready(&PyRange_Type) < 0)
1781 Py_FatalError("Can't initialize range type");
Benjamin Petersonae937c02009-04-18 20:54:08 +00001782
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001783 if (PyType_Ready(&PyDict_Type) < 0)
1784 Py_FatalError("Can't initialize dict type");
Benjamin Petersonae937c02009-04-18 20:54:08 +00001785
Benjamin Petersondb87c992016-11-06 13:01:07 -08001786 if (PyType_Ready(&PyDictKeys_Type) < 0)
1787 Py_FatalError("Can't initialize dict keys type");
1788
1789 if (PyType_Ready(&PyDictValues_Type) < 0)
1790 Py_FatalError("Can't initialize dict values type");
1791
1792 if (PyType_Ready(&PyDictItems_Type) < 0)
1793 Py_FatalError("Can't initialize dict items type");
1794
Eric Snow96c6af92015-05-29 22:21:39 -06001795 if (PyType_Ready(&PyODict_Type) < 0)
1796 Py_FatalError("Can't initialize OrderedDict type");
1797
1798 if (PyType_Ready(&PyODictKeys_Type) < 0)
1799 Py_FatalError("Can't initialize odict_keys type");
1800
1801 if (PyType_Ready(&PyODictItems_Type) < 0)
1802 Py_FatalError("Can't initialize odict_items type");
1803
1804 if (PyType_Ready(&PyODictValues_Type) < 0)
1805 Py_FatalError("Can't initialize odict_values type");
1806
1807 if (PyType_Ready(&PyODictIter_Type) < 0)
1808 Py_FatalError("Can't initialize odict_keyiterator type");
1809
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001810 if (PyType_Ready(&PySet_Type) < 0)
1811 Py_FatalError("Can't initialize set type");
Benjamin Petersonae937c02009-04-18 20:54:08 +00001812
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001813 if (PyType_Ready(&PyUnicode_Type) < 0)
1814 Py_FatalError("Can't initialize str type");
Benjamin Petersonae937c02009-04-18 20:54:08 +00001815
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001816 if (PyType_Ready(&PySlice_Type) < 0)
1817 Py_FatalError("Can't initialize slice type");
Benjamin Petersonae937c02009-04-18 20:54:08 +00001818
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001819 if (PyType_Ready(&PyStaticMethod_Type) < 0)
1820 Py_FatalError("Can't initialize static method type");
Benjamin Petersonae937c02009-04-18 20:54:08 +00001821
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001822 if (PyType_Ready(&PyComplex_Type) < 0)
1823 Py_FatalError("Can't initialize complex type");
Skip Montanaroba1e0f42009-10-18 14:25:35 +00001824
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001825 if (PyType_Ready(&PyFloat_Type) < 0)
1826 Py_FatalError("Can't initialize float type");
Benjamin Petersonae937c02009-04-18 20:54:08 +00001827
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001828 if (PyType_Ready(&PyFrozenSet_Type) < 0)
1829 Py_FatalError("Can't initialize frozenset type");
Benjamin Petersonae937c02009-04-18 20:54:08 +00001830
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001831 if (PyType_Ready(&PyProperty_Type) < 0)
1832 Py_FatalError("Can't initialize property type");
Benjamin Petersonae937c02009-04-18 20:54:08 +00001833
Stefan Krah9a2d99e2012-02-25 12:24:21 +01001834 if (PyType_Ready(&_PyManagedBuffer_Type) < 0)
1835 Py_FatalError("Can't initialize managed buffer type");
1836
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001837 if (PyType_Ready(&PyMemoryView_Type) < 0)
1838 Py_FatalError("Can't initialize memoryview type");
Benjamin Petersonae937c02009-04-18 20:54:08 +00001839
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001840 if (PyType_Ready(&PyTuple_Type) < 0)
1841 Py_FatalError("Can't initialize tuple type");
Benjamin Petersonae937c02009-04-18 20:54:08 +00001842
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001843 if (PyType_Ready(&PyEnum_Type) < 0)
1844 Py_FatalError("Can't initialize enumerate type");
Benjamin Petersonae937c02009-04-18 20:54:08 +00001845
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001846 if (PyType_Ready(&PyReversed_Type) < 0)
1847 Py_FatalError("Can't initialize reversed type");
Guido van Rossum47b9ff62006-08-24 00:41:19 +00001848
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001849 if (PyType_Ready(&PyStdPrinter_Type) < 0)
1850 Py_FatalError("Can't initialize StdPrinter");
Benjamin Petersonfd838e62009-04-20 02:09:13 +00001851
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001852 if (PyType_Ready(&PyCode_Type) < 0)
1853 Py_FatalError("Can't initialize code type");
Benjamin Petersonfd838e62009-04-20 02:09:13 +00001854
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001855 if (PyType_Ready(&PyFrame_Type) < 0)
1856 Py_FatalError("Can't initialize frame type");
Benjamin Petersonfd838e62009-04-20 02:09:13 +00001857
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001858 if (PyType_Ready(&PyCFunction_Type) < 0)
1859 Py_FatalError("Can't initialize builtin function type");
Benjamin Petersonfd838e62009-04-20 02:09:13 +00001860
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001861 if (PyType_Ready(&PyMethod_Type) < 0)
1862 Py_FatalError("Can't initialize method type");
Benjamin Petersonfd838e62009-04-20 02:09:13 +00001863
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001864 if (PyType_Ready(&PyFunction_Type) < 0)
1865 Py_FatalError("Can't initialize function type");
Benjamin Petersonfd838e62009-04-20 02:09:13 +00001866
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001867 if (PyType_Ready(&PyDictProxy_Type) < 0)
1868 Py_FatalError("Can't initialize dict proxy type");
Benjamin Petersonfd838e62009-04-20 02:09:13 +00001869
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001870 if (PyType_Ready(&PyGen_Type) < 0)
1871 Py_FatalError("Can't initialize generator type");
Benjamin Petersonfd838e62009-04-20 02:09:13 +00001872
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001873 if (PyType_Ready(&PyGetSetDescr_Type) < 0)
1874 Py_FatalError("Can't initialize get-set descriptor type");
Benjamin Petersonfd838e62009-04-20 02:09:13 +00001875
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001876 if (PyType_Ready(&PyWrapperDescr_Type) < 0)
1877 Py_FatalError("Can't initialize wrapper type");
Benjamin Petersonfd838e62009-04-20 02:09:13 +00001878
Benjamin Petersoneff61f62011-09-01 16:32:31 -04001879 if (PyType_Ready(&_PyMethodWrapper_Type) < 0)
1880 Py_FatalError("Can't initialize method wrapper type");
1881
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001882 if (PyType_Ready(&PyEllipsis_Type) < 0)
1883 Py_FatalError("Can't initialize ellipsis type");
Benjamin Petersonfd838e62009-04-20 02:09:13 +00001884
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001885 if (PyType_Ready(&PyMemberDescr_Type) < 0)
1886 Py_FatalError("Can't initialize member descriptor type");
Benjamin Peterson8bc5b682009-05-09 18:10:51 +00001887
Barry Warsaw409da152012-06-03 16:18:47 -04001888 if (PyType_Ready(&_PyNamespace_Type) < 0)
1889 Py_FatalError("Can't initialize namespace type");
Benjamin Petersone8ea97f2012-10-30 23:27:52 -04001890
Benjamin Petersonc4311282012-10-30 23:21:10 -04001891 if (PyType_Ready(&PyCapsule_Type) < 0)
1892 Py_FatalError("Can't initialize capsule type");
1893
1894 if (PyType_Ready(&PyLongRangeIter_Type) < 0)
1895 Py_FatalError("Can't initialize long range iterator type");
1896
1897 if (PyType_Ready(&PyCell_Type) < 0)
1898 Py_FatalError("Can't initialize cell type");
1899
1900 if (PyType_Ready(&PyInstanceMethod_Type) < 0)
1901 Py_FatalError("Can't initialize instance method type");
1902
1903 if (PyType_Ready(&PyClassMethodDescr_Type) < 0)
1904 Py_FatalError("Can't initialize class method descr type");
1905
1906 if (PyType_Ready(&PyMethodDescr_Type) < 0)
1907 Py_FatalError("Can't initialize method descr type");
1908
1909 if (PyType_Ready(&PyCallIter_Type) < 0)
1910 Py_FatalError("Can't initialize call iter type");
1911
1912 if (PyType_Ready(&PySeqIter_Type) < 0)
1913 Py_FatalError("Can't initialize sequence iterator type");
Yury Selivanov5376ba92015-06-22 12:19:30 -04001914
1915 if (PyType_Ready(&PyCoro_Type) < 0)
1916 Py_FatalError("Can't initialize coroutine type");
1917
1918 if (PyType_Ready(&_PyCoroWrapper_Type) < 0)
1919 Py_FatalError("Can't initialize coroutine wrapper type");
Guido van Rossumba21a492001-08-16 08:17:26 +00001920}
1921
Guido van Rossum85a5fbb1990-10-14 12:07:46 +00001922
Guido van Rossum84a90321996-05-22 16:34:47 +00001923#ifdef Py_TRACE_REFS
Guido van Rossum85a5fbb1990-10-14 12:07:46 +00001924
Guido van Rossumaacdc9d1996-08-12 21:32:12 +00001925void
Fred Drake100814d2000-07-09 15:48:49 +00001926_Py_NewReference(PyObject *op)
Guido van Rossum85a5fbb1990-10-14 12:07:46 +00001927{
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001928 _Py_INC_REFTOTAL;
1929 op->ob_refcnt = 1;
1930 _Py_AddToAllObjects(op, 1);
1931 _Py_INC_TPALLOCS(op);
Guido van Rossum85a5fbb1990-10-14 12:07:46 +00001932}
1933
Guido van Rossumaacdc9d1996-08-12 21:32:12 +00001934void
Antoine Pitrou9ed5f272013-08-13 20:18:52 +02001935_Py_ForgetReference(PyObject *op)
Guido van Rossum85a5fbb1990-10-14 12:07:46 +00001936{
Guido van Rossumbffd6832000-01-20 22:32:56 +00001937#ifdef SLOW_UNREF_CHECK
Antoine Pitrou9ed5f272013-08-13 20:18:52 +02001938 PyObject *p;
Guido van Rossumbffd6832000-01-20 22:32:56 +00001939#endif
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001940 if (op->ob_refcnt < 0)
1941 Py_FatalError("UNREF negative refcnt");
1942 if (op == &refchain ||
1943 op->_ob_prev->_ob_next != op || op->_ob_next->_ob_prev != op) {
1944 fprintf(stderr, "* ob\n");
1945 _PyObject_Dump(op);
1946 fprintf(stderr, "* op->_ob_prev->_ob_next\n");
1947 _PyObject_Dump(op->_ob_prev->_ob_next);
1948 fprintf(stderr, "* op->_ob_next->_ob_prev\n");
1949 _PyObject_Dump(op->_ob_next->_ob_prev);
1950 Py_FatalError("UNREF invalid object");
1951 }
Guido van Rossum2e8f6141992-09-03 20:32:55 +00001952#ifdef SLOW_UNREF_CHECK
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001953 for (p = refchain._ob_next; p != &refchain; p = p->_ob_next) {
1954 if (p == op)
1955 break;
1956 }
1957 if (p == &refchain) /* Not found */
1958 Py_FatalError("UNREF unknown object");
Guido van Rossum2e8f6141992-09-03 20:32:55 +00001959#endif
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001960 op->_ob_next->_ob_prev = op->_ob_prev;
1961 op->_ob_prev->_ob_next = op->_ob_next;
1962 op->_ob_next = op->_ob_prev = NULL;
1963 _Py_INC_TPFREES(op);
Guido van Rossum3f5da241990-12-20 15:06:42 +00001964}
1965
Guido van Rossumaacdc9d1996-08-12 21:32:12 +00001966void
Fred Drake100814d2000-07-09 15:48:49 +00001967_Py_Dealloc(PyObject *op)
Guido van Rossum3f5da241990-12-20 15:06:42 +00001968{
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001969 destructor dealloc = Py_TYPE(op)->tp_dealloc;
1970 _Py_ForgetReference(op);
1971 (*dealloc)(op);
Guido van Rossum85a5fbb1990-10-14 12:07:46 +00001972}
1973
Tim Peters269b2a62003-04-17 19:52:29 +00001974/* Print all live objects. Because PyObject_Print is called, the
1975 * interpreter must be in a healthy state.
1976 */
Guido van Rossumaacdc9d1996-08-12 21:32:12 +00001977void
Fred Drake100814d2000-07-09 15:48:49 +00001978_Py_PrintReferences(FILE *fp)
Guido van Rossum85a5fbb1990-10-14 12:07:46 +00001979{
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001980 PyObject *op;
1981 fprintf(fp, "Remaining objects:\n");
1982 for (op = refchain._ob_next; op != &refchain; op = op->_ob_next) {
1983 fprintf(fp, "%p [%" PY_FORMAT_SIZE_T "d] ", op, op->ob_refcnt);
1984 if (PyObject_Print(op, fp, 0) != 0)
1985 PyErr_Clear();
1986 putc('\n', fp);
1987 }
Guido van Rossum85a5fbb1990-10-14 12:07:46 +00001988}
1989
Tim Peters269b2a62003-04-17 19:52:29 +00001990/* Print the addresses of all live objects. Unlike _Py_PrintReferences, this
1991 * doesn't make any calls to the Python C API, so is always safe to call.
1992 */
1993void
1994_Py_PrintReferenceAddresses(FILE *fp)
1995{
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001996 PyObject *op;
1997 fprintf(fp, "Remaining object addresses:\n");
1998 for (op = refchain._ob_next; op != &refchain; op = op->_ob_next)
1999 fprintf(fp, "%p [%" PY_FORMAT_SIZE_T "d] %s\n", op,
2000 op->ob_refcnt, Py_TYPE(op)->tp_name);
Tim Peters269b2a62003-04-17 19:52:29 +00002001}
2002
Sjoerd Mullender6ec3c651995-08-29 09:18:14 +00002003PyObject *
Fred Drake100814d2000-07-09 15:48:49 +00002004_Py_GetObjects(PyObject *self, PyObject *args)
Sjoerd Mullender6ec3c651995-08-29 09:18:14 +00002005{
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002006 int i, n;
2007 PyObject *t = NULL;
2008 PyObject *res, *op;
Sjoerd Mullender6ec3c651995-08-29 09:18:14 +00002009
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002010 if (!PyArg_ParseTuple(args, "i|O", &n, &t))
2011 return NULL;
2012 op = refchain._ob_next;
2013 res = PyList_New(0);
2014 if (res == NULL)
2015 return NULL;
2016 for (i = 0; (n == 0 || i < n) && op != &refchain; i++) {
2017 while (op == self || op == args || op == res || op == t ||
2018 (t != NULL && Py_TYPE(op) != (PyTypeObject *) t)) {
2019 op = op->_ob_next;
2020 if (op == &refchain)
2021 return res;
2022 }
2023 if (PyList_Append(res, op) < 0) {
2024 Py_DECREF(res);
2025 return NULL;
2026 }
2027 op = op->_ob_next;
2028 }
2029 return res;
Sjoerd Mullender6ec3c651995-08-29 09:18:14 +00002030}
2031
Guido van Rossum85a5fbb1990-10-14 12:07:46 +00002032#endif
Guido van Rossum97ead3f1996-01-12 01:24:09 +00002033
Benjamin Petersonb173f782009-05-05 22:31:58 +00002034
Guido van Rossum84a90321996-05-22 16:34:47 +00002035/* Hack to force loading of abstract.o */
Martin v. Löwis18e16552006-02-15 17:27:45 +00002036Py_ssize_t (*_Py_abstract_hack)(PyObject *) = PyObject_Size;
Guido van Rossume09fb551997-08-05 02:04:34 +00002037
2038
David Malcolm49526f42012-06-22 14:55:41 -04002039void
2040_PyObject_DebugTypeStats(FILE *out)
2041{
2042 _PyCFunction_DebugMallocStats(out);
2043 _PyDict_DebugMallocStats(out);
2044 _PyFloat_DebugMallocStats(out);
2045 _PyFrame_DebugMallocStats(out);
2046 _PyList_DebugMallocStats(out);
2047 _PyMethod_DebugMallocStats(out);
David Malcolm49526f42012-06-22 14:55:41 -04002048 _PyTuple_DebugMallocStats(out);
2049}
Guido van Rossumb18618d2000-05-03 23:44:39 +00002050
Guido van Rossum86610361998-04-10 22:32:46 +00002051/* These methods are used to control infinite recursion in repr, str, print,
2052 etc. Container objects that may recursively contain themselves,
Martin Panter8d56c022016-05-29 04:13:35 +00002053 e.g. builtin dictionaries and lists, should use Py_ReprEnter() and
Guido van Rossum86610361998-04-10 22:32:46 +00002054 Py_ReprLeave() to avoid infinite recursion.
2055
2056 Py_ReprEnter() returns 0 the first time it is called for a particular
2057 object and 1 every time thereafter. It returns -1 if an exception
2058 occurred. Py_ReprLeave() has no return value.
2059
2060 See dictobject.c and listobject.c for examples of use.
2061*/
2062
Guido van Rossum86610361998-04-10 22:32:46 +00002063int
Fred Drake100814d2000-07-09 15:48:49 +00002064Py_ReprEnter(PyObject *obj)
Guido van Rossum86610361998-04-10 22:32:46 +00002065{
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002066 PyObject *dict;
2067 PyObject *list;
2068 Py_ssize_t i;
Guido van Rossum86610361998-04-10 22:32:46 +00002069
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002070 dict = PyThreadState_GetDict();
Antoine Pitrou04d17d32014-03-31 22:04:38 +02002071 /* Ignore a missing thread-state, so that this function can be called
2072 early on startup. */
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002073 if (dict == NULL)
2074 return 0;
Victor Stinner7a07e452013-11-06 18:57:29 +01002075 list = _PyDict_GetItemId(dict, &PyId_Py_Repr);
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002076 if (list == NULL) {
2077 list = PyList_New(0);
2078 if (list == NULL)
2079 return -1;
Victor Stinner7a07e452013-11-06 18:57:29 +01002080 if (_PyDict_SetItemId(dict, &PyId_Py_Repr, list) < 0)
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002081 return -1;
2082 Py_DECREF(list);
2083 }
2084 i = PyList_GET_SIZE(list);
2085 while (--i >= 0) {
2086 if (PyList_GET_ITEM(list, i) == obj)
2087 return 1;
2088 }
Victor Stinnere901d1f2013-07-17 21:58:41 +02002089 if (PyList_Append(list, obj) < 0)
2090 return -1;
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002091 return 0;
Guido van Rossum86610361998-04-10 22:32:46 +00002092}
2093
2094void
Fred Drake100814d2000-07-09 15:48:49 +00002095Py_ReprLeave(PyObject *obj)
Guido van Rossum86610361998-04-10 22:32:46 +00002096{
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002097 PyObject *dict;
2098 PyObject *list;
2099 Py_ssize_t i;
Victor Stinner1b634932013-07-16 22:24:44 +02002100 PyObject *error_type, *error_value, *error_traceback;
2101
2102 PyErr_Fetch(&error_type, &error_value, &error_traceback);
Guido van Rossum86610361998-04-10 22:32:46 +00002103
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002104 dict = PyThreadState_GetDict();
2105 if (dict == NULL)
Victor Stinner1b634932013-07-16 22:24:44 +02002106 goto finally;
2107
Victor Stinner7a07e452013-11-06 18:57:29 +01002108 list = _PyDict_GetItemId(dict, &PyId_Py_Repr);
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002109 if (list == NULL || !PyList_Check(list))
Victor Stinner1b634932013-07-16 22:24:44 +02002110 goto finally;
2111
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002112 i = PyList_GET_SIZE(list);
2113 /* Count backwards because we always expect obj to be list[-1] */
2114 while (--i >= 0) {
2115 if (PyList_GET_ITEM(list, i) == obj) {
2116 PyList_SetSlice(list, i, i + 1, NULL);
2117 break;
2118 }
2119 }
Victor Stinner1b634932013-07-16 22:24:44 +02002120
2121finally:
2122 /* ignore exceptions because there is no way to report them. */
2123 PyErr_Restore(error_type, error_value, error_traceback);
Guido van Rossum86610361998-04-10 22:32:46 +00002124}
Guido van Rossumd724b232000-03-13 16:01:29 +00002125
Tim Peters803526b2002-07-07 05:13:56 +00002126/* Trashcan support. */
Guido van Rossumd724b232000-03-13 16:01:29 +00002127
Tim Peters803526b2002-07-07 05:13:56 +00002128/* Add op to the _PyTrash_delete_later list. Called when the current
2129 * call-stack depth gets large. op must be a currently untracked gc'ed
2130 * object, with refcount 0. Py_DECREF must already have been called on it.
2131 */
Guido van Rossumd724b232000-03-13 16:01:29 +00002132void
Fred Drake100814d2000-07-09 15:48:49 +00002133_PyTrash_deposit_object(PyObject *op)
Guido van Rossumd724b232000-03-13 16:01:29 +00002134{
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002135 assert(PyObject_IS_GC(op));
Antoine Pitrou796564c2013-07-30 19:59:21 +02002136 assert(_PyGC_REFS(op) == _PyGC_REFS_UNTRACKED);
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002137 assert(op->ob_refcnt == 0);
Eric Snow2ebc5ce2017-09-07 23:51:28 -06002138 _Py_AS_GC(op)->gc.gc_prev = (PyGC_Head *)_PyRuntime.gc.trash_delete_later;
2139 _PyRuntime.gc.trash_delete_later = op;
Guido van Rossumd724b232000-03-13 16:01:29 +00002140}
2141
Antoine Pitrou2b0218a2012-09-06 00:59:49 +02002142/* The equivalent API, using per-thread state recursion info */
2143void
2144_PyTrash_thread_deposit_object(PyObject *op)
2145{
2146 PyThreadState *tstate = PyThreadState_GET();
2147 assert(PyObject_IS_GC(op));
Antoine Pitrou796564c2013-07-30 19:59:21 +02002148 assert(_PyGC_REFS(op) == _PyGC_REFS_UNTRACKED);
Antoine Pitrou2b0218a2012-09-06 00:59:49 +02002149 assert(op->ob_refcnt == 0);
2150 _Py_AS_GC(op)->gc.gc_prev = (PyGC_Head *) tstate->trash_delete_later;
2151 tstate->trash_delete_later = op;
2152}
2153
Tim Peters803526b2002-07-07 05:13:56 +00002154/* Dealloccate all the objects in the _PyTrash_delete_later list. Called when
2155 * the call-stack unwinds again.
2156 */
Guido van Rossumd724b232000-03-13 16:01:29 +00002157void
Fred Drake100814d2000-07-09 15:48:49 +00002158_PyTrash_destroy_chain(void)
Guido van Rossumd724b232000-03-13 16:01:29 +00002159{
Eric Snow2ebc5ce2017-09-07 23:51:28 -06002160 while (_PyRuntime.gc.trash_delete_later) {
2161 PyObject *op = _PyRuntime.gc.trash_delete_later;
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002162 destructor dealloc = Py_TYPE(op)->tp_dealloc;
Neil Schemenauerf589c052002-03-29 03:05:54 +00002163
Eric Snow2ebc5ce2017-09-07 23:51:28 -06002164 _PyRuntime.gc.trash_delete_later =
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002165 (PyObject*) _Py_AS_GC(op)->gc.gc_prev;
Neil Schemenauerf589c052002-03-29 03:05:54 +00002166
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002167 /* Call the deallocator directly. This used to try to
2168 * fool Py_DECREF into calling it indirectly, but
2169 * Py_DECREF was already called on this object, and in
2170 * assorted non-release builds calling Py_DECREF again ends
2171 * up distorting allocation statistics.
2172 */
2173 assert(op->ob_refcnt == 0);
Eric Snow2ebc5ce2017-09-07 23:51:28 -06002174 ++_PyRuntime.gc.trash_delete_nesting;
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002175 (*dealloc)(op);
Eric Snow2ebc5ce2017-09-07 23:51:28 -06002176 --_PyRuntime.gc.trash_delete_nesting;
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002177 }
Guido van Rossumd724b232000-03-13 16:01:29 +00002178}
Thomas Wouters49fd7fa2006-04-21 10:40:58 +00002179
Antoine Pitrou2b0218a2012-09-06 00:59:49 +02002180/* The equivalent API, using per-thread state recursion info */
2181void
2182_PyTrash_thread_destroy_chain(void)
2183{
2184 PyThreadState *tstate = PyThreadState_GET();
Xiang Zhanga66f9c62017-05-13 13:36:14 +08002185 /* We need to increase trash_delete_nesting here, otherwise,
2186 _PyTrash_thread_destroy_chain will be called recursively
2187 and then possibly crash. An example that may crash without
2188 increase:
2189 N = 500000 # need to be large enough
2190 ob = object()
2191 tups = [(ob,) for i in range(N)]
2192 for i in range(49):
2193 tups = [(tup,) for tup in tups]
2194 del tups
2195 */
2196 assert(tstate->trash_delete_nesting == 0);
2197 ++tstate->trash_delete_nesting;
Antoine Pitrou2b0218a2012-09-06 00:59:49 +02002198 while (tstate->trash_delete_later) {
2199 PyObject *op = tstate->trash_delete_later;
2200 destructor dealloc = Py_TYPE(op)->tp_dealloc;
2201
2202 tstate->trash_delete_later =
2203 (PyObject*) _Py_AS_GC(op)->gc.gc_prev;
2204
2205 /* Call the deallocator directly. This used to try to
2206 * fool Py_DECREF into calling it indirectly, but
2207 * Py_DECREF was already called on this object, and in
2208 * assorted non-release builds calling Py_DECREF again ends
2209 * up distorting allocation statistics.
2210 */
2211 assert(op->ob_refcnt == 0);
Antoine Pitrou2b0218a2012-09-06 00:59:49 +02002212 (*dealloc)(op);
Xiang Zhanga66f9c62017-05-13 13:36:14 +08002213 assert(tstate->trash_delete_nesting == 1);
Antoine Pitrou2b0218a2012-09-06 00:59:49 +02002214 }
Xiang Zhanga66f9c62017-05-13 13:36:14 +08002215 --tstate->trash_delete_nesting;
Antoine Pitrou2b0218a2012-09-06 00:59:49 +02002216}
2217
Martin v. Löwis4d0d4712010-12-03 20:14:31 +00002218#ifndef Py_TRACE_REFS
2219/* For Py_LIMITED_API, we need an out-of-line version of _Py_Dealloc.
2220 Define this here, so we can undefine the macro. */
2221#undef _Py_Dealloc
2222PyAPI_FUNC(void) _Py_Dealloc(PyObject *);
2223void
2224_Py_Dealloc(PyObject *op)
2225{
2226 _Py_INC_TPFREES(op) _Py_COUNT_ALLOCS_COMMA
2227 (*Py_TYPE(op)->tp_dealloc)(op);
2228}
2229#endif
2230
Thomas Wouters49fd7fa2006-04-21 10:40:58 +00002231#ifdef __cplusplus
2232}
2233#endif