set_repr(): handle correctly PyUnicode_FromUnicode() error (MemoryError)

Bug found by the Clang Static Analyzer.
This commit is contained in:
Victor Stinner 2011-05-26 14:24:30 +02:00
parent 5572ba7e15
commit a1a807b6ef

View file

@ -607,7 +607,9 @@ set_repr(PySetObject *so)
goto done;
newsize = PyUnicode_GET_SIZE(listrepr);
result = PyUnicode_FromUnicode(NULL, newsize);
if (result) {
if (result == NULL)
goto done;
u = PyUnicode_AS_UNICODE(result);
*u++ = '{';
/* Omit the brackets from the listrepr */
@ -615,8 +617,8 @@ set_repr(PySetObject *so)
newsize-2);
u += newsize-2;
*u = '}';
}
Py_DECREF(listrepr);
if (Py_TYPE(so) != &PySet_Type) {
PyObject *tmp = PyUnicode_FromFormat("%s(%U)",
Py_TYPE(so)->tp_name,