[GLUTEN-8685][VL] Add null check to avoid core dump when rss push par…

…tition data size is large (#8686)
apache · Feb 10, 2025 · 4b741cf · 4b741cf
1 parent c90f170
commit 4b741cf
Showing 1 changed file with 4 additions and 0 deletions.
diff --git a/cpp/core/jni/JniCommon.h b/cpp/core/jni/JniCommon.h
@@ -498,6 +498,10 @@ class JavaRssClient : public RssClient {
       env->ReleaseByteArrayElements(array_, byteArray, JNI_ABORT);
       env->DeleteGlobalRef(array_);
       array_ = env->NewByteArray(size);
+      if (array_ == nullptr) {
+        LOG(WARNING) << "Failed to allocate new byte array size: " << size;
+        throw gluten::GlutenException("Failed to allocate new byte array");
+      }
       array_ = static_cast<jbyteArray>(env->NewGlobalRef(array_));
     }
     env->SetByteArrayRegion(array_, 0, size, (jbyte*)bytes);