[compiler-rt][memprof] adding free_sized/free_aligned_sized intercept… (#154011)

…ions.

Author: devnexen

compiler-rt/lib/memprof/memprof_malloc_linux.cpp

@@ -50,6 +50,24 @@ INTERCEPTOR(void, cfree, void *ptr) {
 }
 #endif // SANITIZER_INTERCEPT_CFREE
 
+#if SANITIZER_INTERCEPT_FREE_SIZED
+INTERCEPTOR(void, free_sized, void *ptr, uptr size) {
+  if (DlsymAlloc::PointerIsMine(ptr))
+    return DlsymAlloc::Free(ptr);
+  GET_STACK_TRACE_FREE;
+  memprof_delete(ptr, size, 0, &stack, FROM_MALLOC);
+}
+#endif // SANITIZER_INTERCEPT_FREE_SIZED
+
+#if SANITIZER_INTERCEPT_FREE_ALIGNED_SIZED
+INTERCEPTOR(void, free_aligned_sized, void *ptr, uptr alignment, uptr size) {
+  if (DlsymAlloc::PointerIsMine(ptr))
+    return DlsymAlloc::Free(ptr);
+  GET_STACK_TRACE_FREE;
+  memprof_delete(ptr, size, alignment, &stack, FROM_MALLOC);
+}
+#endif // SANITIZER_INTERCEPT_FREE_ALIGNED_SIZED
+
 INTERCEPTOR(void *, malloc, uptr size) {
   if (DlsymAlloc::Use())
     return DlsymAlloc::Allocate(size);

compiler-rt/test/memprof/TestCases/free_sized.cpp

@@ -0,0 +1,21 @@
+// RUN: %clangxx_memprof %s -o %t
+
+// RUN: %env_memprof_opts=print_text=true:log_path=stdout %run %t | FileCheck %s
+
+#include <stdlib.h>
+#include <unistd.h>
+
+extern "C" {
+void free_sized(void *ptr, size_t size);
+void free_aligned_sized(void *ptr, size_t alignement, size_t size);
+}
+
+int main() {
+  void *p = aligned_alloc(16, 32);
+  free_aligned_sized(p, 16, 32);
+  p = malloc(10);
+  free_sized(p, 10);
+  return 0;
+}
+// CHECK: Memory allocation stack id
+// CHECK: Memory allocation stack id