MemLicker.hpp

#ifndef _MEM_LICKER_
#define	_MEM_LICKER_

#include <unordered_map>
#include <iostream>
#include <string>

using namespace std;
using ptr		= void*;
struct PtrData
{
	string	file;
	int		line;
	size_t	size;
	PtrData(string f, int l, size_t s)
	{
		file = f;
		line = l;
		size = s;
	}
};


using LeakData	= pair<ptr,	PtrData>;
class MemLicker
{
private:

	static MemLicker*			instance;
	unordered_map<ptr, PtrData> mLeaks;
	MemLicker() : mLeaks() {};

public:
	~MemLicker() { delete instance; instance = nullptr; }
	static MemLicker* getLicker() 
	{ 
		if (instance == nullptr) { instance = new MemLicker(); }
		return instance; 
	};

	template<class T, typename ...Args>
	T* allocate(string file, int line, size_t size = 0, Args&&...args)
	{
		T* ret;
		if(size > 0)	ret = new T[size];
		else			ret = new T(args...);
		mLeaks.insert(LeakData(ret, PtrData(file, line, size > 0 ? sizeof(T)*size : sizeof(T))));
		return ret;
	}

	void deallocate(ptr p)
	{
		auto pData = mLeaks.find(p);
		mLeaks.erase(p);
		if (pData->second.size > 0) delete[]	p;
		else						delete		p;
	}

	void report()
	{
		for(LeakData ldata : mLeaks)
		{
			stringstream s;
			//s << ldata.second.file << "(" << ldata.second.line << "):" << " Leaked " << ldata.second.size << "bytes @ memory address: " << ldata.first << endl;
			OutputDebugStringA(s.str().c_str());
			//cout << s.str() << endl;
		}
	}
};

MemLicker* MemLicker::instance = nullptr;

#endif // !_MEM_LICKER_