Issue #17645: convert an assert() into a proper exception in _Py_Mangle().

022db598 · Antoine Pitrou · 0e7df431 · 022db598
Kaydet (Commit) 022db598 authored Nis 06, 2013 tarafından Antoine Pitrou
Hide whitespace changes
Inline Side-by-side

Showing with 5 additions and 2 deletions

compile.c Python/compile.c +5 -2

No files found.
--- a/Python/compile.c
+++ b/Python/compile.c
@@ -221,8 +221,11 @@ _Py_Mangle(PyObject *privateobj, PyObject *ident)
    }
    plen = strlen(p);
-    assert(1 <= PY_SSIZE_T_MAX - nlen);
+    if (plen + nlen >= PY_SSIZE_T_MAX - 1) {
-    assert(1 + nlen <= PY_SSIZE_T_MAX - plen);
+        PyErr_SetString(PyExc_OverflowError,
+                        "private identifier too large to be mangled");
+        return NULL;
+    }
    ident = PyString_FromStringAndSize(NULL, 1 + nlen + plen);
    if (!ident)