Propagate error when PyByteArray_Resize() fails in bytearray_translate()

CID 715334

Propagate error when PyByteArray_Resize() fails in bytearray_translate()
CID 715334
c731bbe6 · Christian Heimes · 8e090849 · c731bbe6
Commit c731bbe6 authored Jul 21, 2013 by Christian Heimes
Hide whitespace changes
Inline Side-by-side

Showing with 4 additions and 1 deletion

Objects/bytearrayobject.c Objects/bytearrayobject.c +4 -1

No files found.
--- a/Objects/bytearrayobject.c
+++ b/Objects/bytearrayobject.c
@@ -1506,7 +1506,10 @@ bytearray_translate(PyByteArrayObject *self, PyObject *args)
    }
    /* Fix the size of the resulting string */
    if (inlen > 0)
-        PyByteArray_Resize(result, output - output_start);
+        if (PyByteArray_Resize(result, output - output_start) < 0) {
+            Py_CLEAR(result);
+            goto done;
+        }

 done:
    if (tableobj != NULL)