From 037310ea8a3655ac92b8df5d94c9c73cbadfd3be Mon Sep 17 00:00:00 2001 From: Elliott Sales de Andrade Date: Fri, 8 Nov 2024 00:58:24 -0500 Subject: [PATCH] Use std::unique_ptr in pybind11_getbuffer (#5435) * Use std::unique_ptr in pybind11_getbuffer * Move final assignment later --- include/pybind11/detail/class.h | 17 +++++++---------- 1 file changed, 7 insertions(+), 10 deletions(-) diff --git a/include/pybind11/detail/class.h b/include/pybind11/detail/class.h index 6f87f7d4c..88e6d60a9 100644 --- a/include/pybind11/detail/class.h +++ b/include/pybind11/detail/class.h @@ -583,9 +583,9 @@ extern "C" inline int pybind11_getbuffer(PyObject *obj, Py_buffer *view, int fla return -1; } std::memset(view, 0, sizeof(Py_buffer)); - buffer_info *info = nullptr; + std::unique_ptr info = nullptr; try { - info = tinfo->get_buffer(obj, tinfo->get_buffer_data); + info.reset(tinfo->get_buffer(obj, tinfo->get_buffer_data)); } catch (...) { try_translate_exceptions(); raise_from(PyExc_BufferError, "Error getting buffer"); @@ -596,7 +596,6 @@ extern "C" inline int pybind11_getbuffer(PyObject *obj, Py_buffer *view, int fla } if ((flags & PyBUF_WRITABLE) == PyBUF_WRITABLE && info->readonly) { - delete info; // view->obj = nullptr; // Was just memset to 0, so not necessary set_error(PyExc_BufferError, "Writable buffer requested for readonly storage"); return -1; @@ -604,9 +603,6 @@ extern "C" inline int pybind11_getbuffer(PyObject *obj, Py_buffer *view, int fla // Fill in all the information, and then downgrade as requested by the caller, or raise an // error if that's not possible. - view->obj = obj; - view->internal = info; - view->buf = info->ptr; view->itemsize = info->itemsize; view->len = view->itemsize; for (auto s : info->shape) { @@ -624,7 +620,6 @@ extern "C" inline int pybind11_getbuffer(PyObject *obj, Py_buffer *view, int fla if ((flags & PyBUF_C_CONTIGUOUS) == PyBUF_C_CONTIGUOUS) { if (PyBuffer_IsContiguous(view, 'C') == 0) { std::memset(view, 0, sizeof(Py_buffer)); - delete info; set_error(PyExc_BufferError, "C-contiguous buffer requested for discontiguous storage"); return -1; @@ -632,7 +627,6 @@ extern "C" inline int pybind11_getbuffer(PyObject *obj, Py_buffer *view, int fla } else if ((flags & PyBUF_F_CONTIGUOUS) == PyBUF_F_CONTIGUOUS) { if (PyBuffer_IsContiguous(view, 'F') == 0) { std::memset(view, 0, sizeof(Py_buffer)); - delete info; set_error(PyExc_BufferError, "Fortran-contiguous buffer requested for discontiguous storage"); return -1; @@ -640,7 +634,6 @@ extern "C" inline int pybind11_getbuffer(PyObject *obj, Py_buffer *view, int fla } else if ((flags & PyBUF_ANY_CONTIGUOUS) == PyBUF_ANY_CONTIGUOUS) { if (PyBuffer_IsContiguous(view, 'A') == 0) { std::memset(view, 0, sizeof(Py_buffer)); - delete info; set_error(PyExc_BufferError, "Contiguous buffer requested for discontiguous storage"); return -1; } @@ -650,7 +643,6 @@ extern "C" inline int pybind11_getbuffer(PyObject *obj, Py_buffer *view, int fla // https://docs.python.org/3/c-api/buffer.html#contiguity-requests if (PyBuffer_IsContiguous(view, 'C') == 0) { std::memset(view, 0, sizeof(Py_buffer)); - delete info; set_error(PyExc_BufferError, "C-contiguous buffer requested for discontiguous storage"); return -1; @@ -665,6 +657,11 @@ extern "C" inline int pybind11_getbuffer(PyObject *obj, Py_buffer *view, int fla } } + // Set these after all checks so they don't leak out into the caller, and can be automatically + // cleaned up on error. + view->buf = info->ptr; + view->internal = info.release(); + view->obj = obj; Py_INCREF(view->obj); return 0; }