This is an automated email from the ASF dual-hosted git repository. dataroaring pushed a commit to branch branch-3.0 in repository https://gitbox.apache.org/repos/asf/doris.git
The following commit(s) were added to refs/heads/branch-3.0 by this push: new fd5fcebb8f1 branch-3.0: [fix](column) fix ColumnWithTypeAndName::get_nested use-after-free when input Const(Nullable) column #48288 (#49246) fd5fcebb8f1 is described below commit fd5fcebb8f1c6156551fca41d60d31b2ca554b63 Author: github-actions[bot] <41898282+github-actions[bot]@users.noreply.github.com> AuthorDate: Thu Mar 20 14:40:20 2025 +0800 branch-3.0: [fix](column) fix ColumnWithTypeAndName::get_nested use-after-free when input Const(Nullable) column #48288 (#49246) Cherry-picked from #48288 Co-authored-by: Mryange <yanxuech...@selectdb.com> --- be/src/vec/core/column_with_type_and_name.cpp | 5 +-- .../vec/core/column_with_type_and_name_test.cpp | 40 ++++++++++++++++++++++ 2 files changed, 43 insertions(+), 2 deletions(-) diff --git a/be/src/vec/core/column_with_type_and_name.cpp b/be/src/vec/core/column_with_type_and_name.cpp index e93946804ff..39bcc24ccdd 100644 --- a/be/src/vec/core/column_with_type_and_name.cpp +++ b/be/src/vec/core/column_with_type_and_name.cpp @@ -95,8 +95,9 @@ ColumnWithTypeAndName ColumnWithTypeAndName::get_nested(bool replace_null_data_t auto nested_type = assert_cast<const DataTypeNullable*>(type.get())->get_nested_type(); ColumnPtr nested_column = column; if (column) { - nested_column = nested_column->convert_to_full_column_if_const(); - const auto* source_column = assert_cast<const ColumnNullable*>(nested_column.get()); + // A column_ptr is needed here to ensure that the column in convert_to_full_column_if_const is not released. + auto column_ptr = nested_column->convert_to_full_column_if_const(); + const auto* source_column = assert_cast<const ColumnNullable*>(column_ptr.get()); nested_column = source_column->get_nested_column_ptr(); if (replace_null_data_to_default) { diff --git a/be/test/vec/core/column_with_type_and_name_test.cpp b/be/test/vec/core/column_with_type_and_name_test.cpp new file mode 100644 index 00000000000..1cdda959a82 --- /dev/null +++ b/be/test/vec/core/column_with_type_and_name_test.cpp @@ -0,0 +1,40 @@ +// Licensed to the Apache Software Foundation (ASF) under one +// or more contributor license agreements. See the NOTICE file +// distributed with this work for additional information +// regarding copyright ownership. The ASF licenses this file +// to you under the Apache License, Version 2.0 (the +// "License"); you may not use this file except in compliance +// with the License. You may obtain a copy of the License at +// +// http://www.apache.org/licenses/LICENSE-2.0 +// +// Unless required by applicable law or agreed to in writing, +// software distributed under the License is distributed on an +// "AS IS" BASIS, WITHOUT WARRANTIES OR CONDITIONS OF ANY +// KIND, either express or implied. See the License for the +// specific language governing permissions and limitations +// under the License. + +#include "vec/core/column_with_type_and_name.h" + +#include <gtest/gtest.h> + +#include "testutil/column_helper.h" +#include "vec/core/types.h" +#include "vec/data_types/data_type_nullable.h" +#include "vec/data_types/data_type_number.h" + +namespace doris::vectorized { + +TEST(ColumnWithTypeAndNameTest, get_nested_test) { + ColumnWithTypeAndName column_with_type_and_name; + auto null_column = ColumnNullable::create(ColumnHelper::create_column<DataTypeInt32>({1}), + ColumnHelper::create_column<DataTypeUInt8>({true})); + column_with_type_and_name.column = ColumnConst::create(null_column, 3); + column_with_type_and_name.type = + std::make_shared<DataTypeNullable>(std::make_shared<DataTypeInt32>()); + column_with_type_and_name.name = "column_with_type_and_name"; + column_with_type_and_name.get_nested(true); +} + +} // namespace doris::vectorized --------------------------------------------------------------------- To unsubscribe, e-mail: commits-unsubscr...@doris.apache.org For additional commands, e-mail: commits-h...@doris.apache.org