avcodec/h264_slice: Fix overflow in slice offset
[ffmpeg.git] / libavcodec / ffjni.c
index 8333050..f5b581f 100644 (file)
@@ -20,7 +20,6 @@
  * Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA
  */
 
-#include <dlfcn.h>
 #include <jni.h>
 #include <pthread.h>
 #include <stdlib.h>
 #include "jni.h"
 #include "ffjni.h"
 
-static JavaVM *java_vm = NULL;
+static JavaVM *java_vm;
+static pthread_key_t current_env;
+static pthread_once_t once = PTHREAD_ONCE_INIT;
 static pthread_mutex_t lock = PTHREAD_MUTEX_INITIALIZER;
 
-/**
- * Check if JniInvocation has been initialized. Only available on
- * Android >= 4.4.
- *
- * @param log_ctx context used for logging, can be NULL
- * @return 0 on success, < 0 otherwise
- */
-static int check_jni_invocation(void *log_ctx)
+static void jni_detach_env(void *data)
 {
-    int ret = AVERROR_EXTERNAL;
-    void *handle = NULL;
-    void **jni_invocation = NULL;
-
-    handle = dlopen(NULL, RTLD_LOCAL);
-    if (!handle) {
-        goto done;
-    }
-
-    jni_invocation = (void **)dlsym(handle, "_ZN13JniInvocation15jni_invocation_E");
-    if (!jni_invocation) {
-        av_log(log_ctx, AV_LOG_ERROR, "Could not find JniInvocation::jni_invocation_ symbol\n");
-        goto done;
+    if (java_vm) {
+        (*java_vm)->DetachCurrentThread(java_vm);
     }
-
-    ret = !(jni_invocation != NULL && *jni_invocation != NULL);
-
-done:
-    if (handle) {
-        dlclose(handle);
-    }
-
-    return ret;
 }
 
-/**
- * Return created Java virtual machine using private JNI_GetCreatedJavaVMs
- * function from the specified library name.
- *
- * @param name library name used for symbol lookups, can be NULL
- * @param log_ctx context used for logging, can be NULL
- * @return the current Java virtual machine in use
- */
-static JavaVM *get_java_vm(const char *name, void *log_ctx)
+static void jni_create_pthread_key(void)
 {
-    JavaVM *vm = NULL;
-    jsize nb_vm = 0;
-
-    void *handle = NULL;
-    jint (*get_created_java_vms) (JavaVM ** vmBuf, jsize bufLen, jsize *nVMs) = NULL;
-
-    handle = dlopen(name, RTLD_LOCAL);
-    if (!handle) {
-        return NULL;
-    }
-
-    get_created_java_vms = (jint (*)(JavaVM **, jsize, jsize *)) dlsym(handle, "JNI_GetCreatedJavaVMs");
-    if (!get_created_java_vms) {
-        av_log(log_ctx, AV_LOG_ERROR, "Could not find JNI_GetCreatedJavaVMs symbol in library '%s'\n", name);
-        goto done;
-    }
-
-    if (get_created_java_vms(&vm, 1, &nb_vm) != JNI_OK) {
-        av_log(log_ctx, AV_LOG_ERROR, "Could not get created Java virtual machines\n");
-        goto done;
-    }
-
-done:
-    if (handle) {
-        dlclose(handle);
-    }
-
-    return vm;
+    pthread_key_create(&current_env, jni_detach_env);
 }
 
-JNIEnv *ff_jni_attach_env(int *attached, void *log_ctx)
+JNIEnv *ff_jni_get_env(void *log_ctx)
 {
     int ret = 0;
     JNIEnv *env = NULL;
 
-    *attached = 0;
-
     pthread_mutex_lock(&lock);
-    if (java_vm == NULL && (java_vm = av_jni_get_java_vm(log_ctx)) == NULL) {
-
-        av_log(log_ctx, AV_LOG_INFO, "Retrieving current Java virtual machine using Android JniInvocation wrapper\n");
-        if (check_jni_invocation(log_ctx) == 0) {
-            if ((java_vm = get_java_vm(NULL, log_ctx)) != NULL ||
-                (java_vm = get_java_vm("libdvm.so", log_ctx)) != NULL ||
-                (java_vm = get_java_vm("libart.so", log_ctx)) != NULL) {
-                av_log(log_ctx, AV_LOG_INFO, "Found Java virtual machine using Android JniInvocation wrapper\n");
-            }
-        }
+    if (java_vm == NULL) {
+        java_vm = av_jni_get_java_vm(log_ctx);
     }
-    pthread_mutex_unlock(&lock);
 
     if (!java_vm) {
-        av_log(log_ctx, AV_LOG_ERROR, "Could not retrieve a Java virtual machine\n");
-        return NULL;
+        av_log(log_ctx, AV_LOG_ERROR, "No Java virtual machine has been registered\n");
+        goto done;
+    }
+
+    pthread_once(&once, jni_create_pthread_key);
+
+    if ((env = pthread_getspecific(current_env)) != NULL) {
+        goto done;
     }
 
     ret = (*java_vm)->GetEnv(java_vm, (void **)&env, JNI_VERSION_1_6);
@@ -142,7 +76,7 @@ JNIEnv *ff_jni_attach_env(int *attached, void *log_ctx)
             av_log(log_ctx, AV_LOG_ERROR, "Failed to attach the JNI environment to the current thread\n");
             env = NULL;
         } else {
-            *attached = 1;
+            pthread_setspecific(current_env, env);
         }
         break;
     case JNI_OK:
@@ -151,23 +85,15 @@ JNIEnv *ff_jni_attach_env(int *attached, void *log_ctx)
         av_log(log_ctx, AV_LOG_ERROR, "The specified JNI version is not supported\n");
         break;
     default:
-        av_log(log_ctx, AV_LOG_ERROR, "Failed to get the JNI environment attached to this thread");
+        av_log(log_ctx, AV_LOG_ERROR, "Failed to get the JNI environment attached to this thread\n");
         break;
     }
 
+done:
+    pthread_mutex_unlock(&lock);
     return env;
 }
 
