Add PyStackRef_IsMortal

markshannon · markshannon · commit 3a12d188a438 · 2024-12-05T14:25:39.000Z
diff --git a/Include/internal/pycore_stackref.h b/Include/internal/pycore_stackref.h
@@ -267,12 +267,20 @@ PyStackRef_HasCount(_PyStackRef ref)
     return ref.bits & Py_TAG_REFCNT;
 }
 
+/* Does this ref have an embedded refcount and refer to a mortal object (NULL is not mortal) */
 static inline bool
 PyStackRef_HasCountAndMortal(_PyStackRef ref)
 {
     return (ref.bits & Py_TAG_BITS) == Py_TAG_REFCNT;
 }
 
+/* Does this ref refer to a mortal object (NULL is not mortal) */
+static inline bool
+PyStackRef_IsMortal(_PyStackRef ref)
+{
+    return (ref.bits & Py_TAG_BITS) != Py_TAG_IMMORTAL;
+}
+
 static inline PyObject *
 PyStackRef_AsPyObjectBorrow(_PyStackRef ref)
 {

Original file line number	Diff line number	Diff line change
`@@ -267,12 +267,20 @@ PyStackRef_HasCount(_PyStackRef ref)`
`267`	`267`	`return ref.bits & Py_TAG_REFCNT;`
`268`	`268`	`}`
`269`	`269`
	`270`	`+/* Does this ref have an embedded refcount and refer to a mortal object (NULL is not mortal) */`
`270`	`271`	`static inline bool`
`271`	`272`	`PyStackRef_HasCountAndMortal(_PyStackRef ref)`
`272`	`273`	`{`
`273`	`274`	`return (ref.bits & Py_TAG_BITS) == Py_TAG_REFCNT;`
`274`	`275`	`}`
`275`	`276`
	`277`	`+/* Does this ref refer to a mortal object (NULL is not mortal) */`
	`278`	`+static inline bool`
	`279`	`+PyStackRef_IsMortal(_PyStackRef ref)`
	`280`	`+{`
	`281`	`+ return (ref.bits & Py_TAG_BITS) != Py_TAG_IMMORTAL;`
	`282`	`+}`
	`283`	`+`
`276`	`284`	`static inline PyObject *`
`277`	`285`	`PyStackRef_AsPyObjectBorrow(_PyStackRef ref)`
`278`	`286`	`{`