Skip to content

GH-49104: [C++] Fix Segfault in SparseCSFIndex::Equals with mismatched dimensions #49105

New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

Sign up for GitHub

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Merged
rok merged 13 commits into from
Feb 4, 2026
Merged

GH-49104: [C++] Fix Segfault in SparseCSFIndex::Equals with mismatched dimensions #49105

Show file tree
Hide file tree
Changes from all commits
Commits
Show all changes
13 commits
Select commit Hold shift + click to select a range
daa5f73
[C++] Fix Segfault in SparseCSFIndex::Equals with mismatched dimensions
Alirana2829 Jan 31, 2026
3e1cbd6
[C++] Add regression test for SparseCSFIndex::Equals segfault
Alirana2829 Feb 1, 2026
fdf506e
Fix Segfault in SparseCSFIndex::Equals with mismatched dimensions and…
Alirana2829 Feb 1, 2026
c87afdb
Fix lint: Format SparseCSFIndex::Equals to comply with 90-char line l…
Alirana2829 Feb 2, 2026
8dbc21b
Fix test: Use valid CSF index structures (2D vs 3D instead of invalid…
Alirana2829 Feb 2, 2026
900b98d
Remove invalid test causing segfaults with empty tensors
Alirana2829 Feb 2, 2026
76fe338
Revert formatting changes per maintainer feedback
Alirana2829 Feb 3, 2026
a6e5088
Remove redundant axis_order size check per review feedback
Alirana2829 Feb 3, 2026
f9bc643
Update cpp/src/arrow/sparse_tensor.cc
AliRana30 Feb 4, 2026
c1e33a2
Update cpp/src/arrow/sparse_tensor_test.cc
AliRana30 Feb 4, 2026
915dcdc
Fix duplicate function definition from GitHub suggestion
Alirana2829 Feb 4, 2026
b508636
Update cpp/src/arrow/sparse_tensor.cc
AliRana30 Feb 4, 2026
ae4c1b9
Apply clang-format indentation per CI requirement
Alirana2829 Feb 4, 2026
File filter

Filter by extension

Filter by extension
Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
11 changes: 4 additions & 7 deletions cpp/src/arrow/sparse_tensor.cc
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -405,13 +405,10 @@ SparseCSFIndex::SparseCSFIndex(const std::vector<std::shared_ptr<Tensor>>& indpt
std::string SparseCSFIndex::ToString() const { return std::string("SparseCSFIndex"); }

bool SparseCSFIndex::Equals(const SparseCSFIndex& other) const {
for (int64_t i = 0; i < static_cast<int64_t>(indices().size()); ++i) {
if (!indices()[i]->Equals(*other.indices()[i])) return false;
}
for (int64_t i = 0; i < static_cast<int64_t>(indptr().size()); ++i) {
if (!indptr()[i]->Equals(*other.indptr()[i])) return false;
}
return axis_order() == other.axis_order();
auto eq = [](const auto& a, const auto& b) { return a->Equals(*b); };
return axis_order() == other.axis_order() &&
std::ranges::equal(indices(), other.indices(), eq) &&
std::ranges::equal(indptr(), other.indptr(), eq);
}

// ----------------------------------------------------------------------
Expand Down
24 changes: 23 additions & 1 deletion cpp/src/arrow/sparse_tensor_test.cc
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -1641,10 +1641,32 @@ TYPED_TEST_P(TestSparseCSFTensorForIndexValueType, TestNonAscendingShape) {
ASSERT_TRUE(st->Equals(*sparse_tensor));
}

TYPED_TEST_P(TestSparseCSFTensorForIndexValueType, TestEqualityMismatchedDimensions) {
using IndexValueType = TypeParam;
using c_index_value_type = typename IndexValueType::c_type;

// 2D vs 3D - comparing indices with different dimensionality
// 2D CSF: ndim=2, so indptr.size()=1, indices.size()=2
std::vector<int64_t> axis_order_2D = {0, 1};
std::vector<std::vector<c_index_value_type>> indptr_2D = {{0, 1}};
std::vector<std::vector<c_index_value_type>> indices_2D = {{0}, {0}};
auto si_2D = this->MakeSparseCSFIndex(axis_order_2D, indptr_2D, indices_2D);

// 3D CSF: ndim=3, so indptr.size()=2, indices.size()=3
std::vector<int64_t> axis_order_3D = {0, 1, 2};
std::vector<std::vector<c_index_value_type>> indptr_3D = {{0, 1}, {0, 1}};
std::vector<std::vector<c_index_value_type>> indices_3D = {{0}, {0}, {0}};
auto si_3D = this->MakeSparseCSFIndex(axis_order_3D, indptr_3D, indices_3D);

ASSERT_FALSE(si_2D->Equals(*si_3D));
ASSERT_FALSE(si_3D->Equals(*si_2D));
ASSERT_TRUE(si_2D->Equals(*si_2D));
}

REGISTER_TYPED_TEST_SUITE_P(TestSparseCSFTensorForIndexValueType, TestCreateSparseTensor,
TestTensorToSparseTensor, TestSparseTensorToTensor,
TestAlternativeAxisOrder, TestNonAscendingShape,
TestRoundTrip);
TestRoundTrip, TestEqualityMismatchedDimensions);

INSTANTIATE_TYPED_TEST_SUITE_P(TestInt8, TestSparseCSFTensorForIndexValueType, Int8Type);
INSTANTIATE_TYPED_TEST_SUITE_P(TestUInt8, TestSparseCSFTensorForIndexValueType,
Expand Down
Loading