1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
149
150
151
152
153
154
155
156
157
158
159
160
161
162
163
164
165
166
167
168
169
170
171
172
173
174
175
176
177
178
179
180
181
182
183
184
185
186
187
188
189
|
// Userspace emulation of mmap and munmap. Restrictions apply.
//
// This is meant to be complete enough to be compatible with code that uses
// mmap for simple file I/O. It just allocates memory with malloc and reads
// and writes data with pread and pwrite.
#ifdef __wasilibc_unmodified_upstream
#define _WASI_EMULATED_MMAN
#else
#define _WASI_EMULATED_MMAN 1
#endif
#include <stdlib.h>
#include <errno.h>
#include <unistd.h>
#include <string.h>
#include <sys/mman.h>
#include <sys/types.h>
struct map {
int prot;
int flags;
off_t offset;
size_t length;
int fd;
};
void *mmap(void *addr, size_t length, int prot, int flags,
int fd, off_t offset) {
// Check for unsupported flags.
if ((flags & (MAP_PRIVATE | MAP_SHARED)) == 0 ||
(flags & MAP_FIXED) != 0 ||
#ifdef MAP_SHARED_VALIDATE
(flags & MAP_SHARED_VALIDATE) == MAP_SHARED_VALIDATE ||
#endif
#ifdef MAP_NORESERVE
(flags & MAP_NORESERVE) != 0 ||
#endif
#ifdef MAP_GROWSDOWN
(flags & MAP_GROWSDOWN) != 0 ||
#endif
#ifdef MAP_HUGETLB
(flags & MAP_HUGETLB) != 0 ||
#endif
#ifdef MAP_FIXED_NOREPLACE
(flags & MAP_FIXED_NOREPLACE) != 0 ||
#endif
0)
{
errno = EINVAL;
return MAP_FAILED;
}
// Check for unsupported protection requests.
if (prot == PROT_NONE ||
#ifdef PROT_EXEC
(prot & PROT_EXEC) != 0 ||
#endif
0)
{
errno = EINVAL;
return MAP_FAILED;
}
// To be consistent with POSIX.
if (length == 0) {
errno = EINVAL;
return MAP_FAILED;
}
// Check for integer overflow.
size_t buf_len = 0;
if (__builtin_add_overflow(length, sizeof(struct map), &buf_len)) {
errno = ENOMEM;
return MAP_FAILED;
}
// Allocate the memory.
struct map *map = malloc(buf_len);
if (!map) {
errno = ENOMEM;
return MAP_FAILED;
}
// Initialize the header.
map->prot = prot;
map->flags = flags;
map->offset = offset;
map->length = length;
// Initialize the main memory buffer, either with the contents of a file,
// or with zeros.
addr = map + 1;
if ((flags & MAP_ANON) == 0) {
int new_fd = dup(fd);
if (new_fd < 0) {
errno = EINVAL;
free(map);
return NULL;
}
map->fd = new_fd;
char *body = (char *)addr;
while (length > 0) {
const ssize_t nread = pread(fd, body, length, offset);
if (nread < 0) {
if (errno == EINTR)
continue;
return MAP_FAILED;
}
if (nread == 0)
break;
length -= (size_t)nread;
offset += (size_t)nread;
body += (size_t)nread;
}
} else {
map->fd = -1;
memset(addr, 0, length);
}
return addr;
}
int munmap(void *addr, size_t length) {
struct map *map = (struct map *)addr - 1;
// We don't support partial munmapping.
if (map->length != length) {
errno = EINVAL;
return -1;
}
// Release the memory.
free(map);
// Write the data back to the backing file and close
// the file handle
if (map->fd > 0) {
if ((map->prot & PROT_WRITE) != 0) {
msync(addr, length, MS_SYNC);
}
close(map->fd);
}
// Success!
return 0;
}
int msync (void *addr, size_t length, int flags) {
struct map *map = (struct map *)addr - 1;
size_t map_flags = map->flags;
off_t map_offset = map->offset;
size_t map_length = map->length;
int fd = map->fd;
if (length > map_length) {
errno = EINVAL;
return -1;
}
if ((map->prot & PROT_WRITE) != 0) {
errno = EINVAL;
return -1;
}
if ((map_flags & MAP_ANON) == 0) {
char *body = (char *)addr;
while (length > 0) {
const ssize_t nwrite = pwrite(fd, body, length, map_offset);
if (nwrite > 0) {
length -= (size_t)nwrite;
map_offset += (size_t)nwrite;
body += (size_t)nwrite;
} else if (errno == EINTR) {
continue;
} else {
return -1;
}
}
}
return 0;
}
|
