From 54a454714703263d2df73b1555eaf3dc823206f2 Mon Sep 17 00:00:00 2001
From: qingsuijiu <642782632@qq.com>
Date: Tue, 17 Dec 2024 11:15:19 +0000
Subject: [PATCH] ExprCalcPartition performs defensive checks on the types of
 partition expressions and internal partition information.

---
 .../engine/expr/ob_expr_calc_partition_id.cpp | 33 +++++++++++++++++--
 1 file changed, 30 insertions(+), 3 deletions(-)

diff --git a/src/sql/engine/expr/ob_expr_calc_partition_id.cpp b/src/sql/engine/expr/ob_expr_calc_partition_id.cpp
index 2a384cc30..a16a37aa8 100644
--- a/src/sql/engine/expr/ob_expr_calc_partition_id.cpp
+++ b/src/sql/engine/expr/ob_expr_calc_partition_id.cpp
@@ -244,8 +244,18 @@ int ObExprCalcPartitionBase::cg_expr(ObExprCGCtx &expr_cg_ctx,
           if (GET_MIN_CLUSTER_VERSION() < CLUSTER_VERSION_4_3_5_0) {
             fallback = true;
           } else {
-            rt_expr.eval_vector_func_ =
+            ObPartition * const* part_array = table_schema->get_part_array();
+            if (OB_ISNULL(part_array) || OB_ISNULL(part_array[0]) ||
+                OB_ISNULL(part_array[0]->get_high_bound_val().get_obj_ptr())) {
+                ret = OB_INVALID_ARGUMENT;
+                LOG_WARN("partition_array is null", K(ret));
+            } else if (part_expr->obj_meta_.get_type() !=
+                part_array[0]->get_high_bound_val().get_obj_ptr()->get_meta().get_type()) {
+              fallback = true;
+            } else {
+              rt_expr.eval_vector_func_ =
                 ObExprCalcPartitionBase::fast_calc_partition_level_one_vector;
+            }
           }
         } else if (table_schema->is_list_part()) {
           if (GET_MIN_CLUSTER_VERSION() < CLUSTER_VERSION_4_3_5_0) {
@@ -256,8 +266,25 @@ int ObExprCalcPartitionBase::cg_expr(ObExprCGCtx &expr_cg_ctx,
             if (ob_is_json(part_expr_type) || ob_is_urowid(part_expr_type)) {
               fallback = true;
             } else {
-              rt_expr.eval_vector_func_ =
-                    ObExprCalcPartitionBase::fast_calc_partition_level_one_vector;
+              ObPartition * const* part_array = table_schema->get_part_array();
+              if (OB_ISNULL(part_array) || OB_ISNULL(part_array[0])) {
+                  ret = OB_INVALID_ARGUMENT;
+                  LOG_WARN("partition_array is null", K(ret));
+              } else {
+                const ObIArray<common::ObNewRow> &list_row_values =
+                    part_array[0]->get_list_row_values();
+                ObObj *list_part_obj = list_row_values.at(0).cells_;
+                if (OB_ISNULL(list_part_obj)) {
+                  ret = OB_INVALID_ARGUMENT;
+                  LOG_WARN("list_part_obj is null", K(ret));
+                } else if (part_expr->obj_meta_.get_type() !=
+                           list_part_obj->get_meta().get_type()) {
+                  fallback = true;
+                } else {
+                  rt_expr.eval_vector_func_ =
+                      ObExprCalcPartitionBase::fast_calc_partition_level_one_vector;
+                }
+              }
             }
           }
         } else {