2018-06-08 20:15:56 -04:00
|
|
|
using System;
|
|
|
|
using System.Runtime.InteropServices;
|
|
|
|
|
|
|
|
namespace ChocolArm64.Memory
|
|
|
|
{
|
|
|
|
static class AMemoryWin32
|
|
|
|
{
|
|
|
|
private const int MEM_COMMIT = 0x00001000;
|
|
|
|
private const int MEM_RESERVE = 0x00002000;
|
|
|
|
private const int MEM_WRITE_WATCH = 0x00200000;
|
|
|
|
|
|
|
|
private const int PAGE_READWRITE = 0x04;
|
|
|
|
|
|
|
|
private const int MEM_RELEASE = 0x8000;
|
|
|
|
|
|
|
|
private const int WRITE_WATCH_FLAG_RESET = 1;
|
|
|
|
|
|
|
|
[DllImport("kernel32.dll")]
|
|
|
|
private static extern IntPtr VirtualAlloc(IntPtr lpAddress, IntPtr dwSize, int flAllocationType, int flProtect);
|
|
|
|
|
|
|
|
[DllImport("kernel32.dll")]
|
|
|
|
private static extern bool VirtualFree(IntPtr lpAddress, IntPtr dwSize, int dwFreeType);
|
|
|
|
|
|
|
|
[DllImport("kernel32.dll")]
|
|
|
|
private unsafe static extern int GetWriteWatch(
|
|
|
|
int dwFlags,
|
|
|
|
IntPtr lpBaseAddress,
|
|
|
|
IntPtr dwRegionSize,
|
|
|
|
IntPtr[] lpAddresses,
|
|
|
|
long* lpdwCount,
|
|
|
|
long* lpdwGranularity);
|
|
|
|
|
|
|
|
public static IntPtr Allocate(IntPtr Size)
|
|
|
|
{
|
|
|
|
const int Flags = MEM_COMMIT | MEM_RESERVE | MEM_WRITE_WATCH;
|
|
|
|
|
|
|
|
IntPtr Address = VirtualAlloc(IntPtr.Zero, Size, Flags, PAGE_READWRITE);
|
|
|
|
|
|
|
|
if (Address == IntPtr.Zero)
|
|
|
|
{
|
|
|
|
throw new InvalidOperationException();
|
|
|
|
}
|
|
|
|
|
|
|
|
return Address;
|
|
|
|
}
|
|
|
|
|
|
|
|
public static void Free(IntPtr Address)
|
|
|
|
{
|
|
|
|
VirtualFree(Address, IntPtr.Zero, MEM_RELEASE);
|
|
|
|
}
|
|
|
|
|
2018-07-08 15:55:15 -04:00
|
|
|
public unsafe static int GetPageSize(IntPtr Address, IntPtr Size)
|
2018-06-08 20:15:56 -04:00
|
|
|
{
|
|
|
|
IntPtr[] Addresses = new IntPtr[1];
|
|
|
|
|
|
|
|
long Count = Addresses.Length;
|
|
|
|
|
|
|
|
long Granularity;
|
|
|
|
|
2018-07-08 15:55:15 -04:00
|
|
|
GetWriteWatch(
|
|
|
|
0,
|
|
|
|
Address,
|
|
|
|
Size,
|
|
|
|
Addresses,
|
|
|
|
&Count,
|
|
|
|
&Granularity);
|
|
|
|
|
|
|
|
return (int)Granularity;
|
|
|
|
}
|
|
|
|
|
|
|
|
public unsafe static void IsRegionModified(
|
|
|
|
IntPtr Address,
|
|
|
|
IntPtr Size,
|
|
|
|
IntPtr[] Addresses,
|
|
|
|
out int AddrCount)
|
|
|
|
{
|
|
|
|
long Count = Addresses.Length;
|
|
|
|
|
|
|
|
long Granularity;
|
2018-06-08 20:15:56 -04:00
|
|
|
|
|
|
|
GetWriteWatch(
|
2018-07-08 15:55:15 -04:00
|
|
|
WRITE_WATCH_FLAG_RESET,
|
2018-06-08 20:15:56 -04:00
|
|
|
Address,
|
|
|
|
Size,
|
|
|
|
Addresses,
|
|
|
|
&Count,
|
|
|
|
&Granularity);
|
|
|
|
|
2018-07-08 15:55:15 -04:00
|
|
|
AddrCount = (int)Count;
|
2018-06-08 20:15:56 -04:00
|
|
|
}
|
|
|
|
}
|
|
|
|
}
|