SERVER-36243 Use sized deallocation.

Added mongoFree to be used when allocating memory with mongoMalloc.
It has an overload taking size utilizing tc_free_sized if built with tcmalloc.

1 files changed, 20 insertions, 1 deletions

diff --git a/src/mongo/SConscript b/src/mongo/SConscript
index 39b12a203b3..b295b98d9d2 100644
--- a/src/mongo/SConscript
+++ b/src/mongo/SConscript
@@ -48,10 +48,29 @@ env.SConscript(
     ],
 )
 
+allocatorEnv = env.Clone()
+if allocatorEnv['MONGO_ALLOCATOR'] == 'tcmalloc':
+    allocatorEnv.Append(
+            CPPDEFINES=[
+                'MONGO_USE_GPERFTOOLS_TCMALLOC',
+            ]
+        )
+
+    if not use_system_version_of_library('tcmalloc'):
+        # Add in the include path for our vendored tcmalloc.
+        allocatorEnv.InjectThirdParty('gperftools')
+
+        # If our changes to tcmalloc are ever upstreamed, this should become set based on a top
+        # level configure check, though its effects should still be scoped just to these files.
+
+    if not use_system_version_of_library('valgrind'):
+        # Include valgrind since tcmalloc disables itself while running under valgrind
+        allocatorEnv.InjectThirdParty('valgrind')
+
 # NOTE: This library does not really belong here. Its presence here is
 # temporary. Do not add to this library, do not remove from it, and do
 # not declare other libraries in this file.
-env.Library(
+allocatorEnv.Library(
     target='base',
     source=[
         'base/data_range.cpp',