From c2f24cfa8c157b70809701ec3c86c45301c5acf0 Mon Sep 17 00:00:00 2001
From: laihui <laihui@selectdb.com>
Date: Wed, 12 Nov 2025 11:07:03 +0800
Subject: [PATCH] add check when memtable init aggregate function to avoid
 coredump

---
 be/src/olap/memtable.cpp | 4 ++++
 1 file changed, 4 insertions(+)

diff --git a/be/src/olap/memtable.cpp b/be/src/olap/memtable.cpp
index 8a6cb67f714aa8..03a97d876c712a 100644
--- a/be/src/olap/memtable.cpp
+++ b/be/src/olap/memtable.cpp
@@ -104,6 +104,10 @@ void MemTable::_init_columns_offset_by_slot_descs(const std::vector<SlotDescript
 }
 
 void MemTable::_init_agg_functions(const vectorized::Block* block) {
+    if (_num_columns > _column_offset.size()) [[unlikely]] {
+        throw std::runtime_error(fmt::format("num_columns {} is greater than block columns {}",
+                                             _num_columns, _column_offset.size()));
+    }
     for (auto cid = _tablet_schema->num_key_columns(); cid < _num_columns; ++cid) {
         vectorized::AggregateFunctionPtr function;
         if (_keys_type == KeysType::UNIQUE_KEYS && _enable_unique_key_mow) {