1 files changed, 24 insertions, 0 deletions
diff --git a/src/zencore/memtrack/tracemalloc.h b/src/zencore/memtrack/tracemalloc.h
new file mode 100644
index 000000000..54606ac45
--- /dev/null
+++ b/src/zencore/memtrack/tracemalloc.h
@@ -0,0 +1,24 @@
+// Copyright Epic Games, Inc. All Rights Reserved.
+#pragma once
+
+#include <zencore/memory/fmalloc.h>
+#include <zencore/memory/memorytrace.h>
+
+namespace zen {
+
+class FTraceMalloc : public FMalloc
+{
+public:
+	FTraceMalloc(FMalloc* InMalloc);
+	virtual ~FTraceMalloc();
+
+	virtual void* Malloc(SIZE_T Count, uint32 Alignment) override;
+	virtual void* Realloc(void* Original, SIZE_T Count, uint32 Alignment) override;
+	virtual void  Free(void* Original) override;
+
+	virtual void OnMallocInitialized() override { WrappedMalloc->OnMallocInitialized(); }
+
+	FMalloc* WrappedMalloc;
+};
+
+}  // namespace zen