Add PyUnstable_Buffer_ReleaseInInterpreter() and PyUnstable_Buffer_Re…

…leaseInInterpreterAndRawFree().
python · ericsnowcurrently · Oct 9, 2023 · Sep 9, 2023 · Sep 11, 2023 · Sep 11, 2023
commit 5c5662dd9f2dd78ad80cd760c33406d98bd168d7
diff --git a/Include/pybuffer.h b/Include/pybuffer.h
@@ -100,6 +100,10 @@ PyAPI_FUNC(int) PyBuffer_FillInfo(Py_buffer *view, PyObject *o, void *buf,
 
 /* Releases a Py_buffer obtained from getbuffer ParseTuple's "s*". */
 PyAPI_FUNC(void) PyBuffer_Release(Py_buffer *view);
+PyAPI_FUNC(int) PyUnstable_Buffer_ReleaseInInterpreter(
+        PyInterpreterState *interp, Py_buffer *view);
+PyAPI_FUNC(int) PyUnstable_Buffer_ReleaseInInterpreterAndRawFree(
+        PyInterpreterState *interp, Py_buffer *view);
 
 /* Maximum number of dimensions */
 #define PyBUF_MAX_NDIM 64

diff --git a/Objects/abstract.c b/Objects/abstract.c
@@ -806,6 +806,27 @@ PyBuffer_Release(Py_buffer *view)
     Py_DECREF(obj);
 }
 
+static int
+_buffer_release_call(void *arg)
+{
+    PyBuffer_Release((Py_buffer *)arg);
+    return 0;
+}
+
+int
+PyUnstable_Buffer_ReleaseInInterpreter(PyInterpreterState *interp,
+                                       Py_buffer *view)
+{
+    return _Py_CallInInterpreter(interp, _buffer_release_call, view);
+}
+
+int
+PyUnstable_Buffer_ReleaseInInterpreterAndRawFree(PyInterpreterState *interp,
+                                                 Py_buffer *view)
+{
+    return _Py_CallInInterpreterAndRawFree(interp, _buffer_release_call, view);
+}
+
 PyObject *
 PyObject_Format(PyObject *obj, PyObject *format_spec)
 {