From: Victor Stinner <victor.stinner@gmail.com>
Date: Tue, 29 Oct 2013 00:46:24 +0000 (+0100)
Subject: Issue #18408: Fix zipimport, handle PyUnicode_Substring() and get_subname() failures
X-Git-Tag: v3.4.0b1~491
X-Git-Url: http://git.ipfire.org/gitweb.cgi?a=commitdiff_plain;h=af8b7e823360f94714f8eded4a27572914851c7d;p=thirdparty%2FPython%2Fcpython.git

Issue #18408: Fix zipimport, handle PyUnicode_Substring() and get_subname() failures
---

diff --git a/Modules/zipimport.c b/Modules/zipimport.c
index 46b86a0cf4a9..dceca5e7cd4e 100644
--- a/Modules/zipimport.c
+++ b/Modules/zipimport.c
@@ -117,6 +117,8 @@ zipimporter_init(ZipImporter *self, PyObject *args, PyObject *kwds)
         if (flen == -1)
             break;
         filename = PyUnicode_Substring(path, 0, flen);
+        if (filename == NULL)
+            goto error;
     }
     if (filename == NULL) {
         PyErr_SetString(ZipImportError, "not a Zip file");
@@ -469,10 +471,13 @@ zipimporter_load_module(PyObject *obj, PyObject *args)
     if (ispackage) {
         /* add __path__ to the module *before* the code gets
            executed */
-        PyObject *pkgpath, *fullpath;
-        PyObject *subname = get_subname(fullname);
+        PyObject *pkgpath, *fullpath, *subname;
         int err;
 
+        subname = get_subname(fullname);
+        if (subname == NULL)
+            goto error;
+
         fullpath = PyUnicode_FromFormat("%U%c%U%U",
                                 self->archive, SEP,
                                 self->prefix, subname);