Brian Carlstrom | 27ec961 | 2011-09-19 20:20:38 -0700 | [diff] [blame] | 1 | /* |
| 2 | * Copyright (C) 2008 The Android Open Source Project |
| 3 | * |
| 4 | * Licensed under the Apache License, Version 2.0 (the "License"); |
| 5 | * you may not use this file except in compliance with the License. |
| 6 | * You may obtain a copy of the License at |
| 7 | * |
| 8 | * http://www.apache.org/licenses/LICENSE-2.0 |
| 9 | * |
| 10 | * Unless required by applicable law or agreed to in writing, software |
| 11 | * distributed under the License is distributed on an "AS IS" BASIS, |
| 12 | * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. |
| 13 | * See the License for the specific language governing permissions and |
| 14 | * limitations under the License. |
| 15 | */ |
| 16 | |
| 17 | #include "mem_map.h" |
| 18 | |
Elliott Hughes | 6c9c06d | 2011-11-07 16:43:47 -0800 | [diff] [blame] | 19 | #include "ScopedFd.h" |
Elliott Hughes | ecd3a6f | 2012-06-06 18:16:37 -0700 | [diff] [blame] | 20 | #include "stringprintf.h" |
Elliott Hughes | 6c9c06d | 2011-11-07 16:43:47 -0800 | [diff] [blame] | 21 | #include "utils.h" |
| 22 | |
Elliott Hughes | ecd3a6f | 2012-06-06 18:16:37 -0700 | [diff] [blame] | 23 | #include <corkscrew/map_info.h> |
| 24 | |
Elliott Hughes | 6c9c06d | 2011-11-07 16:43:47 -0800 | [diff] [blame] | 25 | #define USE_ASHMEM 1 |
| 26 | |
| 27 | #ifdef USE_ASHMEM |
| 28 | #include <cutils/ashmem.h> |
| 29 | #endif |
| 30 | |
Brian Carlstrom | 27ec961 | 2011-09-19 20:20:38 -0700 | [diff] [blame] | 31 | namespace art { |
| 32 | |
Elliott Hughes | 96970cd | 2012-03-13 15:46:31 -0700 | [diff] [blame] | 33 | #if !defined(NDEBUG) |
| 34 | |
Elliott Hughes | ecd3a6f | 2012-06-06 18:16:37 -0700 | [diff] [blame] | 35 | static std::ostream& operator<<(std::ostream& os, map_info_t* rhs) { |
| 36 | for (map_info_t* m = rhs; m != NULL; m = m->next) { |
Elliott Hughes | 7bda158 | 2012-06-07 11:00:54 -0700 | [diff] [blame] | 37 | os << StringPrintf("0x%08x-0x%08x %c%c %s\n", |
| 38 | static_cast<uint32_t>(m->start), |
| 39 | static_cast<uint32_t>(m->end), |
Elliott Hughes | ecd3a6f | 2012-06-06 18:16:37 -0700 | [diff] [blame] | 40 | m->is_readable ? 'r' : '-', m->is_executable ? 'x' : '-', m->name); |
| 41 | } |
| 42 | return os; |
Brian Carlstrom | 27ec961 | 2011-09-19 20:20:38 -0700 | [diff] [blame] | 43 | } |
| 44 | |
Elliott Hughes | ecd3a6f | 2012-06-06 18:16:37 -0700 | [diff] [blame] | 45 | void CheckMapRequest(byte* addr, size_t byte_count) { |
Brian Carlstrom | 27ec961 | 2011-09-19 20:20:38 -0700 | [diff] [blame] | 46 | if (addr == NULL) { |
| 47 | return; |
| 48 | } |
Elliott Hughes | ecd3a6f | 2012-06-06 18:16:37 -0700 | [diff] [blame] | 49 | |
Elliott Hughes | 96970cd | 2012-03-13 15:46:31 -0700 | [diff] [blame] | 50 | uint32_t base = reinterpret_cast<size_t>(addr); |
Elliott Hughes | ecd3a6f | 2012-06-06 18:16:37 -0700 | [diff] [blame] | 51 | uint32_t limit = base + byte_count; |
Brian Carlstrom | 27ec961 | 2011-09-19 20:20:38 -0700 | [diff] [blame] | 52 | |
Elliott Hughes | ecd3a6f | 2012-06-06 18:16:37 -0700 | [diff] [blame] | 53 | map_info_t* map_info_list = load_map_info_list(getpid()); |
| 54 | for (map_info_t* m = map_info_list; m != NULL; m = m->next) { |
| 55 | CHECK(!(base >= m->start && base < m->end) // start of new within old |
| 56 | && !(limit > m->start && limit < m->end) // end of new within old |
| 57 | && !(base <= m->start && limit > m->end)) // start/end of new includes all of old |
Elliott Hughes | 7bda158 | 2012-06-07 11:00:54 -0700 | [diff] [blame] | 58 | << StringPrintf("Requested region 0x%08x-0x%08x overlaps with existing map 0x%08x-0x%08x (%s)\n", |
| 59 | base, limit, |
| 60 | static_cast<uint32_t>(m->start), static_cast<uint32_t>(m->end), m->name) |
Elliott Hughes | ecd3a6f | 2012-06-06 18:16:37 -0700 | [diff] [blame] | 61 | << map_info_list; |
Elliott Hughes | 96970cd | 2012-03-13 15:46:31 -0700 | [diff] [blame] | 62 | } |
Elliott Hughes | ecd3a6f | 2012-06-06 18:16:37 -0700 | [diff] [blame] | 63 | free_map_info_list(map_info_list); |
Brian Carlstrom | 27ec961 | 2011-09-19 20:20:38 -0700 | [diff] [blame] | 64 | } |
| 65 | |
Elliott Hughes | 96970cd | 2012-03-13 15:46:31 -0700 | [diff] [blame] | 66 | #else |
Elliott Hughes | 1bac54f | 2012-03-16 12:48:31 -0700 | [diff] [blame] | 67 | static void CheckMapRequest(byte*, size_t) { } |
Elliott Hughes | 96970cd | 2012-03-13 15:46:31 -0700 | [diff] [blame] | 68 | #endif |
| 69 | |
Elliott Hughes | ecd3a6f | 2012-06-06 18:16:37 -0700 | [diff] [blame] | 70 | MemMap* MemMap::MapAnonymous(const char* name, byte* addr, size_t byte_count, int prot) { |
| 71 | CHECK_NE(0U, byte_count); |
Brian Carlstrom | 27ec961 | 2011-09-19 20:20:38 -0700 | [diff] [blame] | 72 | CHECK_NE(0, prot); |
Elliott Hughes | ecd3a6f | 2012-06-06 18:16:37 -0700 | [diff] [blame] | 73 | size_t page_aligned_byte_count = RoundUp(byte_count, kPageSize); |
| 74 | CheckMapRequest(addr, page_aligned_byte_count); |
Elliott Hughes | 6c9c06d | 2011-11-07 16:43:47 -0800 | [diff] [blame] | 75 | |
| 76 | #ifdef USE_ASHMEM |
Elliott Hughes | ecd3a6f | 2012-06-06 18:16:37 -0700 | [diff] [blame] | 77 | ScopedFd fd(ashmem_create_region(name, page_aligned_byte_count)); |
Elliott Hughes | 6c9c06d | 2011-11-07 16:43:47 -0800 | [diff] [blame] | 78 | int flags = MAP_PRIVATE; |
| 79 | if (fd.get() == -1) { |
| 80 | PLOG(ERROR) << "ashmem_create_region failed (" << name << ")"; |
| 81 | return NULL; |
| 82 | } |
| 83 | #else |
| 84 | ScopedFd fd(-1); |
| 85 | int flags = MAP_PRIVATE | MAP_ANONYMOUS; |
| 86 | #endif |
| 87 | |
Elliott Hughes | ecd3a6f | 2012-06-06 18:16:37 -0700 | [diff] [blame] | 88 | byte* actual = reinterpret_cast<byte*>(mmap(addr, page_aligned_byte_count, prot, flags, fd.get(), 0)); |
Brian Carlstrom | 27ec961 | 2011-09-19 20:20:38 -0700 | [diff] [blame] | 89 | if (actual == MAP_FAILED) { |
Elliott Hughes | ecd3a6f | 2012-06-06 18:16:37 -0700 | [diff] [blame] | 90 | PLOG(ERROR) << "mmap(" << reinterpret_cast<void*>(addr) << ", " << page_aligned_byte_count |
Elliott Hughes | 5ea8d4b | 2012-05-30 15:21:36 -0700 | [diff] [blame] | 91 | << ", " << prot << ", " << flags << ", " << fd.get() << ", 0) failed for " << name; |
Brian Carlstrom | 27ec961 | 2011-09-19 20:20:38 -0700 | [diff] [blame] | 92 | return NULL; |
| 93 | } |
Mathieu Chartier | 1c23e1e | 2012-10-12 14:14:11 -0700 | [diff] [blame] | 94 | return new MemMap(name, actual, byte_count, actual, page_aligned_byte_count, prot); |
Brian Carlstrom | 27ec961 | 2011-09-19 20:20:38 -0700 | [diff] [blame] | 95 | } |
| 96 | |
Elliott Hughes | ecd3a6f | 2012-06-06 18:16:37 -0700 | [diff] [blame] | 97 | MemMap* MemMap::MapFileAtAddress(byte* addr, size_t byte_count, int prot, int flags, int fd, off_t start) { |
| 98 | CHECK_NE(0U, byte_count); |
Brian Carlstrom | 27ec961 | 2011-09-19 20:20:38 -0700 | [diff] [blame] | 99 | CHECK_NE(0, prot); |
| 100 | CHECK_NE(0, flags & (MAP_SHARED | MAP_PRIVATE)); |
Elliott Hughes | ecd3a6f | 2012-06-06 18:16:37 -0700 | [diff] [blame] | 101 | // Adjust 'offset' and 'byte_count' to be page-aligned. |
Brian Carlstrom | 27ec961 | 2011-09-19 20:20:38 -0700 | [diff] [blame] | 102 | int page_offset = start % kPageSize; |
| 103 | off_t page_aligned_offset = start - page_offset; |
Elliott Hughes | ecd3a6f | 2012-06-06 18:16:37 -0700 | [diff] [blame] | 104 | size_t page_aligned_byte_count = RoundUp(byte_count + page_offset, kPageSize); |
| 105 | CheckMapRequest(addr, page_aligned_byte_count); |
Brian Carlstrom | 27ec961 | 2011-09-19 20:20:38 -0700 | [diff] [blame] | 106 | byte* actual = reinterpret_cast<byte*>(mmap(addr, |
Elliott Hughes | ecd3a6f | 2012-06-06 18:16:37 -0700 | [diff] [blame] | 107 | page_aligned_byte_count, |
Brian Carlstrom | 27ec961 | 2011-09-19 20:20:38 -0700 | [diff] [blame] | 108 | prot, |
| 109 | flags, |
| 110 | fd, |
| 111 | page_aligned_offset)); |
| 112 | if (actual == MAP_FAILED) { |
| 113 | PLOG(ERROR) << "mmap failed"; |
| 114 | return NULL; |
| 115 | } |
Mathieu Chartier | 1c23e1e | 2012-10-12 14:14:11 -0700 | [diff] [blame] | 116 | return new MemMap("file", actual + page_offset, byte_count, actual, page_aligned_byte_count, |
| 117 | prot); |
Brian Carlstrom | 27ec961 | 2011-09-19 20:20:38 -0700 | [diff] [blame] | 118 | } |
| 119 | |
| 120 | MemMap::~MemMap() { |
Ian Rogers | 30fab40 | 2012-01-23 15:43:46 -0800 | [diff] [blame] | 121 | if (base_begin_ == NULL && base_size_ == 0) { |
Brian Carlstrom | 27ec961 | 2011-09-19 20:20:38 -0700 | [diff] [blame] | 122 | return; |
| 123 | } |
Ian Rogers | 30fab40 | 2012-01-23 15:43:46 -0800 | [diff] [blame] | 124 | int result = munmap(base_begin_, base_size_); |
Brian Carlstrom | 27ec961 | 2011-09-19 20:20:38 -0700 | [diff] [blame] | 125 | if (result == -1) { |
| 126 | PLOG(FATAL) << "munmap failed"; |
| 127 | } |
| 128 | } |
| 129 | |
Mathieu Chartier | 1c23e1e | 2012-10-12 14:14:11 -0700 | [diff] [blame] | 130 | MemMap::MemMap(const std::string& name, byte* begin, size_t size, void* base_begin, |
| 131 | size_t base_size, int prot) |
| 132 | : name_(name), begin_(begin), size_(size), base_begin_(base_begin), base_size_(base_size), |
| 133 | prot_(prot) { |
Ian Rogers | 30fab40 | 2012-01-23 15:43:46 -0800 | [diff] [blame] | 134 | CHECK(begin_ != NULL); |
| 135 | CHECK_NE(size_, 0U); |
| 136 | CHECK(base_begin_ != NULL); |
| 137 | CHECK_NE(base_size_, 0U); |
Brian Carlstrom | 27ec961 | 2011-09-19 20:20:38 -0700 | [diff] [blame] | 138 | }; |
| 139 | |
Mathieu Chartier | cc236d7 | 2012-07-20 10:29:05 -0700 | [diff] [blame] | 140 | void MemMap::UnMapAtEnd(byte* new_end) { |
| 141 | DCHECK_GE(new_end, Begin()); |
| 142 | DCHECK_LE(new_end, End()); |
| 143 | size_t unmap_size = End() - new_end; |
| 144 | munmap(new_end, unmap_size); |
| 145 | size_ -= unmap_size; |
| 146 | } |
Logan Chien | d88fa26 | 2012-06-06 15:23:32 +0800 | [diff] [blame] | 147 | |
| 148 | bool MemMap::Protect(int prot) { |
| 149 | if (base_begin_ == NULL && base_size_ == 0) { |
Ian Rogers | 1c849e5 | 2012-06-28 14:00:33 -0700 | [diff] [blame] | 150 | prot_ = prot; |
Logan Chien | d88fa26 | 2012-06-06 15:23:32 +0800 | [diff] [blame] | 151 | return true; |
| 152 | } |
| 153 | |
| 154 | if (mprotect(base_begin_, base_size_, prot) == 0) { |
Ian Rogers | 1c849e5 | 2012-06-28 14:00:33 -0700 | [diff] [blame] | 155 | prot_ = prot; |
Logan Chien | d88fa26 | 2012-06-06 15:23:32 +0800 | [diff] [blame] | 156 | return true; |
| 157 | } |
| 158 | |
Shih-wei Liao | a060ed9 | 2012-06-07 09:25:28 -0700 | [diff] [blame] | 159 | PLOG(ERROR) << "mprotect(" << reinterpret_cast<void*>(base_begin_) << ", " << base_size_ << ", " |
| 160 | << prot << ") failed"; |
Logan Chien | d88fa26 | 2012-06-06 15:23:32 +0800 | [diff] [blame] | 161 | return false; |
| 162 | } |
| 163 | |
Brian Carlstrom | 27ec961 | 2011-09-19 20:20:38 -0700 | [diff] [blame] | 164 | } // namespace art |