Fixed reference leak when read truncated pickle.

ca28eba3 · Serhiy Storchaka · cbbec1c5 · ca28eba3
Commit ca28eba3 authored Dec 01, 2015 by Serhiy Storchaka
Hide whitespace changes
Inline Side-by-side

Showing with 3 additions and 1 deletion

Modules/_pickle.c Modules/_pickle.c +3 -1

No files found.
--- a/Modules/_pickle.c
+++ b/Modules/_pickle.c
@@ -5115,8 +5115,10 @@ load_inst(UnpicklerObject *self)
        return -1;

    if ((len = _Unpickler_Readline(self, &s)) >= 0) {
-        if (len < 2)
+        if (len < 2) {
+            Py_DECREF(module_name);
            return bad_readline();
+        }
        class_name = PyUnicode_DecodeASCII(s, len - 1, "strict");
        if (class_name != NULL) {
            cls = find_class(self, module_name, class_name);