Fix potential bug when resize to less elements

utk-team · Sep 20, 2023 · 049ba55 · 049ba55
1 parent e3424cb
commit 049ba55
Showing 1 changed file with 2 additions and 3 deletions.
diff --git a/include/utk/utils/Pointset.hpp b/include/utk/utils/Pointset.hpp
@@ -181,8 +181,7 @@ namespace utk
             // Always realloc if it previously was a view
             if (isView)
             {
-                uint32_t old_n = N;
-                uint32_t old_d = D;
+                const uint32_t oldC = C;
                 N = n;
                 D = d;
 
@@ -194,7 +193,7 @@ namespace utk
                 // by resetting data
                 std::shared_ptr<T[]> oldData = data;
                 data = std::shared_ptr<T[]>(new T[C]);
-                std::memcpy(data.get(), oldData.get(), old_n * old_d * sizeof(T));
+                std::memcpy(data.get(), oldData.get(), std::min(C, oldC) * sizeof(T));
             }
             else
             {