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 | |
| 3 | #include <sys/mman.h> |
Brian Carlstrom | 578bbdc | 2011-07-21 14:07:47 -0700 | [diff] [blame] | 4 | |
| 5 | #include "assembler.h" |
| 6 | #include "class_linker.h" |
| 7 | #include "common_test.h" |
| 8 | #include "dex_file.h" |
Ian Rogers | cdd1d2d | 2011-08-18 09:58:17 -0700 | [diff] [blame] | 9 | #include "indirect_reference_table.h" |
Brian Carlstrom | 578bbdc | 2011-07-21 14:07:47 -0700 | [diff] [blame] | 10 | #include "jni_compiler.h" |
Ian Rogers | cdd1d2d | 2011-08-18 09:58:17 -0700 | [diff] [blame] | 11 | #include "jni_internal.h" |
Brian Carlstrom | db4d540 | 2011-08-09 12:18:28 -0700 | [diff] [blame] | 12 | #include "mem_map.h" |
Brian Carlstrom | 578bbdc | 2011-07-21 14:07:47 -0700 | [diff] [blame] | 13 | #include "runtime.h" |
Brian Carlstrom | db4d540 | 2011-08-09 12:18:28 -0700 | [diff] [blame] | 14 | #include "scoped_ptr.h" |
Brian Carlstrom | 578bbdc | 2011-07-21 14:07:47 -0700 | [diff] [blame] | 15 | #include "thread.h" |
Ian Rogers | b033c75 | 2011-07-20 12:22:35 -0700 | [diff] [blame] | 16 | #include "gtest/gtest.h" |
| 17 | |
| 18 | namespace art { |
| 19 | |
Brian Carlstrom | f734cf5 | 2011-08-17 16:28:14 -0700 | [diff] [blame] | 20 | class JniCompilerTest : public CommonTest { |
Ian Rogers | b033c75 | 2011-07-20 12:22:35 -0700 | [diff] [blame] | 21 | protected: |
| 22 | virtual void SetUp() { |
Brian Carlstrom | f734cf5 | 2011-08-17 16:28:14 -0700 | [diff] [blame] | 23 | CommonTest::SetUp(); |
Ian Rogers | cdd1d2d | 2011-08-18 09:58:17 -0700 | [diff] [blame] | 24 | dex_.reset(OpenDexFileBase64(kMyClassNativesDex)); |
| 25 | class_loader_ = AllocPathClassLoader(dex_.get()); |
Ian Rogers | b033c75 | 2011-07-20 12:22:35 -0700 | [diff] [blame] | 26 | } |
| 27 | |
Ian Rogers | cdd1d2d | 2011-08-18 09:58:17 -0700 | [diff] [blame] | 28 | void SetupForTest(bool direct, const char* method_name, |
| 29 | const char* method_sig, void* native_fnptr) { |
| 30 | const char* class_name = "LMyClass;"; |
| 31 | Class* klass = class_linker_->FindClass(class_name, class_loader_); |
| 32 | ASSERT_TRUE(klass != NULL); |
| 33 | |
| 34 | Method* method; |
| 35 | if (direct) { |
| 36 | method = klass->FindDirectMethod(method_name, method_sig); |
| 37 | } else { |
| 38 | method = klass->FindVirtualMethod(method_name, method_sig); |
| 39 | } |
| 40 | ASSERT_TRUE(method != NULL); |
| 41 | |
| 42 | // Compile the native method |
| 43 | jni_compiler.Compile(&jni_asm, method); |
| 44 | |
| 45 | env_ = Thread::Current()->GetJniEnv(); |
| 46 | |
| 47 | // TODO: when we support class loaders - env->FindClass(class_name); |
| 48 | IndirectReferenceTable& locals = reinterpret_cast<JNIEnvExt*>(env_)->locals; |
| 49 | uint32_t cookie = IRT_FIRST_SEGMENT; // TODO |
| 50 | IndirectRef klass_ref = locals.Add(cookie, klass); |
| 51 | jklass_ = reinterpret_cast<jclass>(klass_ref); |
| 52 | if (direct) { |
| 53 | jmethod_ = env_->GetStaticMethodID(jklass_, method_name, method_sig); |
| 54 | } else { |
| 55 | jmethod_ = env_->GetMethodID(jklass_, method_name, method_sig); |
| 56 | } |
| 57 | ASSERT_TRUE(jmethod_ != NULL); |
| 58 | |
| 59 | JNINativeMethod methods[] = {{method_name, method_sig, native_fnptr}}; |
| 60 | ASSERT_EQ(JNI_OK, env_->RegisterNatives(jklass_, methods, 1)); |
| 61 | |
| 62 | jmethodID constructor = env_->GetMethodID(jklass_, "<init>", "()V"); |
| 63 | jobj_ = env_->NewObject(jklass_, constructor); |
| 64 | ASSERT_TRUE(jobj_ != NULL); |
Ian Rogers | b033c75 | 2011-07-20 12:22:35 -0700 | [diff] [blame] | 65 | } |
| 66 | |
Ian Rogers | cdd1d2d | 2011-08-18 09:58:17 -0700 | [diff] [blame] | 67 | public: |
| 68 | static jclass jklass_; |
| 69 | static jobject jobj_; |
| 70 | protected: |
| 71 | scoped_ptr<DexFile> dex_; |
| 72 | PathClassLoader* class_loader_; |
| 73 | Assembler jni_asm; |
| 74 | JniCompiler jni_compiler; |
| 75 | JNIEnv* env_; |
| 76 | jmethodID jmethod_; |
Ian Rogers | b033c75 | 2011-07-20 12:22:35 -0700 | [diff] [blame] | 77 | }; |
| 78 | |
Ian Rogers | cdd1d2d | 2011-08-18 09:58:17 -0700 | [diff] [blame] | 79 | jclass JniCompilerTest::jklass_; |
| 80 | jobject JniCompilerTest::jobj_; |
| 81 | |
Ian Rogers | b033c75 | 2011-07-20 12:22:35 -0700 | [diff] [blame] | 82 | int gJava_MyClass_foo_calls = 0; |
Ian Rogers | cdd1d2d | 2011-08-18 09:58:17 -0700 | [diff] [blame] | 83 | void Java_MyClass_foo(JNIEnv* env, jobject thisObj) { |
Ian Rogers | b033c75 | 2011-07-20 12:22:35 -0700 | [diff] [blame] | 84 | EXPECT_EQ(1u, Thread::Current()->NumShbHandles()); |
| 85 | EXPECT_EQ(Thread::kNative, Thread::Current()->GetState()); |
Ian Rogers | cdd1d2d | 2011-08-18 09:58:17 -0700 | [diff] [blame] | 86 | EXPECT_EQ(Thread::Current()->GetJniEnv(), env); |
| 87 | EXPECT_TRUE(thisObj != NULL); |
Ian Rogers | a8cd9f4 | 2011-08-19 16:43:41 -0700 | [diff] [blame^] | 88 | EXPECT_TRUE(env->IsInstanceOf(thisObj, JniCompilerTest::jklass_)); |
Ian Rogers | b033c75 | 2011-07-20 12:22:35 -0700 | [diff] [blame] | 89 | gJava_MyClass_foo_calls++; |
| 90 | } |
| 91 | |
Ian Rogers | cdd1d2d | 2011-08-18 09:58:17 -0700 | [diff] [blame] | 92 | TEST_F(JniCompilerTest, CompileAndRunNoArgMethod) { |
| 93 | SetupForTest(false, "foo", "()V", reinterpret_cast<void*>(&Java_MyClass_foo)); |
| 94 | |
| 95 | EXPECT_EQ(0, gJava_MyClass_foo_calls); |
| 96 | env_->CallNonvirtualVoidMethod(jobj_, jklass_, jmethod_); |
| 97 | EXPECT_EQ(1, gJava_MyClass_foo_calls); |
| 98 | env_->CallNonvirtualVoidMethod(jobj_, jklass_, jmethod_); |
| 99 | EXPECT_EQ(2, gJava_MyClass_foo_calls); |
| 100 | } |
| 101 | |
Ian Rogers | b033c75 | 2011-07-20 12:22:35 -0700 | [diff] [blame] | 102 | int gJava_MyClass_fooI_calls = 0; |
Ian Rogers | cdd1d2d | 2011-08-18 09:58:17 -0700 | [diff] [blame] | 103 | jint Java_MyClass_fooI(JNIEnv* env, jobject thisObj, jint x) { |
Ian Rogers | b033c75 | 2011-07-20 12:22:35 -0700 | [diff] [blame] | 104 | EXPECT_EQ(1u, Thread::Current()->NumShbHandles()); |
| 105 | EXPECT_EQ(Thread::kNative, Thread::Current()->GetState()); |
Ian Rogers | cdd1d2d | 2011-08-18 09:58:17 -0700 | [diff] [blame] | 106 | EXPECT_EQ(Thread::Current()->GetJniEnv(), env); |
| 107 | EXPECT_TRUE(thisObj != NULL); |
Ian Rogers | a8cd9f4 | 2011-08-19 16:43:41 -0700 | [diff] [blame^] | 108 | EXPECT_TRUE(env->IsInstanceOf(thisObj, JniCompilerTest::jklass_)); |
Ian Rogers | b033c75 | 2011-07-20 12:22:35 -0700 | [diff] [blame] | 109 | gJava_MyClass_fooI_calls++; |
| 110 | return x; |
| 111 | } |
| 112 | |
Ian Rogers | cdd1d2d | 2011-08-18 09:58:17 -0700 | [diff] [blame] | 113 | TEST_F(JniCompilerTest, CompileAndRunIntMethod) { |
| 114 | SetupForTest(false, "fooI", "(I)I", |
| 115 | reinterpret_cast<void*>(&Java_MyClass_fooI)); |
| 116 | |
| 117 | EXPECT_EQ(0, gJava_MyClass_fooI_calls); |
| 118 | jint result = env_->CallNonvirtualIntMethod(jobj_, jklass_, jmethod_, 42); |
| 119 | EXPECT_EQ(42, result); |
| 120 | EXPECT_EQ(1, gJava_MyClass_fooI_calls); |
| 121 | result = env_->CallNonvirtualIntMethod(jobj_, jklass_, jmethod_, 0xCAFED00D); |
| 122 | EXPECT_EQ(static_cast<jint>(0xCAFED00D), result); |
| 123 | EXPECT_EQ(2, gJava_MyClass_fooI_calls); |
| 124 | } |
| 125 | |
Ian Rogers | b033c75 | 2011-07-20 12:22:35 -0700 | [diff] [blame] | 126 | int gJava_MyClass_fooII_calls = 0; |
Ian Rogers | cdd1d2d | 2011-08-18 09:58:17 -0700 | [diff] [blame] | 127 | jint Java_MyClass_fooII(JNIEnv* env, jobject thisObj, jint x, jint y) { |
Ian Rogers | b033c75 | 2011-07-20 12:22:35 -0700 | [diff] [blame] | 128 | EXPECT_EQ(1u, Thread::Current()->NumShbHandles()); |
| 129 | EXPECT_EQ(Thread::kNative, Thread::Current()->GetState()); |
Ian Rogers | cdd1d2d | 2011-08-18 09:58:17 -0700 | [diff] [blame] | 130 | EXPECT_EQ(Thread::Current()->GetJniEnv(), env); |
| 131 | EXPECT_TRUE(thisObj != NULL); |
Ian Rogers | a8cd9f4 | 2011-08-19 16:43:41 -0700 | [diff] [blame^] | 132 | EXPECT_TRUE(env->IsInstanceOf(thisObj, JniCompilerTest::jklass_)); |
Ian Rogers | b033c75 | 2011-07-20 12:22:35 -0700 | [diff] [blame] | 133 | gJava_MyClass_fooII_calls++; |
| 134 | return x - y; // non-commutative operator |
| 135 | } |
| 136 | |
Ian Rogers | cdd1d2d | 2011-08-18 09:58:17 -0700 | [diff] [blame] | 137 | TEST_F(JniCompilerTest, CompileAndRunIntIntMethod) { |
| 138 | SetupForTest(false, "fooII", "(II)I", |
| 139 | reinterpret_cast<void*>(&Java_MyClass_fooII)); |
| 140 | |
| 141 | EXPECT_EQ(0, gJava_MyClass_fooII_calls); |
| 142 | jint result = env_->CallNonvirtualIntMethod(jobj_, jklass_, jmethod_, 99, 10); |
| 143 | EXPECT_EQ(99 - 10, result); |
| 144 | EXPECT_EQ(1, gJava_MyClass_fooII_calls); |
| 145 | result = env_->CallNonvirtualIntMethod(jobj_, jklass_, jmethod_, 0xCAFEBABE, |
| 146 | 0xCAFED00D); |
| 147 | EXPECT_EQ(static_cast<jint>(0xCAFEBABE - 0xCAFED00D), result); |
| 148 | EXPECT_EQ(2, gJava_MyClass_fooII_calls); |
| 149 | } |
| 150 | |
Ian Rogers | b033c75 | 2011-07-20 12:22:35 -0700 | [diff] [blame] | 151 | int gJava_MyClass_fooDD_calls = 0; |
Ian Rogers | cdd1d2d | 2011-08-18 09:58:17 -0700 | [diff] [blame] | 152 | jdouble Java_MyClass_fooDD(JNIEnv* env, jobject thisObj, jdouble x, jdouble y) { |
Ian Rogers | b033c75 | 2011-07-20 12:22:35 -0700 | [diff] [blame] | 153 | EXPECT_EQ(1u, Thread::Current()->NumShbHandles()); |
| 154 | EXPECT_EQ(Thread::kNative, Thread::Current()->GetState()); |
Ian Rogers | cdd1d2d | 2011-08-18 09:58:17 -0700 | [diff] [blame] | 155 | EXPECT_EQ(Thread::Current()->GetJniEnv(), env); |
| 156 | EXPECT_TRUE(thisObj != NULL); |
Ian Rogers | a8cd9f4 | 2011-08-19 16:43:41 -0700 | [diff] [blame^] | 157 | EXPECT_TRUE(env->IsInstanceOf(thisObj, JniCompilerTest::jklass_)); |
Ian Rogers | b033c75 | 2011-07-20 12:22:35 -0700 | [diff] [blame] | 158 | gJava_MyClass_fooDD_calls++; |
| 159 | return x - y; // non-commutative operator |
| 160 | } |
| 161 | |
Ian Rogers | cdd1d2d | 2011-08-18 09:58:17 -0700 | [diff] [blame] | 162 | TEST_F(JniCompilerTest, CompileAndRunDoubleDoubleMethod) { |
| 163 | SetupForTest(false, "fooDD", "(DD)D", |
| 164 | reinterpret_cast<void*>(&Java_MyClass_fooDD)); |
| 165 | |
| 166 | EXPECT_EQ(0, gJava_MyClass_fooDD_calls); |
| 167 | jdouble result = env_->CallNonvirtualDoubleMethod(jobj_, jklass_, jmethod_, |
| 168 | 99.0, 10.0); |
| 169 | EXPECT_EQ(99.0 - 10.0, result); |
| 170 | EXPECT_EQ(1, gJava_MyClass_fooDD_calls); |
| 171 | jdouble a = 3.14159265358979323846; |
| 172 | jdouble b = 0.69314718055994530942; |
| 173 | result = env_->CallNonvirtualDoubleMethod(jobj_, jklass_, jmethod_, a, b); |
| 174 | EXPECT_EQ(a - b, result); |
| 175 | EXPECT_EQ(2, gJava_MyClass_fooDD_calls); |
| 176 | } |
| 177 | |
Ian Rogers | b033c75 | 2011-07-20 12:22:35 -0700 | [diff] [blame] | 178 | int gJava_MyClass_fooIOO_calls = 0; |
Ian Rogers | cdd1d2d | 2011-08-18 09:58:17 -0700 | [diff] [blame] | 179 | jobject Java_MyClass_fooIOO(JNIEnv* env, jobject thisObj, jint x, jobject y, |
Ian Rogers | b033c75 | 2011-07-20 12:22:35 -0700 | [diff] [blame] | 180 | jobject z) { |
| 181 | EXPECT_EQ(3u, Thread::Current()->NumShbHandles()); |
| 182 | EXPECT_EQ(Thread::kNative, Thread::Current()->GetState()); |
Ian Rogers | cdd1d2d | 2011-08-18 09:58:17 -0700 | [diff] [blame] | 183 | EXPECT_EQ(Thread::Current()->GetJniEnv(), env); |
| 184 | EXPECT_TRUE(thisObj != NULL); |
Ian Rogers | a8cd9f4 | 2011-08-19 16:43:41 -0700 | [diff] [blame^] | 185 | EXPECT_TRUE(env->IsInstanceOf(thisObj, JniCompilerTest::jklass_)); |
Ian Rogers | b033c75 | 2011-07-20 12:22:35 -0700 | [diff] [blame] | 186 | gJava_MyClass_fooIOO_calls++; |
| 187 | switch (x) { |
| 188 | case 1: |
| 189 | return y; |
| 190 | case 2: |
| 191 | return z; |
| 192 | default: |
Ian Rogers | cdd1d2d | 2011-08-18 09:58:17 -0700 | [diff] [blame] | 193 | return thisObj; |
Ian Rogers | b033c75 | 2011-07-20 12:22:35 -0700 | [diff] [blame] | 194 | } |
| 195 | } |
| 196 | |
Ian Rogers | cdd1d2d | 2011-08-18 09:58:17 -0700 | [diff] [blame] | 197 | TEST_F(JniCompilerTest, CompileAndRunIntObjectObjectMethod) { |
| 198 | SetupForTest(false, "fooIOO", |
| 199 | "(ILjava/lang/Object;Ljava/lang/Object;)Ljava/lang/Object;", |
| 200 | reinterpret_cast<void*>(&Java_MyClass_fooIOO)); |
| 201 | |
| 202 | EXPECT_EQ(0, gJava_MyClass_fooIOO_calls); |
| 203 | jobject result = env_->CallNonvirtualObjectMethod(jobj_, jklass_, jmethod_, 0, NULL, NULL); |
| 204 | EXPECT_TRUE(env_->IsSameObject(jobj_, result)); |
| 205 | EXPECT_EQ(1, gJava_MyClass_fooIOO_calls); |
| 206 | |
| 207 | result = env_->CallNonvirtualObjectMethod(jobj_, jklass_, jmethod_, 0, NULL, jklass_); |
| 208 | EXPECT_TRUE(env_->IsSameObject(jobj_, result)); |
| 209 | EXPECT_EQ(2, gJava_MyClass_fooIOO_calls); |
| 210 | result = env_->CallNonvirtualObjectMethod(jobj_, jklass_, jmethod_, 1, NULL, jklass_); |
| 211 | EXPECT_TRUE(env_->IsSameObject(NULL, result)); |
| 212 | EXPECT_EQ(3, gJava_MyClass_fooIOO_calls); |
| 213 | result = env_->CallNonvirtualObjectMethod(jobj_, jklass_, jmethod_, 2, NULL, jklass_); |
| 214 | EXPECT_TRUE(env_->IsSameObject(jklass_, result)); |
| 215 | EXPECT_EQ(4, gJava_MyClass_fooIOO_calls); |
| 216 | |
| 217 | result = env_->CallNonvirtualObjectMethod(jobj_, jklass_, jmethod_, 0, jklass_, NULL); |
| 218 | EXPECT_TRUE(env_->IsSameObject(jobj_, result)); |
| 219 | EXPECT_EQ(5, gJava_MyClass_fooIOO_calls); |
| 220 | result = env_->CallNonvirtualObjectMethod(jobj_, jklass_, jmethod_, 1, jklass_, NULL); |
| 221 | EXPECT_TRUE(env_->IsSameObject(jklass_, result)); |
| 222 | EXPECT_EQ(6, gJava_MyClass_fooIOO_calls); |
| 223 | result = env_->CallNonvirtualObjectMethod(jobj_, jklass_, jmethod_, 2, jklass_, NULL); |
| 224 | EXPECT_TRUE(env_->IsSameObject(NULL, result)); |
| 225 | EXPECT_EQ(7, gJava_MyClass_fooIOO_calls); |
| 226 | } |
| 227 | |
Ian Rogers | b033c75 | 2011-07-20 12:22:35 -0700 | [diff] [blame] | 228 | int gJava_MyClass_fooSIOO_calls = 0; |
Elliott Hughes | 330304d | 2011-08-12 14:28:05 -0700 | [diff] [blame] | 229 | jobject Java_MyClass_fooSIOO(JNIEnv* env, jclass klass, jint x, jobject y, |
Ian Rogers | b033c75 | 2011-07-20 12:22:35 -0700 | [diff] [blame] | 230 | jobject z) { |
| 231 | EXPECT_EQ(3u, Thread::Current()->NumShbHandles()); |
| 232 | EXPECT_EQ(Thread::kNative, Thread::Current()->GetState()); |
Ian Rogers | cdd1d2d | 2011-08-18 09:58:17 -0700 | [diff] [blame] | 233 | EXPECT_EQ(Thread::Current()->GetJniEnv(), env); |
| 234 | EXPECT_TRUE(klass != NULL); |
Ian Rogers | a8cd9f4 | 2011-08-19 16:43:41 -0700 | [diff] [blame^] | 235 | EXPECT_TRUE(env->IsInstanceOf(JniCompilerTest::jobj_, klass)); |
Ian Rogers | b033c75 | 2011-07-20 12:22:35 -0700 | [diff] [blame] | 236 | gJava_MyClass_fooSIOO_calls++; |
| 237 | switch (x) { |
| 238 | case 1: |
| 239 | return y; |
| 240 | case 2: |
| 241 | return z; |
| 242 | default: |
| 243 | return klass; |
| 244 | } |
| 245 | } |
| 246 | |
Ian Rogers | cdd1d2d | 2011-08-18 09:58:17 -0700 | [diff] [blame] | 247 | |
| 248 | TEST_F(JniCompilerTest, CompileAndRunStaticIntObjectObjectMethod) { |
| 249 | SetupForTest(true, "fooSIOO", |
| 250 | "(ILjava/lang/Object;Ljava/lang/Object;)Ljava/lang/Object;", |
| 251 | reinterpret_cast<void*>(&Java_MyClass_fooSIOO)); |
| 252 | |
| 253 | EXPECT_EQ(0, gJava_MyClass_fooSIOO_calls); |
| 254 | jobject result = env_->CallStaticObjectMethod(jklass_, jmethod_, 0, NULL, NULL); |
| 255 | EXPECT_TRUE(env_->IsSameObject(jklass_, result)); |
| 256 | EXPECT_EQ(1, gJava_MyClass_fooSIOO_calls); |
| 257 | |
| 258 | result = env_->CallStaticObjectMethod(jklass_, jmethod_, 0, NULL, jobj_); |
| 259 | EXPECT_TRUE(env_->IsSameObject(jklass_, result)); |
| 260 | EXPECT_EQ(2, gJava_MyClass_fooSIOO_calls); |
| 261 | result = env_->CallStaticObjectMethod(jklass_, jmethod_, 1, NULL, jobj_); |
| 262 | EXPECT_TRUE(env_->IsSameObject(NULL, result)); |
| 263 | EXPECT_EQ(3, gJava_MyClass_fooSIOO_calls); |
| 264 | result = env_->CallStaticObjectMethod(jklass_, jmethod_, 2, NULL, jobj_); |
| 265 | EXPECT_TRUE(env_->IsSameObject(jobj_, result)); |
| 266 | EXPECT_EQ(4, gJava_MyClass_fooSIOO_calls); |
| 267 | |
| 268 | result = env_->CallStaticObjectMethod(jklass_, jmethod_, 0, jobj_, NULL); |
| 269 | EXPECT_TRUE(env_->IsSameObject(jklass_, result)); |
| 270 | EXPECT_EQ(5, gJava_MyClass_fooSIOO_calls); |
| 271 | result = env_->CallStaticObjectMethod(jklass_, jmethod_, 1, jobj_, NULL); |
| 272 | EXPECT_TRUE(env_->IsSameObject(jobj_, result)); |
| 273 | EXPECT_EQ(6, gJava_MyClass_fooSIOO_calls); |
| 274 | result = env_->CallStaticObjectMethod(jklass_, jmethod_, 2, jobj_, NULL); |
| 275 | EXPECT_TRUE(env_->IsSameObject(NULL, result)); |
| 276 | EXPECT_EQ(7, gJava_MyClass_fooSIOO_calls); |
| 277 | } |
| 278 | |
Ian Rogers | df20fe0 | 2011-07-20 20:34:16 -0700 | [diff] [blame] | 279 | int gJava_MyClass_fooSSIOO_calls = 0; |
Ian Rogers | cdd1d2d | 2011-08-18 09:58:17 -0700 | [diff] [blame] | 280 | jobject Java_MyClass_fooSSIOO(JNIEnv* env, jclass klass, jint x, jobject y, |
Ian Rogers | df20fe0 | 2011-07-20 20:34:16 -0700 | [diff] [blame] | 281 | jobject z) { |
| 282 | EXPECT_EQ(3u, Thread::Current()->NumShbHandles()); |
| 283 | EXPECT_EQ(Thread::kNative, Thread::Current()->GetState()); |
Ian Rogers | cdd1d2d | 2011-08-18 09:58:17 -0700 | [diff] [blame] | 284 | EXPECT_EQ(Thread::Current()->GetJniEnv(), env); |
| 285 | EXPECT_TRUE(klass != NULL); |
Ian Rogers | a8cd9f4 | 2011-08-19 16:43:41 -0700 | [diff] [blame^] | 286 | EXPECT_TRUE(env->IsInstanceOf(JniCompilerTest::jobj_, klass)); |
Ian Rogers | df20fe0 | 2011-07-20 20:34:16 -0700 | [diff] [blame] | 287 | gJava_MyClass_fooSSIOO_calls++; |
| 288 | switch (x) { |
| 289 | case 1: |
| 290 | return y; |
| 291 | case 2: |
| 292 | return z; |
| 293 | default: |
| 294 | return klass; |
| 295 | } |
| 296 | } |
| 297 | |
Ian Rogers | df20fe0 | 2011-07-20 20:34:16 -0700 | [diff] [blame] | 298 | TEST_F(JniCompilerTest, CompileAndRunStaticSynchronizedIntObjectObjectMethod) { |
Ian Rogers | cdd1d2d | 2011-08-18 09:58:17 -0700 | [diff] [blame] | 299 | SetupForTest(true, "fooSSIOO", |
| 300 | "(ILjava/lang/Object;Ljava/lang/Object;)Ljava/lang/Object;", |
| 301 | reinterpret_cast<void*>(&Java_MyClass_fooSSIOO)); |
Carl Shapiro | 419ec7b | 2011-08-03 14:48:33 -0700 | [diff] [blame] | 302 | |
Ian Rogers | df20fe0 | 2011-07-20 20:34:16 -0700 | [diff] [blame] | 303 | EXPECT_EQ(0, gJava_MyClass_fooSSIOO_calls); |
Ian Rogers | cdd1d2d | 2011-08-18 09:58:17 -0700 | [diff] [blame] | 304 | jobject result = env_->CallStaticObjectMethod(jklass_, jmethod_, 0, NULL, NULL); |
| 305 | EXPECT_TRUE(env_->IsSameObject(jklass_, result)); |
Ian Rogers | df20fe0 | 2011-07-20 20:34:16 -0700 | [diff] [blame] | 306 | EXPECT_EQ(1, gJava_MyClass_fooSSIOO_calls); |
Ian Rogers | cdd1d2d | 2011-08-18 09:58:17 -0700 | [diff] [blame] | 307 | |
| 308 | result = env_->CallStaticObjectMethod(jklass_, jmethod_, 0, NULL, jobj_); |
| 309 | EXPECT_TRUE(env_->IsSameObject(jklass_, result)); |
Ian Rogers | df20fe0 | 2011-07-20 20:34:16 -0700 | [diff] [blame] | 310 | EXPECT_EQ(2, gJava_MyClass_fooSSIOO_calls); |
Ian Rogers | cdd1d2d | 2011-08-18 09:58:17 -0700 | [diff] [blame] | 311 | result = env_->CallStaticObjectMethod(jklass_, jmethod_, 1, NULL, jobj_); |
| 312 | EXPECT_TRUE(env_->IsSameObject(NULL, result)); |
Ian Rogers | df20fe0 | 2011-07-20 20:34:16 -0700 | [diff] [blame] | 313 | EXPECT_EQ(3, gJava_MyClass_fooSSIOO_calls); |
Ian Rogers | cdd1d2d | 2011-08-18 09:58:17 -0700 | [diff] [blame] | 314 | result = env_->CallStaticObjectMethod(jklass_, jmethod_, 2, NULL, jobj_); |
| 315 | EXPECT_TRUE(env_->IsSameObject(jobj_, result)); |
Ian Rogers | df20fe0 | 2011-07-20 20:34:16 -0700 | [diff] [blame] | 316 | EXPECT_EQ(4, gJava_MyClass_fooSSIOO_calls); |
Ian Rogers | cdd1d2d | 2011-08-18 09:58:17 -0700 | [diff] [blame] | 317 | |
| 318 | result = env_->CallStaticObjectMethod(jklass_, jmethod_, 0, jobj_, NULL); |
| 319 | EXPECT_TRUE(env_->IsSameObject(jklass_, result)); |
Ian Rogers | df20fe0 | 2011-07-20 20:34:16 -0700 | [diff] [blame] | 320 | EXPECT_EQ(5, gJava_MyClass_fooSSIOO_calls); |
Ian Rogers | cdd1d2d | 2011-08-18 09:58:17 -0700 | [diff] [blame] | 321 | result = env_->CallStaticObjectMethod(jklass_, jmethod_, 1, jobj_, NULL); |
| 322 | EXPECT_TRUE(env_->IsSameObject(jobj_, result)); |
Ian Rogers | df20fe0 | 2011-07-20 20:34:16 -0700 | [diff] [blame] | 323 | EXPECT_EQ(6, gJava_MyClass_fooSSIOO_calls); |
Ian Rogers | cdd1d2d | 2011-08-18 09:58:17 -0700 | [diff] [blame] | 324 | result = env_->CallStaticObjectMethod(jklass_, jmethod_, 2, jobj_, NULL); |
| 325 | EXPECT_TRUE(env_->IsSameObject(NULL, result)); |
Ian Rogers | df20fe0 | 2011-07-20 20:34:16 -0700 | [diff] [blame] | 326 | EXPECT_EQ(7, gJava_MyClass_fooSSIOO_calls); |
| 327 | } |
| 328 | |
Ian Rogers | 45a76cb | 2011-07-21 22:00:15 -0700 | [diff] [blame] | 329 | int gSuspendCounterHandler_calls; |
| 330 | void SuspendCountHandler(Method** frame) { |
Carl Shapiro | 8860c0e | 2011-08-04 17:36:16 -0700 | [diff] [blame] | 331 | EXPECT_TRUE((*frame)->GetName()->Equals("fooI")); |
Ian Rogers | 45a76cb | 2011-07-21 22:00:15 -0700 | [diff] [blame] | 332 | gSuspendCounterHandler_calls++; |
| 333 | Thread::Current()->DecrementSuspendCount(); |
| 334 | } |
Ian Rogers | cdd1d2d | 2011-08-18 09:58:17 -0700 | [diff] [blame] | 335 | |
Ian Rogers | 0d666d8 | 2011-08-14 16:03:46 -0700 | [diff] [blame] | 336 | TEST_F(JniCompilerTest, SuspendCountAcknowledgement) { |
Ian Rogers | cdd1d2d | 2011-08-18 09:58:17 -0700 | [diff] [blame] | 337 | SetupForTest(false, "fooI", "(I)I", |
| 338 | reinterpret_cast<void*>(&Java_MyClass_fooI)); |
Ian Rogers | 45a76cb | 2011-07-21 22:00:15 -0700 | [diff] [blame] | 339 | Thread::Current()->RegisterSuspendCountEntryPoint(&SuspendCountHandler); |
| 340 | |
Ian Rogers | 45a76cb | 2011-07-21 22:00:15 -0700 | [diff] [blame] | 341 | gJava_MyClass_fooI_calls = 0; |
Ian Rogers | cdd1d2d | 2011-08-18 09:58:17 -0700 | [diff] [blame] | 342 | jint result = env_->CallNonvirtualIntMethod(jobj_, jklass_, jmethod_, 42); |
| 343 | EXPECT_EQ(42, result); |
Ian Rogers | 45a76cb | 2011-07-21 22:00:15 -0700 | [diff] [blame] | 344 | EXPECT_EQ(1, gJava_MyClass_fooI_calls); |
| 345 | EXPECT_EQ(0, gSuspendCounterHandler_calls); |
| 346 | Thread::Current()->IncrementSuspendCount(); |
Ian Rogers | cdd1d2d | 2011-08-18 09:58:17 -0700 | [diff] [blame] | 347 | result = env_->CallNonvirtualIntMethod(jobj_, jklass_, jmethod_, 42); |
| 348 | EXPECT_EQ(42, result); |
Ian Rogers | 45a76cb | 2011-07-21 22:00:15 -0700 | [diff] [blame] | 349 | EXPECT_EQ(2, gJava_MyClass_fooI_calls); |
| 350 | EXPECT_EQ(1, gSuspendCounterHandler_calls); |
Ian Rogers | cdd1d2d | 2011-08-18 09:58:17 -0700 | [diff] [blame] | 351 | result = env_->CallNonvirtualIntMethod(jobj_, jklass_, jmethod_, 42); |
| 352 | EXPECT_EQ(42, result); |
Ian Rogers | 45a76cb | 2011-07-21 22:00:15 -0700 | [diff] [blame] | 353 | EXPECT_EQ(3, gJava_MyClass_fooI_calls); |
| 354 | EXPECT_EQ(1, gSuspendCounterHandler_calls); |
| 355 | } |
| 356 | |
| 357 | int gExceptionHandler_calls; |
| 358 | void ExceptionHandler(Method** frame) { |
Carl Shapiro | 8860c0e | 2011-08-04 17:36:16 -0700 | [diff] [blame] | 359 | EXPECT_TRUE((*frame)->GetName()->Equals("foo")); |
Ian Rogers | 45a76cb | 2011-07-21 22:00:15 -0700 | [diff] [blame] | 360 | gExceptionHandler_calls++; |
| 361 | Thread::Current()->ClearException(); |
| 362 | } |
Ian Rogers | cdd1d2d | 2011-08-18 09:58:17 -0700 | [diff] [blame] | 363 | |
Ian Rogers | 45a76cb | 2011-07-21 22:00:15 -0700 | [diff] [blame] | 364 | TEST_F(JniCompilerTest, ExceptionHandling) { |
Ian Rogers | cdd1d2d | 2011-08-18 09:58:17 -0700 | [diff] [blame] | 365 | SetupForTest(false, "foo", "()V", reinterpret_cast<void*>(&Java_MyClass_foo)); |
Ian Rogers | 45a76cb | 2011-07-21 22:00:15 -0700 | [diff] [blame] | 366 | Thread::Current()->RegisterExceptionEntryPoint(&ExceptionHandler); |
| 367 | |
Ian Rogers | 45a76cb | 2011-07-21 22:00:15 -0700 | [diff] [blame] | 368 | gJava_MyClass_foo_calls = 0; |
Ian Rogers | cdd1d2d | 2011-08-18 09:58:17 -0700 | [diff] [blame] | 369 | env_->CallNonvirtualVoidMethod(jobj_, jklass_, jmethod_); |
Ian Rogers | 45a76cb | 2011-07-21 22:00:15 -0700 | [diff] [blame] | 370 | EXPECT_EQ(1, gJava_MyClass_foo_calls); |
| 371 | EXPECT_EQ(0, gExceptionHandler_calls); |
| 372 | // TODO: create a real exception here |
Ian Rogers | cdd1d2d | 2011-08-18 09:58:17 -0700 | [diff] [blame] | 373 | Thread::Current()->SetException(reinterpret_cast<Object*>(jobj_)); |
| 374 | env_->CallNonvirtualVoidMethod(jobj_, jklass_, jmethod_); |
Ian Rogers | 45a76cb | 2011-07-21 22:00:15 -0700 | [diff] [blame] | 375 | EXPECT_EQ(2, gJava_MyClass_foo_calls); |
| 376 | EXPECT_EQ(1, gExceptionHandler_calls); |
Ian Rogers | cdd1d2d | 2011-08-18 09:58:17 -0700 | [diff] [blame] | 377 | env_->CallNonvirtualVoidMethod(jobj_, jklass_, jmethod_); |
Ian Rogers | 45a76cb | 2011-07-21 22:00:15 -0700 | [diff] [blame] | 378 | EXPECT_EQ(3, gJava_MyClass_foo_calls); |
| 379 | EXPECT_EQ(1, gExceptionHandler_calls); |
| 380 | } |
| 381 | |
Ian Rogers | b033c75 | 2011-07-20 12:22:35 -0700 | [diff] [blame] | 382 | } // namespace art |