From 81d4b360e2942e0d6eb5c40769f0e3e2fedaca18 Mon Sep 17 00:00:00 2001 From: Hirokazu Yamamoto Date: Tue, 31 Mar 2009 20:22:13 +0000 Subject: [PATCH] Merged revisions 70879 via svnmerge from svn+ssh://pythondev@svn.python.org/python/trunk ........ r70879 | hirokazu.yamamoto | 2009-04-01 05:14:04 +0900 | 1 line Issue #5387: Fixed mmap.move crash by integer overflow. (take2) ........ --- Lib/test/test_mmap.py | 20 ++++++++++++++++---- Modules/mmapmodule.c | 2 +- 2 files changed, 17 insertions(+), 5 deletions(-) diff --git a/Lib/test/test_mmap.py b/Lib/test/test_mmap.py index a73dfedaa3c..5506956eb65 100644 --- a/Lib/test/test_mmap.py +++ b/Lib/test/test_mmap.py @@ -1,7 +1,7 @@ from test.test_support import TESTFN, run_unittest import mmap import unittest -import os, re +import os, re, itertools PAGESIZE = mmap.PAGESIZE @@ -350,9 +350,21 @@ class MmapTests(unittest.TestCase): self.assertEqual(m[:], expected) m.close() - # should not crash - m = mmap.mmap(-1, 1) - self.assertRaises(ValueError, m.move, 1, 1, -1) + # segfault test (Issue 5387) + m = mmap.mmap(-1, 100) + offsets = [-100, -1, 0, 1, 100] + for source, dest, size in itertools.product(offsets, offsets, offsets): + try: + m.move(source, dest, size) + except ValueError: + pass + self.assertRaises(ValueError, m.move, -1, -1, -1) + self.assertRaises(ValueError, m.move, -1, -1, 0) + self.assertRaises(ValueError, m.move, -1, 0, -1) + self.assertRaises(ValueError, m.move, 0, -1, -1) + self.assertRaises(ValueError, m.move, -1, 0, 0) + self.assertRaises(ValueError, m.move, 0, -1, 0) + self.assertRaises(ValueError, m.move, 0, 0, -1) m.close() def test_anonymous(self): diff --git a/Modules/mmapmodule.c b/Modules/mmapmodule.c index bd7f7cc8889..cbacc2fc252 100644 --- a/Modules/mmapmodule.c +++ b/Modules/mmapmodule.c @@ -617,7 +617,7 @@ mmap_move_method(mmap_object *self, PyObject *args) } else { /* bounds check the values */ unsigned long pos = src > dest ? src : dest; - if (self->size >= pos && count > self->size - pos) { + if (self->size < pos || count > self->size - pos) { PyErr_SetString(PyExc_ValueError, "source or destination out of range"); return NULL;