diff options
author | Tim Peters <tim.peters@gmail.com> | 2002-04-21 17:28:06 (GMT) |
---|---|---|
committer | Tim Peters <tim.peters@gmail.com> | 2002-04-21 17:28:06 (GMT) |
commit | 0eca65c4c5ade46c960ea21e0827b051d0cd8bf7 (patch) | |
tree | f7edd7f59ea4a5ed59a38fe3ec08c278846e041a /Objects/unicodeobject.c | |
parent | 0e0ee598fccac4fec66e2673a97770c286f0a1d7 (diff) | |
download | cpython-0eca65c4c5ade46c960ea21e0827b051d0cd8bf7.zip cpython-0eca65c4c5ade46c960ea21e0827b051d0cd8bf7.tar.gz cpython-0eca65c4c5ade46c960ea21e0827b051d0cd8bf7.tar.bz2 |
PyUnicode_EncodeUTF8(): tightened the memory asserts a bit, and at least
tried to catch some possible arithmetic overflows in the debug build.
Diffstat (limited to 'Objects/unicodeobject.c')
-rw-r--r-- | Objects/unicodeobject.c | 32 |
1 files changed, 20 insertions, 12 deletions
diff --git a/Objects/unicodeobject.c b/Objects/unicodeobject.c index 596cb38..8dbca6d 100644 --- a/Objects/unicodeobject.c +++ b/Objects/unicodeobject.c @@ -1185,15 +1185,17 @@ PyUnicode_EncodeUTF8(const Py_UNICODE *s, { PyObject *v; char *p; - int i = 0; - int overalloc = 2; int len; - - /* Short-cut for emtpy strings */ + int i = 0; + long overalloc = 2; + int nallocated; /* overalloc * size; PyString_ adds one more for \0 */ + + /* Short-cut for empty strings */ if (size == 0) return PyString_FromStringAndSize(NULL, 0); - v = PyString_FromStringAndSize(NULL, overalloc * size); + nallocated = Py_SAFE_DOWNCAST(overalloc * size, long, int); + v = PyString_FromStringAndSize(NULL, nallocated); if (v == NULL) return NULL; @@ -1211,7 +1213,7 @@ PyUnicode_EncodeUTF8(const Py_UNICODE *s, *p++ = (char)(0xc0 | (ch >> 6)); *p++ = (char)(0x80 | (ch & 0x3f)); } - + else { /* Encode UCS2 Unicode ordinals */ if (ch < 0x10000) { @@ -1230,9 +1232,11 @@ PyUnicode_EncodeUTF8(const Py_UNICODE *s, } if (overalloc < 3) { - len = (int)(p - PyString_AS_STRING(v)); + len = Py_SAFE_DOWNCAST(p-PyString_AS_STRING(v), long, int); + assert(len <= nallocated); overalloc = 3; - if (_PyString_Resize(&v, overalloc * size)) + nallocated = Py_SAFE_DOWNCAST(overalloc * size, long, int); + if (_PyString_Resize(&v, nallocated)) goto onError; p = PyString_AS_STRING(v) + len; } @@ -1245,9 +1249,11 @@ PyUnicode_EncodeUTF8(const Py_UNICODE *s, /* Encode UCS4 Unicode ordinals */ encodeUCS4: if (overalloc < 4) { - len = (int)(p - PyString_AS_STRING(v)); + len = Py_SAFE_DOWNCAST(p - PyString_AS_STRING(v), long, int); + assert(len <= nallocated); overalloc = 4; - if (_PyString_Resize(&v, overalloc * size)) + nallocated = Py_SAFE_DOWNCAST(overalloc * size, long, int); + if (_PyString_Resize(&v, nallocated)) goto onError; p = PyString_AS_STRING(v) + len; } @@ -1257,9 +1263,11 @@ PyUnicode_EncodeUTF8(const Py_UNICODE *s, *p++ = (char)(0x80 | (ch & 0x3f)); } } + *p = '\0'; - assert((p - PyString_AS_STRING(v)) <= overalloc*size); - if (_PyString_Resize(&v, (int)(p - PyString_AS_STRING(v)))) + len = Py_SAFE_DOWNCAST(p - PyString_AS_STRING(v), long, int); + assert(len <= nallocated); + if (_PyString_Resize(&v, len)) goto onError; return v; |