Brian Carlstrom | 9ea1cb1 | 2011-08-24 23:18:18 -0700 | [diff] [blame] | 1 | // Copyright 2011 Google Inc. All Rights Reserved. |
| 2 | |
| 3 | #include "compiler.h" |
| 4 | |
Brian Carlstrom | 2cc022b | 2011-08-25 10:05:39 -0700 | [diff] [blame] | 5 | #include "assembler.h" |
Brian Carlstrom | 9ea1cb1 | 2011-08-24 23:18:18 -0700 | [diff] [blame] | 6 | #include "class_linker.h" |
| 7 | #include "dex_cache.h" |
Brian Carlstrom | 2cc022b | 2011-08-25 10:05:39 -0700 | [diff] [blame] | 8 | #include "jni_compiler.h" |
Brian Carlstrom | 9ea1cb1 | 2011-08-24 23:18:18 -0700 | [diff] [blame] | 9 | |
| 10 | extern bool oatCompileMethod(art::Method*, art::InstructionSet); |
| 11 | |
| 12 | namespace art { |
| 13 | |
Brian Carlstrom | 2cc022b | 2011-08-25 10:05:39 -0700 | [diff] [blame] | 14 | // TODO need to specify target |
Brian Carlstrom | 8a48741 | 2011-08-29 20:08:52 -0700 | [diff] [blame] | 15 | void Compiler::CompileAll(const ClassLoader* class_loader) { |
Brian Carlstrom | 9ea1cb1 | 2011-08-24 23:18:18 -0700 | [diff] [blame] | 16 | Resolve(class_loader); |
Brian Carlstrom | 83db772 | 2011-08-26 17:32:56 -0700 | [diff] [blame] | 17 | // TODO add verification step |
| 18 | Compile(class_loader); |
Brian Carlstrom | 9cc262e | 2011-08-28 12:45:30 -0700 | [diff] [blame] | 19 | SetCodeAndDirectMethods(class_loader); |
Brian Carlstrom | 8a48741 | 2011-08-29 20:08:52 -0700 | [diff] [blame] | 20 | } |
| 21 | |
| 22 | void Compiler::CompileOne(Method* method) { |
| 23 | const ClassLoader* class_loader = method->GetDeclaringClass()->GetClassLoader(); |
| 24 | Resolve(class_loader); |
| 25 | // TODO add verification step |
| 26 | CompileMethod(method); |
| 27 | SetCodeAndDirectMethods(class_loader); |
Brian Carlstrom | 9ea1cb1 | 2011-08-24 23:18:18 -0700 | [diff] [blame] | 28 | } |
| 29 | |
| 30 | void Compiler::Resolve(const ClassLoader* class_loader) { |
Brian Carlstrom | 8a48741 | 2011-08-29 20:08:52 -0700 | [diff] [blame] | 31 | const std::vector<const DexFile*>& class_path = ClassLoader::GetClassPath(class_loader); |
Brian Carlstrom | 9ea1cb1 | 2011-08-24 23:18:18 -0700 | [diff] [blame] | 32 | for (size_t i = 0; i != class_path.size(); ++i) { |
| 33 | const DexFile* dex_file = class_path[i]; |
| 34 | CHECK(dex_file != NULL); |
| 35 | ResolveDexFile(class_loader, *dex_file); |
| 36 | } |
| 37 | } |
| 38 | |
| 39 | void Compiler::ResolveDexFile(const ClassLoader* class_loader, const DexFile& dex_file) { |
| 40 | ClassLinker* class_linker = Runtime::Current()->GetClassLinker(); |
| 41 | |
| 42 | // Strings are easy, they always are simply resolved to literals in the same file |
| 43 | DexCache* dex_cache = class_linker->FindDexCache(dex_file); |
| 44 | for (size_t i = 0; i < dex_cache->NumStrings(); i++) { |
| 45 | class_linker->ResolveString(dex_file, i, dex_cache); |
| 46 | } |
| 47 | |
| 48 | // Class derived values are more complicated, they require the linker and loader |
Brian Carlstrom | 1caa2c2 | 2011-08-28 13:02:33 -0700 | [diff] [blame] | 49 | for (size_t i = 0; i < dex_cache->NumResolvedTypes(); i++) { |
Brian Carlstrom | 9ea1cb1 | 2011-08-24 23:18:18 -0700 | [diff] [blame] | 50 | class_linker->ResolveType(dex_file, i, dex_cache, class_loader); |
| 51 | } |
Brian Carlstrom | 1caa2c2 | 2011-08-28 13:02:33 -0700 | [diff] [blame] | 52 | for (size_t i = 0; i < dex_cache->NumResolvedMethods(); i++) { |
Brian Carlstrom | 20cfffa | 2011-08-26 02:31:27 -0700 | [diff] [blame] | 53 | // unknown if direct or virtual, try both |
| 54 | Method* method = class_linker->ResolveMethod(dex_file, i, dex_cache, class_loader, false); |
| 55 | if (method == NULL) { |
| 56 | class_linker->ResolveMethod(dex_file, i, dex_cache, class_loader, true); |
| 57 | } |
Brian Carlstrom | 9ea1cb1 | 2011-08-24 23:18:18 -0700 | [diff] [blame] | 58 | } |
Brian Carlstrom | 1caa2c2 | 2011-08-28 13:02:33 -0700 | [diff] [blame] | 59 | for (size_t i = 0; i < dex_cache->NumResolvedFields(); i++) { |
Brian Carlstrom | 20cfffa | 2011-08-26 02:31:27 -0700 | [diff] [blame] | 60 | // unknown if instance or static, try both |
| 61 | Field* field = class_linker->ResolveField(dex_file, i, dex_cache, class_loader, false); |
| 62 | if (field == NULL) { |
Elliott Hughes | 1f1fa56 | 2011-08-26 10:22:53 -0700 | [diff] [blame] | 63 | class_linker->ResolveField(dex_file, i, dex_cache, class_loader, true); |
Brian Carlstrom | 20cfffa | 2011-08-26 02:31:27 -0700 | [diff] [blame] | 64 | } |
Brian Carlstrom | 9ea1cb1 | 2011-08-24 23:18:18 -0700 | [diff] [blame] | 65 | } |
| 66 | } |
| 67 | |
Brian Carlstrom | 83db772 | 2011-08-26 17:32:56 -0700 | [diff] [blame] | 68 | void Compiler::Compile(const ClassLoader* class_loader) { |
Brian Carlstrom | 8a48741 | 2011-08-29 20:08:52 -0700 | [diff] [blame] | 69 | const std::vector<const DexFile*>& class_path = ClassLoader::GetClassPath(class_loader); |
Brian Carlstrom | 83db772 | 2011-08-26 17:32:56 -0700 | [diff] [blame] | 70 | for (size_t i = 0; i != class_path.size(); ++i) { |
| 71 | const DexFile* dex_file = class_path[i]; |
| 72 | CHECK(dex_file != NULL); |
| 73 | CompileDexFile(class_loader, *dex_file); |
| 74 | } |
| 75 | } |
| 76 | |
Brian Carlstrom | 9ea1cb1 | 2011-08-24 23:18:18 -0700 | [diff] [blame] | 77 | void Compiler::CompileDexFile(const ClassLoader* class_loader, const DexFile& dex_file) { |
| 78 | ClassLinker* class_linker = Runtime::Current()->GetClassLinker(); |
| 79 | for (size_t i = 0; i < dex_file.NumClassDefs(); i++) { |
| 80 | const DexFile::ClassDef& class_def = dex_file.GetClassDef(i); |
| 81 | const char* descriptor = dex_file.GetClassDescriptor(class_def); |
| 82 | Class* klass = class_linker->FindClass(descriptor, class_loader); |
| 83 | CHECK(klass != NULL); |
| 84 | CompileClass(klass); |
| 85 | } |
| 86 | } |
| 87 | |
| 88 | void Compiler::CompileClass(Class* klass) { |
| 89 | for (size_t i = 0; i < klass->NumDirectMethods(); i++) { |
| 90 | CompileMethod(klass->GetDirectMethod(i)); |
| 91 | } |
| 92 | for (size_t i = 0; i < klass->NumVirtualMethods(); i++) { |
| 93 | CompileMethod(klass->GetVirtualMethod(i)); |
| 94 | } |
| 95 | } |
| 96 | |
| 97 | void Compiler::CompileMethod(Method* method) { |
Brian Carlstrom | 2cc022b | 2011-08-25 10:05:39 -0700 | [diff] [blame] | 98 | if (method->IsNative()) { |
Brian Carlstrom | bffb155 | 2011-08-25 12:23:53 -0700 | [diff] [blame] | 99 | // TODO note code will be unmapped when JniCompiler goes out of scope |
Brian Carlstrom | 2cc022b | 2011-08-25 10:05:39 -0700 | [diff] [blame] | 100 | Assembler jni_asm; |
| 101 | JniCompiler jni_compiler; |
| 102 | jni_compiler.Compile(&jni_asm, method); |
| 103 | } else if (method->IsAbstract()) { |
Brian Carlstrom | bffb155 | 2011-08-25 12:23:53 -0700 | [diff] [blame] | 104 | // TODO: This might be also noted in the ClassLinker. |
| 105 | // Probably makes more sense to do here? |
| 106 | UNIMPLEMENTED(FATAL) << "compile stub to throw AbstractMethodError"; |
Brian Carlstrom | 2cc022b | 2011-08-25 10:05:39 -0700 | [diff] [blame] | 107 | } else { |
| 108 | oatCompileMethod(method, kThumb2); |
| 109 | } |
Brian Carlstrom | bffb155 | 2011-08-25 12:23:53 -0700 | [diff] [blame] | 110 | // CHECK(method->HasCode()); // TODO: enable this check ASAP |
Brian Carlstrom | 9ea1cb1 | 2011-08-24 23:18:18 -0700 | [diff] [blame] | 111 | } |
| 112 | |
Brian Carlstrom | 9cc262e | 2011-08-28 12:45:30 -0700 | [diff] [blame] | 113 | void Compiler::SetCodeAndDirectMethods(const ClassLoader* class_loader) { |
Brian Carlstrom | 8a48741 | 2011-08-29 20:08:52 -0700 | [diff] [blame] | 114 | const std::vector<const DexFile*>& class_path = ClassLoader::GetClassPath(class_loader); |
Brian Carlstrom | 83db772 | 2011-08-26 17:32:56 -0700 | [diff] [blame] | 115 | for (size_t i = 0; i != class_path.size(); ++i) { |
| 116 | const DexFile* dex_file = class_path[i]; |
| 117 | CHECK(dex_file != NULL); |
Brian Carlstrom | 8a48741 | 2011-08-29 20:08:52 -0700 | [diff] [blame] | 118 | SetCodeAndDirectMethodsDexFile(*dex_file); |
Brian Carlstrom | 83db772 | 2011-08-26 17:32:56 -0700 | [diff] [blame] | 119 | } |
| 120 | } |
| 121 | |
Brian Carlstrom | 8a48741 | 2011-08-29 20:08:52 -0700 | [diff] [blame] | 122 | void Compiler::SetCodeAndDirectMethodsDexFile(const DexFile& dex_file) { |
Brian Carlstrom | 83db772 | 2011-08-26 17:32:56 -0700 | [diff] [blame] | 123 | ClassLinker* class_linker = Runtime::Current()->GetClassLinker(); |
| 124 | DexCache* dex_cache = class_linker->FindDexCache(dex_file); |
Brian Carlstrom | 9cc262e | 2011-08-28 12:45:30 -0700 | [diff] [blame] | 125 | CodeAndDirectMethods* code_and_direct_methods = dex_cache->GetCodeAndDirectMethods(); |
Brian Carlstrom | 1caa2c2 | 2011-08-28 13:02:33 -0700 | [diff] [blame] | 126 | for (size_t i = 0; i < dex_cache->NumResolvedMethods(); i++) { |
Brian Carlstrom | 83db772 | 2011-08-26 17:32:56 -0700 | [diff] [blame] | 127 | Method* method = dex_cache->GetResolvedMethod(i); |
Brian Carlstrom | 9cc262e | 2011-08-28 12:45:30 -0700 | [diff] [blame] | 128 | if (method == NULL) { |
| 129 | code_and_direct_methods->SetResolvedDirectMethodTrampoline(i); |
| 130 | } else if (method->IsDirect()) { |
| 131 | code_and_direct_methods->SetResolvedDirectMethod(i, method); |
| 132 | } else { |
| 133 | // TODO: we currently leave the entry blank for resolved |
| 134 | // non-direct methods. we could put in an error stub. |
Brian Carlstrom | 83db772 | 2011-08-26 17:32:56 -0700 | [diff] [blame] | 135 | } |
| 136 | } |
| 137 | } |
| 138 | |
Brian Carlstrom | 9ea1cb1 | 2011-08-24 23:18:18 -0700 | [diff] [blame] | 139 | } // namespace art |