Issue #11277: Add tests for mmap crash when using large sparse files on OS X.

Also, reduce code duplication in LargeMmapTests.

Original patch by Steffen Daode Nurpmeso.
This commit is contained in:
Nadeem Vawda 2011-05-07 13:17:16 +02:00
parent 3b066a7e82
commit 0d837ef2a5
1 changed files with 34 additions and 26 deletions

View File

@ -1,4 +1,5 @@
from test.test_support import TESTFN, run_unittest, import_module, unlink, requires
from test.test_support import (TESTFN, run_unittest, import_module, unlink,
requires, _2G, _4G)
import unittest
import os, re, itertools, socket, sys
@ -644,28 +645,20 @@ class LargeMmapTests(unittest.TestCase):
def tearDown(self):
unlink(TESTFN)
def _working_largefile(self):
# Only run if the current filesystem supports large files.
f = open(TESTFN, 'wb', buffering=0)
try:
f.seek(0x80000001)
f.write(b'x')
f.flush()
except (IOError, OverflowError):
raise unittest.SkipTest("filesystem does not have largefile support")
finally:
f.close()
unlink(TESTFN)
def test_large_offset(self):
def _create_test_file(self, num_zeroes, tail):
if sys.platform[:3] == 'win' or sys.platform == 'darwin':
requires('largefile',
'test requires %s bytes and a long time to run' % str(0x180000000))
self._working_largefile()
with open(TESTFN, 'wb') as f:
f.seek(0x14FFFFFFF)
f.write(b" ")
try:
f.seek(num_zeroes)
f.write(tail)
f.flush()
except (IOError, OverflowError):
raise unittest.SkipTest("filesystem does not have largefile support")
def test_large_offset(self):
self._create_test_file(0x14FFFFFFF, b" ")
with open(TESTFN, 'rb') as f:
m = mmap.mmap(f.fileno(), 0, offset=0x140000000, access=mmap.ACCESS_READ)
try:
@ -674,14 +667,7 @@ class LargeMmapTests(unittest.TestCase):
m.close()
def test_large_filesize(self):
if sys.platform[:3] == 'win' or sys.platform == 'darwin':
requires('largefile',
'test requires %s bytes and a long time to run' % str(0x180000000))
self._working_largefile()
with open(TESTFN, 'wb') as f:
f.seek(0x17FFFFFFF)
f.write(b" ")
self._create_test_file(0x17FFFFFFF, b" ")
with open(TESTFN, 'rb') as f:
m = mmap.mmap(f.fileno(), 0x10000, access=mmap.ACCESS_READ)
try:
@ -689,6 +675,28 @@ class LargeMmapTests(unittest.TestCase):
finally:
m.close()
# Issue 11277: mmap() with large (~4GB) sparse files crashes on OS X.
def _test_around_boundary(self, boundary):
tail = b' DEARdear '
start = boundary - len(tail) // 2
end = start + len(tail)
self._create_test_file(start, tail)
with open(TESTFN, 'rb') as f:
m = mmap.mmap(f.fileno(), 0, access=mmap.ACCESS_READ)
try:
self.assertEqual(m[start:end], tail)
finally:
m.close()
@unittest.skipUnless(sys.maxsize > _4G, "test cannot run on 32-bit systems")
def test_around_2GB(self):
self._test_around_boundary(_2G)
@unittest.skipUnless(sys.maxsize > _4G, "test cannot run on 32-bit systems")
def test_around_4GB(self):
self._test_around_boundary(_4G)
def test_main():
run_unittest(MmapTests, LargeMmapTests)