mirror of
https://github.com/python/cpython.git
synced 2025-08-02 08:02:56 +00:00
Issue #23215: Multibyte codecs with custom error handlers that ignores errors
consumed too much memory and raised SystemError or MemoryError. Original patch by Aleksi Torhamo.
This commit is contained in:
parent
a3712a9a6c
commit
a1543cdcd6
3 changed files with 22 additions and 8 deletions
|
@ -182,8 +182,10 @@ expand_encodebuffer(MultibyteEncodeBuffer *buf, Py_ssize_t esize)
|
|||
orgsize = PyBytes_GET_SIZE(buf->outobj);
|
||||
incsize = (esize < (orgsize >> 1) ? (orgsize >> 1) | 1 : esize);
|
||||
|
||||
if (orgsize > PY_SSIZE_T_MAX - incsize)
|
||||
if (orgsize > PY_SSIZE_T_MAX - incsize) {
|
||||
PyErr_NoMemory();
|
||||
return -1;
|
||||
}
|
||||
|
||||
if (_PyBytes_Resize(&buf->outobj, orgsize + incsize) == -1)
|
||||
return -1;
|
||||
|
@ -194,11 +196,11 @@ expand_encodebuffer(MultibyteEncodeBuffer *buf, Py_ssize_t esize)
|
|||
|
||||
return 0;
|
||||
}
|
||||
#define REQUIRE_ENCODEBUFFER(buf, s) { \
|
||||
if ((s) < 1 || (buf)->outbuf + (s) > (buf)->outbuf_end) \
|
||||
#define REQUIRE_ENCODEBUFFER(buf, s) do { \
|
||||
if ((s) < 0 || (s) > (buf)->outbuf_end - (buf)->outbuf) \
|
||||
if (expand_encodebuffer(buf, s) == -1) \
|
||||
goto errorexit; \
|
||||
}
|
||||
} while(0)
|
||||
|
||||
|
||||
/**
|
||||
|
@ -332,10 +334,11 @@ multibytecodec_encerror(MultibyteCodec *codec,
|
|||
|
||||
assert(PyBytes_Check(retstr));
|
||||
retstrsize = PyBytes_GET_SIZE(retstr);
|
||||
REQUIRE_ENCODEBUFFER(buf, retstrsize);
|
||||
|
||||
memcpy(buf->outbuf, PyBytes_AS_STRING(retstr), retstrsize);
|
||||
buf->outbuf += retstrsize;
|
||||
if (retstrsize > 0) {
|
||||
REQUIRE_ENCODEBUFFER(buf, retstrsize);
|
||||
memcpy(buf->outbuf, PyBytes_AS_STRING(retstr), retstrsize);
|
||||
buf->outbuf += retstrsize;
|
||||
}
|
||||
|
||||
newpos = PyLong_AsSsize_t(PyTuple_GET_ITEM(retobj, 1));
|
||||
if (newpos < 0 && !PyErr_Occurred())
|
||||
|
|
Loading…
Add table
Add a link
Reference in a new issue