Brian Carlstrom | e24fa61 | 2011-09-29 00:53:55 -0700 | [diff] [blame] | 1 | // Copyright 2011 Google Inc. All Rights Reserved. |
| 2 | |
| 3 | #include "oat_file.h" |
| 4 | #include "oat_writer.h" |
| 5 | |
| 6 | #include "common_test.h" |
| 7 | |
| 8 | namespace art { |
| 9 | |
| 10 | class OatTest : public CommonTest {}; |
| 11 | |
| 12 | TEST_F(OatTest, WriteRead) { |
| 13 | const bool compile = false; // DISABLED_ due to the time to compile libcore |
Jesse Wilson | 254db0f | 2011-11-16 16:44:11 -0500 | [diff] [blame^] | 14 | ClassLinker* class_linker = Runtime::Current()->GetClassLinker(); |
Brian Carlstrom | e24fa61 | 2011-09-29 00:53:55 -0700 | [diff] [blame] | 15 | |
Brian Carlstrom | 40381fb | 2011-10-19 14:13:40 -0700 | [diff] [blame] | 16 | SirtRef<ClassLoader> class_loader(NULL); |
Brian Carlstrom | e24fa61 | 2011-09-29 00:53:55 -0700 | [diff] [blame] | 17 | if (compile) { |
Brian Carlstrom | aded5f7 | 2011-10-07 17:15:04 -0700 | [diff] [blame] | 18 | compiler_.reset(new Compiler(kThumb2, false)); |
Jesse Wilson | 254db0f | 2011-11-16 16:44:11 -0500 | [diff] [blame^] | 19 | compiler_->CompileAll(class_loader.get(), class_linker->GetBootClassPath()); |
Brian Carlstrom | e24fa61 | 2011-09-29 00:53:55 -0700 | [diff] [blame] | 20 | } |
| 21 | |
| 22 | ScratchFile tmp; |
Elliott Hughes | 234da57 | 2011-11-03 22:13:06 -0700 | [diff] [blame] | 23 | bool success = OatWriter::Create(tmp.GetFile(), class_loader.get(), *compiler_.get()); |
Brian Carlstrom | e24fa61 | 2011-09-29 00:53:55 -0700 | [diff] [blame] | 24 | ASSERT_TRUE(success); |
| 25 | |
| 26 | if (compile) { // OatWriter strips the code, regenerate to compare |
Jesse Wilson | 254db0f | 2011-11-16 16:44:11 -0500 | [diff] [blame^] | 27 | compiler_->CompileAll(class_loader.get(), class_linker->GetBootClassPath()); |
Brian Carlstrom | e24fa61 | 2011-09-29 00:53:55 -0700 | [diff] [blame] | 28 | } |
| 29 | UniquePtr<OatFile> oat_file(OatFile::Open(std::string(tmp.GetFilename()), "", NULL)); |
| 30 | ASSERT_TRUE(oat_file.get() != NULL); |
| 31 | const OatHeader& oat_header = oat_file->GetOatHeader(); |
| 32 | ASSERT_EQ(1U, oat_header.GetDexFileCount()); |
| 33 | |
Brian Carlstrom | e24fa61 | 2011-09-29 00:53:55 -0700 | [diff] [blame] | 34 | const DexFile& dex_file = *java_lang_dex_file_.get(); |
Brian Carlstrom | aded5f7 | 2011-10-07 17:15:04 -0700 | [diff] [blame] | 35 | const OatFile::OatDexFile* oat_dex_file = oat_file->GetOatDexFile(dex_file.GetLocation()); |
Brian Carlstrom | e24fa61 | 2011-09-29 00:53:55 -0700 | [diff] [blame] | 36 | for (size_t i = 0; i < dex_file.NumClassDefs(); i++) { |
| 37 | const DexFile::ClassDef& class_def = dex_file.GetClassDef(i); |
| 38 | const byte* class_data = dex_file.GetClassData(class_def); |
Ian Rogers | 0571d35 | 2011-11-03 19:51:38 -0700 | [diff] [blame] | 39 | size_t num_virtual_methods =0; |
| 40 | if (class_data != NULL) { |
| 41 | ClassDataItemIterator it(dex_file, class_data); |
| 42 | num_virtual_methods = it.NumVirtualMethods(); |
| 43 | } |
Brian Carlstrom | e24fa61 | 2011-09-29 00:53:55 -0700 | [diff] [blame] | 44 | const char* descriptor = dex_file.GetClassDescriptor(class_def); |
| 45 | |
Brian Carlstrom | aded5f7 | 2011-10-07 17:15:04 -0700 | [diff] [blame] | 46 | UniquePtr<const OatFile::OatClass> oat_class(oat_dex_file->GetOatClass(i)); |
Brian Carlstrom | e24fa61 | 2011-09-29 00:53:55 -0700 | [diff] [blame] | 47 | |
Brian Carlstrom | 40381fb | 2011-10-19 14:13:40 -0700 | [diff] [blame] | 48 | Class* klass = class_linker->FindClass(descriptor, class_loader.get()); |
Brian Carlstrom | e24fa61 | 2011-09-29 00:53:55 -0700 | [diff] [blame] | 49 | |
| 50 | size_t method_index = 0; |
| 51 | for (size_t i = 0; i < klass->NumDirectMethods(); i++, method_index++) { |
| 52 | Method* method = klass->GetDirectMethod(i); |
Brian Carlstrom | aded5f7 | 2011-10-07 17:15:04 -0700 | [diff] [blame] | 53 | const OatFile::OatMethod oat_method = oat_class->GetOatMethod(method_index); |
Ian Rogers | 0571d35 | 2011-11-03 19:51:38 -0700 | [diff] [blame] | 54 | const CompiledMethod* compiled_method = |
| 55 | compiler_->GetCompiledMethod(Compiler::MethodReference(&dex_file, |
| 56 | method->GetDexMethodIndex())); |
Brian Carlstrom | 3320cf4 | 2011-10-04 14:58:28 -0700 | [diff] [blame] | 57 | |
| 58 | if (compiled_method == NULL) { |
| 59 | EXPECT_TRUE(oat_method.code_ == NULL) << PrettyMethod(method) << " " << oat_method.code_; |
| 60 | EXPECT_EQ(oat_method.frame_size_in_bytes_, static_cast<uint32_t>(kStackAlignment)); |
Brian Carlstrom | 3320cf4 | 2011-10-04 14:58:28 -0700 | [diff] [blame] | 61 | EXPECT_EQ(oat_method.core_spill_mask_, 0U); |
| 62 | EXPECT_EQ(oat_method.fp_spill_mask_, 0U); |
Brian Carlstrom | e24fa61 | 2011-09-29 00:53:55 -0700 | [diff] [blame] | 63 | } else { |
Brian Carlstrom | 3320cf4 | 2011-10-04 14:58:28 -0700 | [diff] [blame] | 64 | const void* oat_code = oat_method.code_; |
| 65 | uintptr_t oat_code_aligned = RoundDown(reinterpret_cast<uintptr_t>(oat_code), 2); |
| 66 | oat_code = reinterpret_cast<const void*>(oat_code_aligned); |
| 67 | |
| 68 | const std::vector<uint8_t>& code = compiled_method->GetCode(); |
| 69 | size_t code_size = code.size() * sizeof(code[0]); |
| 70 | EXPECT_EQ(0, memcmp(oat_code, &code[0], code_size)) |
| 71 | << PrettyMethod(method) << " " << code_size; |
| 72 | CHECK_EQ(0, memcmp(oat_code, &code[0], code_size)); |
| 73 | EXPECT_EQ(oat_method.frame_size_in_bytes_, compiled_method->GetFrameSizeInBytes()); |
Brian Carlstrom | 3320cf4 | 2011-10-04 14:58:28 -0700 | [diff] [blame] | 74 | EXPECT_EQ(oat_method.core_spill_mask_, compiled_method->GetCoreSpillMask()); |
| 75 | EXPECT_EQ(oat_method.fp_spill_mask_, compiled_method->GetFpSpillMask()); |
Brian Carlstrom | e24fa61 | 2011-09-29 00:53:55 -0700 | [diff] [blame] | 76 | } |
| 77 | } |
| 78 | for (size_t i = 0; i < num_virtual_methods; i++, method_index++) { |
| 79 | Method* method = klass->GetVirtualMethod(i); |
Brian Carlstrom | aded5f7 | 2011-10-07 17:15:04 -0700 | [diff] [blame] | 80 | const OatFile::OatMethod oat_method = oat_class->GetOatMethod(method_index); |
Ian Rogers | 0571d35 | 2011-11-03 19:51:38 -0700 | [diff] [blame] | 81 | const CompiledMethod* compiled_method = |
| 82 | compiler_->GetCompiledMethod(Compiler::MethodReference(&dex_file, |
| 83 | method->GetDexMethodIndex())); |
Brian Carlstrom | 3320cf4 | 2011-10-04 14:58:28 -0700 | [diff] [blame] | 84 | |
| 85 | if (compiled_method == NULL) { |
| 86 | EXPECT_TRUE(oat_method.code_ == NULL) << PrettyMethod(method) << " " << oat_method.code_; |
| 87 | EXPECT_EQ(oat_method.frame_size_in_bytes_, static_cast<uint32_t>(kStackAlignment)); |
Brian Carlstrom | 3320cf4 | 2011-10-04 14:58:28 -0700 | [diff] [blame] | 88 | EXPECT_EQ(oat_method.core_spill_mask_, 0U); |
| 89 | EXPECT_EQ(oat_method.fp_spill_mask_, 0U); |
Brian Carlstrom | e24fa61 | 2011-09-29 00:53:55 -0700 | [diff] [blame] | 90 | } else { |
Brian Carlstrom | 3320cf4 | 2011-10-04 14:58:28 -0700 | [diff] [blame] | 91 | const void* oat_code = oat_method.code_; |
| 92 | EXPECT_TRUE(oat_code != NULL) << PrettyMethod(method); |
| 93 | uintptr_t oat_code_aligned = RoundDown(reinterpret_cast<uintptr_t>(oat_code), 2); |
| 94 | oat_code = reinterpret_cast<const void*>(oat_code_aligned); |
| 95 | |
| 96 | const std::vector<uint8_t>& code = compiled_method->GetCode(); |
| 97 | size_t code_size = code.size() * sizeof(code[0]); |
| 98 | EXPECT_EQ(0, memcmp(oat_code, &code[0], code_size)) |
| 99 | << PrettyMethod(method) << " " << code_size; |
| 100 | CHECK_EQ(0, memcmp(oat_code, &code[0], code_size)); |
| 101 | EXPECT_EQ(oat_method.frame_size_in_bytes_, compiled_method->GetFrameSizeInBytes()); |
Brian Carlstrom | 3320cf4 | 2011-10-04 14:58:28 -0700 | [diff] [blame] | 102 | EXPECT_EQ(oat_method.core_spill_mask_, compiled_method->GetCoreSpillMask()); |
| 103 | EXPECT_EQ(oat_method.fp_spill_mask_, compiled_method->GetFpSpillMask()); |
Brian Carlstrom | e24fa61 | 2011-09-29 00:53:55 -0700 | [diff] [blame] | 104 | } |
| 105 | } |
| 106 | } |
| 107 | } |
| 108 | |
| 109 | } // namespace art |