misc/jni: general code cleanup and refactor

Make it align with mpv's coding conventions in general and avoid unecessary Lavu use.
This commit is contained in:
sfan5 2024-02-25 16:43:00 +01:00
parent 1f3758adea
commit 75ae44472b
1 changed files with 126 additions and 162 deletions

View File

@ -20,11 +20,10 @@
*/ */
#include <libavcodec/jni.h> #include <libavcodec/jni.h>
#include <libavutil/mem.h>
#include <libavutil/bprint.h>
#include <stdlib.h> #include <stdlib.h>
#include "jni.h" #include "jni.h"
#include "mpv_talloc.h"
#include "osdep/threads.h" #include "osdep/threads.h"
static JavaVM *java_vm; static JavaVM *java_vm;
@ -46,13 +45,11 @@ static void jni_create_pthread_key(void)
JNIEnv *mp_jni_get_env(struct mp_log *log) JNIEnv *mp_jni_get_env(struct mp_log *log)
{ {
int ret = 0;
JNIEnv *env = NULL; JNIEnv *env = NULL;
mp_mutex_lock(&lock); mp_mutex_lock(&lock);
if (java_vm == NULL) { if (!java_vm)
java_vm = av_jni_get_java_vm(NULL); java_vm = av_jni_get_java_vm(NULL);
}
if (!java_vm) { if (!java_vm) {
mp_err(log, "No Java virtual machine has been registered\n"); mp_err(log, "No Java virtual machine has been registered\n");
@ -61,11 +58,10 @@ JNIEnv *mp_jni_get_env(struct mp_log *log)
mp_exec_once(&once, jni_create_pthread_key); mp_exec_once(&once, jni_create_pthread_key);
if ((env = pthread_getspecific(current_env)) != NULL) { if ((env = pthread_getspecific(current_env)) != NULL)
goto done; goto done;
}
ret = (*java_vm)->GetEnv(java_vm, (void **)&env, JNI_VERSION_1_6); int ret = (*java_vm)->GetEnv(java_vm, (void **)&env, JNI_VERSION_1_6);
switch(ret) { switch(ret) {
case JNI_EDETACHED: case JNI_EDETACHED:
if ((*java_vm)->AttachCurrentThread(java_vm, &env, NULL) != 0) { if ((*java_vm)->AttachCurrentThread(java_vm, &env, NULL) != 0) {
@ -92,39 +88,27 @@ done:
char *mp_jni_jstring_to_utf_chars(JNIEnv *env, jstring string, struct mp_log *log) char *mp_jni_jstring_to_utf_chars(JNIEnv *env, jstring string, struct mp_log *log)
{ {
char *ret = NULL; if (!string)
const char *utf_chars = NULL;
jboolean copy = 0;
if (!string) {
return NULL; return NULL;
}
utf_chars = (*env)->GetStringUTFChars(env, string, &copy); const char *utf_chars = (*env)->GetStringUTFChars(env, string, NULL);
if ((*env)->ExceptionCheck(env)) { if ((*env)->ExceptionCheck(env)) {
(*env)->ExceptionClear(env); (*env)->ExceptionClear(env);
mp_err(log, "String.getStringUTFChars() threw an exception\n"); mp_err(log, "getStringUTFChars() threw an exception\n");
return NULL; return NULL;
} }
ret = av_strdup(utf_chars); char *ret = talloc_strdup(NULL, utf_chars);
(*env)->ReleaseStringUTFChars(env, string, utf_chars); (*env)->ReleaseStringUTFChars(env, string, utf_chars);
if ((*env)->ExceptionCheck(env)) {
(*env)->ExceptionClear(env);
mp_err(log, "String.releaseStringUTFChars() threw an exception\n");
return NULL;
}
return ret; return ret;
} }
jstring mp_jni_utf_chars_to_jstring(JNIEnv *env, const char *utf_chars, struct mp_log *log) jstring mp_jni_utf_chars_to_jstring(JNIEnv *env, const char *utf_chars,
struct mp_log *log)
{ {
jstring ret; jstring ret = (*env)->NewStringUTF(env, utf_chars);
ret = (*env)->NewStringUTF(env, utf_chars);
if ((*env)->ExceptionCheck(env)) { if ((*env)->ExceptionCheck(env)) {
(*env)->ExceptionClear(env); (*env)->ExceptionClear(env);
mp_err(log, "NewStringUTF() threw an exception\n"); mp_err(log, "NewStringUTF() threw an exception\n");
@ -134,24 +118,19 @@ jstring mp_jni_utf_chars_to_jstring(JNIEnv *env, const char *utf_chars, struct m
return ret; return ret;
} }
int mp_jni_exception_get_summary(JNIEnv *env, jthrowable exception, char **error, struct mp_log *log) int mp_jni_exception_get_summary(JNIEnv *env, jthrowable exception,
char **error, struct mp_log *log)
{ {
int ret = 0; int ret = 0;
AVBPrint bp;
char *name = NULL; char *name = NULL;
char *message = NULL; char *message = NULL;
jclass class_class = NULL; jclass class_class = NULL;
jmethodID get_name_id = NULL;
jclass exception_class = NULL; jclass exception_class = NULL;
jmethodID get_message_id = NULL;
jstring string = NULL; jstring string = NULL;
av_bprint_init(&bp, 0, AV_BPRINT_SIZE_AUTOMATIC); *error = NULL;
exception_class = (*env)->GetObjectClass(env, exception); exception_class = (*env)->GetObjectClass(env, exception);
if ((*env)->ExceptionCheck(env)) { if ((*env)->ExceptionCheck(env)) {
@ -169,7 +148,7 @@ int mp_jni_exception_get_summary(JNIEnv *env, jthrowable exception, char **error
goto done; goto done;
} }
get_name_id = (*env)->GetMethodID(env, class_class, "getName", "()Ljava/lang/String;"); jmethodID get_name_id = (*env)->GetMethodID(env, class_class, "getName", "()Ljava/lang/String;");
if ((*env)->ExceptionCheck(env)) { if ((*env)->ExceptionCheck(env)) {
(*env)->ExceptionClear(env); (*env)->ExceptionClear(env);
mp_err(log, "Could not find method Class.getName()\n"); mp_err(log, "Could not find method Class.getName()\n");
@ -191,10 +170,10 @@ int mp_jni_exception_get_summary(JNIEnv *env, jthrowable exception, char **error
string = NULL; string = NULL;
} }
get_message_id = (*env)->GetMethodID(env, exception_class, "getMessage", "()Ljava/lang/String;"); jmethodID get_message_id = (*env)->GetMethodID(env, exception_class, "getMessage", "()Ljava/lang/String;");
if ((*env)->ExceptionCheck(env)) { if ((*env)->ExceptionCheck(env)) {
(*env)->ExceptionClear(env); (*env)->ExceptionClear(env);
mp_err(log, "Could not find method java/lang/Throwable.getMessage()\n"); mp_err(log, "Could not find method Throwable.getMessage()\n");
ret = -1; ret = -1;
goto done; goto done;
} }
@ -214,21 +193,20 @@ int mp_jni_exception_get_summary(JNIEnv *env, jthrowable exception, char **error
} }
if (name && message) { if (name && message) {
av_bprintf(&bp, "%s: %s", name, message); *error = talloc_asprintf(NULL, "%s: %s", name, message);
} else if (name && !message) { } else if (name && !message) {
av_bprintf(&bp, "%s occurred", name); *error = talloc_asprintf(NULL, "%s occurred", name);
} else if (!name && message) { } else if (!name && message) {
av_bprintf(&bp, "Exception: %s", message); *error = talloc_asprintf(NULL, "Exception: %s", message);
} else { } else {
mp_warn(log, "Could not retrieve exception name and message\n"); mp_warn(log, "Could not retrieve exception name and message\n");
av_bprintf(&bp, "Exception occurred"); *error = talloc_strdup(NULL, "Exception occurred");
} }
ret = av_bprint_finalize(&bp, error);
done: done:
av_free(name); talloc_free(name);
av_free(message); talloc_free(message);
if (class_class) { if (class_class) {
(*env)->DeleteLocalRef(env, class_class); (*env)->DeleteLocalRef(env, class_class);
@ -247,126 +225,117 @@ done:
int mp_jni_exception_check(JNIEnv *env, int logging, struct mp_log *log) int mp_jni_exception_check(JNIEnv *env, int logging, struct mp_log *log)
{ {
int ret; if (!(*env)->ExceptionCheck(env))
jthrowable exception;
char *message = NULL;
if (!(*(env))->ExceptionCheck((env))) {
return 0; return 0;
}
if (!logging) { if (!logging) {
(*(env))->ExceptionClear((env)); (*env)->ExceptionClear(env);
return -1; return -1;
} }
exception = (*env)->ExceptionOccurred(env); jthrowable exception = (*env)->ExceptionOccurred(env);
(*(env))->ExceptionClear((env)); (*env)->ExceptionClear(env);
if ((ret = mp_jni_exception_get_summary(env, exception, &message, log)) < 0) {
(*env)->DeleteLocalRef(env, exception);
return ret;
}
char *message = NULL;
int ret = mp_jni_exception_get_summary(env, exception, &message, log);
(*env)->DeleteLocalRef(env, exception); (*env)->DeleteLocalRef(env, exception);
if (ret < 0)
return ret;
mp_err(log, "%s\n", message); mp_err(log, "%s\n", message);
av_free(message); talloc_free(message);
return -1; return -1;
} }
int mp_jni_init_jfields(JNIEnv *env, void *jfields, const struct MPJniField *jfields_mapping, int global, struct mp_log *log) #define CHECK_EXC_MANDATORY() do { \
if ((ret = mp_jni_exception_check(env, mandatory, log)) < 0 && \
mandatory) { \
goto done; \
} \
} while (0)
int mp_jni_init_jfields(JNIEnv *env, void *jfields,
const struct MPJniField *jfields_mapping,
int global, struct mp_log *log)
{ {
int i, ret = 0; int ret = 0;
jclass last_clazz = NULL; jclass last_clazz = NULL;
for (i = 0; jfields_mapping[i].name; i++) { for (int i = 0; jfields_mapping[i].name; i++) {
int mandatory = jfields_mapping[i].mandatory; bool mandatory = !!jfields_mapping[i].mandatory;
enum MPJniFieldType type = jfields_mapping[i].type; enum MPJniFieldType type = jfields_mapping[i].type;
if (type == MP_JNI_CLASS) { void *jfield = (uint8_t*)jfields + jfields_mapping[i].offset;
jclass clazz;
if (type == MP_JNI_CLASS) {
last_clazz = NULL; last_clazz = NULL;
clazz = (*env)->FindClass(env, jfields_mapping[i].name); jclass clazz = (*env)->FindClass(env, jfields_mapping[i].name);
if ((ret = mp_jni_exception_check(env, mandatory, log)) < 0 && mandatory) { CHECK_EXC_MANDATORY();
goto done;
}
last_clazz = *(jclass*)((uint8_t*)jfields + jfields_mapping[i].offset) = last_clazz = *(jclass*)jfield =
global ? (*env)->NewGlobalRef(env, clazz) : clazz; global ? (*env)->NewGlobalRef(env, clazz) : clazz;
if (global) { if (global)
(*env)->DeleteLocalRef(env, clazz); (*env)->DeleteLocalRef(env, clazz);
}
} else { continue;
if (!last_clazz) {
ret = -1;
break;
}
switch(type) {
case MP_JNI_FIELD: {
jfieldID field_id = (*env)->GetFieldID(env, last_clazz, jfields_mapping[i].method, jfields_mapping[i].signature);
if ((ret = mp_jni_exception_check(env, mandatory, log)) < 0 && mandatory) {
goto done;
}
*(jfieldID*)((uint8_t*)jfields + jfields_mapping[i].offset) = field_id;
break;
}
case MP_JNI_STATIC_FIELD_AS_INT:
case MP_JNI_STATIC_FIELD: {
jfieldID field_id = (*env)->GetStaticFieldID(env, last_clazz, jfields_mapping[i].method, jfields_mapping[i].signature);
if ((ret = mp_jni_exception_check(env, mandatory, log)) < 0 && mandatory) {
goto done;
}
if (type == MP_JNI_STATIC_FIELD_AS_INT) {
if (field_id) {
jint value = (*env)->GetStaticIntField(env, last_clazz, field_id);
if ((ret = mp_jni_exception_check(env, mandatory, log)) < 0 && mandatory) {
goto done;
}
*(jint*)((uint8_t*)jfields + jfields_mapping[i].offset) = value;
}
} else {
*(jfieldID*)((uint8_t*)jfields + jfields_mapping[i].offset) = field_id;
}
break;
}
case MP_JNI_METHOD: {
jmethodID method_id = (*env)->GetMethodID(env, last_clazz, jfields_mapping[i].method, jfields_mapping[i].signature);
if ((ret = mp_jni_exception_check(env, mandatory, log)) < 0 && mandatory) {
goto done;
}
*(jmethodID*)((uint8_t*)jfields + jfields_mapping[i].offset) = method_id;
break;
}
case MP_JNI_STATIC_METHOD: {
jmethodID method_id = (*env)->GetStaticMethodID(env, last_clazz, jfields_mapping[i].method, jfields_mapping[i].signature);
if ((ret = mp_jni_exception_check(env, mandatory, log)) < 0 && mandatory) {
goto done;
}
*(jmethodID*)((uint8_t*)jfields + jfields_mapping[i].offset) = method_id;
break;
}
default:
mp_err(log, "Unknown JNI field type\n");
ret = -1;
goto done;
}
ret = 0;
} }
if (!last_clazz) {
ret = -1;
break;
}
switch (type) {
case MP_JNI_FIELD: {
jfieldID field_id = (*env)->GetFieldID(env, last_clazz,
jfields_mapping[i].method, jfields_mapping[i].signature);
CHECK_EXC_MANDATORY();
*(jfieldID*)jfield = field_id;
break;
}
case MP_JNI_STATIC_FIELD_AS_INT:
case MP_JNI_STATIC_FIELD: {
jfieldID field_id = (*env)->GetStaticFieldID(env, last_clazz,
jfields_mapping[i].method, jfields_mapping[i].signature);
CHECK_EXC_MANDATORY();
if (type == MP_JNI_STATIC_FIELD_AS_INT) {
if (field_id) {
jint value = (*env)->GetStaticIntField(env, last_clazz, field_id);
CHECK_EXC_MANDATORY();
*(jint*)jfield = value;
}
} else {
*(jfieldID*)jfield = field_id;
}
break;
}
case MP_JNI_METHOD: {
jmethodID method_id = (*env)->GetMethodID(env, last_clazz,
jfields_mapping[i].method, jfields_mapping[i].signature);
CHECK_EXC_MANDATORY();
*(jmethodID*)jfield = method_id;
break;
}
case MP_JNI_STATIC_METHOD: {
jmethodID method_id = (*env)->GetStaticMethodID(env, last_clazz,
jfields_mapping[i].method, jfields_mapping[i].signature);
CHECK_EXC_MANDATORY();
*(jmethodID*)jfield = method_id;
break;
}
default:
mp_err(log, "Unknown JNI field type\n");
ret = -1;
goto done;
}
ret = 0;
} }
done: done:
@ -378,16 +347,20 @@ done:
return ret; return ret;
} }
int mp_jni_reset_jfields(JNIEnv *env, void *jfields, const struct MPJniField *jfields_mapping, int global, struct mp_log *log) #undef CHECK_EXC_MANDATORY
{
int i;
for (i = 0; jfields_mapping[i].name; i++) { int mp_jni_reset_jfields(JNIEnv *env, void *jfields,
const struct MPJniField *jfields_mapping,
int global, struct mp_log *log)
{
for (int i = 0; jfields_mapping[i].name; i++) {
enum MPJniFieldType type = jfields_mapping[i].type; enum MPJniFieldType type = jfields_mapping[i].type;
switch(type) { void *jfield = (uint8_t*)jfields + jfields_mapping[i].offset;
switch (type) {
case MP_JNI_CLASS: { case MP_JNI_CLASS: {
jclass clazz = *(jclass*)((uint8_t*)jfields + jfields_mapping[i].offset); jclass clazz = *(jclass*)jfield;
if (!clazz) if (!clazz)
continue; continue;
@ -397,29 +370,20 @@ int mp_jni_reset_jfields(JNIEnv *env, void *jfields, const struct MPJniField *jf
(*env)->DeleteLocalRef(env, clazz); (*env)->DeleteLocalRef(env, clazz);
} }
*(jclass*)((uint8_t*)jfields + jfields_mapping[i].offset) = NULL; *(jclass*)jfield = NULL;
break; break;
} }
case MP_JNI_FIELD: { case MP_JNI_FIELD:
*(jfieldID*)((uint8_t*)jfields + jfields_mapping[i].offset) = NULL; case MP_JNI_STATIC_FIELD:
*(jfieldID*)jfield = NULL;
break; break;
} case MP_JNI_STATIC_FIELD_AS_INT:
case MP_JNI_STATIC_FIELD: { *(jint*)jfield = 0;
*(jfieldID*)((uint8_t*)jfields + jfields_mapping[i].offset) = NULL;
break; break;
} case MP_JNI_METHOD:
case MP_JNI_STATIC_FIELD_AS_INT: { case MP_JNI_STATIC_METHOD:
*(jint*)((uint8_t*)jfields + jfields_mapping[i].offset) = 0; *(jmethodID*)jfield = NULL;
break; break;
}
case MP_JNI_METHOD: {
*(jmethodID*)((uint8_t*)jfields + jfields_mapping[i].offset) = NULL;
break;
}
case MP_JNI_STATIC_METHOD: {
*(jmethodID*)((uint8_t*)jfields + jfields_mapping[i].offset) = NULL;
break;
}
default: default:
mp_err(log, "Unknown JNI field type\n"); mp_err(log, "Unknown JNI field type\n");
} }