check for errors after PyLong_Ssize_t

benjaminp · benjaminp · commit c9e435ee2b3e · 2008-09-30T02:22:04.000Z
patch from Victor Stinner #3977 reviewed by Amaury
diff --git a/Modules/_stringio.c b/Modules/_stringio.c
@@ -177,6 +177,10 @@ stringio_truncate(StringIOObject *self, PyObject *args)
 
     if (PyLong_Check(arg)) {
         size = PyLong_AsSsize_t(arg);
+        if (size == -1 && PyErr_Occurred())
+            return NULL;
+        if (size == -1 && PyErr_Occurred())
+            return NULL;
     }
     else if (arg == Py_None) {
         /* Truncate to current position if no argument is passed. */
