image_writer.cc revision c50d8e11a098cc5c6239aa86b47d4fcf8cbb4899
12faa5f1271587cda765f26bcf2951065300a01ffElliott Hughes/* 22faa5f1271587cda765f26bcf2951065300a01ffElliott Hughes * Copyright (C) 2011 The Android Open Source Project 32faa5f1271587cda765f26bcf2951065300a01ffElliott Hughes * 42faa5f1271587cda765f26bcf2951065300a01ffElliott Hughes * Licensed under the Apache License, Version 2.0 (the "License"); 52faa5f1271587cda765f26bcf2951065300a01ffElliott Hughes * you may not use this file except in compliance with the License. 62faa5f1271587cda765f26bcf2951065300a01ffElliott Hughes * You may obtain a copy of the License at 72faa5f1271587cda765f26bcf2951065300a01ffElliott Hughes * 82faa5f1271587cda765f26bcf2951065300a01ffElliott Hughes * http://www.apache.org/licenses/LICENSE-2.0 92faa5f1271587cda765f26bcf2951065300a01ffElliott Hughes * 102faa5f1271587cda765f26bcf2951065300a01ffElliott Hughes * Unless required by applicable law or agreed to in writing, software 112faa5f1271587cda765f26bcf2951065300a01ffElliott Hughes * distributed under the License is distributed on an "AS IS" BASIS, 122faa5f1271587cda765f26bcf2951065300a01ffElliott Hughes * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. 132faa5f1271587cda765f26bcf2951065300a01ffElliott Hughes * See the License for the specific language governing permissions and 142faa5f1271587cda765f26bcf2951065300a01ffElliott Hughes * limitations under the License. 152faa5f1271587cda765f26bcf2951065300a01ffElliott Hughes */ 16db4d54081f09abcbe97ffdf615874f2809a9e777Brian Carlstrom 17db4d54081f09abcbe97ffdf615874f2809a9e777Brian Carlstrom#include "image_writer.h" 18db4d54081f09abcbe97ffdf615874f2809a9e777Brian Carlstrom 196cd40e5cc59259a0b7636eb5532c76de9b8a7286Brian Carlstrom#include <sys/stat.h> 2090a3369d3b6238f1a4c9b19ca68978dab1c39bc4Elliott Hughes 21db4d54081f09abcbe97ffdf615874f2809a9e777Brian Carlstrom#include <vector> 22db4d54081f09abcbe97ffdf615874f2809a9e777Brian Carlstrom 2307ed66b5ae659c452cbe1ab20c3dbf1d6f546461Elliott Hughes#include "base/logging.h" 24761600567d73b23324ae0251e871c15d6849ffd8Elliott Hughes#include "base/unix_file/fd_file.h" 25a663ea5de4c9ab6b1510fdebd6d8eca77ba699aeBrian Carlstrom#include "class_linker.h" 26ae826983f7903bc0a6bbbe8426bf393fb2f6d747Brian Carlstrom#include "compiled_method.h" 274f6ad8ab428038129b2d0d6c40b7fd625cca15e1Ian Rogers#include "dex_file-inl.h" 287940e44f4517de5e2634a7e07d58d0fb26160513Brian Carlstrom#include "driver/compiler_driver.h" 297940e44f4517de5e2634a7e07d58d0fb26160513Brian Carlstrom#include "elf_writer.h" 301d54e73444e017d3a65234e0f193846f3e27472bIan Rogers#include "gc/accounting/card_table-inl.h" 311d54e73444e017d3a65234e0f193846f3e27472bIan Rogers#include "gc/accounting/heap_bitmap.h" 321d54e73444e017d3a65234e0f193846f3e27472bIan Rogers#include "gc/heap.h" 331d54e73444e017d3a65234e0f193846f3e27472bIan Rogers#include "gc/space/large_object_space.h" 341d54e73444e017d3a65234e0f193846f3e27472bIan Rogers#include "gc/space/space-inl.h" 35db4d54081f09abcbe97ffdf615874f2809a9e777Brian Carlstrom#include "globals.h" 364a289ed61242964b921434de7d375f46480472a1Brian Carlstrom#include "image.h" 37a663ea5de4c9ab6b1510fdebd6d8eca77ba699aeBrian Carlstrom#include "intern_table.h" 382dd0e2cea360bc9206eb88ecc40d259e796c239dIan Rogers#include "mirror/array-inl.h" 392dd0e2cea360bc9206eb88ecc40d259e796c239dIan Rogers#include "mirror/class-inl.h" 402dd0e2cea360bc9206eb88ecc40d259e796c239dIan Rogers#include "mirror/class_loader.h" 4139ebcb800aabedd0ffa6aa4aeac8aa4194c66e61Ian Rogers#include "mirror/dex_cache-inl.h" 422dd0e2cea360bc9206eb88ecc40d259e796c239dIan Rogers#include "mirror/field-inl.h" 432dd0e2cea360bc9206eb88ecc40d259e796c239dIan Rogers#include "mirror/abstract_method-inl.h" 442dd0e2cea360bc9206eb88ecc40d259e796c239dIan Rogers#include "mirror/object-inl.h" 452dd0e2cea360bc9206eb88ecc40d259e796c239dIan Rogers#include "mirror/object_array-inl.h" 46700c8d31733534a3d978b75a03f6f7e177dc7e81Brian Carlstrom#include "oat.h" 470c717dd1c56bd29cf860d0feda8e629dab2cadb3Logan Chien#include "oat_file.h" 486d4d9fcb4f01e287ee29e81cd1c941ee5d11d379Ian Rogers#include "object_utils.h" 491f87008b165d26541d832ff805250afdc89c253dBrian Carlstrom#include "runtime.h" 5000f7d0eaa6bd93d33bf0c1429bf4ba0b3f28abacIan Rogers#include "scoped_thread_state_change.h" 511f5393447b9f45be7918042d9ee7b521376de866Ian Rogers#include "sirt_ref.h" 52a168c83a1d247094e9efb1244b0f73a5f1e1ed97Elliott Hughes#include "UniquePtr.h" 53db4d54081f09abcbe97ffdf615874f2809a9e777Brian Carlstrom#include "utils.h" 54db4d54081f09abcbe97ffdf615874f2809a9e777Brian Carlstrom 553e3d591f781b771de89f3b989830da2b6ac6fac8Brian Carlstromusing ::art::mirror::AbstractMethod; 563e3d591f781b771de89f3b989830da2b6ac6fac8Brian Carlstromusing ::art::mirror::Class; 573e3d591f781b771de89f3b989830da2b6ac6fac8Brian Carlstromusing ::art::mirror::DexCache; 583e3d591f781b771de89f3b989830da2b6ac6fac8Brian Carlstromusing ::art::mirror::EntryPointFromInterpreter; 593e3d591f781b771de89f3b989830da2b6ac6fac8Brian Carlstromusing ::art::mirror::Field; 603e3d591f781b771de89f3b989830da2b6ac6fac8Brian Carlstromusing ::art::mirror::Object; 613e3d591f781b771de89f3b989830da2b6ac6fac8Brian Carlstromusing ::art::mirror::ObjectArray; 623e3d591f781b771de89f3b989830da2b6ac6fac8Brian Carlstromusing ::art::mirror::String; 632dd0e2cea360bc9206eb88ecc40d259e796c239dIan Rogers 64db4d54081f09abcbe97ffdf615874f2809a9e777Brian Carlstromnamespace art { 65db4d54081f09abcbe97ffdf615874f2809a9e777Brian Carlstrom 66a004aa933a58428489e42d77f707c2b063b73747Brian Carlstrombool ImageWriter::Write(const std::string& image_filename, 6730fab40ee5a07af6b8c3b6b0e9438071695a57f4Ian Rogers uintptr_t image_begin, 68ae826983f7903bc0a6bbbe8426bf393fb2f6d747Brian Carlstrom const std::string& oat_filename, 6996391606d8adfc661e1c21703ded1e7a39377a76Brian Carlstrom const std::string& oat_location) { 70a004aa933a58428489e42d77f707c2b063b73747Brian Carlstrom CHECK(!image_filename.empty()); 71aded5f7ab991f3c1132851599d3bc60ff6707eedBrian Carlstrom 7230fab40ee5a07af6b8c3b6b0e9438071695a57f4Ian Rogers CHECK_NE(image_begin, 0U); 7330fab40ee5a07af6b8c3b6b0e9438071695a57f4Ian Rogers image_begin_ = reinterpret_cast<byte*>(image_begin); 7469b15fb098162f19a4c20e6dccdcead04d9c77f0Brian Carlstrom 75ae826983f7903bc0a6bbbe8426bf393fb2f6d747Brian Carlstrom ClassLinker* class_linker = Runtime::Current()->GetClassLinker(); 76ae826983f7903bc0a6bbbe8426bf393fb2f6d747Brian Carlstrom const std::vector<DexCache*>& all_dex_caches = class_linker->GetDexCaches(); 77ae826983f7903bc0a6bbbe8426bf393fb2f6d747Brian Carlstrom for (size_t i = 0; i < all_dex_caches.size(); i++) { 78ae826983f7903bc0a6bbbe8426bf393fb2f6d747Brian Carlstrom DexCache* dex_cache = all_dex_caches[i]; 7964b6d145fa53b8dfb07a8fc2426af13f155d5a4dIan Rogers dex_caches_.insert(dex_cache); 80ae826983f7903bc0a6bbbe8426bf393fb2f6d747Brian Carlstrom } 81ae826983f7903bc0a6bbbe8426bf393fb2f6d747Brian Carlstrom 82700c8d31733534a3d978b75a03f6f7e177dc7e81Brian Carlstrom UniquePtr<File> oat_file(OS::OpenFile(oat_filename.c_str(), true, false)); 83700c8d31733534a3d978b75a03f6f7e177dc7e81Brian Carlstrom if (oat_file.get() == NULL) { 84700c8d31733534a3d978b75a03f6f7e177dc7e81Brian Carlstrom LOG(ERROR) << "Failed to open oat file " << oat_filename << " for " << oat_location; 85e24fa61603a60ade3797e4a0c8b3fccb346cb048Brian Carlstrom return false; 86e24fa61603a60ade3797e4a0c8b3fccb346cb048Brian Carlstrom } 87265091e581c9f643b37e7966890911f09e223269Brian Carlstrom oat_file_ = OatFile::OpenWritable(oat_file.get(), oat_location); 88c50d8e11a098cc5c6239aa86b47d4fcf8cbb4899Brian Carlstrom if (oat_file_ == NULL) { 89c50d8e11a098cc5c6239aa86b47d4fcf8cbb4899Brian Carlstrom LOG(ERROR) << "Failed to open writable oat file " << oat_filename << " for " << oat_location; 90c50d8e11a098cc5c6239aa86b47d4fcf8cbb4899Brian Carlstrom return false; 91c50d8e11a098cc5c6239aa86b47d4fcf8cbb4899Brian Carlstrom } 92b25c3f6a86dc634ce44fb2849385b49465caa84dElliott Hughes class_linker->RegisterOatFile(*oat_file_); 930aba0ba155bef7346bde19e53581200b89ae8a7aJeff Hao interpreter_to_interpreter_entry_offset_ = oat_file_->GetOatHeader().GetInterpreterToInterpreterEntryOffset(); 940aba0ba155bef7346bde19e53581200b89ae8a7aJeff Hao interpreter_to_quick_entry_offset_ = oat_file_->GetOatHeader().GetInterpreterToQuickEntryOffset(); 950aba0ba155bef7346bde19e53581200b89ae8a7aJeff Hao portable_resolution_trampoline_offset_ = oat_file_->GetOatHeader().GetPortableResolutionTrampolineOffset(); 960aba0ba155bef7346bde19e53581200b89ae8a7aJeff Hao quick_resolution_trampoline_offset_ = oat_file_->GetOatHeader().GetQuickResolutionTrampolineOffset(); 97e24fa61603a60ade3797e4a0c8b3fccb346cb048Brian Carlstrom 9800f7d0eaa6bd93d33bf0c1429bf4ba0b3f28abacIan Rogers { 9900f7d0eaa6bd93d33bf0c1429bf4ba0b3f28abacIan Rogers Thread::Current()->TransitionFromSuspendedToRunnable(); 10000f7d0eaa6bd93d33bf0c1429bf4ba0b3f28abacIan Rogers PruneNonImageClasses(); // Remove junk 10100f7d0eaa6bd93d33bf0c1429bf4ba0b3f28abacIan Rogers ComputeLazyFieldsForImageClasses(); // Add useful information 10200f7d0eaa6bd93d33bf0c1429bf4ba0b3f28abacIan Rogers ComputeEagerResolvedStrings(); 10300f7d0eaa6bd93d33bf0c1429bf4ba0b3f28abacIan Rogers Thread::Current()->TransitionFromRunnableToSuspended(kNative); 10400f7d0eaa6bd93d33bf0c1429bf4ba0b3f28abacIan Rogers } 1051d54e73444e017d3a65234e0f193846f3e27472bIan Rogers gc::Heap* heap = Runtime::Current()->GetHeap(); 1061d54e73444e017d3a65234e0f193846f3e27472bIan Rogers heap->CollectGarbage(false); // Remove garbage. 1071d54e73444e017d3a65234e0f193846f3e27472bIan Rogers // Trim size of alloc spaces. 1081d54e73444e017d3a65234e0f193846f3e27472bIan Rogers const std::vector<gc::space::ContinuousSpace*>& spaces = heap->GetContinuousSpaces(); 109b062fdd4cb097fbae69b4bcb479c34d83ecab8caMathieu Chartier // TODO: C++0x auto 1101d54e73444e017d3a65234e0f193846f3e27472bIan Rogers typedef std::vector<gc::space::ContinuousSpace*>::const_iterator It; 1111d54e73444e017d3a65234e0f193846f3e27472bIan Rogers for (It it = spaces.begin(), end = spaces.end(); it != end; ++it) { 1121d54e73444e017d3a65234e0f193846f3e27472bIan Rogers gc::space::ContinuousSpace* space = *it; 1131d54e73444e017d3a65234e0f193846f3e27472bIan Rogers if (space->IsDlMallocSpace()) { 1141d54e73444e017d3a65234e0f193846f3e27472bIan Rogers space->AsDlMallocSpace()->Trim(); 115b062fdd4cb097fbae69b4bcb479c34d83ecab8caMathieu Chartier } 116b062fdd4cb097fbae69b4bcb479c34d83ecab8caMathieu Chartier } 117b062fdd4cb097fbae69b4bcb479c34d83ecab8caMathieu Chartier 118ae826983f7903bc0a6bbbe8426bf393fb2f6d747Brian Carlstrom if (!AllocMemory()) { 119db4d54081f09abcbe97ffdf615874f2809a9e777Brian Carlstrom return false; 120db4d54081f09abcbe97ffdf615874f2809a9e777Brian Carlstrom } 121ae826983f7903bc0a6bbbe8426bf393fb2f6d747Brian Carlstrom#ifndef NDEBUG 1222ce745c06271d5223d57dbf08117b20d5b60694aBrian Carlstrom { // NOLINT(whitespace/braces) 123357e9be24c17a6bc2ae9fb53f25c73503116101dMathieu Chartier ScopedObjectAccess soa(Thread::Current()); 124357e9be24c17a6bc2ae9fb53f25c73503116101dMathieu Chartier CheckNonImageClassesRemoved(); 125357e9be24c17a6bc2ae9fb53f25c73503116101dMathieu Chartier } 126ae826983f7903bc0a6bbbe8426bf393fb2f6d747Brian Carlstrom#endif 127e35517ad40e0713df8420aeb5d3540f146a59f62Mathieu Chartier Thread::Current()->TransitionFromSuspendedToRunnable(); 128700c8d31733534a3d978b75a03f6f7e177dc7e81Brian Carlstrom size_t oat_loaded_size = 0; 129700c8d31733534a3d978b75a03f6f7e177dc7e81Brian Carlstrom size_t oat_data_offset = 0; 13051c2467e8771b56e25ae4f17f66522f979f57a7eBrian Carlstrom ElfWriter::GetOatElfInformation(oat_file.get(), oat_loaded_size, oat_data_offset); 131700c8d31733534a3d978b75a03f6f7e177dc7e81Brian Carlstrom CalculateNewObjectOffsets(oat_loaded_size, oat_data_offset); 132e35517ad40e0713df8420aeb5d3540f146a59f62Mathieu Chartier CopyAndFixupObjects(); 13396391606d8adfc661e1c21703ded1e7a39377a76Brian Carlstrom PatchOatCodeAndMethods(); 134e35517ad40e0713df8420aeb5d3540f146a59f62Mathieu Chartier Thread::Current()->TransitionFromRunnableToSuspended(kNative); 135d8ddfd5eadde1d5f53ef1419f529c799233eaa62Elliott Hughes 136700c8d31733534a3d978b75a03f6f7e177dc7e81Brian Carlstrom UniquePtr<File> image_file(OS::OpenFile(image_filename.c_str(), true)); 137700c8d31733534a3d978b75a03f6f7e177dc7e81Brian Carlstrom if (image_file.get() == NULL) { 138e24fa61603a60ade3797e4a0c8b3fccb346cb048Brian Carlstrom LOG(ERROR) << "Failed to open image file " << image_filename; 139db4d54081f09abcbe97ffdf615874f2809a9e777Brian Carlstrom return false; 140db4d54081f09abcbe97ffdf615874f2809a9e777Brian Carlstrom } 141700c8d31733534a3d978b75a03f6f7e177dc7e81Brian Carlstrom if (fchmod(image_file->Fd(), 0644) != 0) { 1426cd40e5cc59259a0b7636eb5532c76de9b8a7286Brian Carlstrom PLOG(ERROR) << "Failed to make image file world readable: " << image_filename; 1436cd40e5cc59259a0b7636eb5532c76de9b8a7286Brian Carlstrom return EXIT_FAILURE; 1446cd40e5cc59259a0b7636eb5532c76de9b8a7286Brian Carlstrom } 145700c8d31733534a3d978b75a03f6f7e177dc7e81Brian Carlstrom bool success = image_file->WriteFully(image_->Begin(), image_end_); 146e24fa61603a60ade3797e4a0c8b3fccb346cb048Brian Carlstrom if (!success) { 147e24fa61603a60ade3797e4a0c8b3fccb346cb048Brian Carlstrom PLOG(ERROR) << "Failed to write image file " << image_filename; 148e24fa61603a60ade3797e4a0c8b3fccb346cb048Brian Carlstrom return false; 149e24fa61603a60ade3797e4a0c8b3fccb346cb048Brian Carlstrom } 150e24fa61603a60ade3797e4a0c8b3fccb346cb048Brian Carlstrom return true; 151db4d54081f09abcbe97ffdf615874f2809a9e777Brian Carlstrom} 152db4d54081f09abcbe97ffdf615874f2809a9e777Brian Carlstrom 153ae826983f7903bc0a6bbbe8426bf393fb2f6d747Brian Carlstrombool ImageWriter::AllocMemory() { 1541d54e73444e017d3a65234e0f193846f3e27472bIan Rogers gc::Heap* heap = Runtime::Current()->GetHeap(); 1551d54e73444e017d3a65234e0f193846f3e27472bIan Rogers const std::vector<gc::space::ContinuousSpace*>& spaces = heap->GetContinuousSpaces(); 156b062fdd4cb097fbae69b4bcb479c34d83ecab8caMathieu Chartier size_t size = 0; 1571d54e73444e017d3a65234e0f193846f3e27472bIan Rogers // TODO: C++0x auto 1581d54e73444e017d3a65234e0f193846f3e27472bIan Rogers typedef std::vector<gc::space::ContinuousSpace*>::const_iterator It; 1591d54e73444e017d3a65234e0f193846f3e27472bIan Rogers for (It it = spaces.begin(), end = spaces.end(); it != end; ++it) { 1601d54e73444e017d3a65234e0f193846f3e27472bIan Rogers gc::space::ContinuousSpace* space = *it; 1611d54e73444e017d3a65234e0f193846f3e27472bIan Rogers if (space->IsDlMallocSpace()) { 1621d54e73444e017d3a65234e0f193846f3e27472bIan Rogers size += space->Size(); 163b062fdd4cb097fbae69b4bcb479c34d83ecab8caMathieu Chartier } 164b062fdd4cb097fbae69b4bcb479c34d83ecab8caMathieu Chartier } 165b062fdd4cb097fbae69b4bcb479c34d83ecab8caMathieu Chartier 166db4d54081f09abcbe97ffdf615874f2809a9e777Brian Carlstrom int prot = PROT_READ | PROT_WRITE; 167db4d54081f09abcbe97ffdf615874f2809a9e777Brian Carlstrom size_t length = RoundUp(size, kPageSize); 168a40307e8562c20d9b9ce46e773864d07d95512a1Ian Rogers image_.reset(MemMap::MapAnonymous("image writer image", NULL, length, prot)); 16990a3369d3b6238f1a4c9b19ca68978dab1c39bc4Elliott Hughes if (image_.get() == NULL) { 170e24fa61603a60ade3797e4a0c8b3fccb346cb048Brian Carlstrom LOG(ERROR) << "Failed to allocate memory for image file generation"; 171db4d54081f09abcbe97ffdf615874f2809a9e777Brian Carlstrom return false; 172db4d54081f09abcbe97ffdf615874f2809a9e777Brian Carlstrom } 173db4d54081f09abcbe97ffdf615874f2809a9e777Brian Carlstrom return true; 174db4d54081f09abcbe97ffdf615874f2809a9e777Brian Carlstrom} 175db4d54081f09abcbe97ffdf615874f2809a9e777Brian Carlstrom 176d418edaf4df0a410d678389e171ac6d96ae2af15Ian Rogersvoid ImageWriter::ComputeLazyFieldsForImageClasses() { 177d418edaf4df0a410d678389e171ac6d96ae2af15Ian Rogers Runtime* runtime = Runtime::Current(); 178d418edaf4df0a410d678389e171ac6d96ae2af15Ian Rogers ClassLinker* class_linker = runtime->GetClassLinker(); 17900f7d0eaa6bd93d33bf0c1429bf4ba0b3f28abacIan Rogers class_linker->VisitClassesWithoutClassesLock(ComputeLazyFieldsForClassesVisitor, NULL); 180d418edaf4df0a410d678389e171ac6d96ae2af15Ian Rogers} 181d418edaf4df0a410d678389e171ac6d96ae2af15Ian Rogers 1821bac54ffa933fbe9b92b62437577f2f4583eff1aElliott Hughesbool ImageWriter::ComputeLazyFieldsForClassesVisitor(Class* c, void* /*arg*/) { 1831bac54ffa933fbe9b92b62437577f2f4583eff1aElliott Hughes c->ComputeName(); 184d418edaf4df0a410d678389e171ac6d96ae2af15Ian Rogers return true; 185d418edaf4df0a410d678389e171ac6d96ae2af15Ian Rogers} 186d418edaf4df0a410d678389e171ac6d96ae2af15Ian Rogers 187d1f1bf03e30320e5730790678f470ac8a610b6b6Ian Rogersvoid ImageWriter::ComputeEagerResolvedStringsCallback(Object* obj, void* arg) { 188d1f1bf03e30320e5730790678f470ac8a610b6b6Ian Rogers if (!obj->GetClass()->IsStringClass()) { 189d1f1bf03e30320e5730790678f470ac8a610b6b6Ian Rogers return; 190d1f1bf03e30320e5730790678f470ac8a610b6b6Ian Rogers } 191d1f1bf03e30320e5730790678f470ac8a610b6b6Ian Rogers String* string = obj->AsString(); 192637c65b1e431fd90195b71c141b3590bd81cc91aIan Rogers const uint16_t* utf16_string = string->GetCharArray()->GetData() + string->GetOffset(); 193d1f1bf03e30320e5730790678f470ac8a610b6b6Ian Rogers ImageWriter* writer = reinterpret_cast<ImageWriter*>(arg); 194d1f1bf03e30320e5730790678f470ac8a610b6b6Ian Rogers typedef Set::const_iterator CacheIt; // TODO: C++0x auto 195d1f1bf03e30320e5730790678f470ac8a610b6b6Ian Rogers for (CacheIt it = writer->dex_caches_.begin(), end = writer->dex_caches_.end(); it != end; ++it) { 196d1f1bf03e30320e5730790678f470ac8a610b6b6Ian Rogers DexCache* dex_cache = *it; 1974445a7e3398a6143939168097a3aa275b734504dIan Rogers const DexFile& dex_file = *dex_cache->GetDexFile(); 198637c65b1e431fd90195b71c141b3590bd81cc91aIan Rogers const DexFile::StringId* string_id = dex_file.FindStringId(utf16_string); 199d1f1bf03e30320e5730790678f470ac8a610b6b6Ian Rogers if (string_id != NULL) { 200d1f1bf03e30320e5730790678f470ac8a610b6b6Ian Rogers // This string occurs in this dex file, assign the dex cache entry. 201d1f1bf03e30320e5730790678f470ac8a610b6b6Ian Rogers uint32_t string_idx = dex_file.GetIndexForStringId(*string_id); 202d1f1bf03e30320e5730790678f470ac8a610b6b6Ian Rogers if (dex_cache->GetResolvedString(string_idx) == NULL) { 203d1f1bf03e30320e5730790678f470ac8a610b6b6Ian Rogers dex_cache->SetResolvedString(string_idx, string); 204d1f1bf03e30320e5730790678f470ac8a610b6b6Ian Rogers } 205d1f1bf03e30320e5730790678f470ac8a610b6b6Ian Rogers } 206d1f1bf03e30320e5730790678f470ac8a610b6b6Ian Rogers } 207d1f1bf03e30320e5730790678f470ac8a610b6b6Ian Rogers} 208d1f1bf03e30320e5730790678f470ac8a610b6b6Ian Rogers 209357e9be24c17a6bc2ae9fb53f25c73503116101dMathieu Chartiervoid ImageWriter::ComputeEagerResolvedStrings() 210b726dcb581bf72da46527378ccb6889020f0e6e9Ian Rogers SHARED_LOCKS_REQUIRED(Locks::mutator_lock_) { 211b062fdd4cb097fbae69b4bcb479c34d83ecab8caMathieu Chartier // TODO: Check image spaces only? 2121d54e73444e017d3a65234e0f193846f3e27472bIan Rogers gc::Heap* heap = Runtime::Current()->GetHeap(); 213e35517ad40e0713df8420aeb5d3540f146a59f62Mathieu Chartier WriterMutexLock mu(Thread::Current(), *Locks::heap_bitmap_lock_); 214357e9be24c17a6bc2ae9fb53f25c73503116101dMathieu Chartier heap->FlushAllocStack(); 215357e9be24c17a6bc2ae9fb53f25c73503116101dMathieu Chartier heap->GetLiveBitmap()->Walk(ComputeEagerResolvedStringsCallback, this); 216d1f1bf03e30320e5730790678f470ac8a610b6b6Ian Rogers} 217d1f1bf03e30320e5730790678f470ac8a610b6b6Ian Rogers 218ae826983f7903bc0a6bbbe8426bf393fb2f6d747Brian Carlstrombool ImageWriter::IsImageClass(const Class* klass) { 21996391606d8adfc661e1c21703ded1e7a39377a76Brian Carlstrom return compiler_driver_.IsImageClass(ClassHelper(klass).GetDescriptor()); 220ae826983f7903bc0a6bbbe8426bf393fb2f6d747Brian Carlstrom} 221ae826983f7903bc0a6bbbe8426bf393fb2f6d747Brian Carlstrom 222ae826983f7903bc0a6bbbe8426bf393fb2f6d747Brian Carlstromstruct NonImageClasses { 223ae826983f7903bc0a6bbbe8426bf393fb2f6d747Brian Carlstrom ImageWriter* image_writer; 224ae826983f7903bc0a6bbbe8426bf393fb2f6d747Brian Carlstrom std::set<std::string>* non_image_classes; 225ae826983f7903bc0a6bbbe8426bf393fb2f6d747Brian Carlstrom}; 226ae826983f7903bc0a6bbbe8426bf393fb2f6d747Brian Carlstrom 227ae826983f7903bc0a6bbbe8426bf393fb2f6d747Brian Carlstromvoid ImageWriter::PruneNonImageClasses() { 22896391606d8adfc661e1c21703ded1e7a39377a76Brian Carlstrom if (compiler_driver_.GetImageClasses() == NULL) { 229ae826983f7903bc0a6bbbe8426bf393fb2f6d747Brian Carlstrom return; 230ae826983f7903bc0a6bbbe8426bf393fb2f6d747Brian Carlstrom } 231ae826983f7903bc0a6bbbe8426bf393fb2f6d747Brian Carlstrom Runtime* runtime = Runtime::Current(); 232ae826983f7903bc0a6bbbe8426bf393fb2f6d747Brian Carlstrom ClassLinker* class_linker = runtime->GetClassLinker(); 233ae826983f7903bc0a6bbbe8426bf393fb2f6d747Brian Carlstrom 2344b7b9896856171a623d95f86a2676c81c258f408Brian Carlstrom // Make a list of classes we would like to prune. 235ae826983f7903bc0a6bbbe8426bf393fb2f6d747Brian Carlstrom std::set<std::string> non_image_classes; 236ae826983f7903bc0a6bbbe8426bf393fb2f6d747Brian Carlstrom NonImageClasses context; 237ae826983f7903bc0a6bbbe8426bf393fb2f6d747Brian Carlstrom context.image_writer = this; 238ae826983f7903bc0a6bbbe8426bf393fb2f6d747Brian Carlstrom context.non_image_classes = &non_image_classes; 239ae826983f7903bc0a6bbbe8426bf393fb2f6d747Brian Carlstrom class_linker->VisitClasses(NonImageClassesVisitor, &context); 240ae826983f7903bc0a6bbbe8426bf393fb2f6d747Brian Carlstrom 2414b7b9896856171a623d95f86a2676c81c258f408Brian Carlstrom // Remove the undesired classes from the class roots. 242ae826983f7903bc0a6bbbe8426bf393fb2f6d747Brian Carlstrom typedef std::set<std::string>::const_iterator ClassIt; // TODO: C++0x auto 243ae826983f7903bc0a6bbbe8426bf393fb2f6d747Brian Carlstrom for (ClassIt it = non_image_classes.begin(), end = non_image_classes.end(); it != end; ++it) { 244c3b77c7c2971124cbf3b2d9da64e7a8a9a649f2eElliott Hughes class_linker->RemoveClass((*it).c_str(), NULL); 245ae826983f7903bc0a6bbbe8426bf393fb2f6d747Brian Carlstrom } 246ae826983f7903bc0a6bbbe8426bf393fb2f6d747Brian Carlstrom 2474b7b9896856171a623d95f86a2676c81c258f408Brian Carlstrom // Clear references to removed classes from the DexCaches. 24866f19258f9728d4ffe026074d8fd429d639802faMathieu Chartier AbstractMethod* resolution_method = runtime->GetResolutionMethod(); 249ae826983f7903bc0a6bbbe8426bf393fb2f6d747Brian Carlstrom typedef Set::const_iterator CacheIt; // TODO: C++0x auto 250ae826983f7903bc0a6bbbe8426bf393fb2f6d747Brian Carlstrom for (CacheIt it = dex_caches_.begin(), end = dex_caches_.end(); it != end; ++it) { 251ae826983f7903bc0a6bbbe8426bf393fb2f6d747Brian Carlstrom DexCache* dex_cache = *it; 252ae826983f7903bc0a6bbbe8426bf393fb2f6d747Brian Carlstrom for (size_t i = 0; i < dex_cache->NumResolvedTypes(); i++) { 253ae826983f7903bc0a6bbbe8426bf393fb2f6d747Brian Carlstrom Class* klass = dex_cache->GetResolvedType(i); 254ae826983f7903bc0a6bbbe8426bf393fb2f6d747Brian Carlstrom if (klass != NULL && !IsImageClass(klass)) { 255ae826983f7903bc0a6bbbe8426bf393fb2f6d747Brian Carlstrom dex_cache->SetResolvedType(i, NULL); 256ae826983f7903bc0a6bbbe8426bf393fb2f6d747Brian Carlstrom dex_cache->GetInitializedStaticStorage()->Set(i, NULL); 257ae826983f7903bc0a6bbbe8426bf393fb2f6d747Brian Carlstrom } 258ae826983f7903bc0a6bbbe8426bf393fb2f6d747Brian Carlstrom } 259ae826983f7903bc0a6bbbe8426bf393fb2f6d747Brian Carlstrom for (size_t i = 0; i < dex_cache->NumResolvedMethods(); i++) { 26066f19258f9728d4ffe026074d8fd429d639802faMathieu Chartier AbstractMethod* method = dex_cache->GetResolvedMethod(i); 261ae826983f7903bc0a6bbbe8426bf393fb2f6d747Brian Carlstrom if (method != NULL && !IsImageClass(method->GetDeclaringClass())) { 2621984651929744dd603fd082e23eacd877b9bc177Ian Rogers dex_cache->SetResolvedMethod(i, resolution_method); 263ae826983f7903bc0a6bbbe8426bf393fb2f6d747Brian Carlstrom } 264ae826983f7903bc0a6bbbe8426bf393fb2f6d747Brian Carlstrom } 265ae826983f7903bc0a6bbbe8426bf393fb2f6d747Brian Carlstrom for (size_t i = 0; i < dex_cache->NumResolvedFields(); i++) { 266ae826983f7903bc0a6bbbe8426bf393fb2f6d747Brian Carlstrom Field* field = dex_cache->GetResolvedField(i); 267ae826983f7903bc0a6bbbe8426bf393fb2f6d747Brian Carlstrom if (field != NULL && !IsImageClass(field->GetDeclaringClass())) { 268ae826983f7903bc0a6bbbe8426bf393fb2f6d747Brian Carlstrom dex_cache->SetResolvedField(i, NULL); 269ae826983f7903bc0a6bbbe8426bf393fb2f6d747Brian Carlstrom } 270ae826983f7903bc0a6bbbe8426bf393fb2f6d747Brian Carlstrom } 271ae826983f7903bc0a6bbbe8426bf393fb2f6d747Brian Carlstrom } 272ae826983f7903bc0a6bbbe8426bf393fb2f6d747Brian Carlstrom} 273ae826983f7903bc0a6bbbe8426bf393fb2f6d747Brian Carlstrom 274ae826983f7903bc0a6bbbe8426bf393fb2f6d747Brian Carlstrombool ImageWriter::NonImageClassesVisitor(Class* klass, void* arg) { 275ae826983f7903bc0a6bbbe8426bf393fb2f6d747Brian Carlstrom NonImageClasses* context = reinterpret_cast<NonImageClasses*>(arg); 276ae826983f7903bc0a6bbbe8426bf393fb2f6d747Brian Carlstrom if (!context->image_writer->IsImageClass(klass)) { 2776d4d9fcb4f01e287ee29e81cd1c941ee5d11d379Ian Rogers context->non_image_classes->insert(ClassHelper(klass).GetDescriptor()); 278ae826983f7903bc0a6bbbe8426bf393fb2f6d747Brian Carlstrom } 279ae826983f7903bc0a6bbbe8426bf393fb2f6d747Brian Carlstrom return true; 280ae826983f7903bc0a6bbbe8426bf393fb2f6d747Brian Carlstrom} 281ae826983f7903bc0a6bbbe8426bf393fb2f6d747Brian Carlstrom 282357e9be24c17a6bc2ae9fb53f25c73503116101dMathieu Chartiervoid ImageWriter::CheckNonImageClassesRemoved() 283b726dcb581bf72da46527378ccb6889020f0e6e9Ian Rogers SHARED_LOCKS_REQUIRED(Locks::mutator_lock_) { 28496391606d8adfc661e1c21703ded1e7a39377a76Brian Carlstrom if (compiler_driver_.GetImageClasses() == NULL) { 285ae826983f7903bc0a6bbbe8426bf393fb2f6d747Brian Carlstrom return; 286ae826983f7903bc0a6bbbe8426bf393fb2f6d747Brian Carlstrom } 287b062fdd4cb097fbae69b4bcb479c34d83ecab8caMathieu Chartier 2881d54e73444e017d3a65234e0f193846f3e27472bIan Rogers gc::Heap* heap = Runtime::Current()->GetHeap(); 28950b35e2fd1a68cd1240e4a9d9f363e11764957d1Ian Rogers Thread* self = Thread::Current(); 290357e9be24c17a6bc2ae9fb53f25c73503116101dMathieu Chartier { 29150b35e2fd1a68cd1240e4a9d9f363e11764957d1Ian Rogers WriterMutexLock mu(self, *Locks::heap_bitmap_lock_); 292357e9be24c17a6bc2ae9fb53f25c73503116101dMathieu Chartier heap->FlushAllocStack(); 293357e9be24c17a6bc2ae9fb53f25c73503116101dMathieu Chartier } 294357e9be24c17a6bc2ae9fb53f25c73503116101dMathieu Chartier 29550b35e2fd1a68cd1240e4a9d9f363e11764957d1Ian Rogers ReaderMutexLock mu(self, *Locks::heap_bitmap_lock_); 296357e9be24c17a6bc2ae9fb53f25c73503116101dMathieu Chartier heap->GetLiveBitmap()->Walk(CheckNonImageClassesRemovedCallback, this); 297ae826983f7903bc0a6bbbe8426bf393fb2f6d747Brian Carlstrom} 298ae826983f7903bc0a6bbbe8426bf393fb2f6d747Brian Carlstrom 299ae826983f7903bc0a6bbbe8426bf393fb2f6d747Brian Carlstromvoid ImageWriter::CheckNonImageClassesRemovedCallback(Object* obj, void* arg) { 300ae826983f7903bc0a6bbbe8426bf393fb2f6d747Brian Carlstrom ImageWriter* image_writer = reinterpret_cast<ImageWriter*>(arg); 301ae826983f7903bc0a6bbbe8426bf393fb2f6d747Brian Carlstrom if (!obj->IsClass()) { 302ae826983f7903bc0a6bbbe8426bf393fb2f6d747Brian Carlstrom return; 303ae826983f7903bc0a6bbbe8426bf393fb2f6d747Brian Carlstrom } 304ae826983f7903bc0a6bbbe8426bf393fb2f6d747Brian Carlstrom Class* klass = obj->AsClass(); 305c3b77c7c2971124cbf3b2d9da64e7a8a9a649f2eElliott Hughes if (!image_writer->IsImageClass(klass)) { 306c3b77c7c2971124cbf3b2d9da64e7a8a9a649f2eElliott Hughes image_writer->DumpImageClasses(); 307c3b77c7c2971124cbf3b2d9da64e7a8a9a649f2eElliott Hughes CHECK(image_writer->IsImageClass(klass)) << ClassHelper(klass).GetDescriptor() 308c3b77c7c2971124cbf3b2d9da64e7a8a9a649f2eElliott Hughes << " " << PrettyDescriptor(klass); 309c3b77c7c2971124cbf3b2d9da64e7a8a9a649f2eElliott Hughes } 310c3b77c7c2971124cbf3b2d9da64e7a8a9a649f2eElliott Hughes} 311c3b77c7c2971124cbf3b2d9da64e7a8a9a649f2eElliott Hughes 312c3b77c7c2971124cbf3b2d9da64e7a8a9a649f2eElliott Hughesvoid ImageWriter::DumpImageClasses() { 31396391606d8adfc661e1c21703ded1e7a39377a76Brian Carlstrom CompilerDriver::DescriptorSet* image_classes = compiler_driver_.GetImageClasses(); 31496391606d8adfc661e1c21703ded1e7a39377a76Brian Carlstrom CHECK(image_classes != NULL); 315c3b77c7c2971124cbf3b2d9da64e7a8a9a649f2eElliott Hughes typedef std::set<std::string>::const_iterator It; // TODO: C++0x auto 31696391606d8adfc661e1c21703ded1e7a39377a76Brian Carlstrom for (It it = image_classes->begin(), end = image_classes->end(); it != end; ++it) { 317c3b77c7c2971124cbf3b2d9da64e7a8a9a649f2eElliott Hughes LOG(INFO) << " " << *it; 318c3b77c7c2971124cbf3b2d9da64e7a8a9a649f2eElliott Hughes } 319ae826983f7903bc0a6bbbe8426bf393fb2f6d747Brian Carlstrom} 320ae826983f7903bc0a6bbbe8426bf393fb2f6d747Brian Carlstrom 32178128a63b2615744760b7f8ab83df9764a5d4a95Brian Carlstromvoid ImageWriter::CalculateNewObjectOffsetsCallback(Object* obj, void* arg) { 3224e777d4e005375a5a41c4ec3c55783c09565df0dBrian Carlstrom DCHECK(obj != NULL); 3234e777d4e005375a5a41c4ec3c55783c09565df0dBrian Carlstrom DCHECK(arg != NULL); 3244e777d4e005375a5a41c4ec3c55783c09565df0dBrian Carlstrom ImageWriter* image_writer = reinterpret_cast<ImageWriter*>(arg); 325c74255fffb035001304c9a058a2e730a5a1a9604Brian Carlstrom 326c74255fffb035001304c9a058a2e730a5a1a9604Brian Carlstrom // if it is a string, we want to intern it if its not interned. 327dbb4079eb1e7d7738c81a97c8dd2550885c1093aElliott Hughes if (obj->GetClass()->IsStringClass()) { 328c74255fffb035001304c9a058a2e730a5a1a9604Brian Carlstrom // we must be an interned string that was forward referenced and already assigned 329b3bd5f07884f5a1f2b84224363b1372d7c28d447Elliott Hughes if (image_writer->IsImageOffsetAssigned(obj)) { 330c74255fffb035001304c9a058a2e730a5a1a9604Brian Carlstrom DCHECK_EQ(obj, obj->AsString()->Intern()); 331c74255fffb035001304c9a058a2e730a5a1a9604Brian Carlstrom return; 332c74255fffb035001304c9a058a2e730a5a1a9604Brian Carlstrom } 3331f5393447b9f45be7918042d9ee7b521376de866Ian Rogers SirtRef<String> interned(Thread::Current(), obj->AsString()->Intern()); 33440381fb9dc4b4cf274f1e58b2cdf4396202c6189Brian Carlstrom if (obj != interned.get()) { 335b3bd5f07884f5a1f2b84224363b1372d7c28d447Elliott Hughes if (!image_writer->IsImageOffsetAssigned(interned.get())) { 336c74255fffb035001304c9a058a2e730a5a1a9604Brian Carlstrom // interned obj is after us, allocate its location early 33740381fb9dc4b4cf274f1e58b2cdf4396202c6189Brian Carlstrom image_writer->AssignImageOffset(interned.get()); 338c74255fffb035001304c9a058a2e730a5a1a9604Brian Carlstrom } 339c74255fffb035001304c9a058a2e730a5a1a9604Brian Carlstrom // point those looking for this object to the interned version. 340b3bd5f07884f5a1f2b84224363b1372d7c28d447Elliott Hughes image_writer->SetImageOffset(obj, image_writer->GetImageOffset(interned.get())); 341c74255fffb035001304c9a058a2e730a5a1a9604Brian Carlstrom return; 342c74255fffb035001304c9a058a2e730a5a1a9604Brian Carlstrom } 343c74255fffb035001304c9a058a2e730a5a1a9604Brian Carlstrom // else (obj == interned), nothing to do but fall through to the normal case 344c74255fffb035001304c9a058a2e730a5a1a9604Brian Carlstrom } 345c74255fffb035001304c9a058a2e730a5a1a9604Brian Carlstrom 346c74255fffb035001304c9a058a2e730a5a1a9604Brian Carlstrom image_writer->AssignImageOffset(obj); 3474e777d4e005375a5a41c4ec3c55783c09565df0dBrian Carlstrom} 3484e777d4e005375a5a41c4ec3c55783c09565df0dBrian Carlstrom 349e24fa61603a60ade3797e4a0c8b3fccb346cb048Brian CarlstromObjectArray<Object>* ImageWriter::CreateImageRoots() const { 350161928613d3f097108319de60494fab1aab8d48aBrian Carlstrom Runtime* runtime = Runtime::Current(); 351161928613d3f097108319de60494fab1aab8d48aBrian Carlstrom ClassLinker* class_linker = runtime->GetClassLinker(); 352161928613d3f097108319de60494fab1aab8d48aBrian Carlstrom Class* object_array_class = class_linker->FindSystemClass("[Ljava/lang/Object;"); 35350b35e2fd1a68cd1240e4a9d9f363e11764957d1Ian Rogers Thread* self = Thread::Current(); 35458ae9416e197ae68ed12ed43d87407d4dfb15093Brian Carlstrom 35558ae9416e197ae68ed12ed43d87407d4dfb15093Brian Carlstrom // build an Object[] of all the DexCaches used in the source_space_ 35650b35e2fd1a68cd1240e4a9d9f363e11764957d1Ian Rogers ObjectArray<Object>* dex_caches = ObjectArray<Object>::Alloc(self, object_array_class, 357ae826983f7903bc0a6bbbe8426bf393fb2f6d747Brian Carlstrom dex_caches_.size()); 358ae826983f7903bc0a6bbbe8426bf393fb2f6d747Brian Carlstrom int i = 0; 359ae826983f7903bc0a6bbbe8426bf393fb2f6d747Brian Carlstrom typedef Set::const_iterator It; // TODO: C++0x auto 360ae826983f7903bc0a6bbbe8426bf393fb2f6d747Brian Carlstrom for (It it = dex_caches_.begin(), end = dex_caches_.end(); it != end; ++it, ++i) { 361ae826983f7903bc0a6bbbe8426bf393fb2f6d747Brian Carlstrom dex_caches->Set(i, *it); 36258ae9416e197ae68ed12ed43d87407d4dfb15093Brian Carlstrom } 36358ae9416e197ae68ed12ed43d87407d4dfb15093Brian Carlstrom 36458ae9416e197ae68ed12ed43d87407d4dfb15093Brian Carlstrom // build an Object[] of the roots needed to restore the runtime 3651f5393447b9f45be7918042d9ee7b521376de866Ian Rogers SirtRef<ObjectArray<Object> > 36650b35e2fd1a68cd1240e4a9d9f363e11764957d1Ian Rogers image_roots(self, 36750b35e2fd1a68cd1240e4a9d9f363e11764957d1Ian Rogers ObjectArray<Object>::Alloc(self, object_array_class, 36850b35e2fd1a68cd1240e4a9d9f363e11764957d1Ian Rogers ImageHeader::kImageRootsMax)); 3691984651929744dd603fd082e23eacd877b9bc177Ian Rogers image_roots->Set(ImageHeader::kResolutionMethod, runtime->GetResolutionMethod()); 3704f0d07c783afef89703dce32c94440fc8621a29bIan Rogers image_roots->Set(ImageHeader::kCalleeSaveMethod, 3714f0d07c783afef89703dce32c94440fc8621a29bIan Rogers runtime->GetCalleeSaveMethod(Runtime::kSaveAll)); 3724f0d07c783afef89703dce32c94440fc8621a29bIan Rogers image_roots->Set(ImageHeader::kRefsOnlySaveMethod, 3734f0d07c783afef89703dce32c94440fc8621a29bIan Rogers runtime->GetCalleeSaveMethod(Runtime::kRefsOnly)); 3744f0d07c783afef89703dce32c94440fc8621a29bIan Rogers image_roots->Set(ImageHeader::kRefsAndArgsSaveMethod, 3754f0d07c783afef89703dce32c94440fc8621a29bIan Rogers runtime->GetCalleeSaveMethod(Runtime::kRefsAndArgs)); 376e24fa61603a60ade3797e4a0c8b3fccb346cb048Brian Carlstrom image_roots->Set(ImageHeader::kOatLocation, 37750b35e2fd1a68cd1240e4a9d9f363e11764957d1Ian Rogers String::AllocFromModifiedUtf8(self, oat_file_->GetLocation().c_str())); 37858ae9416e197ae68ed12ed43d87407d4dfb15093Brian Carlstrom image_roots->Set(ImageHeader::kDexCaches, 37958ae9416e197ae68ed12ed43d87407d4dfb15093Brian Carlstrom dex_caches); 38034f426c49ac2de8cea70acef6b9ecdd8e62209d2Brian Carlstrom image_roots->Set(ImageHeader::kClassRoots, 38134f426c49ac2de8cea70acef6b9ecdd8e62209d2Brian Carlstrom class_linker->GetClassRoots()); 382e24fa61603a60ade3797e4a0c8b3fccb346cb048Brian Carlstrom for (int i = 0; i < ImageHeader::kImageRootsMax; i++) { 383e24fa61603a60ade3797e4a0c8b3fccb346cb048Brian Carlstrom CHECK(image_roots->Get(i) != NULL); 384e24fa61603a60ade3797e4a0c8b3fccb346cb048Brian Carlstrom } 38540381fb9dc4b4cf274f1e58b2cdf4396202c6189Brian Carlstrom return image_roots.get(); 386161928613d3f097108319de60494fab1aab8d48aBrian Carlstrom} 387161928613d3f097108319de60494fab1aab8d48aBrian Carlstrom 388700c8d31733534a3d978b75a03f6f7e177dc7e81Brian Carlstromvoid ImageWriter::CalculateNewObjectOffsets(size_t oat_loaded_size, size_t oat_data_offset) { 389700c8d31733534a3d978b75a03f6f7e177dc7e81Brian Carlstrom CHECK_NE(0U, oat_loaded_size); 3901f5393447b9f45be7918042d9ee7b521376de866Ian Rogers Thread* self = Thread::Current(); 3911f5393447b9f45be7918042d9ee7b521376de866Ian Rogers SirtRef<ObjectArray<Object> > image_roots(self, CreateImageRoots()); 392161928613d3f097108319de60494fab1aab8d48aBrian Carlstrom 3931d54e73444e017d3a65234e0f193846f3e27472bIan Rogers gc::Heap* heap = Runtime::Current()->GetHeap(); 3941d54e73444e017d3a65234e0f193846f3e27472bIan Rogers const std::vector<gc::space::ContinuousSpace*>& spaces = heap->GetContinuousSpaces(); 395b062fdd4cb097fbae69b4bcb479c34d83ecab8caMathieu Chartier DCHECK(!spaces.empty()); 39630fab40ee5a07af6b8c3b6b0e9438071695a57f4Ian Rogers DCHECK_EQ(0U, image_end_); 397a663ea5de4c9ab6b1510fdebd6d8eca77ba699aeBrian Carlstrom 398161928613d3f097108319de60494fab1aab8d48aBrian Carlstrom // leave space for the header, but do not write it yet, we need to 399161928613d3f097108319de60494fab1aab8d48aBrian Carlstrom // know where image_roots is going to end up 40030fab40ee5a07af6b8c3b6b0e9438071695a57f4Ian Rogers image_end_ += RoundUp(sizeof(ImageHeader), 8); // 64-bit-alignment 401a663ea5de4c9ab6b1510fdebd6d8eca77ba699aeBrian Carlstrom 402357e9be24c17a6bc2ae9fb53f25c73503116101dMathieu Chartier { 403e35517ad40e0713df8420aeb5d3540f146a59f62Mathieu Chartier WriterMutexLock mu(self, *Locks::heap_bitmap_lock_); 404357e9be24c17a6bc2ae9fb53f25c73503116101dMathieu Chartier heap->FlushAllocStack(); 40566f19258f9728d4ffe026074d8fd429d639802faMathieu Chartier // TODO: Image spaces only? 40666f19258f9728d4ffe026074d8fd429d639802faMathieu Chartier // TODO: Add InOrderWalk to heap bitmap. 4071f5393447b9f45be7918042d9ee7b521376de866Ian Rogers const char* old = self->StartAssertNoThreadSuspension("ImageWriter"); 408e0f0cb3d855cb5e926452b5e1ec8457adc4e454eMathieu Chartier DCHECK(heap->GetLargeObjectsSpace()->GetLiveObjects()->IsEmpty()); 4091d54e73444e017d3a65234e0f193846f3e27472bIan Rogers // TODO: C++0x auto 4101d54e73444e017d3a65234e0f193846f3e27472bIan Rogers typedef std::vector<gc::space::ContinuousSpace*>::const_iterator It; 4111d54e73444e017d3a65234e0f193846f3e27472bIan Rogers for (It it = spaces.begin(), end = spaces.end(); it != end; ++it) { 4121d54e73444e017d3a65234e0f193846f3e27472bIan Rogers gc::space::ContinuousSpace* space = *it; 4131d54e73444e017d3a65234e0f193846f3e27472bIan Rogers space->GetLiveBitmap()->InOrderWalk(CalculateNewObjectOffsetsCallback, this); 414357e9be24c17a6bc2ae9fb53f25c73503116101dMathieu Chartier DCHECK_LT(image_end_, image_->Size()); 415357e9be24c17a6bc2ae9fb53f25c73503116101dMathieu Chartier } 4161f5393447b9f45be7918042d9ee7b521376de866Ian Rogers self->EndAssertNoThreadSuspension(old); 417b062fdd4cb097fbae69b4bcb479c34d83ecab8caMathieu Chartier } 418a663ea5de4c9ab6b1510fdebd6d8eca77ba699aeBrian Carlstrom 419700c8d31733534a3d978b75a03f6f7e177dc7e81Brian Carlstrom const byte* oat_file_begin = image_begin_ + RoundUp(image_end_, kPageSize); 420700c8d31733534a3d978b75a03f6f7e177dc7e81Brian Carlstrom const byte* oat_file_end = oat_file_begin + oat_loaded_size; 421700c8d31733534a3d978b75a03f6f7e177dc7e81Brian Carlstrom oat_data_begin_ = oat_file_begin + oat_data_offset; 422700c8d31733534a3d978b75a03f6f7e177dc7e81Brian Carlstrom const byte* oat_data_end = oat_data_begin_ + oat_file_->Size(); 423e24fa61603a60ade3797e4a0c8b3fccb346cb048Brian Carlstrom 424161928613d3f097108319de60494fab1aab8d48aBrian Carlstrom // return to write header at start of image with future location of image_roots 42530fab40ee5a07af6b8c3b6b0e9438071695a57f4Ian Rogers ImageHeader image_header(reinterpret_cast<uint32_t>(image_begin_), 42640381fb9dc4b4cf274f1e58b2cdf4396202c6189Brian Carlstrom reinterpret_cast<uint32_t>(GetImageAddress(image_roots.get())), 427e24fa61603a60ade3797e4a0c8b3fccb346cb048Brian Carlstrom oat_file_->GetOatHeader().GetChecksum(), 428700c8d31733534a3d978b75a03f6f7e177dc7e81Brian Carlstrom reinterpret_cast<uint32_t>(oat_file_begin), 429700c8d31733534a3d978b75a03f6f7e177dc7e81Brian Carlstrom reinterpret_cast<uint32_t>(oat_data_begin_), 430700c8d31733534a3d978b75a03f6f7e177dc7e81Brian Carlstrom reinterpret_cast<uint32_t>(oat_data_end), 431700c8d31733534a3d978b75a03f6f7e177dc7e81Brian Carlstrom reinterpret_cast<uint32_t>(oat_file_end)); 43230fab40ee5a07af6b8c3b6b0e9438071695a57f4Ian Rogers memcpy(image_->Begin(), &image_header, sizeof(image_header)); 433700c8d31733534a3d978b75a03f6f7e177dc7e81Brian Carlstrom 434700c8d31733534a3d978b75a03f6f7e177dc7e81Brian Carlstrom // Note that image_end_ is left at end of used space 435db4d54081f09abcbe97ffdf615874f2809a9e777Brian Carlstrom} 436db4d54081f09abcbe97ffdf615874f2809a9e777Brian Carlstrom 437357e9be24c17a6bc2ae9fb53f25c73503116101dMathieu Chartiervoid ImageWriter::CopyAndFixupObjects() 438b726dcb581bf72da46527378ccb6889020f0e6e9Ian Rogers SHARED_LOCKS_REQUIRED(Locks::mutator_lock_) { 43950b35e2fd1a68cd1240e4a9d9f363e11764957d1Ian Rogers Thread* self = Thread::Current(); 44050b35e2fd1a68cd1240e4a9d9f363e11764957d1Ian Rogers const char* old_cause = self->StartAssertNoThreadSuspension("ImageWriter"); 4411d54e73444e017d3a65234e0f193846f3e27472bIan Rogers gc::Heap* heap = Runtime::Current()->GetHeap(); 4420cfe1fb7060576d047f7f894fc0d8b87de84fcabIan Rogers // TODO: heap validation can't handle this fix up pass 443b3bd5f07884f5a1f2b84224363b1372d7c28d447Elliott Hughes heap->DisableObjectValidation(); 444b062fdd4cb097fbae69b4bcb479c34d83ecab8caMathieu Chartier // TODO: Image spaces only? 445e35517ad40e0713df8420aeb5d3540f146a59f62Mathieu Chartier WriterMutexLock mu(self, *Locks::heap_bitmap_lock_); 446357e9be24c17a6bc2ae9fb53f25c73503116101dMathieu Chartier heap->FlushAllocStack(); 447b062fdd4cb097fbae69b4bcb479c34d83ecab8caMathieu Chartier heap->GetLiveBitmap()->Walk(CopyAndFixupObjectsCallback, this); 44850b35e2fd1a68cd1240e4a9d9f363e11764957d1Ian Rogers self->EndAssertNoThreadSuspension(old_cause); 449db4d54081f09abcbe97ffdf615874f2809a9e777Brian Carlstrom} 450db4d54081f09abcbe97ffdf615874f2809a9e777Brian Carlstrom 45178128a63b2615744760b7f8ab83df9764a5d4a95Brian Carlstromvoid ImageWriter::CopyAndFixupObjectsCallback(Object* object, void* arg) { 4524873d465a1eb6dfbdeddb085c81239d39db60c42Brian Carlstrom DCHECK(object != NULL); 453db4d54081f09abcbe97ffdf615874f2809a9e777Brian Carlstrom DCHECK(arg != NULL); 4544873d465a1eb6dfbdeddb085c81239d39db60c42Brian Carlstrom const Object* obj = object; 455db4d54081f09abcbe97ffdf615874f2809a9e777Brian Carlstrom ImageWriter* image_writer = reinterpret_cast<ImageWriter*>(arg); 456d8ddfd5eadde1d5f53ef1419f529c799233eaa62Elliott Hughes 45769b15fb098162f19a4c20e6dccdcead04d9c77f0Brian Carlstrom // see GetLocalAddress for similar computation 458db4d54081f09abcbe97ffdf615874f2809a9e777Brian Carlstrom size_t offset = image_writer->GetImageOffset(obj); 45930fab40ee5a07af6b8c3b6b0e9438071695a57f4Ian Rogers byte* dst = image_writer->image_->Begin() + offset; 4604873d465a1eb6dfbdeddb085c81239d39db60c42Brian Carlstrom const byte* src = reinterpret_cast<const byte*>(obj); 46104b63fd60de0d9e6f07763511a1a9622accfeed7Elliott Hughes size_t n = obj->SizeOf(); 46230fab40ee5a07af6b8c3b6b0e9438071695a57f4Ian Rogers DCHECK_LT(offset + n, image_writer->image_->Size()); 463db4d54081f09abcbe97ffdf615874f2809a9e777Brian Carlstrom memcpy(dst, src, n); 464db4d54081f09abcbe97ffdf615874f2809a9e777Brian Carlstrom Object* copy = reinterpret_cast<Object*>(dst); 4652dd0e2cea360bc9206eb88ecc40d259e796c239dIan Rogers copy->SetField32(Object::MonitorOffset(), 0, false); // We may have inflated the lock during compilation. 466db4d54081f09abcbe97ffdf615874f2809a9e777Brian Carlstrom image_writer->FixupObject(obj, copy); 467db4d54081f09abcbe97ffdf615874f2809a9e777Brian Carlstrom} 468db4d54081f09abcbe97ffdf615874f2809a9e777Brian Carlstrom 4694873d465a1eb6dfbdeddb085c81239d39db60c42Brian Carlstromvoid ImageWriter::FixupObject(const Object* orig, Object* copy) { 470db4d54081f09abcbe97ffdf615874f2809a9e777Brian Carlstrom DCHECK(orig != NULL); 471db4d54081f09abcbe97ffdf615874f2809a9e777Brian Carlstrom DCHECK(copy != NULL); 4720cfe1fb7060576d047f7f894fc0d8b87de84fcabIan Rogers copy->SetClass(down_cast<Class*>(GetImageAddress(orig->GetClass()))); 4739cff8e13d41825c4f3f0127af061e94b06114fc8Brian Carlstrom // TODO: special case init of pointers to malloc data (or removal of these pointers) 4749cff8e13d41825c4f3f0127af061e94b06114fc8Brian Carlstrom if (orig->IsClass()) { 4759cff8e13d41825c4f3f0127af061e94b06114fc8Brian Carlstrom FixupClass(orig->AsClass(), down_cast<Class*>(copy)); 4769cff8e13d41825c4f3f0127af061e94b06114fc8Brian Carlstrom } else if (orig->IsObjectArray()) { 477db4d54081f09abcbe97ffdf615874f2809a9e777Brian Carlstrom FixupObjectArray(orig->AsObjectArray<Object>(), down_cast<ObjectArray<Object>*>(copy)); 478161928613d3f097108319de60494fab1aab8d48aBrian Carlstrom } else if (orig->IsMethod()) { 47966f19258f9728d4ffe026074d8fd429d639802faMathieu Chartier FixupMethod(orig->AsMethod(), down_cast<AbstractMethod*>(copy)); 480db4d54081f09abcbe97ffdf615874f2809a9e777Brian Carlstrom } else { 481db4d54081f09abcbe97ffdf615874f2809a9e777Brian Carlstrom FixupInstanceFields(orig, copy); 482db4d54081f09abcbe97ffdf615874f2809a9e777Brian Carlstrom } 483db4d54081f09abcbe97ffdf615874f2809a9e777Brian Carlstrom} 484db4d54081f09abcbe97ffdf615874f2809a9e777Brian Carlstrom 4854873d465a1eb6dfbdeddb085c81239d39db60c42Brian Carlstromvoid ImageWriter::FixupClass(const Class* orig, Class* copy) { 4869cff8e13d41825c4f3f0127af061e94b06114fc8Brian Carlstrom FixupInstanceFields(orig, copy); 4874873d465a1eb6dfbdeddb085c81239d39db60c42Brian Carlstrom FixupStaticFields(orig, copy); 4889cff8e13d41825c4f3f0127af061e94b06114fc8Brian Carlstrom} 4899cff8e13d41825c4f3f0127af061e94b06114fc8Brian Carlstrom 49066f19258f9728d4ffe026074d8fd429d639802faMathieu Chartiervoid ImageWriter::FixupMethod(const AbstractMethod* orig, AbstractMethod* copy) { 491a663ea5de4c9ab6b1510fdebd6d8eca77ba699aeBrian Carlstrom FixupInstanceFields(orig, copy); 492e24fa61603a60ade3797e4a0c8b3fccb346cb048Brian Carlstrom 49374180cad94848107cf297d37e72437c5a6eecf1bJeff Hao // OatWriter replaces the code_ with an offset value. 49430fab40ee5a07af6b8c3b6b0e9438071695a57f4Ian Rogers // Here we readjust to a pointer relative to oat_begin_ 4953320cf46afd082398aa401b246e6f301cebdf64dBrian Carlstrom if (orig->IsAbstract()) { 49679fe539587d4c09244172d0168eeed0ec9770466Jeff Hao // Code for abstract methods is set to the abstract method error stub when we load the image. 497aa4a793d7175ceeb1efa215c2af5e52e42b22f0aJeff Hao copy->SetEntryPointFromCompiledCode(NULL); 4980aba0ba155bef7346bde19e53581200b89ae8a7aJeff Hao copy->SetEntryPointFromInterpreter(reinterpret_cast<EntryPointFromInterpreter*> 4990aba0ba155bef7346bde19e53581200b89ae8a7aJeff Hao (GetOatAddress(interpreter_to_interpreter_entry_offset_))); 5003320cf46afd082398aa401b246e6f301cebdf64dBrian Carlstrom return; 5010aba0ba155bef7346bde19e53581200b89ae8a7aJeff Hao } else { 5020aba0ba155bef7346bde19e53581200b89ae8a7aJeff Hao copy->SetEntryPointFromInterpreter(reinterpret_cast<EntryPointFromInterpreter*> 5030aba0ba155bef7346bde19e53581200b89ae8a7aJeff Hao (GetOatAddress(interpreter_to_quick_entry_offset_))); 5043320cf46afd082398aa401b246e6f301cebdf64dBrian Carlstrom } 5053320cf46afd082398aa401b246e6f301cebdf64dBrian Carlstrom 5061984651929744dd603fd082e23eacd877b9bc177Ian Rogers if (orig == Runtime::Current()->GetResolutionMethod()) { 5070aba0ba155bef7346bde19e53581200b89ae8a7aJeff Hao#if defined(ART_USE_PORTABLE_COMPILER) 5080aba0ba155bef7346bde19e53581200b89ae8a7aJeff Hao copy->SetEntryPointFromCompiledCode(GetOatAddress(portable_resolution_trampoline_offset_)); 5090aba0ba155bef7346bde19e53581200b89ae8a7aJeff Hao#else 5100aba0ba155bef7346bde19e53581200b89ae8a7aJeff Hao copy->SetEntryPointFromCompiledCode(GetOatAddress(quick_resolution_trampoline_offset_)); 5110aba0ba155bef7346bde19e53581200b89ae8a7aJeff Hao#endif 5121984651929744dd603fd082e23eacd877b9bc177Ian Rogers return; 5131984651929744dd603fd082e23eacd877b9bc177Ian Rogers } 5141984651929744dd603fd082e23eacd877b9bc177Ian Rogers 5150aba0ba155bef7346bde19e53581200b89ae8a7aJeff Hao // Use original code if it exists. Otherwise, set the code pointer to the resolution trampoline. 5160aba0ba155bef7346bde19e53581200b89ae8a7aJeff Hao const byte* code = GetOatAddress(orig->GetOatCodeOffset()); 5170aba0ba155bef7346bde19e53581200b89ae8a7aJeff Hao if (code != NULL) { 5180aba0ba155bef7346bde19e53581200b89ae8a7aJeff Hao copy->SetEntryPointFromCompiledCode(code); 5190aba0ba155bef7346bde19e53581200b89ae8a7aJeff Hao } else { 5200aba0ba155bef7346bde19e53581200b89ae8a7aJeff Hao#if defined(ART_USE_PORTABLE_COMPILER) 5210aba0ba155bef7346bde19e53581200b89ae8a7aJeff Hao copy->SetEntryPointFromCompiledCode(GetOatAddress(portable_resolution_trampoline_offset_)); 5220aba0ba155bef7346bde19e53581200b89ae8a7aJeff Hao#else 5230aba0ba155bef7346bde19e53581200b89ae8a7aJeff Hao copy->SetEntryPointFromCompiledCode(GetOatAddress(quick_resolution_trampoline_offset_)); 5240aba0ba155bef7346bde19e53581200b89ae8a7aJeff Hao#endif 5250aba0ba155bef7346bde19e53581200b89ae8a7aJeff Hao } 5263320cf46afd082398aa401b246e6f301cebdf64dBrian Carlstrom 527161928613d3f097108319de60494fab1aab8d48aBrian Carlstrom if (orig->IsNative()) { 52879fe539587d4c09244172d0168eeed0ec9770466Jeff Hao // The native method's pointer is set to a stub to lookup via dlsym when we load the image. 5293320cf46afd082398aa401b246e6f301cebdf64dBrian Carlstrom // Note this is not the code_ pointer, that is handled above. 53079fe539587d4c09244172d0168eeed0ec9770466Jeff Hao copy->SetNativeMethod(NULL); 531161928613d3f097108319de60494fab1aab8d48aBrian Carlstrom } else { 5323320cf46afd082398aa401b246e6f301cebdf64dBrian Carlstrom // normal (non-abstract non-native) methods have mapping tables to relocate 5333320cf46afd082398aa401b246e6f301cebdf64dBrian Carlstrom uint32_t mapping_table_off = orig->GetOatMappingTableOffset(); 534ae826983f7903bc0a6bbbe8426bf393fb2f6d747Brian Carlstrom const byte* mapping_table = GetOatAddress(mapping_table_off); 5352dd0e2cea360bc9206eb88ecc40d259e796c239dIan Rogers copy->SetMappingTable(reinterpret_cast<const uint32_t*>(mapping_table)); 5363320cf46afd082398aa401b246e6f301cebdf64dBrian Carlstrom 5373320cf46afd082398aa401b246e6f301cebdf64dBrian Carlstrom uint32_t vmap_table_offset = orig->GetOatVmapTableOffset(); 538ae826983f7903bc0a6bbbe8426bf393fb2f6d747Brian Carlstrom const byte* vmap_table = GetOatAddress(vmap_table_offset); 5392dd0e2cea360bc9206eb88ecc40d259e796c239dIan Rogers copy->SetVmapTable(reinterpret_cast<const uint16_t*>(vmap_table)); 540e7d856b911222aa000ca2be0f8f63f5b292141c3Brian Carlstrom 5410c7abda482f53db3d153c073d1c7a145f84e0626Ian Rogers uint32_t native_gc_map_offset = orig->GetOatNativeGcMapOffset(); 5420c7abda482f53db3d153c073d1c7a145f84e0626Ian Rogers const byte* native_gc_map = GetOatAddress(native_gc_map_offset); 5432dd0e2cea360bc9206eb88ecc40d259e796c239dIan Rogers copy->SetNativeGcMap(reinterpret_cast<const uint8_t*>(native_gc_map)); 544161928613d3f097108319de60494fab1aab8d48aBrian Carlstrom } 545a663ea5de4c9ab6b1510fdebd6d8eca77ba699aeBrian Carlstrom} 546a663ea5de4c9ab6b1510fdebd6d8eca77ba699aeBrian Carlstrom 5474873d465a1eb6dfbdeddb085c81239d39db60c42Brian Carlstromvoid ImageWriter::FixupObjectArray(const ObjectArray<Object>* orig, ObjectArray<Object>* copy) { 548d8ddfd5eadde1d5f53ef1419f529c799233eaa62Elliott Hughes for (int32_t i = 0; i < orig->GetLength(); ++i) { 549db4d54081f09abcbe97ffdf615874f2809a9e777Brian Carlstrom const Object* element = orig->Get(i); 550e35517ad40e0713df8420aeb5d3540f146a59f62Mathieu Chartier copy->SetPtrWithoutChecks(i, GetImageAddress(element)); 551db4d54081f09abcbe97ffdf615874f2809a9e777Brian Carlstrom } 552db4d54081f09abcbe97ffdf615874f2809a9e777Brian Carlstrom} 553db4d54081f09abcbe97ffdf615874f2809a9e777Brian Carlstrom 5544873d465a1eb6dfbdeddb085c81239d39db60c42Brian Carlstromvoid ImageWriter::FixupInstanceFields(const Object* orig, Object* copy) { 5554873d465a1eb6dfbdeddb085c81239d39db60c42Brian Carlstrom DCHECK(orig != NULL); 5564873d465a1eb6dfbdeddb085c81239d39db60c42Brian Carlstrom DCHECK(copy != NULL); 5574873d465a1eb6dfbdeddb085c81239d39db60c42Brian Carlstrom Class* klass = orig->GetClass(); 5584873d465a1eb6dfbdeddb085c81239d39db60c42Brian Carlstrom DCHECK(klass != NULL); 5594873d465a1eb6dfbdeddb085c81239d39db60c42Brian Carlstrom FixupFields(orig, 5604873d465a1eb6dfbdeddb085c81239d39db60c42Brian Carlstrom copy, 5614873d465a1eb6dfbdeddb085c81239d39db60c42Brian Carlstrom klass->GetReferenceInstanceOffsets(), 5624873d465a1eb6dfbdeddb085c81239d39db60c42Brian Carlstrom false); 5634873d465a1eb6dfbdeddb085c81239d39db60c42Brian Carlstrom} 5644873d465a1eb6dfbdeddb085c81239d39db60c42Brian Carlstrom 5654873d465a1eb6dfbdeddb085c81239d39db60c42Brian Carlstromvoid ImageWriter::FixupStaticFields(const Class* orig, Class* copy) { 5664873d465a1eb6dfbdeddb085c81239d39db60c42Brian Carlstrom DCHECK(orig != NULL); 5674873d465a1eb6dfbdeddb085c81239d39db60c42Brian Carlstrom DCHECK(copy != NULL); 5684873d465a1eb6dfbdeddb085c81239d39db60c42Brian Carlstrom FixupFields(orig, 5694873d465a1eb6dfbdeddb085c81239d39db60c42Brian Carlstrom copy, 5704873d465a1eb6dfbdeddb085c81239d39db60c42Brian Carlstrom orig->GetReferenceStaticOffsets(), 5714873d465a1eb6dfbdeddb085c81239d39db60c42Brian Carlstrom true); 5724873d465a1eb6dfbdeddb085c81239d39db60c42Brian Carlstrom} 5734873d465a1eb6dfbdeddb085c81239d39db60c42Brian Carlstrom 5744873d465a1eb6dfbdeddb085c81239d39db60c42Brian Carlstromvoid ImageWriter::FixupFields(const Object* orig, 5754873d465a1eb6dfbdeddb085c81239d39db60c42Brian Carlstrom Object* copy, 5764873d465a1eb6dfbdeddb085c81239d39db60c42Brian Carlstrom uint32_t ref_offsets, 5774873d465a1eb6dfbdeddb085c81239d39db60c42Brian Carlstrom bool is_static) { 578db4d54081f09abcbe97ffdf615874f2809a9e777Brian Carlstrom if (ref_offsets != CLASS_WALK_SUPER) { 579db4d54081f09abcbe97ffdf615874f2809a9e777Brian Carlstrom // Found a reference offset bitmap. Fixup the specified offsets. 580db4d54081f09abcbe97ffdf615874f2809a9e777Brian Carlstrom while (ref_offsets != 0) { 581db4d54081f09abcbe97ffdf615874f2809a9e777Brian Carlstrom size_t right_shift = CLZ(ref_offsets); 5820cfe1fb7060576d047f7f894fc0d8b87de84fcabIan Rogers MemberOffset byte_offset = CLASS_OFFSET_FROM_CLZ(right_shift); 5830cfe1fb7060576d047f7f894fc0d8b87de84fcabIan Rogers const Object* ref = orig->GetFieldObject<const Object*>(byte_offset, false); 584e35517ad40e0713df8420aeb5d3540f146a59f62Mathieu Chartier // Use SetFieldPtr to avoid card marking since we are writing to the image. 585e35517ad40e0713df8420aeb5d3540f146a59f62Mathieu Chartier copy->SetFieldPtr(byte_offset, GetImageAddress(ref), false); 586db4d54081f09abcbe97ffdf615874f2809a9e777Brian Carlstrom ref_offsets &= ~(CLASS_HIGH_BIT >> right_shift); 587db4d54081f09abcbe97ffdf615874f2809a9e777Brian Carlstrom } 588db4d54081f09abcbe97ffdf615874f2809a9e777Brian Carlstrom } else { 5894873d465a1eb6dfbdeddb085c81239d39db60c42Brian Carlstrom // There is no reference offset bitmap. In the non-static case, 5904873d465a1eb6dfbdeddb085c81239d39db60c42Brian Carlstrom // walk up the class inheritance hierarchy and find reference 5914873d465a1eb6dfbdeddb085c81239d39db60c42Brian Carlstrom // offsets the hard way. In the static case, just consider this 5924873d465a1eb6dfbdeddb085c81239d39db60c42Brian Carlstrom // class. 5934873d465a1eb6dfbdeddb085c81239d39db60c42Brian Carlstrom for (const Class *klass = is_static ? orig->AsClass() : orig->GetClass(); 594db4d54081f09abcbe97ffdf615874f2809a9e777Brian Carlstrom klass != NULL; 5954873d465a1eb6dfbdeddb085c81239d39db60c42Brian Carlstrom klass = is_static ? NULL : klass->GetSuperClass()) { 5964873d465a1eb6dfbdeddb085c81239d39db60c42Brian Carlstrom size_t num_reference_fields = (is_static 5974873d465a1eb6dfbdeddb085c81239d39db60c42Brian Carlstrom ? klass->NumReferenceStaticFields() 5984873d465a1eb6dfbdeddb085c81239d39db60c42Brian Carlstrom : klass->NumReferenceInstanceFields()); 5994873d465a1eb6dfbdeddb085c81239d39db60c42Brian Carlstrom for (size_t i = 0; i < num_reference_fields; ++i) { 6004873d465a1eb6dfbdeddb085c81239d39db60c42Brian Carlstrom Field* field = (is_static 6014873d465a1eb6dfbdeddb085c81239d39db60c42Brian Carlstrom ? klass->GetStaticField(i) 6024873d465a1eb6dfbdeddb085c81239d39db60c42Brian Carlstrom : klass->GetInstanceField(i)); 6030cfe1fb7060576d047f7f894fc0d8b87de84fcabIan Rogers MemberOffset field_offset = field->GetOffset(); 6040cfe1fb7060576d047f7f894fc0d8b87de84fcabIan Rogers const Object* ref = orig->GetFieldObject<const Object*>(field_offset, false); 605e35517ad40e0713df8420aeb5d3540f146a59f62Mathieu Chartier // Use SetFieldPtr to avoid card marking since we are writing to the image. 606e35517ad40e0713df8420aeb5d3540f146a59f62Mathieu Chartier copy->SetFieldPtr(field_offset, GetImageAddress(ref), false); 607db4d54081f09abcbe97ffdf615874f2809a9e777Brian Carlstrom } 608db4d54081f09abcbe97ffdf615874f2809a9e777Brian Carlstrom } 609db4d54081f09abcbe97ffdf615874f2809a9e777Brian Carlstrom } 61064b6d145fa53b8dfb07a8fc2426af13f155d5a4dIan Rogers if (!is_static && orig->IsReferenceInstance()) { 61164b6d145fa53b8dfb07a8fc2426af13f155d5a4dIan Rogers // Fix-up referent, that isn't marked as an object field, for References. 61264b6d145fa53b8dfb07a8fc2426af13f155d5a4dIan Rogers Field* field = orig->GetClass()->FindInstanceField("referent", "Ljava/lang/Object;"); 61364b6d145fa53b8dfb07a8fc2426af13f155d5a4dIan Rogers MemberOffset field_offset = field->GetOffset(); 61464b6d145fa53b8dfb07a8fc2426af13f155d5a4dIan Rogers const Object* ref = orig->GetFieldObject<const Object*>(field_offset, false); 61564b6d145fa53b8dfb07a8fc2426af13f155d5a4dIan Rogers // Use SetFieldPtr to avoid card marking since we are writing to the image. 61664b6d145fa53b8dfb07a8fc2426af13f155d5a4dIan Rogers copy->SetFieldPtr(field_offset, GetImageAddress(ref), false); 61764b6d145fa53b8dfb07a8fc2426af13f155d5a4dIan Rogers } 618db4d54081f09abcbe97ffdf615874f2809a9e777Brian Carlstrom} 619db4d54081f09abcbe97ffdf615874f2809a9e777Brian Carlstrom 6201212a022fa5f8ef9585d765b1809521812af882cIan Rogersstatic AbstractMethod* GetTargetMethod(const CompilerDriver::PatchInformation* patch) 621b726dcb581bf72da46527378ccb6889020f0e6e9Ian Rogers SHARED_LOCKS_REQUIRED(Locks::mutator_lock_) { 622f582258f0e296223a091fd64231a203ad71e9649Brian Carlstrom ClassLinker* class_linker = Runtime::Current()->GetClassLinker(); 62300f7d0eaa6bd93d33bf0c1429bf4ba0b3f28abacIan Rogers DexCache* dex_cache = class_linker->FindDexCache(patch->GetDexFile()); 62466f19258f9728d4ffe026074d8fd429d639802faMathieu Chartier AbstractMethod* method = class_linker->ResolveMethod(patch->GetDexFile(), 625f582258f0e296223a091fd64231a203ad71e9649Brian Carlstrom patch->GetTargetMethodIdx(), 62600f7d0eaa6bd93d33bf0c1429bf4ba0b3f28abacIan Rogers dex_cache, 627f582258f0e296223a091fd64231a203ad71e9649Brian Carlstrom NULL, 628c0228b8f02c05ed58bea58490e0d8bdcaf8c5bb8jeffhao NULL, 62908f753d5859936f8d3524e9e4faa6cee353873eaIan Rogers patch->GetTargetInvokeType()); 630f582258f0e296223a091fd64231a203ad71e9649Brian Carlstrom CHECK(method != NULL) 631f582258f0e296223a091fd64231a203ad71e9649Brian Carlstrom << patch->GetDexFile().GetLocation() << " " << patch->GetTargetMethodIdx(); 6320637e27af8e54f39634024dc77e8b094f21782f9Brian Carlstrom CHECK(!method->IsRuntimeMethod()) 6330637e27af8e54f39634024dc77e8b094f21782f9Brian Carlstrom << patch->GetDexFile().GetLocation() << " " << patch->GetTargetMethodIdx(); 63400f7d0eaa6bd93d33bf0c1429bf4ba0b3f28abacIan Rogers CHECK(dex_cache->GetResolvedMethods()->Get(patch->GetTargetMethodIdx()) == method) 6350637e27af8e54f39634024dc77e8b094f21782f9Brian Carlstrom << patch->GetDexFile().GetLocation() << " " << patch->GetReferrerMethodIdx() << " " 63600f7d0eaa6bd93d33bf0c1429bf4ba0b3f28abacIan Rogers << PrettyMethod(dex_cache->GetResolvedMethods()->Get(patch->GetTargetMethodIdx())) << " " 6370637e27af8e54f39634024dc77e8b094f21782f9Brian Carlstrom << PrettyMethod(method); 638f582258f0e296223a091fd64231a203ad71e9649Brian Carlstrom return method; 639f582258f0e296223a091fd64231a203ad71e9649Brian Carlstrom} 640f582258f0e296223a091fd64231a203ad71e9649Brian Carlstrom 64196391606d8adfc661e1c21703ded1e7a39377a76Brian Carlstromvoid ImageWriter::PatchOatCodeAndMethods() { 642e35517ad40e0713df8420aeb5d3540f146a59f62Mathieu Chartier Thread* self = Thread::Current(); 643f582258f0e296223a091fd64231a203ad71e9649Brian Carlstrom ClassLinker* class_linker = Runtime::Current()->GetClassLinker(); 644e35517ad40e0713df8420aeb5d3540f146a59f62Mathieu Chartier const char* old_cause = self->StartAssertNoThreadSuspension("ImageWriter"); 645f582258f0e296223a091fd64231a203ad71e9649Brian Carlstrom 6461212a022fa5f8ef9585d765b1809521812af882cIan Rogers typedef std::vector<const CompilerDriver::PatchInformation*> Patches; 64796391606d8adfc661e1c21703ded1e7a39377a76Brian Carlstrom const Patches& code_to_patch = compiler_driver_.GetCodeToPatch(); 648f582258f0e296223a091fd64231a203ad71e9649Brian Carlstrom for (size_t i = 0; i < code_to_patch.size(); i++) { 6491212a022fa5f8ef9585d765b1809521812af882cIan Rogers const CompilerDriver::PatchInformation* patch = code_to_patch[i]; 65066f19258f9728d4ffe026074d8fd429d639802faMathieu Chartier AbstractMethod* target = GetTargetMethod(patch); 651f582258f0e296223a091fd64231a203ad71e9649Brian Carlstrom uint32_t code = reinterpret_cast<uint32_t>(class_linker->GetOatCodeFor(target)); 652f582258f0e296223a091fd64231a203ad71e9649Brian Carlstrom uint32_t code_base = reinterpret_cast<uint32_t>(&oat_file_->GetOatHeader()); 653f582258f0e296223a091fd64231a203ad71e9649Brian Carlstrom uint32_t code_offset = code - code_base; 654f582258f0e296223a091fd64231a203ad71e9649Brian Carlstrom SetPatchLocation(patch, reinterpret_cast<uint32_t>(GetOatAddress(code_offset))); 655f582258f0e296223a091fd64231a203ad71e9649Brian Carlstrom } 656f582258f0e296223a091fd64231a203ad71e9649Brian Carlstrom 65796391606d8adfc661e1c21703ded1e7a39377a76Brian Carlstrom const Patches& methods_to_patch = compiler_driver_.GetMethodsToPatch(); 658f582258f0e296223a091fd64231a203ad71e9649Brian Carlstrom for (size_t i = 0; i < methods_to_patch.size(); i++) { 6591212a022fa5f8ef9585d765b1809521812af882cIan Rogers const CompilerDriver::PatchInformation* patch = methods_to_patch[i]; 66066f19258f9728d4ffe026074d8fd429d639802faMathieu Chartier AbstractMethod* target = GetTargetMethod(patch); 661f582258f0e296223a091fd64231a203ad71e9649Brian Carlstrom SetPatchLocation(patch, reinterpret_cast<uint32_t>(GetImageAddress(target))); 662f582258f0e296223a091fd64231a203ad71e9649Brian Carlstrom } 663a85b8376740641f13dc078aea6be94d15a800803Brian Carlstrom 664a85b8376740641f13dc078aea6be94d15a800803Brian Carlstrom // Update the image header with the new checksum after patching 665a85b8376740641f13dc078aea6be94d15a800803Brian Carlstrom ImageHeader* image_header = reinterpret_cast<ImageHeader*>(image_->Begin()); 666a85b8376740641f13dc078aea6be94d15a800803Brian Carlstrom image_header->SetOatChecksum(oat_file_->GetOatHeader().GetChecksum()); 667e35517ad40e0713df8420aeb5d3540f146a59f62Mathieu Chartier self->EndAssertNoThreadSuspension(old_cause); 668f582258f0e296223a091fd64231a203ad71e9649Brian Carlstrom} 669f582258f0e296223a091fd64231a203ad71e9649Brian Carlstrom 6701212a022fa5f8ef9585d765b1809521812af882cIan Rogersvoid ImageWriter::SetPatchLocation(const CompilerDriver::PatchInformation* patch, uint32_t value) { 671f582258f0e296223a091fd64231a203ad71e9649Brian Carlstrom ClassLinker* class_linker = Runtime::Current()->GetClassLinker(); 672e35517ad40e0713df8420aeb5d3540f146a59f62Mathieu Chartier const void* oat_code = class_linker->GetOatCodeFor(patch->GetDexFile(), 673e35517ad40e0713df8420aeb5d3540f146a59f62Mathieu Chartier patch->GetReferrerMethodIdx()); 674a85b8376740641f13dc078aea6be94d15a800803Brian Carlstrom OatHeader& oat_header = const_cast<OatHeader&>(oat_file_->GetOatHeader()); 675f582258f0e296223a091fd64231a203ad71e9649Brian Carlstrom // TODO: make this Thumb2 specific 676e35517ad40e0713df8420aeb5d3540f146a59f62Mathieu Chartier uint8_t* base = reinterpret_cast<uint8_t*>(reinterpret_cast<uint32_t>(oat_code) & ~0x1); 677f582258f0e296223a091fd64231a203ad71e9649Brian Carlstrom uint32_t* patch_location = reinterpret_cast<uint32_t*>(base + patch->GetLiteralOffset()); 678f582258f0e296223a091fd64231a203ad71e9649Brian Carlstrom#ifndef NDEBUG 679f582258f0e296223a091fd64231a203ad71e9649Brian Carlstrom const DexFile::MethodId& id = patch->GetDexFile().GetMethodId(patch->GetTargetMethodIdx()); 680f582258f0e296223a091fd64231a203ad71e9649Brian Carlstrom uint32_t expected = reinterpret_cast<uint32_t>(&id); 681f582258f0e296223a091fd64231a203ad71e9649Brian Carlstrom uint32_t actual = *patch_location; 682f582258f0e296223a091fd64231a203ad71e9649Brian Carlstrom CHECK(actual == expected || actual == value) << std::hex 683f582258f0e296223a091fd64231a203ad71e9649Brian Carlstrom << "actual=" << actual 684f582258f0e296223a091fd64231a203ad71e9649Brian Carlstrom << "expected=" << expected 685f582258f0e296223a091fd64231a203ad71e9649Brian Carlstrom << "value=" << value; 686f582258f0e296223a091fd64231a203ad71e9649Brian Carlstrom#endif 687b25c3f6a86dc634ce44fb2849385b49465caa84dElliott Hughes *patch_location = value; 688a85b8376740641f13dc078aea6be94d15a800803Brian Carlstrom oat_header.UpdateChecksum(patch_location, sizeof(value)); 689f582258f0e296223a091fd64231a203ad71e9649Brian Carlstrom} 690f582258f0e296223a091fd64231a203ad71e9649Brian Carlstrom 691db4d54081f09abcbe97ffdf615874f2809a9e777Brian Carlstrom} // namespace art 692