From 91e564db9f20c61bc7e9f319c97536315484d06b Mon Sep 17 00:00:00 2001
From: mwish <maplewish117@gmail.com>
Date: Fri, 12 Jul 2024 11:02:55 +0800
Subject: [PATCH] one more check

---
 cpp/src/parquet/encryption/internal_file_decryptor.cc | 3 +++
 1 file changed, 3 insertions(+)

diff --git a/cpp/src/parquet/encryption/internal_file_decryptor.cc b/cpp/src/parquet/encryption/internal_file_decryptor.cc
index 26b8a9c2064d7..9252368bd7d4b 100644
--- a/cpp/src/parquet/encryption/internal_file_decryptor.cc
+++ b/cpp/src/parquet/encryption/internal_file_decryptor.cc
@@ -205,6 +205,9 @@ std::shared_ptr<Decryptor> InternalFileDecryptor::GetColumnDecryptor(
   std::lock_guard<std::mutex> lock(mutex_);
   auto aes_decryptor =
       encryption::AesDecryptor::Make(algorithm_, key_len, metadata, &all_decryptors_);
+  if (ARROW_PREDICT_FALSE(aes_decryptor == nullptr)) {
+    throw ParquetException("Failed to create AES decryptor");
+  }
   return std::make_shared<Decryptor>(std::move(aes_decryptor), column_key, file_aad_, aad,
                                      pool_);
 }