Elliott Hughes | 2faa5f1 | 2012-01-30 14:42:07 -0800 | [diff] [blame] | 1 | /* |
| 2 | * Copyright (C) 2011 The Android Open Source Project |
| 3 | * |
| 4 | * Licensed under the Apache License, Version 2.0 (the "License"); |
| 5 | * you may not use this file except in compliance with the License. |
| 6 | * You may obtain a copy of the License at |
| 7 | * |
| 8 | * http://www.apache.org/licenses/LICENSE-2.0 |
| 9 | * |
| 10 | * Unless required by applicable law or agreed to in writing, software |
| 11 | * distributed under the License is distributed on an "AS IS" BASIS, |
| 12 | * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. |
| 13 | * See the License for the specific language governing permissions and |
| 14 | * limitations under the License. |
| 15 | */ |
Brian Carlstrom | db4d540 | 2011-08-09 12:18:28 -0700 | [diff] [blame] | 16 | |
Brian Carlstrom | a1ce1fe | 2014-02-24 23:23:58 -0800 | [diff] [blame] | 17 | #include "image.h" |
| 18 | |
Ian Rogers | 700a402 | 2014-05-19 16:49:03 -0700 | [diff] [blame] | 19 | #include <memory> |
Ian Rogers | 0cfe1fb | 2011-08-26 03:29:44 -0700 | [diff] [blame] | 20 | #include <string> |
| 21 | #include <vector> |
| 22 | |
Ian Rogers | e63db27 | 2014-07-15 15:36:11 -0700 | [diff] [blame] | 23 | #include "base/unix_file/fd_file.h" |
Vladimir Marko | 3481ba2 | 2015-04-13 12:22:36 +0100 | [diff] [blame] | 24 | #include "class_linker-inl.h" |
Brian Carlstrom | a1ce1fe | 2014-02-24 23:23:58 -0800 | [diff] [blame] | 25 | #include "common_compiler_test.h" |
Vladimir Marko | 10c1356 | 2015-11-25 14:33:36 +0000 | [diff] [blame] | 26 | #include "dwarf/method_debug_info.h" |
Tong Shen | 62d1ca3 | 2014-09-03 17:24:56 -0700 | [diff] [blame] | 27 | #include "elf_writer.h" |
Vladimir Marko | 10c1356 | 2015-11-25 14:33:36 +0000 | [diff] [blame] | 28 | #include "elf_writer_quick.h" |
Ian Rogers | 1d54e73 | 2013-05-02 21:10:01 -0700 | [diff] [blame] | 29 | #include "gc/space/image_space.h" |
Ian Rogers | e63db27 | 2014-07-15 15:36:11 -0700 | [diff] [blame] | 30 | #include "image_writer.h" |
Ian Rogers | d9c4fc9 | 2013-10-01 19:45:43 -0700 | [diff] [blame] | 31 | #include "lock_word.h" |
| 32 | #include "mirror/object-inl.h" |
Ian Rogers | e63db27 | 2014-07-15 15:36:11 -0700 | [diff] [blame] | 33 | #include "oat_writer.h" |
| 34 | #include "scoped_thread_state_change.h" |
Brian Carlstrom | 51c2467 | 2013-07-11 16:00:56 -0700 | [diff] [blame] | 35 | #include "signal_catcher.h" |
buzbee | c143c55 | 2011-08-20 17:38:58 -0700 | [diff] [blame] | 36 | #include "utils.h" |
Brian Carlstrom | db4d540 | 2011-08-09 12:18:28 -0700 | [diff] [blame] | 37 | |
Brian Carlstrom | db4d540 | 2011-08-09 12:18:28 -0700 | [diff] [blame] | 38 | namespace art { |
| 39 | |
Brian Carlstrom | a1ce1fe | 2014-02-24 23:23:58 -0800 | [diff] [blame] | 40 | class ImageTest : public CommonCompilerTest { |
Ian Rogers | 10c5b78 | 2013-01-10 10:40:53 -0800 | [diff] [blame] | 41 | protected: |
| 42 | virtual void SetUp() { |
Brian Carlstrom | 700c8d3 | 2012-11-05 10:42:02 -0800 | [diff] [blame] | 43 | ReserveImageSpace(); |
Brian Carlstrom | a1ce1fe | 2014-02-24 23:23:58 -0800 | [diff] [blame] | 44 | CommonCompilerTest::SetUp(); |
Ian Rogers | 10c5b78 | 2013-01-10 10:40:53 -0800 | [diff] [blame] | 45 | } |
Mathieu Chartier | ceb07b3 | 2015-12-10 09:33:21 -0800 | [diff] [blame] | 46 | void TestWriteRead(ImageHeader::StorageMode storage_mode); |
Ian Rogers | 10c5b78 | 2013-01-10 10:40:53 -0800 | [diff] [blame] | 47 | }; |
Brian Carlstrom | db4d540 | 2011-08-09 12:18:28 -0700 | [diff] [blame] | 48 | |
Mathieu Chartier | ceb07b3 | 2015-12-10 09:33:21 -0800 | [diff] [blame] | 49 | void ImageTest::TestWriteRead(ImageHeader::StorageMode storage_mode) { |
| 50 | // TODO: Test does not currently work with optimizing. |
| 51 | CreateCompilerDriver(Compiler::kQuick, kRuntimeISA); |
| 52 | ClassLinker* class_linker = Runtime::Current()->GetClassLinker(); |
| 53 | // Enable write for dex2dex. |
| 54 | for (const DexFile* dex_file : class_linker->GetBootClassPath()) { |
| 55 | dex_file->EnableWrite(); |
| 56 | } |
Brian Carlstrom | 0e12bdc | 2014-05-14 17:44:28 -0700 | [diff] [blame] | 57 | // Create a generic location tmp file, to be the base of the .art and .oat temporary files. |
| 58 | ScratchFile location; |
| 59 | ScratchFile image_location(location, ".art"); |
| 60 | |
| 61 | std::string image_filename(GetSystemImageFilename(image_location.GetFilename().c_str(), |
| 62 | kRuntimeISA)); |
| 63 | size_t pos = image_filename.rfind('/'); |
| 64 | CHECK_NE(pos, std::string::npos) << image_filename; |
| 65 | std::string image_dir(image_filename, 0, pos); |
| 66 | int mkdir_result = mkdir(image_dir.c_str(), 0700); |
| 67 | CHECK_EQ(0, mkdir_result) << image_dir; |
| 68 | ScratchFile image_file(OS::CreateEmptyFile(image_filename.c_str())); |
| 69 | |
| 70 | std::string oat_filename(image_filename, 0, image_filename.size() - 3); |
| 71 | oat_filename += "oat"; |
| 72 | ScratchFile oat_file(OS::CreateEmptyFile(oat_filename.c_str())); |
| 73 | |
Vladimir Marko | f4da675 | 2014-08-01 19:04:18 +0100 | [diff] [blame] | 74 | const uintptr_t requested_image_base = ART_BASE_ADDRESS; |
Jeff Hao | dcdc85b | 2015-12-04 14:06:18 -0800 | [diff] [blame] | 75 | std::unordered_map<const DexFile*, const char*> dex_file_to_oat_filename_map; |
| 76 | std::vector<const char*> oat_filename_vector(1, oat_filename.c_str()); |
| 77 | for (const DexFile* dex_file : class_linker->GetBootClassPath()) { |
| 78 | dex_file_to_oat_filename_map.emplace(dex_file, oat_filename.c_str()); |
| 79 | } |
Mathieu Chartier | da5b28a | 2015-11-05 08:03:47 -0800 | [diff] [blame] | 80 | std::unique_ptr<ImageWriter> writer(new ImageWriter(*compiler_driver_, |
| 81 | requested_image_base, |
| 82 | /*compile_pic*/false, |
Mathieu Chartier | ceb07b3 | 2015-12-10 09:33:21 -0800 | [diff] [blame] | 83 | /*compile_app_image*/false, |
Jeff Hao | dcdc85b | 2015-12-04 14:06:18 -0800 | [diff] [blame] | 84 | storage_mode, |
| 85 | oat_filename_vector, |
| 86 | dex_file_to_oat_filename_map)); |
Igor Murashkin | 4677476 | 2014-10-22 11:37:02 -0700 | [diff] [blame] | 87 | // TODO: compile_pic should be a test argument. |
Ian Rogers | 00f7d0e | 2012-07-19 15:28:27 -0700 | [diff] [blame] | 88 | { |
Brian Carlstrom | 700c8d3 | 2012-11-05 10:42:02 -0800 | [diff] [blame] | 89 | { |
Mathieu Chartier | 2cebb24 | 2015-04-21 16:50:40 -0700 | [diff] [blame] | 90 | jobject class_loader = nullptr; |
Ian Rogers | 5fe9af7 | 2013-11-14 00:17:20 -0800 | [diff] [blame] | 91 | TimingLogger timings("ImageTest::WriteRead", false, false); |
Mathieu Chartier | f5997b4 | 2014-06-20 10:37:54 -0700 | [diff] [blame] | 92 | TimingLogger::ScopedTiming t("CompileAll", &timings); |
Vladimir Marko | d1eaf0d | 2015-10-29 12:18:29 +0000 | [diff] [blame] | 93 | compiler_driver_->SetDexFilesForOatFile(class_linker->GetBootClassPath()); |
Ian Rogers | 3d50407 | 2014-03-01 09:16:49 -0800 | [diff] [blame] | 94 | compiler_driver_->CompileAll(class_loader, class_linker->GetBootClassPath(), &timings); |
Brian Carlstrom | 9639160 | 2013-06-13 19:49:50 -0700 | [diff] [blame] | 95 | |
Mathieu Chartier | f5997b4 | 2014-06-20 10:37:54 -0700 | [diff] [blame] | 96 | t.NewTiming("WriteElf"); |
Andreas Gampe | 22f8e5c | 2014-07-09 11:38:21 -0700 | [diff] [blame] | 97 | SafeMap<std::string, std::string> key_value_store; |
Vladimir Marko | 9bdf108 | 2016-01-21 12:15:52 +0000 | [diff] [blame] | 98 | const std::vector<const DexFile*>& dex_files = class_linker->GetBootClassPath(); |
Vladimir Marko | 10c1356 | 2015-11-25 14:33:36 +0000 | [diff] [blame] | 99 | std::unique_ptr<ElfWriter> elf_writer = CreateElfWriterQuick( |
| 100 | compiler_driver_->GetInstructionSet(), |
| 101 | &compiler_driver_->GetCompilerOptions(), |
| 102 | oat_file.GetFile()); |
Vladimir Marko | 919f553 | 2016-01-20 19:13:01 +0000 | [diff] [blame] | 103 | elf_writer->Start(); |
Vladimir Marko | 9bdf108 | 2016-01-21 12:15:52 +0000 | [diff] [blame] | 104 | OatWriter oat_writer(/*compiling_boot_image*/true, &timings); |
Vladimir Marko | 919f553 | 2016-01-20 19:13:01 +0000 | [diff] [blame] | 105 | OutputStream* rodata = elf_writer->StartRoData(); |
Vladimir Marko | 9bdf108 | 2016-01-21 12:15:52 +0000 | [diff] [blame] | 106 | for (const DexFile* dex_file : dex_files) { |
| 107 | ArrayRef<const uint8_t> raw_dex_file( |
| 108 | reinterpret_cast<const uint8_t*>(&dex_file->GetHeader()), |
| 109 | dex_file->GetHeader().file_size_); |
| 110 | oat_writer.AddRawDexFileSource(raw_dex_file, |
| 111 | dex_file->GetLocation().c_str(), |
| 112 | dex_file->GetLocationChecksum()); |
| 113 | } |
| 114 | std::unique_ptr<MemMap> opened_dex_files_map; |
| 115 | std::vector<std::unique_ptr<const DexFile>> opened_dex_files; |
| 116 | bool dex_files_ok = oat_writer.WriteAndOpenDexFiles( |
| 117 | rodata, |
| 118 | oat_file.GetFile(), |
| 119 | compiler_driver_->GetInstructionSet(), |
| 120 | compiler_driver_->GetInstructionSetFeatures(), |
| 121 | &key_value_store, |
| 122 | &opened_dex_files_map, |
| 123 | &opened_dex_files); |
| 124 | ASSERT_TRUE(dex_files_ok); |
| 125 | oat_writer.PrepareLayout(compiler_driver_.get(), writer.get(), dex_files); |
| 126 | bool image_space_ok = writer->PrepareImageAddressSpace(); |
| 127 | ASSERT_TRUE(image_space_ok); |
| 128 | |
Vladimir Marko | 10c1356 | 2015-11-25 14:33:36 +0000 | [diff] [blame] | 129 | bool rodata_ok = oat_writer.WriteRodata(rodata); |
| 130 | ASSERT_TRUE(rodata_ok); |
| 131 | elf_writer->EndRoData(rodata); |
| 132 | |
| 133 | OutputStream* text = elf_writer->StartText(); |
| 134 | bool text_ok = oat_writer.WriteCode(text); |
| 135 | ASSERT_TRUE(text_ok); |
| 136 | elf_writer->EndText(text); |
| 137 | |
Vladimir Marko | 9bdf108 | 2016-01-21 12:15:52 +0000 | [diff] [blame] | 138 | bool header_ok = oat_writer.WriteHeader(elf_writer->GetStream(), 0u, 0u, 0u); |
| 139 | ASSERT_TRUE(header_ok); |
| 140 | |
Vladimir Marko | 10c1356 | 2015-11-25 14:33:36 +0000 | [diff] [blame] | 141 | elf_writer->SetBssSize(oat_writer.GetBssSize()); |
Vladimir Marko | 10c1356 | 2015-11-25 14:33:36 +0000 | [diff] [blame] | 142 | elf_writer->WriteDynamicSection(); |
Vladimir Marko | 49b0f45 | 2015-12-10 13:49:19 +0000 | [diff] [blame] | 143 | elf_writer->WriteDebugInfo(oat_writer.GetMethodDebugInfo()); |
| 144 | elf_writer->WritePatchLocations(oat_writer.GetAbsolutePatchLocations()); |
Vladimir Marko | 10c1356 | 2015-11-25 14:33:36 +0000 | [diff] [blame] | 145 | |
Vladimir Marko | 9bdf108 | 2016-01-21 12:15:52 +0000 | [diff] [blame] | 146 | bool success = elf_writer->End(); |
Vladimir Marko | 10c1356 | 2015-11-25 14:33:36 +0000 | [diff] [blame] | 147 | |
Brian Carlstrom | c50d8e1 | 2013-07-23 22:35:16 -0700 | [diff] [blame] | 148 | ASSERT_TRUE(success); |
Ian Rogers | 00f7d0e | 2012-07-19 15:28:27 -0700 | [diff] [blame] | 149 | } |
Ian Rogers | 0571d35 | 2011-11-03 19:51:38 -0700 | [diff] [blame] | 150 | } |
Brian Carlstrom | 0e12bdc | 2014-05-14 17:44:28 -0700 | [diff] [blame] | 151 | // Workound bug that mcld::Linker::emit closes oat_file by reopening as dup_oat. |
Ian Rogers | 700a402 | 2014-05-19 16:49:03 -0700 | [diff] [blame] | 152 | std::unique_ptr<File> dup_oat(OS::OpenFileReadWrite(oat_file.GetFilename().c_str())); |
Mathieu Chartier | 2cebb24 | 2015-04-21 16:50:40 -0700 | [diff] [blame] | 153 | ASSERT_TRUE(dup_oat.get() != nullptr); |
Mathieu Chartier | 357e9be | 2012-08-01 11:00:14 -0700 | [diff] [blame] | 154 | |
jeffhao | 8161c03 | 2012-10-31 15:50:00 -0700 | [diff] [blame] | 155 | { |
Jeff Hao | dcdc85b | 2015-12-04 14:06:18 -0800 | [diff] [blame] | 156 | std::vector<const char*> dup_oat_filename(1, dup_oat->GetPath().c_str()); |
| 157 | std::vector<const char*> dup_image_filename(1, image_file.GetFilename().c_str()); |
Mathieu Chartier | fbc3108 | 2016-01-24 11:59:56 -0800 | [diff] [blame] | 158 | bool success_image = writer->Write(kInvalidFd, |
| 159 | dup_image_filename, |
| 160 | kInvalidFd, |
| 161 | dup_oat_filename, |
| 162 | dup_oat_filename[0]); |
jeffhao | 8161c03 | 2012-10-31 15:50:00 -0700 | [diff] [blame] | 163 | ASSERT_TRUE(success_image); |
Jeff Hao | dcdc85b | 2015-12-04 14:06:18 -0800 | [diff] [blame] | 164 | bool success_fixup = ElfWriter::Fixup(dup_oat.get(), |
| 165 | writer->GetOatDataBegin(dup_oat_filename[0])); |
Brian Carlstrom | 700c8d3 | 2012-11-05 10:42:02 -0800 | [diff] [blame] | 166 | ASSERT_TRUE(success_fixup); |
Andreas Gampe | 4303ba9 | 2014-11-06 01:00:46 -0800 | [diff] [blame] | 167 | |
| 168 | ASSERT_EQ(dup_oat->FlushCloseOrErase(), 0) << "Could not flush and close oat file " |
| 169 | << oat_file.GetFilename(); |
jeffhao | 8161c03 | 2012-10-31 15:50:00 -0700 | [diff] [blame] | 170 | } |
Brian Carlstrom | 4a289ed | 2011-08-16 17:17:49 -0700 | [diff] [blame] | 171 | |
Mathieu Chartier | 68c868e | 2015-06-01 16:33:53 -0700 | [diff] [blame] | 172 | uint64_t image_file_size; |
Brian Carlstrom | 4a289ed | 2011-08-16 17:17:49 -0700 | [diff] [blame] | 173 | { |
Ian Rogers | 700a402 | 2014-05-19 16:49:03 -0700 | [diff] [blame] | 174 | std::unique_ptr<File> file(OS::OpenFileForReading(image_file.GetFilename().c_str())); |
Mathieu Chartier | 2cebb24 | 2015-04-21 16:50:40 -0700 | [diff] [blame] | 175 | ASSERT_TRUE(file.get() != nullptr); |
Brian Carlstrom | 4a289ed | 2011-08-16 17:17:49 -0700 | [diff] [blame] | 176 | ImageHeader image_header; |
Andreas Gampe | 4303ba9 | 2014-11-06 01:00:46 -0800 | [diff] [blame] | 177 | ASSERT_EQ(file->ReadFully(&image_header, sizeof(image_header)), true); |
Brian Carlstrom | 4a289ed | 2011-08-16 17:17:49 -0700 | [diff] [blame] | 178 | ASSERT_TRUE(image_header.IsValid()); |
Mathieu Chartier | e401d14 | 2015-04-22 13:56:20 -0700 | [diff] [blame] | 179 | const auto& bitmap_section = image_header.GetImageSection(ImageHeader::kSectionImageBitmap); |
| 180 | ASSERT_GE(bitmap_section.Offset(), sizeof(image_header)); |
| 181 | ASSERT_NE(0U, bitmap_section.Size()); |
Brian Carlstrom | 69b15fb | 2011-09-03 12:25:21 -0700 | [diff] [blame] | 182 | |
Ian Rogers | 1d54e73 | 2013-05-02 21:10:01 -0700 | [diff] [blame] | 183 | gc::Heap* heap = Runtime::Current()->GetHeap(); |
Mathieu Chartier | a9d82fe | 2016-01-25 20:06:11 -0800 | [diff] [blame] | 184 | ASSERT_TRUE(heap->HaveContinuousSpaces()); |
Mathieu Chartier | 590fee9 | 2013-09-13 13:46:47 -0700 | [diff] [blame] | 185 | gc::space::ContinuousSpace* space = heap->GetNonMovingSpace(); |
Brian Carlstrom | 3320cf4 | 2011-10-04 14:58:28 -0700 | [diff] [blame] | 186 | ASSERT_FALSE(space->IsImageSpace()); |
Mathieu Chartier | 2cebb24 | 2015-04-21 16:50:40 -0700 | [diff] [blame] | 187 | ASSERT_TRUE(space != nullptr); |
Hiroshi Yamauchi | cf58d4a | 2013-09-26 14:21:22 -0700 | [diff] [blame] | 188 | ASSERT_TRUE(space->IsMallocSpace()); |
Mathieu Chartier | 68c868e | 2015-06-01 16:33:53 -0700 | [diff] [blame] | 189 | |
| 190 | image_file_size = file->GetLength(); |
Brian Carlstrom | 4a289ed | 2011-08-16 17:17:49 -0700 | [diff] [blame] | 191 | } |
Brian Carlstrom | 8a43659 | 2011-08-15 21:27:23 -0700 | [diff] [blame] | 192 | |
Mathieu Chartier | 2cebb24 | 2015-04-21 16:50:40 -0700 | [diff] [blame] | 193 | ASSERT_TRUE(compiler_driver_->GetImageClasses() != nullptr); |
Andreas Gampe | b1fcead | 2015-04-20 18:53:51 -0700 | [diff] [blame] | 194 | std::unordered_set<std::string> image_classes(*compiler_driver_->GetImageClasses()); |
Brian Carlstrom | 9639160 | 2013-06-13 19:49:50 -0700 | [diff] [blame] | 195 | |
Mathieu Chartier | 0e4627e | 2012-10-23 16:13:36 -0700 | [diff] [blame] | 196 | // Need to delete the compiler since it has worker threads which are attached to runtime. |
Ian Rogers | 1212a02 | 2013-03-04 10:48:41 -0800 | [diff] [blame] | 197 | compiler_driver_.reset(); |
Mathieu Chartier | 0e4627e | 2012-10-23 16:13:36 -0700 | [diff] [blame] | 198 | |
Ian Rogers | 10c5b78 | 2013-01-10 10:40:53 -0800 | [diff] [blame] | 199 | // Tear down old runtime before making a new one, clearing out misc state. |
Mathieu Chartier | 6e88ef6 | 2014-10-14 15:01:24 -0700 | [diff] [blame] | 200 | |
| 201 | // Remove the reservation of the memory for use to load the image. |
| 202 | // Need to do this before we reset the runtime. |
| 203 | UnreserveImageSpace(); |
| 204 | writer.reset(nullptr); |
| 205 | |
Ian Rogers | 10c5b78 | 2013-01-10 10:40:53 -0800 | [diff] [blame] | 206 | runtime_.reset(); |
Mathieu Chartier | 2cebb24 | 2015-04-21 16:50:40 -0700 | [diff] [blame] | 207 | java_lang_dex_file_ = nullptr; |
Brian Carlstrom | 9cff8e1 | 2011-08-18 16:47:29 -0700 | [diff] [blame] | 208 | |
Mathieu Chartier | 6e88ef6 | 2014-10-14 15:01:24 -0700 | [diff] [blame] | 209 | MemMap::Init(); |
Przemyslaw Szczepaniak | 5c0cf12 | 2015-09-30 17:41:38 +0100 | [diff] [blame] | 210 | std::unique_ptr<const DexFile> dex(LoadExpectSingleDexFile(GetLibCoreDexFileNames()[0].c_str())); |
Brian Carlstrom | 8a43659 | 2011-08-15 21:27:23 -0700 | [diff] [blame] | 211 | |
Ian Rogers | e63db27 | 2014-07-15 15:36:11 -0700 | [diff] [blame] | 212 | RuntimeOptions options; |
Brian Carlstrom | 58ae941 | 2011-10-04 00:56:06 -0700 | [diff] [blame] | 213 | std::string image("-Ximage:"); |
Brian Carlstrom | 0e12bdc | 2014-05-14 17:44:28 -0700 | [diff] [blame] | 214 | image.append(image_location.GetFilename()); |
Mathieu Chartier | 2cebb24 | 2015-04-21 16:50:40 -0700 | [diff] [blame] | 215 | options.push_back(std::make_pair(image.c_str(), static_cast<void*>(nullptr))); |
Alex Light | 6e183f2 | 2014-07-18 14:57:04 -0700 | [diff] [blame] | 216 | // By default the compiler this creates will not include patch information. |
| 217 | options.push_back(std::make_pair("-Xnorelocate", nullptr)); |
Brian Carlstrom | 8a43659 | 2011-08-15 21:27:23 -0700 | [diff] [blame] | 218 | |
Ian Rogers | 00f7d0e | 2012-07-19 15:28:27 -0700 | [diff] [blame] | 219 | if (!Runtime::Create(options, false)) { |
| 220 | LOG(FATAL) << "Failed to create runtime"; |
| 221 | return; |
| 222 | } |
| 223 | runtime_.reset(Runtime::Current()); |
| 224 | // Runtime::Create acquired the mutator_lock_ that is normally given away when we Runtime::Start, |
| 225 | // give it away now and then switch to a more managable ScopedObjectAccess. |
| 226 | Thread::Current()->TransitionFromRunnableToSuspended(kNative); |
| 227 | ScopedObjectAccess soa(Thread::Current()); |
Mathieu Chartier | 2cebb24 | 2015-04-21 16:50:40 -0700 | [diff] [blame] | 228 | ASSERT_TRUE(runtime_.get() != nullptr); |
Brian Carlstrom | 8a43659 | 2011-08-15 21:27:23 -0700 | [diff] [blame] | 229 | class_linker_ = runtime_->GetClassLinker(); |
Brian Carlstrom | 9cff8e1 | 2011-08-18 16:47:29 -0700 | [diff] [blame] | 230 | |
Ian Rogers | 1d54e73 | 2013-05-02 21:10:01 -0700 | [diff] [blame] | 231 | gc::Heap* heap = Runtime::Current()->GetHeap(); |
Jeff Hao | dcdc85b | 2015-12-04 14:06:18 -0800 | [diff] [blame] | 232 | ASSERT_TRUE(heap->HasBootImageSpace()); |
Hiroshi Yamauchi | cf58d4a | 2013-09-26 14:21:22 -0700 | [diff] [blame] | 233 | ASSERT_TRUE(heap->GetNonMovingSpace()->IsMallocSpace()); |
Brian Carlstrom | 9cff8e1 | 2011-08-18 16:47:29 -0700 | [diff] [blame] | 234 | |
Jeff Hao | dcdc85b | 2015-12-04 14:06:18 -0800 | [diff] [blame] | 235 | // We loaded the runtime with an explicit image, so it must exist. |
| 236 | gc::space::ImageSpace* image_space = heap->GetBootImageSpaces()[0]; |
Mathieu Chartier | 68c868e | 2015-06-01 16:33:53 -0700 | [diff] [blame] | 237 | ASSERT_TRUE(image_space != nullptr); |
Mathieu Chartier | ceb07b3 | 2015-12-10 09:33:21 -0800 | [diff] [blame] | 238 | if (storage_mode == ImageHeader::kStorageModeUncompressed) { |
| 239 | // Uncompressed, image should be smaller than file. |
| 240 | ASSERT_LE(image_space->Size(), image_file_size); |
| 241 | } else { |
| 242 | // Compressed, file should be smaller than image. |
| 243 | ASSERT_LE(image_file_size, image_space->Size()); |
| 244 | } |
Mathieu Chartier | 68c868e | 2015-06-01 16:33:53 -0700 | [diff] [blame] | 245 | |
Mathieu Chartier | 31e8925 | 2013-08-28 11:29:12 -0700 | [diff] [blame] | 246 | image_space->VerifyImageAllocations(); |
Ian Rogers | 1373595 | 2014-10-08 12:43:28 -0700 | [diff] [blame] | 247 | uint8_t* image_begin = image_space->Begin(); |
| 248 | uint8_t* image_end = image_space->End(); |
Ian Rogers | 30fab40 | 2012-01-23 15:43:46 -0800 | [diff] [blame] | 249 | CHECK_EQ(requested_image_base, reinterpret_cast<uintptr_t>(image_begin)); |
Elliott Hughes | b3bd5f0 | 2012-03-08 21:05:27 -0800 | [diff] [blame] | 250 | for (size_t i = 0; i < dex->NumClassDefs(); ++i) { |
Brian Carlstrom | d2fbb2b | 2011-08-23 11:57:08 -0700 | [diff] [blame] | 251 | const DexFile::ClassDef& class_def = dex->GetClassDef(i); |
Brian Carlstrom | 9cff8e1 | 2011-08-18 16:47:29 -0700 | [diff] [blame] | 252 | const char* descriptor = dex->GetClassDescriptor(class_def); |
Ian Rogers | 9837939 | 2014-02-24 16:53:16 -0800 | [diff] [blame] | 253 | mirror::Class* klass = class_linker_->FindSystemClass(soa.Self(), descriptor); |
Mathieu Chartier | c528dba | 2013-11-26 12:00:11 -0800 | [diff] [blame] | 254 | EXPECT_TRUE(klass != nullptr) << descriptor; |
Brian Carlstrom | 9639160 | 2013-06-13 19:49:50 -0700 | [diff] [blame] | 255 | if (image_classes.find(descriptor) != image_classes.end()) { |
Mathieu Chartier | c528dba | 2013-11-26 12:00:11 -0800 | [diff] [blame] | 256 | // Image classes should be located inside the image. |
Ian Rogers | 1373595 | 2014-10-08 12:43:28 -0700 | [diff] [blame] | 257 | EXPECT_LT(image_begin, reinterpret_cast<uint8_t*>(klass)) << descriptor; |
| 258 | EXPECT_LT(reinterpret_cast<uint8_t*>(klass), image_end) << descriptor; |
Brian Carlstrom | 9639160 | 2013-06-13 19:49:50 -0700 | [diff] [blame] | 259 | } else { |
Ian Rogers | 1373595 | 2014-10-08 12:43:28 -0700 | [diff] [blame] | 260 | EXPECT_TRUE(reinterpret_cast<uint8_t*>(klass) >= image_end || |
| 261 | reinterpret_cast<uint8_t*>(klass) < image_begin) << descriptor; |
Brian Carlstrom | 9639160 | 2013-06-13 19:49:50 -0700 | [diff] [blame] | 262 | } |
Mathieu Chartier | 4d7f61d | 2014-04-17 14:43:39 -0700 | [diff] [blame] | 263 | EXPECT_TRUE(Monitor::IsValidLockWord(klass->GetLockWord(false))); |
Brian Carlstrom | 9cff8e1 | 2011-08-18 16:47:29 -0700 | [diff] [blame] | 264 | } |
Brian Carlstrom | 0e12bdc | 2014-05-14 17:44:28 -0700 | [diff] [blame] | 265 | |
| 266 | image_file.Unlink(); |
| 267 | oat_file.Unlink(); |
| 268 | int rmdir_result = rmdir(image_dir.c_str()); |
| 269 | CHECK_EQ(0, rmdir_result); |
Brian Carlstrom | db4d540 | 2011-08-09 12:18:28 -0700 | [diff] [blame] | 270 | } |
| 271 | |
Mathieu Chartier | ceb07b3 | 2015-12-10 09:33:21 -0800 | [diff] [blame] | 272 | TEST_F(ImageTest, WriteReadUncompressed) { |
| 273 | TestWriteRead(ImageHeader::kStorageModeUncompressed); |
| 274 | } |
| 275 | |
| 276 | TEST_F(ImageTest, WriteReadLZ4) { |
| 277 | TestWriteRead(ImageHeader::kStorageModeLZ4); |
| 278 | } |
| 279 | |
Brian Carlstrom | 179486a | 2013-09-03 11:51:42 -0700 | [diff] [blame] | 280 | TEST_F(ImageTest, ImageHeaderIsValid) { |
| 281 | uint32_t image_begin = ART_BASE_ADDRESS; |
| 282 | uint32_t image_size_ = 16 * KB; |
Brian Carlstrom | 179486a | 2013-09-03 11:51:42 -0700 | [diff] [blame] | 283 | uint32_t image_roots = ART_BASE_ADDRESS + (1 * KB); |
| 284 | uint32_t oat_checksum = 0; |
| 285 | uint32_t oat_file_begin = ART_BASE_ADDRESS + (4 * KB); // page aligned |
| 286 | uint32_t oat_data_begin = ART_BASE_ADDRESS + (8 * KB); // page aligned |
| 287 | uint32_t oat_data_end = ART_BASE_ADDRESS + (9 * KB); |
| 288 | uint32_t oat_file_end = ART_BASE_ADDRESS + (10 * KB); |
Mathieu Chartier | e401d14 | 2015-04-22 13:56:20 -0700 | [diff] [blame] | 289 | ImageSection sections[ImageHeader::kSectionCount]; |
Brian Carlstrom | 179486a | 2013-09-03 11:51:42 -0700 | [diff] [blame] | 290 | ImageHeader image_header(image_begin, |
| 291 | image_size_, |
Mathieu Chartier | e401d14 | 2015-04-22 13:56:20 -0700 | [diff] [blame] | 292 | sections, |
Brian Carlstrom | 179486a | 2013-09-03 11:51:42 -0700 | [diff] [blame] | 293 | image_roots, |
| 294 | oat_checksum, |
| 295 | oat_file_begin, |
| 296 | oat_data_begin, |
| 297 | oat_data_end, |
Igor Murashkin | 4677476 | 2014-10-22 11:37:02 -0700 | [diff] [blame] | 298 | oat_file_end, |
Mathieu Chartier | fbc3108 | 2016-01-24 11:59:56 -0800 | [diff] [blame] | 299 | /*boot_image_begin*/0U, |
| 300 | /*boot_image_size*/0U, |
| 301 | /*boot_oat_begin*/0U, |
| 302 | /*boot_oat_size_*/0U, |
Mathieu Chartier | e401d14 | 2015-04-22 13:56:20 -0700 | [diff] [blame] | 303 | sizeof(void*), |
Mathieu Chartier | ceb07b3 | 2015-12-10 09:33:21 -0800 | [diff] [blame] | 304 | /*compile_pic*/false, |
Mathieu Chartier | fbc3108 | 2016-01-24 11:59:56 -0800 | [diff] [blame] | 305 | /*is_pic*/false, |
Mathieu Chartier | ceb07b3 | 2015-12-10 09:33:21 -0800 | [diff] [blame] | 306 | ImageHeader::kDefaultStorageMode, |
| 307 | /*data_size*/0u); |
Brian Carlstrom | 179486a | 2013-09-03 11:51:42 -0700 | [diff] [blame] | 308 | ASSERT_TRUE(image_header.IsValid()); |
Mathieu Chartier | fbc3108 | 2016-01-24 11:59:56 -0800 | [diff] [blame] | 309 | ASSERT_TRUE(!image_header.IsAppImage()); |
Brian Carlstrom | 179486a | 2013-09-03 11:51:42 -0700 | [diff] [blame] | 310 | |
| 311 | char* magic = const_cast<char*>(image_header.GetMagic()); |
| 312 | strcpy(magic, ""); // bad magic |
| 313 | ASSERT_FALSE(image_header.IsValid()); |
| 314 | strcpy(magic, "art\n000"); // bad version |
| 315 | ASSERT_FALSE(image_header.IsValid()); |
| 316 | } |
| 317 | |
Brian Carlstrom | db4d540 | 2011-08-09 12:18:28 -0700 | [diff] [blame] | 318 | } // namespace art |