Commit | Line | Data |
---|---|---|
81c4586f PC |
1 | |
2 | #include <windows.h> | |
3 | #include <errno.h> | |
4 | #include <io.h> | |
5 | ||
6 | #include "mman.h" | |
7 | ||
8 | #ifndef FILE_MAP_EXECUTE | |
9 | #define FILE_MAP_EXECUTE 0x0020 | |
10 | #endif /* FILE_MAP_EXECUTE */ | |
11 | ||
12 | static int __map_mman_error(const DWORD err, const int deferr) | |
13 | { | |
14 | if (err == 0) | |
15 | return 0; | |
16 | //TODO: implement | |
17 | return err; | |
18 | } | |
19 | ||
20 | static DWORD __map_mmap_prot_page(const int prot) | |
21 | { | |
22 | DWORD protect = 0; | |
23 | ||
24 | if (prot == PROT_NONE) | |
25 | return protect; | |
26 | ||
27 | if ((prot & PROT_EXEC) != 0) | |
28 | { | |
29 | protect = ((prot & PROT_WRITE) != 0) ? | |
30 | PAGE_EXECUTE_READWRITE : PAGE_EXECUTE_READ; | |
31 | } | |
32 | else | |
33 | { | |
34 | protect = ((prot & PROT_WRITE) != 0) ? | |
35 | PAGE_READWRITE : PAGE_READONLY; | |
36 | } | |
37 | ||
38 | return protect; | |
39 | } | |
40 | ||
41 | static DWORD __map_mmap_prot_file(const int prot) | |
42 | { | |
43 | DWORD desiredAccess = 0; | |
44 | ||
45 | if (prot == PROT_NONE) | |
46 | return desiredAccess; | |
47 | ||
48 | if ((prot & PROT_READ) != 0) | |
49 | desiredAccess |= FILE_MAP_READ; | |
50 | if ((prot & PROT_WRITE) != 0) | |
51 | desiredAccess |= FILE_MAP_WRITE; | |
52 | if ((prot & PROT_EXEC) != 0) | |
53 | desiredAccess |= FILE_MAP_EXECUTE; | |
54 | ||
55 | return desiredAccess; | |
56 | } | |
57 | ||
58 | void* mmap(void *addr, size_t len, int prot, int flags, int fildes, OffsetType off) | |
59 | { | |
60 | HANDLE fm, h; | |
61 | ||
62 | void * map = MAP_FAILED; | |
63 | ||
64 | #ifdef _MSC_VER | |
65 | #pragma warning(push) | |
66 | #pragma warning(disable: 4293) | |
67 | #endif | |
68 | ||
69 | const DWORD dwFileOffsetLow = (sizeof(OffsetType) <= sizeof(DWORD)) ? | |
70 | (DWORD)off : (DWORD)(off & 0xFFFFFFFFL); | |
71 | const DWORD dwFileOffsetHigh = (sizeof(OffsetType) <= sizeof(DWORD)) ? | |
72 | (DWORD)0 : (DWORD)((off >> 32) & 0xFFFFFFFFL); | |
73 | const DWORD protect = __map_mmap_prot_page(prot); | |
74 | const DWORD desiredAccess = __map_mmap_prot_file(prot); | |
75 | ||
76 | const OffsetType maxSize = off + (OffsetType)len; | |
77 | ||
78 | const DWORD dwMaxSizeLow = (sizeof(OffsetType) <= sizeof(DWORD)) ? | |
79 | (DWORD)maxSize : (DWORD)(maxSize & 0xFFFFFFFFL); | |
80 | const DWORD dwMaxSizeHigh = (sizeof(OffsetType) <= sizeof(DWORD)) ? | |
81 | (DWORD)0 : (DWORD)((maxSize >> 32) & 0xFFFFFFFFL); | |
82 | ||
83 | #ifdef _MSC_VER | |
84 | #pragma warning(pop) | |
85 | #endif | |
86 | ||
87 | errno = 0; | |
88 | ||
89 | if (len == 0 | |
90 | /* Usupported protection combinations */ | |
91 | || prot == PROT_EXEC) | |
92 | { | |
93 | errno = EINVAL; | |
94 | return MAP_FAILED; | |
95 | } | |
96 | ||
97 | h = ((flags & MAP_ANONYMOUS) == 0) ? | |
98 | (HANDLE)_get_osfhandle(fildes) : INVALID_HANDLE_VALUE; | |
99 | ||
100 | if ((flags & MAP_ANONYMOUS) == 0 && h == INVALID_HANDLE_VALUE) | |
101 | { | |
102 | errno = EBADF; | |
103 | return MAP_FAILED; | |
104 | } | |
105 | ||
106 | fm = CreateFileMapping(h, NULL, protect, dwMaxSizeHigh, dwMaxSizeLow, NULL); | |
107 | ||
108 | if (fm == NULL) | |
109 | { | |
110 | errno = __map_mman_error(GetLastError(), EPERM); | |
111 | return MAP_FAILED; | |
112 | } | |
113 | ||
114 | if ((flags & MAP_FIXED) == 0) | |
115 | { | |
116 | map = MapViewOfFile(fm, desiredAccess, dwFileOffsetHigh, dwFileOffsetLow, len); | |
117 | } | |
118 | else | |
119 | { | |
120 | map = MapViewOfFileEx(fm, desiredAccess, dwFileOffsetHigh, dwFileOffsetLow, len, addr); | |
121 | } | |
122 | ||
123 | CloseHandle(fm); | |
124 | ||
125 | if (map == NULL) | |
126 | { | |
127 | errno = __map_mman_error(GetLastError(), EPERM); | |
128 | return MAP_FAILED; | |
129 | } | |
130 | ||
131 | return map; | |
132 | } | |
133 | ||
134 | int munmap(void *addr, size_t len) | |
135 | { | |
136 | if (UnmapViewOfFile(addr)) | |
137 | return 0; | |
138 | ||
139 | errno = __map_mman_error(GetLastError(), EPERM); | |
140 | ||
141 | return -1; | |
142 | } | |
143 | ||
144 | int _mprotect(void *addr, size_t len, int prot) | |
145 | { | |
146 | DWORD newProtect = __map_mmap_prot_page(prot); | |
147 | DWORD oldProtect = 0; | |
148 | ||
149 | if (VirtualProtect(addr, len, newProtect, &oldProtect)) | |
150 | return 0; | |
151 | ||
152 | errno = __map_mman_error(GetLastError(), EPERM); | |
153 | ||
154 | return -1; | |
155 | } | |
156 | ||
157 | int msync(void *addr, size_t len, int flags) | |
158 | { | |
159 | if (FlushViewOfFile(addr, len)) | |
160 | return 0; | |
161 | ||
162 | errno = __map_mman_error(GetLastError(), EPERM); | |
163 | ||
164 | return -1; | |
165 | } | |
166 | ||
167 | int mlock(const void *addr, size_t len) | |
168 | { | |
169 | if (VirtualLock((LPVOID)addr, len)) | |
170 | return 0; | |
171 | ||
172 | errno = __map_mman_error(GetLastError(), EPERM); | |
173 | ||
174 | return -1; | |
175 | } | |
176 | ||
177 | int munlock(const void *addr, size_t len) | |
178 | { | |
179 | if (VirtualUnlock((LPVOID)addr, len)) | |
180 | return 0; | |
181 | ||
182 | errno = __map_mman_error(GetLastError(), EPERM); | |
183 | ||
184 | return -1; | |
185 | } |