From a863185e57bcecc352c7d446bd102f6872fc8879 Mon Sep 17 00:00:00 2001
From: bkaradzic <branimirkaradzic@gmail.com>
Date: Sun, 6 Oct 2013 11:09:55 -0700
Subject: [PATCH] Added guard for memory tracking.

---
 src/bgfx.cpp | 24 +++++++++++++++++++++---
 1 file changed, 21 insertions(+), 3 deletions(-)

diff --git a/src/bgfx.cpp b/src/bgfx.cpp
index 3396f084..0e7220a5 100755
--- a/src/bgfx.cpp
+++ b/src/bgfx.cpp
@@ -114,15 +114,22 @@ namespace bgfx
 	{
 	public:
 		AllocatorStub()
+#if BGFX_CONFIG_DEBUG
 			: m_numBlocks(0)
 			, m_maxBlocks(0)
+#endif // BGFX_CONFIG_DEBUG
 		{
 		}
 
 		virtual void* alloc(size_t _size, const char* _file, uint32_t _line) BX_OVERRIDE
 		{
-			++m_numBlocks;
-			m_maxBlocks = bx::uint32_max(m_maxBlocks, m_numBlocks);
+#if BGFX_CONFIG_DEBUG
+			{
+				bx::LwMutexScope scope(m_mutex);
+				++m_numBlocks;
+				m_maxBlocks = bx::uint32_max(m_maxBlocks, m_numBlocks);
+			}
+#endif // BGFX_CONFIG_DEBUG
 
 			BX_UNUSED(_file, _line);
 			return ::malloc(_size);
@@ -132,7 +139,12 @@ namespace bgfx
 		{
 			if (NULL != _ptr)
 			{
-				--m_numBlocks;
+#if BGFX_CONFIG_DEBUG
+				{
+					bx::LwMutexScope scope(m_mutex);
+					--m_numBlocks;
+				}
+#endif // BGFX_CONFIG_DEBUG
 
 				BX_UNUSED(_file, _line);
 				::free(_ptr);
@@ -141,11 +153,14 @@ namespace bgfx
 
 		virtual void* realloc(void* _ptr, size_t _size, const char* _file, uint32_t _line) BX_OVERRIDE
 		{
+#if BGFX_CONFIG_DEBUG
 			if (NULL == _ptr)
 			{
+				bx::LwMutexScope scope(m_mutex);
 				++m_numBlocks;
 				m_maxBlocks = bx::uint32_max(m_maxBlocks, m_numBlocks);
 			}
+#endif // BGFX_CONFIG_DEBUG
 
 			BX_UNUSED(_file, _line);
 			return ::realloc(_ptr, _size);
@@ -157,8 +172,11 @@ namespace bgfx
 		}
 
 	private:
+#if BGFX_CONFIG_DEBUG
+		bx::LwMutex m_mutex;
 		uint32_t m_numBlocks;
 		uint32_t m_maxBlocks;
+#endif // BGFX_CONFIG_DEBUG
 	};
 
 	static CallbackStub* s_callbackStub = NULL;