mirror of
https://github.com/python/cpython.git
synced 2025-07-24 11:44:31 +00:00
Issue #14172: Fix reference leak when marshalling a buffer-like object (other than a bytes object).
This commit is contained in:
commit
b2677c7397
3 changed files with 30 additions and 4 deletions
|
@ -409,11 +409,12 @@ w_object(PyObject *v, WFILE *p)
|
|||
else if (PyObject_CheckBuffer(v)) {
|
||||
/* Write unknown buffer-style objects as a string */
|
||||
char *s;
|
||||
PyBufferProcs *pb = v->ob_type->tp_as_buffer;
|
||||
Py_buffer view;
|
||||
if ((*pb->bf_getbuffer)(v, &view, PyBUF_SIMPLE) != 0) {
|
||||
if (PyObject_GetBuffer(v, &view, PyBUF_SIMPLE) != 0) {
|
||||
w_byte(TYPE_UNKNOWN, p);
|
||||
p->depth--;
|
||||
p->error = WFERR_UNMARSHALLABLE;
|
||||
return;
|
||||
}
|
||||
w_byte(TYPE_STRING, p);
|
||||
n = view.len;
|
||||
|
@ -425,8 +426,7 @@ w_object(PyObject *v, WFILE *p)
|
|||
}
|
||||
w_long((long)n, p);
|
||||
w_string(s, (int)n, p);
|
||||
if (pb->bf_releasebuffer != NULL)
|
||||
(*pb->bf_releasebuffer)(v, &view);
|
||||
PyBuffer_Release(&view);
|
||||
}
|
||||
else {
|
||||
w_byte(TYPE_UNKNOWN, p);
|
||||
|
|
Loading…
Add table
Add a link
Reference in a new issue