fix overflow checking in PyBytes_Repr (closes #22519)

This commit is contained in:
Benjamin Peterson 2014-09-29 19:01:18 -04:00
parent 3bbb2e4844
commit 42ff105539
2 changed files with 18 additions and 12 deletions

View File

@ -10,6 +10,8 @@ What's New in Python 3.3.6 release candidate 1?
Core and Builtins Core and Builtins
----------------- -----------------
- Issue #22519: Fix overflow checking in PyBytes_Repr.
- Issue #22518: Fix integer overflow issues in latin-1 encoding. - Issue #22518: Fix integer overflow issues in latin-1 encoding.
Library Library

View File

@ -593,28 +593,27 @@ PyBytes_Repr(PyObject *obj, int smartquotes)
newsize = 3; /* b'' */ newsize = 3; /* b'' */
s = (unsigned char*)op->ob_sval; s = (unsigned char*)op->ob_sval;
for (i = 0; i < length; i++) { for (i = 0; i < length; i++) {
Py_ssize_t incr = 1;
switch(s[i]) { switch(s[i]) {
case '\'': squotes++; newsize++; break; case '\'': squotes++; break;
case '"': dquotes++; newsize++; break; case '"': dquotes++; break;
case '\\': case '\t': case '\n': case '\r': case '\\': case '\t': case '\n': case '\r':
newsize += 2; break; /* \C */ incr = 2; break; /* \C */
default: default:
if (s[i] < ' ' || s[i] >= 0x7f) if (s[i] < ' ' || s[i] >= 0x7f)
newsize += 4; /* \xHH */ incr = 4; /* \xHH */
else
newsize++;
} }
if (newsize > PY_SSIZE_T_MAX - incr)
goto overflow;
newsize += incr;
} }
quote = '\''; quote = '\'';
if (smartquotes && squotes && !dquotes) if (smartquotes && squotes && !dquotes)
quote = '"'; quote = '"';
if (squotes && quote == '\'') if (squotes && quote == '\'') {
if (newsize > PY_SSIZE_T_MAX - squotes)
goto overflow;
newsize += squotes; newsize += squotes;
if (newsize > (PY_SSIZE_T_MAX - sizeof(PyUnicodeObject) - 1)) {
PyErr_SetString(PyExc_OverflowError,
"bytes object is too large to make repr");
return NULL;
} }
v = PyUnicode_New(newsize, 127); v = PyUnicode_New(newsize, 127);
@ -646,6 +645,11 @@ PyBytes_Repr(PyObject *obj, int smartquotes)
*p++ = quote; *p++ = quote;
assert(_PyUnicode_CheckConsistency(v, 1)); assert(_PyUnicode_CheckConsistency(v, 1));
return v; return v;
overflow:
PyErr_SetString(PyExc_OverflowError,
"bytes object is too large to make repr");
return NULL;
} }
static PyObject * static PyObject *