Nadeem Vawda | ced1056 | 2011-05-07 13:01:50 +0200 | [diff] [blame] | 1 | from test.support import (TESTFN, run_unittest, import_module, unlink, |
| 2 | requires, _2G, _4G) |
Thomas Wouters | 89f507f | 2006-12-13 04:49:30 +0000 | [diff] [blame] | 3 | import unittest |
Benjamin Peterson | 5dc8fab | 2010-10-31 01:44:49 +0000 | [diff] [blame] | 4 | import os |
| 5 | import re |
| 6 | import itertools |
| 7 | import socket |
| 8 | import sys |
Andrew M. Kuchling | e81b9cf | 2000-03-30 21:15:29 +0000 | [diff] [blame] | 9 | |
R. David Murray | a21e4ca | 2009-03-31 23:16:50 +0000 | [diff] [blame] | 10 | # Skip test if we can't import mmap. |
| 11 | mmap = import_module('mmap') |
| 12 | |
Andrew M. Kuchling | e81b9cf | 2000-03-30 21:15:29 +0000 | [diff] [blame] | 13 | PAGESIZE = mmap.PAGESIZE |
| 14 | |
Thomas Wouters | 89f507f | 2006-12-13 04:49:30 +0000 | [diff] [blame] | 15 | class MmapTests(unittest.TestCase): |
Fred Drake | 004d5e6 | 2000-10-23 17:22:08 +0000 | [diff] [blame] | 16 | |
Thomas Wouters | 89f507f | 2006-12-13 04:49:30 +0000 | [diff] [blame] | 17 | def setUp(self): |
| 18 | if os.path.exists(TESTFN): |
| 19 | os.unlink(TESTFN) |
Fred Drake | 004d5e6 | 2000-10-23 17:22:08 +0000 | [diff] [blame] | 20 | |
Thomas Wouters | 89f507f | 2006-12-13 04:49:30 +0000 | [diff] [blame] | 21 | def tearDown(self): |
Tim Peters | fd69208 | 2001-05-10 20:03:04 +0000 | [diff] [blame] | 22 | try: |
Fred Drake | 6278799 | 2001-05-11 14:29:21 +0000 | [diff] [blame] | 23 | os.unlink(TESTFN) |
Tim Peters | fd69208 | 2001-05-10 20:03:04 +0000 | [diff] [blame] | 24 | except OSError: |
| 25 | pass |
| 26 | |
Thomas Wouters | 89f507f | 2006-12-13 04:49:30 +0000 | [diff] [blame] | 27 | def test_basic(self): |
| 28 | # Test mmap module on Unix systems and Windows |
| 29 | |
| 30 | # Create a file to be mmap'ed. |
Guido van Rossum | b358a2c | 2007-07-16 19:29:02 +0000 | [diff] [blame] | 31 | f = open(TESTFN, 'bw+') |
Thomas Wouters | 89f507f | 2006-12-13 04:49:30 +0000 | [diff] [blame] | 32 | try: |
| 33 | # Write 2 pages worth of data to the file |
Guido van Rossum | b358a2c | 2007-07-16 19:29:02 +0000 | [diff] [blame] | 34 | f.write(b'\0'* PAGESIZE) |
| 35 | f.write(b'foo') |
| 36 | f.write(b'\0'* (PAGESIZE-3) ) |
Thomas Wouters | 89f507f | 2006-12-13 04:49:30 +0000 | [diff] [blame] | 37 | f.flush() |
| 38 | m = mmap.mmap(f.fileno(), 2 * PAGESIZE) |
Guido van Rossum | 456fe5d | 2007-07-16 19:42:05 +0000 | [diff] [blame] | 39 | finally: |
Thomas Wouters | 89f507f | 2006-12-13 04:49:30 +0000 | [diff] [blame] | 40 | f.close() |
| 41 | |
Guido van Rossum | 456fe5d | 2007-07-16 19:42:05 +0000 | [diff] [blame] | 42 | # Simple sanity checks |
Thomas Wouters | 89f507f | 2006-12-13 04:49:30 +0000 | [diff] [blame] | 43 | |
Guido van Rossum | 456fe5d | 2007-07-16 19:42:05 +0000 | [diff] [blame] | 44 | tp = str(type(m)) # SF bug 128713: segfaulted on Linux |
Benjamin Peterson | e099b37 | 2009-04-04 17:09:35 +0000 | [diff] [blame] | 45 | self.assertEqual(m.find(b'foo'), PAGESIZE) |
Thomas Wouters | 89f507f | 2006-12-13 04:49:30 +0000 | [diff] [blame] | 46 | |
Guido van Rossum | 456fe5d | 2007-07-16 19:42:05 +0000 | [diff] [blame] | 47 | self.assertEqual(len(m), 2*PAGESIZE) |
Thomas Wouters | 89f507f | 2006-12-13 04:49:30 +0000 | [diff] [blame] | 48 | |
Guido van Rossum | 98297ee | 2007-11-06 21:34:58 +0000 | [diff] [blame] | 49 | self.assertEqual(m[0], 0) |
Guido van Rossum | 456fe5d | 2007-07-16 19:42:05 +0000 | [diff] [blame] | 50 | self.assertEqual(m[0:3], b'\0\0\0') |
Thomas Wouters | 89f507f | 2006-12-13 04:49:30 +0000 | [diff] [blame] | 51 | |
Hirokazu Yamamoto | 0654ccd | 2009-02-18 16:38:00 +0000 | [diff] [blame] | 52 | # Shouldn't crash on boundary (Issue #5292) |
| 53 | self.assertRaises(IndexError, m.__getitem__, len(m)) |
| 54 | self.assertRaises(IndexError, m.__setitem__, len(m), b'\0') |
| 55 | |
Guido van Rossum | 456fe5d | 2007-07-16 19:42:05 +0000 | [diff] [blame] | 56 | # Modify the file's content |
Guido van Rossum | 98297ee | 2007-11-06 21:34:58 +0000 | [diff] [blame] | 57 | m[0] = b'3'[0] |
Guido van Rossum | 456fe5d | 2007-07-16 19:42:05 +0000 | [diff] [blame] | 58 | m[PAGESIZE +3: PAGESIZE +3+3] = b'bar' |
Thomas Wouters | 89f507f | 2006-12-13 04:49:30 +0000 | [diff] [blame] | 59 | |
Guido van Rossum | 456fe5d | 2007-07-16 19:42:05 +0000 | [diff] [blame] | 60 | # Check that the modification worked |
Guido van Rossum | 98297ee | 2007-11-06 21:34:58 +0000 | [diff] [blame] | 61 | self.assertEqual(m[0], b'3'[0]) |
Guido van Rossum | 456fe5d | 2007-07-16 19:42:05 +0000 | [diff] [blame] | 62 | self.assertEqual(m[0:3], b'3\0\0') |
| 63 | self.assertEqual(m[PAGESIZE-1 : PAGESIZE + 7], b'\0foobar\0') |
Thomas Wouters | 89f507f | 2006-12-13 04:49:30 +0000 | [diff] [blame] | 64 | |
Guido van Rossum | 456fe5d | 2007-07-16 19:42:05 +0000 | [diff] [blame] | 65 | m.flush() |
Thomas Wouters | 89f507f | 2006-12-13 04:49:30 +0000 | [diff] [blame] | 66 | |
Guido van Rossum | 456fe5d | 2007-07-16 19:42:05 +0000 | [diff] [blame] | 67 | # Test doing a regular expression match in an mmap'ed file |
Antoine Pitrou | fd03645 | 2008-08-19 17:56:33 +0000 | [diff] [blame] | 68 | match = re.search(b'[A-Za-z]+', m) |
Guido van Rossum | 456fe5d | 2007-07-16 19:42:05 +0000 | [diff] [blame] | 69 | if match is None: |
| 70 | self.fail('regex match on mmap failed!') |
| 71 | else: |
| 72 | start, end = match.span(0) |
| 73 | length = end - start |
Thomas Wouters | 89f507f | 2006-12-13 04:49:30 +0000 | [diff] [blame] | 74 | |
Guido van Rossum | 456fe5d | 2007-07-16 19:42:05 +0000 | [diff] [blame] | 75 | self.assertEqual(start, PAGESIZE) |
| 76 | self.assertEqual(end, PAGESIZE + 6) |
Thomas Wouters | 89f507f | 2006-12-13 04:49:30 +0000 | [diff] [blame] | 77 | |
Guido van Rossum | 456fe5d | 2007-07-16 19:42:05 +0000 | [diff] [blame] | 78 | # test seeking around (try to overflow the seek implementation) |
| 79 | m.seek(0,0) |
| 80 | self.assertEqual(m.tell(), 0) |
| 81 | m.seek(42,1) |
| 82 | self.assertEqual(m.tell(), 42) |
| 83 | m.seek(0,2) |
| 84 | self.assertEqual(m.tell(), len(m)) |
Thomas Wouters | 89f507f | 2006-12-13 04:49:30 +0000 | [diff] [blame] | 85 | |
Guido van Rossum | 456fe5d | 2007-07-16 19:42:05 +0000 | [diff] [blame] | 86 | # Try to seek to negative position... |
| 87 | self.assertRaises(ValueError, m.seek, -1) |
Thomas Wouters | 89f507f | 2006-12-13 04:49:30 +0000 | [diff] [blame] | 88 | |
Guido van Rossum | 456fe5d | 2007-07-16 19:42:05 +0000 | [diff] [blame] | 89 | # Try to seek beyond end of mmap... |
| 90 | self.assertRaises(ValueError, m.seek, 1, 2) |
Thomas Wouters | 89f507f | 2006-12-13 04:49:30 +0000 | [diff] [blame] | 91 | |
Guido van Rossum | 456fe5d | 2007-07-16 19:42:05 +0000 | [diff] [blame] | 92 | # Try to seek to negative position... |
| 93 | self.assertRaises(ValueError, m.seek, -len(m)-1, 2) |
Thomas Wouters | 89f507f | 2006-12-13 04:49:30 +0000 | [diff] [blame] | 94 | |
Guido van Rossum | 456fe5d | 2007-07-16 19:42:05 +0000 | [diff] [blame] | 95 | # Try resizing map |
| 96 | try: |
| 97 | m.resize(512) |
| 98 | except SystemError: |
| 99 | # resize() not supported |
| 100 | # No messages are printed, since the output of this test suite |
| 101 | # would then be different across platforms. |
| 102 | pass |
| 103 | else: |
| 104 | # resize() is supported |
| 105 | self.assertEqual(len(m), 512) |
| 106 | # Check that we can no longer seek beyond the new size. |
| 107 | self.assertRaises(ValueError, m.seek, 513, 0) |
| 108 | |
| 109 | # Check that the underlying file is truncated too |
| 110 | # (bug #728515) |
Victor Stinner | a6d2c76 | 2011-06-30 18:20:11 +0200 | [diff] [blame] | 111 | f = open(TESTFN, 'rb') |
Thomas Wouters | 89f507f | 2006-12-13 04:49:30 +0000 | [diff] [blame] | 112 | try: |
Thomas Wouters | 89f507f | 2006-12-13 04:49:30 +0000 | [diff] [blame] | 113 | f.seek(0, 2) |
| 114 | self.assertEqual(f.tell(), 512) |
Guido van Rossum | 456fe5d | 2007-07-16 19:42:05 +0000 | [diff] [blame] | 115 | finally: |
Thomas Wouters | 89f507f | 2006-12-13 04:49:30 +0000 | [diff] [blame] | 116 | f.close() |
Guido van Rossum | 456fe5d | 2007-07-16 19:42:05 +0000 | [diff] [blame] | 117 | self.assertEqual(m.size(), 512) |
Thomas Wouters | 89f507f | 2006-12-13 04:49:30 +0000 | [diff] [blame] | 118 | |
Guido van Rossum | 456fe5d | 2007-07-16 19:42:05 +0000 | [diff] [blame] | 119 | m.close() |
Thomas Wouters | 89f507f | 2006-12-13 04:49:30 +0000 | [diff] [blame] | 120 | |
| 121 | def test_access_parameter(self): |
| 122 | # Test for "access" keyword parameter |
Tim Peters | 5ebfd36 | 2001-11-13 23:11:19 +0000 | [diff] [blame] | 123 | mapsize = 10 |
Benjamin Peterson | 5dc8fab | 2010-10-31 01:44:49 +0000 | [diff] [blame] | 124 | with open(TESTFN, "wb") as fp: |
| 125 | fp.write(b"a"*mapsize) |
| 126 | with open(TESTFN, "rb") as f: |
| 127 | m = mmap.mmap(f.fileno(), mapsize, access=mmap.ACCESS_READ) |
| 128 | self.assertEqual(m[:], b'a'*mapsize, "Readonly memory map data incorrect.") |
Tim Peters | 5ebfd36 | 2001-11-13 23:11:19 +0000 | [diff] [blame] | 129 | |
Benjamin Peterson | 5dc8fab | 2010-10-31 01:44:49 +0000 | [diff] [blame] | 130 | # Ensuring that readonly mmap can't be slice assigned |
| 131 | try: |
| 132 | m[:] = b'b'*mapsize |
| 133 | except TypeError: |
| 134 | pass |
| 135 | else: |
| 136 | self.fail("Able to write to readonly memory map") |
Tim Peters | 5ebfd36 | 2001-11-13 23:11:19 +0000 | [diff] [blame] | 137 | |
Benjamin Peterson | 5dc8fab | 2010-10-31 01:44:49 +0000 | [diff] [blame] | 138 | # Ensuring that readonly mmap can't be item assigned |
| 139 | try: |
| 140 | m[0] = b'b' |
| 141 | except TypeError: |
| 142 | pass |
| 143 | else: |
| 144 | self.fail("Able to write to readonly memory map") |
Tim Peters | 5ebfd36 | 2001-11-13 23:11:19 +0000 | [diff] [blame] | 145 | |
Benjamin Peterson | 5dc8fab | 2010-10-31 01:44:49 +0000 | [diff] [blame] | 146 | # Ensuring that readonly mmap can't be write() to |
| 147 | try: |
| 148 | m.seek(0,0) |
| 149 | m.write(b'abc') |
| 150 | except TypeError: |
| 151 | pass |
| 152 | else: |
| 153 | self.fail("Able to write to readonly memory map") |
Tim Peters | 5ebfd36 | 2001-11-13 23:11:19 +0000 | [diff] [blame] | 154 | |
Benjamin Peterson | 5dc8fab | 2010-10-31 01:44:49 +0000 | [diff] [blame] | 155 | # Ensuring that readonly mmap can't be write_byte() to |
| 156 | try: |
| 157 | m.seek(0,0) |
| 158 | m.write_byte(b'd') |
| 159 | except TypeError: |
| 160 | pass |
| 161 | else: |
| 162 | self.fail("Able to write to readonly memory map") |
Tim Peters | 5ebfd36 | 2001-11-13 23:11:19 +0000 | [diff] [blame] | 163 | |
Benjamin Peterson | 5dc8fab | 2010-10-31 01:44:49 +0000 | [diff] [blame] | 164 | # Ensuring that readonly mmap can't be resized |
| 165 | try: |
| 166 | m.resize(2*mapsize) |
| 167 | except SystemError: # resize is not universally supported |
| 168 | pass |
| 169 | except TypeError: |
| 170 | pass |
| 171 | else: |
| 172 | self.fail("Able to resize readonly memory map") |
| 173 | with open(TESTFN, "rb") as fp: |
| 174 | self.assertEqual(fp.read(), b'a'*mapsize, |
| 175 | "Readonly memory map data file was modified") |
Tim Peters | 5ebfd36 | 2001-11-13 23:11:19 +0000 | [diff] [blame] | 176 | |
Thomas Wouters | 89f507f | 2006-12-13 04:49:30 +0000 | [diff] [blame] | 177 | # Opening mmap with size too big |
Benjamin Peterson | 5dc8fab | 2010-10-31 01:44:49 +0000 | [diff] [blame] | 178 | with open(TESTFN, "r+b") as f: |
| 179 | try: |
| 180 | m = mmap.mmap(f.fileno(), mapsize+1) |
| 181 | except ValueError: |
| 182 | # we do not expect a ValueError on Windows |
| 183 | # CAUTION: This also changes the size of the file on disk, and |
| 184 | # later tests assume that the length hasn't changed. We need to |
| 185 | # repair that. |
| 186 | if sys.platform.startswith('win'): |
| 187 | self.fail("Opening mmap with size+1 should work on Windows.") |
| 188 | else: |
| 189 | # we expect a ValueError on Unix, but not on Windows |
| 190 | if not sys.platform.startswith('win'): |
| 191 | self.fail("Opening mmap with size+1 should raise ValueError.") |
| 192 | m.close() |
Neal Norwitz | b567392 | 2002-09-05 21:48:07 +0000 | [diff] [blame] | 193 | if sys.platform.startswith('win'): |
Benjamin Peterson | 5dc8fab | 2010-10-31 01:44:49 +0000 | [diff] [blame] | 194 | # Repair damage from the resizing test. |
| 195 | with open(TESTFN, 'r+b') as f: |
| 196 | f.truncate(mapsize) |
Neal Norwitz | b567392 | 2002-09-05 21:48:07 +0000 | [diff] [blame] | 197 | |
Thomas Wouters | 89f507f | 2006-12-13 04:49:30 +0000 | [diff] [blame] | 198 | # Opening mmap with access=ACCESS_WRITE |
Benjamin Peterson | 5dc8fab | 2010-10-31 01:44:49 +0000 | [diff] [blame] | 199 | with open(TESTFN, "r+b") as f: |
| 200 | m = mmap.mmap(f.fileno(), mapsize, access=mmap.ACCESS_WRITE) |
| 201 | # Modifying write-through memory map |
| 202 | m[:] = b'c'*mapsize |
| 203 | self.assertEqual(m[:], b'c'*mapsize, |
| 204 | "Write-through memory map memory not updated properly.") |
| 205 | m.flush() |
| 206 | m.close() |
| 207 | with open(TESTFN, 'rb') as f: |
| 208 | stuff = f.read() |
Guido van Rossum | b358a2c | 2007-07-16 19:29:02 +0000 | [diff] [blame] | 209 | self.assertEqual(stuff, b'c'*mapsize, |
Tim Peters | 5ebfd36 | 2001-11-13 23:11:19 +0000 | [diff] [blame] | 210 | "Write-through memory map data file not updated properly.") |
| 211 | |
Thomas Wouters | 89f507f | 2006-12-13 04:49:30 +0000 | [diff] [blame] | 212 | # Opening mmap with access=ACCESS_COPY |
Benjamin Peterson | 5dc8fab | 2010-10-31 01:44:49 +0000 | [diff] [blame] | 213 | with open(TESTFN, "r+b") as f: |
| 214 | m = mmap.mmap(f.fileno(), mapsize, access=mmap.ACCESS_COPY) |
| 215 | # Modifying copy-on-write memory map |
| 216 | m[:] = b'd'*mapsize |
| 217 | self.assertEqual(m[:], b'd' * mapsize, |
| 218 | "Copy-on-write memory map data not written correctly.") |
| 219 | m.flush() |
| 220 | with open(TESTFN, "rb") as fp: |
| 221 | self.assertEqual(fp.read(), b'c'*mapsize, |
| 222 | "Copy-on-write test data file should not be modified.") |
| 223 | # Ensuring copy-on-write maps cannot be resized |
| 224 | self.assertRaises(TypeError, m.resize, 2*mapsize) |
| 225 | m.close() |
Thomas Wouters | 89f507f | 2006-12-13 04:49:30 +0000 | [diff] [blame] | 226 | |
| 227 | # Ensuring invalid access parameter raises exception |
Benjamin Peterson | 5dc8fab | 2010-10-31 01:44:49 +0000 | [diff] [blame] | 228 | with open(TESTFN, "r+b") as f: |
| 229 | self.assertRaises(ValueError, mmap.mmap, f.fileno(), mapsize, access=4) |
Tim Peters | 5ebfd36 | 2001-11-13 23:11:19 +0000 | [diff] [blame] | 230 | |
| 231 | if os.name == "posix": |
Tim Peters | 00cafa0 | 2001-11-13 23:39:47 +0000 | [diff] [blame] | 232 | # Try incompatible flags, prot and access parameters. |
Benjamin Peterson | 5dc8fab | 2010-10-31 01:44:49 +0000 | [diff] [blame] | 233 | with open(TESTFN, "r+b") as f: |
| 234 | self.assertRaises(ValueError, mmap.mmap, f.fileno(), mapsize, |
| 235 | flags=mmap.MAP_PRIVATE, |
| 236 | prot=mmap.PROT_READ, access=mmap.ACCESS_WRITE) |
Tim Peters | 5ebfd36 | 2001-11-13 23:11:19 +0000 | [diff] [blame] | 237 | |
Antoine Pitrou | 7b50c2c | 2011-03-06 01:47:18 +0100 | [diff] [blame] | 238 | # Try writing with PROT_EXEC and without PROT_WRITE |
| 239 | prot = mmap.PROT_READ | getattr(mmap, 'PROT_EXEC', 0) |
Antoine Pitrou | 16a0a0b | 2011-03-06 01:11:03 +0100 | [diff] [blame] | 240 | with open(TESTFN, "r+b") as f: |
Antoine Pitrou | 7b50c2c | 2011-03-06 01:47:18 +0100 | [diff] [blame] | 241 | m = mmap.mmap(f.fileno(), mapsize, prot=prot) |
Antoine Pitrou | 16a0a0b | 2011-03-06 01:11:03 +0100 | [diff] [blame] | 242 | self.assertRaises(TypeError, m.write, b"abcdef") |
| 243 | self.assertRaises(TypeError, m.write_byte, 0) |
| 244 | m.close() |
| 245 | |
Thomas Wouters | 89f507f | 2006-12-13 04:49:30 +0000 | [diff] [blame] | 246 | def test_bad_file_desc(self): |
| 247 | # Try opening a bad file descriptor... |
| 248 | self.assertRaises(mmap.error, mmap.mmap, -2, 4096) |
Neal Norwitz | 3b4fff8 | 2006-01-11 08:54:45 +0000 | [diff] [blame] | 249 | |
Thomas Wouters | 89f507f | 2006-12-13 04:49:30 +0000 | [diff] [blame] | 250 | def test_tougher_find(self): |
| 251 | # Do a tougher .find() test. SF bug 515943 pointed out that, in 2.2, |
| 252 | # searching for data with embedded \0 bytes didn't work. |
Benjamin Peterson | 5dc8fab | 2010-10-31 01:44:49 +0000 | [diff] [blame] | 253 | with open(TESTFN, 'wb+') as f: |
Tim Peters | c9ffa06 | 2002-03-08 05:43:32 +0000 | [diff] [blame] | 254 | |
Benjamin Peterson | 5dc8fab | 2010-10-31 01:44:49 +0000 | [diff] [blame] | 255 | data = b'aabaac\x00deef\x00\x00aa\x00' |
| 256 | n = len(data) |
| 257 | f.write(data) |
| 258 | f.flush() |
| 259 | m = mmap.mmap(f.fileno(), n) |
Tim Peters | c9ffa06 | 2002-03-08 05:43:32 +0000 | [diff] [blame] | 260 | |
| 261 | for start in range(n+1): |
| 262 | for finish in range(start, n+1): |
| 263 | slice = data[start : finish] |
Thomas Wouters | 89f507f | 2006-12-13 04:49:30 +0000 | [diff] [blame] | 264 | self.assertEqual(m.find(slice), data.find(slice)) |
Guido van Rossum | b358a2c | 2007-07-16 19:29:02 +0000 | [diff] [blame] | 265 | self.assertEqual(m.find(slice + b'x'), -1) |
Tim Peters | ddc82ea | 2003-01-13 21:38:45 +0000 | [diff] [blame] | 266 | m.close() |
Tim Peters | c9ffa06 | 2002-03-08 05:43:32 +0000 | [diff] [blame] | 267 | |
Georg Brandl | fceab5a | 2008-01-19 20:08:23 +0000 | [diff] [blame] | 268 | def test_find_end(self): |
| 269 | # test the new 'end' parameter works as expected |
Benjamin Peterson | e099b37 | 2009-04-04 17:09:35 +0000 | [diff] [blame] | 270 | f = open(TESTFN, 'wb+') |
| 271 | data = b'one two ones' |
Georg Brandl | fceab5a | 2008-01-19 20:08:23 +0000 | [diff] [blame] | 272 | n = len(data) |
| 273 | f.write(data) |
| 274 | f.flush() |
| 275 | m = mmap.mmap(f.fileno(), n) |
| 276 | f.close() |
| 277 | |
Benjamin Peterson | e099b37 | 2009-04-04 17:09:35 +0000 | [diff] [blame] | 278 | self.assertEqual(m.find(b'one'), 0) |
| 279 | self.assertEqual(m.find(b'ones'), 8) |
| 280 | self.assertEqual(m.find(b'one', 0, -1), 0) |
| 281 | self.assertEqual(m.find(b'one', 1), 8) |
| 282 | self.assertEqual(m.find(b'one', 1, -1), 8) |
| 283 | self.assertEqual(m.find(b'one', 1, -2), -1) |
Georg Brandl | fceab5a | 2008-01-19 20:08:23 +0000 | [diff] [blame] | 284 | |
| 285 | |
| 286 | def test_rfind(self): |
| 287 | # test the new 'end' parameter works as expected |
Benjamin Peterson | e099b37 | 2009-04-04 17:09:35 +0000 | [diff] [blame] | 288 | f = open(TESTFN, 'wb+') |
| 289 | data = b'one two ones' |
Georg Brandl | fceab5a | 2008-01-19 20:08:23 +0000 | [diff] [blame] | 290 | n = len(data) |
| 291 | f.write(data) |
| 292 | f.flush() |
| 293 | m = mmap.mmap(f.fileno(), n) |
| 294 | f.close() |
| 295 | |
Benjamin Peterson | e099b37 | 2009-04-04 17:09:35 +0000 | [diff] [blame] | 296 | self.assertEqual(m.rfind(b'one'), 8) |
| 297 | self.assertEqual(m.rfind(b'one '), 0) |
| 298 | self.assertEqual(m.rfind(b'one', 0, -1), 8) |
| 299 | self.assertEqual(m.rfind(b'one', 0, -2), 0) |
| 300 | self.assertEqual(m.rfind(b'one', 1, -1), 8) |
| 301 | self.assertEqual(m.rfind(b'one', 1, -2), -1) |
Georg Brandl | fceab5a | 2008-01-19 20:08:23 +0000 | [diff] [blame] | 302 | |
| 303 | |
Thomas Wouters | 89f507f | 2006-12-13 04:49:30 +0000 | [diff] [blame] | 304 | def test_double_close(self): |
| 305 | # make sure a double close doesn't crash on Solaris (Bug# 665913) |
Guido van Rossum | b358a2c | 2007-07-16 19:29:02 +0000 | [diff] [blame] | 306 | f = open(TESTFN, 'wb+') |
Tim Peters | c9ffa06 | 2002-03-08 05:43:32 +0000 | [diff] [blame] | 307 | |
Guido van Rossum | b358a2c | 2007-07-16 19:29:02 +0000 | [diff] [blame] | 308 | f.write(2**16 * b'a') # Arbitrary character |
Neal Norwitz | e604c02 | 2003-01-10 20:52:16 +0000 | [diff] [blame] | 309 | f.close() |
| 310 | |
Victor Stinner | a6d2c76 | 2011-06-30 18:20:11 +0200 | [diff] [blame] | 311 | f = open(TESTFN, 'rb') |
Tim Peters | ddc82ea | 2003-01-13 21:38:45 +0000 | [diff] [blame] | 312 | mf = mmap.mmap(f.fileno(), 2**16, access=mmap.ACCESS_READ) |
Neal Norwitz | e604c02 | 2003-01-10 20:52:16 +0000 | [diff] [blame] | 313 | mf.close() |
| 314 | mf.close() |
| 315 | f.close() |
| 316 | |
Thomas Wouters | 89f507f | 2006-12-13 04:49:30 +0000 | [diff] [blame] | 317 | def test_entire_file(self): |
| 318 | # test mapping of entire file by passing 0 for map length |
| 319 | if hasattr(os, "stat"): |
Guido van Rossum | b358a2c | 2007-07-16 19:29:02 +0000 | [diff] [blame] | 320 | f = open(TESTFN, "wb+") |
Tim Peters | c9ffa06 | 2002-03-08 05:43:32 +0000 | [diff] [blame] | 321 | |
Guido van Rossum | b358a2c | 2007-07-16 19:29:02 +0000 | [diff] [blame] | 322 | f.write(2**16 * b'm') # Arbitrary character |
Martin v. Löwis | 7fe60c0 | 2005-03-03 11:22:44 +0000 | [diff] [blame] | 323 | f.close() |
| 324 | |
| 325 | f = open(TESTFN, "rb+") |
Tim Peters | eba28be | 2005-03-28 01:08:02 +0000 | [diff] [blame] | 326 | mf = mmap.mmap(f.fileno(), 0) |
Thomas Wouters | 89f507f | 2006-12-13 04:49:30 +0000 | [diff] [blame] | 327 | self.assertEqual(len(mf), 2**16, "Map size should equal file size.") |
Guido van Rossum | b358a2c | 2007-07-16 19:29:02 +0000 | [diff] [blame] | 328 | self.assertEqual(mf.read(2**16), 2**16 * b"m") |
Martin v. Löwis | 7fe60c0 | 2005-03-03 11:22:44 +0000 | [diff] [blame] | 329 | mf.close() |
| 330 | f.close() |
| 331 | |
Antoine Pitrou | 85f4615 | 2011-01-15 16:17:07 +0000 | [diff] [blame] | 332 | def test_length_0_offset(self): |
| 333 | # Issue #10916: test mapping of remainder of file by passing 0 for |
| 334 | # map length with an offset doesn't cause a segfault. |
| 335 | if not hasattr(os, "stat"): |
| 336 | self.skipTest("needs os.stat") |
Antoine Pitrou | d0ebc75 | 2011-01-15 17:25:58 +0000 | [diff] [blame] | 337 | # NOTE: allocation granularity is currently 65536 under Win64, |
| 338 | # and therefore the minimum offset alignment. |
| 339 | with open(TESTFN, "wb") as f: |
| 340 | f.write((65536 * 2) * b'm') # Arbitrary character |
Antoine Pitrou | 85f4615 | 2011-01-15 16:17:07 +0000 | [diff] [blame] | 341 | |
| 342 | with open(TESTFN, "rb") as f: |
Antoine Pitrou | d0ebc75 | 2011-01-15 17:25:58 +0000 | [diff] [blame] | 343 | with mmap.mmap(f.fileno(), 0, offset=65536, access=mmap.ACCESS_READ) as mf: |
| 344 | self.assertRaises(IndexError, mf.__getitem__, 80000) |
Antoine Pitrou | 85f4615 | 2011-01-15 16:17:07 +0000 | [diff] [blame] | 345 | |
Antoine Pitrou | 305bc9e | 2011-01-20 21:07:24 +0000 | [diff] [blame] | 346 | def test_length_0_large_offset(self): |
| 347 | # Issue #10959: test mapping of a file by passing 0 for |
| 348 | # map length with a large offset doesn't cause a segfault. |
| 349 | if not hasattr(os, "stat"): |
| 350 | self.skipTest("needs os.stat") |
| 351 | |
| 352 | with open(TESTFN, "wb") as f: |
| 353 | f.write(115699 * b'm') # Arbitrary character |
| 354 | |
| 355 | with open(TESTFN, "w+b") as f: |
| 356 | self.assertRaises(ValueError, mmap.mmap, f.fileno(), 0, |
| 357 | offset=2147418112) |
| 358 | |
Thomas Wouters | 89f507f | 2006-12-13 04:49:30 +0000 | [diff] [blame] | 359 | def test_move(self): |
| 360 | # make move works everywhere (64-bit format problem earlier) |
Guido van Rossum | b358a2c | 2007-07-16 19:29:02 +0000 | [diff] [blame] | 361 | f = open(TESTFN, 'wb+') |
Tim Peters | eba28be | 2005-03-28 01:08:02 +0000 | [diff] [blame] | 362 | |
Guido van Rossum | b358a2c | 2007-07-16 19:29:02 +0000 | [diff] [blame] | 363 | f.write(b"ABCDEabcde") # Arbitrary character |
Neal Norwitz | 8856fb7 | 2005-12-18 03:34:22 +0000 | [diff] [blame] | 364 | f.flush() |
| 365 | |
| 366 | mf = mmap.mmap(f.fileno(), 10) |
| 367 | mf.move(5, 0, 5) |
Guido van Rossum | b358a2c | 2007-07-16 19:29:02 +0000 | [diff] [blame] | 368 | self.assertEqual(mf[:], b"ABCDEABCDE", "Map move should have duplicated front 5") |
Neal Norwitz | 8856fb7 | 2005-12-18 03:34:22 +0000 | [diff] [blame] | 369 | mf.close() |
| 370 | f.close() |
| 371 | |
Hirokazu Yamamoto | 16caab0 | 2009-03-31 13:44:06 +0000 | [diff] [blame] | 372 | # more excessive test |
| 373 | data = b"0123456789" |
| 374 | for dest in range(len(data)): |
| 375 | for src in range(len(data)): |
| 376 | for count in range(len(data) - max(dest, src)): |
| 377 | expected = data[:dest] + data[src:src+count] + data[dest+count:] |
| 378 | m = mmap.mmap(-1, len(data)) |
| 379 | m[:] = data |
| 380 | m.move(dest, src, count) |
| 381 | self.assertEqual(m[:], expected) |
| 382 | m.close() |
| 383 | |
Hirokazu Yamamoto | 2ca1501 | 2009-03-31 20:43:56 +0000 | [diff] [blame] | 384 | # segfault test (Issue 5387) |
| 385 | m = mmap.mmap(-1, 100) |
| 386 | offsets = [-100, -1, 0, 1, 100] |
| 387 | for source, dest, size in itertools.product(offsets, offsets, offsets): |
| 388 | try: |
| 389 | m.move(source, dest, size) |
| 390 | except ValueError: |
| 391 | pass |
Benjamin Peterson | ef3e4c2 | 2009-04-11 19:48:14 +0000 | [diff] [blame] | 392 | |
| 393 | offsets = [(-1, -1, -1), (-1, -1, 0), (-1, 0, -1), (0, -1, -1), |
| 394 | (-1, 0, 0), (0, -1, 0), (0, 0, -1)] |
| 395 | for source, dest, size in offsets: |
| 396 | self.assertRaises(ValueError, m.move, source, dest, size) |
| 397 | |
Hirokazu Yamamoto | 16caab0 | 2009-03-31 13:44:06 +0000 | [diff] [blame] | 398 | m.close() |
| 399 | |
Benjamin Peterson | ef3e4c2 | 2009-04-11 19:48:14 +0000 | [diff] [blame] | 400 | m = mmap.mmap(-1, 1) # single byte |
| 401 | self.assertRaises(ValueError, m.move, 0, 0, 2) |
| 402 | self.assertRaises(ValueError, m.move, 1, 0, 1) |
| 403 | self.assertRaises(ValueError, m.move, 0, 1, 1) |
| 404 | m.move(0, 0, 1) |
| 405 | m.move(0, 0, 0) |
| 406 | |
| 407 | |
Thomas Wouters | 89f507f | 2006-12-13 04:49:30 +0000 | [diff] [blame] | 408 | def test_anonymous(self): |
| 409 | # anonymous mmap.mmap(-1, PAGE) |
| 410 | m = mmap.mmap(-1, PAGESIZE) |
Guido van Rossum | 805365e | 2007-05-07 22:24:25 +0000 | [diff] [blame] | 411 | for x in range(PAGESIZE): |
Guido van Rossum | 98297ee | 2007-11-06 21:34:58 +0000 | [diff] [blame] | 412 | self.assertEqual(m[x], 0, |
| 413 | "anonymously mmap'ed contents should be zero") |
Neal Norwitz | 8856fb7 | 2005-12-18 03:34:22 +0000 | [diff] [blame] | 414 | |
Guido van Rossum | 805365e | 2007-05-07 22:24:25 +0000 | [diff] [blame] | 415 | for x in range(PAGESIZE): |
Guido van Rossum | 98297ee | 2007-11-06 21:34:58 +0000 | [diff] [blame] | 416 | b = x & 0xff |
Guido van Rossum | b358a2c | 2007-07-16 19:29:02 +0000 | [diff] [blame] | 417 | m[x] = b |
| 418 | self.assertEqual(m[x], b) |
Neal Norwitz | 0e6bc8c | 2006-02-05 05:45:43 +0000 | [diff] [blame] | 419 | |
Charles-François Natali | 4dd453c | 2011-06-08 19:18:14 +0200 | [diff] [blame] | 420 | def test_read_all(self): |
| 421 | m = mmap.mmap(-1, 16) |
| 422 | self.addCleanup(m.close) |
| 423 | |
| 424 | # With no parameters, or None or a negative argument, reads all |
| 425 | m.write(bytes(range(16))) |
| 426 | m.seek(0) |
| 427 | self.assertEqual(m.read(), bytes(range(16))) |
| 428 | m.seek(8) |
| 429 | self.assertEqual(m.read(), bytes(range(8, 16))) |
| 430 | m.seek(16) |
| 431 | self.assertEqual(m.read(), b'') |
| 432 | m.seek(3) |
| 433 | self.assertEqual(m.read(None), bytes(range(3, 16))) |
| 434 | m.seek(4) |
| 435 | self.assertEqual(m.read(-1), bytes(range(4, 16))) |
| 436 | m.seek(5) |
| 437 | self.assertEqual(m.read(-2), bytes(range(5, 16))) |
| 438 | m.seek(9) |
| 439 | self.assertEqual(m.read(-42), bytes(range(9, 16))) |
| 440 | |
| 441 | def test_read_invalid_arg(self): |
| 442 | m = mmap.mmap(-1, 16) |
| 443 | self.addCleanup(m.close) |
| 444 | |
| 445 | self.assertRaises(TypeError, m.read, 'foo') |
| 446 | self.assertRaises(TypeError, m.read, 5.5) |
| 447 | self.assertRaises(TypeError, m.read, [1, 2, 3]) |
| 448 | |
Thomas Wouters | ed03b41 | 2007-08-28 21:37:11 +0000 | [diff] [blame] | 449 | def test_extended_getslice(self): |
| 450 | # Test extended slicing by comparing with list slicing. |
| 451 | s = bytes(reversed(range(256))) |
| 452 | m = mmap.mmap(-1, len(s)) |
| 453 | m[:] = s |
| 454 | self.assertEqual(m[:], s) |
| 455 | indices = (0, None, 1, 3, 19, 300, -1, -2, -31, -300) |
| 456 | for start in indices: |
| 457 | for stop in indices: |
| 458 | # Skip step 0 (invalid) |
| 459 | for step in indices[1:]: |
| 460 | self.assertEqual(m[start:stop:step], |
| 461 | s[start:stop:step]) |
| 462 | |
| 463 | def test_extended_set_del_slice(self): |
| 464 | # Test extended slicing by comparing with list slicing. |
| 465 | s = bytes(reversed(range(256))) |
| 466 | m = mmap.mmap(-1, len(s)) |
| 467 | indices = (0, None, 1, 3, 19, 300, -1, -2, -31, -300) |
| 468 | for start in indices: |
| 469 | for stop in indices: |
| 470 | # Skip invalid step 0 |
| 471 | for step in indices[1:]: |
| 472 | m[:] = s |
| 473 | self.assertEqual(m[:], s) |
| 474 | L = list(s) |
| 475 | # Make sure we have a slice of exactly the right length, |
| 476 | # but with different data. |
| 477 | data = L[start:stop:step] |
| 478 | data = bytes(reversed(data)) |
| 479 | L[start:stop:step] = data |
| 480 | m[start:stop:step] = data |
Ezio Melotti | b3aedd4 | 2010-11-20 19:04:17 +0000 | [diff] [blame] | 481 | self.assertEqual(m[:], bytes(L)) |
Thomas Wouters | ed03b41 | 2007-08-28 21:37:11 +0000 | [diff] [blame] | 482 | |
Guido van Rossum | 8ce8a78 | 2007-11-01 19:42:39 +0000 | [diff] [blame] | 483 | def make_mmap_file (self, f, halfsize): |
| 484 | # Write 2 pages worth of data to the file |
| 485 | f.write (b'\0' * halfsize) |
| 486 | f.write (b'foo') |
| 487 | f.write (b'\0' * (halfsize - 3)) |
| 488 | f.flush () |
| 489 | return mmap.mmap (f.fileno(), 0) |
| 490 | |
| 491 | def test_offset (self): |
| 492 | f = open (TESTFN, 'w+b') |
| 493 | |
| 494 | try: # unlink TESTFN no matter what |
| 495 | halfsize = mmap.ALLOCATIONGRANULARITY |
| 496 | m = self.make_mmap_file (f, halfsize) |
| 497 | m.close () |
| 498 | f.close () |
| 499 | |
| 500 | mapsize = halfsize * 2 |
| 501 | # Try invalid offset |
| 502 | f = open(TESTFN, "r+b") |
| 503 | for offset in [-2, -1, None]: |
| 504 | try: |
| 505 | m = mmap.mmap(f.fileno(), mapsize, offset=offset) |
| 506 | self.assertEqual(0, 1) |
| 507 | except (ValueError, TypeError, OverflowError): |
| 508 | pass |
| 509 | else: |
| 510 | self.assertEqual(0, 0) |
| 511 | f.close() |
| 512 | |
| 513 | # Try valid offset, hopefully 8192 works on all OSes |
| 514 | f = open(TESTFN, "r+b") |
| 515 | m = mmap.mmap(f.fileno(), mapsize - halfsize, offset=halfsize) |
| 516 | self.assertEqual(m[0:3], b'foo') |
| 517 | f.close() |
Hirokazu Yamamoto | 0654ccd | 2009-02-18 16:38:00 +0000 | [diff] [blame] | 518 | |
| 519 | # Try resizing map |
| 520 | try: |
| 521 | m.resize(512) |
| 522 | except SystemError: |
| 523 | pass |
| 524 | else: |
| 525 | # resize() is supported |
| 526 | self.assertEqual(len(m), 512) |
| 527 | # Check that we can no longer seek beyond the new size. |
| 528 | self.assertRaises(ValueError, m.seek, 513, 0) |
| 529 | # Check that the content is not changed |
| 530 | self.assertEqual(m[0:3], b'foo') |
| 531 | |
| 532 | # Check that the underlying file is truncated too |
Victor Stinner | a6d2c76 | 2011-06-30 18:20:11 +0200 | [diff] [blame] | 533 | f = open(TESTFN, 'rb') |
Hirokazu Yamamoto | 0654ccd | 2009-02-18 16:38:00 +0000 | [diff] [blame] | 534 | f.seek(0, 2) |
| 535 | self.assertEqual(f.tell(), halfsize + 512) |
| 536 | f.close() |
| 537 | self.assertEqual(m.size(), halfsize + 512) |
| 538 | |
Guido van Rossum | 8ce8a78 | 2007-11-01 19:42:39 +0000 | [diff] [blame] | 539 | m.close() |
| 540 | |
| 541 | finally: |
| 542 | f.close() |
| 543 | try: |
| 544 | os.unlink(TESTFN) |
| 545 | except OSError: |
| 546 | pass |
| 547 | |
Christian Heimes | 1af737c | 2008-01-23 08:24:23 +0000 | [diff] [blame] | 548 | def test_subclass(self): |
| 549 | class anon_mmap(mmap.mmap): |
| 550 | def __new__(klass, *args, **kwargs): |
| 551 | return mmap.mmap.__new__(klass, -1, *args, **kwargs) |
| 552 | anon_mmap(PAGESIZE) |
| 553 | |
Christian Heimes | a156e09 | 2008-02-16 07:38:31 +0000 | [diff] [blame] | 554 | def test_prot_readonly(self): |
Christian Heimes | 18c6689 | 2008-02-17 13:31:39 +0000 | [diff] [blame] | 555 | if not hasattr(mmap, 'PROT_READ'): |
| 556 | return |
Christian Heimes | a156e09 | 2008-02-16 07:38:31 +0000 | [diff] [blame] | 557 | mapsize = 10 |
Benjamin Peterson | 5dc8fab | 2010-10-31 01:44:49 +0000 | [diff] [blame] | 558 | with open(TESTFN, "wb") as fp: |
| 559 | fp.write(b"a"*mapsize) |
Christian Heimes | a156e09 | 2008-02-16 07:38:31 +0000 | [diff] [blame] | 560 | f = open(TESTFN, "rb") |
| 561 | m = mmap.mmap(f.fileno(), mapsize, prot=mmap.PROT_READ) |
| 562 | self.assertRaises(TypeError, m.write, "foo") |
Martin v. Löwis | e1e9f23 | 2008-04-01 06:17:46 +0000 | [diff] [blame] | 563 | f.close() |
Christian Heimes | 1af737c | 2008-01-23 08:24:23 +0000 | [diff] [blame] | 564 | |
Christian Heimes | 7131fd9 | 2008-02-19 14:21:46 +0000 | [diff] [blame] | 565 | def test_error(self): |
Benjamin Peterson | c9c0f20 | 2009-06-30 23:06:06 +0000 | [diff] [blame] | 566 | self.assertTrue(issubclass(mmap.error, EnvironmentError)) |
Benjamin Peterson | 577473f | 2010-01-19 00:09:57 +0000 | [diff] [blame] | 567 | self.assertIn("mmap.error", str(mmap.error)) |
Christian Heimes | 7131fd9 | 2008-02-19 14:21:46 +0000 | [diff] [blame] | 568 | |
Hirokazu Yamamoto | 39c6dea | 2009-02-28 10:56:50 +0000 | [diff] [blame] | 569 | def test_io_methods(self): |
| 570 | data = b"0123456789" |
Benjamin Peterson | 5dc8fab | 2010-10-31 01:44:49 +0000 | [diff] [blame] | 571 | with open(TESTFN, "wb") as fp: |
| 572 | fp.write(b"x"*len(data)) |
Hirokazu Yamamoto | 39c6dea | 2009-02-28 10:56:50 +0000 | [diff] [blame] | 573 | f = open(TESTFN, "r+b") |
| 574 | m = mmap.mmap(f.fileno(), len(data)) |
| 575 | f.close() |
| 576 | # Test write_byte() |
| 577 | for i in range(len(data)): |
Ezio Melotti | b3aedd4 | 2010-11-20 19:04:17 +0000 | [diff] [blame] | 578 | self.assertEqual(m.tell(), i) |
Benjamin Peterson | e099b37 | 2009-04-04 17:09:35 +0000 | [diff] [blame] | 579 | m.write_byte(data[i]) |
Ezio Melotti | b3aedd4 | 2010-11-20 19:04:17 +0000 | [diff] [blame] | 580 | self.assertEqual(m.tell(), i+1) |
Benjamin Peterson | e099b37 | 2009-04-04 17:09:35 +0000 | [diff] [blame] | 581 | self.assertRaises(ValueError, m.write_byte, b"x"[0]) |
Ezio Melotti | b3aedd4 | 2010-11-20 19:04:17 +0000 | [diff] [blame] | 582 | self.assertEqual(m[:], data) |
Hirokazu Yamamoto | 39c6dea | 2009-02-28 10:56:50 +0000 | [diff] [blame] | 583 | # Test read_byte() |
| 584 | m.seek(0) |
| 585 | for i in range(len(data)): |
Ezio Melotti | b3aedd4 | 2010-11-20 19:04:17 +0000 | [diff] [blame] | 586 | self.assertEqual(m.tell(), i) |
| 587 | self.assertEqual(m.read_byte(), data[i]) |
| 588 | self.assertEqual(m.tell(), i+1) |
Hirokazu Yamamoto | 39c6dea | 2009-02-28 10:56:50 +0000 | [diff] [blame] | 589 | self.assertRaises(ValueError, m.read_byte) |
| 590 | # Test read() |
| 591 | m.seek(3) |
Ezio Melotti | b3aedd4 | 2010-11-20 19:04:17 +0000 | [diff] [blame] | 592 | self.assertEqual(m.read(3), b"345") |
| 593 | self.assertEqual(m.tell(), 6) |
Hirokazu Yamamoto | 39c6dea | 2009-02-28 10:56:50 +0000 | [diff] [blame] | 594 | # Test write() |
| 595 | m.seek(3) |
| 596 | m.write(b"bar") |
Ezio Melotti | b3aedd4 | 2010-11-20 19:04:17 +0000 | [diff] [blame] | 597 | self.assertEqual(m.tell(), 6) |
| 598 | self.assertEqual(m[:], b"012bar6789") |
Hirokazu Yamamoto | 39c6dea | 2009-02-28 10:56:50 +0000 | [diff] [blame] | 599 | m.seek(8) |
| 600 | self.assertRaises(ValueError, m.write, b"bar") |
| 601 | |
Hirokazu Yamamoto | 3cdd5cb | 2010-11-04 12:09:08 +0000 | [diff] [blame] | 602 | def test_non_ascii_byte(self): |
| 603 | for b in (129, 200, 255): # > 128 |
| 604 | m = mmap.mmap(-1, 1) |
| 605 | m.write_byte(b) |
Ezio Melotti | b3aedd4 | 2010-11-20 19:04:17 +0000 | [diff] [blame] | 606 | self.assertEqual(m[0], b) |
Hirokazu Yamamoto | 3cdd5cb | 2010-11-04 12:09:08 +0000 | [diff] [blame] | 607 | m.seek(0) |
Ezio Melotti | b3aedd4 | 2010-11-20 19:04:17 +0000 | [diff] [blame] | 608 | self.assertEqual(m.read_byte(), b) |
Hirokazu Yamamoto | 3cdd5cb | 2010-11-04 12:09:08 +0000 | [diff] [blame] | 609 | m.close() |
| 610 | |
Hirokazu Yamamoto | 10c9937 | 2009-02-28 12:21:53 +0000 | [diff] [blame] | 611 | if os.name == 'nt': |
| 612 | def test_tagname(self): |
| 613 | data1 = b"0123456789" |
| 614 | data2 = b"abcdefghij" |
| 615 | assert len(data1) == len(data2) |
Hirokazu Yamamoto | 8e722bc | 2009-03-05 14:33:01 +0000 | [diff] [blame] | 616 | |
Hirokazu Yamamoto | 10c9937 | 2009-02-28 12:21:53 +0000 | [diff] [blame] | 617 | # Test same tag |
| 618 | m1 = mmap.mmap(-1, len(data1), tagname="foo") |
| 619 | m1[:] = data1 |
| 620 | m2 = mmap.mmap(-1, len(data2), tagname="foo") |
| 621 | m2[:] = data2 |
Ezio Melotti | b3aedd4 | 2010-11-20 19:04:17 +0000 | [diff] [blame] | 622 | self.assertEqual(m1[:], data2) |
| 623 | self.assertEqual(m2[:], data2) |
Hirokazu Yamamoto | 8e722bc | 2009-03-05 14:33:01 +0000 | [diff] [blame] | 624 | m2.close() |
| 625 | m1.close() |
| 626 | |
Ezio Melotti | 1392500 | 2011-03-16 11:05:33 +0200 | [diff] [blame] | 627 | # Test different tag |
Hirokazu Yamamoto | 10c9937 | 2009-02-28 12:21:53 +0000 | [diff] [blame] | 628 | m1 = mmap.mmap(-1, len(data1), tagname="foo") |
| 629 | m1[:] = data1 |
| 630 | m2 = mmap.mmap(-1, len(data2), tagname="boo") |
| 631 | m2[:] = data2 |
Ezio Melotti | b3aedd4 | 2010-11-20 19:04:17 +0000 | [diff] [blame] | 632 | self.assertEqual(m1[:], data1) |
| 633 | self.assertEqual(m2[:], data2) |
Hirokazu Yamamoto | 8e722bc | 2009-03-05 14:33:01 +0000 | [diff] [blame] | 634 | m2.close() |
| 635 | m1.close() |
Hirokazu Yamamoto | 10c9937 | 2009-02-28 12:21:53 +0000 | [diff] [blame] | 636 | |
Hirokazu Yamamoto | 8e722bc | 2009-03-05 14:33:01 +0000 | [diff] [blame] | 637 | def test_crasher_on_windows(self): |
Hirokazu Yamamoto | 10c9937 | 2009-02-28 12:21:53 +0000 | [diff] [blame] | 638 | # Should not crash (Issue 1733986) |
| 639 | m = mmap.mmap(-1, 1000, tagname="foo") |
| 640 | try: |
| 641 | mmap.mmap(-1, 5000, tagname="foo")[:] # same tagname, but larger size |
| 642 | except: |
| 643 | pass |
Hirokazu Yamamoto | 8e722bc | 2009-03-05 14:33:01 +0000 | [diff] [blame] | 644 | m.close() |
| 645 | |
| 646 | # Should not crash (Issue 5385) |
Benjamin Peterson | 5dc8fab | 2010-10-31 01:44:49 +0000 | [diff] [blame] | 647 | with open(TESTFN, "wb") as fp: |
| 648 | fp.write(b"x"*10) |
Hirokazu Yamamoto | 9b78925 | 2009-03-05 15:00:28 +0000 | [diff] [blame] | 649 | f = open(TESTFN, "r+b") |
| 650 | m = mmap.mmap(f.fileno(), 0) |
| 651 | f.close() |
Hirokazu Yamamoto | 8e722bc | 2009-03-05 14:33:01 +0000 | [diff] [blame] | 652 | try: |
Hirokazu Yamamoto | 9b78925 | 2009-03-05 15:00:28 +0000 | [diff] [blame] | 653 | m.resize(0) # will raise WindowsError |
Hirokazu Yamamoto | 8e722bc | 2009-03-05 14:33:01 +0000 | [diff] [blame] | 654 | except: |
| 655 | pass |
| 656 | try: |
| 657 | m[:] |
| 658 | except: |
| 659 | pass |
| 660 | m.close() |
| 661 | |
Brian Curtin | ea47eaa | 2010-08-01 15:26:26 +0000 | [diff] [blame] | 662 | def test_invalid_descriptor(self): |
| 663 | # socket file descriptors are valid, but out of range |
| 664 | # for _get_osfhandle, causing a crash when validating the |
| 665 | # parameters to _get_osfhandle. |
| 666 | s = socket.socket() |
| 667 | try: |
| 668 | with self.assertRaises(mmap.error): |
| 669 | m = mmap.mmap(s.fileno(), 10) |
| 670 | finally: |
| 671 | s.close() |
| 672 | |
Georg Brandl | 0bccc18 | 2010-08-01 14:50:00 +0000 | [diff] [blame] | 673 | def test_context_manager(self): |
| 674 | with mmap.mmap(-1, 10) as m: |
| 675 | self.assertFalse(m.closed) |
| 676 | self.assertTrue(m.closed) |
| 677 | |
| 678 | def test_context_manager_exception(self): |
| 679 | # Test that the IOError gets passed through |
| 680 | with self.assertRaises(Exception) as exc: |
| 681 | with mmap.mmap(-1, 10) as m: |
| 682 | raise IOError |
| 683 | self.assertIsInstance(exc.exception, IOError, |
| 684 | "wrong exception raised in context manager") |
| 685 | self.assertTrue(m.closed, "context manager failed") |
| 686 | |
Antoine Pitrou | 97696cb | 2011-02-21 23:46:27 +0000 | [diff] [blame] | 687 | class LargeMmapTests(unittest.TestCase): |
| 688 | |
| 689 | def setUp(self): |
| 690 | unlink(TESTFN) |
| 691 | |
| 692 | def tearDown(self): |
| 693 | unlink(TESTFN) |
| 694 | |
Nadeem Vawda | 909f6d2 | 2011-05-07 14:14:53 +0200 | [diff] [blame] | 695 | def _make_test_file(self, num_zeroes, tail): |
Antoine Pitrou | 97696cb | 2011-02-21 23:46:27 +0000 | [diff] [blame] | 696 | if sys.platform[:3] == 'win' or sys.platform == 'darwin': |
| 697 | requires('largefile', |
| 698 | 'test requires %s bytes and a long time to run' % str(0x180000000)) |
Nadeem Vawda | 909f6d2 | 2011-05-07 14:14:53 +0200 | [diff] [blame] | 699 | f = open(TESTFN, 'w+b') |
| 700 | try: |
| 701 | f.seek(num_zeroes) |
| 702 | f.write(tail) |
| 703 | f.flush() |
| 704 | except (IOError, OverflowError): |
Nadeem Vawda | 7420b70 | 2011-05-07 14:35:05 +0200 | [diff] [blame] | 705 | f.close() |
Nadeem Vawda | 909f6d2 | 2011-05-07 14:14:53 +0200 | [diff] [blame] | 706 | raise unittest.SkipTest("filesystem does not have largefile support") |
| 707 | return f |
Antoine Pitrou | 97696cb | 2011-02-21 23:46:27 +0000 | [diff] [blame] | 708 | |
Nadeem Vawda | ced1056 | 2011-05-07 13:01:50 +0200 | [diff] [blame] | 709 | def test_large_offset(self): |
Nadeem Vawda | 909f6d2 | 2011-05-07 14:14:53 +0200 | [diff] [blame] | 710 | with self._make_test_file(0x14FFFFFFF, b" ") as f: |
Antoine Pitrou | 97696cb | 2011-02-21 23:46:27 +0000 | [diff] [blame] | 711 | with mmap.mmap(f.fileno(), 0, offset=0x140000000, access=mmap.ACCESS_READ) as m: |
| 712 | self.assertEqual(m[0xFFFFFFF], 32) |
| 713 | |
| 714 | def test_large_filesize(self): |
Nadeem Vawda | 909f6d2 | 2011-05-07 14:14:53 +0200 | [diff] [blame] | 715 | with self._make_test_file(0x17FFFFFFF, b" ") as f: |
Antoine Pitrou | 97696cb | 2011-02-21 23:46:27 +0000 | [diff] [blame] | 716 | with mmap.mmap(f.fileno(), 0x10000, access=mmap.ACCESS_READ) as m: |
| 717 | self.assertEqual(m.size(), 0x180000000) |
| 718 | |
Nadeem Vawda | ced1056 | 2011-05-07 13:01:50 +0200 | [diff] [blame] | 719 | # Issue 11277: mmap() with large (~4GB) sparse files crashes on OS X. |
| 720 | |
| 721 | def _test_around_boundary(self, boundary): |
| 722 | tail = b' DEARdear ' |
| 723 | start = boundary - len(tail) // 2 |
| 724 | end = start + len(tail) |
Nadeem Vawda | 909f6d2 | 2011-05-07 14:14:53 +0200 | [diff] [blame] | 725 | with self._make_test_file(start, tail) as f: |
Nadeem Vawda | c2bb073 | 2011-05-07 13:08:54 +0200 | [diff] [blame] | 726 | with mmap.mmap(f.fileno(), 0, access=mmap.ACCESS_READ) as m: |
Nadeem Vawda | ced1056 | 2011-05-07 13:01:50 +0200 | [diff] [blame] | 727 | self.assertEqual(m[start:end], tail) |
Nadeem Vawda | ced1056 | 2011-05-07 13:01:50 +0200 | [diff] [blame] | 728 | |
| 729 | @unittest.skipUnless(sys.maxsize > _4G, "test cannot run on 32-bit systems") |
| 730 | def test_around_2GB(self): |
| 731 | self._test_around_boundary(_2G) |
| 732 | |
| 733 | @unittest.skipUnless(sys.maxsize > _4G, "test cannot run on 32-bit systems") |
| 734 | def test_around_4GB(self): |
| 735 | self._test_around_boundary(_4G) |
| 736 | |
Christian Heimes | 7131fd9 | 2008-02-19 14:21:46 +0000 | [diff] [blame] | 737 | |
Thomas Wouters | 89f507f | 2006-12-13 04:49:30 +0000 | [diff] [blame] | 738 | def test_main(): |
Antoine Pitrou | 97696cb | 2011-02-21 23:46:27 +0000 | [diff] [blame] | 739 | run_unittest(MmapTests, LargeMmapTests) |
Andrew M. Kuchling | e81b9cf | 2000-03-30 21:15:29 +0000 | [diff] [blame] | 740 | |
Thomas Wouters | 89f507f | 2006-12-13 04:49:30 +0000 | [diff] [blame] | 741 | if __name__ == '__main__': |
| 742 | test_main() |