Elliott Hughes | 2faa5f1 | 2012-01-30 14:42:07 -0800 | [diff] [blame] | 1 | /* |
| 2 | * Copyright (C) 2011 The Android Open Source Project |
| 3 | * |
| 4 | * Licensed under the Apache License, Version 2.0 (the "License"); |
| 5 | * you may not use this file except in compliance with the License. |
| 6 | * You may obtain a copy of the License at |
| 7 | * |
| 8 | * http://www.apache.org/licenses/LICENSE-2.0 |
| 9 | * |
| 10 | * Unless required by applicable law or agreed to in writing, software |
| 11 | * distributed under the License is distributed on an "AS IS" BASIS, |
| 12 | * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. |
| 13 | * See the License for the specific language governing permissions and |
| 14 | * limitations under the License. |
| 15 | */ |
Brian Carlstrom | e24fa61 | 2011-09-29 00:53:55 -0700 | [diff] [blame] | 16 | |
| 17 | #include "oat_file.h" |
| 18 | #include "oat_writer.h" |
| 19 | |
| 20 | #include "common_test.h" |
| 21 | |
| 22 | namespace art { |
| 23 | |
Logan Chien | eeb7edf | 2012-03-09 20:38:39 +0800 | [diff] [blame] | 24 | class OatTest : public CommonTest { |
| 25 | protected: |
Mathieu Chartier | 66f1925 | 2012-09-18 08:57:04 -0700 | [diff] [blame] | 26 | void CheckMethod(AbstractMethod* method, |
Logan Chien | eeb7edf | 2012-03-09 20:38:39 +0800 | [diff] [blame] | 27 | const OatFile::OatMethod& oat_method, |
Ian Rogers | 00f7d0e | 2012-07-19 15:28:27 -0700 | [diff] [blame] | 28 | const DexFile* dex_file) |
Ian Rogers | b726dcb | 2012-09-05 08:57:23 -0700 | [diff] [blame] | 29 | SHARED_LOCKS_REQUIRED(Locks::mutator_lock_) { |
Logan Chien | eeb7edf | 2012-03-09 20:38:39 +0800 | [diff] [blame] | 30 | const CompiledMethod* compiled_method = |
| 31 | compiler_->GetCompiledMethod(Compiler::MethodReference(dex_file, |
| 32 | method->GetDexMethodIndex())); |
| 33 | |
| 34 | if (compiled_method == NULL) { |
| 35 | EXPECT_TRUE(oat_method.GetCode() == NULL) << PrettyMethod(method) << " " |
| 36 | << oat_method.GetCode(); |
TDYa127 | 42d4d73 | 2012-03-23 19:10:56 -0700 | [diff] [blame] | 37 | #if !defined(ART_USE_LLVM_COMPILER) |
Logan Chien | eeb7edf | 2012-03-09 20:38:39 +0800 | [diff] [blame] | 38 | EXPECT_EQ(oat_method.GetFrameSizeInBytes(), static_cast<uint32_t>(kStackAlignment)); |
| 39 | EXPECT_EQ(oat_method.GetCoreSpillMask(), 0U); |
| 40 | EXPECT_EQ(oat_method.GetFpSpillMask(), 0U); |
TDYa127 | 42d4d73 | 2012-03-23 19:10:56 -0700 | [diff] [blame] | 41 | #endif |
Logan Chien | eeb7edf | 2012-03-09 20:38:39 +0800 | [diff] [blame] | 42 | } else { |
| 43 | const void* oat_code = oat_method.GetCode(); |
| 44 | EXPECT_TRUE(oat_code != NULL) << PrettyMethod(method); |
| 45 | uintptr_t oat_code_aligned = RoundDown(reinterpret_cast<uintptr_t>(oat_code), 2); |
| 46 | oat_code = reinterpret_cast<const void*>(oat_code_aligned); |
| 47 | |
| 48 | const std::vector<uint8_t>& code = compiled_method->GetCode(); |
| 49 | size_t code_size = code.size() * sizeof(code[0]); |
| 50 | EXPECT_EQ(0, memcmp(oat_code, &code[0], code_size)) |
| 51 | << PrettyMethod(method) << " " << code_size; |
| 52 | CHECK_EQ(0, memcmp(oat_code, &code[0], code_size)); |
TDYa127 | 42d4d73 | 2012-03-23 19:10:56 -0700 | [diff] [blame] | 53 | #if !defined(ART_USE_LLVM_COMPILER) |
Logan Chien | eeb7edf | 2012-03-09 20:38:39 +0800 | [diff] [blame] | 54 | EXPECT_EQ(oat_method.GetFrameSizeInBytes(), compiled_method->GetFrameSizeInBytes()); |
| 55 | EXPECT_EQ(oat_method.GetCoreSpillMask(), compiled_method->GetCoreSpillMask()); |
| 56 | EXPECT_EQ(oat_method.GetFpSpillMask(), compiled_method->GetFpSpillMask()); |
TDYa127 | 42d4d73 | 2012-03-23 19:10:56 -0700 | [diff] [blame] | 57 | #endif |
Logan Chien | eeb7edf | 2012-03-09 20:38:39 +0800 | [diff] [blame] | 58 | } |
| 59 | } |
| 60 | }; |
Brian Carlstrom | e24fa61 | 2011-09-29 00:53:55 -0700 | [diff] [blame] | 61 | |
| 62 | TEST_F(OatTest, WriteRead) { |
| 63 | const bool compile = false; // DISABLED_ due to the time to compile libcore |
Jesse Wilson | 254db0f | 2011-11-16 16:44:11 -0500 | [diff] [blame] | 64 | ClassLinker* class_linker = Runtime::Current()->GetClassLinker(); |
Brian Carlstrom | e24fa61 | 2011-09-29 00:53:55 -0700 | [diff] [blame] | 65 | |
Ian Rogers | 00f7d0e | 2012-07-19 15:28:27 -0700 | [diff] [blame] | 66 | jobject class_loader = NULL; |
Brian Carlstrom | e24fa61 | 2011-09-29 00:53:55 -0700 | [diff] [blame] | 67 | if (compile) { |
buzbee | c531cef | 2012-10-18 07:09:20 -0700 | [diff] [blame] | 68 | // TODO: make selectable |
| 69 | #if defined(ART_USE_PORTABLE_COMPILER) |
| 70 | CompilerBackend compiler_backend = kPortable; |
| 71 | #elif defined(ART_USE_LLVM_COMPILER) |
| 72 | CompilerBackend compiler_backend = kIceland; // TODO: remove |
| 73 | #else |
| 74 | CompilerBackend compiler_backend = kQuick; |
| 75 | #endif |
| 76 | compiler_.reset(new Compiler(compiler_backend, kThumb2, false, 2, false, NULL, true, true)); |
Ian Rogers | 00f7d0e | 2012-07-19 15:28:27 -0700 | [diff] [blame] | 77 | compiler_->CompileAll(class_loader, class_linker->GetBootClassPath()); |
Brian Carlstrom | e24fa61 | 2011-09-29 00:53:55 -0700 | [diff] [blame] | 78 | } |
| 79 | |
Ian Rogers | 00f7d0e | 2012-07-19 15:28:27 -0700 | [diff] [blame] | 80 | ScopedObjectAccess soa(Thread::Current()); |
Brian Carlstrom | e24fa61 | 2011-09-29 00:53:55 -0700 | [diff] [blame] | 81 | ScratchFile tmp; |
Brian Carlstrom | 81f3ca1 | 2012-03-17 00:27:35 -0700 | [diff] [blame] | 82 | bool success = OatWriter::Create(tmp.GetFile(), |
Ian Rogers | 00f7d0e | 2012-07-19 15:28:27 -0700 | [diff] [blame] | 83 | class_loader, |
Brian Carlstrom | 81f3ca1 | 2012-03-17 00:27:35 -0700 | [diff] [blame] | 84 | class_linker->GetBootClassPath(), |
| 85 | 42U, |
Brian Carlstrom | 28db012 | 2012-10-18 16:20:41 -0700 | [diff] [blame] | 86 | 4096U, |
Brian Carlstrom | 81f3ca1 | 2012-03-17 00:27:35 -0700 | [diff] [blame] | 87 | "lue.art", |
| 88 | *compiler_.get()); |
Brian Carlstrom | e24fa61 | 2011-09-29 00:53:55 -0700 | [diff] [blame] | 89 | ASSERT_TRUE(success); |
| 90 | |
| 91 | if (compile) { // OatWriter strips the code, regenerate to compare |
Ian Rogers | 00f7d0e | 2012-07-19 15:28:27 -0700 | [diff] [blame] | 92 | compiler_->CompileAll(class_loader, class_linker->GetBootClassPath()); |
Brian Carlstrom | e24fa61 | 2011-09-29 00:53:55 -0700 | [diff] [blame] | 93 | } |
Logan Chien | 0c717dd | 2012-03-28 18:31:07 +0800 | [diff] [blame] | 94 | UniquePtr<OatFile> oat_file(OatFile::Open(tmp.GetFilename(), |
| 95 | tmp.GetFilename(), |
| 96 | NULL, |
| 97 | OatFile::kRelocNone)); |
Brian Carlstrom | e24fa61 | 2011-09-29 00:53:55 -0700 | [diff] [blame] | 98 | ASSERT_TRUE(oat_file.get() != NULL); |
| 99 | const OatHeader& oat_header = oat_file->GetOatHeader(); |
Brian Carlstrom | f852fb2 | 2012-10-19 11:01:58 -0700 | [diff] [blame] | 100 | ASSERT_TRUE(oat_header.IsValid()); |
Brian Carlstrom | e24fa61 | 2011-09-29 00:53:55 -0700 | [diff] [blame] | 101 | ASSERT_EQ(1U, oat_header.GetDexFileCount()); |
Brian Carlstrom | 28db012 | 2012-10-18 16:20:41 -0700 | [diff] [blame] | 102 | ASSERT_EQ(42U, oat_header.GetImageFileLocationOatChecksum()); |
| 103 | ASSERT_EQ(4096U, oat_header.GetImageFileLocationOatBegin()); |
Brian Carlstrom | 81f3ca1 | 2012-03-17 00:27:35 -0700 | [diff] [blame] | 104 | ASSERT_EQ("lue.art", oat_header.GetImageFileLocation()); |
Brian Carlstrom | e24fa61 | 2011-09-29 00:53:55 -0700 | [diff] [blame] | 105 | |
Brian Carlstrom | a004aa9 | 2012-02-08 18:05:09 -0800 | [diff] [blame] | 106 | const DexFile* dex_file = java_lang_dex_file_; |
| 107 | const OatFile::OatDexFile* oat_dex_file = oat_file->GetOatDexFile(dex_file->GetLocation()); |
| 108 | CHECK_EQ(dex_file->GetLocationChecksum(), oat_dex_file->GetDexFileLocationChecksum()); |
| 109 | for (size_t i = 0; i < dex_file->NumClassDefs(); i++) { |
| 110 | const DexFile::ClassDef& class_def = dex_file->GetClassDef(i); |
| 111 | const byte* class_data = dex_file->GetClassData(class_def); |
Ian Rogers | 0571d35 | 2011-11-03 19:51:38 -0700 | [diff] [blame] | 112 | size_t num_virtual_methods =0; |
| 113 | if (class_data != NULL) { |
Brian Carlstrom | a004aa9 | 2012-02-08 18:05:09 -0800 | [diff] [blame] | 114 | ClassDataItemIterator it(*dex_file, class_data); |
Ian Rogers | 0571d35 | 2011-11-03 19:51:38 -0700 | [diff] [blame] | 115 | num_virtual_methods = it.NumVirtualMethods(); |
| 116 | } |
Brian Carlstrom | a004aa9 | 2012-02-08 18:05:09 -0800 | [diff] [blame] | 117 | const char* descriptor = dex_file->GetClassDescriptor(class_def); |
Brian Carlstrom | e24fa61 | 2011-09-29 00:53:55 -0700 | [diff] [blame] | 118 | |
Brian Carlstrom | aded5f7 | 2011-10-07 17:15:04 -0700 | [diff] [blame] | 119 | UniquePtr<const OatFile::OatClass> oat_class(oat_dex_file->GetOatClass(i)); |
Brian Carlstrom | e24fa61 | 2011-09-29 00:53:55 -0700 | [diff] [blame] | 120 | |
Ian Rogers | 00f7d0e | 2012-07-19 15:28:27 -0700 | [diff] [blame] | 121 | Class* klass = class_linker->FindClass(descriptor, NULL); |
Brian Carlstrom | e24fa61 | 2011-09-29 00:53:55 -0700 | [diff] [blame] | 122 | |
| 123 | size_t method_index = 0; |
| 124 | for (size_t i = 0; i < klass->NumDirectMethods(); i++, method_index++) { |
Logan Chien | eeb7edf | 2012-03-09 20:38:39 +0800 | [diff] [blame] | 125 | CheckMethod(klass->GetDirectMethod(i), |
| 126 | oat_class->GetOatMethod(method_index), dex_file); |
Brian Carlstrom | e24fa61 | 2011-09-29 00:53:55 -0700 | [diff] [blame] | 127 | } |
| 128 | for (size_t i = 0; i < num_virtual_methods; i++, method_index++) { |
Logan Chien | eeb7edf | 2012-03-09 20:38:39 +0800 | [diff] [blame] | 129 | CheckMethod(klass->GetVirtualMethod(i), |
| 130 | oat_class->GetOatMethod(method_index), dex_file); |
Brian Carlstrom | e24fa61 | 2011-09-29 00:53:55 -0700 | [diff] [blame] | 131 | } |
| 132 | } |
| 133 | } |
| 134 | |
Brian Carlstrom | 341df94 | 2012-06-27 12:29:22 -0700 | [diff] [blame] | 135 | TEST_F(OatTest, OatHeaderSizeCheck) { |
| 136 | // If this test is failing and you have to update these constants, |
| 137 | // it is time to update OatHeader::kOatVersion |
Brian Carlstrom | 28db012 | 2012-10-18 16:20:41 -0700 | [diff] [blame] | 138 | EXPECT_EQ(36U, sizeof(OatHeader)); |
TDYa127 | 5833853 | 2012-07-11 23:16:29 -0700 | [diff] [blame] | 139 | #if !defined(ART_USE_LLVM_COMPILER) |
Brian Carlstrom | 341df94 | 2012-06-27 12:29:22 -0700 | [diff] [blame] | 140 | EXPECT_EQ(32U, sizeof(OatMethodOffsets)); |
TDYa127 | 5833853 | 2012-07-11 23:16:29 -0700 | [diff] [blame] | 141 | #else |
| 142 | // ART-LLVM has a extra 4 bytes field: proxy_stub_offset_ |
| 143 | EXPECT_EQ(36U, sizeof(OatMethodOffsets)); |
| 144 | #endif |
Brian Carlstrom | 341df94 | 2012-06-27 12:29:22 -0700 | [diff] [blame] | 145 | } |
| 146 | |
Brian Carlstrom | f852fb2 | 2012-10-19 11:01:58 -0700 | [diff] [blame] | 147 | TEST_F(OatTest, OatHeaderIsValid) { |
| 148 | InstructionSet instruction_set = kX86; |
| 149 | std::vector<const DexFile*> dex_files; |
| 150 | uint32_t image_file_location_oat_checksum = 0; |
| 151 | uint32_t image_file_location_oat_begin = 0; |
| 152 | const std::string image_file_location; |
| 153 | OatHeader oat_header(instruction_set, |
| 154 | &dex_files, |
| 155 | image_file_location_oat_checksum, |
| 156 | image_file_location_oat_begin, |
| 157 | image_file_location); |
| 158 | ASSERT_TRUE(oat_header.IsValid()); |
| 159 | |
| 160 | char* magic = const_cast<char*>(oat_header.GetMagic()); |
| 161 | strcpy(magic, ""); // bad magic |
| 162 | ASSERT_FALSE(oat_header.IsValid()); |
| 163 | strcpy(magic, "oat\n000"); // bad version |
| 164 | ASSERT_FALSE(oat_header.IsValid()); |
| 165 | } |
| 166 | |
Brian Carlstrom | e24fa61 | 2011-09-29 00:53:55 -0700 | [diff] [blame] | 167 | } // namespace art |