YimMenu/src/memory/byte_patch.hpp

#pragma once

namespace memory
{
	class byte_patch
	{
	public:
		virtual ~byte_patch();

		void apply() const;

		void restore() const;

		void remove() const;

		template <typename TAddr>
		static const std::unique_ptr<byte_patch>& make(TAddr address, std::remove_pointer_t<std::remove_reference_t<TAddr>> value)
		{
			return m_patches.emplace_back(
				std::unique_ptr<byte_patch>(new byte_patch(address, value)));
		}

		static void restore_all();

	private:
		template <typename TAddr>
		byte_patch(TAddr address, std::remove_pointer_t<std::remove_reference_t<TAddr>> value)
			: m_address(address)
		{
			m_size = sizeof(std::remove_pointer_t<std::remove_reference_t<TAddr>>);
			m_original_bytes = std::make_unique<uint8_t[]>(m_size);
			m_value = std::make_unique<uint8_t[]>(m_size);

			memcpy(m_original_bytes.get(), m_address, m_size);
			memcpy(m_value.get(), &value, m_size);
		}

	protected:
		static inline std::vector<std::unique_ptr<byte_patch>> m_patches;

	private:
		void* m_address;
		std::unique_ptr<uint8_t[]> m_value;
		std::unique_ptr<uint8_t[]> m_original_bytes;
		std::size_t m_size;

		friend bool operator== (const std::unique_ptr<byte_patch>& a, const byte_patch* b);
	};
}