OLD | NEW |
| (Empty) |
1 // Copyright (c) 2010 The Chromium Authors. All rights reserved. | |
2 // Use of this source code is governed by a BSD-style license that can be | |
3 // found in the LICENSE file. | |
4 | |
5 #include <errno.h> | |
6 #include <fcntl.h> | |
7 #include <linux/unistd.h> | |
8 #include <signal.h> | |
9 #include <stdarg.h> | |
10 #include <stdlib.h> | |
11 #include <sys/ptrace.h> | |
12 #include <sys/types.h> | |
13 #include <sys/wait.h> | |
14 | |
15 #include "library.h" | |
16 #include "maps.h" | |
17 #include "sandbox_impl.h" | |
18 | |
19 namespace playground { | |
20 | |
21 Maps::Maps(int proc_self_maps) : | |
22 proc_self_maps_(proc_self_maps), | |
23 begin_iter_(this, true, false), | |
24 end_iter_(this, false, true), | |
25 vsyscall_(0) { | |
26 Sandbox::SysCalls sys; | |
27 if (proc_self_maps_ >= 0 && | |
28 !sys.lseek(proc_self_maps_, 0, SEEK_SET)) { | |
29 char buf[256] = { 0 }; | |
30 int len = 0, rc = 1; | |
31 bool long_line = false; | |
32 do { | |
33 if (rc > 0) { | |
34 rc = Sandbox::read(sys, proc_self_maps_, buf + len, | |
35 sizeof(buf) - len - 1); | |
36 if (rc > 0) { | |
37 len += rc; | |
38 } | |
39 } | |
40 char *ptr = buf; | |
41 if (!long_line) { | |
42 long_line = true; | |
43 unsigned long start = strtoul(ptr, &ptr, 16); | |
44 unsigned long stop = strtoul(ptr + 1, &ptr, 16); | |
45 while (*ptr == ' ' || *ptr == '\t') ++ptr; | |
46 char *perm_ptr = ptr; | |
47 while (*ptr && *ptr != ' ' && *ptr != '\t') ++ptr; | |
48 string perm(perm_ptr, ptr - perm_ptr); | |
49 unsigned long offset = strtoul(ptr, &ptr, 16); | |
50 while (*ptr == ' ' || *ptr == '\t') ++ptr; | |
51 char *id_ptr = ptr; | |
52 while (*ptr && *ptr != ' ' && *ptr != '\t') ++ptr; | |
53 while (*ptr == ' ' || *ptr == '\t') ++ptr; | |
54 while (*ptr && *ptr != ' ' && *ptr != '\t') ++ptr; | |
55 string id(id_ptr, ptr - id_ptr); | |
56 while (*ptr == ' ' || *ptr == '\t') ++ptr; | |
57 char *library_ptr = ptr; | |
58 while (*ptr && *ptr != ' ' && *ptr != '\t' && *ptr != '\n') ++ptr; | |
59 string library(library_ptr, ptr - library_ptr); | |
60 bool isVDSO = false; | |
61 if (library == "[vdso]") { | |
62 // /proc/self/maps has a misleading file offset in the [vdso] entry. | |
63 // Override it with a sane value. | |
64 offset = 0; | |
65 isVDSO = true; | |
66 } else if (library == "[vsyscall]") { | |
67 vsyscall_ = reinterpret_cast<char *>(start); | |
68 } else if (library.empty() || library[0] == '[') { | |
69 goto skip_entry; | |
70 } | |
71 int prot = 0; | |
72 if (perm.find('r') != string::npos) { | |
73 prot |= PROT_READ; | |
74 } | |
75 if (perm.find('w') != string::npos) { | |
76 prot |= PROT_WRITE; | |
77 } | |
78 if (perm.find('x') != string::npos) { | |
79 prot |= PROT_EXEC; | |
80 } | |
81 if ((prot & (PROT_EXEC | PROT_READ)) == 0) { | |
82 goto skip_entry; | |
83 } | |
84 Library* lib = &libs_[id + ' ' + library]; | |
85 lib->setLibraryInfo(this); | |
86 lib->addMemoryRange(reinterpret_cast<void *>(start), | |
87 reinterpret_cast<void *>(stop), | |
88 Elf_Addr(offset), | |
89 prot, isVDSO); | |
90 } | |
91 skip_entry: | |
92 for (;;) { | |
93 if (!*ptr || *ptr++ == '\n') { | |
94 long_line = false; | |
95 memmove(buf, ptr, len - (ptr - buf)); | |
96 memset(buf + len - (ptr - buf), 0, ptr - buf); | |
97 len -= (ptr - buf); | |
98 break; | |
99 } | |
100 } | |
101 } while (len || long_line); | |
102 } | |
103 } | |
104 | |
105 Maps::Iterator::Iterator(Maps* maps, bool at_beginning, bool at_end) | |
106 : maps_(maps), | |
107 at_beginning_(at_beginning), | |
108 at_end_(at_end) { | |
109 } | |
110 | |
111 Maps::LibraryMap::iterator& Maps::Iterator::getIterator() const { | |
112 if (at_beginning_) { | |
113 iter_ = maps_->libs_.begin(); | |
114 } else if (at_end_) { | |
115 iter_ = maps_->libs_.end(); | |
116 } | |
117 return iter_; | |
118 } | |
119 | |
120 Maps::Iterator Maps::Iterator::begin() { | |
121 return maps_->begin_iter_; | |
122 } | |
123 | |
124 Maps::Iterator Maps::Iterator::end() { | |
125 return maps_->end_iter_; | |
126 } | |
127 | |
128 Maps::Iterator& Maps::Iterator::operator++() { | |
129 getIterator().operator++(); | |
130 at_beginning_ = false; | |
131 return *this; | |
132 } | |
133 | |
134 Maps::Iterator Maps::Iterator::operator++(int i) { | |
135 getIterator().operator++(i); | |
136 at_beginning_ = false; | |
137 return *this; | |
138 } | |
139 | |
140 Library* Maps::Iterator::operator*() const { | |
141 return &getIterator().operator*().second; | |
142 } | |
143 | |
144 bool Maps::Iterator::operator==(const Maps::Iterator& iter) const { | |
145 return getIterator().operator==(iter.getIterator()); | |
146 } | |
147 | |
148 bool Maps::Iterator::operator!=(const Maps::Iterator& iter) const { | |
149 return !operator==(iter); | |
150 } | |
151 | |
152 Maps::string Maps::Iterator::name() const { | |
153 return getIterator()->first; | |
154 } | |
155 | |
156 // Test whether a line ends with "[stack]"; used for identifying the | |
157 // stack entry of /proc/self/maps. | |
158 static bool isStackLine(char* buf, char* end) { | |
159 char* ptr = buf; | |
160 for ( ; *ptr != '\n' && ptr < end; ++ptr) | |
161 ; | |
162 if (ptr < end && ptr - 7 > buf) { | |
163 return (memcmp(ptr - 7, "[stack]", 7) == 0); | |
164 } | |
165 return false; | |
166 } | |
167 | |
168 char* Maps::allocNearAddr(char* addr_target, size_t size, int prot) const { | |
169 // We try to allocate memory within 1.5GB of a target address. This means, | |
170 // we will be able to perform relative 32bit jumps from the target address. | |
171 const unsigned long kMaxDistance = 1536 << 20; | |
172 // In most of the code below, we just care about the numeric value of | |
173 // the address. | |
174 const long addr = reinterpret_cast<long>(addr_target); | |
175 size = (size + 4095) & ~4095; | |
176 Sandbox::SysCalls sys; | |
177 if (sys.lseek(proc_self_maps_, 0, SEEK_SET)) { | |
178 return NULL; | |
179 } | |
180 | |
181 // Iterate through lines of /proc/self/maps to consider each mapped | |
182 // region one at a time, looking for a gap between regions to allocate. | |
183 char buf[256] = { 0 }; | |
184 int len = 0, rc = 1; | |
185 bool long_line = false; | |
186 unsigned long gap_start = 0x10000; | |
187 void* new_addr; | |
188 do { | |
189 if (rc > 0) { | |
190 do { | |
191 rc = Sandbox::read(sys, proc_self_maps_, buf + len, | |
192 sizeof(buf) - len - 1); | |
193 if (rc > 0) { | |
194 len += rc; | |
195 } | |
196 } while (rc > 0 && len < (int)sizeof(buf) - 1); | |
197 } | |
198 char *ptr = buf; | |
199 if (!long_line) { | |
200 long_line = true; | |
201 // Maps lines have the form "<start address>-<end address> ... <name>". | |
202 unsigned long gap_end = strtoul(ptr, &ptr, 16); | |
203 unsigned long map_end = strtoul(ptr + 1, &ptr, 16); | |
204 | |
205 // gap_start to gap_end now covers the region of empty space before | |
206 // the current line. Now we try to see if there's a place within the | |
207 // gap we can use. | |
208 | |
209 if (gap_end - gap_start >= size) { | |
210 // Is the gap before our target address? | |
211 if (addr - static_cast<long>(gap_end) >= 0) { | |
212 if (addr - (gap_end - size) < kMaxDistance) { | |
213 unsigned long position; | |
214 if (isStackLine(ptr, buf + len)) { | |
215 // If we're adjacent to the stack, try to stay away from | |
216 // the GROWS_DOWN region. Pick the farthest away region that | |
217 // is still within the gap. | |
218 | |
219 if (static_cast<unsigned long>(addr) < kMaxDistance || // Underfl
ow protection. | |
220 static_cast<unsigned long>(addr) - kMaxDistance < gap_start) { | |
221 position = gap_start; | |
222 } else { | |
223 position = (addr - kMaxDistance) & ~4095; | |
224 if (position < gap_start) { | |
225 position = gap_start; | |
226 } | |
227 } | |
228 } else { | |
229 // Otherwise, take the end of the region. | |
230 position = gap_end - size; | |
231 } | |
232 new_addr = reinterpret_cast<char *>(sys.MMAP | |
233 (reinterpret_cast<void *>(position), size, prot, | |
234 MAP_PRIVATE|MAP_ANONYMOUS|MAP_FIXED, -1, 0)); | |
235 if (new_addr != MAP_FAILED) { | |
236 goto done; | |
237 } | |
238 } | |
239 } else if (gap_start + size - addr < kMaxDistance) { | |
240 // Gap is after the address. Above checks that we can wrap around | |
241 // through 0 to a space we'd use. | |
242 new_addr = reinterpret_cast<char *>(sys.MMAP | |
243 (reinterpret_cast<void *>(gap_start), size, prot, | |
244 MAP_PRIVATE|MAP_ANONYMOUS|MAP_FIXED, -1 ,0)); | |
245 if (new_addr != MAP_FAILED) { | |
246 goto done; | |
247 } | |
248 } | |
249 } | |
250 gap_start = map_end; | |
251 } | |
252 for (;;) { | |
253 if (!*ptr || *ptr++ == '\n') { | |
254 long_line = false; | |
255 memmove(buf, ptr, len - (ptr - buf)); | |
256 memset(buf + len - (ptr - buf), 0, ptr - buf); | |
257 len -= (ptr - buf); | |
258 break; | |
259 } | |
260 } | |
261 } while (len || long_line); | |
262 new_addr = NULL; | |
263 done: | |
264 return reinterpret_cast<char*>(new_addr); | |
265 } | |
266 | |
267 } // namespace | |
OLD | NEW |