/***********************************************************/
/* ntux: native translation und extension */
/* Copyright (C) 2016--2018 Z. Gilboa */
/* Released under GPLv2 and GPLv3; see COPYING.NTUX. */
/***********************************************************/
#include <psxxfi/xfi_base.h>
#include <psxxfi/xfi_acl.h>
#include <ntapi/nt_memory.h>
#include <ntapi/nt_file.h>
#include <ntapi/nt_process.h>
#include <stddef.h>
#include "ntux_memfn_impl.h"
#include "ntux_driver_impl.h"
void * ntux_calloc(size_t n, size_t size)
{
struct ntux_memory_block block;
if (!n || (size > (size_t)(-1) / n))
return 0;
size *= n;
size += 0xf;
size &= ~(size_t)0xf;
block.addr = 0;
block.size = size + sizeof(block);
if (__xfi_allocate_virtual_memory(
NT_CURRENT_PROCESS_HANDLE,
&block.addr,
0,
&block.size,
NT_MEM_COMMIT,
NT_PAGE_READWRITE))
return 0;
block.used = size;
block.avail = block.size - block.used;
return (char *)block.addr + offsetof(struct ntux_memory_block,buffer);
}
void * ntux_balloc(
struct ntux_memory_block * cache,
struct ntux_memory_block * block,
size_t n, size_t size)
{
char * baddr;
void * addr = 0;
if (!n || (size > (size_t)(-1) / n))
return 0;
if (!cache || !cache->addr)
addr = ntux_calloc(n,size);
size *= n;
size += 0xf;
size &= ~(size_t)0xf;
if (size < block->avail)
addr = ntux_calloc(n,size);
/* newly allocated block? */
if (addr) {
baddr = addr;
baddr -= offsetof(struct ntux_memory_block,buffer);
__xfi_memcpy(
(uintptr_t *)block,
(uintptr_t *)baddr,
sizeof(*block));
return addr;
}
/* cache */
addr = &block->buffer[block->used / sizeof(size_t)];
block->used += size;
block->avail -= size;
return addr;
}
void ntux_free(void * addr)
{
struct ntux_memory_block block;
block.addr = addr;
block.size = 0;
__xfi_free_virtual_memory(
NT_CURRENT_PROCESS_HANDLE,
&block.addr,
&block.size,
NT_MEM_RELEASE);
}