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