Elliott Hughes | 2faa5f1 | 2012-01-30 14:42:07 -0800 | [diff] [blame] | 1 | /* |
| 2 | * Copyright (C) 2011 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 | */ |
Brian Carlstrom | e24fa61 | 2011-09-29 00:53:55 -0700 | [diff] [blame] | 16 | |
| 17 | #include "oat_file.h" |
| 18 | |
Brian Carlstrom | 700c8d3 | 2012-11-05 10:42:02 -0800 | [diff] [blame] | 19 | #include <dlfcn.h> |
Calin Juravle | 4e1d579 | 2014-07-15 23:56:47 +0100 | [diff] [blame] | 20 | #include <string.h> |
Andreas Gampe | daab38c | 2014-09-12 18:38:24 -0700 | [diff] [blame] | 21 | #include <unistd.h> |
Brian Carlstrom | 700c8d3 | 2012-11-05 10:42:02 -0800 | [diff] [blame] | 22 | |
Andreas Gampe | 7848da4 | 2015-04-09 11:15:04 -0700 | [diff] [blame^] | 23 | #include <cstdlib> |
Richard Uhler | e5fed03 | 2015-03-18 08:21:11 -0700 | [diff] [blame] | 24 | #include <sstream> |
| 25 | |
Brian Carlstrom | ba150c3 | 2013-08-27 17:31:03 -0700 | [diff] [blame] | 26 | #include "base/bit_vector.h" |
Elliott Hughes | 1aa246d | 2012-12-13 09:29:36 -0800 | [diff] [blame] | 27 | #include "base/stl_util.h" |
Elliott Hughes | 7616005 | 2012-12-12 16:31:20 -0800 | [diff] [blame] | 28 | #include "base/unix_file/fd_file.h" |
Brian Carlstrom | 700c8d3 | 2012-11-05 10:42:02 -0800 | [diff] [blame] | 29 | #include "elf_file.h" |
Alex Light | 84d7605 | 2014-08-22 17:49:35 -0700 | [diff] [blame] | 30 | #include "elf_utils.h" |
Brian Carlstrom | 700c8d3 | 2012-11-05 10:42:02 -0800 | [diff] [blame] | 31 | #include "oat.h" |
Brian Carlstrom | ea46f95 | 2013-07-30 01:26:50 -0700 | [diff] [blame] | 32 | #include "mirror/art_method.h" |
| 33 | #include "mirror/art_method-inl.h" |
Ian Rogers | 2dd0e2c | 2013-01-24 12:42:14 -0800 | [diff] [blame] | 34 | #include "mirror/class.h" |
Ian Rogers | 4f6ad8a | 2013-03-18 15:27:28 -0700 | [diff] [blame] | 35 | #include "mirror/object-inl.h" |
Brian Carlstrom | e24fa61 | 2011-09-29 00:53:55 -0700 | [diff] [blame] | 36 | #include "os.h" |
Andreas Gampe | 22f8e5c | 2014-07-09 11:38:21 -0700 | [diff] [blame] | 37 | #include "runtime.h" |
Ian Rogers | 2dd0e2c | 2013-01-24 12:42:14 -0800 | [diff] [blame] | 38 | #include "utils.h" |
Ian Rogers | 1809a72 | 2013-08-09 22:05:32 -0700 | [diff] [blame] | 39 | #include "vmap_table.h" |
Brian Carlstrom | e24fa61 | 2011-09-29 00:53:55 -0700 | [diff] [blame] | 40 | |
| 41 | namespace art { |
| 42 | |
Richard Uhler | e5fed03 | 2015-03-18 08:21:11 -0700 | [diff] [blame] | 43 | std::string OatFile::ResolveRelativeEncodedDexLocation( |
| 44 | const char* abs_dex_location, const std::string& rel_dex_location) { |
| 45 | if (abs_dex_location != nullptr && rel_dex_location[0] != '/') { |
| 46 | // Strip :classes<N>.dex used for secondary multidex files. |
| 47 | std::string base = DexFile::GetBaseLocation(rel_dex_location); |
| 48 | std::string multidex_suffix = DexFile::GetMultiDexSuffix(rel_dex_location); |
| 49 | |
| 50 | // Check if the base is a suffix of the provided abs_dex_location. |
| 51 | std::string target_suffix = "/" + base; |
| 52 | std::string abs_location(abs_dex_location); |
| 53 | if (abs_location.size() > target_suffix.size()) { |
| 54 | size_t pos = abs_location.size() - target_suffix.size(); |
| 55 | if (abs_location.compare(pos, std::string::npos, target_suffix) == 0) { |
| 56 | return abs_location + multidex_suffix; |
| 57 | } |
| 58 | } |
| 59 | } |
| 60 | return rel_dex_location; |
| 61 | } |
| 62 | |
Brian Carlstrom | 700c8d3 | 2012-11-05 10:42:02 -0800 | [diff] [blame] | 63 | void OatFile::CheckLocation(const std::string& location) { |
Brian Carlstrom | 7a967b3 | 2012-03-28 15:23:10 -0700 | [diff] [blame] | 64 | CHECK(!location.empty()); |
Brian Carlstrom | 700c8d3 | 2012-11-05 10:42:02 -0800 | [diff] [blame] | 65 | } |
| 66 | |
Alex Light | 84d7605 | 2014-08-22 17:49:35 -0700 | [diff] [blame] | 67 | OatFile* OatFile::OpenWithElfFile(ElfFile* elf_file, |
| 68 | const std::string& location, |
Richard Uhler | e5fed03 | 2015-03-18 08:21:11 -0700 | [diff] [blame] | 69 | const char* abs_dex_location, |
Alex Light | 84d7605 | 2014-08-22 17:49:35 -0700 | [diff] [blame] | 70 | std::string* error_msg) { |
| 71 | std::unique_ptr<OatFile> oat_file(new OatFile(location, false)); |
| 72 | oat_file->elf_file_.reset(elf_file); |
Tong Shen | 62d1ca3 | 2014-09-03 17:24:56 -0700 | [diff] [blame] | 73 | uint64_t offset, size; |
| 74 | bool has_section = elf_file->GetSectionOffsetAndSize(".rodata", &offset, &size); |
| 75 | CHECK(has_section); |
| 76 | oat_file->begin_ = elf_file->Begin() + offset; |
| 77 | oat_file->end_ = elf_file->Begin() + size + offset; |
Vladimir Marko | 5c42c29 | 2015-02-25 12:02:49 +0000 | [diff] [blame] | 78 | // Ignore the optional .bss section when opening non-executable. |
Richard Uhler | e5fed03 | 2015-03-18 08:21:11 -0700 | [diff] [blame] | 79 | return oat_file->Setup(abs_dex_location, error_msg) ? oat_file.release() : nullptr; |
Brian Carlstrom | 700c8d3 | 2012-11-05 10:42:02 -0800 | [diff] [blame] | 80 | } |
| 81 | |
| 82 | OatFile* OatFile::Open(const std::string& filename, |
| 83 | const std::string& location, |
Ian Rogers | 1373595 | 2014-10-08 12:43:28 -0700 | [diff] [blame] | 84 | uint8_t* requested_base, |
Igor Murashkin | 4677476 | 2014-10-22 11:37:02 -0700 | [diff] [blame] | 85 | uint8_t* oat_file_begin, |
Ian Rogers | 8d31bbd | 2013-10-13 10:44:14 -0700 | [diff] [blame] | 86 | bool executable, |
Richard Uhler | e5fed03 | 2015-03-18 08:21:11 -0700 | [diff] [blame] | 87 | const char* abs_dex_location, |
Ian Rogers | 8d31bbd | 2013-10-13 10:44:14 -0700 | [diff] [blame] | 88 | std::string* error_msg) { |
Brian Carlstrom | 700c8d3 | 2012-11-05 10:42:02 -0800 | [diff] [blame] | 89 | CHECK(!filename.empty()) << location; |
Andreas Gampe | 7ba6496 | 2014-10-23 11:37:40 -0700 | [diff] [blame] | 90 | CheckLocation(location); |
Andreas Gampe | 22f8e5c | 2014-07-09 11:38:21 -0700 | [diff] [blame] | 91 | std::unique_ptr<OatFile> ret; |
Elliott Hughes | 956af0f | 2014-12-11 14:34:28 -0800 | [diff] [blame] | 92 | // If we aren't trying to execute, we just use our own ElfFile loader for a couple reasons: |
| 93 | // |
| 94 | // On target, dlopen may fail when compiling due to selinux restrictions on installd. |
| 95 | // |
| 96 | // We use our own ELF loader for Quick to deal with legacy apps that |
| 97 | // open a generated dex file by name, remove the file, then open |
| 98 | // another generated dex file with the same name. http://b/10614658 |
| 99 | // |
| 100 | // On host, dlopen is expected to fail when cross compiling, so fall back to OpenElfFile. |
| 101 | std::unique_ptr<File> file(OS::OpenFileForReading(filename.c_str())); |
| 102 | if (file.get() == NULL) { |
| 103 | *error_msg = StringPrintf("Failed to open oat filename for reading: %s", strerror(errno)); |
| 104 | return nullptr; |
Brian Carlstrom | 700c8d3 | 2012-11-05 10:42:02 -0800 | [diff] [blame] | 105 | } |
Elliott Hughes | 956af0f | 2014-12-11 14:34:28 -0800 | [diff] [blame] | 106 | ret.reset(OpenElfFile(file.get(), location, requested_base, oat_file_begin, false, executable, |
Richard Uhler | e5fed03 | 2015-03-18 08:21:11 -0700 | [diff] [blame] | 107 | abs_dex_location, error_msg)); |
Elliott Hughes | 956af0f | 2014-12-11 14:34:28 -0800 | [diff] [blame] | 108 | |
| 109 | // It would be nice to unlink here. But we might have opened the file created by the |
| 110 | // ScopedLock, which we better not delete to avoid races. TODO: Investigate how to fix the API |
| 111 | // to allow removal when we know the ELF must be borked. |
Dave Allison | 69dfe51 | 2014-07-11 17:11:58 +0000 | [diff] [blame] | 112 | return ret.release(); |
Brian Carlstrom | 700c8d3 | 2012-11-05 10:42:02 -0800 | [diff] [blame] | 113 | } |
| 114 | |
Richard Uhler | e5fed03 | 2015-03-18 08:21:11 -0700 | [diff] [blame] | 115 | OatFile* OatFile::OpenWritable(File* file, const std::string& location, |
| 116 | const char* abs_dex_location, |
| 117 | std::string* error_msg) { |
Brian Carlstrom | 700c8d3 | 2012-11-05 10:42:02 -0800 | [diff] [blame] | 118 | CheckLocation(location); |
Richard Uhler | e5fed03 | 2015-03-18 08:21:11 -0700 | [diff] [blame] | 119 | return OpenElfFile(file, location, nullptr, nullptr, true, false, abs_dex_location, error_msg); |
Brian Carlstrom | 700c8d3 | 2012-11-05 10:42:02 -0800 | [diff] [blame] | 120 | } |
| 121 | |
Richard Uhler | e5fed03 | 2015-03-18 08:21:11 -0700 | [diff] [blame] | 122 | OatFile* OatFile::OpenReadable(File* file, const std::string& location, |
| 123 | const char* abs_dex_location, |
| 124 | std::string* error_msg) { |
Alex Light | a59dd80 | 2014-07-02 16:28:08 -0700 | [diff] [blame] | 125 | CheckLocation(location); |
Richard Uhler | e5fed03 | 2015-03-18 08:21:11 -0700 | [diff] [blame] | 126 | return OpenElfFile(file, location, nullptr, nullptr, false, false, abs_dex_location, error_msg); |
Alex Light | a59dd80 | 2014-07-02 16:28:08 -0700 | [diff] [blame] | 127 | } |
| 128 | |
Brian Carlstrom | 700c8d3 | 2012-11-05 10:42:02 -0800 | [diff] [blame] | 129 | OatFile* OatFile::OpenElfFile(File* file, |
| 130 | const std::string& location, |
Ian Rogers | 1373595 | 2014-10-08 12:43:28 -0700 | [diff] [blame] | 131 | uint8_t* requested_base, |
Igor Murashkin | 4677476 | 2014-10-22 11:37:02 -0700 | [diff] [blame] | 132 | uint8_t* oat_file_begin, |
Brian Carlstrom | f1d3455 | 2013-07-12 20:22:23 -0700 | [diff] [blame] | 133 | bool writable, |
Ian Rogers | 8d31bbd | 2013-10-13 10:44:14 -0700 | [diff] [blame] | 134 | bool executable, |
Richard Uhler | e5fed03 | 2015-03-18 08:21:11 -0700 | [diff] [blame] | 135 | const char* abs_dex_location, |
Ian Rogers | 8d31bbd | 2013-10-13 10:44:14 -0700 | [diff] [blame] | 136 | std::string* error_msg) { |
Alex Light | 9dcc457 | 2014-08-14 14:16:26 -0700 | [diff] [blame] | 137 | std::unique_ptr<OatFile> oat_file(new OatFile(location, executable)); |
Igor Murashkin | 4677476 | 2014-10-22 11:37:02 -0700 | [diff] [blame] | 138 | bool success = oat_file->ElfFileOpen(file, requested_base, oat_file_begin, writable, executable, |
Richard Uhler | e5fed03 | 2015-03-18 08:21:11 -0700 | [diff] [blame] | 139 | abs_dex_location, error_msg); |
Brian Carlstrom | e24fa61 | 2011-09-29 00:53:55 -0700 | [diff] [blame] | 140 | if (!success) { |
Ian Rogers | 8d31bbd | 2013-10-13 10:44:14 -0700 | [diff] [blame] | 141 | CHECK(!error_msg->empty()); |
| 142 | return nullptr; |
Brian Carlstrom | e24fa61 | 2011-09-29 00:53:55 -0700 | [diff] [blame] | 143 | } |
| 144 | return oat_file.release(); |
| 145 | } |
| 146 | |
Alex Light | 9dcc457 | 2014-08-14 14:16:26 -0700 | [diff] [blame] | 147 | OatFile::OatFile(const std::string& location, bool is_executable) |
Vladimir Marko | 5c42c29 | 2015-02-25 12:02:49 +0000 | [diff] [blame] | 148 | : location_(location), begin_(NULL), end_(NULL), bss_begin_(nullptr), bss_end_(nullptr), |
| 149 | is_executable_(is_executable), dlopen_handle_(NULL), |
Vladimir Marko | 3f5838d | 2014-08-07 18:07:18 +0100 | [diff] [blame] | 150 | secondary_lookup_lock_("OatFile secondary lookup lock", kOatFileSecondaryLookupLock) { |
Brian Carlstrom | a004aa9 | 2012-02-08 18:05:09 -0800 | [diff] [blame] | 151 | CHECK(!location_.empty()); |
| 152 | } |
Brian Carlstrom | e24fa61 | 2011-09-29 00:53:55 -0700 | [diff] [blame] | 153 | |
| 154 | OatFile::~OatFile() { |
Vladimir Marko | aa4497d | 2014-09-05 14:01:17 +0100 | [diff] [blame] | 155 | STLDeleteElements(&oat_dex_files_storage_); |
Brian Carlstrom | 700c8d3 | 2012-11-05 10:42:02 -0800 | [diff] [blame] | 156 | if (dlopen_handle_ != NULL) { |
| 157 | dlclose(dlopen_handle_); |
| 158 | } |
Brian Carlstrom | e24fa61 | 2011-09-29 00:53:55 -0700 | [diff] [blame] | 159 | } |
| 160 | |
Igor Murashkin | 4677476 | 2014-10-22 11:37:02 -0700 | [diff] [blame] | 161 | bool OatFile::ElfFileOpen(File* file, uint8_t* requested_base, uint8_t* oat_file_begin, |
| 162 | bool writable, bool executable, |
Richard Uhler | e5fed03 | 2015-03-18 08:21:11 -0700 | [diff] [blame] | 163 | const char* abs_dex_location, |
Ian Rogers | 8d31bbd | 2013-10-13 10:44:14 -0700 | [diff] [blame] | 164 | std::string* error_msg) { |
Igor Murashkin | 4677476 | 2014-10-22 11:37:02 -0700 | [diff] [blame] | 165 | // TODO: rename requested_base to oat_data_begin |
| 166 | elf_file_.reset(ElfFile::Open(file, writable, /*program_header_only*/true, error_msg, |
| 167 | oat_file_begin)); |
Ian Rogers | 8d31bbd | 2013-10-13 10:44:14 -0700 | [diff] [blame] | 168 | if (elf_file_.get() == nullptr) { |
| 169 | DCHECK(!error_msg->empty()); |
Brian Carlstrom | 700c8d3 | 2012-11-05 10:42:02 -0800 | [diff] [blame] | 170 | return false; |
| 171 | } |
Ian Rogers | 8d31bbd | 2013-10-13 10:44:14 -0700 | [diff] [blame] | 172 | bool loaded = elf_file_->Load(executable, error_msg); |
Brian Carlstrom | 700c8d3 | 2012-11-05 10:42:02 -0800 | [diff] [blame] | 173 | if (!loaded) { |
Ian Rogers | 8d31bbd | 2013-10-13 10:44:14 -0700 | [diff] [blame] | 174 | DCHECK(!error_msg->empty()); |
Brian Carlstrom | 700c8d3 | 2012-11-05 10:42:02 -0800 | [diff] [blame] | 175 | return false; |
| 176 | } |
| 177 | begin_ = elf_file_->FindDynamicSymbolAddress("oatdata"); |
| 178 | if (begin_ == NULL) { |
Ian Rogers | 8d31bbd | 2013-10-13 10:44:14 -0700 | [diff] [blame] | 179 | *error_msg = StringPrintf("Failed to find oatdata symbol in '%s'", file->GetPath().c_str()); |
Brian Carlstrom | 700c8d3 | 2012-11-05 10:42:02 -0800 | [diff] [blame] | 180 | return false; |
| 181 | } |
| 182 | if (requested_base != NULL && begin_ != requested_base) { |
Andreas Gampe | a6dfdae | 2015-02-24 15:50:19 -0800 | [diff] [blame] | 183 | PrintFileToLog("/proc/self/maps", LogSeverity::WARNING); |
Ian Rogers | 8d31bbd | 2013-10-13 10:44:14 -0700 | [diff] [blame] | 184 | *error_msg = StringPrintf("Failed to find oatdata symbol at expected address: " |
Andreas Gampe | a6dfdae | 2015-02-24 15:50:19 -0800 | [diff] [blame] | 185 | "oatdata=%p != expected=%p. See process maps in the log.", |
Ian Rogers | 8d31bbd | 2013-10-13 10:44:14 -0700 | [diff] [blame] | 186 | begin_, requested_base); |
Brian Carlstrom | 700c8d3 | 2012-11-05 10:42:02 -0800 | [diff] [blame] | 187 | return false; |
| 188 | } |
| 189 | end_ = elf_file_->FindDynamicSymbolAddress("oatlastword"); |
| 190 | if (end_ == NULL) { |
Ian Rogers | 8d31bbd | 2013-10-13 10:44:14 -0700 | [diff] [blame] | 191 | *error_msg = StringPrintf("Failed to find oatlastword symbol in '%s'", file->GetPath().c_str()); |
Brian Carlstrom | 700c8d3 | 2012-11-05 10:42:02 -0800 | [diff] [blame] | 192 | return false; |
| 193 | } |
| 194 | // Readjust to be non-inclusive upper bound. |
| 195 | end_ += sizeof(uint32_t); |
Vladimir Marko | 5c42c29 | 2015-02-25 12:02:49 +0000 | [diff] [blame] | 196 | |
| 197 | bss_begin_ = elf_file_->FindDynamicSymbolAddress("oatbss"); |
| 198 | if (bss_begin_ == nullptr) { |
| 199 | // No .bss section. Clear dlerror(). |
| 200 | bss_end_ = nullptr; |
| 201 | dlerror(); |
| 202 | } else { |
| 203 | bss_end_ = elf_file_->FindDynamicSymbolAddress("oatbsslastword"); |
| 204 | if (bss_end_ == nullptr) { |
| 205 | *error_msg = StringPrintf("Failed to find oatbasslastword symbol in '%s'", |
| 206 | file->GetPath().c_str()); |
| 207 | return false; |
| 208 | } |
| 209 | // Readjust to be non-inclusive upper bound. |
| 210 | bss_end_ += sizeof(uint32_t); |
| 211 | } |
| 212 | |
Richard Uhler | e5fed03 | 2015-03-18 08:21:11 -0700 | [diff] [blame] | 213 | return Setup(abs_dex_location, error_msg); |
Brian Carlstrom | 700c8d3 | 2012-11-05 10:42:02 -0800 | [diff] [blame] | 214 | } |
Brian Carlstrom | 6e3b1d9 | 2012-01-11 01:36:32 -0800 | [diff] [blame] | 215 | |
Richard Uhler | e5fed03 | 2015-03-18 08:21:11 -0700 | [diff] [blame] | 216 | bool OatFile::Setup(const char* abs_dex_location, std::string* error_msg) { |
Brian Carlstrom | f1b3030 | 2013-03-28 10:35:32 -0700 | [diff] [blame] | 217 | if (!GetOatHeader().IsValid()) { |
Andreas Gampe | 2bcb3b2 | 2014-12-12 15:25:14 -0800 | [diff] [blame] | 218 | std::string cause = GetOatHeader().GetValidationErrorMessage(); |
| 219 | *error_msg = StringPrintf("Invalid oat header for '%s': %s", GetLocation().c_str(), |
| 220 | cause.c_str()); |
Brian Carlstrom | f1b3030 | 2013-03-28 10:35:32 -0700 | [diff] [blame] | 221 | return false; |
| 222 | } |
Ian Rogers | 1373595 | 2014-10-08 12:43:28 -0700 | [diff] [blame] | 223 | const uint8_t* oat = Begin(); |
Brian Carlstrom | e24fa61 | 2011-09-29 00:53:55 -0700 | [diff] [blame] | 224 | oat += sizeof(OatHeader); |
Brian Carlstrom | fb331d7 | 2013-07-25 22:00:16 -0700 | [diff] [blame] | 225 | if (oat > End()) { |
Ian Rogers | 8d31bbd | 2013-10-13 10:44:14 -0700 | [diff] [blame] | 226 | *error_msg = StringPrintf("In oat file '%s' found truncated OatHeader", GetLocation().c_str()); |
Brian Carlstrom | fb331d7 | 2013-07-25 22:00:16 -0700 | [diff] [blame] | 227 | return false; |
| 228 | } |
Brian Carlstrom | 81f3ca1 | 2012-03-17 00:27:35 -0700 | [diff] [blame] | 229 | |
Andreas Gampe | 22f8e5c | 2014-07-09 11:38:21 -0700 | [diff] [blame] | 230 | oat += GetOatHeader().GetKeyValueStoreSize(); |
Brian Carlstrom | fb331d7 | 2013-07-25 22:00:16 -0700 | [diff] [blame] | 231 | if (oat > End()) { |
Andreas Gampe | 22f8e5c | 2014-07-09 11:38:21 -0700 | [diff] [blame] | 232 | *error_msg = StringPrintf("In oat file '%s' found truncated variable-size data: " |
Ian Rogers | 8d31bbd | 2013-10-13 10:44:14 -0700 | [diff] [blame] | 233 | "%p + %zd + %ud <= %p", GetLocation().c_str(), |
Andreas Gampe | 22f8e5c | 2014-07-09 11:38:21 -0700 | [diff] [blame] | 234 | Begin(), sizeof(OatHeader), GetOatHeader().GetKeyValueStoreSize(), |
Ian Rogers | 8d31bbd | 2013-10-13 10:44:14 -0700 | [diff] [blame] | 235 | End()); |
Brian Carlstrom | fb331d7 | 2013-07-25 22:00:16 -0700 | [diff] [blame] | 236 | return false; |
| 237 | } |
| 238 | |
Vladimir Marko | aa4497d | 2014-09-05 14:01:17 +0100 | [diff] [blame] | 239 | uint32_t dex_file_count = GetOatHeader().GetDexFileCount(); |
| 240 | oat_dex_files_storage_.reserve(dex_file_count); |
| 241 | for (size_t i = 0; i < dex_file_count; i++) { |
Dmitry Petrochenko | 83bef92 | 2014-02-03 12:34:59 +0700 | [diff] [blame] | 242 | uint32_t dex_file_location_size = *reinterpret_cast<const uint32_t*>(oat); |
Ian Rogers | 8d31bbd | 2013-10-13 10:44:14 -0700 | [diff] [blame] | 243 | if (UNLIKELY(dex_file_location_size == 0U)) { |
| 244 | *error_msg = StringPrintf("In oat file '%s' found OatDexFile #%zd with empty location name", |
| 245 | GetLocation().c_str(), i); |
Brian Carlstrom | fb331d7 | 2013-07-25 22:00:16 -0700 | [diff] [blame] | 246 | return false; |
| 247 | } |
Brian Carlstrom | e24fa61 | 2011-09-29 00:53:55 -0700 | [diff] [blame] | 248 | oat += sizeof(dex_file_location_size); |
Ian Rogers | 8d31bbd | 2013-10-13 10:44:14 -0700 | [diff] [blame] | 249 | if (UNLIKELY(oat > End())) { |
| 250 | *error_msg = StringPrintf("In oat file '%s' found OatDexFile #%zd truncated after dex file " |
| 251 | "location size", GetLocation().c_str(), i); |
Brian Carlstrom | fb331d7 | 2013-07-25 22:00:16 -0700 | [diff] [blame] | 252 | return false; |
| 253 | } |
Brian Carlstrom | e24fa61 | 2011-09-29 00:53:55 -0700 | [diff] [blame] | 254 | |
| 255 | const char* dex_file_location_data = reinterpret_cast<const char*>(oat); |
| 256 | oat += dex_file_location_size; |
Ian Rogers | 8d31bbd | 2013-10-13 10:44:14 -0700 | [diff] [blame] | 257 | if (UNLIKELY(oat > End())) { |
| 258 | *error_msg = StringPrintf("In oat file '%s' found OatDexFile #%zd with truncated dex file " |
| 259 | "location", GetLocation().c_str(), i); |
Brian Carlstrom | fb331d7 | 2013-07-25 22:00:16 -0700 | [diff] [blame] | 260 | return false; |
| 261 | } |
Brian Carlstrom | e24fa61 | 2011-09-29 00:53:55 -0700 | [diff] [blame] | 262 | |
Richard Uhler | e5fed03 | 2015-03-18 08:21:11 -0700 | [diff] [blame] | 263 | std::string dex_file_location = ResolveRelativeEncodedDexLocation( |
| 264 | abs_dex_location, |
| 265 | std::string(dex_file_location_data, dex_file_location_size)); |
Brian Carlstrom | e24fa61 | 2011-09-29 00:53:55 -0700 | [diff] [blame] | 266 | |
| 267 | uint32_t dex_file_checksum = *reinterpret_cast<const uint32_t*>(oat); |
| 268 | oat += sizeof(dex_file_checksum); |
Ian Rogers | 8d31bbd | 2013-10-13 10:44:14 -0700 | [diff] [blame] | 269 | if (UNLIKELY(oat > End())) { |
| 270 | *error_msg = StringPrintf("In oat file '%s' found OatDexFile #%zd for '%s' truncated after " |
| 271 | "dex file checksum", GetLocation().c_str(), i, |
| 272 | dex_file_location.c_str()); |
Brian Carlstrom | fb331d7 | 2013-07-25 22:00:16 -0700 | [diff] [blame] | 273 | return false; |
| 274 | } |
Brian Carlstrom | e24fa61 | 2011-09-29 00:53:55 -0700 | [diff] [blame] | 275 | |
Brian Carlstrom | 8952189 | 2011-12-07 22:05:07 -0800 | [diff] [blame] | 276 | uint32_t dex_file_offset = *reinterpret_cast<const uint32_t*>(oat); |
Ian Rogers | 8d31bbd | 2013-10-13 10:44:14 -0700 | [diff] [blame] | 277 | if (UNLIKELY(dex_file_offset == 0U)) { |
| 278 | *error_msg = StringPrintf("In oat file '%s' found OatDexFile #%zd for '%s' with zero dex " |
| 279 | "file offset", GetLocation().c_str(), i, dex_file_location.c_str()); |
Brian Carlstrom | fb331d7 | 2013-07-25 22:00:16 -0700 | [diff] [blame] | 280 | return false; |
| 281 | } |
Ian Rogers | 8d31bbd | 2013-10-13 10:44:14 -0700 | [diff] [blame] | 282 | if (UNLIKELY(dex_file_offset > Size())) { |
| 283 | *error_msg = StringPrintf("In oat file '%s' found OatDexFile #%zd for '%s' with dex file " |
| 284 | "offset %ud > %zd", GetLocation().c_str(), i, |
| 285 | dex_file_location.c_str(), dex_file_offset, Size()); |
Brian Carlstrom | fb331d7 | 2013-07-25 22:00:16 -0700 | [diff] [blame] | 286 | return false; |
| 287 | } |
Brian Carlstrom | 8952189 | 2011-12-07 22:05:07 -0800 | [diff] [blame] | 288 | oat += sizeof(dex_file_offset); |
Ian Rogers | 8d31bbd | 2013-10-13 10:44:14 -0700 | [diff] [blame] | 289 | if (UNLIKELY(oat > End())) { |
| 290 | *error_msg = StringPrintf("In oat file '%s' found OatDexFile #%zd for '%s' truncated " |
Brian Carlstrom | 37628b7 | 2014-10-28 13:54:26 -0700 | [diff] [blame] | 291 | "after dex file offsets", GetLocation().c_str(), i, |
Ian Rogers | 8d31bbd | 2013-10-13 10:44:14 -0700 | [diff] [blame] | 292 | dex_file_location.c_str()); |
Brian Carlstrom | fb331d7 | 2013-07-25 22:00:16 -0700 | [diff] [blame] | 293 | return false; |
| 294 | } |
Brian Carlstrom | 8952189 | 2011-12-07 22:05:07 -0800 | [diff] [blame] | 295 | |
Brian Carlstrom | 700c8d3 | 2012-11-05 10:42:02 -0800 | [diff] [blame] | 296 | const uint8_t* dex_file_pointer = Begin() + dex_file_offset; |
Ian Rogers | 8d31bbd | 2013-10-13 10:44:14 -0700 | [diff] [blame] | 297 | if (UNLIKELY(!DexFile::IsMagicValid(dex_file_pointer))) { |
| 298 | *error_msg = StringPrintf("In oat file '%s' found OatDexFile #%zd for '%s' with invalid " |
Brian Carlstrom | 37628b7 | 2014-10-28 13:54:26 -0700 | [diff] [blame] | 299 | "dex file magic '%s'", GetLocation().c_str(), i, |
Ian Rogers | 8d31bbd | 2013-10-13 10:44:14 -0700 | [diff] [blame] | 300 | dex_file_location.c_str(), dex_file_pointer); |
Brian Carlstrom | fb331d7 | 2013-07-25 22:00:16 -0700 | [diff] [blame] | 301 | return false; |
| 302 | } |
Ian Rogers | 8d31bbd | 2013-10-13 10:44:14 -0700 | [diff] [blame] | 303 | if (UNLIKELY(!DexFile::IsVersionValid(dex_file_pointer))) { |
| 304 | *error_msg = StringPrintf("In oat file '%s' found OatDexFile #%zd for '%s' with invalid " |
Brian Carlstrom | 37628b7 | 2014-10-28 13:54:26 -0700 | [diff] [blame] | 305 | "dex file version '%s'", GetLocation().c_str(), i, |
Ian Rogers | 8d31bbd | 2013-10-13 10:44:14 -0700 | [diff] [blame] | 306 | dex_file_location.c_str(), dex_file_pointer); |
Brian Carlstrom | fb331d7 | 2013-07-25 22:00:16 -0700 | [diff] [blame] | 307 | return false; |
| 308 | } |
Brian Carlstrom | 6e3b1d9 | 2012-01-11 01:36:32 -0800 | [diff] [blame] | 309 | const DexFile::Header* header = reinterpret_cast<const DexFile::Header*>(dex_file_pointer); |
| 310 | const uint32_t* methods_offsets_pointer = reinterpret_cast<const uint32_t*>(oat); |
Brian Carlstrom | 8952189 | 2011-12-07 22:05:07 -0800 | [diff] [blame] | 311 | |
Brian Carlstrom | 6e3b1d9 | 2012-01-11 01:36:32 -0800 | [diff] [blame] | 312 | oat += (sizeof(*methods_offsets_pointer) * header->class_defs_size_); |
Ian Rogers | 8d31bbd | 2013-10-13 10:44:14 -0700 | [diff] [blame] | 313 | if (UNLIKELY(oat > End())) { |
| 314 | *error_msg = StringPrintf("In oat file '%s' found OatDexFile #%zd for '%s' with truncated " |
Brian Carlstrom | 37628b7 | 2014-10-28 13:54:26 -0700 | [diff] [blame] | 315 | "method offsets", GetLocation().c_str(), i, |
Ian Rogers | 8d31bbd | 2013-10-13 10:44:14 -0700 | [diff] [blame] | 316 | dex_file_location.c_str()); |
Brian Carlstrom | fb331d7 | 2013-07-25 22:00:16 -0700 | [diff] [blame] | 317 | return false; |
| 318 | } |
Brian Carlstrom | e24fa61 | 2011-09-29 00:53:55 -0700 | [diff] [blame] | 319 | |
Vladimir Marko | aa4497d | 2014-09-05 14:01:17 +0100 | [diff] [blame] | 320 | std::string canonical_location = DexFile::GetDexCanonicalLocation(dex_file_location.c_str()); |
| 321 | |
| 322 | // Create the OatDexFile and add it to the owning container. |
Vladimir Marko | 539690a | 2014-06-05 18:36:42 +0100 | [diff] [blame] | 323 | OatDexFile* oat_dex_file = new OatDexFile(this, |
| 324 | dex_file_location, |
Vladimir Marko | aa4497d | 2014-09-05 14:01:17 +0100 | [diff] [blame] | 325 | canonical_location, |
Vladimir Marko | 539690a | 2014-06-05 18:36:42 +0100 | [diff] [blame] | 326 | dex_file_checksum, |
| 327 | dex_file_pointer, |
| 328 | methods_offsets_pointer); |
Vladimir Marko | aa4497d | 2014-09-05 14:01:17 +0100 | [diff] [blame] | 329 | oat_dex_files_storage_.push_back(oat_dex_file); |
| 330 | |
| 331 | // Add the location and canonical location (if different) to the oat_dex_files_ table. |
Vladimir Marko | 3f5838d | 2014-08-07 18:07:18 +0100 | [diff] [blame] | 332 | StringPiece key(oat_dex_file->GetDexFileLocation()); |
Vladimir Marko | 539690a | 2014-06-05 18:36:42 +0100 | [diff] [blame] | 333 | oat_dex_files_.Put(key, oat_dex_file); |
Vladimir Marko | aa4497d | 2014-09-05 14:01:17 +0100 | [diff] [blame] | 334 | if (canonical_location != dex_file_location) { |
| 335 | StringPiece canonical_key(oat_dex_file->GetCanonicalDexFileLocation()); |
| 336 | oat_dex_files_.Put(canonical_key, oat_dex_file); |
| 337 | } |
Brian Carlstrom | e24fa61 | 2011-09-29 00:53:55 -0700 | [diff] [blame] | 338 | } |
Brian Carlstrom | f1b3030 | 2013-03-28 10:35:32 -0700 | [diff] [blame] | 339 | return true; |
Brian Carlstrom | e24fa61 | 2011-09-29 00:53:55 -0700 | [diff] [blame] | 340 | } |
| 341 | |
| 342 | const OatHeader& OatFile::GetOatHeader() const { |
Ian Rogers | 30fab40 | 2012-01-23 15:43:46 -0800 | [diff] [blame] | 343 | return *reinterpret_cast<const OatHeader*>(Begin()); |
Brian Carlstrom | e24fa61 | 2011-09-29 00:53:55 -0700 | [diff] [blame] | 344 | } |
| 345 | |
Ian Rogers | 1373595 | 2014-10-08 12:43:28 -0700 | [diff] [blame] | 346 | const uint8_t* OatFile::Begin() const { |
Brian Carlstrom | 700c8d3 | 2012-11-05 10:42:02 -0800 | [diff] [blame] | 347 | CHECK(begin_ != NULL); |
| 348 | return begin_; |
Brian Carlstrom | e24fa61 | 2011-09-29 00:53:55 -0700 | [diff] [blame] | 349 | } |
| 350 | |
Ian Rogers | 1373595 | 2014-10-08 12:43:28 -0700 | [diff] [blame] | 351 | const uint8_t* OatFile::End() const { |
Brian Carlstrom | 700c8d3 | 2012-11-05 10:42:02 -0800 | [diff] [blame] | 352 | CHECK(end_ != NULL); |
| 353 | return end_; |
Brian Carlstrom | e24fa61 | 2011-09-29 00:53:55 -0700 | [diff] [blame] | 354 | } |
| 355 | |
Vladimir Marko | 5c42c29 | 2015-02-25 12:02:49 +0000 | [diff] [blame] | 356 | const uint8_t* OatFile::BssBegin() const { |
| 357 | return bss_begin_; |
| 358 | } |
| 359 | |
| 360 | const uint8_t* OatFile::BssEnd() const { |
| 361 | return bss_end_; |
| 362 | } |
| 363 | |
Ian Rogers | 8d31bbd | 2013-10-13 10:44:14 -0700 | [diff] [blame] | 364 | const OatFile::OatDexFile* OatFile::GetOatDexFile(const char* dex_location, |
| 365 | const uint32_t* dex_location_checksum, |
Ian Rogers | 7fe2c69 | 2011-12-06 16:35:59 -0800 | [diff] [blame] | 366 | bool warn_if_not_found) const { |
Vladimir Marko | 3f5838d | 2014-08-07 18:07:18 +0100 | [diff] [blame] | 367 | // NOTE: We assume here that the canonical location for a given dex_location never |
| 368 | // changes. If it does (i.e. some symlink used by the filename changes) we may return |
| 369 | // an incorrect OatDexFile. As long as we have a checksum to check, we shall return |
| 370 | // an identical file or fail; otherwise we may see some unpredictable failures. |
Calin Juravle | 4e1d579 | 2014-07-15 23:56:47 +0100 | [diff] [blame] | 371 | |
Vladimir Marko | 3f5838d | 2014-08-07 18:07:18 +0100 | [diff] [blame] | 372 | // TODO: Additional analysis of usage patterns to see if this can be simplified |
| 373 | // without any performance loss, for example by not doing the first lock-free lookup. |
| 374 | |
| 375 | const OatFile::OatDexFile* oat_dex_file = nullptr; |
| 376 | StringPiece key(dex_location); |
| 377 | // Try to find the key cheaply in the oat_dex_files_ map which holds dex locations |
| 378 | // directly mentioned in the oat file and doesn't require locking. |
| 379 | auto primary_it = oat_dex_files_.find(key); |
| 380 | if (primary_it != oat_dex_files_.end()) { |
| 381 | oat_dex_file = primary_it->second; |
| 382 | DCHECK(oat_dex_file != nullptr); |
| 383 | } else { |
| 384 | // This dex_location is not one of the dex locations directly mentioned in the |
| 385 | // oat file. The correct lookup is via the canonical location but first see in |
| 386 | // the secondary_oat_dex_files_ whether we've looked up this location before. |
| 387 | MutexLock mu(Thread::Current(), secondary_lookup_lock_); |
| 388 | auto secondary_lb = secondary_oat_dex_files_.lower_bound(key); |
| 389 | if (secondary_lb != secondary_oat_dex_files_.end() && key == secondary_lb->first) { |
| 390 | oat_dex_file = secondary_lb->second; // May be nullptr. |
| 391 | } else { |
| 392 | // We haven't seen this dex_location before, we must check the canonical location. |
Vladimir Marko | 3f5838d | 2014-08-07 18:07:18 +0100 | [diff] [blame] | 393 | std::string dex_canonical_location = DexFile::GetDexCanonicalLocation(dex_location); |
Vladimir Marko | aa4497d | 2014-09-05 14:01:17 +0100 | [diff] [blame] | 394 | if (dex_canonical_location != dex_location) { |
| 395 | StringPiece canonical_key(dex_canonical_location); |
| 396 | auto canonical_it = oat_dex_files_.find(canonical_key); |
| 397 | if (canonical_it != oat_dex_files_.end()) { |
| 398 | oat_dex_file = canonical_it->second; |
| 399 | } // else keep nullptr. |
Vladimir Marko | 3f5838d | 2014-08-07 18:07:18 +0100 | [diff] [blame] | 400 | } // else keep nullptr. |
| 401 | |
| 402 | // Copy the key to the string_cache_ and store the result in secondary map. |
| 403 | string_cache_.emplace_back(key.data(), key.length()); |
| 404 | StringPiece key_copy(string_cache_.back()); |
| 405 | secondary_oat_dex_files_.PutBefore(secondary_lb, key_copy, oat_dex_file); |
Brian Carlstrom | 756ee4e | 2013-10-03 15:46:12 -0700 | [diff] [blame] | 406 | } |
| 407 | } |
Vladimir Marko | 3f5838d | 2014-08-07 18:07:18 +0100 | [diff] [blame] | 408 | if (oat_dex_file != nullptr && |
| 409 | (dex_location_checksum == nullptr || |
| 410 | oat_dex_file->GetDexFileLocationChecksum() == *dex_location_checksum)) { |
| 411 | return oat_dex_file; |
| 412 | } |
Brian Carlstrom | 756ee4e | 2013-10-03 15:46:12 -0700 | [diff] [blame] | 413 | |
| 414 | if (warn_if_not_found) { |
Vladimir Marko | 3f5838d | 2014-08-07 18:07:18 +0100 | [diff] [blame] | 415 | std::string dex_canonical_location = DexFile::GetDexCanonicalLocation(dex_location); |
Brian Carlstrom | 0d6adac | 2014-02-05 17:39:16 -0800 | [diff] [blame] | 416 | std::string checksum("<unspecified>"); |
| 417 | if (dex_location_checksum != NULL) { |
| 418 | checksum = StringPrintf("0x%08x", *dex_location_checksum); |
| 419 | } |
Brian Carlstrom | 756ee4e | 2013-10-03 15:46:12 -0700 | [diff] [blame] | 420 | LOG(WARNING) << "Failed to find OatDexFile for DexFile " << dex_location |
Calin Juravle | 4e1d579 | 2014-07-15 23:56:47 +0100 | [diff] [blame] | 421 | << " ( canonical path " << dex_canonical_location << ")" |
Brian Carlstrom | 0d6adac | 2014-02-05 17:39:16 -0800 | [diff] [blame] | 422 | << " with checksum " << checksum << " in OatFile " << GetLocation(); |
Brian Carlstrom | 756ee4e | 2013-10-03 15:46:12 -0700 | [diff] [blame] | 423 | if (kIsDebugBuild) { |
Vladimir Marko | aa4497d | 2014-09-05 14:01:17 +0100 | [diff] [blame] | 424 | for (const OatDexFile* odf : oat_dex_files_storage_) { |
Brian Carlstrom | 756ee4e | 2013-10-03 15:46:12 -0700 | [diff] [blame] | 425 | LOG(WARNING) << "OatFile " << GetLocation() |
Vladimir Marko | aa4497d | 2014-09-05 14:01:17 +0100 | [diff] [blame] | 426 | << " contains OatDexFile " << odf->GetDexFileLocation() |
| 427 | << " (canonical path " << odf->GetCanonicalDexFileLocation() << ")" |
| 428 | << " with checksum 0x" << std::hex << odf->GetDexFileLocationChecksum(); |
Brian Carlstrom | 7571e8b | 2013-08-12 17:04:14 -0700 | [diff] [blame] | 429 | } |
Ian Rogers | 7fe2c69 | 2011-12-06 16:35:59 -0800 | [diff] [blame] | 430 | } |
Brian Carlstrom | e24fa61 | 2011-09-29 00:53:55 -0700 | [diff] [blame] | 431 | } |
Calin Juravle | 4e1d579 | 2014-07-15 23:56:47 +0100 | [diff] [blame] | 432 | |
Brian Carlstrom | 756ee4e | 2013-10-03 15:46:12 -0700 | [diff] [blame] | 433 | return NULL; |
Brian Carlstrom | aded5f7 | 2011-10-07 17:15:04 -0700 | [diff] [blame] | 434 | } |
| 435 | |
Brian Carlstrom | e24fa61 | 2011-09-29 00:53:55 -0700 | [diff] [blame] | 436 | OatFile::OatDexFile::OatDexFile(const OatFile* oat_file, |
Elliott Hughes | aa6a588 | 2012-01-13 19:39:16 -0800 | [diff] [blame] | 437 | const std::string& dex_file_location, |
Vladimir Marko | aa4497d | 2014-09-05 14:01:17 +0100 | [diff] [blame] | 438 | const std::string& canonical_dex_file_location, |
Brian Carlstrom | 5b332c8 | 2012-02-01 15:02:31 -0800 | [diff] [blame] | 439 | uint32_t dex_file_location_checksum, |
Ian Rogers | 1373595 | 2014-10-08 12:43:28 -0700 | [diff] [blame] | 440 | const uint8_t* dex_file_pointer, |
Brian Carlstrom | 0755ec5 | 2012-01-11 15:19:46 -0800 | [diff] [blame] | 441 | const uint32_t* oat_class_offsets_pointer) |
Brian Carlstrom | e24fa61 | 2011-09-29 00:53:55 -0700 | [diff] [blame] | 442 | : oat_file_(oat_file), |
| 443 | dex_file_location_(dex_file_location), |
Vladimir Marko | aa4497d | 2014-09-05 14:01:17 +0100 | [diff] [blame] | 444 | canonical_dex_file_location_(canonical_dex_file_location), |
Brian Carlstrom | 5b332c8 | 2012-02-01 15:02:31 -0800 | [diff] [blame] | 445 | dex_file_location_checksum_(dex_file_location_checksum), |
Brian Carlstrom | 8952189 | 2011-12-07 22:05:07 -0800 | [diff] [blame] | 446 | dex_file_pointer_(dex_file_pointer), |
Brian Carlstrom | 0755ec5 | 2012-01-11 15:19:46 -0800 | [diff] [blame] | 447 | oat_class_offsets_pointer_(oat_class_offsets_pointer) {} |
Brian Carlstrom | e24fa61 | 2011-09-29 00:53:55 -0700 | [diff] [blame] | 448 | |
| 449 | OatFile::OatDexFile::~OatDexFile() {} |
| 450 | |
Ian Rogers | 05f28c6 | 2012-10-23 18:12:13 -0700 | [diff] [blame] | 451 | size_t OatFile::OatDexFile::FileSize() const { |
| 452 | return reinterpret_cast<const DexFile::Header*>(dex_file_pointer_)->file_size_; |
| 453 | } |
| 454 | |
Richard Uhler | fbef44d | 2014-12-23 09:48:51 -0800 | [diff] [blame] | 455 | std::unique_ptr<const DexFile> OatFile::OatDexFile::OpenDexFile(std::string* error_msg) const { |
Ian Rogers | 05f28c6 | 2012-10-23 18:12:13 -0700 | [diff] [blame] | 456 | return DexFile::Open(dex_file_pointer_, FileSize(), dex_file_location_, |
Richard Uhler | 07b3c23 | 2015-03-31 15:57:54 -0700 | [diff] [blame] | 457 | dex_file_location_checksum_, this, error_msg); |
Brian Carlstrom | 8952189 | 2011-12-07 22:05:07 -0800 | [diff] [blame] | 458 | } |
| 459 | |
Brian Carlstrom | 2cbaccb | 2014-09-14 20:34:17 -0700 | [diff] [blame] | 460 | uint32_t OatFile::OatDexFile::GetOatClassOffset(uint16_t class_def_index) const { |
| 461 | return oat_class_offsets_pointer_[class_def_index]; |
| 462 | } |
| 463 | |
Vladimir Marko | d3c5beb | 2014-04-11 16:32:51 +0100 | [diff] [blame] | 464 | OatFile::OatClass OatFile::OatDexFile::GetOatClass(uint16_t class_def_index) const { |
Brian Carlstrom | 2cbaccb | 2014-09-14 20:34:17 -0700 | [diff] [blame] | 465 | uint32_t oat_class_offset = GetOatClassOffset(class_def_index); |
Brian Carlstrom | 0755ec5 | 2012-01-11 15:19:46 -0800 | [diff] [blame] | 466 | |
Ian Rogers | 1373595 | 2014-10-08 12:43:28 -0700 | [diff] [blame] | 467 | const uint8_t* oat_class_pointer = oat_file_->Begin() + oat_class_offset; |
Brian Carlstrom | 7571e8b | 2013-08-12 17:04:14 -0700 | [diff] [blame] | 468 | CHECK_LT(oat_class_pointer, oat_file_->End()) << oat_file_->GetLocation(); |
Brian Carlstrom | 0755ec5 | 2012-01-11 15:19:46 -0800 | [diff] [blame] | 469 | |
Ian Rogers | 1373595 | 2014-10-08 12:43:28 -0700 | [diff] [blame] | 470 | const uint8_t* status_pointer = oat_class_pointer; |
Brian Carlstrom | ba150c3 | 2013-08-27 17:31:03 -0700 | [diff] [blame] | 471 | CHECK_LT(status_pointer, oat_file_->End()) << oat_file_->GetLocation(); |
| 472 | mirror::Class::Status status = |
| 473 | static_cast<mirror::Class::Status>(*reinterpret_cast<const int16_t*>(status_pointer)); |
| 474 | CHECK_LT(status, mirror::Class::kStatusMax); |
| 475 | |
Ian Rogers | 1373595 | 2014-10-08 12:43:28 -0700 | [diff] [blame] | 476 | const uint8_t* type_pointer = status_pointer + sizeof(uint16_t); |
Brian Carlstrom | ba150c3 | 2013-08-27 17:31:03 -0700 | [diff] [blame] | 477 | CHECK_LT(type_pointer, oat_file_->End()) << oat_file_->GetLocation(); |
| 478 | OatClassType type = static_cast<OatClassType>(*reinterpret_cast<const uint16_t*>(type_pointer)); |
| 479 | CHECK_LT(type, kOatClassMax); |
| 480 | |
Ian Rogers | 1373595 | 2014-10-08 12:43:28 -0700 | [diff] [blame] | 481 | const uint8_t* after_type_pointer = type_pointer + sizeof(int16_t); |
Brian Carlstrom | cd937a9 | 2014-03-04 22:53:23 -0800 | [diff] [blame] | 482 | CHECK_LE(after_type_pointer, oat_file_->End()) << oat_file_->GetLocation(); |
Brian Carlstrom | ba150c3 | 2013-08-27 17:31:03 -0700 | [diff] [blame] | 483 | |
Brian Carlstrom | cd937a9 | 2014-03-04 22:53:23 -0800 | [diff] [blame] | 484 | uint32_t bitmap_size = 0; |
Ian Rogers | 1373595 | 2014-10-08 12:43:28 -0700 | [diff] [blame] | 485 | const uint8_t* bitmap_pointer = nullptr; |
| 486 | const uint8_t* methods_pointer = nullptr; |
Ian Rogers | 97b52f8 | 2014-08-14 11:34:07 -0700 | [diff] [blame] | 487 | if (type != kOatClassNoneCompiled) { |
| 488 | if (type == kOatClassSomeCompiled) { |
| 489 | bitmap_size = static_cast<uint32_t>(*reinterpret_cast<const uint32_t*>(after_type_pointer)); |
| 490 | bitmap_pointer = after_type_pointer + sizeof(bitmap_size); |
| 491 | CHECK_LE(bitmap_pointer, oat_file_->End()) << oat_file_->GetLocation(); |
| 492 | methods_pointer = bitmap_pointer + bitmap_size; |
| 493 | } else { |
| 494 | methods_pointer = after_type_pointer; |
| 495 | } |
| 496 | CHECK_LE(methods_pointer, oat_file_->End()) << oat_file_->GetLocation(); |
Brian Carlstrom | cd937a9 | 2014-03-04 22:53:23 -0800 | [diff] [blame] | 497 | } |
Brian Carlstrom | 0755ec5 | 2012-01-11 15:19:46 -0800 | [diff] [blame] | 498 | |
Richard Uhler | 07b3c23 | 2015-03-31 15:57:54 -0700 | [diff] [blame] | 499 | return OatFile::OatClass(oat_file_, |
| 500 | status, |
| 501 | type, |
| 502 | bitmap_size, |
| 503 | reinterpret_cast<const uint32_t*>(bitmap_pointer), |
| 504 | reinterpret_cast<const OatMethodOffsets*>(methods_pointer)); |
Brian Carlstrom | e24fa61 | 2011-09-29 00:53:55 -0700 | [diff] [blame] | 505 | } |
| 506 | |
Brian Carlstrom | 0755ec5 | 2012-01-11 15:19:46 -0800 | [diff] [blame] | 507 | OatFile::OatClass::OatClass(const OatFile* oat_file, |
Ian Rogers | 2dd0e2c | 2013-01-24 12:42:14 -0800 | [diff] [blame] | 508 | mirror::Class::Status status, |
Brian Carlstrom | ba150c3 | 2013-08-27 17:31:03 -0700 | [diff] [blame] | 509 | OatClassType type, |
| 510 | uint32_t bitmap_size, |
| 511 | const uint32_t* bitmap_pointer, |
Brian Carlstrom | 0755ec5 | 2012-01-11 15:19:46 -0800 | [diff] [blame] | 512 | const OatMethodOffsets* methods_pointer) |
Brian Carlstrom | ba150c3 | 2013-08-27 17:31:03 -0700 | [diff] [blame] | 513 | : oat_file_(oat_file), status_(status), type_(type), |
Vladimir Marko | d3c5beb | 2014-04-11 16:32:51 +0100 | [diff] [blame] | 514 | bitmap_(bitmap_pointer), methods_pointer_(methods_pointer) { |
Brian Carlstrom | ba150c3 | 2013-08-27 17:31:03 -0700 | [diff] [blame] | 515 | switch (type_) { |
| 516 | case kOatClassAllCompiled: { |
| 517 | CHECK_EQ(0U, bitmap_size); |
Brian Carlstrom | cd937a9 | 2014-03-04 22:53:23 -0800 | [diff] [blame] | 518 | CHECK(bitmap_pointer == nullptr); |
Ian Rogers | 97b52f8 | 2014-08-14 11:34:07 -0700 | [diff] [blame] | 519 | CHECK(methods_pointer != nullptr); |
Brian Carlstrom | ba150c3 | 2013-08-27 17:31:03 -0700 | [diff] [blame] | 520 | break; |
| 521 | } |
| 522 | case kOatClassSomeCompiled: { |
| 523 | CHECK_NE(0U, bitmap_size); |
Brian Carlstrom | cd937a9 | 2014-03-04 22:53:23 -0800 | [diff] [blame] | 524 | CHECK(bitmap_pointer != nullptr); |
Ian Rogers | 97b52f8 | 2014-08-14 11:34:07 -0700 | [diff] [blame] | 525 | CHECK(methods_pointer != nullptr); |
Brian Carlstrom | ba150c3 | 2013-08-27 17:31:03 -0700 | [diff] [blame] | 526 | break; |
| 527 | } |
| 528 | case kOatClassNoneCompiled: { |
| 529 | CHECK_EQ(0U, bitmap_size); |
Brian Carlstrom | cd937a9 | 2014-03-04 22:53:23 -0800 | [diff] [blame] | 530 | CHECK(bitmap_pointer == nullptr); |
Ian Rogers | 97b52f8 | 2014-08-14 11:34:07 -0700 | [diff] [blame] | 531 | CHECK(methods_pointer_ == nullptr); |
Brian Carlstrom | ba150c3 | 2013-08-27 17:31:03 -0700 | [diff] [blame] | 532 | break; |
| 533 | } |
| 534 | case kOatClassMax: { |
| 535 | LOG(FATAL) << "Invalid OatClassType " << type_; |
| 536 | break; |
| 537 | } |
| 538 | } |
| 539 | } |
Brian Carlstrom | e24fa61 | 2011-09-29 00:53:55 -0700 | [diff] [blame] | 540 | |
Brian Carlstrom | 2cbaccb | 2014-09-14 20:34:17 -0700 | [diff] [blame] | 541 | uint32_t OatFile::OatClass::GetOatMethodOffsetsOffset(uint32_t method_index) const { |
| 542 | const OatMethodOffsets* oat_method_offsets = GetOatMethodOffsets(method_index); |
| 543 | if (oat_method_offsets == nullptr) { |
| 544 | return 0u; |
| 545 | } |
| 546 | return reinterpret_cast<const uint8_t*>(oat_method_offsets) - oat_file_->Begin(); |
| 547 | } |
| 548 | |
| 549 | const OatMethodOffsets* OatFile::OatClass::GetOatMethodOffsets(uint32_t method_index) const { |
Vladimir Marko | d3c5beb | 2014-04-11 16:32:51 +0100 | [diff] [blame] | 550 | // NOTE: We don't keep the number of methods and cannot do a bounds check for method_index. |
Brian Carlstrom | 2cbaccb | 2014-09-14 20:34:17 -0700 | [diff] [blame] | 551 | if (methods_pointer_ == nullptr) { |
Brian Carlstrom | ba150c3 | 2013-08-27 17:31:03 -0700 | [diff] [blame] | 552 | CHECK_EQ(kOatClassNoneCompiled, type_); |
Brian Carlstrom | 2cbaccb | 2014-09-14 20:34:17 -0700 | [diff] [blame] | 553 | return nullptr; |
Brian Carlstrom | ba150c3 | 2013-08-27 17:31:03 -0700 | [diff] [blame] | 554 | } |
| 555 | size_t methods_pointer_index; |
Brian Carlstrom | 2cbaccb | 2014-09-14 20:34:17 -0700 | [diff] [blame] | 556 | if (bitmap_ == nullptr) { |
Brian Carlstrom | ba150c3 | 2013-08-27 17:31:03 -0700 | [diff] [blame] | 557 | CHECK_EQ(kOatClassAllCompiled, type_); |
| 558 | methods_pointer_index = method_index; |
| 559 | } else { |
| 560 | CHECK_EQ(kOatClassSomeCompiled, type_); |
Vladimir Marko | d3c5beb | 2014-04-11 16:32:51 +0100 | [diff] [blame] | 561 | if (!BitVector::IsBitSet(bitmap_, method_index)) { |
Brian Carlstrom | 2cbaccb | 2014-09-14 20:34:17 -0700 | [diff] [blame] | 562 | return nullptr; |
Brian Carlstrom | ba150c3 | 2013-08-27 17:31:03 -0700 | [diff] [blame] | 563 | } |
Vladimir Marko | d3c5beb | 2014-04-11 16:32:51 +0100 | [diff] [blame] | 564 | size_t num_set_bits = BitVector::NumSetBits(bitmap_, method_index); |
| 565 | methods_pointer_index = num_set_bits; |
Brian Carlstrom | ba150c3 | 2013-08-27 17:31:03 -0700 | [diff] [blame] | 566 | } |
| 567 | const OatMethodOffsets& oat_method_offsets = methods_pointer_[methods_pointer_index]; |
Brian Carlstrom | 2cbaccb | 2014-09-14 20:34:17 -0700 | [diff] [blame] | 568 | return &oat_method_offsets; |
| 569 | } |
| 570 | |
| 571 | const OatFile::OatMethod OatFile::OatClass::GetOatMethod(uint32_t method_index) const { |
| 572 | const OatMethodOffsets* oat_method_offsets = GetOatMethodOffsets(method_index); |
| 573 | if (oat_method_offsets == nullptr) { |
Mathieu Chartier | 957ca1c | 2014-11-21 16:51:29 -0800 | [diff] [blame] | 574 | return OatMethod(nullptr, 0); |
Brian Carlstrom | 2cbaccb | 2014-09-14 20:34:17 -0700 | [diff] [blame] | 575 | } |
| 576 | if (oat_file_->IsExecutable() || |
| 577 | Runtime::Current() == nullptr || // This case applies for oatdump. |
Mathieu Chartier | e5f13e5 | 2015-02-24 09:37:21 -0800 | [diff] [blame] | 578 | Runtime::Current()->IsAotCompiler()) { |
Mathieu Chartier | 957ca1c | 2014-11-21 16:51:29 -0800 | [diff] [blame] | 579 | return OatMethod(oat_file_->Begin(), oat_method_offsets->code_offset_); |
Alex Light | 9dcc457 | 2014-08-14 14:16:26 -0700 | [diff] [blame] | 580 | } |
Mathieu Chartier | e5f13e5 | 2015-02-24 09:37:21 -0800 | [diff] [blame] | 581 | // We aren't allowed to use the compiled code. We just force it down the interpreted / jit |
| 582 | // version. |
| 583 | return OatMethod(oat_file_->Begin(), 0); |
Brian Carlstrom | 3320cf4 | 2011-10-04 14:58:28 -0700 | [diff] [blame] | 584 | } |
| 585 | |
Brian Carlstrom | ea46f95 | 2013-07-30 01:26:50 -0700 | [diff] [blame] | 586 | void OatFile::OatMethod::LinkMethod(mirror::ArtMethod* method) const { |
Brian Carlstrom | 3320cf4 | 2011-10-04 14:58:28 -0700 | [diff] [blame] | 587 | CHECK(method != NULL); |
Ian Rogers | ef7d42f | 2014-01-06 12:55:46 -0800 | [diff] [blame] | 588 | method->SetEntryPointFromQuickCompiledCode(GetQuickCode()); |
Brian Carlstrom | ae82698 | 2011-11-09 01:33:42 -0800 | [diff] [blame] | 589 | } |
| 590 | |
Andreas Gampe | 7ba6496 | 2014-10-23 11:37:40 -0700 | [diff] [blame] | 591 | bool OatFile::IsPic() const { |
Igor Murashkin | 4677476 | 2014-10-22 11:37:02 -0700 | [diff] [blame] | 592 | return GetOatHeader().IsPic(); |
Andreas Gampe | 7ba6496 | 2014-10-23 11:37:40 -0700 | [diff] [blame] | 593 | // TODO: Check against oat_patches. b/18144996 |
| 594 | } |
| 595 | |
Andreas Gampe | 7848da4 | 2015-04-09 11:15:04 -0700 | [diff] [blame^] | 596 | static constexpr char kDexClassPathEncodingSeparator = '*'; |
| 597 | |
| 598 | std::string OatFile::EncodeDexFileDependencies(const std::vector<const DexFile*>& dex_files) { |
| 599 | std::ostringstream out; |
| 600 | |
| 601 | for (const DexFile* dex_file : dex_files) { |
| 602 | out << dex_file->GetLocation().c_str(); |
| 603 | out << kDexClassPathEncodingSeparator; |
| 604 | out << dex_file->GetLocationChecksum(); |
| 605 | out << kDexClassPathEncodingSeparator; |
| 606 | } |
| 607 | |
| 608 | return out.str(); |
| 609 | } |
| 610 | |
| 611 | bool OatFile::CheckStaticDexFileDependencies(const char* dex_dependencies, std::string* msg) { |
| 612 | if (dex_dependencies == nullptr || dex_dependencies[0] == 0) { |
| 613 | // No dependencies. |
| 614 | return true; |
| 615 | } |
| 616 | |
| 617 | // Assumption: this is not performance-critical. So it's OK to do this with a std::string and |
| 618 | // Split() instead of manual parsing of the combined char*. |
| 619 | std::vector<std::string> split; |
| 620 | Split(dex_dependencies, kDexClassPathEncodingSeparator, &split); |
| 621 | if (split.size() % 2 != 0) { |
| 622 | // Expected pairs of location and checksum. |
| 623 | *msg = StringPrintf("Odd number of elements in dependency list %s", dex_dependencies); |
| 624 | return false; |
| 625 | } |
| 626 | |
| 627 | for (auto it = split.begin(), end = split.end(); it != end; it += 2) { |
| 628 | std::string& location = *it; |
| 629 | std::string& checksum = *(it + 1); |
| 630 | int64_t converted = strtoll(checksum.c_str(), nullptr, 10); |
| 631 | if (converted == 0) { |
| 632 | // Conversion error. |
| 633 | *msg = StringPrintf("Conversion error for %s", checksum.c_str()); |
| 634 | return false; |
| 635 | } |
| 636 | |
| 637 | uint32_t dex_checksum; |
| 638 | std::string error_msg; |
| 639 | if (DexFile::GetChecksum(DexFile::GetDexCanonicalLocation(location.c_str()).c_str(), |
| 640 | &dex_checksum, |
| 641 | &error_msg)) { |
| 642 | if (converted != dex_checksum) { |
| 643 | *msg = StringPrintf("Checksums don't match for %s: %" PRId64 " vs %u", |
| 644 | location.c_str(), converted, dex_checksum); |
| 645 | return false; |
| 646 | } |
| 647 | } else { |
| 648 | // Problem retrieving checksum. |
| 649 | // TODO: odex files? |
| 650 | *msg = StringPrintf("Could not retrieve checksum for %s: %s", location.c_str(), |
| 651 | error_msg.c_str()); |
| 652 | return false; |
| 653 | } |
| 654 | } |
| 655 | |
| 656 | return true; |
| 657 | } |
| 658 | |
| 659 | bool OatFile::GetDexLocationsFromDependencies(const char* dex_dependencies, |
| 660 | std::vector<std::string>* locations) { |
| 661 | DCHECK(locations != nullptr); |
| 662 | if (dex_dependencies == nullptr || dex_dependencies[0] == 0) { |
| 663 | return true; |
| 664 | } |
| 665 | |
| 666 | // Assumption: this is not performance-critical. So it's OK to do this with a std::string and |
| 667 | // Split() instead of manual parsing of the combined char*. |
| 668 | std::vector<std::string> split; |
| 669 | Split(dex_dependencies, kDexClassPathEncodingSeparator, &split); |
| 670 | if (split.size() % 2 != 0) { |
| 671 | // Expected pairs of location and checksum. |
| 672 | return false; |
| 673 | } |
| 674 | |
| 675 | for (auto it = split.begin(), end = split.end(); it != end; it += 2) { |
| 676 | locations->push_back(*it); |
| 677 | } |
| 678 | |
| 679 | return true; |
| 680 | } |
| 681 | |
Brian Carlstrom | e24fa61 | 2011-09-29 00:53:55 -0700 | [diff] [blame] | 682 | } // namespace art |