From 4b741cf61a74ec71e0ed3a986bf659ed9469cfad Mon Sep 17 00:00:00 2001
From: Terry Wang <zjuwangg@foxmail.com>
Date: Mon, 10 Feb 2025 17:22:45 +0800
Subject: [PATCH] [GLUTEN-8685][VL] Add null check to avoid core dump when rss
 push partition data size is large (#8686)

---
 cpp/core/jni/JniCommon.h | 4 ++++
 1 file changed, 4 insertions(+)

diff --git a/cpp/core/jni/JniCommon.h b/cpp/core/jni/JniCommon.h
index 8f40398a4132..246658834d48 100644
--- 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);