2014-08-01 18:27:48 +02:00
|
|
|
#include "stdafx.h"
|
2014-08-31 11:54:12 +02:00
|
|
|
#include "Memory.h"
|
2015-02-07 00:39:51 +01:00
|
|
|
#include "Emu/System.h"
|
2015-07-15 13:58:13 +02:00
|
|
|
#include "Utilities/Thread.h"
|
2015-01-07 03:14:00 +01:00
|
|
|
#include "Emu/CPU/CPUThread.h"
|
|
|
|
|
#include "Emu/Cell/PPUThread.h"
|
2015-07-04 01:22:24 +02:00
|
|
|
#include "Emu/Cell/SPUThread.h"
|
2016-05-13 16:01:48 +02:00
|
|
|
#include "Emu/PSP2/ARMv7Thread.h"
|
2014-08-31 11:54:12 +02:00
|
|
|
|
2015-02-07 00:39:51 +01:00
|
|
|
#ifdef _WIN32
|
|
|
|
|
#include <Windows.h>
|
|
|
|
|
#else
|
|
|
|
|
#include <sys/mman.h>
|
2015-02-07 14:59:10 +01:00
|
|
|
#include <sys/stat.h>
|
|
|
|
|
#include <fcntl.h>
|
2015-02-07 16:35:54 +01:00
|
|
|
#include <unistd.h>
|
|
|
|
|
#include <sys/types.h>
|
2015-02-07 00:39:51 +01:00
|
|
|
|
|
|
|
|
/* OS X uses MAP_ANON instead of MAP_ANONYMOUS */
|
|
|
|
|
#ifndef MAP_ANONYMOUS
|
|
|
|
|
#define MAP_ANONYMOUS MAP_ANON
|
|
|
|
|
#endif
|
|
|
|
|
#endif
|
|
|
|
|
|
2016-04-25 12:49:12 +02:00
|
|
|
#include "wait_engine.h"
|
|
|
|
|
|
2014-11-19 15:16:30 +01:00
|
|
|
namespace vm
|
|
|
|
|
{
|
2016-02-01 22:52:49 +01:00
|
|
|
thread_local u64 g_tls_fault_count{};
|
|
|
|
|
|
2015-09-26 22:46:04 +02:00
|
|
|
template<std::size_t Size> struct mapped_ptr_deleter
|
2015-02-07 00:39:51 +01:00
|
|
|
{
|
2015-09-26 22:46:04 +02:00
|
|
|
void operator ()(void* ptr)
|
|
|
|
|
{
|
2015-02-07 00:39:51 +01:00
|
|
|
#ifdef _WIN32
|
2015-09-26 22:46:04 +02:00
|
|
|
::UnmapViewOfFile(ptr);
|
|
|
|
|
#else
|
|
|
|
|
::munmap(ptr, Size);
|
|
|
|
|
#endif
|
|
|
|
|
}
|
|
|
|
|
};
|
2015-02-07 00:39:51 +01:00
|
|
|
|
2015-09-26 22:46:04 +02:00
|
|
|
using mapped_ptr_t = std::unique_ptr<u8[], mapped_ptr_deleter<0x100000000>>;
|
2015-02-07 00:39:51 +01:00
|
|
|
|
2015-09-26 22:46:04 +02:00
|
|
|
std::array<mapped_ptr_t, 2> initialize()
|
|
|
|
|
{
|
|
|
|
|
#ifdef _WIN32
|
|
|
|
|
const HANDLE memory_handle = ::CreateFileMapping(INVALID_HANDLE_VALUE, NULL, PAGE_READWRITE | SEC_RESERVE, 0x1, 0x0, NULL);
|
|
|
|
|
|
|
|
|
|
if (memory_handle == NULL)
|
|
|
|
|
{
|
2015-12-18 12:11:18 +01:00
|
|
|
MessageBoxA(0, fmt::format("CreateFileMapping() failed (0x%x).", GetLastError()).c_str(), "vm::initialize()", MB_ICONERROR);
|
|
|
|
|
std::abort();
|
2015-09-26 22:46:04 +02:00
|
|
|
}
|
2015-02-07 14:46:42 +01:00
|
|
|
|
2015-09-26 22:46:04 +02:00
|
|
|
mapped_ptr_t base_addr(static_cast<u8*>(::MapViewOfFile(memory_handle, FILE_MAP_WRITE, 0, 0, 0x100000000)));
|
|
|
|
|
mapped_ptr_t priv_addr(static_cast<u8*>(::MapViewOfFile(memory_handle, FILE_MAP_WRITE, 0, 0, 0x100000000)));
|
|
|
|
|
|
|
|
|
|
::CloseHandle(memory_handle);
|
2015-02-07 00:39:51 +01:00
|
|
|
#else
|
2015-09-26 22:46:04 +02:00
|
|
|
const int memory_handle = ::shm_open("/rpcs3_vm", O_RDWR | O_CREAT | O_EXCL, S_IRUSR | S_IWUSR);
|
2015-02-07 14:46:42 +01:00
|
|
|
|
2015-02-07 16:35:54 +01:00
|
|
|
if (memory_handle == -1)
|
|
|
|
|
{
|
2015-12-18 12:11:18 +01:00
|
|
|
std::printf("shm_open('/rpcs3_vm') failed (%d).\n", errno);
|
|
|
|
|
std::abort();
|
2015-02-07 16:35:54 +01:00
|
|
|
}
|
|
|
|
|
|
2015-09-26 22:46:04 +02:00
|
|
|
if (::ftruncate(memory_handle, 0x100000000) == -1)
|
2015-02-13 16:26:42 +01:00
|
|
|
{
|
2015-12-18 12:11:18 +01:00
|
|
|
std::printf("ftruncate(memory_handle) failed (%d).\n", errno);
|
2015-09-26 22:46:04 +02:00
|
|
|
::shm_unlink("/rpcs3_vm");
|
2015-12-18 12:11:18 +01:00
|
|
|
std::abort();
|
2015-02-13 16:26:42 +01:00
|
|
|
}
|
2015-02-07 16:35:54 +01:00
|
|
|
|
2015-09-26 22:46:04 +02:00
|
|
|
mapped_ptr_t base_addr(static_cast<u8*>(::mmap(nullptr, 0x100000000, PROT_NONE, MAP_SHARED, memory_handle, 0)));
|
|
|
|
|
mapped_ptr_t priv_addr(static_cast<u8*>(::mmap(nullptr, 0x100000000, PROT_NONE, MAP_SHARED, memory_handle, 0)));
|
2015-02-07 16:35:54 +01:00
|
|
|
|
2015-09-26 22:46:04 +02:00
|
|
|
::shm_unlink("/rpcs3_vm");
|
|
|
|
|
#endif
|
2015-02-07 14:46:42 +01:00
|
|
|
|
2015-12-18 12:11:18 +01:00
|
|
|
std::printf("vm::g_base_addr = %p\nvm::g_priv_addr = %p\n", base_addr.get(), priv_addr.get());
|
2015-07-11 22:44:53 +02:00
|
|
|
|
2015-09-26 22:46:04 +02:00
|
|
|
return{ std::move(base_addr), std::move(priv_addr) };
|
2015-02-07 00:39:51 +01:00
|
|
|
}
|
|
|
|
|
|
2015-09-26 22:46:04 +02:00
|
|
|
const auto g_addr_set = vm::initialize();
|
2015-02-07 00:39:51 +01:00
|
|
|
|
2015-09-26 22:46:04 +02:00
|
|
|
u8* const g_base_addr = g_addr_set[0].get();
|
|
|
|
|
u8* const g_priv_addr = g_addr_set[1].get();
|
2015-02-07 00:39:51 +01:00
|
|
|
|
2015-09-18 00:41:14 +02:00
|
|
|
std::array<atomic_t<u8>, 0x100000000ull / 4096> g_pages{}; // information about every page
|
2015-02-13 15:04:03 +01:00
|
|
|
|
2015-09-26 22:46:04 +02:00
|
|
|
std::vector<std::shared_ptr<block_t>> g_locations; // memory locations
|
|
|
|
|
|
2016-05-13 16:01:48 +02:00
|
|
|
access_violation::access_violation(u64 addr, const char* cause)
|
|
|
|
|
: std::runtime_error(fmt::exception("Access violation %s address 0x%llx", cause, addr))
|
2015-02-07 14:46:42 +01:00
|
|
|
{
|
2016-05-13 16:01:48 +02:00
|
|
|
g_tls_fault_count &= ~(1ull << 63);
|
|
|
|
|
}
|
2015-02-07 14:46:42 +01:00
|
|
|
|
2016-05-13 16:01:48 +02:00
|
|
|
using reservation_mutex_t = std::mutex;
|
2015-02-07 00:39:51 +01:00
|
|
|
|
2016-05-13 16:01:48 +02:00
|
|
|
thread_ctrl* volatile g_reservation_owner = nullptr;
|
2015-02-07 00:39:51 +01:00
|
|
|
|
2015-02-07 14:46:42 +01:00
|
|
|
u32 g_reservation_addr = 0;
|
2015-02-08 14:38:08 +01:00
|
|
|
u32 g_reservation_size = 0;
|
2015-02-07 14:46:42 +01:00
|
|
|
|
2015-07-21 22:14:04 +02:00
|
|
|
thread_local bool g_tls_did_break_reservation = false;
|
|
|
|
|
|
2015-02-07 14:46:42 +01:00
|
|
|
reservation_mutex_t g_reservation_mutex;
|
|
|
|
|
|
2015-02-08 16:25:50 +01:00
|
|
|
void _reservation_set(u32 addr, bool no_access = false)
|
2015-02-07 00:39:51 +01:00
|
|
|
{
|
2015-02-07 14:46:42 +01:00
|
|
|
#ifdef _WIN32
|
|
|
|
|
DWORD old;
|
2015-09-26 22:46:04 +02:00
|
|
|
if (!::VirtualProtect(vm::base(addr & ~0xfff), 4096, no_access ? PAGE_NOACCESS : PAGE_READONLY, &old))
|
2015-02-07 14:46:42 +01:00
|
|
|
#else
|
2015-09-26 22:46:04 +02:00
|
|
|
if (::mprotect(vm::base(addr & ~0xfff), 4096, no_access ? PROT_NONE : PROT_READ))
|
2015-02-07 14:46:42 +01:00
|
|
|
#endif
|
|
|
|
|
{
|
2015-07-02 03:54:36 +02:00
|
|
|
throw EXCEPTION("System failure (addr=0x%x)", addr);
|
2015-02-07 14:46:42 +01:00
|
|
|
}
|
|
|
|
|
}
|
|
|
|
|
|
2015-07-21 22:14:04 +02:00
|
|
|
bool _reservation_break(u32 addr)
|
2015-02-07 14:46:42 +01:00
|
|
|
{
|
2015-02-07 00:39:51 +01:00
|
|
|
if (g_reservation_addr >> 12 == addr >> 12)
|
|
|
|
|
{
|
|
|
|
|
#ifdef _WIN32
|
2015-02-07 23:09:23 +01:00
|
|
|
DWORD old;
|
2015-09-26 22:46:04 +02:00
|
|
|
if (!::VirtualProtect(vm::base(addr & ~0xfff), 4096, PAGE_READWRITE, &old))
|
2015-02-07 00:39:51 +01:00
|
|
|
#else
|
2015-09-26 22:46:04 +02:00
|
|
|
if (::mprotect(vm::base(addr & ~0xfff), 4096, PROT_READ | PROT_WRITE))
|
2015-02-07 00:39:51 +01:00
|
|
|
#endif
|
|
|
|
|
{
|
2015-07-02 03:54:36 +02:00
|
|
|
throw EXCEPTION("System failure (addr=0x%x)", addr);
|
2015-02-07 00:39:51 +01:00
|
|
|
}
|
|
|
|
|
|
|
|
|
|
g_reservation_addr = 0;
|
2015-02-08 14:38:08 +01:00
|
|
|
g_reservation_size = 0;
|
2015-07-12 13:52:55 +02:00
|
|
|
g_reservation_owner = nullptr;
|
2015-07-21 22:14:04 +02:00
|
|
|
|
|
|
|
|
return true;
|
2015-02-07 00:39:51 +01:00
|
|
|
}
|
2015-07-21 22:14:04 +02:00
|
|
|
|
|
|
|
|
return false;
|
2015-02-07 00:39:51 +01:00
|
|
|
}
|
|
|
|
|
|
2015-07-11 22:44:53 +02:00
|
|
|
void reservation_break(u32 addr)
|
2015-02-07 14:46:42 +01:00
|
|
|
{
|
2015-07-15 13:58:13 +02:00
|
|
|
std::unique_lock<reservation_mutex_t> lock(g_reservation_mutex);
|
|
|
|
|
|
|
|
|
|
const u32 raddr = g_reservation_addr;
|
|
|
|
|
const u32 rsize = g_reservation_size;
|
2015-02-07 14:46:42 +01:00
|
|
|
|
2015-07-15 13:58:13 +02:00
|
|
|
if ((g_tls_did_break_reservation = _reservation_break(addr)))
|
|
|
|
|
{
|
2016-04-25 12:49:12 +02:00
|
|
|
lock.unlock(), vm::notify_at(raddr, rsize);
|
2015-07-15 13:58:13 +02:00
|
|
|
}
|
2015-02-07 14:46:42 +01:00
|
|
|
}
|
|
|
|
|
|
2015-07-11 22:44:53 +02:00
|
|
|
void reservation_acquire(void* data, u32 addr, u32 size)
|
2015-02-07 00:39:51 +01:00
|
|
|
{
|
2015-07-11 22:44:53 +02:00
|
|
|
std::lock_guard<reservation_mutex_t> lock(g_reservation_mutex);
|
2015-02-07 00:39:51 +01:00
|
|
|
|
2015-07-21 22:14:04 +02:00
|
|
|
const u64 align = 0x80000000ull >> cntlz32(size);
|
|
|
|
|
|
|
|
|
|
if (!size || !addr || size > 4096 || size != align || addr & (align - 1))
|
|
|
|
|
{
|
|
|
|
|
throw EXCEPTION("Invalid arguments (addr=0x%x, size=0x%x)", addr, size);
|
|
|
|
|
}
|
2015-02-07 00:39:51 +01:00
|
|
|
|
2015-09-18 00:41:14 +02:00
|
|
|
const u8 flags = g_pages[addr >> 12];
|
2015-02-07 00:39:51 +01:00
|
|
|
|
2015-07-11 22:44:53 +02:00
|
|
|
if (!(flags & page_writable) || !(flags & page_allocated) || (flags & page_no_reservations))
|
|
|
|
|
{
|
|
|
|
|
throw EXCEPTION("Invalid page flags (addr=0x%x, size=0x%x, flags=0x%x)", addr, size, flags);
|
|
|
|
|
}
|
2015-02-07 00:39:51 +01:00
|
|
|
|
2015-07-21 22:14:04 +02:00
|
|
|
// break the reservation
|
|
|
|
|
g_tls_did_break_reservation = g_reservation_owner && _reservation_break(g_reservation_addr);
|
2015-02-07 21:06:35 +01:00
|
|
|
|
2015-07-11 22:44:53 +02:00
|
|
|
// change memory protection to read-only
|
|
|
|
|
_reservation_set(addr);
|
2015-02-07 00:39:51 +01:00
|
|
|
|
2015-07-11 22:44:53 +02:00
|
|
|
// may not be necessary
|
|
|
|
|
_mm_mfence();
|
2015-02-07 00:39:51 +01:00
|
|
|
|
2015-07-11 22:44:53 +02:00
|
|
|
// set additional information
|
|
|
|
|
g_reservation_addr = addr;
|
|
|
|
|
g_reservation_size = size;
|
2015-11-26 09:06:29 +01:00
|
|
|
g_reservation_owner = thread_ctrl::get_current();
|
2015-02-07 00:39:51 +01:00
|
|
|
|
2015-07-11 22:44:53 +02:00
|
|
|
// copy data
|
2015-09-26 22:46:04 +02:00
|
|
|
std::memcpy(data, vm::base(addr), size);
|
2015-02-16 18:19:17 +01:00
|
|
|
}
|
|
|
|
|
|
2015-02-07 00:39:51 +01:00
|
|
|
bool reservation_update(u32 addr, const void* data, u32 size)
|
|
|
|
|
{
|
2015-07-15 13:58:13 +02:00
|
|
|
std::unique_lock<reservation_mutex_t> lock(g_reservation_mutex);
|
2015-07-11 22:44:53 +02:00
|
|
|
|
2015-07-21 22:14:04 +02:00
|
|
|
const u64 align = 0x80000000ull >> cntlz32(size);
|
|
|
|
|
|
|
|
|
|
if (!size || !addr || size > 4096 || size != align || addr & (align - 1))
|
|
|
|
|
{
|
|
|
|
|
throw EXCEPTION("Invalid arguments (addr=0x%x, size=0x%x)", addr, size);
|
|
|
|
|
}
|
2015-02-07 00:39:51 +01:00
|
|
|
|
2015-11-26 09:06:29 +01:00
|
|
|
if (g_reservation_owner != thread_ctrl::get_current() || g_reservation_addr != addr || g_reservation_size != size)
|
2015-02-07 00:39:51 +01:00
|
|
|
{
|
|
|
|
|
// atomic update failed
|
|
|
|
|
return false;
|
|
|
|
|
}
|
|
|
|
|
|
2015-02-08 16:25:50 +01:00
|
|
|
// change memory protection to no access
|
|
|
|
|
_reservation_set(addr, true);
|
|
|
|
|
|
2015-02-07 00:39:51 +01:00
|
|
|
// update memory using privileged access
|
2015-09-26 22:46:04 +02:00
|
|
|
std::memcpy(vm::base_priv(addr), data, size);
|
2015-02-07 00:39:51 +01:00
|
|
|
|
|
|
|
|
// free the reservation and restore memory protection
|
2015-02-07 14:46:42 +01:00
|
|
|
_reservation_break(addr);
|
2015-02-07 00:39:51 +01:00
|
|
|
|
2015-07-15 13:58:13 +02:00
|
|
|
// notify waiter
|
2016-04-25 12:49:12 +02:00
|
|
|
lock.unlock(), vm::notify_at(addr, size);
|
2015-07-15 13:58:13 +02:00
|
|
|
|
2015-02-07 00:39:51 +01:00
|
|
|
// atomic update succeeded
|
|
|
|
|
return true;
|
|
|
|
|
}
|
|
|
|
|
|
2015-02-17 01:08:23 +01:00
|
|
|
bool reservation_query(u32 addr, u32 size, bool is_writing, std::function<bool()> callback)
|
2015-02-07 00:39:51 +01:00
|
|
|
{
|
2015-07-15 13:58:13 +02:00
|
|
|
std::unique_lock<reservation_mutex_t> lock(g_reservation_mutex);
|
2015-02-07 00:39:51 +01:00
|
|
|
|
2015-02-13 15:04:03 +01:00
|
|
|
if (!check_addr(addr))
|
2015-02-07 00:39:51 +01:00
|
|
|
{
|
2015-02-13 15:04:03 +01:00
|
|
|
return false;
|
2015-02-07 00:39:51 +01:00
|
|
|
}
|
|
|
|
|
|
2015-02-17 01:08:23 +01:00
|
|
|
// check if current reservation and address may overlap
|
|
|
|
|
if (g_reservation_addr >> 12 == addr >> 12 && is_writing)
|
2015-02-08 16:25:50 +01:00
|
|
|
{
|
2015-07-15 13:58:13 +02:00
|
|
|
const bool result = callback();
|
|
|
|
|
|
|
|
|
|
if (result && size && addr + size - 1 >= g_reservation_addr && g_reservation_addr + g_reservation_size - 1 >= addr)
|
2015-02-16 02:53:53 +01:00
|
|
|
{
|
2015-07-15 13:58:13 +02:00
|
|
|
const u32 raddr = g_reservation_addr;
|
|
|
|
|
const u32 rsize = g_reservation_size;
|
|
|
|
|
|
2015-02-17 01:08:23 +01:00
|
|
|
// break the reservation if overlap
|
2015-07-15 13:58:13 +02:00
|
|
|
if ((g_tls_did_break_reservation = _reservation_break(addr)))
|
|
|
|
|
{
|
2016-04-25 12:49:12 +02:00
|
|
|
lock.unlock(), vm::notify_at(raddr, rsize);
|
2015-07-15 13:58:13 +02:00
|
|
|
}
|
2015-02-16 02:53:53 +01:00
|
|
|
}
|
2015-07-15 13:58:13 +02:00
|
|
|
|
|
|
|
|
return result;
|
2015-02-08 16:25:50 +01:00
|
|
|
}
|
|
|
|
|
|
2015-02-07 00:39:51 +01:00
|
|
|
return true;
|
|
|
|
|
}
|
|
|
|
|
|
2016-05-13 16:01:48 +02:00
|
|
|
bool reservation_test(thread_ctrl* current)
|
2015-07-12 13:52:55 +02:00
|
|
|
{
|
|
|
|
|
const auto owner = g_reservation_owner;
|
|
|
|
|
|
2015-07-15 20:11:32 +02:00
|
|
|
return owner && owner == current;
|
2015-07-12 13:52:55 +02:00
|
|
|
}
|
|
|
|
|
|
2015-02-07 00:39:51 +01:00
|
|
|
void reservation_free()
|
|
|
|
|
{
|
2016-04-25 12:49:12 +02:00
|
|
|
auto thread = thread_ctrl::get_current();
|
|
|
|
|
|
|
|
|
|
if (reservation_test(thread))
|
2015-02-07 00:39:51 +01:00
|
|
|
{
|
2015-07-04 21:23:10 +02:00
|
|
|
std::lock_guard<reservation_mutex_t> lock(g_reservation_mutex);
|
|
|
|
|
|
2016-04-25 12:49:12 +02:00
|
|
|
if (g_reservation_owner && g_reservation_owner == thread)
|
2015-07-19 03:56:33 +02:00
|
|
|
{
|
|
|
|
|
g_tls_did_break_reservation = _reservation_break(g_reservation_addr);
|
|
|
|
|
}
|
2015-02-07 00:39:51 +01:00
|
|
|
}
|
|
|
|
|
}
|
2014-08-01 18:27:48 +02:00
|
|
|
|
2015-02-07 14:46:42 +01:00
|
|
|
void reservation_op(u32 addr, u32 size, std::function<void()> proc)
|
|
|
|
|
{
|
2015-07-15 13:58:13 +02:00
|
|
|
std::unique_lock<reservation_mutex_t> lock(g_reservation_mutex);
|
2015-07-11 22:44:53 +02:00
|
|
|
|
2015-07-21 22:14:04 +02:00
|
|
|
const u64 align = 0x80000000ull >> cntlz32(size);
|
2015-02-07 14:46:42 +01:00
|
|
|
|
2015-07-21 22:14:04 +02:00
|
|
|
if (!size || !addr || size > 4096 || size != align || addr & (align - 1))
|
|
|
|
|
{
|
|
|
|
|
throw EXCEPTION("Invalid arguments (addr=0x%x, size=0x%x)", addr, size);
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
g_tls_did_break_reservation = false;
|
|
|
|
|
|
|
|
|
|
// check and possibly break previous reservation
|
2015-11-26 09:06:29 +01:00
|
|
|
if (g_reservation_owner != thread_ctrl::get_current() || g_reservation_addr != addr || g_reservation_size != size)
|
2015-02-07 14:46:42 +01:00
|
|
|
{
|
2015-07-19 03:56:33 +02:00
|
|
|
if (g_reservation_owner)
|
|
|
|
|
{
|
|
|
|
|
_reservation_break(g_reservation_addr);
|
|
|
|
|
}
|
2015-07-21 22:14:04 +02:00
|
|
|
|
|
|
|
|
g_tls_did_break_reservation = true;
|
2015-02-07 14:46:42 +01:00
|
|
|
}
|
|
|
|
|
|
2015-02-08 16:25:50 +01:00
|
|
|
// change memory protection to no access
|
|
|
|
|
_reservation_set(addr, true);
|
2015-02-07 16:35:54 +01:00
|
|
|
|
2015-02-07 14:46:42 +01:00
|
|
|
// set additional information
|
|
|
|
|
g_reservation_addr = addr;
|
2015-02-08 14:38:08 +01:00
|
|
|
g_reservation_size = size;
|
2015-11-26 09:06:29 +01:00
|
|
|
g_reservation_owner = thread_ctrl::get_current();
|
2015-02-07 14:46:42 +01:00
|
|
|
|
2015-02-08 16:25:50 +01:00
|
|
|
// may not be necessary
|
2015-02-07 21:06:35 +01:00
|
|
|
_mm_mfence();
|
|
|
|
|
|
2015-02-07 14:46:42 +01:00
|
|
|
// do the operation
|
|
|
|
|
proc();
|
|
|
|
|
|
|
|
|
|
// remove the reservation
|
|
|
|
|
_reservation_break(addr);
|
2015-07-15 13:58:13 +02:00
|
|
|
|
|
|
|
|
// notify waiter
|
2016-04-25 12:49:12 +02:00
|
|
|
lock.unlock(), vm::notify_at(addr, size);
|
2015-02-07 14:46:42 +01:00
|
|
|
}
|
|
|
|
|
|
2015-07-11 22:44:53 +02:00
|
|
|
void _page_map(u32 addr, u32 size, u8 flags)
|
2014-08-01 18:27:48 +02:00
|
|
|
{
|
2015-09-26 22:46:04 +02:00
|
|
|
if (!size || (size | addr) % 4096 || flags & page_allocated)
|
|
|
|
|
{
|
|
|
|
|
throw EXCEPTION("Invalid arguments (addr=0x%x, size=0x%x)", addr, size);
|
|
|
|
|
}
|
2015-02-13 15:04:03 +01:00
|
|
|
|
|
|
|
|
for (u32 i = addr / 4096; i < addr / 4096 + size / 4096; i++)
|
|
|
|
|
{
|
2015-09-18 00:41:14 +02:00
|
|
|
if (g_pages[i])
|
2015-02-13 15:04:03 +01:00
|
|
|
{
|
2015-07-02 03:54:36 +02:00
|
|
|
throw EXCEPTION("Memory already mapped (addr=0x%x, size=0x%x, flags=0x%x, current_addr=0x%x)", addr, size, flags, i * 4096);
|
2015-02-13 15:04:03 +01:00
|
|
|
}
|
|
|
|
|
}
|
|
|
|
|
|
2015-09-26 22:46:04 +02:00
|
|
|
void* real_addr = vm::base(addr);
|
|
|
|
|
void* priv_addr = vm::base_priv(addr);
|
2015-02-13 15:04:03 +01:00
|
|
|
|
|
|
|
|
#ifdef _WIN32
|
|
|
|
|
auto protection = flags & page_writable ? PAGE_READWRITE : (flags & page_readable ? PAGE_READONLY : PAGE_NOACCESS);
|
2015-09-26 22:46:04 +02:00
|
|
|
if (!::VirtualAlloc(priv_addr, size, MEM_COMMIT, PAGE_READWRITE) || !::VirtualAlloc(real_addr, size, MEM_COMMIT, protection))
|
2015-02-13 15:04:03 +01:00
|
|
|
#else
|
|
|
|
|
auto protection = flags & page_writable ? PROT_WRITE | PROT_READ : (flags & page_readable ? PROT_READ : PROT_NONE);
|
2015-09-26 22:46:04 +02:00
|
|
|
if (::mprotect(priv_addr, size, PROT_READ | PROT_WRITE) || ::mprotect(real_addr, size, protection))
|
2015-02-13 15:04:03 +01:00
|
|
|
#endif
|
|
|
|
|
{
|
2015-07-02 03:54:36 +02:00
|
|
|
throw EXCEPTION("System failure (addr=0x%x, size=0x%x, flags=0x%x)", addr, size, flags);
|
2015-02-13 15:04:03 +01:00
|
|
|
}
|
|
|
|
|
|
|
|
|
|
for (u32 i = addr / 4096; i < addr / 4096 + size / 4096; i++)
|
|
|
|
|
{
|
2015-07-11 22:44:53 +02:00
|
|
|
if (g_pages[i].exchange(flags | page_allocated))
|
2015-02-13 15:04:03 +01:00
|
|
|
{
|
2015-07-02 03:54:36 +02:00
|
|
|
throw EXCEPTION("Concurrent access (addr=0x%x, size=0x%x, flags=0x%x, current_addr=0x%x)", addr, size, flags, i * 4096);
|
2015-02-13 15:04:03 +01:00
|
|
|
}
|
|
|
|
|
}
|
|
|
|
|
|
2015-09-26 22:46:04 +02:00
|
|
|
std::memset(priv_addr, 0, size); // ???
|
2015-02-13 15:04:03 +01:00
|
|
|
}
|
|
|
|
|
|
|
|
|
|
bool page_protect(u32 addr, u32 size, u8 flags_test, u8 flags_set, u8 flags_clear)
|
|
|
|
|
{
|
2015-07-11 22:44:53 +02:00
|
|
|
std::lock_guard<reservation_mutex_t> lock(g_reservation_mutex);
|
|
|
|
|
|
2015-09-26 22:46:04 +02:00
|
|
|
if (!size || (size | addr) % 4096)
|
|
|
|
|
{
|
|
|
|
|
throw EXCEPTION("Invalid arguments (addr=0x%x, size=0x%x)", addr, size);
|
|
|
|
|
}
|
2015-02-13 15:04:03 +01:00
|
|
|
|
2015-09-26 22:46:04 +02:00
|
|
|
const u8 flags_inv = flags_set & flags_clear;
|
2015-02-13 15:04:03 +01:00
|
|
|
|
|
|
|
|
flags_test |= page_allocated;
|
|
|
|
|
|
|
|
|
|
for (u32 i = addr / 4096; i < addr / 4096 + size / 4096; i++)
|
|
|
|
|
{
|
2015-09-18 00:41:14 +02:00
|
|
|
if ((g_pages[i] & flags_test) != (flags_test | page_allocated))
|
2015-02-13 15:04:03 +01:00
|
|
|
{
|
|
|
|
|
return false;
|
|
|
|
|
}
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
if (!flags_inv && !flags_set && !flags_clear)
|
|
|
|
|
{
|
|
|
|
|
return true;
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
for (u32 i = addr / 4096; i < addr / 4096 + size / 4096; i++)
|
|
|
|
|
{
|
|
|
|
|
_reservation_break(i * 4096);
|
|
|
|
|
|
2016-02-01 22:52:49 +01:00
|
|
|
const u8 f1 = g_pages[i].fetch_or(flags_set & ~flags_inv) & (page_writable | page_readable);
|
|
|
|
|
g_pages[i].fetch_and(~(flags_clear & ~flags_inv));
|
2015-07-11 22:44:53 +02:00
|
|
|
const u8 f2 = (g_pages[i] ^= flags_inv) & (page_writable | page_readable);
|
2015-02-13 15:04:03 +01:00
|
|
|
|
|
|
|
|
if (f1 != f2)
|
|
|
|
|
{
|
2015-09-26 22:46:04 +02:00
|
|
|
void* real_addr = vm::base(i * 4096);
|
2015-02-13 15:04:03 +01:00
|
|
|
|
|
|
|
|
#ifdef _WIN32
|
|
|
|
|
DWORD old;
|
|
|
|
|
|
|
|
|
|
auto protection = f2 & page_writable ? PAGE_READWRITE : (f2 & page_readable ? PAGE_READONLY : PAGE_NOACCESS);
|
2015-09-26 22:46:04 +02:00
|
|
|
if (!::VirtualProtect(real_addr, 4096, protection, &old))
|
2015-02-13 15:04:03 +01:00
|
|
|
#else
|
|
|
|
|
auto protection = f2 & page_writable ? PROT_WRITE | PROT_READ : (f2 & page_readable ? PROT_READ : PROT_NONE);
|
2015-09-26 22:46:04 +02:00
|
|
|
if (::mprotect(real_addr, 4096, protection))
|
2015-02-13 15:04:03 +01:00
|
|
|
#endif
|
|
|
|
|
{
|
2015-07-02 03:54:36 +02:00
|
|
|
throw EXCEPTION("System failure (addr=0x%x, size=0x%x, flags_test=0x%x, flags_set=0x%x, flags_clear=0x%x)", addr, size, flags_test, flags_set, flags_clear);
|
2015-02-13 15:04:03 +01:00
|
|
|
}
|
|
|
|
|
}
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
return true;
|
|
|
|
|
}
|
|
|
|
|
|
2015-07-12 13:52:55 +02:00
|
|
|
void _page_unmap(u32 addr, u32 size)
|
2015-02-13 15:04:03 +01:00
|
|
|
{
|
2015-09-26 22:46:04 +02:00
|
|
|
if (!size || (size | addr) % 4096)
|
|
|
|
|
{
|
|
|
|
|
throw EXCEPTION("Invalid arguments (addr=0x%x, size=0x%x)", addr, size);
|
|
|
|
|
}
|
2015-07-11 22:44:53 +02:00
|
|
|
|
2015-02-13 15:04:03 +01:00
|
|
|
for (u32 i = addr / 4096; i < addr / 4096 + size / 4096; i++)
|
|
|
|
|
{
|
2015-09-18 00:41:14 +02:00
|
|
|
if ((g_pages[i] & page_allocated) == 0)
|
2015-02-13 15:04:03 +01:00
|
|
|
{
|
2015-07-02 03:54:36 +02:00
|
|
|
throw EXCEPTION("Memory not mapped (addr=0x%x, size=0x%x, current_addr=0x%x)", addr, size, i * 4096);
|
2015-02-13 15:04:03 +01:00
|
|
|
}
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
for (u32 i = addr / 4096; i < addr / 4096 + size / 4096; i++)
|
|
|
|
|
{
|
|
|
|
|
_reservation_break(i * 4096);
|
|
|
|
|
|
2015-07-11 22:44:53 +02:00
|
|
|
if (!(g_pages[i].exchange(0) & page_allocated))
|
2015-02-13 15:04:03 +01:00
|
|
|
{
|
2015-07-02 03:54:36 +02:00
|
|
|
throw EXCEPTION("Concurrent access (addr=0x%x, size=0x%x, current_addr=0x%x)", addr, size, i * 4096);
|
2015-02-13 15:04:03 +01:00
|
|
|
}
|
|
|
|
|
}
|
|
|
|
|
|
2015-09-26 22:46:04 +02:00
|
|
|
void* real_addr = vm::base(addr);
|
|
|
|
|
void* priv_addr = vm::base_priv(addr);
|
2015-02-13 15:04:03 +01:00
|
|
|
|
|
|
|
|
#ifdef _WIN32
|
|
|
|
|
DWORD old;
|
|
|
|
|
|
2015-09-26 22:46:04 +02:00
|
|
|
if (!::VirtualProtect(real_addr, size, PAGE_NOACCESS, &old) || !::VirtualProtect(priv_addr, size, PAGE_NOACCESS, &old))
|
2015-02-13 15:04:03 +01:00
|
|
|
#else
|
2015-09-26 22:46:04 +02:00
|
|
|
if (::mprotect(real_addr, size, PROT_NONE) || ::mprotect(priv_addr, size, PROT_NONE))
|
2015-02-13 15:04:03 +01:00
|
|
|
#endif
|
|
|
|
|
{
|
2015-07-02 03:54:36 +02:00
|
|
|
throw EXCEPTION("System failure (addr=0x%x, size=0x%x)", addr, size);
|
2015-02-13 15:04:03 +01:00
|
|
|
}
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
bool check_addr(u32 addr, u32 size)
|
|
|
|
|
{
|
|
|
|
|
if (addr + (size - 1) < addr)
|
|
|
|
|
{
|
|
|
|
|
return false;
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
for (u32 i = addr / 4096; i <= (addr + size - 1) / 4096; i++)
|
|
|
|
|
{
|
2015-09-18 00:41:14 +02:00
|
|
|
if ((g_pages[i] & page_allocated) == 0)
|
2015-02-13 15:04:03 +01:00
|
|
|
{
|
|
|
|
|
return false;
|
|
|
|
|
}
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
return true;
|
2014-08-01 18:27:48 +02:00
|
|
|
}
|
|
|
|
|
|
2015-07-11 22:44:53 +02:00
|
|
|
u32 alloc(u32 size, memory_location_t location, u32 align)
|
2014-08-01 18:27:48 +02:00
|
|
|
{
|
2015-07-11 22:44:53 +02:00
|
|
|
const auto block = get(location);
|
|
|
|
|
|
|
|
|
|
if (!block)
|
|
|
|
|
{
|
|
|
|
|
throw EXCEPTION("Invalid memory location (%d)", location);
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
return block->alloc(size, align);
|
2014-08-01 18:27:48 +02:00
|
|
|
}
|
|
|
|
|
|
2015-07-11 22:44:53 +02:00
|
|
|
u32 falloc(u32 addr, u32 size, memory_location_t location)
|
2014-08-01 18:27:48 +02:00
|
|
|
{
|
2015-07-11 22:44:53 +02:00
|
|
|
const auto block = get(location, addr);
|
|
|
|
|
|
|
|
|
|
if (!block)
|
|
|
|
|
{
|
|
|
|
|
throw EXCEPTION("Invalid memory location (%d, addr=0x%x)", location, addr);
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
return block->falloc(addr, size);
|
2014-08-01 18:27:48 +02:00
|
|
|
}
|
|
|
|
|
|
2015-07-11 22:44:53 +02:00
|
|
|
bool dealloc(u32 addr, memory_location_t location)
|
2014-08-01 18:27:48 +02:00
|
|
|
{
|
2015-07-11 22:44:53 +02:00
|
|
|
const auto block = get(location, addr);
|
|
|
|
|
|
|
|
|
|
if (!block)
|
|
|
|
|
{
|
|
|
|
|
throw EXCEPTION("Invalid memory location (%d, addr=0x%x)", location, addr);
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
return block->dealloc(addr);
|
2014-08-01 18:27:48 +02:00
|
|
|
}
|
|
|
|
|
|
2015-08-19 13:04:58 +02:00
|
|
|
void dealloc_verbose_nothrow(u32 addr, memory_location_t location) noexcept
|
|
|
|
|
{
|
|
|
|
|
const auto block = get(location, addr);
|
|
|
|
|
|
|
|
|
|
if (!block)
|
|
|
|
|
{
|
2015-11-26 09:06:29 +01:00
|
|
|
LOG_ERROR(MEMORY, "vm::dealloc(): invalid memory location (%d, addr=0x%x)\n", location, addr);
|
2015-08-19 13:04:58 +02:00
|
|
|
return;
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
if (!block->dealloc(addr))
|
|
|
|
|
{
|
2015-11-26 09:06:29 +01:00
|
|
|
LOG_ERROR(MEMORY, "vm::dealloc(): deallocation failed (addr=0x%x)\n", addr);
|
2015-08-19 13:04:58 +02:00
|
|
|
return;
|
|
|
|
|
}
|
|
|
|
|
}
|
|
|
|
|
|
2015-07-11 22:44:53 +02:00
|
|
|
bool block_t::try_alloc(u32 addr, u32 size)
|
2014-11-19 15:16:30 +01:00
|
|
|
{
|
2015-07-11 22:44:53 +02:00
|
|
|
// check if memory area is already mapped
|
|
|
|
|
for (u32 i = addr / 4096; i <= (addr + size - 1) / 4096; i++)
|
|
|
|
|
{
|
2015-09-18 00:41:14 +02:00
|
|
|
if (g_pages[i])
|
2015-07-11 22:44:53 +02:00
|
|
|
{
|
|
|
|
|
return false;
|
|
|
|
|
}
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
// try to reserve "physical" memory
|
|
|
|
|
if (!used.atomic_op([=](u32& used) -> bool
|
|
|
|
|
{
|
|
|
|
|
if (used > this->size)
|
|
|
|
|
{
|
|
|
|
|
throw EXCEPTION("Unexpected memory amount used (0x%x)", used);
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
if (used + size > this->size)
|
|
|
|
|
{
|
|
|
|
|
return false;
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
used += size;
|
|
|
|
|
|
|
|
|
|
return true;
|
|
|
|
|
}))
|
|
|
|
|
{
|
|
|
|
|
return false;
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
// map memory pages
|
|
|
|
|
_page_map(addr, size, page_readable | page_writable);
|
|
|
|
|
|
|
|
|
|
// add entry
|
|
|
|
|
m_map[addr] = size;
|
|
|
|
|
|
|
|
|
|
return true;
|
2014-11-19 15:16:30 +01:00
|
|
|
}
|
|
|
|
|
|
2015-07-11 22:44:53 +02:00
|
|
|
block_t::~block_t()
|
2014-11-19 15:16:30 +01:00
|
|
|
{
|
2015-07-12 13:52:55 +02:00
|
|
|
std::lock_guard<reservation_mutex_t> lock(g_reservation_mutex);
|
|
|
|
|
|
2015-07-11 22:44:53 +02:00
|
|
|
// deallocate all memory
|
|
|
|
|
for (auto& entry : m_map)
|
|
|
|
|
{
|
2015-07-21 22:14:04 +02:00
|
|
|
_page_unmap(entry.first, entry.second);
|
2015-07-11 22:44:53 +02:00
|
|
|
}
|
2014-11-19 15:16:30 +01:00
|
|
|
}
|
|
|
|
|
|
2015-07-11 22:44:53 +02:00
|
|
|
u32 block_t::alloc(u32 size, u32 align)
|
2014-11-19 15:16:30 +01:00
|
|
|
{
|
2015-07-11 22:44:53 +02:00
|
|
|
std::lock_guard<std::mutex> lock(m_mutex);
|
|
|
|
|
|
|
|
|
|
// align to minimal page size
|
|
|
|
|
size = ::align(size, 4096);
|
|
|
|
|
|
|
|
|
|
// check alignment (it's page allocation, so passing small values there is just silly)
|
|
|
|
|
if (align < 4096 || align != (0x80000000u >> cntlz32(align)))
|
2014-11-19 15:16:30 +01:00
|
|
|
{
|
2015-07-11 22:44:53 +02:00
|
|
|
throw EXCEPTION("Invalid alignment (size=0x%x, align=0x%x)", size, align);
|
2014-11-19 15:16:30 +01:00
|
|
|
}
|
2015-07-11 22:44:53 +02:00
|
|
|
|
|
|
|
|
// return if size is invalid
|
|
|
|
|
if (!size || size > this->size)
|
2014-11-19 15:16:30 +01:00
|
|
|
{
|
2015-07-12 13:52:55 +02:00
|
|
|
return 0;
|
2014-11-19 15:16:30 +01:00
|
|
|
}
|
2015-07-11 22:44:53 +02:00
|
|
|
|
|
|
|
|
// search for an appropriate place (unoptimized)
|
|
|
|
|
for (u32 addr = ::align(this->addr, align); addr < this->addr + this->size - 1; addr += align)
|
2014-11-19 15:16:30 +01:00
|
|
|
{
|
2015-07-11 22:44:53 +02:00
|
|
|
if (try_alloc(addr, size))
|
|
|
|
|
{
|
|
|
|
|
return addr;
|
|
|
|
|
}
|
2015-07-12 13:52:55 +02:00
|
|
|
|
2015-09-18 00:41:14 +02:00
|
|
|
if (used + size > this->size)
|
2015-07-12 13:52:55 +02:00
|
|
|
{
|
|
|
|
|
return 0;
|
|
|
|
|
}
|
2014-11-19 15:16:30 +01:00
|
|
|
}
|
|
|
|
|
|
2015-07-12 13:52:55 +02:00
|
|
|
return 0;
|
2015-07-11 22:44:53 +02:00
|
|
|
}
|
|
|
|
|
|
|
|
|
|
u32 block_t::falloc(u32 addr, u32 size)
|
|
|
|
|
{
|
|
|
|
|
std::lock_guard<std::mutex> lock(m_mutex);
|
|
|
|
|
|
|
|
|
|
// align to minimal page size
|
|
|
|
|
size = ::align(size, 4096);
|
|
|
|
|
|
|
|
|
|
// return if addr or size is invalid
|
|
|
|
|
if (!size || size > this->size || addr < this->addr || addr + size - 1 >= this->addr + this->size - 1)
|
2014-11-19 15:16:30 +01:00
|
|
|
{
|
2015-07-12 13:52:55 +02:00
|
|
|
return 0;
|
2014-11-19 15:16:30 +01:00
|
|
|
}
|
2015-07-11 22:44:53 +02:00
|
|
|
|
|
|
|
|
if (!try_alloc(addr, size))
|
2014-11-19 15:16:30 +01:00
|
|
|
{
|
2015-07-12 13:52:55 +02:00
|
|
|
return 0;
|
2014-11-19 15:16:30 +01:00
|
|
|
}
|
2015-07-11 22:44:53 +02:00
|
|
|
|
|
|
|
|
return addr;
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
bool block_t::dealloc(u32 addr)
|
|
|
|
|
{
|
|
|
|
|
std::lock_guard<std::mutex> lock(m_mutex);
|
|
|
|
|
|
|
|
|
|
const auto found = m_map.find(addr);
|
|
|
|
|
|
|
|
|
|
if (found != m_map.end())
|
2014-11-19 15:16:30 +01:00
|
|
|
{
|
2015-07-11 22:44:53 +02:00
|
|
|
const u32 size = found->second;
|
|
|
|
|
|
|
|
|
|
// remove entry
|
|
|
|
|
m_map.erase(found);
|
|
|
|
|
|
|
|
|
|
// return "physical" memory
|
|
|
|
|
used -= size;
|
|
|
|
|
|
2015-07-12 13:52:55 +02:00
|
|
|
// unmap memory pages
|
2015-07-21 22:14:04 +02:00
|
|
|
std::lock_guard<reservation_mutex_t>{ g_reservation_mutex }, _page_unmap(addr, size);
|
2015-07-12 13:52:55 +02:00
|
|
|
|
2015-07-11 22:44:53 +02:00
|
|
|
return true;
|
2014-11-19 15:16:30 +01:00
|
|
|
}
|
|
|
|
|
|
2015-07-11 22:44:53 +02:00
|
|
|
return false;
|
|
|
|
|
}
|
|
|
|
|
|
2015-07-12 13:52:55 +02:00
|
|
|
std::shared_ptr<block_t> map(u32 addr, u32 size, u64 flags)
|
2015-07-11 22:44:53 +02:00
|
|
|
{
|
|
|
|
|
std::lock_guard<reservation_mutex_t> lock(g_reservation_mutex);
|
2014-11-19 15:16:30 +01:00
|
|
|
|
2015-07-12 13:52:55 +02:00
|
|
|
if (!size || (size | addr) % 4096)
|
2014-11-19 15:16:30 +01:00
|
|
|
{
|
2015-07-12 13:52:55 +02:00
|
|
|
throw EXCEPTION("Invalid arguments (addr=0x%x, size=0x%x)", addr, size);
|
2014-11-19 15:16:30 +01:00
|
|
|
}
|
2015-07-11 22:44:53 +02:00
|
|
|
|
|
|
|
|
for (auto& block : g_locations)
|
2014-11-19 15:16:30 +01:00
|
|
|
{
|
2015-07-11 22:44:53 +02:00
|
|
|
if (block->addr >= addr && block->addr <= addr + size - 1)
|
|
|
|
|
{
|
|
|
|
|
return nullptr;
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
if (addr >= block->addr && addr <= block->addr + block->size - 1)
|
|
|
|
|
{
|
|
|
|
|
return nullptr;
|
|
|
|
|
}
|
2014-11-19 15:16:30 +01:00
|
|
|
}
|
2015-07-11 22:44:53 +02:00
|
|
|
|
|
|
|
|
for (u32 i = addr / 4096; i < addr / 4096 + size / 4096; i++)
|
|
|
|
|
{
|
2015-09-18 00:41:14 +02:00
|
|
|
if (g_pages[i])
|
2015-07-11 22:44:53 +02:00
|
|
|
{
|
2015-07-21 22:14:04 +02:00
|
|
|
throw EXCEPTION("Unexpected pages allocated (current_addr=0x%x)", i * 4096);
|
2015-07-11 22:44:53 +02:00
|
|
|
}
|
|
|
|
|
}
|
|
|
|
|
|
2015-07-12 13:52:55 +02:00
|
|
|
auto block = std::make_shared<block_t>(addr, size, flags);
|
2015-07-11 22:44:53 +02:00
|
|
|
|
|
|
|
|
g_locations.emplace_back(block);
|
|
|
|
|
|
|
|
|
|
return block;
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
std::shared_ptr<block_t> unmap(u32 addr)
|
|
|
|
|
{
|
|
|
|
|
std::lock_guard<reservation_mutex_t> lock(g_reservation_mutex);
|
|
|
|
|
|
|
|
|
|
for (auto it = g_locations.begin(); it != g_locations.end(); it++)
|
2014-11-19 15:16:30 +01:00
|
|
|
{
|
2015-07-11 22:44:53 +02:00
|
|
|
if (*it && (*it)->addr == addr)
|
|
|
|
|
{
|
|
|
|
|
auto block = std::move(*it);
|
|
|
|
|
g_locations.erase(it);
|
|
|
|
|
return block;
|
|
|
|
|
}
|
2014-11-19 15:16:30 +01:00
|
|
|
}
|
|
|
|
|
|
2015-07-11 22:44:53 +02:00
|
|
|
return nullptr;
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
std::shared_ptr<block_t> get(memory_location_t location, u32 addr)
|
|
|
|
|
{
|
|
|
|
|
std::lock_guard<reservation_mutex_t> lock(g_reservation_mutex);
|
|
|
|
|
|
|
|
|
|
if (location != any)
|
|
|
|
|
{
|
|
|
|
|
// return selected location
|
|
|
|
|
if (location < g_locations.size())
|
|
|
|
|
{
|
|
|
|
|
return g_locations[location];
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
return nullptr;
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
// search location by address
|
|
|
|
|
for (auto& block : g_locations)
|
|
|
|
|
{
|
|
|
|
|
if (addr >= block->addr && addr <= block->addr + block->size - 1)
|
|
|
|
|
{
|
|
|
|
|
return block;
|
|
|
|
|
}
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
return nullptr;
|
|
|
|
|
}
|
|
|
|
|
|
2016-04-25 12:49:12 +02:00
|
|
|
extern void start();
|
|
|
|
|
|
2015-07-11 22:44:53 +02:00
|
|
|
namespace ps3
|
|
|
|
|
{
|
2014-11-19 15:16:30 +01:00
|
|
|
void init()
|
|
|
|
|
{
|
2015-07-11 22:44:53 +02:00
|
|
|
g_locations =
|
|
|
|
|
{
|
|
|
|
|
std::make_shared<block_t>(0x00010000, 0x1FFF0000), // main
|
|
|
|
|
std::make_shared<block_t>(0x20000000, 0x10000000), // user
|
|
|
|
|
std::make_shared<block_t>(0xC0000000, 0x10000000), // video
|
|
|
|
|
std::make_shared<block_t>(0xD0000000, 0x10000000), // stack
|
2015-09-26 22:46:04 +02:00
|
|
|
std::make_shared<block_t>(0xE0000000, 0x20000000), // SPU reserved
|
2015-07-11 22:44:53 +02:00
|
|
|
};
|
2015-07-15 13:58:13 +02:00
|
|
|
|
|
|
|
|
vm::start();
|
2014-11-19 15:16:30 +01:00
|
|
|
}
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
namespace psv
|
|
|
|
|
{
|
|
|
|
|
void init()
|
|
|
|
|
{
|
2015-07-11 22:44:53 +02:00
|
|
|
g_locations =
|
|
|
|
|
{
|
|
|
|
|
std::make_shared<block_t>(0x81000000, 0x10000000), // RAM
|
|
|
|
|
std::make_shared<block_t>(0x91000000, 0x2F000000), // user
|
2015-09-26 22:46:04 +02:00
|
|
|
std::make_shared<block_t>(0xC0000000, 0x10000000), // video (arbitrarily)
|
|
|
|
|
std::make_shared<block_t>(0xD0000000, 0x10000000), // stack (arbitrarily)
|
2015-07-11 22:44:53 +02:00
|
|
|
};
|
2015-07-15 13:58:13 +02:00
|
|
|
|
|
|
|
|
vm::start();
|
2014-11-19 15:16:30 +01:00
|
|
|
}
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
namespace psp
|
|
|
|
|
{
|
|
|
|
|
void init()
|
|
|
|
|
{
|
2015-07-11 22:44:53 +02:00
|
|
|
g_locations =
|
|
|
|
|
{
|
|
|
|
|
std::make_shared<block_t>(0x08000000, 0x02000000), // RAM
|
|
|
|
|
std::make_shared<block_t>(0x08800000, 0x01800000), // user
|
|
|
|
|
std::make_shared<block_t>(0x04000000, 0x00200000), // VRAM
|
|
|
|
|
nullptr, // stack
|
|
|
|
|
|
|
|
|
|
std::make_shared<block_t>(0x00010000, 0x00004000), // scratchpad
|
|
|
|
|
std::make_shared<block_t>(0x88000000, 0x00800000), // kernel
|
|
|
|
|
};
|
2015-07-15 13:58:13 +02:00
|
|
|
|
|
|
|
|
vm::start();
|
2014-11-19 15:16:30 +01:00
|
|
|
}
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
void close()
|
2014-08-01 18:27:48 +02:00
|
|
|
{
|
2015-07-11 22:44:53 +02:00
|
|
|
g_locations.clear();
|
2014-08-01 18:27:48 +02:00
|
|
|
}
|
2015-01-07 03:14:00 +01:00
|
|
|
|
2015-09-26 22:46:04 +02:00
|
|
|
u32 stack_push(u32 size, u32 align_v)
|
2015-01-07 03:14:00 +01:00
|
|
|
{
|
2016-02-01 22:52:49 +01:00
|
|
|
if (auto cpu = get_current_cpu_thread()) switch (cpu->type)
|
2015-01-07 03:14:00 +01:00
|
|
|
{
|
2016-02-01 22:52:49 +01:00
|
|
|
case cpu_type::ppu:
|
2015-01-07 03:14:00 +01:00
|
|
|
{
|
2015-09-26 22:46:04 +02:00
|
|
|
PPUThread& context = static_cast<PPUThread&>(*cpu);
|
2015-01-07 03:14:00 +01:00
|
|
|
|
2016-02-01 22:52:49 +01:00
|
|
|
const u32 old_pos = vm::cast(context.GPR[1], HERE);
|
2015-09-26 22:46:04 +02:00
|
|
|
context.GPR[1] -= align(size + 4, 8); // room minimal possible size
|
2015-04-18 03:35:58 +02:00
|
|
|
context.GPR[1] &= ~(align_v - 1); // fix stack alignment
|
2015-01-07 03:14:00 +01:00
|
|
|
|
2015-07-01 00:25:52 +02:00
|
|
|
if (context.GPR[1] < context.stack_addr)
|
2015-01-07 03:14:00 +01:00
|
|
|
{
|
2015-07-04 01:22:24 +02:00
|
|
|
throw EXCEPTION("Stack overflow (size=0x%x, align=0x%x, SP=0x%llx, stack=*0x%x)", size, align_v, old_pos, context.stack_addr);
|
2015-01-07 03:14:00 +01:00
|
|
|
}
|
|
|
|
|
else
|
|
|
|
|
{
|
2015-09-26 22:46:04 +02:00
|
|
|
const u32 addr = static_cast<u32>(context.GPR[1]);
|
|
|
|
|
vm::ps3::_ref<nse_t<u32>>(addr + size) = old_pos;
|
2016-02-01 22:52:49 +01:00
|
|
|
std::memset(vm::base(addr), 0, size);
|
2015-09-26 22:46:04 +02:00
|
|
|
return addr;
|
2015-01-07 03:14:00 +01:00
|
|
|
}
|
|
|
|
|
}
|
|
|
|
|
|
2016-02-01 22:52:49 +01:00
|
|
|
case cpu_type::spu:
|
2015-01-07 03:14:00 +01:00
|
|
|
{
|
2015-09-26 22:46:04 +02:00
|
|
|
SPUThread& context = static_cast<SPUThread&>(*cpu);
|
2015-07-04 01:22:24 +02:00
|
|
|
|
2015-09-26 22:46:04 +02:00
|
|
|
const u32 old_pos = context.gpr[1]._u32[3];
|
|
|
|
|
context.gpr[1]._u32[3] -= align(size + 4, 16);
|
2015-08-26 04:54:06 +02:00
|
|
|
context.gpr[1]._u32[3] &= ~(align_v - 1);
|
2015-07-04 01:22:24 +02:00
|
|
|
|
2015-08-26 04:54:06 +02:00
|
|
|
if (context.gpr[1]._u32[3] >= 0x40000) // extremely rough
|
2015-07-04 01:22:24 +02:00
|
|
|
{
|
|
|
|
|
throw EXCEPTION("Stack overflow (size=0x%x, align=0x%x, SP=LS:0x%05x)", size, align_v, old_pos);
|
|
|
|
|
}
|
|
|
|
|
else
|
|
|
|
|
{
|
2015-09-26 22:46:04 +02:00
|
|
|
const u32 addr = context.gpr[1]._u32[3] + context.offset;
|
|
|
|
|
vm::ps3::_ref<nse_t<u32>>(addr + size) = old_pos;
|
|
|
|
|
return addr;
|
2015-07-04 01:22:24 +02:00
|
|
|
}
|
2015-01-07 03:14:00 +01:00
|
|
|
}
|
|
|
|
|
|
2016-02-01 22:52:49 +01:00
|
|
|
case cpu_type::arm:
|
2015-01-07 03:14:00 +01:00
|
|
|
{
|
2016-02-01 22:52:49 +01:00
|
|
|
ARMv7Thread& context = static_cast<ARMv7Thread&>(*cpu);
|
2015-02-01 14:52:34 +01:00
|
|
|
|
2015-09-26 22:46:04 +02:00
|
|
|
const u32 old_pos = context.SP;
|
|
|
|
|
context.SP -= align(size + 4, 4); // room minimal possible size
|
2015-02-01 14:52:34 +01:00
|
|
|
context.SP &= ~(align_v - 1); // fix stack alignment
|
|
|
|
|
|
2015-07-01 00:25:52 +02:00
|
|
|
if (context.SP < context.stack_addr)
|
2015-02-01 14:52:34 +01:00
|
|
|
{
|
2015-07-04 01:22:24 +02:00
|
|
|
throw EXCEPTION("Stack overflow (size=0x%x, align=0x%x, SP=0x%x, stack=*0x%x)", size, align_v, context.SP, context.stack_addr);
|
2015-02-01 14:52:34 +01:00
|
|
|
}
|
|
|
|
|
else
|
|
|
|
|
{
|
2015-09-26 22:46:04 +02:00
|
|
|
vm::psv::_ref<nse_t<u32>>(context.SP + size) = old_pos;
|
2015-02-01 14:52:34 +01:00
|
|
|
return context.SP;
|
|
|
|
|
}
|
2015-01-07 03:14:00 +01:00
|
|
|
}
|
|
|
|
|
|
|
|
|
|
default:
|
|
|
|
|
{
|
2016-02-01 22:52:49 +01:00
|
|
|
throw EXCEPTION("Invalid thread type (%u)", cpu->type);
|
2015-01-07 03:14:00 +01:00
|
|
|
}
|
|
|
|
|
}
|
2015-09-26 22:46:04 +02:00
|
|
|
|
|
|
|
|
throw EXCEPTION("Invalid thread");
|
2015-01-07 03:14:00 +01:00
|
|
|
}
|
|
|
|
|
|
2016-02-01 22:52:49 +01:00
|
|
|
void stack_pop_verbose(u32 addr, u32 size) noexcept
|
2015-01-07 03:14:00 +01:00
|
|
|
{
|
2016-02-01 22:52:49 +01:00
|
|
|
if (auto cpu = get_current_cpu_thread()) switch (cpu->type)
|
2015-01-07 03:14:00 +01:00
|
|
|
{
|
2016-02-01 22:52:49 +01:00
|
|
|
case cpu_type::ppu:
|
2015-01-07 03:14:00 +01:00
|
|
|
{
|
2015-09-26 22:46:04 +02:00
|
|
|
PPUThread& context = static_cast<PPUThread&>(*cpu);
|
2015-04-18 03:35:58 +02:00
|
|
|
|
2015-07-04 01:22:24 +02:00
|
|
|
if (context.GPR[1] != addr)
|
2015-04-18 03:35:58 +02:00
|
|
|
{
|
2016-02-01 22:52:49 +01:00
|
|
|
LOG_ERROR(MEMORY, "Stack inconsistency (addr=0x%x, SP=0x%llx, size=0x%x)", addr, context.GPR[1], size);
|
|
|
|
|
return;
|
2015-04-18 03:35:58 +02:00
|
|
|
}
|
2015-01-07 03:14:00 +01:00
|
|
|
|
2015-09-26 22:46:04 +02:00
|
|
|
context.GPR[1] = vm::ps3::_ref<nse_t<u32>>(context.GPR[1] + size);
|
2015-01-07 03:14:00 +01:00
|
|
|
return;
|
|
|
|
|
}
|
|
|
|
|
|
2016-02-01 22:52:49 +01:00
|
|
|
case cpu_type::spu:
|
2015-01-07 03:14:00 +01:00
|
|
|
{
|
2015-09-26 22:46:04 +02:00
|
|
|
SPUThread& context = static_cast<SPUThread&>(*cpu);
|
2015-07-04 01:22:24 +02:00
|
|
|
|
2015-08-26 04:54:06 +02:00
|
|
|
if (context.gpr[1]._u32[3] + context.offset != addr)
|
2015-07-04 01:22:24 +02:00
|
|
|
{
|
2016-02-01 22:52:49 +01:00
|
|
|
LOG_ERROR(MEMORY, "Stack inconsistency (addr=0x%x, SP=LS:0x%05x, size=0x%x)", addr, context.gpr[1]._u32[3], size);
|
|
|
|
|
return;
|
2015-07-04 01:22:24 +02:00
|
|
|
}
|
|
|
|
|
|
2015-09-26 22:46:04 +02:00
|
|
|
context.gpr[1]._u32[3] = vm::ps3::_ref<nse_t<u32>>(context.gpr[1]._u32[3] + context.offset + size);
|
2015-01-07 03:14:00 +01:00
|
|
|
return;
|
|
|
|
|
}
|
|
|
|
|
|
2016-02-01 22:52:49 +01:00
|
|
|
case cpu_type::arm:
|
2015-01-07 03:14:00 +01:00
|
|
|
{
|
2016-02-01 22:52:49 +01:00
|
|
|
ARMv7Thread& context = static_cast<ARMv7Thread&>(*cpu);
|
2015-02-02 10:14:49 +01:00
|
|
|
|
2015-07-04 01:22:24 +02:00
|
|
|
if (context.SP != addr)
|
2015-04-18 03:35:58 +02:00
|
|
|
{
|
2016-02-01 22:52:49 +01:00
|
|
|
LOG_ERROR(MEMORY, "Stack inconsistency (addr=0x%x, SP=0x%x, size=0x%x)", addr, context.SP, size);
|
|
|
|
|
return;
|
2015-04-18 03:35:58 +02:00
|
|
|
}
|
|
|
|
|
|
2015-09-26 22:46:04 +02:00
|
|
|
context.SP = vm::psv::_ref<nse_t<u32>>(context.SP + size);
|
2015-01-07 03:14:00 +01:00
|
|
|
return;
|
|
|
|
|
}
|
|
|
|
|
}
|
2016-02-01 22:52:49 +01:00
|
|
|
}
|
2015-09-26 22:46:04 +02:00
|
|
|
|
2016-02-01 22:52:49 +01:00
|
|
|
[[noreturn]] void throw_access_violation(u64 addr, const char* cause)
|
|
|
|
|
{
|
|
|
|
|
throw access_violation(addr, cause);
|
2015-01-07 03:14:00 +01:00
|
|
|
}
|
2015-02-07 16:35:54 +01:00
|
|
|
}
|