diff options
-rw-r--r-- | Modules/pcremodule.c | 29 |
1 files changed, 26 insertions, 3 deletions
diff --git a/Modules/pcremodule.c b/Modules/pcremodule.c index fa98696..3b61697 100644 --- a/Modules/pcremodule.c +++ b/Modules/pcremodule.c @@ -489,8 +489,19 @@ PyPcre_expand(self, args) if (start!=i) { - PyList_Append(results, - PyString_FromStringAndSize((char *)repl+start, i-start)); + int status; + PyObject *s = PyString_FromStringAndSize( + (char *)repl+start, i-start); + if (s == NULL) { + Py_DECREF(results); + return NULL; + } + status = PyList_Append(results, s); + Py_DECREF(s); + if (status < 0) { + Py_DECREF(results); + return NULL; + } total_len += i-start; } i++; @@ -574,7 +585,19 @@ PyPcre_expand(self, args) if (start!=i) { - PyList_Append(results, PyString_FromStringAndSize((char *)repl+start, i-start)); + int status; + PyObject *s = PyString_FromStringAndSize((char *)repl+start, + i-start); + if (s == NULL) { + Py_DECREF(results); + return NULL; + } + status = PyList_Append(results, s); + Py_DECREF(s); + if (status < 0) { + Py_DECREF(results); + return NULL; + } total_len += i-start; } |