image_writer.cc revision 637c65b1e431fd90195b71c141b3590bd81cc91a
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" 271212a022fa5f8ef9585d765b1809521812af882cIan Rogers#include "compiler/driver/compiler_driver.h" 284f6ad8ab428038129b2d0d6c40b7fd625cca15e1Ian Rogers#include "dex_file-inl.h" 292dd0e2cea360bc9206eb88ecc40d259e796c239dIan Rogers#include "gc/card_table-inl.h" 301c23e1edb7361bbaec6e57fca86d8d3797960ad2Mathieu Chartier#include "gc/large_object_space.h" 311c23e1edb7361bbaec6e57fca86d8d3797960ad2Mathieu Chartier#include "gc/space.h" 32db4d54081f09abcbe97ffdf615874f2809a9e777Brian Carlstrom#include "globals.h" 33db4d54081f09abcbe97ffdf615874f2809a9e777Brian Carlstrom#include "heap.h" 344a289ed61242964b921434de7d375f46480472a1Brian Carlstrom#include "image.h" 35a663ea5de4c9ab6b1510fdebd6d8eca77ba699aeBrian Carlstrom#include "intern_table.h" 362dd0e2cea360bc9206eb88ecc40d259e796c239dIan Rogers#include "mirror/array-inl.h" 372dd0e2cea360bc9206eb88ecc40d259e796c239dIan Rogers#include "mirror/class-inl.h" 382dd0e2cea360bc9206eb88ecc40d259e796c239dIan Rogers#include "mirror/class_loader.h" 3939ebcb800aabedd0ffa6aa4aeac8aa4194c66e61Ian Rogers#include "mirror/dex_cache-inl.h" 402dd0e2cea360bc9206eb88ecc40d259e796c239dIan Rogers#include "mirror/field-inl.h" 412dd0e2cea360bc9206eb88ecc40d259e796c239dIan Rogers#include "mirror/abstract_method-inl.h" 422dd0e2cea360bc9206eb88ecc40d259e796c239dIan Rogers#include "mirror/object-inl.h" 432dd0e2cea360bc9206eb88ecc40d259e796c239dIan Rogers#include "mirror/object_array-inl.h" 44700c8d31733534a3d978b75a03f6f7e177dc7e81Brian Carlstrom#include "oat.h" 450c717dd1c56bd29cf860d0feda8e629dab2cadb3Logan Chien#include "oat_file.h" 466d4d9fcb4f01e287ee29e81cd1c941ee5d11d379Ian Rogers#include "object_utils.h" 471f87008b165d26541d832ff805250afdc89c253dBrian Carlstrom#include "runtime.h" 4800f7d0eaa6bd93d33bf0c1429bf4ba0b3f28abacIan Rogers#include "scoped_thread_state_change.h" 491f5393447b9f45be7918042d9ee7b521376de866Ian Rogers#include "sirt_ref.h" 50a168c83a1d247094e9efb1244b0f73a5f1e1ed97Elliott Hughes#include "UniquePtr.h" 51db4d54081f09abcbe97ffdf615874f2809a9e777Brian Carlstrom#include "utils.h" 52db4d54081f09abcbe97ffdf615874f2809a9e777Brian Carlstrom 532dd0e2cea360bc9206eb88ecc40d259e796c239dIan Rogersusing namespace art::mirror; 542dd0e2cea360bc9206eb88ecc40d259e796c239dIan Rogers 55db4d54081f09abcbe97ffdf615874f2809a9e777Brian Carlstromnamespace art { 56db4d54081f09abcbe97ffdf615874f2809a9e777Brian Carlstrom 57a004aa933a58428489e42d77f707c2b063b73747Brian Carlstrombool ImageWriter::Write(const std::string& image_filename, 5830fab40ee5a07af6b8c3b6b0e9438071695a57f4Ian Rogers uintptr_t image_begin, 59ae826983f7903bc0a6bbbe8426bf393fb2f6d747Brian Carlstrom const std::string& oat_filename, 60f582258f0e296223a091fd64231a203ad71e9649Brian Carlstrom const std::string& oat_location, 611212a022fa5f8ef9585d765b1809521812af882cIan Rogers const CompilerDriver& compiler_driver) { 62a004aa933a58428489e42d77f707c2b063b73747Brian Carlstrom CHECK(!image_filename.empty()); 63aded5f7ab991f3c1132851599d3bc60ff6707eedBrian Carlstrom 6430fab40ee5a07af6b8c3b6b0e9438071695a57f4Ian Rogers CHECK_NE(image_begin, 0U); 6530fab40ee5a07af6b8c3b6b0e9438071695a57f4Ian Rogers image_begin_ = reinterpret_cast<byte*>(image_begin); 6669b15fb098162f19a4c20e6dccdcead04d9c77f0Brian Carlstrom 67b3bd5f07884f5a1f2b84224363b1372d7c28d447Elliott Hughes Heap* heap = Runtime::Current()->GetHeap(); 68b062fdd4cb097fbae69b4bcb479c34d83ecab8caMathieu Chartier const Spaces& spaces = heap->GetSpaces(); 6969b15fb098162f19a4c20e6dccdcead04d9c77f0Brian Carlstrom 70ae826983f7903bc0a6bbbe8426bf393fb2f6d747Brian Carlstrom ClassLinker* class_linker = Runtime::Current()->GetClassLinker(); 71ae826983f7903bc0a6bbbe8426bf393fb2f6d747Brian Carlstrom const std::vector<DexCache*>& all_dex_caches = class_linker->GetDexCaches(); 72ae826983f7903bc0a6bbbe8426bf393fb2f6d747Brian Carlstrom for (size_t i = 0; i < all_dex_caches.size(); i++) { 73ae826983f7903bc0a6bbbe8426bf393fb2f6d747Brian Carlstrom DexCache* dex_cache = all_dex_caches[i]; 7464b6d145fa53b8dfb07a8fc2426af13f155d5a4dIan Rogers dex_caches_.insert(dex_cache); 75ae826983f7903bc0a6bbbe8426bf393fb2f6d747Brian Carlstrom } 76ae826983f7903bc0a6bbbe8426bf393fb2f6d747Brian Carlstrom 77700c8d31733534a3d978b75a03f6f7e177dc7e81Brian Carlstrom UniquePtr<File> oat_file(OS::OpenFile(oat_filename.c_str(), true, false)); 78700c8d31733534a3d978b75a03f6f7e177dc7e81Brian Carlstrom if (oat_file.get() == NULL) { 79700c8d31733534a3d978b75a03f6f7e177dc7e81Brian Carlstrom LOG(ERROR) << "Failed to open oat file " << oat_filename << " for " << oat_location; 80e24fa61603a60ade3797e4a0c8b3fccb346cb048Brian Carlstrom return false; 81e24fa61603a60ade3797e4a0c8b3fccb346cb048Brian Carlstrom } 82265091e581c9f643b37e7966890911f09e223269Brian Carlstrom oat_file_ = OatFile::OpenWritable(oat_file.get(), oat_location); 83b25c3f6a86dc634ce44fb2849385b49465caa84dElliott Hughes class_linker->RegisterOatFile(*oat_file_); 84e24fa61603a60ade3797e4a0c8b3fccb346cb048Brian Carlstrom 8500f7d0eaa6bd93d33bf0c1429bf4ba0b3f28abacIan Rogers { 8600f7d0eaa6bd93d33bf0c1429bf4ba0b3f28abacIan Rogers Thread::Current()->TransitionFromSuspendedToRunnable(); 8700f7d0eaa6bd93d33bf0c1429bf4ba0b3f28abacIan Rogers PruneNonImageClasses(); // Remove junk 8800f7d0eaa6bd93d33bf0c1429bf4ba0b3f28abacIan Rogers ComputeLazyFieldsForImageClasses(); // Add useful information 8900f7d0eaa6bd93d33bf0c1429bf4ba0b3f28abacIan Rogers ComputeEagerResolvedStrings(); 9000f7d0eaa6bd93d33bf0c1429bf4ba0b3f28abacIan Rogers Thread::Current()->TransitionFromRunnableToSuspended(kNative); 9100f7d0eaa6bd93d33bf0c1429bf4ba0b3f28abacIan Rogers } 92b3bd5f07884f5a1f2b84224363b1372d7c28d447Elliott Hughes heap->CollectGarbage(false); // Remove garbage 93b062fdd4cb097fbae69b4bcb479c34d83ecab8caMathieu Chartier // Trim size of alloc spaces 94b062fdd4cb097fbae69b4bcb479c34d83ecab8caMathieu Chartier // TODO: C++0x auto 95b062fdd4cb097fbae69b4bcb479c34d83ecab8caMathieu Chartier for (Spaces::const_iterator cur = spaces.begin(); cur != spaces.end(); ++cur) { 96b062fdd4cb097fbae69b4bcb479c34d83ecab8caMathieu Chartier if ((*cur)->IsAllocSpace()) { 97b062fdd4cb097fbae69b4bcb479c34d83ecab8caMathieu Chartier (*cur)->AsAllocSpace()->Trim(); 98b062fdd4cb097fbae69b4bcb479c34d83ecab8caMathieu Chartier } 99b062fdd4cb097fbae69b4bcb479c34d83ecab8caMathieu Chartier } 100b062fdd4cb097fbae69b4bcb479c34d83ecab8caMathieu Chartier 101ae826983f7903bc0a6bbbe8426bf393fb2f6d747Brian Carlstrom if (!AllocMemory()) { 102db4d54081f09abcbe97ffdf615874f2809a9e777Brian Carlstrom return false; 103db4d54081f09abcbe97ffdf615874f2809a9e777Brian Carlstrom } 104ae826983f7903bc0a6bbbe8426bf393fb2f6d747Brian Carlstrom#ifndef NDEBUG 105357e9be24c17a6bc2ae9fb53f25c73503116101dMathieu Chartier { 106357e9be24c17a6bc2ae9fb53f25c73503116101dMathieu Chartier ScopedObjectAccess soa(Thread::Current()); 107357e9be24c17a6bc2ae9fb53f25c73503116101dMathieu Chartier CheckNonImageClassesRemoved(); 108357e9be24c17a6bc2ae9fb53f25c73503116101dMathieu Chartier } 109ae826983f7903bc0a6bbbe8426bf393fb2f6d747Brian Carlstrom#endif 110e35517ad40e0713df8420aeb5d3540f146a59f62Mathieu Chartier Thread::Current()->TransitionFromSuspendedToRunnable(); 111700c8d31733534a3d978b75a03f6f7e177dc7e81Brian Carlstrom size_t oat_loaded_size = 0; 112700c8d31733534a3d978b75a03f6f7e177dc7e81Brian Carlstrom size_t oat_data_offset = 0; 1131212a022fa5f8ef9585d765b1809521812af882cIan Rogers compiler_driver.GetOatElfInformation(oat_file.get(), oat_loaded_size, oat_data_offset); 114700c8d31733534a3d978b75a03f6f7e177dc7e81Brian Carlstrom CalculateNewObjectOffsets(oat_loaded_size, oat_data_offset); 115e35517ad40e0713df8420aeb5d3540f146a59f62Mathieu Chartier CopyAndFixupObjects(); 1161212a022fa5f8ef9585d765b1809521812af882cIan Rogers PatchOatCodeAndMethods(compiler_driver); 117e35517ad40e0713df8420aeb5d3540f146a59f62Mathieu Chartier Thread::Current()->TransitionFromRunnableToSuspended(kNative); 118d8ddfd5eadde1d5f53ef1419f529c799233eaa62Elliott Hughes 119700c8d31733534a3d978b75a03f6f7e177dc7e81Brian Carlstrom UniquePtr<File> image_file(OS::OpenFile(image_filename.c_str(), true)); 120700c8d31733534a3d978b75a03f6f7e177dc7e81Brian Carlstrom if (image_file.get() == NULL) { 121e24fa61603a60ade3797e4a0c8b3fccb346cb048Brian Carlstrom LOG(ERROR) << "Failed to open image file " << image_filename; 122db4d54081f09abcbe97ffdf615874f2809a9e777Brian Carlstrom return false; 123db4d54081f09abcbe97ffdf615874f2809a9e777Brian Carlstrom } 124700c8d31733534a3d978b75a03f6f7e177dc7e81Brian Carlstrom if (fchmod(image_file->Fd(), 0644) != 0) { 1256cd40e5cc59259a0b7636eb5532c76de9b8a7286Brian Carlstrom PLOG(ERROR) << "Failed to make image file world readable: " << image_filename; 1266cd40e5cc59259a0b7636eb5532c76de9b8a7286Brian Carlstrom return EXIT_FAILURE; 1276cd40e5cc59259a0b7636eb5532c76de9b8a7286Brian Carlstrom } 128700c8d31733534a3d978b75a03f6f7e177dc7e81Brian Carlstrom bool success = image_file->WriteFully(image_->Begin(), image_end_); 129e24fa61603a60ade3797e4a0c8b3fccb346cb048Brian Carlstrom if (!success) { 130e24fa61603a60ade3797e4a0c8b3fccb346cb048Brian Carlstrom PLOG(ERROR) << "Failed to write image file " << image_filename; 131e24fa61603a60ade3797e4a0c8b3fccb346cb048Brian Carlstrom return false; 132e24fa61603a60ade3797e4a0c8b3fccb346cb048Brian Carlstrom } 133e24fa61603a60ade3797e4a0c8b3fccb346cb048Brian Carlstrom return true; 134db4d54081f09abcbe97ffdf615874f2809a9e777Brian Carlstrom} 135db4d54081f09abcbe97ffdf615874f2809a9e777Brian Carlstrom 136ae826983f7903bc0a6bbbe8426bf393fb2f6d747Brian Carlstrombool ImageWriter::AllocMemory() { 1372fde53367dbe721e5273c34b590e67112322cc9eMathieu Chartier const Spaces& spaces = Runtime::Current()->GetHeap()->GetSpaces(); 138b062fdd4cb097fbae69b4bcb479c34d83ecab8caMathieu Chartier size_t size = 0; 1392fde53367dbe721e5273c34b590e67112322cc9eMathieu Chartier for (Spaces::const_iterator it = spaces.begin(); it != spaces.end(); ++it) { 1402fde53367dbe721e5273c34b590e67112322cc9eMathieu Chartier if ((*it)->IsAllocSpace()) { 1412fde53367dbe721e5273c34b590e67112322cc9eMathieu Chartier size += (*it)->Size(); 142b062fdd4cb097fbae69b4bcb479c34d83ecab8caMathieu Chartier } 143b062fdd4cb097fbae69b4bcb479c34d83ecab8caMathieu Chartier } 144b062fdd4cb097fbae69b4bcb479c34d83ecab8caMathieu Chartier 145db4d54081f09abcbe97ffdf615874f2809a9e777Brian Carlstrom int prot = PROT_READ | PROT_WRITE; 146db4d54081f09abcbe97ffdf615874f2809a9e777Brian Carlstrom size_t length = RoundUp(size, kPageSize); 147a40307e8562c20d9b9ce46e773864d07d95512a1Ian Rogers image_.reset(MemMap::MapAnonymous("image writer image", NULL, length, prot)); 14890a3369d3b6238f1a4c9b19ca68978dab1c39bc4Elliott Hughes if (image_.get() == NULL) { 149e24fa61603a60ade3797e4a0c8b3fccb346cb048Brian Carlstrom LOG(ERROR) << "Failed to allocate memory for image file generation"; 150db4d54081f09abcbe97ffdf615874f2809a9e777Brian Carlstrom return false; 151db4d54081f09abcbe97ffdf615874f2809a9e777Brian Carlstrom } 152db4d54081f09abcbe97ffdf615874f2809a9e777Brian Carlstrom return true; 153db4d54081f09abcbe97ffdf615874f2809a9e777Brian Carlstrom} 154db4d54081f09abcbe97ffdf615874f2809a9e777Brian Carlstrom 155d418edaf4df0a410d678389e171ac6d96ae2af15Ian Rogersvoid ImageWriter::ComputeLazyFieldsForImageClasses() { 156d418edaf4df0a410d678389e171ac6d96ae2af15Ian Rogers Runtime* runtime = Runtime::Current(); 157d418edaf4df0a410d678389e171ac6d96ae2af15Ian Rogers ClassLinker* class_linker = runtime->GetClassLinker(); 15800f7d0eaa6bd93d33bf0c1429bf4ba0b3f28abacIan Rogers class_linker->VisitClassesWithoutClassesLock(ComputeLazyFieldsForClassesVisitor, NULL); 159d418edaf4df0a410d678389e171ac6d96ae2af15Ian Rogers} 160d418edaf4df0a410d678389e171ac6d96ae2af15Ian Rogers 1611bac54ffa933fbe9b92b62437577f2f4583eff1aElliott Hughesbool ImageWriter::ComputeLazyFieldsForClassesVisitor(Class* c, void* /*arg*/) { 1621bac54ffa933fbe9b92b62437577f2f4583eff1aElliott Hughes c->ComputeName(); 163d418edaf4df0a410d678389e171ac6d96ae2af15Ian Rogers return true; 164d418edaf4df0a410d678389e171ac6d96ae2af15Ian Rogers} 165d418edaf4df0a410d678389e171ac6d96ae2af15Ian Rogers 166d1f1bf03e30320e5730790678f470ac8a610b6b6Ian Rogersvoid ImageWriter::ComputeEagerResolvedStringsCallback(Object* obj, void* arg) { 167d1f1bf03e30320e5730790678f470ac8a610b6b6Ian Rogers if (!obj->GetClass()->IsStringClass()) { 168d1f1bf03e30320e5730790678f470ac8a610b6b6Ian Rogers return; 169d1f1bf03e30320e5730790678f470ac8a610b6b6Ian Rogers } 170d1f1bf03e30320e5730790678f470ac8a610b6b6Ian Rogers String* string = obj->AsString(); 171637c65b1e431fd90195b71c141b3590bd81cc91aIan Rogers const uint16_t* utf16_string = string->GetCharArray()->GetData() + string->GetOffset(); 172d1f1bf03e30320e5730790678f470ac8a610b6b6Ian Rogers ImageWriter* writer = reinterpret_cast<ImageWriter*>(arg); 173d1f1bf03e30320e5730790678f470ac8a610b6b6Ian Rogers typedef Set::const_iterator CacheIt; // TODO: C++0x auto 174d1f1bf03e30320e5730790678f470ac8a610b6b6Ian Rogers for (CacheIt it = writer->dex_caches_.begin(), end = writer->dex_caches_.end(); it != end; ++it) { 175d1f1bf03e30320e5730790678f470ac8a610b6b6Ian Rogers DexCache* dex_cache = *it; 1764445a7e3398a6143939168097a3aa275b734504dIan Rogers const DexFile& dex_file = *dex_cache->GetDexFile(); 177637c65b1e431fd90195b71c141b3590bd81cc91aIan Rogers const DexFile::StringId* string_id = dex_file.FindStringId(utf16_string); 178d1f1bf03e30320e5730790678f470ac8a610b6b6Ian Rogers if (string_id != NULL) { 179d1f1bf03e30320e5730790678f470ac8a610b6b6Ian Rogers // This string occurs in this dex file, assign the dex cache entry. 180d1f1bf03e30320e5730790678f470ac8a610b6b6Ian Rogers uint32_t string_idx = dex_file.GetIndexForStringId(*string_id); 181d1f1bf03e30320e5730790678f470ac8a610b6b6Ian Rogers if (dex_cache->GetResolvedString(string_idx) == NULL) { 182d1f1bf03e30320e5730790678f470ac8a610b6b6Ian Rogers dex_cache->SetResolvedString(string_idx, string); 183d1f1bf03e30320e5730790678f470ac8a610b6b6Ian Rogers } 184d1f1bf03e30320e5730790678f470ac8a610b6b6Ian Rogers } 185d1f1bf03e30320e5730790678f470ac8a610b6b6Ian Rogers } 186d1f1bf03e30320e5730790678f470ac8a610b6b6Ian Rogers} 187d1f1bf03e30320e5730790678f470ac8a610b6b6Ian Rogers 188357e9be24c17a6bc2ae9fb53f25c73503116101dMathieu Chartiervoid ImageWriter::ComputeEagerResolvedStrings() 189b726dcb581bf72da46527378ccb6889020f0e6e9Ian Rogers SHARED_LOCKS_REQUIRED(Locks::mutator_lock_) { 190b062fdd4cb097fbae69b4bcb479c34d83ecab8caMathieu Chartier // TODO: Check image spaces only? 191357e9be24c17a6bc2ae9fb53f25c73503116101dMathieu Chartier Heap* heap = Runtime::Current()->GetHeap(); 192e35517ad40e0713df8420aeb5d3540f146a59f62Mathieu Chartier WriterMutexLock mu(Thread::Current(), *Locks::heap_bitmap_lock_); 193357e9be24c17a6bc2ae9fb53f25c73503116101dMathieu Chartier heap->FlushAllocStack(); 194357e9be24c17a6bc2ae9fb53f25c73503116101dMathieu Chartier heap->GetLiveBitmap()->Walk(ComputeEagerResolvedStringsCallback, this); 195d1f1bf03e30320e5730790678f470ac8a610b6b6Ian Rogers} 196d1f1bf03e30320e5730790678f470ac8a610b6b6Ian Rogers 197ae826983f7903bc0a6bbbe8426bf393fb2f6d747Brian Carlstrombool ImageWriter::IsImageClass(const Class* klass) { 198ae826983f7903bc0a6bbbe8426bf393fb2f6d747Brian Carlstrom if (image_classes_ == NULL) { 199ae826983f7903bc0a6bbbe8426bf393fb2f6d747Brian Carlstrom return true; 200ae826983f7903bc0a6bbbe8426bf393fb2f6d747Brian Carlstrom } 201ae826983f7903bc0a6bbbe8426bf393fb2f6d747Brian Carlstrom while (klass->IsArrayClass()) { 202ae826983f7903bc0a6bbbe8426bf393fb2f6d747Brian Carlstrom klass = klass->GetComponentType(); 203ae826983f7903bc0a6bbbe8426bf393fb2f6d747Brian Carlstrom } 204c3b77c7c2971124cbf3b2d9da64e7a8a9a649f2eElliott Hughes if (klass->IsPrimitive()) { 205ae826983f7903bc0a6bbbe8426bf393fb2f6d747Brian Carlstrom return true; 206ae826983f7903bc0a6bbbe8426bf393fb2f6d747Brian Carlstrom } 2076d4d9fcb4f01e287ee29e81cd1c941ee5d11d379Ian Rogers const std::string descriptor(ClassHelper(klass).GetDescriptor()); 208ae826983f7903bc0a6bbbe8426bf393fb2f6d747Brian Carlstrom return image_classes_->find(descriptor) != image_classes_->end(); 209ae826983f7903bc0a6bbbe8426bf393fb2f6d747Brian Carlstrom} 210ae826983f7903bc0a6bbbe8426bf393fb2f6d747Brian Carlstrom 211ae826983f7903bc0a6bbbe8426bf393fb2f6d747Brian Carlstrom 212ae826983f7903bc0a6bbbe8426bf393fb2f6d747Brian Carlstromstruct NonImageClasses { 213ae826983f7903bc0a6bbbe8426bf393fb2f6d747Brian Carlstrom ImageWriter* image_writer; 214ae826983f7903bc0a6bbbe8426bf393fb2f6d747Brian Carlstrom std::set<std::string>* non_image_classes; 215ae826983f7903bc0a6bbbe8426bf393fb2f6d747Brian Carlstrom}; 216ae826983f7903bc0a6bbbe8426bf393fb2f6d747Brian Carlstrom 217ae826983f7903bc0a6bbbe8426bf393fb2f6d747Brian Carlstromvoid ImageWriter::PruneNonImageClasses() { 218ae826983f7903bc0a6bbbe8426bf393fb2f6d747Brian Carlstrom if (image_classes_ == NULL) { 219ae826983f7903bc0a6bbbe8426bf393fb2f6d747Brian Carlstrom return; 220ae826983f7903bc0a6bbbe8426bf393fb2f6d747Brian Carlstrom } 221ae826983f7903bc0a6bbbe8426bf393fb2f6d747Brian Carlstrom Runtime* runtime = Runtime::Current(); 222ae826983f7903bc0a6bbbe8426bf393fb2f6d747Brian Carlstrom ClassLinker* class_linker = runtime->GetClassLinker(); 223ae826983f7903bc0a6bbbe8426bf393fb2f6d747Brian Carlstrom 224ae826983f7903bc0a6bbbe8426bf393fb2f6d747Brian Carlstrom std::set<std::string> non_image_classes; 225ae826983f7903bc0a6bbbe8426bf393fb2f6d747Brian Carlstrom NonImageClasses context; 226ae826983f7903bc0a6bbbe8426bf393fb2f6d747Brian Carlstrom context.image_writer = this; 227ae826983f7903bc0a6bbbe8426bf393fb2f6d747Brian Carlstrom context.non_image_classes = &non_image_classes; 228ae826983f7903bc0a6bbbe8426bf393fb2f6d747Brian Carlstrom class_linker->VisitClasses(NonImageClassesVisitor, &context); 229ae826983f7903bc0a6bbbe8426bf393fb2f6d747Brian Carlstrom 230ae826983f7903bc0a6bbbe8426bf393fb2f6d747Brian Carlstrom typedef std::set<std::string>::const_iterator ClassIt; // TODO: C++0x auto 231ae826983f7903bc0a6bbbe8426bf393fb2f6d747Brian Carlstrom for (ClassIt it = non_image_classes.begin(), end = non_image_classes.end(); it != end; ++it) { 232c3b77c7c2971124cbf3b2d9da64e7a8a9a649f2eElliott Hughes class_linker->RemoveClass((*it).c_str(), NULL); 233ae826983f7903bc0a6bbbe8426bf393fb2f6d747Brian Carlstrom } 234ae826983f7903bc0a6bbbe8426bf393fb2f6d747Brian Carlstrom 23566f19258f9728d4ffe026074d8fd429d639802faMathieu Chartier AbstractMethod* resolution_method = runtime->GetResolutionMethod(); 236ae826983f7903bc0a6bbbe8426bf393fb2f6d747Brian Carlstrom typedef Set::const_iterator CacheIt; // TODO: C++0x auto 237ae826983f7903bc0a6bbbe8426bf393fb2f6d747Brian Carlstrom for (CacheIt it = dex_caches_.begin(), end = dex_caches_.end(); it != end; ++it) { 238ae826983f7903bc0a6bbbe8426bf393fb2f6d747Brian Carlstrom DexCache* dex_cache = *it; 239ae826983f7903bc0a6bbbe8426bf393fb2f6d747Brian Carlstrom for (size_t i = 0; i < dex_cache->NumResolvedTypes(); i++) { 240ae826983f7903bc0a6bbbe8426bf393fb2f6d747Brian Carlstrom Class* klass = dex_cache->GetResolvedType(i); 241ae826983f7903bc0a6bbbe8426bf393fb2f6d747Brian Carlstrom if (klass != NULL && !IsImageClass(klass)) { 242ae826983f7903bc0a6bbbe8426bf393fb2f6d747Brian Carlstrom dex_cache->SetResolvedType(i, NULL); 243ae826983f7903bc0a6bbbe8426bf393fb2f6d747Brian Carlstrom dex_cache->GetInitializedStaticStorage()->Set(i, NULL); 244ae826983f7903bc0a6bbbe8426bf393fb2f6d747Brian Carlstrom } 245ae826983f7903bc0a6bbbe8426bf393fb2f6d747Brian Carlstrom } 246ae826983f7903bc0a6bbbe8426bf393fb2f6d747Brian Carlstrom for (size_t i = 0; i < dex_cache->NumResolvedMethods(); i++) { 24766f19258f9728d4ffe026074d8fd429d639802faMathieu Chartier AbstractMethod* method = dex_cache->GetResolvedMethod(i); 248ae826983f7903bc0a6bbbe8426bf393fb2f6d747Brian Carlstrom if (method != NULL && !IsImageClass(method->GetDeclaringClass())) { 2491984651929744dd603fd082e23eacd877b9bc177Ian Rogers dex_cache->SetResolvedMethod(i, resolution_method); 250ae826983f7903bc0a6bbbe8426bf393fb2f6d747Brian Carlstrom } 251ae826983f7903bc0a6bbbe8426bf393fb2f6d747Brian Carlstrom } 252ae826983f7903bc0a6bbbe8426bf393fb2f6d747Brian Carlstrom for (size_t i = 0; i < dex_cache->NumResolvedFields(); i++) { 253ae826983f7903bc0a6bbbe8426bf393fb2f6d747Brian Carlstrom Field* field = dex_cache->GetResolvedField(i); 254ae826983f7903bc0a6bbbe8426bf393fb2f6d747Brian Carlstrom if (field != NULL && !IsImageClass(field->GetDeclaringClass())) { 255ae826983f7903bc0a6bbbe8426bf393fb2f6d747Brian Carlstrom dex_cache->SetResolvedField(i, NULL); 256ae826983f7903bc0a6bbbe8426bf393fb2f6d747Brian Carlstrom } 257ae826983f7903bc0a6bbbe8426bf393fb2f6d747Brian Carlstrom } 258ae826983f7903bc0a6bbbe8426bf393fb2f6d747Brian Carlstrom } 259ae826983f7903bc0a6bbbe8426bf393fb2f6d747Brian Carlstrom} 260ae826983f7903bc0a6bbbe8426bf393fb2f6d747Brian Carlstrom 261ae826983f7903bc0a6bbbe8426bf393fb2f6d747Brian Carlstrombool ImageWriter::NonImageClassesVisitor(Class* klass, void* arg) { 262ae826983f7903bc0a6bbbe8426bf393fb2f6d747Brian Carlstrom NonImageClasses* context = reinterpret_cast<NonImageClasses*>(arg); 263ae826983f7903bc0a6bbbe8426bf393fb2f6d747Brian Carlstrom if (!context->image_writer->IsImageClass(klass)) { 2646d4d9fcb4f01e287ee29e81cd1c941ee5d11d379Ian Rogers context->non_image_classes->insert(ClassHelper(klass).GetDescriptor()); 265ae826983f7903bc0a6bbbe8426bf393fb2f6d747Brian Carlstrom } 266ae826983f7903bc0a6bbbe8426bf393fb2f6d747Brian Carlstrom return true; 267ae826983f7903bc0a6bbbe8426bf393fb2f6d747Brian Carlstrom} 268ae826983f7903bc0a6bbbe8426bf393fb2f6d747Brian Carlstrom 269357e9be24c17a6bc2ae9fb53f25c73503116101dMathieu Chartiervoid ImageWriter::CheckNonImageClassesRemoved() 270b726dcb581bf72da46527378ccb6889020f0e6e9Ian Rogers SHARED_LOCKS_REQUIRED(Locks::mutator_lock_) { 271ae826983f7903bc0a6bbbe8426bf393fb2f6d747Brian Carlstrom if (image_classes_ == NULL) { 272ae826983f7903bc0a6bbbe8426bf393fb2f6d747Brian Carlstrom return; 273ae826983f7903bc0a6bbbe8426bf393fb2f6d747Brian Carlstrom } 274b062fdd4cb097fbae69b4bcb479c34d83ecab8caMathieu Chartier 275357e9be24c17a6bc2ae9fb53f25c73503116101dMathieu Chartier Heap* heap = Runtime::Current()->GetHeap(); 27650b35e2fd1a68cd1240e4a9d9f363e11764957d1Ian Rogers Thread* self = Thread::Current(); 277357e9be24c17a6bc2ae9fb53f25c73503116101dMathieu Chartier { 27850b35e2fd1a68cd1240e4a9d9f363e11764957d1Ian Rogers WriterMutexLock mu(self, *Locks::heap_bitmap_lock_); 279357e9be24c17a6bc2ae9fb53f25c73503116101dMathieu Chartier heap->FlushAllocStack(); 280357e9be24c17a6bc2ae9fb53f25c73503116101dMathieu Chartier } 281357e9be24c17a6bc2ae9fb53f25c73503116101dMathieu Chartier 28250b35e2fd1a68cd1240e4a9d9f363e11764957d1Ian Rogers ReaderMutexLock mu(self, *Locks::heap_bitmap_lock_); 283357e9be24c17a6bc2ae9fb53f25c73503116101dMathieu Chartier heap->GetLiveBitmap()->Walk(CheckNonImageClassesRemovedCallback, this); 284ae826983f7903bc0a6bbbe8426bf393fb2f6d747Brian Carlstrom} 285ae826983f7903bc0a6bbbe8426bf393fb2f6d747Brian Carlstrom 286ae826983f7903bc0a6bbbe8426bf393fb2f6d747Brian Carlstromvoid ImageWriter::CheckNonImageClassesRemovedCallback(Object* obj, void* arg) { 287ae826983f7903bc0a6bbbe8426bf393fb2f6d747Brian Carlstrom ImageWriter* image_writer = reinterpret_cast<ImageWriter*>(arg); 288ae826983f7903bc0a6bbbe8426bf393fb2f6d747Brian Carlstrom if (!obj->IsClass()) { 289ae826983f7903bc0a6bbbe8426bf393fb2f6d747Brian Carlstrom return; 290ae826983f7903bc0a6bbbe8426bf393fb2f6d747Brian Carlstrom } 291ae826983f7903bc0a6bbbe8426bf393fb2f6d747Brian Carlstrom Class* klass = obj->AsClass(); 292c3b77c7c2971124cbf3b2d9da64e7a8a9a649f2eElliott Hughes if (!image_writer->IsImageClass(klass)) { 293c3b77c7c2971124cbf3b2d9da64e7a8a9a649f2eElliott Hughes image_writer->DumpImageClasses(); 294c3b77c7c2971124cbf3b2d9da64e7a8a9a649f2eElliott Hughes CHECK(image_writer->IsImageClass(klass)) << ClassHelper(klass).GetDescriptor() 295c3b77c7c2971124cbf3b2d9da64e7a8a9a649f2eElliott Hughes << " " << PrettyDescriptor(klass); 296c3b77c7c2971124cbf3b2d9da64e7a8a9a649f2eElliott Hughes } 297c3b77c7c2971124cbf3b2d9da64e7a8a9a649f2eElliott Hughes} 298c3b77c7c2971124cbf3b2d9da64e7a8a9a649f2eElliott Hughes 299c3b77c7c2971124cbf3b2d9da64e7a8a9a649f2eElliott Hughesvoid ImageWriter::DumpImageClasses() { 300c3b77c7c2971124cbf3b2d9da64e7a8a9a649f2eElliott Hughes typedef std::set<std::string>::const_iterator It; // TODO: C++0x auto 301c3b77c7c2971124cbf3b2d9da64e7a8a9a649f2eElliott Hughes for (It it = image_classes_->begin(), end = image_classes_->end(); it != end; ++it) { 302c3b77c7c2971124cbf3b2d9da64e7a8a9a649f2eElliott Hughes LOG(INFO) << " " << *it; 303c3b77c7c2971124cbf3b2d9da64e7a8a9a649f2eElliott Hughes } 304ae826983f7903bc0a6bbbe8426bf393fb2f6d747Brian Carlstrom} 305ae826983f7903bc0a6bbbe8426bf393fb2f6d747Brian Carlstrom 30678128a63b2615744760b7f8ab83df9764a5d4a95Brian Carlstromvoid ImageWriter::CalculateNewObjectOffsetsCallback(Object* obj, void* arg) { 3074e777d4e005375a5a41c4ec3c55783c09565df0dBrian Carlstrom DCHECK(obj != NULL); 3084e777d4e005375a5a41c4ec3c55783c09565df0dBrian Carlstrom DCHECK(arg != NULL); 3094e777d4e005375a5a41c4ec3c55783c09565df0dBrian Carlstrom ImageWriter* image_writer = reinterpret_cast<ImageWriter*>(arg); 310c74255fffb035001304c9a058a2e730a5a1a9604Brian Carlstrom 311c74255fffb035001304c9a058a2e730a5a1a9604Brian Carlstrom // if it is a string, we want to intern it if its not interned. 312dbb4079eb1e7d7738c81a97c8dd2550885c1093aElliott Hughes if (obj->GetClass()->IsStringClass()) { 313c74255fffb035001304c9a058a2e730a5a1a9604Brian Carlstrom // we must be an interned string that was forward referenced and already assigned 314b3bd5f07884f5a1f2b84224363b1372d7c28d447Elliott Hughes if (image_writer->IsImageOffsetAssigned(obj)) { 315c74255fffb035001304c9a058a2e730a5a1a9604Brian Carlstrom DCHECK_EQ(obj, obj->AsString()->Intern()); 316c74255fffb035001304c9a058a2e730a5a1a9604Brian Carlstrom return; 317c74255fffb035001304c9a058a2e730a5a1a9604Brian Carlstrom } 3181f5393447b9f45be7918042d9ee7b521376de866Ian Rogers SirtRef<String> interned(Thread::Current(), obj->AsString()->Intern()); 31940381fb9dc4b4cf274f1e58b2cdf4396202c6189Brian Carlstrom if (obj != interned.get()) { 320b3bd5f07884f5a1f2b84224363b1372d7c28d447Elliott Hughes if (!image_writer->IsImageOffsetAssigned(interned.get())) { 321c74255fffb035001304c9a058a2e730a5a1a9604Brian Carlstrom // interned obj is after us, allocate its location early 32240381fb9dc4b4cf274f1e58b2cdf4396202c6189Brian Carlstrom image_writer->AssignImageOffset(interned.get()); 323c74255fffb035001304c9a058a2e730a5a1a9604Brian Carlstrom } 324c74255fffb035001304c9a058a2e730a5a1a9604Brian Carlstrom // point those looking for this object to the interned version. 325b3bd5f07884f5a1f2b84224363b1372d7c28d447Elliott Hughes image_writer->SetImageOffset(obj, image_writer->GetImageOffset(interned.get())); 326c74255fffb035001304c9a058a2e730a5a1a9604Brian Carlstrom return; 327c74255fffb035001304c9a058a2e730a5a1a9604Brian Carlstrom } 328c74255fffb035001304c9a058a2e730a5a1a9604Brian Carlstrom // else (obj == interned), nothing to do but fall through to the normal case 329c74255fffb035001304c9a058a2e730a5a1a9604Brian Carlstrom } 330c74255fffb035001304c9a058a2e730a5a1a9604Brian Carlstrom 331c74255fffb035001304c9a058a2e730a5a1a9604Brian Carlstrom image_writer->AssignImageOffset(obj); 3324e777d4e005375a5a41c4ec3c55783c09565df0dBrian Carlstrom} 3334e777d4e005375a5a41c4ec3c55783c09565df0dBrian Carlstrom 334e24fa61603a60ade3797e4a0c8b3fccb346cb048Brian CarlstromObjectArray<Object>* ImageWriter::CreateImageRoots() const { 335161928613d3f097108319de60494fab1aab8d48aBrian Carlstrom Runtime* runtime = Runtime::Current(); 336161928613d3f097108319de60494fab1aab8d48aBrian Carlstrom ClassLinker* class_linker = runtime->GetClassLinker(); 337161928613d3f097108319de60494fab1aab8d48aBrian Carlstrom Class* object_array_class = class_linker->FindSystemClass("[Ljava/lang/Object;"); 33850b35e2fd1a68cd1240e4a9d9f363e11764957d1Ian Rogers Thread* self = Thread::Current(); 33958ae9416e197ae68ed12ed43d87407d4dfb15093Brian Carlstrom 34058ae9416e197ae68ed12ed43d87407d4dfb15093Brian Carlstrom // build an Object[] of all the DexCaches used in the source_space_ 34150b35e2fd1a68cd1240e4a9d9f363e11764957d1Ian Rogers ObjectArray<Object>* dex_caches = ObjectArray<Object>::Alloc(self, object_array_class, 342ae826983f7903bc0a6bbbe8426bf393fb2f6d747Brian Carlstrom dex_caches_.size()); 343ae826983f7903bc0a6bbbe8426bf393fb2f6d747Brian Carlstrom int i = 0; 344ae826983f7903bc0a6bbbe8426bf393fb2f6d747Brian Carlstrom typedef Set::const_iterator It; // TODO: C++0x auto 345ae826983f7903bc0a6bbbe8426bf393fb2f6d747Brian Carlstrom for (It it = dex_caches_.begin(), end = dex_caches_.end(); it != end; ++it, ++i) { 346ae826983f7903bc0a6bbbe8426bf393fb2f6d747Brian Carlstrom dex_caches->Set(i, *it); 34758ae9416e197ae68ed12ed43d87407d4dfb15093Brian Carlstrom } 34858ae9416e197ae68ed12ed43d87407d4dfb15093Brian Carlstrom 34958ae9416e197ae68ed12ed43d87407d4dfb15093Brian Carlstrom // build an Object[] of the roots needed to restore the runtime 3501f5393447b9f45be7918042d9ee7b521376de866Ian Rogers SirtRef<ObjectArray<Object> > 35150b35e2fd1a68cd1240e4a9d9f363e11764957d1Ian Rogers image_roots(self, 35250b35e2fd1a68cd1240e4a9d9f363e11764957d1Ian Rogers ObjectArray<Object>::Alloc(self, object_array_class, 35350b35e2fd1a68cd1240e4a9d9f363e11764957d1Ian Rogers ImageHeader::kImageRootsMax)); 3541984651929744dd603fd082e23eacd877b9bc177Ian Rogers image_roots->Set(ImageHeader::kResolutionMethod, runtime->GetResolutionMethod()); 3554f0d07c783afef89703dce32c94440fc8621a29bIan Rogers image_roots->Set(ImageHeader::kCalleeSaveMethod, 3564f0d07c783afef89703dce32c94440fc8621a29bIan Rogers runtime->GetCalleeSaveMethod(Runtime::kSaveAll)); 3574f0d07c783afef89703dce32c94440fc8621a29bIan Rogers image_roots->Set(ImageHeader::kRefsOnlySaveMethod, 3584f0d07c783afef89703dce32c94440fc8621a29bIan Rogers runtime->GetCalleeSaveMethod(Runtime::kRefsOnly)); 3594f0d07c783afef89703dce32c94440fc8621a29bIan Rogers image_roots->Set(ImageHeader::kRefsAndArgsSaveMethod, 3604f0d07c783afef89703dce32c94440fc8621a29bIan Rogers runtime->GetCalleeSaveMethod(Runtime::kRefsAndArgs)); 361e24fa61603a60ade3797e4a0c8b3fccb346cb048Brian Carlstrom image_roots->Set(ImageHeader::kOatLocation, 36250b35e2fd1a68cd1240e4a9d9f363e11764957d1Ian Rogers String::AllocFromModifiedUtf8(self, oat_file_->GetLocation().c_str())); 36358ae9416e197ae68ed12ed43d87407d4dfb15093Brian Carlstrom image_roots->Set(ImageHeader::kDexCaches, 36458ae9416e197ae68ed12ed43d87407d4dfb15093Brian Carlstrom dex_caches); 36534f426c49ac2de8cea70acef6b9ecdd8e62209d2Brian Carlstrom image_roots->Set(ImageHeader::kClassRoots, 36634f426c49ac2de8cea70acef6b9ecdd8e62209d2Brian Carlstrom class_linker->GetClassRoots()); 367e24fa61603a60ade3797e4a0c8b3fccb346cb048Brian Carlstrom for (int i = 0; i < ImageHeader::kImageRootsMax; i++) { 368e24fa61603a60ade3797e4a0c8b3fccb346cb048Brian Carlstrom CHECK(image_roots->Get(i) != NULL); 369e24fa61603a60ade3797e4a0c8b3fccb346cb048Brian Carlstrom } 37040381fb9dc4b4cf274f1e58b2cdf4396202c6189Brian Carlstrom return image_roots.get(); 371161928613d3f097108319de60494fab1aab8d48aBrian Carlstrom} 372161928613d3f097108319de60494fab1aab8d48aBrian Carlstrom 373700c8d31733534a3d978b75a03f6f7e177dc7e81Brian Carlstromvoid ImageWriter::CalculateNewObjectOffsets(size_t oat_loaded_size, size_t oat_data_offset) { 374700c8d31733534a3d978b75a03f6f7e177dc7e81Brian Carlstrom CHECK_NE(0U, oat_loaded_size); 3751f5393447b9f45be7918042d9ee7b521376de866Ian Rogers Thread* self = Thread::Current(); 3761f5393447b9f45be7918042d9ee7b521376de866Ian Rogers SirtRef<ObjectArray<Object> > image_roots(self, CreateImageRoots()); 377161928613d3f097108319de60494fab1aab8d48aBrian Carlstrom 378b062fdd4cb097fbae69b4bcb479c34d83ecab8caMathieu Chartier Heap* heap = Runtime::Current()->GetHeap(); 379e0f0cb3d855cb5e926452b5e1ec8457adc4e454eMathieu Chartier const Spaces& spaces = heap->GetSpaces(); 380b062fdd4cb097fbae69b4bcb479c34d83ecab8caMathieu Chartier DCHECK(!spaces.empty()); 38130fab40ee5a07af6b8c3b6b0e9438071695a57f4Ian Rogers DCHECK_EQ(0U, image_end_); 382a663ea5de4c9ab6b1510fdebd6d8eca77ba699aeBrian Carlstrom 383161928613d3f097108319de60494fab1aab8d48aBrian Carlstrom // leave space for the header, but do not write it yet, we need to 384161928613d3f097108319de60494fab1aab8d48aBrian Carlstrom // know where image_roots is going to end up 38530fab40ee5a07af6b8c3b6b0e9438071695a57f4Ian Rogers image_end_ += RoundUp(sizeof(ImageHeader), 8); // 64-bit-alignment 386a663ea5de4c9ab6b1510fdebd6d8eca77ba699aeBrian Carlstrom 387357e9be24c17a6bc2ae9fb53f25c73503116101dMathieu Chartier { 388e35517ad40e0713df8420aeb5d3540f146a59f62Mathieu Chartier WriterMutexLock mu(self, *Locks::heap_bitmap_lock_); 389357e9be24c17a6bc2ae9fb53f25c73503116101dMathieu Chartier heap->FlushAllocStack(); 39066f19258f9728d4ffe026074d8fd429d639802faMathieu Chartier // TODO: Image spaces only? 39166f19258f9728d4ffe026074d8fd429d639802faMathieu Chartier // TODO: Add InOrderWalk to heap bitmap. 3921f5393447b9f45be7918042d9ee7b521376de866Ian Rogers const char* old = self->StartAssertNoThreadSuspension("ImageWriter"); 393e0f0cb3d855cb5e926452b5e1ec8457adc4e454eMathieu Chartier DCHECK(heap->GetLargeObjectsSpace()->GetLiveObjects()->IsEmpty()); 394e0f0cb3d855cb5e926452b5e1ec8457adc4e454eMathieu Chartier for (Spaces::const_iterator it = spaces.begin(); it != spaces.end(); ++it) { 39566f19258f9728d4ffe026074d8fd429d639802faMathieu Chartier (*it)->GetLiveBitmap()->InOrderWalk(CalculateNewObjectOffsetsCallback, this); 396357e9be24c17a6bc2ae9fb53f25c73503116101dMathieu Chartier DCHECK_LT(image_end_, image_->Size()); 397357e9be24c17a6bc2ae9fb53f25c73503116101dMathieu Chartier } 3981f5393447b9f45be7918042d9ee7b521376de866Ian Rogers self->EndAssertNoThreadSuspension(old); 399b062fdd4cb097fbae69b4bcb479c34d83ecab8caMathieu Chartier } 400a663ea5de4c9ab6b1510fdebd6d8eca77ba699aeBrian Carlstrom 401700c8d31733534a3d978b75a03f6f7e177dc7e81Brian Carlstrom const byte* oat_file_begin = image_begin_ + RoundUp(image_end_, kPageSize); 402700c8d31733534a3d978b75a03f6f7e177dc7e81Brian Carlstrom const byte* oat_file_end = oat_file_begin + oat_loaded_size; 403700c8d31733534a3d978b75a03f6f7e177dc7e81Brian Carlstrom oat_data_begin_ = oat_file_begin + oat_data_offset; 404700c8d31733534a3d978b75a03f6f7e177dc7e81Brian Carlstrom const byte* oat_data_end = oat_data_begin_ + oat_file_->Size(); 405e24fa61603a60ade3797e4a0c8b3fccb346cb048Brian Carlstrom 406161928613d3f097108319de60494fab1aab8d48aBrian Carlstrom // return to write header at start of image with future location of image_roots 40730fab40ee5a07af6b8c3b6b0e9438071695a57f4Ian Rogers ImageHeader image_header(reinterpret_cast<uint32_t>(image_begin_), 40840381fb9dc4b4cf274f1e58b2cdf4396202c6189Brian Carlstrom reinterpret_cast<uint32_t>(GetImageAddress(image_roots.get())), 409e24fa61603a60ade3797e4a0c8b3fccb346cb048Brian Carlstrom oat_file_->GetOatHeader().GetChecksum(), 410700c8d31733534a3d978b75a03f6f7e177dc7e81Brian Carlstrom reinterpret_cast<uint32_t>(oat_file_begin), 411700c8d31733534a3d978b75a03f6f7e177dc7e81Brian Carlstrom reinterpret_cast<uint32_t>(oat_data_begin_), 412700c8d31733534a3d978b75a03f6f7e177dc7e81Brian Carlstrom reinterpret_cast<uint32_t>(oat_data_end), 413700c8d31733534a3d978b75a03f6f7e177dc7e81Brian Carlstrom reinterpret_cast<uint32_t>(oat_file_end)); 41430fab40ee5a07af6b8c3b6b0e9438071695a57f4Ian Rogers memcpy(image_->Begin(), &image_header, sizeof(image_header)); 415700c8d31733534a3d978b75a03f6f7e177dc7e81Brian Carlstrom 416700c8d31733534a3d978b75a03f6f7e177dc7e81Brian Carlstrom // Note that image_end_ is left at end of used space 417db4d54081f09abcbe97ffdf615874f2809a9e777Brian Carlstrom} 418db4d54081f09abcbe97ffdf615874f2809a9e777Brian Carlstrom 419357e9be24c17a6bc2ae9fb53f25c73503116101dMathieu Chartiervoid ImageWriter::CopyAndFixupObjects() 420b726dcb581bf72da46527378ccb6889020f0e6e9Ian Rogers SHARED_LOCKS_REQUIRED(Locks::mutator_lock_) { 42150b35e2fd1a68cd1240e4a9d9f363e11764957d1Ian Rogers Thread* self = Thread::Current(); 42250b35e2fd1a68cd1240e4a9d9f363e11764957d1Ian Rogers const char* old_cause = self->StartAssertNoThreadSuspension("ImageWriter"); 423b3bd5f07884f5a1f2b84224363b1372d7c28d447Elliott Hughes Heap* heap = Runtime::Current()->GetHeap(); 4240cfe1fb7060576d047f7f894fc0d8b87de84fcabIan Rogers // TODO: heap validation can't handle this fix up pass 425b3bd5f07884f5a1f2b84224363b1372d7c28d447Elliott Hughes heap->DisableObjectValidation(); 426b062fdd4cb097fbae69b4bcb479c34d83ecab8caMathieu Chartier // TODO: Image spaces only? 427e35517ad40e0713df8420aeb5d3540f146a59f62Mathieu Chartier WriterMutexLock mu(self, *Locks::heap_bitmap_lock_); 428357e9be24c17a6bc2ae9fb53f25c73503116101dMathieu Chartier heap->FlushAllocStack(); 429b062fdd4cb097fbae69b4bcb479c34d83ecab8caMathieu Chartier heap->GetLiveBitmap()->Walk(CopyAndFixupObjectsCallback, this); 43050b35e2fd1a68cd1240e4a9d9f363e11764957d1Ian Rogers self->EndAssertNoThreadSuspension(old_cause); 431db4d54081f09abcbe97ffdf615874f2809a9e777Brian Carlstrom} 432db4d54081f09abcbe97ffdf615874f2809a9e777Brian Carlstrom 43378128a63b2615744760b7f8ab83df9764a5d4a95Brian Carlstromvoid ImageWriter::CopyAndFixupObjectsCallback(Object* object, void* arg) { 4344873d465a1eb6dfbdeddb085c81239d39db60c42Brian Carlstrom DCHECK(object != NULL); 435db4d54081f09abcbe97ffdf615874f2809a9e777Brian Carlstrom DCHECK(arg != NULL); 4364873d465a1eb6dfbdeddb085c81239d39db60c42Brian Carlstrom const Object* obj = object; 437db4d54081f09abcbe97ffdf615874f2809a9e777Brian Carlstrom ImageWriter* image_writer = reinterpret_cast<ImageWriter*>(arg); 438d8ddfd5eadde1d5f53ef1419f529c799233eaa62Elliott Hughes 43969b15fb098162f19a4c20e6dccdcead04d9c77f0Brian Carlstrom // see GetLocalAddress for similar computation 440db4d54081f09abcbe97ffdf615874f2809a9e777Brian Carlstrom size_t offset = image_writer->GetImageOffset(obj); 44130fab40ee5a07af6b8c3b6b0e9438071695a57f4Ian Rogers byte* dst = image_writer->image_->Begin() + offset; 4424873d465a1eb6dfbdeddb085c81239d39db60c42Brian Carlstrom const byte* src = reinterpret_cast<const byte*>(obj); 44304b63fd60de0d9e6f07763511a1a9622accfeed7Elliott Hughes size_t n = obj->SizeOf(); 44430fab40ee5a07af6b8c3b6b0e9438071695a57f4Ian Rogers DCHECK_LT(offset + n, image_writer->image_->Size()); 445db4d54081f09abcbe97ffdf615874f2809a9e777Brian Carlstrom memcpy(dst, src, n); 446db4d54081f09abcbe97ffdf615874f2809a9e777Brian Carlstrom Object* copy = reinterpret_cast<Object*>(dst); 4472dd0e2cea360bc9206eb88ecc40d259e796c239dIan Rogers copy->SetField32(Object::MonitorOffset(), 0, false); // We may have inflated the lock during compilation. 448db4d54081f09abcbe97ffdf615874f2809a9e777Brian Carlstrom image_writer->FixupObject(obj, copy); 449db4d54081f09abcbe97ffdf615874f2809a9e777Brian Carlstrom} 450db4d54081f09abcbe97ffdf615874f2809a9e777Brian Carlstrom 4514873d465a1eb6dfbdeddb085c81239d39db60c42Brian Carlstromvoid ImageWriter::FixupObject(const Object* orig, Object* copy) { 452db4d54081f09abcbe97ffdf615874f2809a9e777Brian Carlstrom DCHECK(orig != NULL); 453db4d54081f09abcbe97ffdf615874f2809a9e777Brian Carlstrom DCHECK(copy != NULL); 4540cfe1fb7060576d047f7f894fc0d8b87de84fcabIan Rogers copy->SetClass(down_cast<Class*>(GetImageAddress(orig->GetClass()))); 4559cff8e13d41825c4f3f0127af061e94b06114fc8Brian Carlstrom // TODO: special case init of pointers to malloc data (or removal of these pointers) 4569cff8e13d41825c4f3f0127af061e94b06114fc8Brian Carlstrom if (orig->IsClass()) { 4579cff8e13d41825c4f3f0127af061e94b06114fc8Brian Carlstrom FixupClass(orig->AsClass(), down_cast<Class*>(copy)); 4589cff8e13d41825c4f3f0127af061e94b06114fc8Brian Carlstrom } else if (orig->IsObjectArray()) { 459db4d54081f09abcbe97ffdf615874f2809a9e777Brian Carlstrom FixupObjectArray(orig->AsObjectArray<Object>(), down_cast<ObjectArray<Object>*>(copy)); 460161928613d3f097108319de60494fab1aab8d48aBrian Carlstrom } else if (orig->IsMethod()) { 46166f19258f9728d4ffe026074d8fd429d639802faMathieu Chartier FixupMethod(orig->AsMethod(), down_cast<AbstractMethod*>(copy)); 462db4d54081f09abcbe97ffdf615874f2809a9e777Brian Carlstrom } else { 463db4d54081f09abcbe97ffdf615874f2809a9e777Brian Carlstrom FixupInstanceFields(orig, copy); 464db4d54081f09abcbe97ffdf615874f2809a9e777Brian Carlstrom } 465db4d54081f09abcbe97ffdf615874f2809a9e777Brian Carlstrom} 466db4d54081f09abcbe97ffdf615874f2809a9e777Brian Carlstrom 4674873d465a1eb6dfbdeddb085c81239d39db60c42Brian Carlstromvoid ImageWriter::FixupClass(const Class* orig, Class* copy) { 4689cff8e13d41825c4f3f0127af061e94b06114fc8Brian Carlstrom FixupInstanceFields(orig, copy); 4694873d465a1eb6dfbdeddb085c81239d39db60c42Brian Carlstrom FixupStaticFields(orig, copy); 4709cff8e13d41825c4f3f0127af061e94b06114fc8Brian Carlstrom} 4719cff8e13d41825c4f3f0127af061e94b06114fc8Brian Carlstrom 47266f19258f9728d4ffe026074d8fd429d639802faMathieu Chartiervoid ImageWriter::FixupMethod(const AbstractMethod* orig, AbstractMethod* copy) { 473a663ea5de4c9ab6b1510fdebd6d8eca77ba699aeBrian Carlstrom FixupInstanceFields(orig, copy); 474e24fa61603a60ade3797e4a0c8b3fccb346cb048Brian Carlstrom 47574180cad94848107cf297d37e72437c5a6eecf1bJeff Hao // OatWriter replaces the code_ with an offset value. 47630fab40ee5a07af6b8c3b6b0e9438071695a57f4Ian Rogers // Here we readjust to a pointer relative to oat_begin_ 4773320cf46afd082398aa401b246e6f301cebdf64dBrian Carlstrom if (orig->IsAbstract()) { 47879fe539587d4c09244172d0168eeed0ec9770466Jeff Hao // Code for abstract methods is set to the abstract method error stub when we load the image. 479aa4a793d7175ceeb1efa215c2af5e52e42b22f0aJeff Hao copy->SetEntryPointFromCompiledCode(NULL); 4803320cf46afd082398aa401b246e6f301cebdf64dBrian Carlstrom return; 4813320cf46afd082398aa401b246e6f301cebdf64dBrian Carlstrom } 4823320cf46afd082398aa401b246e6f301cebdf64dBrian Carlstrom 4831984651929744dd603fd082e23eacd877b9bc177Ian Rogers if (orig == Runtime::Current()->GetResolutionMethod()) { 48458df32711162823647356afe25ae2cde04133563Jeff Hao // The resolution method's code is set to the resolution trampoline when we load the image. 485aa4a793d7175ceeb1efa215c2af5e52e42b22f0aJeff Hao copy->SetEntryPointFromCompiledCode(NULL); 4861984651929744dd603fd082e23eacd877b9bc177Ian Rogers return; 4871984651929744dd603fd082e23eacd877b9bc177Ian Rogers } 4881984651929744dd603fd082e23eacd877b9bc177Ian Rogers 48958df32711162823647356afe25ae2cde04133563Jeff Hao // Non-abstract methods have code 490aa4a793d7175ceeb1efa215c2af5e52e42b22f0aJeff Hao copy->SetEntryPointFromCompiledCode(GetOatAddress(orig->GetOatCodeOffset())); 4913320cf46afd082398aa401b246e6f301cebdf64dBrian Carlstrom 492161928613d3f097108319de60494fab1aab8d48aBrian Carlstrom if (orig->IsNative()) { 49379fe539587d4c09244172d0168eeed0ec9770466Jeff Hao // The native method's pointer is set to a stub to lookup via dlsym when we load the image. 4943320cf46afd082398aa401b246e6f301cebdf64dBrian Carlstrom // Note this is not the code_ pointer, that is handled above. 49579fe539587d4c09244172d0168eeed0ec9770466Jeff Hao copy->SetNativeMethod(NULL); 496161928613d3f097108319de60494fab1aab8d48aBrian Carlstrom } else { 4973320cf46afd082398aa401b246e6f301cebdf64dBrian Carlstrom // normal (non-abstract non-native) methods have mapping tables to relocate 4983320cf46afd082398aa401b246e6f301cebdf64dBrian Carlstrom uint32_t mapping_table_off = orig->GetOatMappingTableOffset(); 499ae826983f7903bc0a6bbbe8426bf393fb2f6d747Brian Carlstrom const byte* mapping_table = GetOatAddress(mapping_table_off); 5002dd0e2cea360bc9206eb88ecc40d259e796c239dIan Rogers copy->SetMappingTable(reinterpret_cast<const uint32_t*>(mapping_table)); 5013320cf46afd082398aa401b246e6f301cebdf64dBrian Carlstrom 5023320cf46afd082398aa401b246e6f301cebdf64dBrian Carlstrom uint32_t vmap_table_offset = orig->GetOatVmapTableOffset(); 503ae826983f7903bc0a6bbbe8426bf393fb2f6d747Brian Carlstrom const byte* vmap_table = GetOatAddress(vmap_table_offset); 5042dd0e2cea360bc9206eb88ecc40d259e796c239dIan Rogers copy->SetVmapTable(reinterpret_cast<const uint16_t*>(vmap_table)); 505e7d856b911222aa000ca2be0f8f63f5b292141c3Brian Carlstrom 5060c7abda482f53db3d153c073d1c7a145f84e0626Ian Rogers uint32_t native_gc_map_offset = orig->GetOatNativeGcMapOffset(); 5070c7abda482f53db3d153c073d1c7a145f84e0626Ian Rogers const byte* native_gc_map = GetOatAddress(native_gc_map_offset); 5082dd0e2cea360bc9206eb88ecc40d259e796c239dIan Rogers copy->SetNativeGcMap(reinterpret_cast<const uint8_t*>(native_gc_map)); 509161928613d3f097108319de60494fab1aab8d48aBrian Carlstrom } 510a663ea5de4c9ab6b1510fdebd6d8eca77ba699aeBrian Carlstrom} 511a663ea5de4c9ab6b1510fdebd6d8eca77ba699aeBrian Carlstrom 5124873d465a1eb6dfbdeddb085c81239d39db60c42Brian Carlstromvoid ImageWriter::FixupObjectArray(const ObjectArray<Object>* orig, ObjectArray<Object>* copy) { 513d8ddfd5eadde1d5f53ef1419f529c799233eaa62Elliott Hughes for (int32_t i = 0; i < orig->GetLength(); ++i) { 514db4d54081f09abcbe97ffdf615874f2809a9e777Brian Carlstrom const Object* element = orig->Get(i); 515e35517ad40e0713df8420aeb5d3540f146a59f62Mathieu Chartier copy->SetPtrWithoutChecks(i, GetImageAddress(element)); 516db4d54081f09abcbe97ffdf615874f2809a9e777Brian Carlstrom } 517db4d54081f09abcbe97ffdf615874f2809a9e777Brian Carlstrom} 518db4d54081f09abcbe97ffdf615874f2809a9e777Brian Carlstrom 5194873d465a1eb6dfbdeddb085c81239d39db60c42Brian Carlstromvoid ImageWriter::FixupInstanceFields(const Object* orig, Object* copy) { 5204873d465a1eb6dfbdeddb085c81239d39db60c42Brian Carlstrom DCHECK(orig != NULL); 5214873d465a1eb6dfbdeddb085c81239d39db60c42Brian Carlstrom DCHECK(copy != NULL); 5224873d465a1eb6dfbdeddb085c81239d39db60c42Brian Carlstrom Class* klass = orig->GetClass(); 5234873d465a1eb6dfbdeddb085c81239d39db60c42Brian Carlstrom DCHECK(klass != NULL); 5244873d465a1eb6dfbdeddb085c81239d39db60c42Brian Carlstrom FixupFields(orig, 5254873d465a1eb6dfbdeddb085c81239d39db60c42Brian Carlstrom copy, 5264873d465a1eb6dfbdeddb085c81239d39db60c42Brian Carlstrom klass->GetReferenceInstanceOffsets(), 5274873d465a1eb6dfbdeddb085c81239d39db60c42Brian Carlstrom false); 5284873d465a1eb6dfbdeddb085c81239d39db60c42Brian Carlstrom} 5294873d465a1eb6dfbdeddb085c81239d39db60c42Brian Carlstrom 5304873d465a1eb6dfbdeddb085c81239d39db60c42Brian Carlstromvoid ImageWriter::FixupStaticFields(const Class* orig, Class* copy) { 5314873d465a1eb6dfbdeddb085c81239d39db60c42Brian Carlstrom DCHECK(orig != NULL); 5324873d465a1eb6dfbdeddb085c81239d39db60c42Brian Carlstrom DCHECK(copy != NULL); 5334873d465a1eb6dfbdeddb085c81239d39db60c42Brian Carlstrom FixupFields(orig, 5344873d465a1eb6dfbdeddb085c81239d39db60c42Brian Carlstrom copy, 5354873d465a1eb6dfbdeddb085c81239d39db60c42Brian Carlstrom orig->GetReferenceStaticOffsets(), 5364873d465a1eb6dfbdeddb085c81239d39db60c42Brian Carlstrom true); 5374873d465a1eb6dfbdeddb085c81239d39db60c42Brian Carlstrom} 5384873d465a1eb6dfbdeddb085c81239d39db60c42Brian Carlstrom 5394873d465a1eb6dfbdeddb085c81239d39db60c42Brian Carlstromvoid ImageWriter::FixupFields(const Object* orig, 5404873d465a1eb6dfbdeddb085c81239d39db60c42Brian Carlstrom Object* copy, 5414873d465a1eb6dfbdeddb085c81239d39db60c42Brian Carlstrom uint32_t ref_offsets, 5424873d465a1eb6dfbdeddb085c81239d39db60c42Brian Carlstrom bool is_static) { 543db4d54081f09abcbe97ffdf615874f2809a9e777Brian Carlstrom if (ref_offsets != CLASS_WALK_SUPER) { 544db4d54081f09abcbe97ffdf615874f2809a9e777Brian Carlstrom // Found a reference offset bitmap. Fixup the specified offsets. 545db4d54081f09abcbe97ffdf615874f2809a9e777Brian Carlstrom while (ref_offsets != 0) { 546db4d54081f09abcbe97ffdf615874f2809a9e777Brian Carlstrom size_t right_shift = CLZ(ref_offsets); 5470cfe1fb7060576d047f7f894fc0d8b87de84fcabIan Rogers MemberOffset byte_offset = CLASS_OFFSET_FROM_CLZ(right_shift); 5480cfe1fb7060576d047f7f894fc0d8b87de84fcabIan Rogers const Object* ref = orig->GetFieldObject<const Object*>(byte_offset, false); 549e35517ad40e0713df8420aeb5d3540f146a59f62Mathieu Chartier // Use SetFieldPtr to avoid card marking since we are writing to the image. 550e35517ad40e0713df8420aeb5d3540f146a59f62Mathieu Chartier copy->SetFieldPtr(byte_offset, GetImageAddress(ref), false); 551db4d54081f09abcbe97ffdf615874f2809a9e777Brian Carlstrom ref_offsets &= ~(CLASS_HIGH_BIT >> right_shift); 552db4d54081f09abcbe97ffdf615874f2809a9e777Brian Carlstrom } 553db4d54081f09abcbe97ffdf615874f2809a9e777Brian Carlstrom } else { 5544873d465a1eb6dfbdeddb085c81239d39db60c42Brian Carlstrom // There is no reference offset bitmap. In the non-static case, 5554873d465a1eb6dfbdeddb085c81239d39db60c42Brian Carlstrom // walk up the class inheritance hierarchy and find reference 5564873d465a1eb6dfbdeddb085c81239d39db60c42Brian Carlstrom // offsets the hard way. In the static case, just consider this 5574873d465a1eb6dfbdeddb085c81239d39db60c42Brian Carlstrom // class. 5584873d465a1eb6dfbdeddb085c81239d39db60c42Brian Carlstrom for (const Class *klass = is_static ? orig->AsClass() : orig->GetClass(); 559db4d54081f09abcbe97ffdf615874f2809a9e777Brian Carlstrom klass != NULL; 5604873d465a1eb6dfbdeddb085c81239d39db60c42Brian Carlstrom klass = is_static ? NULL : klass->GetSuperClass()) { 5614873d465a1eb6dfbdeddb085c81239d39db60c42Brian Carlstrom size_t num_reference_fields = (is_static 5624873d465a1eb6dfbdeddb085c81239d39db60c42Brian Carlstrom ? klass->NumReferenceStaticFields() 5634873d465a1eb6dfbdeddb085c81239d39db60c42Brian Carlstrom : klass->NumReferenceInstanceFields()); 5644873d465a1eb6dfbdeddb085c81239d39db60c42Brian Carlstrom for (size_t i = 0; i < num_reference_fields; ++i) { 5654873d465a1eb6dfbdeddb085c81239d39db60c42Brian Carlstrom Field* field = (is_static 5664873d465a1eb6dfbdeddb085c81239d39db60c42Brian Carlstrom ? klass->GetStaticField(i) 5674873d465a1eb6dfbdeddb085c81239d39db60c42Brian Carlstrom : klass->GetInstanceField(i)); 5680cfe1fb7060576d047f7f894fc0d8b87de84fcabIan Rogers MemberOffset field_offset = field->GetOffset(); 5690cfe1fb7060576d047f7f894fc0d8b87de84fcabIan Rogers const Object* ref = orig->GetFieldObject<const Object*>(field_offset, false); 570e35517ad40e0713df8420aeb5d3540f146a59f62Mathieu Chartier // Use SetFieldPtr to avoid card marking since we are writing to the image. 571e35517ad40e0713df8420aeb5d3540f146a59f62Mathieu Chartier copy->SetFieldPtr(field_offset, GetImageAddress(ref), false); 572db4d54081f09abcbe97ffdf615874f2809a9e777Brian Carlstrom } 573db4d54081f09abcbe97ffdf615874f2809a9e777Brian Carlstrom } 574db4d54081f09abcbe97ffdf615874f2809a9e777Brian Carlstrom } 57564b6d145fa53b8dfb07a8fc2426af13f155d5a4dIan Rogers if (!is_static && orig->IsReferenceInstance()) { 57664b6d145fa53b8dfb07a8fc2426af13f155d5a4dIan Rogers // Fix-up referent, that isn't marked as an object field, for References. 57764b6d145fa53b8dfb07a8fc2426af13f155d5a4dIan Rogers Field* field = orig->GetClass()->FindInstanceField("referent", "Ljava/lang/Object;"); 57864b6d145fa53b8dfb07a8fc2426af13f155d5a4dIan Rogers MemberOffset field_offset = field->GetOffset(); 57964b6d145fa53b8dfb07a8fc2426af13f155d5a4dIan Rogers const Object* ref = orig->GetFieldObject<const Object*>(field_offset, false); 58064b6d145fa53b8dfb07a8fc2426af13f155d5a4dIan Rogers // Use SetFieldPtr to avoid card marking since we are writing to the image. 58164b6d145fa53b8dfb07a8fc2426af13f155d5a4dIan Rogers copy->SetFieldPtr(field_offset, GetImageAddress(ref), false); 58264b6d145fa53b8dfb07a8fc2426af13f155d5a4dIan Rogers } 583db4d54081f09abcbe97ffdf615874f2809a9e777Brian Carlstrom} 584db4d54081f09abcbe97ffdf615874f2809a9e777Brian Carlstrom 5851212a022fa5f8ef9585d765b1809521812af882cIan Rogersstatic AbstractMethod* GetTargetMethod(const CompilerDriver::PatchInformation* patch) 586b726dcb581bf72da46527378ccb6889020f0e6e9Ian Rogers SHARED_LOCKS_REQUIRED(Locks::mutator_lock_) { 587f582258f0e296223a091fd64231a203ad71e9649Brian Carlstrom ClassLinker* class_linker = Runtime::Current()->GetClassLinker(); 58800f7d0eaa6bd93d33bf0c1429bf4ba0b3f28abacIan Rogers DexCache* dex_cache = class_linker->FindDexCache(patch->GetDexFile()); 58966f19258f9728d4ffe026074d8fd429d639802faMathieu Chartier AbstractMethod* method = class_linker->ResolveMethod(patch->GetDexFile(), 590f582258f0e296223a091fd64231a203ad71e9649Brian Carlstrom patch->GetTargetMethodIdx(), 59100f7d0eaa6bd93d33bf0c1429bf4ba0b3f28abacIan Rogers dex_cache, 592f582258f0e296223a091fd64231a203ad71e9649Brian Carlstrom NULL, 593c0228b8f02c05ed58bea58490e0d8bdcaf8c5bb8jeffhao NULL, 59408f753d5859936f8d3524e9e4faa6cee353873eaIan Rogers patch->GetTargetInvokeType()); 595f582258f0e296223a091fd64231a203ad71e9649Brian Carlstrom CHECK(method != NULL) 596f582258f0e296223a091fd64231a203ad71e9649Brian Carlstrom << patch->GetDexFile().GetLocation() << " " << patch->GetTargetMethodIdx(); 5970637e27af8e54f39634024dc77e8b094f21782f9Brian Carlstrom CHECK(!method->IsRuntimeMethod()) 5980637e27af8e54f39634024dc77e8b094f21782f9Brian Carlstrom << patch->GetDexFile().GetLocation() << " " << patch->GetTargetMethodIdx(); 59900f7d0eaa6bd93d33bf0c1429bf4ba0b3f28abacIan Rogers CHECK(dex_cache->GetResolvedMethods()->Get(patch->GetTargetMethodIdx()) == method) 6000637e27af8e54f39634024dc77e8b094f21782f9Brian Carlstrom << patch->GetDexFile().GetLocation() << " " << patch->GetReferrerMethodIdx() << " " 60100f7d0eaa6bd93d33bf0c1429bf4ba0b3f28abacIan Rogers << PrettyMethod(dex_cache->GetResolvedMethods()->Get(patch->GetTargetMethodIdx())) << " " 6020637e27af8e54f39634024dc77e8b094f21782f9Brian Carlstrom << PrettyMethod(method); 603f582258f0e296223a091fd64231a203ad71e9649Brian Carlstrom return method; 604f582258f0e296223a091fd64231a203ad71e9649Brian Carlstrom} 605f582258f0e296223a091fd64231a203ad71e9649Brian Carlstrom 6061212a022fa5f8ef9585d765b1809521812af882cIan Rogersvoid ImageWriter::PatchOatCodeAndMethods(const CompilerDriver& compiler) { 607e35517ad40e0713df8420aeb5d3540f146a59f62Mathieu Chartier Thread* self = Thread::Current(); 608f582258f0e296223a091fd64231a203ad71e9649Brian Carlstrom ClassLinker* class_linker = Runtime::Current()->GetClassLinker(); 609e35517ad40e0713df8420aeb5d3540f146a59f62Mathieu Chartier const char* old_cause = self->StartAssertNoThreadSuspension("ImageWriter"); 610f582258f0e296223a091fd64231a203ad71e9649Brian Carlstrom 6111212a022fa5f8ef9585d765b1809521812af882cIan Rogers typedef std::vector<const CompilerDriver::PatchInformation*> Patches; 612e35517ad40e0713df8420aeb5d3540f146a59f62Mathieu Chartier const Patches& code_to_patch = compiler.GetCodeToPatch(); 613f582258f0e296223a091fd64231a203ad71e9649Brian Carlstrom for (size_t i = 0; i < code_to_patch.size(); i++) { 6141212a022fa5f8ef9585d765b1809521812af882cIan Rogers const CompilerDriver::PatchInformation* patch = code_to_patch[i]; 61566f19258f9728d4ffe026074d8fd429d639802faMathieu Chartier AbstractMethod* target = GetTargetMethod(patch); 616f582258f0e296223a091fd64231a203ad71e9649Brian Carlstrom uint32_t code = reinterpret_cast<uint32_t>(class_linker->GetOatCodeFor(target)); 617f582258f0e296223a091fd64231a203ad71e9649Brian Carlstrom uint32_t code_base = reinterpret_cast<uint32_t>(&oat_file_->GetOatHeader()); 618f582258f0e296223a091fd64231a203ad71e9649Brian Carlstrom uint32_t code_offset = code - code_base; 619f582258f0e296223a091fd64231a203ad71e9649Brian Carlstrom SetPatchLocation(patch, reinterpret_cast<uint32_t>(GetOatAddress(code_offset))); 620f582258f0e296223a091fd64231a203ad71e9649Brian Carlstrom } 621f582258f0e296223a091fd64231a203ad71e9649Brian Carlstrom 622e35517ad40e0713df8420aeb5d3540f146a59f62Mathieu Chartier const Patches& methods_to_patch = compiler.GetMethodsToPatch(); 623f582258f0e296223a091fd64231a203ad71e9649Brian Carlstrom for (size_t i = 0; i < methods_to_patch.size(); i++) { 6241212a022fa5f8ef9585d765b1809521812af882cIan Rogers const CompilerDriver::PatchInformation* patch = methods_to_patch[i]; 62566f19258f9728d4ffe026074d8fd429d639802faMathieu Chartier AbstractMethod* target = GetTargetMethod(patch); 626f582258f0e296223a091fd64231a203ad71e9649Brian Carlstrom SetPatchLocation(patch, reinterpret_cast<uint32_t>(GetImageAddress(target))); 627f582258f0e296223a091fd64231a203ad71e9649Brian Carlstrom } 628a85b8376740641f13dc078aea6be94d15a800803Brian Carlstrom 629a85b8376740641f13dc078aea6be94d15a800803Brian Carlstrom // Update the image header with the new checksum after patching 630a85b8376740641f13dc078aea6be94d15a800803Brian Carlstrom ImageHeader* image_header = reinterpret_cast<ImageHeader*>(image_->Begin()); 631a85b8376740641f13dc078aea6be94d15a800803Brian Carlstrom image_header->SetOatChecksum(oat_file_->GetOatHeader().GetChecksum()); 632e35517ad40e0713df8420aeb5d3540f146a59f62Mathieu Chartier self->EndAssertNoThreadSuspension(old_cause); 633f582258f0e296223a091fd64231a203ad71e9649Brian Carlstrom} 634f582258f0e296223a091fd64231a203ad71e9649Brian Carlstrom 6351212a022fa5f8ef9585d765b1809521812af882cIan Rogersvoid ImageWriter::SetPatchLocation(const CompilerDriver::PatchInformation* patch, uint32_t value) { 636f582258f0e296223a091fd64231a203ad71e9649Brian Carlstrom ClassLinker* class_linker = Runtime::Current()->GetClassLinker(); 637e35517ad40e0713df8420aeb5d3540f146a59f62Mathieu Chartier const void* oat_code = class_linker->GetOatCodeFor(patch->GetDexFile(), 638e35517ad40e0713df8420aeb5d3540f146a59f62Mathieu Chartier patch->GetReferrerMethodIdx()); 639a85b8376740641f13dc078aea6be94d15a800803Brian Carlstrom OatHeader& oat_header = const_cast<OatHeader&>(oat_file_->GetOatHeader()); 640f582258f0e296223a091fd64231a203ad71e9649Brian Carlstrom // TODO: make this Thumb2 specific 641e35517ad40e0713df8420aeb5d3540f146a59f62Mathieu Chartier uint8_t* base = reinterpret_cast<uint8_t*>(reinterpret_cast<uint32_t>(oat_code) & ~0x1); 642f582258f0e296223a091fd64231a203ad71e9649Brian Carlstrom uint32_t* patch_location = reinterpret_cast<uint32_t*>(base + patch->GetLiteralOffset()); 643f582258f0e296223a091fd64231a203ad71e9649Brian Carlstrom#ifndef NDEBUG 644f582258f0e296223a091fd64231a203ad71e9649Brian Carlstrom const DexFile::MethodId& id = patch->GetDexFile().GetMethodId(patch->GetTargetMethodIdx()); 645f582258f0e296223a091fd64231a203ad71e9649Brian Carlstrom uint32_t expected = reinterpret_cast<uint32_t>(&id); 646f582258f0e296223a091fd64231a203ad71e9649Brian Carlstrom uint32_t actual = *patch_location; 647f582258f0e296223a091fd64231a203ad71e9649Brian Carlstrom CHECK(actual == expected || actual == value) << std::hex 648f582258f0e296223a091fd64231a203ad71e9649Brian Carlstrom << "actual=" << actual 649f582258f0e296223a091fd64231a203ad71e9649Brian Carlstrom << "expected=" << expected 650f582258f0e296223a091fd64231a203ad71e9649Brian Carlstrom << "value=" << value; 651f582258f0e296223a091fd64231a203ad71e9649Brian Carlstrom#endif 652b25c3f6a86dc634ce44fb2849385b49465caa84dElliott Hughes *patch_location = value; 653a85b8376740641f13dc078aea6be94d15a800803Brian Carlstrom oat_header.UpdateChecksum(patch_location, sizeof(value)); 654f582258f0e296223a091fd64231a203ad71e9649Brian Carlstrom} 655f582258f0e296223a091fd64231a203ad71e9649Brian Carlstrom 656db4d54081f09abcbe97ffdf615874f2809a9e777Brian Carlstrom} // namespace art 657