blob: a5c0ae0eaf065a3f835e76d01e0e08f7e66a48ab [file] [log] [blame]
Guido van Rossum09fdf072000-03-31 01:17:07 +00001/*
2 / Author: Sam Rushing <rushing@nightmare.com>
Andrew M. Kuchling10f9c072001-11-05 21:25:42 +00003 / Hacked for Unix by AMK
Guido van Rossum09fdf072000-03-31 01:17:07 +00004 / $Id$
Andrew M. Kuchling1ed7d2d2000-03-30 21:14:30 +00005
Guido van Rossum8ce8a782007-11-01 19:42:39 +00006 / Modified to support mmap with offset - to map a 'window' of a file
7 / Author: Yotam Medini yotamm@mellanox.co.il
8 /
Andrew M. Kuchling1ed7d2d2000-03-30 21:14:30 +00009 / mmapmodule.cpp -- map a view of a file into memory
10 /
11 / todo: need permission flags, perhaps a 'chsize' analog
12 / not all functions check range yet!!!
13 /
14 /
Mark Hammond071864a2000-07-30 02:46:26 +000015 / This version of mmapmodule.c has been changed significantly
16 / from the original mmapfile.c on which it was based.
17 / The original version of mmapfile is maintained by Sam at
Andrew M. Kuchling1ed7d2d2000-03-30 21:14:30 +000018 / ftp://squirl.nightmare.com/pub/python/python-ext.
19*/
20
Martin v. Löwiscfe7e092006-02-17 06:59:14 +000021#define PY_SSIZE_T_CLEAN
Guido van Rossum09fdf072000-03-31 01:17:07 +000022#include <Python.h>
Antoine Pitrouc53204b2013-08-05 23:17:30 +020023#include "structmember.h"
Guido van Rossum09fdf072000-03-31 01:17:07 +000024
Martin v. Löwis6238d2b2002-06-30 15:26:10 +000025#ifndef MS_WINDOWS
Andrew M. Kuchling1ed7d2d2000-03-30 21:14:30 +000026#define UNIX
Benjamin Petersonf6b5cad2015-08-02 12:15:30 -070027# ifdef HAVE_FCNTL_H
Victor Stinnera6cd0cf2011-05-02 01:05:37 +020028# include <fcntl.h>
Benjamin Petersonf6b5cad2015-08-02 12:15:30 -070029# endif /* HAVE_FCNTL_H */
Andrew M. Kuchling1ed7d2d2000-03-30 21:14:30 +000030#endif
31
Martin v. Löwis6238d2b2002-06-30 15:26:10 +000032#ifdef MS_WINDOWS
Andrew M. Kuchling1ed7d2d2000-03-30 21:14:30 +000033#include <windows.h>
Fred Drake145f96e2000-10-01 17:50:46 +000034static int
35my_getpagesize(void)
36{
Antoine Pitrouf95a1b32010-05-09 15:52:27 +000037 SYSTEM_INFO si;
38 GetSystemInfo(&si);
39 return si.dwPageSize;
Fred Drake145f96e2000-10-01 17:50:46 +000040}
Guido van Rossum8ce8a782007-11-01 19:42:39 +000041
42static int
43my_getallocationgranularity (void)
44{
45
Antoine Pitrouf95a1b32010-05-09 15:52:27 +000046 SYSTEM_INFO si;
47 GetSystemInfo(&si);
48 return si.dwAllocationGranularity;
Guido van Rossum8ce8a782007-11-01 19:42:39 +000049}
50
Andrew M. Kuchling1ed7d2d2000-03-30 21:14:30 +000051#endif
52
53#ifdef UNIX
Andrew M. Kuchling1ed7d2d2000-03-30 21:14:30 +000054#include <sys/mman.h>
Andrew M. Kuchling7b9fb922000-06-17 22:41:22 +000055#include <sys/stat.h>
Guido van Rossum4b36e6b2000-09-25 13:16:15 +000056
Fred Drake145f96e2000-10-01 17:50:46 +000057#if defined(HAVE_SYSCONF) && defined(_SC_PAGESIZE)
58static int
59my_getpagesize(void)
60{
Antoine Pitrouf95a1b32010-05-09 15:52:27 +000061 return sysconf(_SC_PAGESIZE);
Fred Drake145f96e2000-10-01 17:50:46 +000062}
Guido van Rossum8ce8a782007-11-01 19:42:39 +000063
64#define my_getallocationgranularity my_getpagesize
Fred Drake145f96e2000-10-01 17:50:46 +000065#else
66#define my_getpagesize getpagesize
67#endif
68
Guido van Rossum4b36e6b2000-09-25 13:16:15 +000069#endif /* UNIX */
70
Andrew M. Kuchling1ed7d2d2000-03-30 21:14:30 +000071#include <string.h>
Thomas Wouters0e3f5912006-08-11 14:57:12 +000072
73#ifdef HAVE_SYS_TYPES_H
Andrew M. Kuchling1ed7d2d2000-03-30 21:14:30 +000074#include <sys/types.h>
Thomas Wouters0e3f5912006-08-11 14:57:12 +000075#endif /* HAVE_SYS_TYPES_H */
Andrew M. Kuchling1ed7d2d2000-03-30 21:14:30 +000076
Neal Norwitz3eaf2b52006-02-16 08:08:54 +000077/* Prefer MAP_ANONYMOUS since MAP_ANON is deprecated according to man page. */
Neal Norwitz0e6bc8c2006-02-05 05:45:43 +000078#if !defined(MAP_ANONYMOUS) && defined(MAP_ANON)
79# define MAP_ANONYMOUS MAP_ANON
80#endif
81
Tim Peters5ebfd362001-11-13 23:11:19 +000082typedef enum
83{
Antoine Pitrouf95a1b32010-05-09 15:52:27 +000084 ACCESS_DEFAULT,
85 ACCESS_READ,
86 ACCESS_WRITE,
87 ACCESS_COPY
Tim Peters5ebfd362001-11-13 23:11:19 +000088} access_mode;
89
Andrew M. Kuchling1ed7d2d2000-03-30 21:14:30 +000090typedef struct {
Antoine Pitrouf95a1b32010-05-09 15:52:27 +000091 PyObject_HEAD
92 char * data;
Benjamin Petersoncd04db02016-10-05 21:45:48 -070093 Py_ssize_t size;
94 Py_ssize_t pos; /* relative to offset */
Antoine Pitrou97696cb2011-02-21 23:46:27 +000095#ifdef MS_WINDOWS
Benjamin Petersonaf580df2016-09-06 10:46:49 -070096 long long offset;
Antoine Pitrou97696cb2011-02-21 23:46:27 +000097#else
98 off_t offset;
99#endif
Hai Shi06cd5b62019-10-21 14:31:46 +0800100 Py_ssize_t exports;
Andrew M. Kuchling1ed7d2d2000-03-30 21:14:30 +0000101
Martin v. Löwis6238d2b2002-06-30 15:26:10 +0000102#ifdef MS_WINDOWS
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000103 HANDLE map_handle;
104 HANDLE file_handle;
105 char * tagname;
Andrew M. Kuchling1ed7d2d2000-03-30 21:14:30 +0000106#endif
107
108#ifdef UNIX
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000109 int fd;
Andrew M. Kuchling1ed7d2d2000-03-30 21:14:30 +0000110#endif
Tim Peters5ebfd362001-11-13 23:11:19 +0000111
Antoine Pitrouc53204b2013-08-05 23:17:30 +0200112 PyObject *weakreflist;
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000113 access_mode access;
Andrew M. Kuchling1ed7d2d2000-03-30 21:14:30 +0000114} mmap_object;
115
Tim Peters5ebfd362001-11-13 23:11:19 +0000116
Andrew M. Kuchling1ed7d2d2000-03-30 21:14:30 +0000117static void
Fredrik Lundh54cf3dc2000-07-08 22:05:01 +0000118mmap_object_dealloc(mmap_object *m_obj)
Andrew M. Kuchling1ed7d2d2000-03-30 21:14:30 +0000119{
Martin v. Löwis6238d2b2002-06-30 15:26:10 +0000120#ifdef MS_WINDOWS
Davide Rizzodc078942019-03-06 18:08:31 +0100121 Py_BEGIN_ALLOW_THREADS
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000122 if (m_obj->data != NULL)
123 UnmapViewOfFile (m_obj->data);
124 if (m_obj->map_handle != NULL)
125 CloseHandle (m_obj->map_handle);
126 if (m_obj->file_handle != INVALID_HANDLE_VALUE)
127 CloseHandle (m_obj->file_handle);
Davide Rizzodc078942019-03-06 18:08:31 +0100128 Py_END_ALLOW_THREADS
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000129 if (m_obj->tagname)
130 PyMem_Free(m_obj->tagname);
Martin v. Löwis6238d2b2002-06-30 15:26:10 +0000131#endif /* MS_WINDOWS */
Andrew M. Kuchling1ed7d2d2000-03-30 21:14:30 +0000132
133#ifdef UNIX
Davide Rizzodc078942019-03-06 18:08:31 +0100134 Py_BEGIN_ALLOW_THREADS
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000135 if (m_obj->fd >= 0)
136 (void) close(m_obj->fd);
137 if (m_obj->data!=NULL) {
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000138 munmap(m_obj->data, m_obj->size);
139 }
Davide Rizzobb9593a2019-03-06 16:52:34 +0100140 Py_END_ALLOW_THREADS
Davide Rizzodc078942019-03-06 18:08:31 +0100141#endif /* UNIX */
Andrew M. Kuchling1ed7d2d2000-03-30 21:14:30 +0000142
Antoine Pitrouc53204b2013-08-05 23:17:30 +0200143 if (m_obj->weakreflist != NULL)
144 PyObject_ClearWeakRefs((PyObject *) m_obj);
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000145 Py_TYPE(m_obj)->tp_free((PyObject*)m_obj);
Andrew M. Kuchling1ed7d2d2000-03-30 21:14:30 +0000146}
147
148static PyObject *
Thomas Wouters4d70c3d2006-06-08 14:42:34 +0000149mmap_close_method(mmap_object *self, PyObject *unused)
Andrew M. Kuchling1ed7d2d2000-03-30 21:14:30 +0000150{
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000151 if (self->exports > 0) {
152 PyErr_SetString(PyExc_BufferError, "cannot close "\
153 "exported pointers exist");
154 return NULL;
155 }
Martin v. Löwis6238d2b2002-06-30 15:26:10 +0000156#ifdef MS_WINDOWS
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000157 /* For each resource we maintain, we need to check
158 the value is valid, and if so, free the resource
159 and set the member value to an invalid value so
160 the dealloc does not attempt to resource clearing
161 again.
162 TODO - should we check for errors in the close operations???
163 */
Davide Rizzobb9593a2019-03-06 16:52:34 +0100164 HANDLE map_handle = self->map_handle;
165 HANDLE file_handle = self->file_handle;
166 char *data = self->data;
167 self->map_handle = NULL;
168 self->file_handle = INVALID_HANDLE_VALUE;
169 self->data = NULL;
170 Py_BEGIN_ALLOW_THREADS
171 if (data != NULL) {
172 UnmapViewOfFile(data);
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000173 }
Davide Rizzobb9593a2019-03-06 16:52:34 +0100174 if (map_handle != NULL) {
175 CloseHandle(map_handle);
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000176 }
Davide Rizzobb9593a2019-03-06 16:52:34 +0100177 if (file_handle != INVALID_HANDLE_VALUE) {
178 CloseHandle(file_handle);
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000179 }
Davide Rizzobb9593a2019-03-06 16:52:34 +0100180 Py_END_ALLOW_THREADS
Martin v. Löwis6238d2b2002-06-30 15:26:10 +0000181#endif /* MS_WINDOWS */
Andrew M. Kuchling1ed7d2d2000-03-30 21:14:30 +0000182
183#ifdef UNIX
Davide Rizzobb9593a2019-03-06 16:52:34 +0100184 int fd = self->fd;
185 char *data = self->data;
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000186 self->fd = -1;
Davide Rizzobb9593a2019-03-06 16:52:34 +0100187 self->data = NULL;
188 Py_BEGIN_ALLOW_THREADS
189 if (0 <= fd)
190 (void) close(fd);
191 if (data != NULL) {
192 munmap(data, self->size);
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000193 }
Davide Rizzobb9593a2019-03-06 16:52:34 +0100194 Py_END_ALLOW_THREADS
Andrew M. Kuchling1ed7d2d2000-03-30 21:14:30 +0000195#endif
196
Serhiy Storchaka228b12e2017-01-23 09:47:21 +0200197 Py_RETURN_NONE;
Andrew M. Kuchling1ed7d2d2000-03-30 21:14:30 +0000198}
199
Martin v. Löwis6238d2b2002-06-30 15:26:10 +0000200#ifdef MS_WINDOWS
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000201#define CHECK_VALID(err) \
202do { \
203 if (self->map_handle == NULL) { \
204 PyErr_SetString(PyExc_ValueError, "mmap closed or invalid"); \
205 return err; \
206 } \
Andrew M. Kuchling1ed7d2d2000-03-30 21:14:30 +0000207} while (0)
Martin v. Löwis6238d2b2002-06-30 15:26:10 +0000208#endif /* MS_WINDOWS */
Andrew M. Kuchling1ed7d2d2000-03-30 21:14:30 +0000209
210#ifdef UNIX
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000211#define CHECK_VALID(err) \
212do { \
213 if (self->data == NULL) { \
214 PyErr_SetString(PyExc_ValueError, "mmap closed or invalid"); \
215 return err; \
216 } \
Andrew M. Kuchling1ed7d2d2000-03-30 21:14:30 +0000217} while (0)
218#endif /* UNIX */
219
220static PyObject *
Fredrik Lundh54cf3dc2000-07-08 22:05:01 +0000221mmap_read_byte_method(mmap_object *self,
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000222 PyObject *unused)
Andrew M. Kuchling1ed7d2d2000-03-30 21:14:30 +0000223{
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000224 CHECK_VALID(NULL);
Benjamin Petersoncd04db02016-10-05 21:45:48 -0700225 if (self->pos >= self->size) {
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000226 PyErr_SetString(PyExc_ValueError, "read byte out of range");
227 return NULL;
228 }
Benjamin Petersoncd04db02016-10-05 21:45:48 -0700229 return PyLong_FromLong((unsigned char)self->data[self->pos++]);
Andrew M. Kuchling1ed7d2d2000-03-30 21:14:30 +0000230}
231
232static PyObject *
Fredrik Lundh54cf3dc2000-07-08 22:05:01 +0000233mmap_read_line_method(mmap_object *self,
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000234 PyObject *unused)
Andrew M. Kuchling1ed7d2d2000-03-30 21:14:30 +0000235{
Benjamin Petersoncd04db02016-10-05 21:45:48 -0700236 Py_ssize_t remaining;
237 char *start, *eol;
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000238 PyObject *result;
Andrew M. Kuchling1ed7d2d2000-03-30 21:14:30 +0000239
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000240 CHECK_VALID(NULL);
Andrew M. Kuchling1ed7d2d2000-03-30 21:14:30 +0000241
Benjamin Petersoncd04db02016-10-05 21:45:48 -0700242 remaining = (self->pos < self->size) ? self->size - self->pos : 0;
243 if (!remaining)
244 return PyBytes_FromString("");
245 start = self->data + self->pos;
246 eol = memchr(start, '\n', remaining);
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000247 if (!eol)
Benjamin Petersoncd04db02016-10-05 21:45:48 -0700248 eol = self->data + self->size;
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000249 else
Benjamin Petersoncd04db02016-10-05 21:45:48 -0700250 ++eol; /* advance past newline */
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000251 result = PyBytes_FromStringAndSize(start, (eol - start));
252 self->pos += (eol - start);
253 return result;
Andrew M. Kuchling1ed7d2d2000-03-30 21:14:30 +0000254}
255
256static PyObject *
Fredrik Lundh54cf3dc2000-07-08 22:05:01 +0000257mmap_read_method(mmap_object *self,
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000258 PyObject *args)
Andrew M. Kuchling1ed7d2d2000-03-30 21:14:30 +0000259{
Benjamin Peterson8f1cdc62016-10-05 23:32:09 -0700260 Py_ssize_t num_bytes = PY_SSIZE_T_MAX, remaining;
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000261 PyObject *result;
Andrew M. Kuchling1ed7d2d2000-03-30 21:14:30 +0000262
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000263 CHECK_VALID(NULL);
Serhiy Storchaka762bf402017-03-30 09:15:31 +0300264 if (!PyArg_ParseTuple(args, "|O&:read", _Py_convert_optional_to_ssize_t, &num_bytes))
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000265 return(NULL);
Andrew M. Kuchling1ed7d2d2000-03-30 21:14:30 +0000266
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000267 /* silently 'adjust' out-of-range requests */
Benjamin Petersoncd04db02016-10-05 21:45:48 -0700268 remaining = (self->pos < self->size) ? self->size - self->pos : 0;
269 if (num_bytes < 0 || num_bytes > remaining)
270 num_bytes = remaining;
271 result = PyBytes_FromStringAndSize(&self->data[self->pos], num_bytes);
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000272 self->pos += num_bytes;
273 return result;
Andrew M. Kuchling1ed7d2d2000-03-30 21:14:30 +0000274}
275
276static PyObject *
Georg Brandlfceab5a2008-01-19 20:08:23 +0000277mmap_gfind(mmap_object *self,
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000278 PyObject *args,
279 int reverse)
Andrew M. Kuchling1ed7d2d2000-03-30 21:14:30 +0000280{
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000281 Py_ssize_t start = self->pos;
282 Py_ssize_t end = self->size;
Serhiy Storchaka8490f5a2015-03-20 09:00:36 +0200283 Py_buffer view;
Andrew M. Kuchling1ed7d2d2000-03-30 21:14:30 +0000284
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000285 CHECK_VALID(NULL);
Serhiy Storchaka8490f5a2015-03-20 09:00:36 +0200286 if (!PyArg_ParseTuple(args, reverse ? "y*|nn:rfind" : "y*|nn:find",
287 &view, &start, &end)) {
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000288 return NULL;
289 } else {
290 const char *p, *start_p, *end_p;
291 int sign = reverse ? -1 : 1;
Serhiy Storchaka8490f5a2015-03-20 09:00:36 +0200292 const char *needle = view.buf;
293 Py_ssize_t len = view.len;
Greg Stein834f4dd2001-05-14 09:32:26 +0000294
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000295 if (start < 0)
296 start += self->size;
297 if (start < 0)
298 start = 0;
Benjamin Petersoncd04db02016-10-05 21:45:48 -0700299 else if (start > self->size)
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000300 start = self->size;
Greg Stein834f4dd2001-05-14 09:32:26 +0000301
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000302 if (end < 0)
303 end += self->size;
304 if (end < 0)
305 end = 0;
Benjamin Petersoncd04db02016-10-05 21:45:48 -0700306 else if (end > self->size)
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000307 end = self->size;
Georg Brandlfceab5a2008-01-19 20:08:23 +0000308
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000309 start_p = self->data + start;
310 end_p = self->data + end;
Georg Brandlfceab5a2008-01-19 20:08:23 +0000311
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000312 for (p = (reverse ? end_p - len : start_p);
313 (p >= start_p) && (p + len <= end_p); p += sign) {
314 Py_ssize_t i;
315 for (i = 0; i < len && needle[i] == p[i]; ++i)
316 /* nothing */;
317 if (i == len) {
Serhiy Storchaka8490f5a2015-03-20 09:00:36 +0200318 PyBuffer_Release(&view);
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000319 return PyLong_FromSsize_t(p - self->data);
320 }
321 }
Serhiy Storchaka8490f5a2015-03-20 09:00:36 +0200322 PyBuffer_Release(&view);
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000323 return PyLong_FromLong(-1);
324 }
Andrew M. Kuchling1ed7d2d2000-03-30 21:14:30 +0000325}
326
Georg Brandlfceab5a2008-01-19 20:08:23 +0000327static PyObject *
328mmap_find_method(mmap_object *self,
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000329 PyObject *args)
Georg Brandlfceab5a2008-01-19 20:08:23 +0000330{
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000331 return mmap_gfind(self, args, 0);
Georg Brandlfceab5a2008-01-19 20:08:23 +0000332}
333
334static PyObject *
335mmap_rfind_method(mmap_object *self,
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000336 PyObject *args)
Georg Brandlfceab5a2008-01-19 20:08:23 +0000337{
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000338 return mmap_gfind(self, args, 1);
Georg Brandlfceab5a2008-01-19 20:08:23 +0000339}
340
Tim Petersec0a5f02006-02-16 23:47:20 +0000341static int
Sean Reifscheider54cf12b2007-09-17 17:55:36 +0000342is_writable(mmap_object *self)
Tim Peters5ebfd362001-11-13 23:11:19 +0000343{
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000344 if (self->access != ACCESS_READ)
345 return 1;
346 PyErr_Format(PyExc_TypeError, "mmap can't modify a readonly memory map.");
347 return 0;
Tim Peters5ebfd362001-11-13 23:11:19 +0000348}
349
Tim Petersec0a5f02006-02-16 23:47:20 +0000350static int
Tim Peters5ebfd362001-11-13 23:11:19 +0000351is_resizeable(mmap_object *self)
352{
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000353 if (self->exports > 0) {
354 PyErr_SetString(PyExc_BufferError,
355 "mmap can't resize with extant buffers exported.");
356 return 0;
357 }
358 if ((self->access == ACCESS_WRITE) || (self->access == ACCESS_DEFAULT))
359 return 1;
360 PyErr_Format(PyExc_TypeError,
361 "mmap can't resize a readonly or copy-on-write memory map.");
362 return 0;
Tim Peters5ebfd362001-11-13 23:11:19 +0000363}
364
365
Andrew M. Kuchling1ed7d2d2000-03-30 21:14:30 +0000366static PyObject *
Fredrik Lundh54cf3dc2000-07-08 22:05:01 +0000367mmap_write_method(mmap_object *self,
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000368 PyObject *args)
Andrew M. Kuchling1ed7d2d2000-03-30 21:14:30 +0000369{
Serhiy Storchaka8490f5a2015-03-20 09:00:36 +0200370 Py_buffer data;
Andrew M. Kuchling1ed7d2d2000-03-30 21:14:30 +0000371
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000372 CHECK_VALID(NULL);
Serhiy Storchaka8490f5a2015-03-20 09:00:36 +0200373 if (!PyArg_ParseTuple(args, "y*:write", &data))
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000374 return(NULL);
Andrew M. Kuchling1ed7d2d2000-03-30 21:14:30 +0000375
Benjamin Peterson37768362016-10-05 23:29:07 -0700376 if (!is_writable(self)) {
377 PyBuffer_Release(&data);
Benjamin Petersoncd04db02016-10-05 21:45:48 -0700378 return NULL;
Benjamin Peterson37768362016-10-05 23:29:07 -0700379 }
Benjamin Petersoncd04db02016-10-05 21:45:48 -0700380
381 if (self->pos > self->size || self->size - self->pos < data.len) {
Serhiy Storchaka8490f5a2015-03-20 09:00:36 +0200382 PyBuffer_Release(&data);
Benjamin Petersoncd04db02016-10-05 21:45:48 -0700383 PyErr_SetString(PyExc_ValueError, "data out of range");
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000384 return NULL;
385 }
Serhiy Storchaka8490f5a2015-03-20 09:00:36 +0200386
Benjamin Petersoncd04db02016-10-05 21:45:48 -0700387 memcpy(&self->data[self->pos], data.buf, data.len);
388 self->pos += data.len;
Serhiy Storchaka8490f5a2015-03-20 09:00:36 +0200389 PyBuffer_Release(&data);
Benjamin Peterson87845bc2016-10-05 22:54:19 -0700390 return PyLong_FromSsize_t(data.len);
Andrew M. Kuchling1ed7d2d2000-03-30 21:14:30 +0000391}
392
393static PyObject *
Fredrik Lundh54cf3dc2000-07-08 22:05:01 +0000394mmap_write_byte_method(mmap_object *self,
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000395 PyObject *args)
Andrew M. Kuchling1ed7d2d2000-03-30 21:14:30 +0000396{
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000397 char value;
Andrew M. Kuchling1ed7d2d2000-03-30 21:14:30 +0000398
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000399 CHECK_VALID(NULL);
400 if (!PyArg_ParseTuple(args, "b:write_byte", &value))
401 return(NULL);
Andrew M. Kuchling1ed7d2d2000-03-30 21:14:30 +0000402
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000403 if (!is_writable(self))
404 return NULL;
Hirokazu Yamamoto39c6dea2009-02-28 10:56:50 +0000405
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000406 if (self->pos < self->size) {
Benjamin Petersoncd04db02016-10-05 21:45:48 -0700407 self->data[self->pos++] = value;
Serhiy Storchaka228b12e2017-01-23 09:47:21 +0200408 Py_RETURN_NONE;
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000409 }
410 else {
411 PyErr_SetString(PyExc_ValueError, "write byte out of range");
412 return NULL;
413 }
Andrew M. Kuchling1ed7d2d2000-03-30 21:14:30 +0000414}
Tim Petersec0a5f02006-02-16 23:47:20 +0000415
Andrew M. Kuchling1ed7d2d2000-03-30 21:14:30 +0000416static PyObject *
Fredrik Lundh54cf3dc2000-07-08 22:05:01 +0000417mmap_size_method(mmap_object *self,
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000418 PyObject *unused)
Andrew M. Kuchling1ed7d2d2000-03-30 21:14:30 +0000419{
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000420 CHECK_VALID(NULL);
Andrew M. Kuchling1ed7d2d2000-03-30 21:14:30 +0000421
Martin v. Löwis6238d2b2002-06-30 15:26:10 +0000422#ifdef MS_WINDOWS
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000423 if (self->file_handle != INVALID_HANDLE_VALUE) {
424 DWORD low,high;
Benjamin Petersonaf580df2016-09-06 10:46:49 -0700425 long long size;
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000426 low = GetFileSize(self->file_handle, &high);
427 if (low == INVALID_FILE_SIZE) {
428 /* It might be that the function appears to have failed,
429 when indeed its size equals INVALID_FILE_SIZE */
430 DWORD error = GetLastError();
431 if (error != NO_ERROR)
432 return PyErr_SetFromWindowsErr(error);
433 }
434 if (!high && low < LONG_MAX)
435 return PyLong_FromLong((long)low);
Benjamin Petersonaf580df2016-09-06 10:46:49 -0700436 size = (((long long)high)<<32) + low;
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000437 return PyLong_FromLongLong(size);
438 } else {
439 return PyLong_FromSsize_t(self->size);
440 }
Martin v. Löwis6238d2b2002-06-30 15:26:10 +0000441#endif /* MS_WINDOWS */
Andrew M. Kuchling1ed7d2d2000-03-30 21:14:30 +0000442
443#ifdef UNIX
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000444 {
Victor Stinnere134a7f2015-03-30 10:09:31 +0200445 struct _Py_stat_struct status;
446 if (_Py_fstat(self->fd, &status) == -1)
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000447 return NULL;
Antoine Pitrou97696cb2011-02-21 23:46:27 +0000448#ifdef HAVE_LARGEFILE_SUPPORT
Victor Stinnere134a7f2015-03-30 10:09:31 +0200449 return PyLong_FromLongLong(status.st_size);
Antoine Pitrou97696cb2011-02-21 23:46:27 +0000450#else
Victor Stinnere134a7f2015-03-30 10:09:31 +0200451 return PyLong_FromLong(status.st_size);
Antoine Pitrou97696cb2011-02-21 23:46:27 +0000452#endif
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000453 }
Andrew M. Kuchling1ed7d2d2000-03-30 21:14:30 +0000454#endif /* UNIX */
455}
456
457/* This assumes that you want the entire file mapped,
458 / and when recreating the map will make the new file
459 / have the new size
460 /
461 / Is this really necessary? This could easily be done
462 / from python by just closing and re-opening with the
463 / new size?
464 */
465
466static PyObject *
Fredrik Lundh54cf3dc2000-07-08 22:05:01 +0000467mmap_resize_method(mmap_object *self,
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000468 PyObject *args)
Andrew M. Kuchling1ed7d2d2000-03-30 21:14:30 +0000469{
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000470 Py_ssize_t new_size;
471 CHECK_VALID(NULL);
472 if (!PyArg_ParseTuple(args, "n:resize", &new_size) ||
473 !is_resizeable(self)) {
474 return NULL;
Benjamin Petersoncd04db02016-10-05 21:45:48 -0700475 }
476 if (new_size < 0 || PY_SSIZE_T_MAX - new_size < self->offset) {
477 PyErr_SetString(PyExc_ValueError, "new size out of range");
478 return NULL;
479 }
480
481 {
Martin v. Löwis6238d2b2002-06-30 15:26:10 +0000482#ifdef MS_WINDOWS
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000483 DWORD dwErrCode = 0;
484 DWORD off_hi, off_lo, newSizeLow, newSizeHigh;
485 /* First, unmap the file view */
486 UnmapViewOfFile(self->data);
487 self->data = NULL;
488 /* Close the mapping object */
489 CloseHandle(self->map_handle);
490 self->map_handle = NULL;
491 /* Move to the desired EOF position */
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000492 newSizeHigh = (DWORD)((self->offset + new_size) >> 32);
493 newSizeLow = (DWORD)((self->offset + new_size) & 0xFFFFFFFF);
494 off_hi = (DWORD)(self->offset >> 32);
495 off_lo = (DWORD)(self->offset & 0xFFFFFFFF);
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000496 SetFilePointer(self->file_handle,
497 newSizeLow, &newSizeHigh, FILE_BEGIN);
498 /* Change the size of the file */
499 SetEndOfFile(self->file_handle);
500 /* Create another mapping object and remap the file view */
501 self->map_handle = CreateFileMapping(
502 self->file_handle,
503 NULL,
504 PAGE_READWRITE,
505 0,
506 0,
507 self->tagname);
508 if (self->map_handle != NULL) {
509 self->data = (char *) MapViewOfFile(self->map_handle,
510 FILE_MAP_WRITE,
511 off_hi,
512 off_lo,
513 new_size);
514 if (self->data != NULL) {
515 self->size = new_size;
Serhiy Storchaka228b12e2017-01-23 09:47:21 +0200516 Py_RETURN_NONE;
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000517 } else {
518 dwErrCode = GetLastError();
519 CloseHandle(self->map_handle);
520 self->map_handle = NULL;
521 }
522 } else {
523 dwErrCode = GetLastError();
524 }
525 PyErr_SetFromWindowsErr(dwErrCode);
526 return NULL;
Martin v. Löwis6238d2b2002-06-30 15:26:10 +0000527#endif /* MS_WINDOWS */
Andrew M. Kuchling1ed7d2d2000-03-30 21:14:30 +0000528
529#ifdef UNIX
Tim Petersec0a5f02006-02-16 23:47:20 +0000530#ifndef HAVE_MREMAP
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000531 PyErr_SetString(PyExc_SystemError,
532 "mmap: resizing not available--no mremap()");
533 return NULL;
Andrew M. Kuchling1ed7d2d2000-03-30 21:14:30 +0000534#else
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000535 void *newmap;
Armin Rigo335ffe82005-09-20 19:04:02 +0000536
Benjamin Petersoncd04db02016-10-05 21:45:48 -0700537 if (self->fd != -1 && ftruncate(self->fd, self->offset + new_size) == -1) {
Antoine Pitrou6b4883d2011-10-12 02:54:14 +0200538 PyErr_SetFromErrno(PyExc_OSError);
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000539 return NULL;
540 }
Andrew M. Kuchling1ed7d2d2000-03-30 21:14:30 +0000541
Andrew M. Kuchling6fef30e2000-06-18 14:51:21 +0000542#ifdef MREMAP_MAYMOVE
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000543 newmap = mremap(self->data, self->size, new_size, MREMAP_MAYMOVE);
Andrew M. Kuchling6fef30e2000-06-18 14:51:21 +0000544#else
Benjamin Petersoncd04db02016-10-05 21:45:48 -0700545#if defined(__NetBSD__)
546 newmap = mremap(self->data, self->size, self->data, new_size, 0);
547#else
548 newmap = mremap(self->data, self->size, new_size, 0);
549#endif /* __NetBSD__ */
Andrew M. Kuchling6fef30e2000-06-18 14:51:21 +0000550#endif
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000551 if (newmap == (void *)-1)
552 {
Antoine Pitrou6b4883d2011-10-12 02:54:14 +0200553 PyErr_SetFromErrno(PyExc_OSError);
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000554 return NULL;
555 }
556 self->data = newmap;
557 self->size = new_size;
Serhiy Storchaka228b12e2017-01-23 09:47:21 +0200558 Py_RETURN_NONE;
Andrew M. Kuchling6fef30e2000-06-18 14:51:21 +0000559#endif /* HAVE_MREMAP */
Andrew M. Kuchling1ed7d2d2000-03-30 21:14:30 +0000560#endif /* UNIX */
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000561 }
Andrew M. Kuchling1ed7d2d2000-03-30 21:14:30 +0000562}
563
564static PyObject *
Thomas Wouters4d70c3d2006-06-08 14:42:34 +0000565mmap_tell_method(mmap_object *self, PyObject *unused)
Andrew M. Kuchling1ed7d2d2000-03-30 21:14:30 +0000566{
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000567 CHECK_VALID(NULL);
568 return PyLong_FromSize_t(self->pos);
Andrew M. Kuchling1ed7d2d2000-03-30 21:14:30 +0000569}
570
571static PyObject *
Fredrik Lundh54cf3dc2000-07-08 22:05:01 +0000572mmap_flush_method(mmap_object *self, PyObject *args)
Andrew M. Kuchling1ed7d2d2000-03-30 21:14:30 +0000573{
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000574 Py_ssize_t offset = 0;
575 Py_ssize_t size = self->size;
576 CHECK_VALID(NULL);
577 if (!PyArg_ParseTuple(args, "|nn:flush", &offset, &size))
578 return NULL;
Benjamin Petersoncd04db02016-10-05 21:45:48 -0700579 if (size < 0 || offset < 0 || self->size - offset < size) {
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000580 PyErr_SetString(PyExc_ValueError, "flush values out of range");
581 return NULL;
582 }
R. David Murraye194dd62010-10-18 01:14:06 +0000583
584 if (self->access == ACCESS_READ || self->access == ACCESS_COPY)
Berker Peksage7d4b2f2018-08-22 21:21:05 +0300585 Py_RETURN_NONE;
R. David Murraye194dd62010-10-18 01:14:06 +0000586
Christian Heimesaf98da12008-01-27 15:18:18 +0000587#ifdef MS_WINDOWS
Berker Peksage7d4b2f2018-08-22 21:21:05 +0300588 if (!FlushViewOfFile(self->data+offset, size)) {
589 PyErr_SetFromWindowsErr(GetLastError());
590 return NULL;
591 }
592 Py_RETURN_NONE;
Christian Heimesaf98da12008-01-27 15:18:18 +0000593#elif defined(UNIX)
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000594 /* XXX flags for msync? */
595 if (-1 == msync(self->data + offset, size, MS_SYNC)) {
Antoine Pitrou6b4883d2011-10-12 02:54:14 +0200596 PyErr_SetFromErrno(PyExc_OSError);
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000597 return NULL;
598 }
Berker Peksage7d4b2f2018-08-22 21:21:05 +0300599 Py_RETURN_NONE;
Christian Heimesaf98da12008-01-27 15:18:18 +0000600#else
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000601 PyErr_SetString(PyExc_ValueError, "flush not supported on this system");
602 return NULL;
Christian Heimesaf98da12008-01-27 15:18:18 +0000603#endif
Andrew M. Kuchling1ed7d2d2000-03-30 21:14:30 +0000604}
605
606static PyObject *
Fredrik Lundh54cf3dc2000-07-08 22:05:01 +0000607mmap_seek_method(mmap_object *self, PyObject *args)
Andrew M. Kuchling1ed7d2d2000-03-30 21:14:30 +0000608{
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000609 Py_ssize_t dist;
610 int how=0;
611 CHECK_VALID(NULL);
612 if (!PyArg_ParseTuple(args, "n|i:seek", &dist, &how))
613 return NULL;
614 else {
Benjamin Petersoncd04db02016-10-05 21:45:48 -0700615 Py_ssize_t where;
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000616 switch (how) {
617 case 0: /* relative to start */
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000618 where = dist;
619 break;
620 case 1: /* relative to current position */
Benjamin Petersoncd04db02016-10-05 21:45:48 -0700621 if (PY_SSIZE_T_MAX - self->pos < dist)
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000622 goto onoutofrange;
623 where = self->pos + dist;
624 break;
625 case 2: /* relative to end */
Benjamin Petersoncd04db02016-10-05 21:45:48 -0700626 if (PY_SSIZE_T_MAX - self->size < dist)
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000627 goto onoutofrange;
628 where = self->size + dist;
629 break;
630 default:
631 PyErr_SetString(PyExc_ValueError, "unknown seek type");
632 return NULL;
633 }
Benjamin Petersoncd04db02016-10-05 21:45:48 -0700634 if (where > self->size || where < 0)
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000635 goto onoutofrange;
636 self->pos = where;
Serhiy Storchaka228b12e2017-01-23 09:47:21 +0200637 Py_RETURN_NONE;
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000638 }
Andrew M. Kuchling70d27422000-06-18 04:45:14 +0000639
Tim Peters5ebfd362001-11-13 23:11:19 +0000640 onoutofrange:
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000641 PyErr_SetString(PyExc_ValueError, "seek out of range");
642 return NULL;
Andrew M. Kuchling1ed7d2d2000-03-30 21:14:30 +0000643}
644
645static PyObject *
Fredrik Lundh54cf3dc2000-07-08 22:05:01 +0000646mmap_move_method(mmap_object *self, PyObject *args)
Andrew M. Kuchling1ed7d2d2000-03-30 21:14:30 +0000647{
Benjamin Petersoncd04db02016-10-05 21:45:48 -0700648 Py_ssize_t dest, src, cnt;
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000649 CHECK_VALID(NULL);
Benjamin Petersoncd04db02016-10-05 21:45:48 -0700650 if (!PyArg_ParseTuple(args, "nnn:move", &dest, &src, &cnt) ||
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000651 !is_writable(self)) {
652 return NULL;
653 } else {
654 /* bounds check the values */
Benjamin Petersoncd04db02016-10-05 21:45:48 -0700655 if (dest < 0 || src < 0 || cnt < 0)
656 goto bounds;
657 if (self->size - dest < cnt || self->size - src < cnt)
658 goto bounds;
659
660 memmove(&self->data[dest], &self->data[src], cnt);
661
Serhiy Storchaka228b12e2017-01-23 09:47:21 +0200662 Py_RETURN_NONE;
Benjamin Petersoncd04db02016-10-05 21:45:48 -0700663
664 bounds:
665 PyErr_SetString(PyExc_ValueError,
666 "source, destination, or count out of range");
667 return NULL;
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000668 }
Andrew M. Kuchling1ed7d2d2000-03-30 21:14:30 +0000669}
670
Georg Brandl0bccc182010-08-01 14:50:00 +0000671static PyObject *
Serhiy Storchakad4f9cf52018-11-27 19:34:35 +0200672mmap_closed_get(mmap_object *self, void *Py_UNUSED(ignored))
Georg Brandl0bccc182010-08-01 14:50:00 +0000673{
674#ifdef MS_WINDOWS
675 return PyBool_FromLong(self->map_handle == NULL ? 1 : 0);
676#elif defined(UNIX)
677 return PyBool_FromLong(self->data == NULL ? 1 : 0);
678#endif
679}
680
681static PyObject *
682mmap__enter__method(mmap_object *self, PyObject *args)
683{
684 CHECK_VALID(NULL);
685
686 Py_INCREF(self);
687 return (PyObject *)self;
688}
689
690static PyObject *
691mmap__exit__method(PyObject *self, PyObject *args)
692{
Martin v. Löwisbd928fe2011-10-14 10:20:37 +0200693 _Py_IDENTIFIER(close);
Martin v. Löwisafe55bb2011-10-09 10:38:36 +0200694
Jeroen Demeyer762f93f2019-07-08 10:19:25 +0200695 return _PyObject_CallMethodIdNoArgs(self, &PyId_close);
Georg Brandl0bccc182010-08-01 14:50:00 +0000696}
697
Taine Zhaod8ca2352019-10-17 18:41:35 +0800698static PyObject *
699mmap__repr__method(PyObject *self)
700{
701 mmap_object *mobj = (mmap_object *)self;
702
703#ifdef MS_WINDOWS
704#define _Py_FORMAT_OFFSET "lld"
705 if (mobj->map_handle == NULL)
706#elif defined(UNIX)
707# ifdef HAVE_LARGEFILE_SUPPORT
708# define _Py_FORMAT_OFFSET "lld"
709# else
710# define _Py_FORMAT_OFFSET "ld"
711# endif
712 if (mobj->data == NULL)
713#endif
714 {
715 return PyUnicode_FromFormat("<%s closed=True>", Py_TYPE(self)->tp_name);
716 } else {
717 const char *access_str;
718
719 switch (mobj->access) {
720 case ACCESS_DEFAULT:
721 access_str = "ACCESS_DEFAULT";
722 break;
723 case ACCESS_READ:
724 access_str = "ACCESS_READ";
725 break;
726 case ACCESS_WRITE:
727 access_str = "ACCESS_WRITE";
728 break;
729 case ACCESS_COPY:
730 access_str = "ACCESS_COPY";
731 break;
732 default:
733 Py_UNREACHABLE();
734 }
735
736 return PyUnicode_FromFormat("<%s closed=False, access=%s, length=%zd, "
737 "pos=%zd, offset=%" _Py_FORMAT_OFFSET ">",
738 Py_TYPE(self)->tp_name, access_str,
739 mobj->size, mobj->pos, mobj->offset);
740 }
741}
742
Serhiy Storchaka76b47652014-08-19 17:11:20 +0300743#ifdef MS_WINDOWS
744static PyObject *
745mmap__sizeof__method(mmap_object *self, void *unused)
746{
747 Py_ssize_t res;
748
Serhiy Storchaka5c4064e2015-12-19 20:05:25 +0200749 res = _PyObject_SIZE(Py_TYPE(self));
Serhiy Storchaka76b47652014-08-19 17:11:20 +0300750 if (self->tagname)
751 res += strlen(self->tagname) + 1;
752 return PyLong_FromSsize_t(res);
753}
754#endif
755
Zackery Spytz02db6962019-05-27 10:48:17 -0600756#ifdef HAVE_MADVISE
757static PyObject *
758mmap_madvise_method(mmap_object *self, PyObject *args)
759{
760 int option;
761 Py_ssize_t start = 0, length;
762
763 CHECK_VALID(NULL);
764 length = self->size;
765
766 if (!PyArg_ParseTuple(args, "i|nn:madvise", &option, &start, &length)) {
767 return NULL;
768 }
769
770 if (start < 0 || start >= self->size) {
771 PyErr_SetString(PyExc_ValueError, "madvise start out of bounds");
772 return NULL;
773 }
774 if (length < 0) {
775 PyErr_SetString(PyExc_ValueError, "madvise length invalid");
776 return NULL;
777 }
778 if (PY_SSIZE_T_MAX - start < length) {
779 PyErr_SetString(PyExc_OverflowError, "madvise length too large");
780 return NULL;
781 }
782
783 if (start + length > self->size) {
784 length = self->size - start;
785 }
786
787 if (madvise(self->data + start, length, option) != 0) {
788 PyErr_SetFromErrno(PyExc_OSError);
789 return NULL;
790 }
791
792 Py_RETURN_NONE;
793}
794#endif // HAVE_MADVISE
795
Andrew M. Kuchling1ed7d2d2000-03-30 21:14:30 +0000796static struct PyMethodDef mmap_object_methods[] = {
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000797 {"close", (PyCFunction) mmap_close_method, METH_NOARGS},
798 {"find", (PyCFunction) mmap_find_method, METH_VARARGS},
799 {"rfind", (PyCFunction) mmap_rfind_method, METH_VARARGS},
800 {"flush", (PyCFunction) mmap_flush_method, METH_VARARGS},
Zackery Spytz02db6962019-05-27 10:48:17 -0600801#ifdef HAVE_MADVISE
802 {"madvise", (PyCFunction) mmap_madvise_method, METH_VARARGS},
803#endif
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000804 {"move", (PyCFunction) mmap_move_method, METH_VARARGS},
805 {"read", (PyCFunction) mmap_read_method, METH_VARARGS},
806 {"read_byte", (PyCFunction) mmap_read_byte_method, METH_NOARGS},
807 {"readline", (PyCFunction) mmap_read_line_method, METH_NOARGS},
808 {"resize", (PyCFunction) mmap_resize_method, METH_VARARGS},
809 {"seek", (PyCFunction) mmap_seek_method, METH_VARARGS},
810 {"size", (PyCFunction) mmap_size_method, METH_NOARGS},
811 {"tell", (PyCFunction) mmap_tell_method, METH_NOARGS},
812 {"write", (PyCFunction) mmap_write_method, METH_VARARGS},
813 {"write_byte", (PyCFunction) mmap_write_byte_method, METH_VARARGS},
Georg Brandl0bccc182010-08-01 14:50:00 +0000814 {"__enter__", (PyCFunction) mmap__enter__method, METH_NOARGS},
815 {"__exit__", (PyCFunction) mmap__exit__method, METH_VARARGS},
Serhiy Storchaka76b47652014-08-19 17:11:20 +0300816#ifdef MS_WINDOWS
817 {"__sizeof__", (PyCFunction) mmap__sizeof__method, METH_NOARGS},
818#endif
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000819 {NULL, NULL} /* sentinel */
Andrew M. Kuchling1ed7d2d2000-03-30 21:14:30 +0000820};
821
Georg Brandl0bccc182010-08-01 14:50:00 +0000822static PyGetSetDef mmap_object_getset[] = {
823 {"closed", (getter) mmap_closed_get, NULL, NULL},
824 {NULL}
825};
826
827
Andrew M. Kuchling1ed7d2d2000-03-30 21:14:30 +0000828/* Functions for treating an mmap'ed file as a buffer */
829
Travis E. Oliphantb99f7622007-08-18 11:21:56 +0000830static int
Guido van Rossum98297ee2007-11-06 21:34:58 +0000831mmap_buffer_getbuf(mmap_object *self, Py_buffer *view, int flags)
Andrew M. Kuchling1ed7d2d2000-03-30 21:14:30 +0000832{
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000833 CHECK_VALID(-1);
834 if (PyBuffer_FillInfo(view, (PyObject*)self, self->data, self->size,
835 (self->access == ACCESS_READ), flags) < 0)
836 return -1;
837 self->exports++;
838 return 0;
Andrew M. Kuchling1ed7d2d2000-03-30 21:14:30 +0000839}
840
Travis E. Oliphantb99f7622007-08-18 11:21:56 +0000841static void
Travis E. Oliphant8ae62b62007-09-23 02:00:13 +0000842mmap_buffer_releasebuf(mmap_object *self, Py_buffer *view)
Tim Petersec0a5f02006-02-16 23:47:20 +0000843{
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000844 self->exports--;
Andrew M. Kuchling1ed7d2d2000-03-30 21:14:30 +0000845}
846
Martin v. Löwis18e16552006-02-15 17:27:45 +0000847static Py_ssize_t
Fredrik Lundh54cf3dc2000-07-08 22:05:01 +0000848mmap_length(mmap_object *self)
Andrew M. Kuchling1ed7d2d2000-03-30 21:14:30 +0000849{
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000850 CHECK_VALID(-1);
851 return self->size;
Andrew M. Kuchling1ed7d2d2000-03-30 21:14:30 +0000852}
853
854static PyObject *
Martin v. Löwis18e16552006-02-15 17:27:45 +0000855mmap_item(mmap_object *self, Py_ssize_t i)
Andrew M. Kuchling1ed7d2d2000-03-30 21:14:30 +0000856{
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000857 CHECK_VALID(NULL);
Benjamin Petersoncd04db02016-10-05 21:45:48 -0700858 if (i < 0 || i >= self->size) {
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000859 PyErr_SetString(PyExc_IndexError, "mmap index out of range");
860 return NULL;
861 }
862 return PyBytes_FromStringAndSize(self->data + i, 1);
Andrew M. Kuchling1ed7d2d2000-03-30 21:14:30 +0000863}
864
865static PyObject *
Thomas Woutersed03b412007-08-28 21:37:11 +0000866mmap_subscript(mmap_object *self, PyObject *item)
867{
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000868 CHECK_VALID(NULL);
869 if (PyIndex_Check(item)) {
870 Py_ssize_t i = PyNumber_AsSsize_t(item, PyExc_IndexError);
871 if (i == -1 && PyErr_Occurred())
872 return NULL;
873 if (i < 0)
874 i += self->size;
Benjamin Petersoncd04db02016-10-05 21:45:48 -0700875 if (i < 0 || i >= self->size) {
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000876 PyErr_SetString(PyExc_IndexError,
877 "mmap index out of range");
878 return NULL;
879 }
880 return PyLong_FromLong(Py_CHARMASK(self->data[i]));
881 }
882 else if (PySlice_Check(item)) {
883 Py_ssize_t start, stop, step, slicelen;
Thomas Woutersed03b412007-08-28 21:37:11 +0000884
Serhiy Storchakab879fe82017-04-08 09:53:51 +0300885 if (PySlice_Unpack(item, &start, &stop, &step) < 0) {
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000886 return NULL;
887 }
Serhiy Storchakab879fe82017-04-08 09:53:51 +0300888 slicelen = PySlice_AdjustIndices(self->size, &start, &stop, step);
Guido van Rossum98297ee2007-11-06 21:34:58 +0000889
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000890 if (slicelen <= 0)
891 return PyBytes_FromStringAndSize("", 0);
892 else if (step == 1)
893 return PyBytes_FromStringAndSize(self->data + start,
894 slicelen);
895 else {
896 char *result_buf = (char *)PyMem_Malloc(slicelen);
Zackery Spytz14514d92019-05-17 01:13:03 -0600897 size_t cur;
898 Py_ssize_t i;
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000899 PyObject *result;
Thomas Woutersed03b412007-08-28 21:37:11 +0000900
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000901 if (result_buf == NULL)
902 return PyErr_NoMemory();
903 for (cur = start, i = 0; i < slicelen;
904 cur += step, i++) {
905 result_buf[i] = self->data[cur];
906 }
907 result = PyBytes_FromStringAndSize(result_buf,
908 slicelen);
909 PyMem_Free(result_buf);
910 return result;
911 }
912 }
913 else {
914 PyErr_SetString(PyExc_TypeError,
915 "mmap indices must be integers");
916 return NULL;
917 }
Thomas Woutersed03b412007-08-28 21:37:11 +0000918}
919
Andrew M. Kuchling1ed7d2d2000-03-30 21:14:30 +0000920static int
Martin v. Löwis18e16552006-02-15 17:27:45 +0000921mmap_ass_item(mmap_object *self, Py_ssize_t i, PyObject *v)
Andrew M. Kuchling1ed7d2d2000-03-30 21:14:30 +0000922{
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000923 const char *buf;
Tim Petersec0a5f02006-02-16 23:47:20 +0000924
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000925 CHECK_VALID(-1);
Benjamin Petersoncd04db02016-10-05 21:45:48 -0700926 if (i < 0 || i >= self->size) {
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000927 PyErr_SetString(PyExc_IndexError, "mmap index out of range");
928 return -1;
929 }
930 if (v == NULL) {
931 PyErr_SetString(PyExc_TypeError,
932 "mmap object doesn't support item deletion");
933 return -1;
934 }
935 if (! (PyBytes_Check(v) && PyBytes_Size(v)==1) ) {
936 PyErr_SetString(PyExc_IndexError,
937 "mmap assignment must be length-1 bytes()");
938 return -1;
939 }
940 if (!is_writable(self))
941 return -1;
942 buf = PyBytes_AsString(v);
943 self->data[i] = buf[0];
944 return 0;
Andrew M. Kuchling1ed7d2d2000-03-30 21:14:30 +0000945}
946
Thomas Woutersed03b412007-08-28 21:37:11 +0000947static int
948mmap_ass_subscript(mmap_object *self, PyObject *item, PyObject *value)
949{
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000950 CHECK_VALID(-1);
Thomas Woutersed03b412007-08-28 21:37:11 +0000951
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000952 if (!is_writable(self))
953 return -1;
Guido van Rossum98297ee2007-11-06 21:34:58 +0000954
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000955 if (PyIndex_Check(item)) {
956 Py_ssize_t i = PyNumber_AsSsize_t(item, PyExc_IndexError);
957 Py_ssize_t v;
Thomas Woutersed03b412007-08-28 21:37:11 +0000958
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000959 if (i == -1 && PyErr_Occurred())
960 return -1;
961 if (i < 0)
962 i += self->size;
Benjamin Petersoncd04db02016-10-05 21:45:48 -0700963 if (i < 0 || i >= self->size) {
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000964 PyErr_SetString(PyExc_IndexError,
965 "mmap index out of range");
966 return -1;
967 }
968 if (value == NULL) {
969 PyErr_SetString(PyExc_TypeError,
970 "mmap doesn't support item deletion");
971 return -1;
972 }
973 if (!PyIndex_Check(value)) {
974 PyErr_SetString(PyExc_TypeError,
975 "mmap item value must be an int");
976 return -1;
977 }
978 v = PyNumber_AsSsize_t(value, PyExc_TypeError);
979 if (v == -1 && PyErr_Occurred())
980 return -1;
981 if (v < 0 || v > 255) {
982 PyErr_SetString(PyExc_ValueError,
983 "mmap item value must be "
984 "in range(0, 256)");
985 return -1;
986 }
Antoine Pitrou22e41552010-08-15 18:07:50 +0000987 self->data[i] = (char) v;
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000988 return 0;
989 }
990 else if (PySlice_Check(item)) {
991 Py_ssize_t start, stop, step, slicelen;
992 Py_buffer vbuf;
Guido van Rossum98297ee2007-11-06 21:34:58 +0000993
Serhiy Storchakab879fe82017-04-08 09:53:51 +0300994 if (PySlice_Unpack(item, &start, &stop, &step) < 0) {
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000995 return -1;
996 }
Serhiy Storchakab879fe82017-04-08 09:53:51 +0300997 slicelen = PySlice_AdjustIndices(self->size, &start, &stop, step);
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000998 if (value == NULL) {
999 PyErr_SetString(PyExc_TypeError,
1000 "mmap object doesn't support slice deletion");
1001 return -1;
1002 }
1003 if (PyObject_GetBuffer(value, &vbuf, PyBUF_SIMPLE) < 0)
1004 return -1;
1005 if (vbuf.len != slicelen) {
1006 PyErr_SetString(PyExc_IndexError,
1007 "mmap slice assignment is wrong size");
1008 PyBuffer_Release(&vbuf);
1009 return -1;
1010 }
Thomas Woutersed03b412007-08-28 21:37:11 +00001011
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001012 if (slicelen == 0) {
1013 }
1014 else if (step == 1) {
1015 memcpy(self->data + start, vbuf.buf, slicelen);
1016 }
1017 else {
Zackery Spytz14514d92019-05-17 01:13:03 -06001018 size_t cur;
1019 Py_ssize_t i;
Guido van Rossum98297ee2007-11-06 21:34:58 +00001020
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001021 for (cur = start, i = 0;
1022 i < slicelen;
1023 cur += step, i++)
1024 {
1025 self->data[cur] = ((char *)vbuf.buf)[i];
1026 }
1027 }
1028 PyBuffer_Release(&vbuf);
1029 return 0;
1030 }
1031 else {
1032 PyErr_SetString(PyExc_TypeError,
1033 "mmap indices must be integer");
1034 return -1;
1035 }
Thomas Woutersed03b412007-08-28 21:37:11 +00001036}
1037
Andrew M. Kuchling1ed7d2d2000-03-30 21:14:30 +00001038static PySequenceMethods mmap_as_sequence = {
Stefan Krah23186992012-03-06 15:37:36 +01001039 (lenfunc)mmap_length, /*sq_length*/
Zackery Spytze9e39762018-06-05 06:59:41 -06001040 0, /*sq_concat*/
1041 0, /*sq_repeat*/
Stefan Krah23186992012-03-06 15:37:36 +01001042 (ssizeargfunc)mmap_item, /*sq_item*/
1043 0, /*sq_slice*/
1044 (ssizeobjargproc)mmap_ass_item, /*sq_ass_item*/
1045 0, /*sq_ass_slice*/
Andrew M. Kuchling1ed7d2d2000-03-30 21:14:30 +00001046};
1047
Thomas Woutersed03b412007-08-28 21:37:11 +00001048static PyMappingMethods mmap_as_mapping = {
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001049 (lenfunc)mmap_length,
1050 (binaryfunc)mmap_subscript,
1051 (objobjargproc)mmap_ass_subscript,
Thomas Woutersed03b412007-08-28 21:37:11 +00001052};
1053
Andrew M. Kuchling1ed7d2d2000-03-30 21:14:30 +00001054static PyBufferProcs mmap_as_buffer = {
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001055 (getbufferproc)mmap_buffer_getbuf,
1056 (releasebufferproc)mmap_buffer_releasebuf,
Andrew M. Kuchling1ed7d2d2000-03-30 21:14:30 +00001057};
1058
Georg Brandl86def6c2008-01-21 20:36:10 +00001059static PyObject *
1060new_mmap_object(PyTypeObject *type, PyObject *args, PyObject *kwdict);
1061
Christian Heimese1c98112008-01-21 11:20:28 +00001062PyDoc_STRVAR(mmap_doc,
1063"Windows: mmap(fileno, length[, tagname[, access[, offset]]])\n\
1064\n\
1065Maps length bytes from the file specified by the file handle fileno,\n\
1066and returns a mmap object. If length is larger than the current size\n\
1067of the file, the file is extended to contain length bytes. If length\n\
1068is 0, the maximum length of the map is the current size of the file,\n\
1069except that if the file is empty Windows raises an exception (you cannot\n\
1070create an empty mapping on Windows).\n\
1071\n\
1072Unix: mmap(fileno, length[, flags[, prot[, access[, offset]]]])\n\
1073\n\
1074Maps length bytes from the file specified by the file descriptor fileno,\n\
1075and returns a mmap object. If length is 0, the maximum length of the map\n\
1076will be the current size of the file when mmap is called.\n\
1077flags specifies the nature of the mapping. MAP_PRIVATE creates a\n\
1078private copy-on-write mapping, so changes to the contents of the mmap\n\
Benjamin Petersonc4fe6f32008-08-19 18:57:56 +00001079object will be private to this process, and MAP_SHARED creates a mapping\n\
Christian Heimese1c98112008-01-21 11:20:28 +00001080that's shared with all other processes mapping the same areas of the file.\n\
1081The default value is MAP_SHARED.\n\
1082\n\
1083To map anonymous memory, pass -1 as the fileno (both versions).");
1084
1085
Andrew M. Kuchling1ed7d2d2000-03-30 21:14:30 +00001086static PyTypeObject mmap_object_type = {
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001087 PyVarObject_HEAD_INIT(NULL, 0)
1088 "mmap.mmap", /* tp_name */
Peter Eisentraut0e0bc4e2018-09-10 18:46:08 +02001089 sizeof(mmap_object), /* tp_basicsize */
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001090 0, /* tp_itemsize */
1091 /* methods */
Taine Zhaod8ca2352019-10-17 18:41:35 +08001092 (destructor)mmap_object_dealloc, /* tp_dealloc */
Jeroen Demeyer530f5062019-05-31 04:13:39 +02001093 0, /* tp_vectorcall_offset */
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001094 0, /* tp_getattr */
1095 0, /* tp_setattr */
Jeroen Demeyer530f5062019-05-31 04:13:39 +02001096 0, /* tp_as_async */
Taine Zhaod8ca2352019-10-17 18:41:35 +08001097 (reprfunc)mmap__repr__method, /* tp_repr */
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001098 0, /* tp_as_number */
Taine Zhaod8ca2352019-10-17 18:41:35 +08001099 &mmap_as_sequence, /* tp_as_sequence */
1100 &mmap_as_mapping, /* tp_as_mapping */
1101 0, /* tp_hash */
1102 0, /* tp_call */
1103 0, /* tp_str */
1104 PyObject_GenericGetAttr, /* tp_getattro */
1105 0, /* tp_setattro */
1106 &mmap_as_buffer, /* tp_as_buffer */
1107 Py_TPFLAGS_DEFAULT | Py_TPFLAGS_BASETYPE, /* tp_flags */
1108 mmap_doc, /* tp_doc */
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001109 0, /* tp_traverse */
1110 0, /* tp_clear */
1111 0, /* tp_richcompare */
Antoine Pitrouc53204b2013-08-05 23:17:30 +02001112 offsetof(mmap_object, weakreflist), /* tp_weaklistoffset */
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001113 0, /* tp_iter */
1114 0, /* tp_iternext */
1115 mmap_object_methods, /* tp_methods */
1116 0, /* tp_members */
Georg Brandl0bccc182010-08-01 14:50:00 +00001117 mmap_object_getset, /* tp_getset */
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001118 0, /* tp_base */
1119 0, /* tp_dict */
1120 0, /* tp_descr_get */
1121 0, /* tp_descr_set */
1122 0, /* tp_dictoffset */
Stefan Krah23186992012-03-06 15:37:36 +01001123 0, /* tp_init */
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001124 PyType_GenericAlloc, /* tp_alloc */
1125 new_mmap_object, /* tp_new */
Stefan Krah23186992012-03-06 15:37:36 +01001126 PyObject_Del, /* tp_free */
Andrew M. Kuchling1ed7d2d2000-03-30 21:14:30 +00001127};
1128
Andrew M. Kuchling70d27422000-06-18 04:45:14 +00001129
Tim Petersec0a5f02006-02-16 23:47:20 +00001130#ifdef UNIX
Antoine Pitrou97696cb2011-02-21 23:46:27 +00001131#ifdef HAVE_LARGEFILE_SUPPORT
1132#define _Py_PARSE_OFF_T "L"
1133#else
1134#define _Py_PARSE_OFF_T "l"
1135#endif
1136
Andrew M. Kuchling1ed7d2d2000-03-30 21:14:30 +00001137static PyObject *
Georg Brandl86def6c2008-01-21 20:36:10 +00001138new_mmap_object(PyTypeObject *type, PyObject *args, PyObject *kwdict)
Andrew M. Kuchling1ed7d2d2000-03-30 21:14:30 +00001139{
Victor Stinnere134a7f2015-03-30 10:09:31 +02001140 struct _Py_stat_struct status;
Zackery Spytzd6e14042018-03-14 14:08:01 -06001141 int fstat_result = -1;
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001142 mmap_object *m_obj;
Antoine Pitrou97696cb2011-02-21 23:46:27 +00001143 Py_ssize_t map_size;
1144 off_t offset = 0;
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001145 int fd, flags = MAP_SHARED, prot = PROT_WRITE | PROT_READ;
1146 int devzero = -1;
1147 int access = (int)ACCESS_DEFAULT;
1148 static char *keywords[] = {"fileno", "length",
Stefan Krah23186992012-03-06 15:37:36 +01001149 "flags", "prot",
1150 "access", "offset", NULL};
Andrew M. Kuchling1ed7d2d2000-03-30 21:14:30 +00001151
Benjamin Petersoncd04db02016-10-05 21:45:48 -07001152 if (!PyArg_ParseTupleAndKeywords(args, kwdict, "in|iii" _Py_PARSE_OFF_T, keywords,
1153 &fd, &map_size, &flags, &prot,
Antoine Pitrou97696cb2011-02-21 23:46:27 +00001154 &access, &offset))
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001155 return NULL;
Benjamin Petersoncd04db02016-10-05 21:45:48 -07001156 if (map_size < 0) {
1157 PyErr_SetString(PyExc_OverflowError,
Zackery Spytz9308dea2018-03-21 00:02:37 -06001158 "memory mapped length must be positive");
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001159 return NULL;
Benjamin Petersoncd04db02016-10-05 21:45:48 -07001160 }
Antoine Pitrou97696cb2011-02-21 23:46:27 +00001161 if (offset < 0) {
1162 PyErr_SetString(PyExc_OverflowError,
1163 "memory mapped offset must be positive");
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001164 return NULL;
Antoine Pitrou97696cb2011-02-21 23:46:27 +00001165 }
Tim Peters5ebfd362001-11-13 23:11:19 +00001166
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001167 if ((access != (int)ACCESS_DEFAULT) &&
1168 ((flags != MAP_SHARED) || (prot != (PROT_WRITE | PROT_READ))))
1169 return PyErr_Format(PyExc_ValueError,
1170 "mmap can't specify both access and flags, prot.");
1171 switch ((access_mode)access) {
1172 case ACCESS_READ:
1173 flags = MAP_SHARED;
1174 prot = PROT_READ;
1175 break;
1176 case ACCESS_WRITE:
1177 flags = MAP_SHARED;
1178 prot = PROT_READ | PROT_WRITE;
1179 break;
1180 case ACCESS_COPY:
1181 flags = MAP_PRIVATE;
1182 prot = PROT_READ | PROT_WRITE;
1183 break;
1184 case ACCESS_DEFAULT:
Antoine Pitrou16a0a0b2011-03-06 01:11:03 +01001185 /* map prot to access type */
1186 if ((prot & PROT_READ) && (prot & PROT_WRITE)) {
1187 /* ACCESS_DEFAULT */
1188 }
1189 else if (prot & PROT_WRITE) {
1190 access = ACCESS_WRITE;
1191 }
1192 else {
1193 access = ACCESS_READ;
1194 }
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001195 break;
1196 default:
1197 return PyErr_Format(PyExc_ValueError,
1198 "mmap invalid access parameter.");
1199 }
Neal Norwitzb5673922002-09-05 21:48:07 +00001200
Steve Dowerb82e17e2019-05-23 08:45:22 -07001201 if (PySys_Audit("mmap.__new__", "ini" _Py_PARSE_OFF_T,
Zackery Spytz08286d52019-06-21 09:31:59 -06001202 fd, map_size, access, offset) < 0) {
Steve Dowerb82e17e2019-05-23 08:45:22 -07001203 return NULL;
1204 }
1205
Victor Stinnera6cd0cf2011-05-02 01:05:37 +02001206#ifdef __APPLE__
1207 /* Issue #11277: fsync(2) is not enough on OS X - a special, OS X specific
1208 fcntl(2) is necessary to force DISKSYNC and get around mmap(2) bug */
1209 if (fd != -1)
1210 (void)fcntl(fd, F_FULLFSYNC);
1211#endif
Nir Soffer4484f9d2018-03-12 01:39:22 +02001212
1213 if (fd != -1) {
1214 Py_BEGIN_ALLOW_THREADS
1215 fstat_result = _Py_fstat_noraise(fd, &status);
1216 Py_END_ALLOW_THREADS
1217 }
1218
1219 if (fd != -1 && fstat_result == 0 && S_ISREG(status.st_mode)) {
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001220 if (map_size == 0) {
Victor Stinnere134a7f2015-03-30 10:09:31 +02001221 if (status.st_size == 0) {
Jesus Cea941bfcc2012-09-10 00:27:55 +02001222 PyErr_SetString(PyExc_ValueError,
1223 "cannot mmap an empty file");
1224 return NULL;
1225 }
Victor Stinnere134a7f2015-03-30 10:09:31 +02001226 if (offset >= status.st_size) {
Antoine Pitrou305bc9e2011-01-20 21:07:24 +00001227 PyErr_SetString(PyExc_ValueError,
1228 "mmap offset is greater than file size");
1229 return NULL;
1230 }
Victor Stinnere134a7f2015-03-30 10:09:31 +02001231 if (status.st_size - offset > PY_SSIZE_T_MAX) {
Antoine Pitrou97696cb2011-02-21 23:46:27 +00001232 PyErr_SetString(PyExc_ValueError,
1233 "mmap length is too large");
Richard Oudkerk0d09ba82013-02-13 12:18:03 +00001234 return NULL;
1235 }
Victor Stinnere134a7f2015-03-30 10:09:31 +02001236 map_size = (Py_ssize_t) (status.st_size - offset);
Benjamin Petersoncd04db02016-10-05 21:45:48 -07001237 } else if (offset > status.st_size || status.st_size - offset < map_size) {
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001238 PyErr_SetString(PyExc_ValueError,
1239 "mmap length is greater than file size");
1240 return NULL;
1241 }
1242 }
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001243 m_obj = (mmap_object *)type->tp_alloc(type, 0);
1244 if (m_obj == NULL) {return NULL;}
1245 m_obj->data = NULL;
Benjamin Petersoncd04db02016-10-05 21:45:48 -07001246 m_obj->size = map_size;
1247 m_obj->pos = 0;
Antoine Pitrouc53204b2013-08-05 23:17:30 +02001248 m_obj->weakreflist = NULL;
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001249 m_obj->exports = 0;
1250 m_obj->offset = offset;
1251 if (fd == -1) {
1252 m_obj->fd = -1;
1253 /* Assume the caller wants to map anonymous memory.
1254 This is the same behaviour as Windows. mmap.mmap(-1, size)
1255 on both Windows and Unix map anonymous memory.
1256 */
Neal Norwitz0e6bc8c2006-02-05 05:45:43 +00001257#ifdef MAP_ANONYMOUS
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001258 /* BSD way to map anonymous memory */
1259 flags |= MAP_ANONYMOUS;
Lihua Zhao4fb15022019-05-21 18:50:14 +08001260
1261 /* VxWorks only supports MAP_ANONYMOUS with MAP_PRIVATE flag */
1262#ifdef __VXWORKS__
1263 flags &= ~MAP_SHARED;
1264 flags |= MAP_PRIVATE;
1265#endif
1266
Neal Norwitz0e6bc8c2006-02-05 05:45:43 +00001267#else
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001268 /* SVR4 method to map anonymous memory is to open /dev/zero */
Victor Stinnerdaf45552013-08-28 00:53:59 +02001269 fd = devzero = _Py_open("/dev/zero", O_RDWR);
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001270 if (devzero == -1) {
1271 Py_DECREF(m_obj);
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001272 return NULL;
1273 }
Neal Norwitz0e6bc8c2006-02-05 05:45:43 +00001274#endif
Victor Stinnerdaf45552013-08-28 00:53:59 +02001275 }
1276 else {
1277 m_obj->fd = _Py_dup(fd);
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001278 if (m_obj->fd == -1) {
1279 Py_DECREF(m_obj);
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001280 return NULL;
1281 }
1282 }
Neal Norwitz0e6bc8c2006-02-05 05:45:43 +00001283
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001284 m_obj->data = mmap(NULL, map_size,
1285 prot, flags,
1286 fd, offset);
Neal Norwitz0e6bc8c2006-02-05 05:45:43 +00001287
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001288 if (devzero != -1) {
1289 close(devzero);
1290 }
1291
1292 if (m_obj->data == (char *)-1) {
1293 m_obj->data = NULL;
1294 Py_DECREF(m_obj);
Antoine Pitrou6b4883d2011-10-12 02:54:14 +02001295 PyErr_SetFromErrno(PyExc_OSError);
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001296 return NULL;
1297 }
1298 m_obj->access = (access_mode)access;
1299 return (PyObject *)m_obj;
Andrew M. Kuchling1ed7d2d2000-03-30 21:14:30 +00001300}
1301#endif /* UNIX */
1302
Martin v. Löwis6238d2b2002-06-30 15:26:10 +00001303#ifdef MS_WINDOWS
Antoine Pitrou97696cb2011-02-21 23:46:27 +00001304
1305/* A note on sizes and offsets: while the actual map size must hold in a
1306 Py_ssize_t, both the total file size and the start offset can be longer
Benjamin Petersonaf580df2016-09-06 10:46:49 -07001307 than a Py_ssize_t, so we use long long which is always 64-bit.
Antoine Pitrou97696cb2011-02-21 23:46:27 +00001308*/
1309
Andrew M. Kuchling1ed7d2d2000-03-30 21:14:30 +00001310static PyObject *
Georg Brandl86def6c2008-01-21 20:36:10 +00001311new_mmap_object(PyTypeObject *type, PyObject *args, PyObject *kwdict)
Andrew M. Kuchling1ed7d2d2000-03-30 21:14:30 +00001312{
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001313 mmap_object *m_obj;
Antoine Pitrou97696cb2011-02-21 23:46:27 +00001314 Py_ssize_t map_size;
Benjamin Petersonaf580df2016-09-06 10:46:49 -07001315 long long offset = 0, size;
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001316 DWORD off_hi; /* upper 32 bits of offset */
1317 DWORD off_lo; /* lower 32 bits of offset */
1318 DWORD size_hi; /* upper 32 bits of size */
1319 DWORD size_lo; /* lower 32 bits of size */
Serhiy Storchakae2f92de2017-11-11 13:06:26 +02001320 const char *tagname = "";
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001321 DWORD dwErr = 0;
1322 int fileno;
1323 HANDLE fh = 0;
1324 int access = (access_mode)ACCESS_DEFAULT;
1325 DWORD flProtect, dwDesiredAccess;
1326 static char *keywords[] = { "fileno", "length",
Stefan Krah23186992012-03-06 15:37:36 +01001327 "tagname",
1328 "access", "offset", NULL };
Andrew M. Kuchling1ed7d2d2000-03-30 21:14:30 +00001329
Benjamin Petersoncd04db02016-10-05 21:45:48 -07001330 if (!PyArg_ParseTupleAndKeywords(args, kwdict, "in|ziL", keywords,
1331 &fileno, &map_size,
Antoine Pitrou97696cb2011-02-21 23:46:27 +00001332 &tagname, &access, &offset)) {
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001333 return NULL;
1334 }
Tim Peters5ebfd362001-11-13 23:11:19 +00001335
Steve Dowerb82e17e2019-05-23 08:45:22 -07001336 if (PySys_Audit("mmap.__new__", "iniL",
1337 fileno, map_size, access, offset) < 0) {
1338 return NULL;
1339 }
1340
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001341 switch((access_mode)access) {
1342 case ACCESS_READ:
1343 flProtect = PAGE_READONLY;
1344 dwDesiredAccess = FILE_MAP_READ;
1345 break;
1346 case ACCESS_DEFAULT: case ACCESS_WRITE:
1347 flProtect = PAGE_READWRITE;
1348 dwDesiredAccess = FILE_MAP_WRITE;
1349 break;
1350 case ACCESS_COPY:
1351 flProtect = PAGE_WRITECOPY;
1352 dwDesiredAccess = FILE_MAP_COPY;
1353 break;
1354 default:
1355 return PyErr_Format(PyExc_ValueError,
1356 "mmap invalid access parameter.");
1357 }
Tim Peters5ebfd362001-11-13 23:11:19 +00001358
Benjamin Petersoncd04db02016-10-05 21:45:48 -07001359 if (map_size < 0) {
1360 PyErr_SetString(PyExc_OverflowError,
Zackery Spytz9308dea2018-03-21 00:02:37 -06001361 "memory mapped length must be positive");
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001362 return NULL;
Benjamin Petersoncd04db02016-10-05 21:45:48 -07001363 }
Antoine Pitrou97696cb2011-02-21 23:46:27 +00001364 if (offset < 0) {
1365 PyErr_SetString(PyExc_OverflowError,
1366 "memory mapped offset must be positive");
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001367 return NULL;
Antoine Pitrou97696cb2011-02-21 23:46:27 +00001368 }
Tim Petersec0a5f02006-02-16 23:47:20 +00001369
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001370 /* assume -1 and 0 both mean invalid filedescriptor
1371 to 'anonymously' map memory.
1372 XXX: fileno == 0 is a valid fd, but was accepted prior to 2.5.
1373 XXX: Should this code be added?
1374 if (fileno == 0)
1375 PyErr_WarnEx(PyExc_DeprecationWarning,
1376 "don't use 0 for anonymous memory",
1377 1);
1378 */
1379 if (fileno != -1 && fileno != 0) {
Brian Curtinea47eaa2010-08-01 15:26:26 +00001380 /* Ensure that fileno is within the CRT's valid range */
Steve Dower8fc89802015-04-12 00:26:27 -04001381 _Py_BEGIN_SUPPRESS_IPH
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001382 fh = (HANDLE)_get_osfhandle(fileno);
Steve Dower8fc89802015-04-12 00:26:27 -04001383 _Py_END_SUPPRESS_IPH
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001384 if (fh==(HANDLE)-1) {
Antoine Pitrou6b4883d2011-10-12 02:54:14 +02001385 PyErr_SetFromErrno(PyExc_OSError);
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001386 return NULL;
1387 }
1388 /* Win9x appears to need us seeked to zero */
1389 lseek(fileno, 0, SEEK_SET);
1390 }
Andrew M. Kuchling1ed7d2d2000-03-30 21:14:30 +00001391
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001392 m_obj = (mmap_object *)type->tp_alloc(type, 0);
1393 if (m_obj == NULL)
1394 return NULL;
1395 /* Set every field to an invalid marker, so we can safely
1396 destruct the object in the face of failure */
1397 m_obj->data = NULL;
1398 m_obj->file_handle = INVALID_HANDLE_VALUE;
1399 m_obj->map_handle = NULL;
1400 m_obj->tagname = NULL;
1401 m_obj->offset = offset;
Mark Hammond2cbed002000-07-30 02:22:43 +00001402
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001403 if (fh) {
1404 /* It is necessary to duplicate the handle, so the
1405 Python code can close it on us */
1406 if (!DuplicateHandle(
1407 GetCurrentProcess(), /* source process handle */
1408 fh, /* handle to be duplicated */
1409 GetCurrentProcess(), /* target proc handle */
1410 (LPHANDLE)&m_obj->file_handle, /* result */
1411 0, /* access - ignored due to options value */
1412 FALSE, /* inherited by child processes? */
1413 DUPLICATE_SAME_ACCESS)) { /* options */
1414 dwErr = GetLastError();
1415 Py_DECREF(m_obj);
1416 PyErr_SetFromWindowsErr(dwErr);
1417 return NULL;
1418 }
1419 if (!map_size) {
1420 DWORD low,high;
1421 low = GetFileSize(fh, &high);
1422 /* low might just happen to have the value INVALID_FILE_SIZE;
1423 so we need to check the last error also. */
1424 if (low == INVALID_FILE_SIZE &&
1425 (dwErr = GetLastError()) != NO_ERROR) {
1426 Py_DECREF(m_obj);
1427 return PyErr_SetFromWindowsErr(dwErr);
1428 }
Guido van Rossum98297ee2007-11-06 21:34:58 +00001429
Benjamin Petersonaf580df2016-09-06 10:46:49 -07001430 size = (((long long) high) << 32) + low;
Jesus Cea1f2799b2012-09-10 22:49:50 +02001431 if (size == 0) {
1432 PyErr_SetString(PyExc_ValueError,
1433 "cannot mmap an empty file");
Jesus Ceae8db3562012-09-10 22:58:07 +02001434 Py_DECREF(m_obj);
Jesus Cea1f2799b2012-09-10 22:49:50 +02001435 return NULL;
1436 }
Antoine Pitrou97696cb2011-02-21 23:46:27 +00001437 if (offset >= size) {
Antoine Pitrou305bc9e2011-01-20 21:07:24 +00001438 PyErr_SetString(PyExc_ValueError,
1439 "mmap offset is greater than file size");
1440 Py_DECREF(m_obj);
1441 return NULL;
1442 }
Richard Oudkerk0d09ba82013-02-13 12:18:03 +00001443 if (size - offset > PY_SSIZE_T_MAX) {
1444 PyErr_SetString(PyExc_ValueError,
1445 "mmap length is too large");
1446 Py_DECREF(m_obj);
1447 return NULL;
1448 }
1449 m_obj->size = (Py_ssize_t) (size - offset);
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001450 } else {
1451 m_obj->size = map_size;
Antoine Pitrou97696cb2011-02-21 23:46:27 +00001452 size = offset + map_size;
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001453 }
1454 }
1455 else {
1456 m_obj->size = map_size;
Antoine Pitrou97696cb2011-02-21 23:46:27 +00001457 size = offset + map_size;
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001458 }
Guido van Rossum09fdf072000-03-31 01:17:07 +00001459
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001460 /* set the initial position */
1461 m_obj->pos = (size_t) 0;
Guido van Rossum09fdf072000-03-31 01:17:07 +00001462
Antoine Pitrouc53204b2013-08-05 23:17:30 +02001463 m_obj->weakreflist = NULL;
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001464 m_obj->exports = 0;
1465 /* set the tag name */
1466 if (tagname != NULL && *tagname != '\0') {
1467 m_obj->tagname = PyMem_Malloc(strlen(tagname)+1);
1468 if (m_obj->tagname == NULL) {
1469 PyErr_NoMemory();
1470 Py_DECREF(m_obj);
1471 return NULL;
1472 }
1473 strcpy(m_obj->tagname, tagname);
1474 }
1475 else
1476 m_obj->tagname = NULL;
Mark Hammond2cbed002000-07-30 02:22:43 +00001477
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001478 m_obj->access = (access_mode)access;
Antoine Pitrou97696cb2011-02-21 23:46:27 +00001479 size_hi = (DWORD)(size >> 32);
1480 size_lo = (DWORD)(size & 0xFFFFFFFF);
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001481 off_hi = (DWORD)(offset >> 32);
1482 off_lo = (DWORD)(offset & 0xFFFFFFFF);
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001483 /* For files, it would be sufficient to pass 0 as size.
1484 For anonymous maps, we have to pass the size explicitly. */
1485 m_obj->map_handle = CreateFileMapping(m_obj->file_handle,
1486 NULL,
1487 flProtect,
1488 size_hi,
1489 size_lo,
1490 m_obj->tagname);
1491 if (m_obj->map_handle != NULL) {
1492 m_obj->data = (char *) MapViewOfFile(m_obj->map_handle,
1493 dwDesiredAccess,
1494 off_hi,
1495 off_lo,
1496 m_obj->size);
1497 if (m_obj->data != NULL)
1498 return (PyObject *)m_obj;
1499 else {
1500 dwErr = GetLastError();
1501 CloseHandle(m_obj->map_handle);
1502 m_obj->map_handle = NULL;
1503 }
1504 } else
1505 dwErr = GetLastError();
1506 Py_DECREF(m_obj);
1507 PyErr_SetFromWindowsErr(dwErr);
1508 return NULL;
Andrew M. Kuchling1ed7d2d2000-03-30 21:14:30 +00001509}
Martin v. Löwis6238d2b2002-06-30 15:26:10 +00001510#endif /* MS_WINDOWS */
Andrew M. Kuchling1ed7d2d2000-03-30 21:14:30 +00001511
Thomas Wouters00ee7ba2006-08-21 19:07:27 +00001512static void
1513setint(PyObject *d, const char *name, long value)
1514{
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001515 PyObject *o = PyLong_FromLong(value);
Hai Shi56a45142019-09-16 00:56:57 -05001516 if (o) {
1517 PyDict_SetItemString(d, name, o);
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001518 Py_DECREF(o);
1519 }
Thomas Wouters00ee7ba2006-08-21 19:07:27 +00001520}
1521
Martin v. Löwis1a214512008-06-11 05:26:20 +00001522
1523static struct PyModuleDef mmapmodule = {
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001524 PyModuleDef_HEAD_INIT,
1525 "mmap",
1526 NULL,
1527 -1,
1528 NULL,
1529 NULL,
1530 NULL,
1531 NULL,
1532 NULL
Martin v. Löwis1a214512008-06-11 05:26:20 +00001533};
1534
Mark Hammond62b1ab12002-07-23 06:31:15 +00001535PyMODINIT_FUNC
Martin v. Löwis1a214512008-06-11 05:26:20 +00001536PyInit_mmap(void)
Andrew M. Kuchling1ed7d2d2000-03-30 21:14:30 +00001537{
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001538 PyObject *dict, *module;
Tim Peters2caf8df2001-01-14 05:05:51 +00001539
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001540 if (PyType_Ready(&mmap_object_type) < 0)
1541 return NULL;
Tim Peters2caf8df2001-01-14 05:05:51 +00001542
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001543 module = PyModule_Create(&mmapmodule);
1544 if (module == NULL)
1545 return NULL;
1546 dict = PyModule_GetDict(module);
1547 if (!dict)
1548 return NULL;
Antoine Pitrou6b4883d2011-10-12 02:54:14 +02001549 PyDict_SetItemString(dict, "error", PyExc_OSError);
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001550 PyDict_SetItemString(dict, "mmap", (PyObject*) &mmap_object_type);
Andrew M. Kuchling1ed7d2d2000-03-30 21:14:30 +00001551#ifdef PROT_EXEC
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001552 setint(dict, "PROT_EXEC", PROT_EXEC);
Andrew M. Kuchling1ed7d2d2000-03-30 21:14:30 +00001553#endif
1554#ifdef PROT_READ
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001555 setint(dict, "PROT_READ", PROT_READ);
Andrew M. Kuchling1ed7d2d2000-03-30 21:14:30 +00001556#endif
1557#ifdef PROT_WRITE
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001558 setint(dict, "PROT_WRITE", PROT_WRITE);
Andrew M. Kuchling1ed7d2d2000-03-30 21:14:30 +00001559#endif
1560
1561#ifdef MAP_SHARED
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001562 setint(dict, "MAP_SHARED", MAP_SHARED);
Andrew M. Kuchling1ed7d2d2000-03-30 21:14:30 +00001563#endif
1564#ifdef MAP_PRIVATE
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001565 setint(dict, "MAP_PRIVATE", MAP_PRIVATE);
Andrew M. Kuchling1ed7d2d2000-03-30 21:14:30 +00001566#endif
1567#ifdef MAP_DENYWRITE
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001568 setint(dict, "MAP_DENYWRITE", MAP_DENYWRITE);
Andrew M. Kuchling1ed7d2d2000-03-30 21:14:30 +00001569#endif
1570#ifdef MAP_EXECUTABLE
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001571 setint(dict, "MAP_EXECUTABLE", MAP_EXECUTABLE);
Andrew M. Kuchling1ed7d2d2000-03-30 21:14:30 +00001572#endif
Neal Norwitz0e6bc8c2006-02-05 05:45:43 +00001573#ifdef MAP_ANONYMOUS
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001574 setint(dict, "MAP_ANON", MAP_ANONYMOUS);
1575 setint(dict, "MAP_ANONYMOUS", MAP_ANONYMOUS);
Andrew M. Kuchling1ed7d2d2000-03-30 21:14:30 +00001576#endif
1577
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001578 setint(dict, "PAGESIZE", (long)my_getpagesize());
Andrew M. Kuchling961fe172000-06-03 19:41:42 +00001579
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001580 setint(dict, "ALLOCATIONGRANULARITY", (long)my_getallocationgranularity());
Guido van Rossum8ce8a782007-11-01 19:42:39 +00001581
Justus Schwabedal5a8a84b2017-11-07 15:51:43 -05001582 setint(dict, "ACCESS_DEFAULT", ACCESS_DEFAULT);
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001583 setint(dict, "ACCESS_READ", ACCESS_READ);
1584 setint(dict, "ACCESS_WRITE", ACCESS_WRITE);
1585 setint(dict, "ACCESS_COPY", ACCESS_COPY);
Zackery Spytz02db6962019-05-27 10:48:17 -06001586
1587#ifdef HAVE_MADVISE
1588 // Conventional advice values
1589#ifdef MADV_NORMAL
1590 setint(dict, "MADV_NORMAL", MADV_NORMAL);
1591#endif
1592#ifdef MADV_RANDOM
1593 setint(dict, "MADV_RANDOM", MADV_RANDOM);
1594#endif
1595#ifdef MADV_SEQUENTIAL
1596 setint(dict, "MADV_SEQUENTIAL", MADV_SEQUENTIAL);
1597#endif
1598#ifdef MADV_WILLNEED
1599 setint(dict, "MADV_WILLNEED", MADV_WILLNEED);
1600#endif
1601#ifdef MADV_DONTNEED
1602 setint(dict, "MADV_DONTNEED", MADV_DONTNEED);
1603#endif
1604
1605 // Linux-specific advice values
1606#ifdef MADV_REMOVE
1607 setint(dict, "MADV_REMOVE", MADV_REMOVE);
1608#endif
1609#ifdef MADV_DONTFORK
1610 setint(dict, "MADV_DONTFORK", MADV_DONTFORK);
1611#endif
1612#ifdef MADV_DOFORK
1613 setint(dict, "MADV_DOFORK", MADV_DOFORK);
1614#endif
1615#ifdef MADV_HWPOISON
1616 setint(dict, "MADV_HWPOISON", MADV_HWPOISON);
1617#endif
1618#ifdef MADV_MERGEABLE
1619 setint(dict, "MADV_MERGEABLE", MADV_MERGEABLE);
1620#endif
1621#ifdef MADV_UNMERGEABLE
1622 setint(dict, "MADV_UNMERGEABLE", MADV_UNMERGEABLE);
1623#endif
1624#ifdef MADV_SOFT_OFFLINE
1625 setint(dict, "MADV_SOFT_OFFLINE", MADV_SOFT_OFFLINE);
1626#endif
1627#ifdef MADV_HUGEPAGE
1628 setint(dict, "MADV_HUGEPAGE", MADV_HUGEPAGE);
1629#endif
1630#ifdef MADV_NOHUGEPAGE
1631 setint(dict, "MADV_NOHUGEPAGE", MADV_NOHUGEPAGE);
1632#endif
1633#ifdef MADV_DONTDUMP
1634 setint(dict, "MADV_DONTDUMP", MADV_DONTDUMP);
1635#endif
1636#ifdef MADV_DODUMP
1637 setint(dict, "MADV_DODUMP", MADV_DODUMP);
1638#endif
1639#ifdef MADV_FREE // (Also present on FreeBSD and macOS.)
1640 setint(dict, "MADV_FREE", MADV_FREE);
1641#endif
1642
1643 // FreeBSD-specific
1644#ifdef MADV_NOSYNC
1645 setint(dict, "MADV_NOSYNC", MADV_NOSYNC);
1646#endif
1647#ifdef MADV_AUTOSYNC
1648 setint(dict, "MADV_AUTOSYNC", MADV_AUTOSYNC);
1649#endif
1650#ifdef MADV_NOCORE
1651 setint(dict, "MADV_NOCORE", MADV_NOCORE);
1652#endif
1653#ifdef MADV_CORE
1654 setint(dict, "MADV_CORE", MADV_CORE);
1655#endif
1656#ifdef MADV_PROTECT
1657 setint(dict, "MADV_PROTECT", MADV_PROTECT);
1658#endif
1659#endif // HAVE_MADVISE
1660
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001661 return module;
Tim Peters5ebfd362001-11-13 23:11:19 +00001662}