Issue #6344: Fixed a crash of mmap.read() when passed a negative argument.

Reviewed by Amaury Forgeot d'Arc.
This commit is contained in:
Hirokazu Yamamoto 2009-06-29 13:25:16 +00:00
parent bcff47a6ec
commit 8839fd7372
2 changed files with 15 additions and 3 deletions

View file

@ -12,6 +12,8 @@ What's New in Python 2.7 alpha 1
Core and Builtins Core and Builtins
----------------- -----------------
- Issue #6344: Fixed a crash of mmap.read() when passed a negative argument.
- Issue #4856: Remove checks for win NT. - Issue #4856: Remove checks for win NT.
- Issue #2016: Fixed a crash in a corner case where the dictionary of keyword - Issue #2016: Fixed a crash in a corner case where the dictionary of keyword

View file

@ -232,7 +232,7 @@ static PyObject *
mmap_read_method(mmap_object *self, mmap_read_method(mmap_object *self,
PyObject *args) PyObject *args)
{ {
Py_ssize_t num_bytes; Py_ssize_t num_bytes, n;
PyObject *result; PyObject *result;
CHECK_VALID(NULL); CHECK_VALID(NULL);
@ -240,8 +240,18 @@ mmap_read_method(mmap_object *self,
return(NULL); return(NULL);
/* silently 'adjust' out-of-range requests */ /* silently 'adjust' out-of-range requests */
if (num_bytes > self->size - self->pos) { assert(self->size >= self->pos);
num_bytes -= (self->pos+num_bytes) - self->size; n = self->size - self->pos;
/* The difference can overflow, only if self->size is greater than
* PY_SSIZE_T_MAX. But then the operation cannot possibly succeed,
* because the mapped area and the returned string each need more
* than half of the addressable memory. So we clip the size, and let
* the code below raise MemoryError.
*/
if (n < 0)
n = PY_SSIZE_T_MAX;
if (num_bytes < 0 || num_bytes > n) {
num_bytes = n;
} }
result = Py_BuildValue("s#", self->data+self->pos, num_bytes); result = Py_BuildValue("s#", self->data+self->pos, num_bytes);
self->pos += num_bytes; self->pos += num_bytes;