[VL] Fix allocate and free memory

apache · Jan 9, 2025 · 8ba547b · 8ba547b
1 parent 73ee147
commit 8ba547b
Show file tree

Hide file tree

Showing 2 changed files with 9 additions and 3 deletions.
diff --git a/cpp/core/benchmarks/CompressionBenchmark.cc b/cpp/core/benchmarks/CompressionBenchmark.cc
@@ -48,7 +48,9 @@ void printTrace(void) {
   for (i = 0; i < size; i++)
     printf("    %s\n", strings[i]);
   puts("");
-  free(strings);
+  if (strings != nullptr) {
+    free(strings);
+  }
 }
 
 using arrow::RecordBatchReader;

diff --git a/cpp/core/memory/MemoryAllocator.cc b/cpp/core/memory/MemoryAllocator.cc
@@ -172,14 +172,18 @@ bool StdMemoryAllocator::reallocateAligned(void* p, uint64_t alignment, int64_t
     return false;
   }
   memcpy(reallocatedP, p, std::min(size, newSize));
-  std::free(p);
+  if (p != nullptr) {
+    std::free(p);
+  }
   *out = reallocatedP;
   bytes_ += (newSize - size);
   return true;
 }
 
 bool StdMemoryAllocator::free(void* p, int64_t size) {
-  std::free(p);
+  if (p != nullptr) {
+    std::free(p);
+  }
   bytes_ -= size;
   return true;
 }