Ensure sufficient space for local references

Author: adamretter

java/rocksjni/env.cc

@@ -137,6 +137,12 @@ jobjectArray Java_org_rocksdb_Env_getThreadList(JNIEnv* env, jclass,
 
   // object[]
   const jsize len = static_cast<jsize>(thread_status.size());
+  if (env->EnsureLocalCapacity((len * 2) + 1) != JNI_OK) {
+    // out of memory
+    env->ExceptionClear();
+    ROCKSDB_NAMESPACE::RocksDBExceptionJni::ThrowNew(env, "Insufficient JNI local references for thread list");
+    return nullptr;
+  }
   jobjectArray jthread_status = env->NewObjectArray(
       len, ROCKSDB_NAMESPACE::ThreadStatusJni::getJClass(env), nullptr);
   if (jthread_status == nullptr) {

java/rocksjni/transaction.cc

@@ -673,7 +673,7 @@ void txn_write_kv_parts_helper(JNIEnv* env,
   // in order to avoid the local reference limit.
   // The user needs to be a aware that there is a limit to the number of parts
   // which can be gathered.
-  if (env->EnsureLocalCapacity(jkey_parts_len + jvalue_parts_len) != 0) {
+  if (env->EnsureLocalCapacity(jkey_parts_len + jvalue_parts_len) != JNI_OK) {
     // no space for all the jobjects we store up
     env->ExceptionClear();
     ROCKSDB_NAMESPACE::RocksDBExceptionJni::ThrowNew(