Ian Rogers | b033c75 | 2011-07-20 12:22:35 -0700 | [diff] [blame] | 1 | // Copyright 2011 Google Inc. All Rights Reserved. |
Ian Rogers | b033c75 | 2011-07-20 12:22:35 -0700 | [diff] [blame] | 2 | |
Elliott Hughes | 90a3369 | 2011-08-30 13:27:07 -0700 | [diff] [blame] | 3 | #include "jni_compiler.h" |
| 4 | |
Ian Rogers | b033c75 | 2011-07-20 12:22:35 -0700 | [diff] [blame] | 5 | #include <sys/mman.h> |
Brian Carlstrom | 578bbdc | 2011-07-21 14:07:47 -0700 | [diff] [blame] | 6 | |
Elliott Hughes | 90a3369 | 2011-08-30 13:27:07 -0700 | [diff] [blame] | 7 | #include "UniquePtr.h" |
Brian Carlstrom | 578bbdc | 2011-07-21 14:07:47 -0700 | [diff] [blame] | 8 | #include "assembler.h" |
| 9 | #include "class_linker.h" |
| 10 | #include "common_test.h" |
| 11 | #include "dex_file.h" |
Elliott Hughes | 90a3369 | 2011-08-30 13:27:07 -0700 | [diff] [blame] | 12 | #include "gtest/gtest.h" |
Ian Rogers | cdd1d2d | 2011-08-18 09:58:17 -0700 | [diff] [blame] | 13 | #include "indirect_reference_table.h" |
Ian Rogers | cdd1d2d | 2011-08-18 09:58:17 -0700 | [diff] [blame] | 14 | #include "jni_internal.h" |
Brian Carlstrom | db4d540 | 2011-08-09 12:18:28 -0700 | [diff] [blame] | 15 | #include "mem_map.h" |
Brian Carlstrom | 578bbdc | 2011-07-21 14:07:47 -0700 | [diff] [blame] | 16 | #include "runtime.h" |
| 17 | #include "thread.h" |
Ian Rogers | b033c75 | 2011-07-20 12:22:35 -0700 | [diff] [blame] | 18 | |
| 19 | namespace art { |
| 20 | |
Brian Carlstrom | f734cf5 | 2011-08-17 16:28:14 -0700 | [diff] [blame] | 21 | class JniCompilerTest : public CommonTest { |
Ian Rogers | b033c75 | 2011-07-20 12:22:35 -0700 | [diff] [blame] | 22 | protected: |
| 23 | virtual void SetUp() { |
Brian Carlstrom | f734cf5 | 2011-08-17 16:28:14 -0700 | [diff] [blame] | 24 | CommonTest::SetUp(); |
Brian Carlstrom | 9f30b38 | 2011-08-28 22:41:38 -0700 | [diff] [blame] | 25 | dex_.reset(OpenTestDexFile("MyClassNatives")); |
Ian Rogers | cdd1d2d | 2011-08-18 09:58:17 -0700 | [diff] [blame] | 26 | class_loader_ = AllocPathClassLoader(dex_.get()); |
buzbee | c143c55 | 2011-08-20 17:38:58 -0700 | [diff] [blame] | 27 | Thread::Current()->SetClassLoaderOverride(class_loader_); |
Ian Rogers | b033c75 | 2011-07-20 12:22:35 -0700 | [diff] [blame] | 28 | } |
| 29 | |
Ian Rogers | cdd1d2d | 2011-08-18 09:58:17 -0700 | [diff] [blame] | 30 | void SetupForTest(bool direct, const char* method_name, |
| 31 | const char* method_sig, void* native_fnptr) { |
buzbee | c143c55 | 2011-08-20 17:38:58 -0700 | [diff] [blame] | 32 | env_ = Thread::Current()->GetJniEnv(); |
Ian Rogers | cdd1d2d | 2011-08-18 09:58:17 -0700 | [diff] [blame] | 33 | |
buzbee | c143c55 | 2011-08-20 17:38:58 -0700 | [diff] [blame] | 34 | jklass_ = env_->FindClass("MyClass"); |
| 35 | ASSERT_TRUE(jklass_ != NULL); |
| 36 | |
| 37 | Class* c = class_linker_->FindClass("LMyClass;", class_loader_); |
Ian Rogers | cdd1d2d | 2011-08-18 09:58:17 -0700 | [diff] [blame] | 38 | Method* method; |
| 39 | if (direct) { |
buzbee | c143c55 | 2011-08-20 17:38:58 -0700 | [diff] [blame] | 40 | method = c->FindDirectMethod(method_name, method_sig); |
Ian Rogers | cdd1d2d | 2011-08-18 09:58:17 -0700 | [diff] [blame] | 41 | } else { |
buzbee | c143c55 | 2011-08-20 17:38:58 -0700 | [diff] [blame] | 42 | method = c->FindVirtualMethod(method_name, method_sig); |
Ian Rogers | cdd1d2d | 2011-08-18 09:58:17 -0700 | [diff] [blame] | 43 | } |
| 44 | ASSERT_TRUE(method != NULL); |
| 45 | |
| 46 | // Compile the native method |
| 47 | jni_compiler.Compile(&jni_asm, method); |
buzbee | c143c55 | 2011-08-20 17:38:58 -0700 | [diff] [blame] | 48 | ASSERT_TRUE(method->HasCode()); |
Ian Rogers | cdd1d2d | 2011-08-18 09:58:17 -0700 | [diff] [blame] | 49 | |
Ian Rogers | cdd1d2d | 2011-08-18 09:58:17 -0700 | [diff] [blame] | 50 | if (direct) { |
| 51 | jmethod_ = env_->GetStaticMethodID(jklass_, method_name, method_sig); |
| 52 | } else { |
| 53 | jmethod_ = env_->GetMethodID(jklass_, method_name, method_sig); |
| 54 | } |
| 55 | ASSERT_TRUE(jmethod_ != NULL); |
| 56 | |
| 57 | JNINativeMethod methods[] = {{method_name, method_sig, native_fnptr}}; |
| 58 | ASSERT_EQ(JNI_OK, env_->RegisterNatives(jklass_, methods, 1)); |
| 59 | |
| 60 | jmethodID constructor = env_->GetMethodID(jklass_, "<init>", "()V"); |
| 61 | jobj_ = env_->NewObject(jklass_, constructor); |
| 62 | ASSERT_TRUE(jobj_ != NULL); |
Ian Rogers | b033c75 | 2011-07-20 12:22:35 -0700 | [diff] [blame] | 63 | } |
| 64 | |
Ian Rogers | cdd1d2d | 2011-08-18 09:58:17 -0700 | [diff] [blame] | 65 | public: |
| 66 | static jclass jklass_; |
| 67 | static jobject jobj_; |
| 68 | protected: |
Elliott Hughes | 90a3369 | 2011-08-30 13:27:07 -0700 | [diff] [blame] | 69 | UniquePtr<const DexFile> dex_; |
Brian Carlstrom | 8a48741 | 2011-08-29 20:08:52 -0700 | [diff] [blame] | 70 | const PathClassLoader* class_loader_; |
Ian Rogers | cdd1d2d | 2011-08-18 09:58:17 -0700 | [diff] [blame] | 71 | Assembler jni_asm; |
| 72 | JniCompiler jni_compiler; |
| 73 | JNIEnv* env_; |
| 74 | jmethodID jmethod_; |
Ian Rogers | b033c75 | 2011-07-20 12:22:35 -0700 | [diff] [blame] | 75 | }; |
| 76 | |
Ian Rogers | cdd1d2d | 2011-08-18 09:58:17 -0700 | [diff] [blame] | 77 | jclass JniCompilerTest::jklass_; |
| 78 | jobject JniCompilerTest::jobj_; |
| 79 | |
Ian Rogers | b033c75 | 2011-07-20 12:22:35 -0700 | [diff] [blame] | 80 | int gJava_MyClass_foo_calls = 0; |
Ian Rogers | cdd1d2d | 2011-08-18 09:58:17 -0700 | [diff] [blame] | 81 | void Java_MyClass_foo(JNIEnv* env, jobject thisObj) { |
Ian Rogers | 408f79a | 2011-08-23 18:22:33 -0700 | [diff] [blame] | 82 | EXPECT_EQ(1u, Thread::Current()->NumSirtReferences()); |
Ian Rogers | b033c75 | 2011-07-20 12:22:35 -0700 | [diff] [blame] | 83 | EXPECT_EQ(Thread::kNative, Thread::Current()->GetState()); |
Ian Rogers | cdd1d2d | 2011-08-18 09:58:17 -0700 | [diff] [blame] | 84 | EXPECT_EQ(Thread::Current()->GetJniEnv(), env); |
| 85 | EXPECT_TRUE(thisObj != NULL); |
Ian Rogers | a8cd9f4 | 2011-08-19 16:43:41 -0700 | [diff] [blame] | 86 | EXPECT_TRUE(env->IsInstanceOf(thisObj, JniCompilerTest::jklass_)); |
Ian Rogers | b033c75 | 2011-07-20 12:22:35 -0700 | [diff] [blame] | 87 | gJava_MyClass_foo_calls++; |
| 88 | } |
| 89 | |
Ian Rogers | cdd1d2d | 2011-08-18 09:58:17 -0700 | [diff] [blame] | 90 | TEST_F(JniCompilerTest, CompileAndRunNoArgMethod) { |
| 91 | SetupForTest(false, "foo", "()V", reinterpret_cast<void*>(&Java_MyClass_foo)); |
| 92 | |
| 93 | EXPECT_EQ(0, gJava_MyClass_foo_calls); |
| 94 | env_->CallNonvirtualVoidMethod(jobj_, jklass_, jmethod_); |
| 95 | EXPECT_EQ(1, gJava_MyClass_foo_calls); |
| 96 | env_->CallNonvirtualVoidMethod(jobj_, jklass_, jmethod_); |
| 97 | EXPECT_EQ(2, gJava_MyClass_foo_calls); |
| 98 | } |
| 99 | |
Ian Rogers | b033c75 | 2011-07-20 12:22:35 -0700 | [diff] [blame] | 100 | int gJava_MyClass_fooI_calls = 0; |
Ian Rogers | cdd1d2d | 2011-08-18 09:58:17 -0700 | [diff] [blame] | 101 | jint Java_MyClass_fooI(JNIEnv* env, jobject thisObj, jint x) { |
Ian Rogers | 408f79a | 2011-08-23 18:22:33 -0700 | [diff] [blame] | 102 | EXPECT_EQ(1u, Thread::Current()->NumSirtReferences()); |
Ian Rogers | b033c75 | 2011-07-20 12:22:35 -0700 | [diff] [blame] | 103 | EXPECT_EQ(Thread::kNative, Thread::Current()->GetState()); |
Ian Rogers | cdd1d2d | 2011-08-18 09:58:17 -0700 | [diff] [blame] | 104 | EXPECT_EQ(Thread::Current()->GetJniEnv(), env); |
| 105 | EXPECT_TRUE(thisObj != NULL); |
Ian Rogers | a8cd9f4 | 2011-08-19 16:43:41 -0700 | [diff] [blame] | 106 | EXPECT_TRUE(env->IsInstanceOf(thisObj, JniCompilerTest::jklass_)); |
Ian Rogers | b033c75 | 2011-07-20 12:22:35 -0700 | [diff] [blame] | 107 | gJava_MyClass_fooI_calls++; |
| 108 | return x; |
| 109 | } |
| 110 | |
Ian Rogers | cdd1d2d | 2011-08-18 09:58:17 -0700 | [diff] [blame] | 111 | TEST_F(JniCompilerTest, CompileAndRunIntMethod) { |
| 112 | SetupForTest(false, "fooI", "(I)I", |
| 113 | reinterpret_cast<void*>(&Java_MyClass_fooI)); |
| 114 | |
| 115 | EXPECT_EQ(0, gJava_MyClass_fooI_calls); |
| 116 | jint result = env_->CallNonvirtualIntMethod(jobj_, jklass_, jmethod_, 42); |
| 117 | EXPECT_EQ(42, result); |
| 118 | EXPECT_EQ(1, gJava_MyClass_fooI_calls); |
| 119 | result = env_->CallNonvirtualIntMethod(jobj_, jklass_, jmethod_, 0xCAFED00D); |
| 120 | EXPECT_EQ(static_cast<jint>(0xCAFED00D), result); |
| 121 | EXPECT_EQ(2, gJava_MyClass_fooI_calls); |
| 122 | } |
| 123 | |
Ian Rogers | b033c75 | 2011-07-20 12:22:35 -0700 | [diff] [blame] | 124 | int gJava_MyClass_fooII_calls = 0; |
Ian Rogers | cdd1d2d | 2011-08-18 09:58:17 -0700 | [diff] [blame] | 125 | jint Java_MyClass_fooII(JNIEnv* env, jobject thisObj, jint x, jint y) { |
Ian Rogers | 408f79a | 2011-08-23 18:22:33 -0700 | [diff] [blame] | 126 | EXPECT_EQ(1u, Thread::Current()->NumSirtReferences()); |
Ian Rogers | b033c75 | 2011-07-20 12:22:35 -0700 | [diff] [blame] | 127 | EXPECT_EQ(Thread::kNative, Thread::Current()->GetState()); |
Ian Rogers | cdd1d2d | 2011-08-18 09:58:17 -0700 | [diff] [blame] | 128 | EXPECT_EQ(Thread::Current()->GetJniEnv(), env); |
| 129 | EXPECT_TRUE(thisObj != NULL); |
Ian Rogers | a8cd9f4 | 2011-08-19 16:43:41 -0700 | [diff] [blame] | 130 | EXPECT_TRUE(env->IsInstanceOf(thisObj, JniCompilerTest::jklass_)); |
Ian Rogers | b033c75 | 2011-07-20 12:22:35 -0700 | [diff] [blame] | 131 | gJava_MyClass_fooII_calls++; |
| 132 | return x - y; // non-commutative operator |
| 133 | } |
| 134 | |
Ian Rogers | cdd1d2d | 2011-08-18 09:58:17 -0700 | [diff] [blame] | 135 | TEST_F(JniCompilerTest, CompileAndRunIntIntMethod) { |
| 136 | SetupForTest(false, "fooII", "(II)I", |
| 137 | reinterpret_cast<void*>(&Java_MyClass_fooII)); |
| 138 | |
| 139 | EXPECT_EQ(0, gJava_MyClass_fooII_calls); |
| 140 | jint result = env_->CallNonvirtualIntMethod(jobj_, jklass_, jmethod_, 99, 10); |
| 141 | EXPECT_EQ(99 - 10, result); |
| 142 | EXPECT_EQ(1, gJava_MyClass_fooII_calls); |
| 143 | result = env_->CallNonvirtualIntMethod(jobj_, jklass_, jmethod_, 0xCAFEBABE, |
| 144 | 0xCAFED00D); |
| 145 | EXPECT_EQ(static_cast<jint>(0xCAFEBABE - 0xCAFED00D), result); |
| 146 | EXPECT_EQ(2, gJava_MyClass_fooII_calls); |
| 147 | } |
| 148 | |
Ian Rogers | b033c75 | 2011-07-20 12:22:35 -0700 | [diff] [blame] | 149 | int gJava_MyClass_fooDD_calls = 0; |
Ian Rogers | cdd1d2d | 2011-08-18 09:58:17 -0700 | [diff] [blame] | 150 | jdouble Java_MyClass_fooDD(JNIEnv* env, jobject thisObj, jdouble x, jdouble y) { |
Ian Rogers | 408f79a | 2011-08-23 18:22:33 -0700 | [diff] [blame] | 151 | EXPECT_EQ(1u, Thread::Current()->NumSirtReferences()); |
Ian Rogers | b033c75 | 2011-07-20 12:22:35 -0700 | [diff] [blame] | 152 | EXPECT_EQ(Thread::kNative, Thread::Current()->GetState()); |
Ian Rogers | cdd1d2d | 2011-08-18 09:58:17 -0700 | [diff] [blame] | 153 | EXPECT_EQ(Thread::Current()->GetJniEnv(), env); |
| 154 | EXPECT_TRUE(thisObj != NULL); |
Ian Rogers | a8cd9f4 | 2011-08-19 16:43:41 -0700 | [diff] [blame] | 155 | EXPECT_TRUE(env->IsInstanceOf(thisObj, JniCompilerTest::jklass_)); |
Ian Rogers | b033c75 | 2011-07-20 12:22:35 -0700 | [diff] [blame] | 156 | gJava_MyClass_fooDD_calls++; |
| 157 | return x - y; // non-commutative operator |
| 158 | } |
| 159 | |
Ian Rogers | cdd1d2d | 2011-08-18 09:58:17 -0700 | [diff] [blame] | 160 | TEST_F(JniCompilerTest, CompileAndRunDoubleDoubleMethod) { |
| 161 | SetupForTest(false, "fooDD", "(DD)D", |
| 162 | reinterpret_cast<void*>(&Java_MyClass_fooDD)); |
| 163 | |
| 164 | EXPECT_EQ(0, gJava_MyClass_fooDD_calls); |
| 165 | jdouble result = env_->CallNonvirtualDoubleMethod(jobj_, jklass_, jmethod_, |
| 166 | 99.0, 10.0); |
| 167 | EXPECT_EQ(99.0 - 10.0, result); |
| 168 | EXPECT_EQ(1, gJava_MyClass_fooDD_calls); |
| 169 | jdouble a = 3.14159265358979323846; |
| 170 | jdouble b = 0.69314718055994530942; |
| 171 | result = env_->CallNonvirtualDoubleMethod(jobj_, jklass_, jmethod_, a, b); |
| 172 | EXPECT_EQ(a - b, result); |
| 173 | EXPECT_EQ(2, gJava_MyClass_fooDD_calls); |
| 174 | } |
| 175 | |
Ian Rogers | b033c75 | 2011-07-20 12:22:35 -0700 | [diff] [blame] | 176 | int gJava_MyClass_fooIOO_calls = 0; |
Ian Rogers | cdd1d2d | 2011-08-18 09:58:17 -0700 | [diff] [blame] | 177 | jobject Java_MyClass_fooIOO(JNIEnv* env, jobject thisObj, jint x, jobject y, |
Ian Rogers | b033c75 | 2011-07-20 12:22:35 -0700 | [diff] [blame] | 178 | jobject z) { |
Ian Rogers | 408f79a | 2011-08-23 18:22:33 -0700 | [diff] [blame] | 179 | EXPECT_EQ(3u, Thread::Current()->NumSirtReferences()); |
Ian Rogers | b033c75 | 2011-07-20 12:22:35 -0700 | [diff] [blame] | 180 | EXPECT_EQ(Thread::kNative, Thread::Current()->GetState()); |
Ian Rogers | cdd1d2d | 2011-08-18 09:58:17 -0700 | [diff] [blame] | 181 | EXPECT_EQ(Thread::Current()->GetJniEnv(), env); |
| 182 | EXPECT_TRUE(thisObj != NULL); |
Ian Rogers | a8cd9f4 | 2011-08-19 16:43:41 -0700 | [diff] [blame] | 183 | EXPECT_TRUE(env->IsInstanceOf(thisObj, JniCompilerTest::jklass_)); |
Ian Rogers | b033c75 | 2011-07-20 12:22:35 -0700 | [diff] [blame] | 184 | gJava_MyClass_fooIOO_calls++; |
| 185 | switch (x) { |
| 186 | case 1: |
| 187 | return y; |
| 188 | case 2: |
| 189 | return z; |
| 190 | default: |
Ian Rogers | cdd1d2d | 2011-08-18 09:58:17 -0700 | [diff] [blame] | 191 | return thisObj; |
Ian Rogers | b033c75 | 2011-07-20 12:22:35 -0700 | [diff] [blame] | 192 | } |
| 193 | } |
| 194 | |
Ian Rogers | cdd1d2d | 2011-08-18 09:58:17 -0700 | [diff] [blame] | 195 | TEST_F(JniCompilerTest, CompileAndRunIntObjectObjectMethod) { |
| 196 | SetupForTest(false, "fooIOO", |
| 197 | "(ILjava/lang/Object;Ljava/lang/Object;)Ljava/lang/Object;", |
| 198 | reinterpret_cast<void*>(&Java_MyClass_fooIOO)); |
| 199 | |
| 200 | EXPECT_EQ(0, gJava_MyClass_fooIOO_calls); |
| 201 | jobject result = env_->CallNonvirtualObjectMethod(jobj_, jklass_, jmethod_, 0, NULL, NULL); |
| 202 | EXPECT_TRUE(env_->IsSameObject(jobj_, result)); |
| 203 | EXPECT_EQ(1, gJava_MyClass_fooIOO_calls); |
| 204 | |
| 205 | result = env_->CallNonvirtualObjectMethod(jobj_, jklass_, jmethod_, 0, NULL, jklass_); |
| 206 | EXPECT_TRUE(env_->IsSameObject(jobj_, result)); |
| 207 | EXPECT_EQ(2, gJava_MyClass_fooIOO_calls); |
| 208 | result = env_->CallNonvirtualObjectMethod(jobj_, jklass_, jmethod_, 1, NULL, jklass_); |
| 209 | EXPECT_TRUE(env_->IsSameObject(NULL, result)); |
| 210 | EXPECT_EQ(3, gJava_MyClass_fooIOO_calls); |
| 211 | result = env_->CallNonvirtualObjectMethod(jobj_, jklass_, jmethod_, 2, NULL, jklass_); |
| 212 | EXPECT_TRUE(env_->IsSameObject(jklass_, result)); |
| 213 | EXPECT_EQ(4, gJava_MyClass_fooIOO_calls); |
| 214 | |
| 215 | result = env_->CallNonvirtualObjectMethod(jobj_, jklass_, jmethod_, 0, jklass_, NULL); |
| 216 | EXPECT_TRUE(env_->IsSameObject(jobj_, result)); |
| 217 | EXPECT_EQ(5, gJava_MyClass_fooIOO_calls); |
| 218 | result = env_->CallNonvirtualObjectMethod(jobj_, jklass_, jmethod_, 1, jklass_, NULL); |
| 219 | EXPECT_TRUE(env_->IsSameObject(jklass_, result)); |
| 220 | EXPECT_EQ(6, gJava_MyClass_fooIOO_calls); |
| 221 | result = env_->CallNonvirtualObjectMethod(jobj_, jklass_, jmethod_, 2, jklass_, NULL); |
| 222 | EXPECT_TRUE(env_->IsSameObject(NULL, result)); |
| 223 | EXPECT_EQ(7, gJava_MyClass_fooIOO_calls); |
| 224 | } |
| 225 | |
Shih-wei Liao | 82da44b | 2011-09-01 00:38:04 -0700 | [diff] [blame] | 226 | int gJava_MyClass_fooSII_calls = 0; |
| 227 | jint Java_MyClass_fooSII(JNIEnv* env, jclass klass, jint x, jint y) { |
| 228 | EXPECT_EQ(1u, Thread::Current()->NumSirtReferences()); |
| 229 | EXPECT_EQ(Thread::kNative, Thread::Current()->GetState()); |
| 230 | EXPECT_EQ(Thread::Current()->GetJniEnv(), env); |
| 231 | EXPECT_TRUE(klass != NULL); |
| 232 | EXPECT_TRUE(env->IsInstanceOf(JniCompilerTest::jobj_, klass)); |
| 233 | gJava_MyClass_fooSII_calls++; |
| 234 | return x + y; |
| 235 | } |
| 236 | |
| 237 | |
| 238 | TEST_F(JniCompilerTest, CompileAndRunStaticIntIntMethod) { |
| 239 | SetupForTest(true, "fooSII", |
| 240 | "(II)I", |
| 241 | reinterpret_cast<void*>(&Java_MyClass_fooSII)); |
| 242 | |
| 243 | EXPECT_EQ(0, gJava_MyClass_fooSII_calls); |
| 244 | jint result = env_->CallStaticIntMethod(jklass_, jmethod_, 20, 30); |
| 245 | EXPECT_EQ(50, result); |
| 246 | EXPECT_EQ(1, gJava_MyClass_fooSII_calls); |
| 247 | } |
| 248 | |
Ian Rogers | b033c75 | 2011-07-20 12:22:35 -0700 | [diff] [blame] | 249 | int gJava_MyClass_fooSIOO_calls = 0; |
Elliott Hughes | 330304d | 2011-08-12 14:28:05 -0700 | [diff] [blame] | 250 | jobject Java_MyClass_fooSIOO(JNIEnv* env, jclass klass, jint x, jobject y, |
Ian Rogers | b033c75 | 2011-07-20 12:22:35 -0700 | [diff] [blame] | 251 | jobject z) { |
Ian Rogers | 408f79a | 2011-08-23 18:22:33 -0700 | [diff] [blame] | 252 | EXPECT_EQ(3u, Thread::Current()->NumSirtReferences()); |
Ian Rogers | b033c75 | 2011-07-20 12:22:35 -0700 | [diff] [blame] | 253 | EXPECT_EQ(Thread::kNative, Thread::Current()->GetState()); |
Ian Rogers | cdd1d2d | 2011-08-18 09:58:17 -0700 | [diff] [blame] | 254 | EXPECT_EQ(Thread::Current()->GetJniEnv(), env); |
| 255 | EXPECT_TRUE(klass != NULL); |
Ian Rogers | a8cd9f4 | 2011-08-19 16:43:41 -0700 | [diff] [blame] | 256 | EXPECT_TRUE(env->IsInstanceOf(JniCompilerTest::jobj_, klass)); |
Ian Rogers | b033c75 | 2011-07-20 12:22:35 -0700 | [diff] [blame] | 257 | gJava_MyClass_fooSIOO_calls++; |
| 258 | switch (x) { |
| 259 | case 1: |
| 260 | return y; |
| 261 | case 2: |
| 262 | return z; |
| 263 | default: |
| 264 | return klass; |
| 265 | } |
| 266 | } |
| 267 | |
Ian Rogers | cdd1d2d | 2011-08-18 09:58:17 -0700 | [diff] [blame] | 268 | |
| 269 | TEST_F(JniCompilerTest, CompileAndRunStaticIntObjectObjectMethod) { |
| 270 | SetupForTest(true, "fooSIOO", |
| 271 | "(ILjava/lang/Object;Ljava/lang/Object;)Ljava/lang/Object;", |
| 272 | reinterpret_cast<void*>(&Java_MyClass_fooSIOO)); |
| 273 | |
| 274 | EXPECT_EQ(0, gJava_MyClass_fooSIOO_calls); |
| 275 | jobject result = env_->CallStaticObjectMethod(jklass_, jmethod_, 0, NULL, NULL); |
| 276 | EXPECT_TRUE(env_->IsSameObject(jklass_, result)); |
| 277 | EXPECT_EQ(1, gJava_MyClass_fooSIOO_calls); |
| 278 | |
| 279 | result = env_->CallStaticObjectMethod(jklass_, jmethod_, 0, NULL, jobj_); |
| 280 | EXPECT_TRUE(env_->IsSameObject(jklass_, result)); |
| 281 | EXPECT_EQ(2, gJava_MyClass_fooSIOO_calls); |
| 282 | result = env_->CallStaticObjectMethod(jklass_, jmethod_, 1, NULL, jobj_); |
| 283 | EXPECT_TRUE(env_->IsSameObject(NULL, result)); |
| 284 | EXPECT_EQ(3, gJava_MyClass_fooSIOO_calls); |
| 285 | result = env_->CallStaticObjectMethod(jklass_, jmethod_, 2, NULL, jobj_); |
| 286 | EXPECT_TRUE(env_->IsSameObject(jobj_, result)); |
| 287 | EXPECT_EQ(4, gJava_MyClass_fooSIOO_calls); |
| 288 | |
| 289 | result = env_->CallStaticObjectMethod(jklass_, jmethod_, 0, jobj_, NULL); |
| 290 | EXPECT_TRUE(env_->IsSameObject(jklass_, result)); |
| 291 | EXPECT_EQ(5, gJava_MyClass_fooSIOO_calls); |
| 292 | result = env_->CallStaticObjectMethod(jklass_, jmethod_, 1, jobj_, NULL); |
| 293 | EXPECT_TRUE(env_->IsSameObject(jobj_, result)); |
| 294 | EXPECT_EQ(6, gJava_MyClass_fooSIOO_calls); |
| 295 | result = env_->CallStaticObjectMethod(jklass_, jmethod_, 2, jobj_, NULL); |
| 296 | EXPECT_TRUE(env_->IsSameObject(NULL, result)); |
| 297 | EXPECT_EQ(7, gJava_MyClass_fooSIOO_calls); |
| 298 | } |
| 299 | |
Ian Rogers | df20fe0 | 2011-07-20 20:34:16 -0700 | [diff] [blame] | 300 | int gJava_MyClass_fooSSIOO_calls = 0; |
Ian Rogers | cdd1d2d | 2011-08-18 09:58:17 -0700 | [diff] [blame] | 301 | jobject Java_MyClass_fooSSIOO(JNIEnv* env, jclass klass, jint x, jobject y, |
Ian Rogers | df20fe0 | 2011-07-20 20:34:16 -0700 | [diff] [blame] | 302 | jobject z) { |
Ian Rogers | 408f79a | 2011-08-23 18:22:33 -0700 | [diff] [blame] | 303 | EXPECT_EQ(3u, Thread::Current()->NumSirtReferences()); |
Ian Rogers | df20fe0 | 2011-07-20 20:34:16 -0700 | [diff] [blame] | 304 | EXPECT_EQ(Thread::kNative, Thread::Current()->GetState()); |
Ian Rogers | cdd1d2d | 2011-08-18 09:58:17 -0700 | [diff] [blame] | 305 | EXPECT_EQ(Thread::Current()->GetJniEnv(), env); |
| 306 | EXPECT_TRUE(klass != NULL); |
Ian Rogers | a8cd9f4 | 2011-08-19 16:43:41 -0700 | [diff] [blame] | 307 | EXPECT_TRUE(env->IsInstanceOf(JniCompilerTest::jobj_, klass)); |
Ian Rogers | df20fe0 | 2011-07-20 20:34:16 -0700 | [diff] [blame] | 308 | gJava_MyClass_fooSSIOO_calls++; |
| 309 | switch (x) { |
| 310 | case 1: |
| 311 | return y; |
| 312 | case 2: |
| 313 | return z; |
| 314 | default: |
| 315 | return klass; |
| 316 | } |
| 317 | } |
| 318 | |
Ian Rogers | df20fe0 | 2011-07-20 20:34:16 -0700 | [diff] [blame] | 319 | TEST_F(JniCompilerTest, CompileAndRunStaticSynchronizedIntObjectObjectMethod) { |
Ian Rogers | cdd1d2d | 2011-08-18 09:58:17 -0700 | [diff] [blame] | 320 | SetupForTest(true, "fooSSIOO", |
| 321 | "(ILjava/lang/Object;Ljava/lang/Object;)Ljava/lang/Object;", |
| 322 | reinterpret_cast<void*>(&Java_MyClass_fooSSIOO)); |
Carl Shapiro | 419ec7b | 2011-08-03 14:48:33 -0700 | [diff] [blame] | 323 | |
Ian Rogers | df20fe0 | 2011-07-20 20:34:16 -0700 | [diff] [blame] | 324 | EXPECT_EQ(0, gJava_MyClass_fooSSIOO_calls); |
Ian Rogers | cdd1d2d | 2011-08-18 09:58:17 -0700 | [diff] [blame] | 325 | jobject result = env_->CallStaticObjectMethod(jklass_, jmethod_, 0, NULL, NULL); |
| 326 | EXPECT_TRUE(env_->IsSameObject(jklass_, result)); |
Ian Rogers | df20fe0 | 2011-07-20 20:34:16 -0700 | [diff] [blame] | 327 | EXPECT_EQ(1, gJava_MyClass_fooSSIOO_calls); |
Ian Rogers | cdd1d2d | 2011-08-18 09:58:17 -0700 | [diff] [blame] | 328 | |
| 329 | result = env_->CallStaticObjectMethod(jklass_, jmethod_, 0, NULL, jobj_); |
| 330 | EXPECT_TRUE(env_->IsSameObject(jklass_, result)); |
Ian Rogers | df20fe0 | 2011-07-20 20:34:16 -0700 | [diff] [blame] | 331 | EXPECT_EQ(2, gJava_MyClass_fooSSIOO_calls); |
Ian Rogers | cdd1d2d | 2011-08-18 09:58:17 -0700 | [diff] [blame] | 332 | result = env_->CallStaticObjectMethod(jklass_, jmethod_, 1, NULL, jobj_); |
| 333 | EXPECT_TRUE(env_->IsSameObject(NULL, result)); |
Ian Rogers | df20fe0 | 2011-07-20 20:34:16 -0700 | [diff] [blame] | 334 | EXPECT_EQ(3, gJava_MyClass_fooSSIOO_calls); |
Ian Rogers | cdd1d2d | 2011-08-18 09:58:17 -0700 | [diff] [blame] | 335 | result = env_->CallStaticObjectMethod(jklass_, jmethod_, 2, NULL, jobj_); |
| 336 | EXPECT_TRUE(env_->IsSameObject(jobj_, result)); |
Ian Rogers | df20fe0 | 2011-07-20 20:34:16 -0700 | [diff] [blame] | 337 | EXPECT_EQ(4, gJava_MyClass_fooSSIOO_calls); |
Ian Rogers | cdd1d2d | 2011-08-18 09:58:17 -0700 | [diff] [blame] | 338 | |
| 339 | result = env_->CallStaticObjectMethod(jklass_, jmethod_, 0, jobj_, NULL); |
| 340 | EXPECT_TRUE(env_->IsSameObject(jklass_, result)); |
Ian Rogers | df20fe0 | 2011-07-20 20:34:16 -0700 | [diff] [blame] | 341 | EXPECT_EQ(5, gJava_MyClass_fooSSIOO_calls); |
Ian Rogers | cdd1d2d | 2011-08-18 09:58:17 -0700 | [diff] [blame] | 342 | result = env_->CallStaticObjectMethod(jklass_, jmethod_, 1, jobj_, NULL); |
| 343 | EXPECT_TRUE(env_->IsSameObject(jobj_, result)); |
Ian Rogers | df20fe0 | 2011-07-20 20:34:16 -0700 | [diff] [blame] | 344 | EXPECT_EQ(6, gJava_MyClass_fooSSIOO_calls); |
Ian Rogers | cdd1d2d | 2011-08-18 09:58:17 -0700 | [diff] [blame] | 345 | result = env_->CallStaticObjectMethod(jklass_, jmethod_, 2, jobj_, NULL); |
| 346 | EXPECT_TRUE(env_->IsSameObject(NULL, result)); |
Ian Rogers | df20fe0 | 2011-07-20 20:34:16 -0700 | [diff] [blame] | 347 | EXPECT_EQ(7, gJava_MyClass_fooSSIOO_calls); |
| 348 | } |
| 349 | |
Ian Rogers | 45a76cb | 2011-07-21 22:00:15 -0700 | [diff] [blame] | 350 | int gSuspendCounterHandler_calls; |
| 351 | void SuspendCountHandler(Method** frame) { |
Ian Rogers | 0cfe1fb | 2011-08-26 03:29:44 -0700 | [diff] [blame] | 352 | EXPECT_EQ(Thread::kNative, Thread::Current()->GetState()); |
| 353 | Thread::Current()->SetState(Thread::kRunnable); |
Carl Shapiro | 8860c0e | 2011-08-04 17:36:16 -0700 | [diff] [blame] | 354 | EXPECT_TRUE((*frame)->GetName()->Equals("fooI")); |
Ian Rogers | 45a76cb | 2011-07-21 22:00:15 -0700 | [diff] [blame] | 355 | gSuspendCounterHandler_calls++; |
| 356 | Thread::Current()->DecrementSuspendCount(); |
Ian Rogers | 0cfe1fb | 2011-08-26 03:29:44 -0700 | [diff] [blame] | 357 | Thread::Current()->SetState(Thread::kNative); |
Ian Rogers | 45a76cb | 2011-07-21 22:00:15 -0700 | [diff] [blame] | 358 | } |
Ian Rogers | cdd1d2d | 2011-08-18 09:58:17 -0700 | [diff] [blame] | 359 | |
Ian Rogers | 0d666d8 | 2011-08-14 16:03:46 -0700 | [diff] [blame] | 360 | TEST_F(JniCompilerTest, SuspendCountAcknowledgement) { |
Ian Rogers | cdd1d2d | 2011-08-18 09:58:17 -0700 | [diff] [blame] | 361 | SetupForTest(false, "fooI", "(I)I", |
| 362 | reinterpret_cast<void*>(&Java_MyClass_fooI)); |
Ian Rogers | 45a76cb | 2011-07-21 22:00:15 -0700 | [diff] [blame] | 363 | Thread::Current()->RegisterSuspendCountEntryPoint(&SuspendCountHandler); |
| 364 | |
Ian Rogers | 45a76cb | 2011-07-21 22:00:15 -0700 | [diff] [blame] | 365 | gJava_MyClass_fooI_calls = 0; |
Ian Rogers | cdd1d2d | 2011-08-18 09:58:17 -0700 | [diff] [blame] | 366 | jint result = env_->CallNonvirtualIntMethod(jobj_, jklass_, jmethod_, 42); |
| 367 | EXPECT_EQ(42, result); |
Ian Rogers | 45a76cb | 2011-07-21 22:00:15 -0700 | [diff] [blame] | 368 | EXPECT_EQ(1, gJava_MyClass_fooI_calls); |
| 369 | EXPECT_EQ(0, gSuspendCounterHandler_calls); |
| 370 | Thread::Current()->IncrementSuspendCount(); |
Ian Rogers | cdd1d2d | 2011-08-18 09:58:17 -0700 | [diff] [blame] | 371 | result = env_->CallNonvirtualIntMethod(jobj_, jklass_, jmethod_, 42); |
| 372 | EXPECT_EQ(42, result); |
Ian Rogers | 45a76cb | 2011-07-21 22:00:15 -0700 | [diff] [blame] | 373 | EXPECT_EQ(2, gJava_MyClass_fooI_calls); |
| 374 | EXPECT_EQ(1, gSuspendCounterHandler_calls); |
Ian Rogers | cdd1d2d | 2011-08-18 09:58:17 -0700 | [diff] [blame] | 375 | result = env_->CallNonvirtualIntMethod(jobj_, jklass_, jmethod_, 42); |
| 376 | EXPECT_EQ(42, result); |
Ian Rogers | 45a76cb | 2011-07-21 22:00:15 -0700 | [diff] [blame] | 377 | EXPECT_EQ(3, gJava_MyClass_fooI_calls); |
| 378 | EXPECT_EQ(1, gSuspendCounterHandler_calls); |
| 379 | } |
| 380 | |
| 381 | int gExceptionHandler_calls; |
| 382 | void ExceptionHandler(Method** frame) { |
Ian Rogers | 0cfe1fb | 2011-08-26 03:29:44 -0700 | [diff] [blame] | 383 | EXPECT_EQ(Thread::kNative, Thread::Current()->GetState()); |
| 384 | Thread::Current()->SetState(Thread::kRunnable); |
Elliott Hughes | a250199 | 2011-08-26 19:39:54 -0700 | [diff] [blame] | 385 | EXPECT_TRUE((*frame)->GetName()->Equals("throwException")); |
Ian Rogers | 45a76cb | 2011-07-21 22:00:15 -0700 | [diff] [blame] | 386 | gExceptionHandler_calls++; |
| 387 | Thread::Current()->ClearException(); |
Ian Rogers | 0cfe1fb | 2011-08-26 03:29:44 -0700 | [diff] [blame] | 388 | Thread::Current()->SetState(Thread::kNative); |
Ian Rogers | 45a76cb | 2011-07-21 22:00:15 -0700 | [diff] [blame] | 389 | } |
Ian Rogers | cdd1d2d | 2011-08-18 09:58:17 -0700 | [diff] [blame] | 390 | |
Elliott Hughes | a250199 | 2011-08-26 19:39:54 -0700 | [diff] [blame] | 391 | void Java_MyClass_throwException(JNIEnv* env, jobject) { |
| 392 | jclass c = env->FindClass("java/lang/RuntimeException"); |
| 393 | env->ThrowNew(c, "hello"); |
| 394 | } |
Ian Rogers | 45a76cb | 2011-07-21 22:00:15 -0700 | [diff] [blame] | 395 | |
Elliott Hughes | a250199 | 2011-08-26 19:39:54 -0700 | [diff] [blame] | 396 | TEST_F(JniCompilerTest, ExceptionHandling) { |
| 397 | Thread::Current()->RegisterExceptionEntryPoint(&ExceptionHandler); |
| 398 | gExceptionHandler_calls = 0; |
Ian Rogers | 45a76cb | 2011-07-21 22:00:15 -0700 | [diff] [blame] | 399 | gJava_MyClass_foo_calls = 0; |
Elliott Hughes | a250199 | 2011-08-26 19:39:54 -0700 | [diff] [blame] | 400 | |
| 401 | SetupForTest(false, "foo", "()V", reinterpret_cast<void*>(&Java_MyClass_foo)); |
Ian Rogers | cdd1d2d | 2011-08-18 09:58:17 -0700 | [diff] [blame] | 402 | env_->CallNonvirtualVoidMethod(jobj_, jklass_, jmethod_); |
Ian Rogers | 45a76cb | 2011-07-21 22:00:15 -0700 | [diff] [blame] | 403 | EXPECT_EQ(1, gJava_MyClass_foo_calls); |
| 404 | EXPECT_EQ(0, gExceptionHandler_calls); |
Elliott Hughes | a250199 | 2011-08-26 19:39:54 -0700 | [diff] [blame] | 405 | |
| 406 | SetupForTest(false, "throwException", "()V", reinterpret_cast<void*>(&Java_MyClass_throwException)); |
| 407 | env_->CallNonvirtualVoidMethod(jobj_, jklass_, jmethod_); |
| 408 | EXPECT_EQ(1, gJava_MyClass_foo_calls); |
| 409 | EXPECT_EQ(1, gExceptionHandler_calls); |
| 410 | |
| 411 | SetupForTest(false, "foo", "()V", reinterpret_cast<void*>(&Java_MyClass_foo)); |
Ian Rogers | cdd1d2d | 2011-08-18 09:58:17 -0700 | [diff] [blame] | 412 | env_->CallNonvirtualVoidMethod(jobj_, jklass_, jmethod_); |
Ian Rogers | 45a76cb | 2011-07-21 22:00:15 -0700 | [diff] [blame] | 413 | EXPECT_EQ(2, gJava_MyClass_foo_calls); |
| 414 | EXPECT_EQ(1, gExceptionHandler_calls); |
Ian Rogers | 45a76cb | 2011-07-21 22:00:15 -0700 | [diff] [blame] | 415 | } |
| 416 | |
Shih-wei Liao | ff0f9be | 2011-08-29 15:43:53 -0700 | [diff] [blame] | 417 | jint Java_MyClass_nativeUpCall(JNIEnv* env, jobject thisObj, jint i) { |
| 418 | if (i <= 0) { |
Ian Rogers | 0cfe1fb | 2011-08-26 03:29:44 -0700 | [diff] [blame] | 419 | EXPECT_EQ(Thread::kNative, Thread::Current()->GetState()); |
| 420 | Thread::Current()->SetState(Thread::kRunnable); |
Shih-wei Liao | ff0f9be | 2011-08-29 15:43:53 -0700 | [diff] [blame] | 421 | ObjectArray<StackTraceElement>* trace_array = Thread::Current()->AllocStackTrace(); |
| 422 | EXPECT_TRUE(trace_array != NULL); |
| 423 | EXPECT_EQ(11, trace_array->GetLength()); |
| 424 | |
| 425 | for (int32_t i = 0; i < trace_array->GetLength(); ++i) { |
| 426 | EXPECT_EQ(-2, trace_array->Get(i)->GetLineNumber()); |
| 427 | EXPECT_STREQ("MyClassNatives.java", trace_array->Get(i)->GetFileName()->ToModifiedUtf8().c_str()); |
| 428 | EXPECT_STREQ("MyClass", trace_array->Get(i)->GetDeclaringClass()->ToModifiedUtf8().c_str()); |
| 429 | EXPECT_STREQ("fooI", trace_array->Get(i)->GetMethodName()->ToModifiedUtf8().c_str()); |
| 430 | } |
Ian Rogers | 0cfe1fb | 2011-08-26 03:29:44 -0700 | [diff] [blame] | 431 | Thread::Current()->SetState(Thread::kNative); |
Shih-wei Liao | ff0f9be | 2011-08-29 15:43:53 -0700 | [diff] [blame] | 432 | return 0; |
| 433 | } else { |
| 434 | jclass jklass = env->FindClass("MyClass"); |
| 435 | EXPECT_TRUE(jklass != NULL); |
| 436 | jmethodID jmethod = env->GetMethodID(jklass, "fooI", "(I)I"); |
| 437 | EXPECT_TRUE(jmethod != NULL); |
| 438 | |
| 439 | jint result = env->CallNonvirtualIntMethod(thisObj, jklass, jmethod, i - 1); |
| 440 | return i + result; |
| 441 | } |
| 442 | } |
| 443 | |
| 444 | TEST_F(JniCompilerTest, NativeStackTraceElement) { |
| 445 | SetupForTest(false, "fooI", "(I)I", reinterpret_cast<void*>(&Java_MyClass_nativeUpCall)); |
| 446 | jint result = env_->CallNonvirtualIntMethod(jobj_, jklass_, jmethod_, 10); |
| 447 | EXPECT_EQ(55, result); |
| 448 | } |
| 449 | |
Ian Rogers | b033c75 | 2011-07-20 12:22:35 -0700 | [diff] [blame] | 450 | } // namespace art |