Fix refleaks introduced by 83da67651687.

758153ba · Antoine Pitrou · e45c0c5c · 758153ba
Commit 758153ba authored May 12, 2012 by Antoine Pitrou
Hide whitespace changes
Inline Side-by-side

Showing with 8 additions and 2 deletions

Objects/unicodeobject.c Objects/unicodeobject.c +8 -2

No files found.
--- a/Objects/unicodeobject.c
+++ b/Objects/unicodeobject.c
@@ -9079,8 +9079,11 @@ PyUnicode_Count(PyObject *str,
    buf1 = PyUnicode_DATA(str_obj);
    buf2 = PyUnicode_DATA(sub_obj);
    if (kind2 != kind) {
-        if (kind2 > kind)
+        if (kind2 > kind) {
+            Py_DECREF(sub_obj);
+            Py_DECREF(str_obj);
            return 0;
+        }
        buf2 = _PyUnicode_AsKind(sub_obj, kind);
    }
    if (!buf2)
@@ -10659,8 +10662,11 @@ PyUnicode_Contains(PyObject *container, PyObject *element)
    buf1 = PyUnicode_DATA(str);
    buf2 = PyUnicode_DATA(sub);
    if (kind2 != kind) {
-        if (kind2 > kind)
+        if (kind2 > kind) {
+            Py_DECREF(sub);
+            Py_DECREF(str);
            return 0;
+        }
        buf2 = _PyUnicode_AsKind(sub, kind);
    }
    if (!buf2) {