/**

* @file byte_patch.cpp

*

* @copyright GNU General Public License Version 2.

* This file is part of YimMenu.

* YimMenu is free software: you can redistribute it and/or modify it under the terms of the GNU General Public License as published by the Free Software Foundation, either version 2 of the License, or (at your option) any later version.

* YimMenu is distributed in the hope that it will be useful, but WITHOUT ANY WARRANTY; without even the implied warranty of MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU General Public License for more details.

* You should have received a copy of the GNU General Public License along with YimMenu. If not, see <https://www.gnu.org/licenses/>.

*/

#include "byte_patch.hpp"

namespace memory

{

byte_patch::~byte_patch()

{

restore();

}

void byte_patch::apply() const

{

DWORD old_protect;

VirtualProtect(m_address, m_size, PAGE_EXECUTE_READWRITE, (PDWORD)&old_protect);

memcpy(m_address, m_value.get(), m_size);

DWORD unused;

VirtualProtect(m_address, m_size, old_protect, &unused);

}

void byte_patch::restore() const

{

DWORD old_protect;

VirtualProtect(m_address, m_size, PAGE_EXECUTE_READWRITE, (PDWORD)&old_protect);

memcpy(m_address, m_original_bytes.get(), m_size);

DWORD unused;

VirtualProtect(m_address, m_size, old_protect, &unused);

}

void byte_patch::remove() const

{

std::erase_if(m_patches, [this](auto& el) {

return el.get() == this;

});

}

void byte_patch::restore_all()

{

m_patches.clear();

}

bool operator==(const std::shared_ptr<byte_patch>& a, const byte_patch* b)

{

return a->m_address == b->m_address;

}

}