Blob Blame History Raw
/********************************************************/
/*  ntapi: Native API core library                      */
/*  Copyright (C) 2013--2016  Z. Gilboa                 */
/*  Released under GPLv2 and GPLv3; see COPYING.NTAPI.  */
/********************************************************/

#include <psxtypes/psxtypes.h>
#include <ntapi/nt_port.h>
#include <ntapi/nt_vmount.h>
#include <ntapi/ntapi.h>
#include "ntapi_impl.h"


typedef struct nt_vms_cache_interface {
	nt_vms_system *		vms_sys;
	struct dalist_ex	cache;
	size_t			alloc_size;
	uintptr_t		buffer[1];
} nt_vms_cache_context;


typedef struct _nt_vms_cache_record {
	void *			hfile;
	uint32_t		dev_name_hash;
	nt_large_integer	index_number;
	intptr_t		client_key;
	intptr_t		server_key;
} nt_vms_cache_record;


int32_t __stdcall	__ntapi_vms_cache_free(
	__in	nt_vms_cache		vms_cache)
{
	int32_t		status;
	void *		region_addr;
	size_t		region_size;

	/* validation */
	if (!vms_cache)
		return NT_STATUS_INVALID_PARAMETER;

	/* free memory */
	region_addr = vms_cache;
	region_size = vms_cache->alloc_size;

	status = __ntapi->zw_free_virtual_memory(
		NT_CURRENT_PROCESS_HANDLE,
		&region_addr,
		&region_size,
		NT_MEM_RELEASE);

	return status;
}

/* vms optional cache functions */
nt_vms_cache __stdcall	__ntapi_vms_cache_alloc(
	__in	nt_vms_system *		vms_sys,
	__in	uint32_t		flags	__reserved,
	__in	void *			options	__reserved,
	__out	int32_t *		status	__optional)
{
	int32_t			_status;
	void *			buffer;
	size_t			buffer_size;
	nt_vms_cache_context *	vms_cache;

	/* unused params */
	(void)flags;
	(void)options;

	/* status */
	if (!status) status = &_status;

	/* validation */
	if (!vms_sys) {
		*status = NT_STATUS_INVALID_PARAMETER;
		return (nt_vms_cache)0;
	}

	/* calculate size */
	buffer_size =  sizeof(nt_vms_cache_context);
	buffer_size += vms_sys->vms_points_cap * (sizeof(nt_vms_cache_record) - sizeof(uintptr_t));

	/* allocate buffer */
	*status = __ntapi->zw_allocate_virtual_memory(
		NT_CURRENT_PROCESS_HANDLE,
		&buffer,
		0,
		&buffer_size,
		NT_MEM_COMMIT,
		NT_PAGE_READWRITE);

	if (*status) return (nt_vms_cache)0;

	/* init vms cache */
	vms_cache = (nt_vms_cache_context *)buffer;
	vms_cache->vms_sys	= vms_sys;
	vms_cache->alloc_size	= buffer_size;

	/* init list */
	*status = dalist_init_ex(
		&vms_cache->cache,
		sizeof(nt_vms_cache_record),
		0x1000,
		__ntapi->zw_allocate_virtual_memory,
		DALIST_MEMFN_NT_ALLOCATE_VIRTUAL_MEMORY);

	if (*status != DALIST_OK) {
		*status = NT_STATUS_UNSUCCESSFUL;
		__ntapi_vms_cache_free(vms_cache);
		return (nt_vms_cache)0;
	}

	/* set list buffer */
	buffer_size -= (size_t)&(((nt_vms_cache_context *)0)->buffer);

	*status = dalist_deposit_memory_block(
		&vms_cache->cache,
		&vms_cache->buffer,
		buffer_size);

	return vms_cache;
}


int32_t __stdcall	__ntapi_vms_cache_record_append(
	__in	nt_vms_cache		cache,
	__in	void *			hfile,
	__in	uint32_t		dev_name_hash,
	__in	nt_large_integer	index_number,
	__in	intptr_t		client_key,
	__in	intptr_t		server_key)
{
	int32_t			status;
	struct dalist_node_ex *	node;
	nt_vms_cache_record *	cache_record;

	status = dalist_get_node_by_key(
			&cache->cache,
			&node,
			(uintptr_t)hfile,
			DALIST_NODE_TYPE_EXISTING,
			(uintptr_t *)0);

	if (status != DALIST_OK)
		status = NT_STATUS_INTERNAL_ERROR;
	else if (node)
		status = NT_STATUS_OBJECTID_EXISTS;
	else {
		status = dalist_get_free_node(&cache->cache,(void **)&node);

		if (status == DALIST_OK) {
			cache_record = (nt_vms_cache_record *)&node->dblock;

			__ntapi->tt_aligned_block_memset(
				node,
				0,
				(uintptr_t)&((struct dalist_node_ex *)0)->dblock + sizeof(*cache_record));

			node->key = (uintptr_t)hfile;

			cache_record->hfile		= hfile;
			cache_record->dev_name_hash	= dev_name_hash;
			cache_record->index_number.quad	= index_number.quad;
			cache_record->client_key	= client_key;
			cache_record->server_key	= server_key;

			status = dalist_insert_node_by_key(
				&cache->cache,
				node);

			if (status != DALIST_OK)
				dalist_deposit_free_node(
					&cache->cache,
					node);
		}
	}

	return status;
}


int32_t __stdcall	__ntapi_vms_cache_record_remove(
	__in	nt_vms_cache		cache,
	__in	void *			hfile,
	__in	uint32_t		dev_name_hash,
	__in	nt_large_integer	index_number)
{
	int32_t			status;
	struct dalist_node_ex *	node;

	(void)dev_name_hash;
	(void)index_number;

	status = dalist_get_node_by_key(
			&cache->cache,
			&node,
			(uintptr_t)hfile,
			DALIST_NODE_TYPE_EXISTING,
			(uintptr_t *)0);

	if (status != DALIST_OK)
		status = NT_STATUS_INTERNAL_ERROR;
	else if (node)
		status = NT_STATUS_INVALID_PARAMETER;
	else {
		status = dalist_discard_node(
			&cache->cache,
			node);

		if (status != DALIST_OK)
			status = NT_STATUS_INTERNAL_ERROR;
	}

	return status;
}