YimMenu/BigBaseV2/src/memory/byte_patch.hpp

#pragma once

namespace memory
{
	class byte_patch
	{
	public:
		virtual ~byte_patch()
		{
			memcpy(m_address, m_original_bytes.data(), m_original_bytes.size());
		}

		void restore() const
		{
			if (const auto it = std::find(m_patches.begin(), m_patches.end(), this); it != m_patches.end())
			{
				m_patches.erase(it);
			}
		}

		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()
		{
			m_patches.clear();
		}

	private:
		template <typename TAddr>
		byte_patch(TAddr address, std::remove_pointer_t<std::remove_reference_t<TAddr>> value)
			: m_address(address)
		{
			constexpr auto size = sizeof(std::remove_pointer_t<std::remove_reference_t<TAddr>>);
			m_original_bytes.resize(size);
			memcpy(m_original_bytes.data(), m_address, size);

			*address = value;
		}

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

	private:
		void* m_address;
		std::vector<uint8_t> m_original_bytes;

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

	bool operator== (const std::unique_ptr<byte_patch>& a, const byte_patch* b)
	{
		return a->m_address == b->m_address;
	}
}