Add Windows support to tanya.memory.mmappool

This commit is contained in:
Nathan Sashihara 2023-09-22 16:04:36 -04:00
parent 07b388eecb
commit 1c5e18b92e

View File

@ -19,6 +19,22 @@ import tanya.memory.allocator;
import tanya.memory.op;
import tanya.os.error;
version (Windows)
{
import core.sys.windows.basetsd : SIZE_T;
import core.sys.windows.windef : BOOL, DWORD;
import core.sys.windows.winnt : MEM_COMMIT, MEM_RELEASE, PAGE_READWRITE, PVOID;
extern (Windows)
private PVOID VirtualAlloc(PVOID, SIZE_T, DWORD, DWORD)
@nogc nothrow pure @system;
extern (Windows)
private BOOL VirtualFree(shared PVOID, SIZE_T, DWORD)
@nogc nothrow pure @system;
}
else
{
extern(C) pragma(mangle, "mmap")
private void* mapMemory(void *addr, size_t length, int prot, int flags, int fd, off_t offset)
@nogc nothrow pure @system;
@ -26,6 +42,7 @@ private void* mapMemory(void *addr, size_t length, int prot, int flags, int fd,
extern(C) pragma(mangle, "munmap")
private bool unmapMemory(shared void* addr, size_t length)
@nogc nothrow pure @system;
}
/*
* This allocator allocates memory in regions (multiple of 64 KB for example).
@ -192,6 +209,9 @@ final class MmapPool : Allocator
{
block.region.next.prev = block.region.prev;
}
version (Windows)
return VirtualFree(block.region, 0, MEM_RELEASE) != 0;
else
return unmapMemory(block.region, block.region.size) == 0;
}
// Merge blocks if neigbours are free.
@ -380,6 +400,19 @@ final class MmapPool : Allocator
{
return null;
}
version (Windows)
{
void* p = VirtualAlloc(null,
regionSize,
MEM_COMMIT,
PAGE_READWRITE);
if (p is null)
{
return null;
}
}
else
{
void* p = mapMemory(null,
regionSize,
PROT_READ | PROT_WRITE,
@ -390,6 +423,7 @@ final class MmapPool : Allocator
{
return null;
}
}
Region region = cast(Region) p;
region.blocks = 1;