one more check

mapleFU · Jul 12, 2024 · 91e564d · 91e564d
1 parent cb6af0d
commit 91e564d
Showing 1 changed file with 3 additions and 0 deletions.
diff --git 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_);
 }