-int ff_jni_detach_env(void *log_ctx)
-{
-    if (java_vm == NULL) {
-        av_log(log_ctx, AV_LOG_ERROR, "No Java virtual machine has been registered\n");
-        return AVERROR(EINVAL);
-    }
-
-    return (*java_vm)->DetachCurrentThread(java_vm);
-}
-
 char *ff_jni_jstring_to_utf_chars(JNIEnv *env, jstring string, void *log_ctx)
 {
     char *ret = NULL;
@@ -227,7 +153,7 @@ int ff_jni_exception_get_summary(JNIEnv *env, jthrowable exception, char **error
     jclass exception_class = NULL;
     jmethodID get_message_id = NULL;
 
-    jstring string;
+    jstring string = NULL;
 
     av_bprint_init(&bp, 0, AV_BPRINT_SIZE_AUTOMATIC);
 
@@ -298,7 +224,7 @@ int ff_jni_exception_get_summary(JNIEnv *env, jthrowable exception, char **error
     } else if (!name && message) {
         av_bprintf(&bp, "Exception: %s", message);
     } else {
-        av_log(log_ctx, AV_LOG_WARNING, "Could not retreive exception name and message\n");
+        av_log(log_ctx, AV_LOG_WARNING, "Could not retrieve exception name and message\n");
         av_bprintf(&bp, "Exception occurred");
     }
 
@@ -377,6 +303,11 @@ int ff_jni_init_jfields(JNIEnv *env, void *jfields, const struct FFJniField *jfi
 
             last_clazz = *(jclass*)((uint8_t*)jfields + jfields_mapping[i].offset) =
                     global ? (*env)->NewGlobalRef(env, clazz) : clazz;
+
+            if (global) {
+                (*env)->DeleteLocalRef(env, clazz);
+            }
+
         } else {
 
             if (!last_clazz) {
@@ -426,6 +357,8 @@ int ff_jni_init_jfields(JNIEnv *env, void *jfields, const struct FFJniField *jfi
                 ret = AVERROR(EINVAL);
                 goto done;
             }
+
+            ret = 0;
         }
     }