1*9190c2a8SAndroid Build Coastguard Worker /*
2*9190c2a8SAndroid Build Coastguard Worker * Copyright (C) 2019 The Android Open Source Project
3*9190c2a8SAndroid Build Coastguard Worker *
4*9190c2a8SAndroid Build Coastguard Worker * Licensed under the Apache License, Version 2.0 (the "License");
5*9190c2a8SAndroid Build Coastguard Worker * you may not use this file except in compliance with the License.
6*9190c2a8SAndroid Build Coastguard Worker * You may obtain a copy of the License at
7*9190c2a8SAndroid Build Coastguard Worker *
8*9190c2a8SAndroid Build Coastguard Worker * http://www.apache.org/licenses/LICENSE-2.0
9*9190c2a8SAndroid Build Coastguard Worker *
10*9190c2a8SAndroid Build Coastguard Worker * Unless required by applicable law or agreed to in writing, software
11*9190c2a8SAndroid Build Coastguard Worker * distributed under the License is distributed on an "AS IS" BASIS,
12*9190c2a8SAndroid Build Coastguard Worker * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
13*9190c2a8SAndroid Build Coastguard Worker * See the License for the specific language governing permissions and
14*9190c2a8SAndroid Build Coastguard Worker * limitations under the License.
15*9190c2a8SAndroid Build Coastguard Worker */
16*9190c2a8SAndroid Build Coastguard Worker #pragma once
17*9190c2a8SAndroid Build Coastguard Worker
18*9190c2a8SAndroid Build Coastguard Worker #include <jni.h>
19*9190c2a8SAndroid Build Coastguard Worker #include <log/log.h>
20*9190c2a8SAndroid Build Coastguard Worker
21*9190c2a8SAndroid Build Coastguard Worker // JNI helpers.
FindClassOrDie(JNIEnv * env,const char * class_name)22*9190c2a8SAndroid Build Coastguard Worker static inline jclass FindClassOrDie(JNIEnv* env, const char* class_name) {
23*9190c2a8SAndroid Build Coastguard Worker jclass clazz = env->FindClass(class_name);
24*9190c2a8SAndroid Build Coastguard Worker LOG_ALWAYS_FATAL_IF(clazz == NULL, "Unable to find class %s", class_name);
25*9190c2a8SAndroid Build Coastguard Worker return clazz;
26*9190c2a8SAndroid Build Coastguard Worker }
27*9190c2a8SAndroid Build Coastguard Worker
GetMethodIDOrDie(JNIEnv * env,jclass clazz,const char * method_name,const char * method_signature)28*9190c2a8SAndroid Build Coastguard Worker static inline jmethodID GetMethodIDOrDie(JNIEnv* env, jclass clazz, const char* method_name,
29*9190c2a8SAndroid Build Coastguard Worker const char* method_signature) {
30*9190c2a8SAndroid Build Coastguard Worker jmethodID res = env->GetMethodID(clazz, method_name, method_signature);
31*9190c2a8SAndroid Build Coastguard Worker LOG_ALWAYS_FATAL_IF(res == NULL, "Unable to find method %s", method_name);
32*9190c2a8SAndroid Build Coastguard Worker return res;
33*9190c2a8SAndroid Build Coastguard Worker }
34*9190c2a8SAndroid Build Coastguard Worker
GetStaticMethodIDOrDie(JNIEnv * env,jclass clazz,const char * method_name,const char * method_signature)35*9190c2a8SAndroid Build Coastguard Worker static inline jmethodID GetStaticMethodIDOrDie(JNIEnv* env, jclass clazz, const char* method_name,
36*9190c2a8SAndroid Build Coastguard Worker const char* method_signature) {
37*9190c2a8SAndroid Build Coastguard Worker jmethodID res = env->GetStaticMethodID(clazz, method_name, method_signature);
38*9190c2a8SAndroid Build Coastguard Worker LOG_ALWAYS_FATAL_IF(res == NULL, "Unable to find static method %s", method_name);
39*9190c2a8SAndroid Build Coastguard Worker return res;
40*9190c2a8SAndroid Build Coastguard Worker }
41*9190c2a8SAndroid Build Coastguard Worker
GetFieldIDOrDie(JNIEnv * env,jclass clazz,const char * field_name,const char * field_signature)42*9190c2a8SAndroid Build Coastguard Worker static inline jfieldID GetFieldIDOrDie(JNIEnv* env, jclass clazz, const char* field_name,
43*9190c2a8SAndroid Build Coastguard Worker const char* field_signature) {
44*9190c2a8SAndroid Build Coastguard Worker jfieldID res = env->GetFieldID(clazz, field_name, field_signature);
45*9190c2a8SAndroid Build Coastguard Worker LOG_ALWAYS_FATAL_IF(res == NULL, "Unable to find static field %s", field_name);
46*9190c2a8SAndroid Build Coastguard Worker return res;
47*9190c2a8SAndroid Build Coastguard Worker }
48*9190c2a8SAndroid Build Coastguard Worker
GetStaticFieldIDOrDie(JNIEnv * env,jclass clazz,const char * field_name,const char * field_signature)49*9190c2a8SAndroid Build Coastguard Worker static inline jfieldID GetStaticFieldIDOrDie(JNIEnv* env, jclass clazz, const char* field_name,
50*9190c2a8SAndroid Build Coastguard Worker const char* field_signature) {
51*9190c2a8SAndroid Build Coastguard Worker jfieldID res = env->GetStaticFieldID(clazz, field_name, field_signature);
52*9190c2a8SAndroid Build Coastguard Worker LOG_ALWAYS_FATAL_IF(res == NULL, "Unable to find static field %s", field_name);
53*9190c2a8SAndroid Build Coastguard Worker return res;
54*9190c2a8SAndroid Build Coastguard Worker }
55*9190c2a8SAndroid Build Coastguard Worker
GetStaticIntFieldValueOrDie(JNIEnv * env,jclass clazz,const char * fieldName)56*9190c2a8SAndroid Build Coastguard Worker static inline jint GetStaticIntFieldValueOrDie(JNIEnv* env, jclass clazz, const char* fieldName) {
57*9190c2a8SAndroid Build Coastguard Worker jfieldID res = env->GetStaticFieldID(clazz, fieldName, "I");
58*9190c2a8SAndroid Build Coastguard Worker LOG_ALWAYS_FATAL_IF(res == NULL, "Unable to find static field %s", fieldName);
59*9190c2a8SAndroid Build Coastguard Worker return env->GetStaticIntField(clazz, res);
60*9190c2a8SAndroid Build Coastguard Worker }
61*9190c2a8SAndroid Build Coastguard Worker
GetStringField(JNIEnv * env,jobject obj,jfieldID field)62*9190c2a8SAndroid Build Coastguard Worker static inline jstring GetStringField(JNIEnv* env, jobject obj, jfieldID field) {
63*9190c2a8SAndroid Build Coastguard Worker return reinterpret_cast<jstring>(env->GetObjectField(obj, field));
64*9190c2a8SAndroid Build Coastguard Worker }
65*9190c2a8SAndroid Build Coastguard Worker
GetByteArrayField(JNIEnv * env,jobject obj,jfieldID field)66*9190c2a8SAndroid Build Coastguard Worker static inline jbyteArray GetByteArrayField(JNIEnv* env, jobject obj, jfieldID field) {
67*9190c2a8SAndroid Build Coastguard Worker return reinterpret_cast<jbyteArray>(env->GetObjectField(obj, field));
68*9190c2a8SAndroid Build Coastguard Worker }
69*9190c2a8SAndroid Build Coastguard Worker
GetJNIEnvironment(JavaVM * vm)70*9190c2a8SAndroid Build Coastguard Worker static inline JNIEnv* GetJNIEnvironment(JavaVM* vm) {
71*9190c2a8SAndroid Build Coastguard Worker JNIEnv* env;
72*9190c2a8SAndroid Build Coastguard Worker if (vm->GetEnv(reinterpret_cast<void**>(&env), JNI_VERSION_1_6) != JNI_OK) {
73*9190c2a8SAndroid Build Coastguard Worker return 0;
74*9190c2a8SAndroid Build Coastguard Worker }
75*9190c2a8SAndroid Build Coastguard Worker return env;
76*9190c2a8SAndroid Build Coastguard Worker }
77*9190c2a8SAndroid Build Coastguard Worker
GetOrAttachJNIEnvironment(JavaVM * jvm)78*9190c2a8SAndroid Build Coastguard Worker static inline JNIEnv* GetOrAttachJNIEnvironment(JavaVM* jvm) {
79*9190c2a8SAndroid Build Coastguard Worker JNIEnv* env = GetJNIEnvironment(jvm);
80*9190c2a8SAndroid Build Coastguard Worker if (!env) {
81*9190c2a8SAndroid Build Coastguard Worker int result = jvm->AttachCurrentThread(&env, nullptr);
82*9190c2a8SAndroid Build Coastguard Worker CHECK_EQ(result, JNI_OK) << "thread attach failed";
83*9190c2a8SAndroid Build Coastguard Worker struct VmDetacher {
84*9190c2a8SAndroid Build Coastguard Worker VmDetacher(JavaVM* vm) : mVm(vm) {}
85*9190c2a8SAndroid Build Coastguard Worker ~VmDetacher() { mVm->DetachCurrentThread(); }
86*9190c2a8SAndroid Build Coastguard Worker
87*9190c2a8SAndroid Build Coastguard Worker private:
88*9190c2a8SAndroid Build Coastguard Worker JavaVM* const mVm;
89*9190c2a8SAndroid Build Coastguard Worker };
90*9190c2a8SAndroid Build Coastguard Worker static thread_local VmDetacher detacher(jvm);
91*9190c2a8SAndroid Build Coastguard Worker }
92*9190c2a8SAndroid Build Coastguard Worker return env;
93*9190c2a8SAndroid Build Coastguard Worker }
94