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