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: |
| 26 | void CheckMethod(Method* method, |
| 27 | const OatFile::OatMethod& oat_method, |
| 28 | const DexFile* dex_file) { |
| 29 | const CompiledMethod* compiled_method = |
| 30 | compiler_->GetCompiledMethod(Compiler::MethodReference(dex_file, |
| 31 | method->GetDexMethodIndex())); |
| 32 | |
| 33 | if (compiled_method == NULL) { |
| 34 | EXPECT_TRUE(oat_method.GetCode() == NULL) << PrettyMethod(method) << " " |
| 35 | << oat_method.GetCode(); |
TDYa127 | 42d4d73 | 2012-03-23 19:10:56 -0700 | [diff] [blame] | 36 | #if !defined(ART_USE_LLVM_COMPILER) |
Logan Chien | eeb7edf | 2012-03-09 20:38:39 +0800 | [diff] [blame] | 37 | EXPECT_EQ(oat_method.GetFrameSizeInBytes(), static_cast<uint32_t>(kStackAlignment)); |
| 38 | EXPECT_EQ(oat_method.GetCoreSpillMask(), 0U); |
| 39 | EXPECT_EQ(oat_method.GetFpSpillMask(), 0U); |
TDYa127 | 42d4d73 | 2012-03-23 19:10:56 -0700 | [diff] [blame] | 40 | #endif |
Logan Chien | eeb7edf | 2012-03-09 20:38:39 +0800 | [diff] [blame] | 41 | } else { |
| 42 | const void* oat_code = oat_method.GetCode(); |
| 43 | EXPECT_TRUE(oat_code != NULL) << PrettyMethod(method); |
| 44 | uintptr_t oat_code_aligned = RoundDown(reinterpret_cast<uintptr_t>(oat_code), 2); |
| 45 | oat_code = reinterpret_cast<const void*>(oat_code_aligned); |
| 46 | |
| 47 | const std::vector<uint8_t>& code = compiled_method->GetCode(); |
| 48 | size_t code_size = code.size() * sizeof(code[0]); |
| 49 | EXPECT_EQ(0, memcmp(oat_code, &code[0], code_size)) |
| 50 | << PrettyMethod(method) << " " << code_size; |
| 51 | CHECK_EQ(0, memcmp(oat_code, &code[0], code_size)); |
TDYa127 | 42d4d73 | 2012-03-23 19:10:56 -0700 | [diff] [blame] | 52 | #if !defined(ART_USE_LLVM_COMPILER) |
Logan Chien | eeb7edf | 2012-03-09 20:38:39 +0800 | [diff] [blame] | 53 | EXPECT_EQ(oat_method.GetFrameSizeInBytes(), compiled_method->GetFrameSizeInBytes()); |
| 54 | EXPECT_EQ(oat_method.GetCoreSpillMask(), compiled_method->GetCoreSpillMask()); |
| 55 | EXPECT_EQ(oat_method.GetFpSpillMask(), compiled_method->GetFpSpillMask()); |
TDYa127 | 42d4d73 | 2012-03-23 19:10:56 -0700 | [diff] [blame] | 56 | #endif |
Logan Chien | eeb7edf | 2012-03-09 20:38:39 +0800 | [diff] [blame] | 57 | } |
| 58 | } |
| 59 | }; |
Brian Carlstrom | e24fa61 | 2011-09-29 00:53:55 -0700 | [diff] [blame] | 60 | |
| 61 | TEST_F(OatTest, WriteRead) { |
| 62 | const bool compile = false; // DISABLED_ due to the time to compile libcore |
Jesse Wilson | 254db0f | 2011-11-16 16:44:11 -0500 | [diff] [blame] | 63 | ClassLinker* class_linker = Runtime::Current()->GetClassLinker(); |
Brian Carlstrom | e24fa61 | 2011-09-29 00:53:55 -0700 | [diff] [blame] | 64 | |
Brian Carlstrom | 40381fb | 2011-10-19 14:13:40 -0700 | [diff] [blame] | 65 | SirtRef<ClassLoader> class_loader(NULL); |
Brian Carlstrom | e24fa61 | 2011-09-29 00:53:55 -0700 | [diff] [blame] | 66 | if (compile) { |
Brian Carlstrom | ba0668e | 2012-03-26 13:14:07 -0700 | [diff] [blame] | 67 | compiler_.reset(new Compiler(kThumb2, false, 2, false, NULL, true, true)); |
Jesse Wilson | 254db0f | 2011-11-16 16:44:11 -0500 | [diff] [blame] | 68 | compiler_->CompileAll(class_loader.get(), class_linker->GetBootClassPath()); |
Brian Carlstrom | e24fa61 | 2011-09-29 00:53:55 -0700 | [diff] [blame] | 69 | } |
| 70 | |
| 71 | ScratchFile tmp; |
Brian Carlstrom | 81f3ca1 | 2012-03-17 00:27:35 -0700 | [diff] [blame] | 72 | bool success = OatWriter::Create(tmp.GetFile(), |
| 73 | class_loader.get(), |
| 74 | class_linker->GetBootClassPath(), |
| 75 | 42U, |
| 76 | "lue.art", |
| 77 | *compiler_.get()); |
Brian Carlstrom | e24fa61 | 2011-09-29 00:53:55 -0700 | [diff] [blame] | 78 | ASSERT_TRUE(success); |
| 79 | |
| 80 | if (compile) { // OatWriter strips the code, regenerate to compare |
Jesse Wilson | 254db0f | 2011-11-16 16:44:11 -0500 | [diff] [blame] | 81 | compiler_->CompileAll(class_loader.get(), class_linker->GetBootClassPath()); |
Brian Carlstrom | e24fa61 | 2011-09-29 00:53:55 -0700 | [diff] [blame] | 82 | } |
Brian Carlstrom | a004aa9 | 2012-02-08 18:05:09 -0800 | [diff] [blame] | 83 | UniquePtr<OatFile> oat_file(OatFile::Open(tmp.GetFilename(), tmp.GetFilename(), NULL)); |
Brian Carlstrom | e24fa61 | 2011-09-29 00:53:55 -0700 | [diff] [blame] | 84 | ASSERT_TRUE(oat_file.get() != NULL); |
| 85 | const OatHeader& oat_header = oat_file->GetOatHeader(); |
| 86 | ASSERT_EQ(1U, oat_header.GetDexFileCount()); |
Brian Carlstrom | 81f3ca1 | 2012-03-17 00:27:35 -0700 | [diff] [blame] | 87 | ASSERT_EQ(42U, oat_header.GetImageFileLocationChecksum()); |
| 88 | ASSERT_EQ("lue.art", oat_header.GetImageFileLocation()); |
Brian Carlstrom | e24fa61 | 2011-09-29 00:53:55 -0700 | [diff] [blame] | 89 | |
Brian Carlstrom | a004aa9 | 2012-02-08 18:05:09 -0800 | [diff] [blame] | 90 | const DexFile* dex_file = java_lang_dex_file_; |
| 91 | const OatFile::OatDexFile* oat_dex_file = oat_file->GetOatDexFile(dex_file->GetLocation()); |
| 92 | CHECK_EQ(dex_file->GetLocationChecksum(), oat_dex_file->GetDexFileLocationChecksum()); |
| 93 | for (size_t i = 0; i < dex_file->NumClassDefs(); i++) { |
| 94 | const DexFile::ClassDef& class_def = dex_file->GetClassDef(i); |
| 95 | const byte* class_data = dex_file->GetClassData(class_def); |
Ian Rogers | 0571d35 | 2011-11-03 19:51:38 -0700 | [diff] [blame] | 96 | size_t num_virtual_methods =0; |
| 97 | if (class_data != NULL) { |
Brian Carlstrom | a004aa9 | 2012-02-08 18:05:09 -0800 | [diff] [blame] | 98 | ClassDataItemIterator it(*dex_file, class_data); |
Ian Rogers | 0571d35 | 2011-11-03 19:51:38 -0700 | [diff] [blame] | 99 | num_virtual_methods = it.NumVirtualMethods(); |
| 100 | } |
Brian Carlstrom | a004aa9 | 2012-02-08 18:05:09 -0800 | [diff] [blame] | 101 | const char* descriptor = dex_file->GetClassDescriptor(class_def); |
Brian Carlstrom | e24fa61 | 2011-09-29 00:53:55 -0700 | [diff] [blame] | 102 | |
Brian Carlstrom | aded5f7 | 2011-10-07 17:15:04 -0700 | [diff] [blame] | 103 | UniquePtr<const OatFile::OatClass> oat_class(oat_dex_file->GetOatClass(i)); |
Brian Carlstrom | e24fa61 | 2011-09-29 00:53:55 -0700 | [diff] [blame] | 104 | |
Brian Carlstrom | 40381fb | 2011-10-19 14:13:40 -0700 | [diff] [blame] | 105 | Class* klass = class_linker->FindClass(descriptor, class_loader.get()); |
Brian Carlstrom | e24fa61 | 2011-09-29 00:53:55 -0700 | [diff] [blame] | 106 | |
| 107 | size_t method_index = 0; |
| 108 | for (size_t i = 0; i < klass->NumDirectMethods(); i++, method_index++) { |
Logan Chien | eeb7edf | 2012-03-09 20:38:39 +0800 | [diff] [blame] | 109 | CheckMethod(klass->GetDirectMethod(i), |
| 110 | oat_class->GetOatMethod(method_index), dex_file); |
Brian Carlstrom | e24fa61 | 2011-09-29 00:53:55 -0700 | [diff] [blame] | 111 | } |
| 112 | for (size_t i = 0; i < num_virtual_methods; i++, method_index++) { |
Logan Chien | eeb7edf | 2012-03-09 20:38:39 +0800 | [diff] [blame] | 113 | CheckMethod(klass->GetVirtualMethod(i), |
| 114 | oat_class->GetOatMethod(method_index), dex_file); |
Brian Carlstrom | e24fa61 | 2011-09-29 00:53:55 -0700 | [diff] [blame] | 115 | } |
| 116 | } |
| 117 | } |
| 118 | |
| 119 | } // namespace art |