Propagate error when PyByteArray_Resize() fails in bytearray_translate()

CID 715334
lysnikolaou · Jul 21, 2013 · 6895947 · 6895947
2 parents 8ded80d + c731bbe
commit 6895947
Showing 1 changed file with 4 additions and 1 deletion.
diff --git a/Objects/bytearrayobject.c b/Objects/bytearrayobject.c
@@ -1508,7 +1508,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)