mirror of
https://github.com/shadps4-emu/shadPS4.git
synced 2024-12-29 19:16:07 +00:00
improved memory_alloc function
This commit is contained in:
parent
9103a545a2
commit
bc2facaee4
|
@ -1,4 +1,5 @@
|
||||||
#include "../Core/PS4/Loader/Elf.h"
|
#include "../Core/PS4/Loader/Elf.h"
|
||||||
|
#include "Memory.h"
|
||||||
|
|
||||||
#ifdef _WIN64
|
#ifdef _WIN64
|
||||||
#include <windows.h>
|
#include <windows.h>
|
||||||
|
@ -11,15 +12,33 @@
|
||||||
namespace Memory
|
namespace Memory
|
||||||
{
|
{
|
||||||
namespace VirtualMemory {
|
namespace VirtualMemory {
|
||||||
|
static DWORD convertMemoryMode(MemoryMode mode)
|
||||||
|
{
|
||||||
|
switch (mode)
|
||||||
|
{
|
||||||
|
case MemoryMode::Read: return PAGE_READONLY;
|
||||||
|
case MemoryMode::Write:
|
||||||
|
case MemoryMode::ReadWrite: return PAGE_READWRITE;
|
||||||
|
|
||||||
u64 memory_alloc(u64 address, u64 size)
|
case MemoryMode::Execute: return PAGE_EXECUTE;
|
||||||
|
case MemoryMode::ExecuteRead: return PAGE_EXECUTE_READ;
|
||||||
|
case MemoryMode::ExecuteWrite:
|
||||||
|
case MemoryMode::ExecuteReadWrite: return PAGE_EXECUTE_READWRITE;
|
||||||
|
|
||||||
|
case MemoryMode::NoAccess: return PAGE_NOACCESS;
|
||||||
|
default:
|
||||||
|
return PAGE_NOACCESS;
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
u64 memory_alloc(u64 address, u64 size, MemoryMode mode)
|
||||||
{
|
{
|
||||||
//TODO it supports only execute_read_write mode
|
//TODO it supports only execute_read_write mode
|
||||||
#ifdef _WIN64
|
#ifdef _WIN64
|
||||||
auto ptr = reinterpret_cast<uintptr_t>(VirtualAlloc(reinterpret_cast<LPVOID>(static_cast<uintptr_t>(address)),
|
auto ptr = reinterpret_cast<uintptr_t>(VirtualAlloc(reinterpret_cast<LPVOID>(static_cast<uintptr_t>(address)),
|
||||||
size,
|
size,
|
||||||
static_cast<DWORD>(MEM_COMMIT) | static_cast<DWORD>(MEM_RESERVE),
|
static_cast<DWORD>(MEM_COMMIT) | static_cast<DWORD>(MEM_RESERVE),
|
||||||
PAGE_EXECUTE_READWRITE));
|
convertMemoryMode(mode)));
|
||||||
|
|
||||||
if (ptr == 0)
|
if (ptr == 0)
|
||||||
{
|
{
|
||||||
|
|
|
@ -1,11 +1,24 @@
|
||||||
#pragma once
|
#pragma once
|
||||||
|
#include "../types.h"
|
||||||
|
|
||||||
constexpr u64 SYSTEM_RESERVED = 0x800000000u;
|
constexpr u64 SYSTEM_RESERVED = 0x800000000u;
|
||||||
constexpr u64 CODE_BASE_OFFSET = 0x100000000u;
|
constexpr u64 CODE_BASE_OFFSET = 0x100000000u;
|
||||||
|
|
||||||
namespace Memory
|
namespace Memory
|
||||||
{
|
{
|
||||||
|
enum class MemoryMode : u32
|
||||||
|
{
|
||||||
|
NoAccess = 0,
|
||||||
|
Read = 1,
|
||||||
|
Write = 2,
|
||||||
|
ReadWrite =3,
|
||||||
|
Execute = 4,
|
||||||
|
ExecuteRead = 5,
|
||||||
|
ExecuteWrite = 6,
|
||||||
|
ExecuteReadWrite = 7,
|
||||||
|
};
|
||||||
|
|
||||||
namespace VirtualMemory {
|
namespace VirtualMemory {
|
||||||
u64 memory_alloc(u64 address, u64 size);
|
u64 memory_alloc(u64 address, u64 size, MemoryMode mode);
|
||||||
}
|
}
|
||||||
}
|
}
|
|
@ -109,7 +109,7 @@ void Linker::LoadModuleToMemory(Module* m)
|
||||||
u64 base_size = calculate_base_size(elf_header, elf_pheader);
|
u64 base_size = calculate_base_size(elf_header, elf_pheader);
|
||||||
m->aligned_base_size = (base_size & ~(static_cast<u64>(0x1000) - 1)) + 0x1000;//align base size to 0x1000 block size (TODO is that the default block size or it can be changed?
|
m->aligned_base_size = (base_size & ~(static_cast<u64>(0x1000) - 1)) + 0x1000;//align base size to 0x1000 block size (TODO is that the default block size or it can be changed?
|
||||||
|
|
||||||
m->base_virtual_addr = Memory::VirtualMemory::memory_alloc(g_load_addr, m->aligned_base_size);
|
m->base_virtual_addr = Memory::VirtualMemory::memory_alloc(g_load_addr, m->aligned_base_size, Memory::MemoryMode::ExecuteReadWrite);
|
||||||
|
|
||||||
LOG_INFO_IF(debug_loader, "====Load Module to Memory ========\n");
|
LOG_INFO_IF(debug_loader, "====Load Module to Memory ========\n");
|
||||||
LOG_INFO_IF(debug_loader, "base_virtual_addr ......: {:#018x}\n", m->base_virtual_addr);
|
LOG_INFO_IF(debug_loader, "base_virtual_addr ......: {:#018x}\n", m->base_virtual_addr);
|
||||||
|
|
Loading…
Reference in a new issue