Donetsk/hook_lib/common/utils/memory.hpp

#pragma once

#include <mutex>
#include <vector>

namespace utils
{
	class memory final
	{
	public:
		class allocator final
		{
		public:
			~allocator();

			void clear();

			void free(void* data);

			void free(const void* data);

			void* allocate(size_t length);

			template <typename T>
			T* allocate()
			{
				return this->allocate_array<T>(1);
			}

			template <typename T>
			T* allocate_array(const size_t count = 1)
			{
				return static_cast<T*>(this->allocate(count * sizeof(T)));
			}

			bool empty() const;

			char* duplicate_string(const std::string& string);

			bool find(const void* data);

		private:
			std::mutex mutex_;
			std::vector<void*> pool_;
		};

		static void* allocate(size_t length);

		template <typename T>
		static T* allocate()
		{
			return allocate_array<T>(1);
		}

		template <typename T>
		static T* allocate_array(const size_t count = 1)
		{
			return static_cast<T*>(allocate(count * sizeof(T)));
		}

		static char* duplicate_string(const std::string& string);

		static void free(void* data);
		static void free(const void* data);

		static bool is_set(const void* mem, char chr, size_t length);

		static bool is_bad_read_ptr(const void* ptr);
		static bool is_bad_code_ptr(const void* ptr);
		static bool is_rdata_ptr(void* ptr);

		static allocator* get_allocator();

	private:
		static allocator mem_allocator_;
	};
}