/********************************************************/
/*  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"


static int32_t __stdcall __ntapi_vms_ref_count_inc_dec(
	__in	void *			hvms,
	__in	nt_vms_ref_count_info *	ref_cnt_info,
	__in	int32_t			vms_opcode)
{
	int32_t			status;
	nt_vms_daemon_msg	msg;

	/* msg */
	__ntapi->tt_aligned_block_memset(&msg,0,sizeof(msg));

	msg.header.msg_type	= NT_LPC_NEW_MESSAGE;
	msg.header.data_size	= sizeof(msg.data);
	msg.header.msg_size	= sizeof(msg);
	msg.data.msginfo.opcode	= vms_opcode;

	/* copy ref count info to msg */
	__ntapi->tt_aligned_block_memcpy(
		(uintptr_t *)&(msg.data.refcntinfo),
		(uintptr_t *)ref_cnt_info,
		sizeof(*ref_cnt_info));

	/* zw_request_wait_reply_port */
	status = __ntapi->zw_request_wait_reply_port(
		hvms,
		&msg,
		&msg);

	if (status) return status;

	/* return info */
	__ntapi->tt_aligned_block_memcpy(
		(uintptr_t *)ref_cnt_info,
		(uintptr_t *)&(msg.data.refcntinfo),
		sizeof(*ref_cnt_info));

	/* return vms status */
	return status ? status : msg.data.msginfo.status;
}


int32_t __stdcall __ntapi_vms_ref_count_inc(
	__in	void *			hvms,
	__in	nt_vms_ref_count_info *	ref_cnt_info)
{
	return __ntapi_vms_ref_count_inc_dec(
		hvms,
		ref_cnt_info,
		NT_VMS_REF_COUNT_INC);
}


int32_t __stdcall __ntapi_vms_ref_count_dec(
	__in	void *			hvms,
	__in	nt_vms_ref_count_info *	ref_cnt_info)
{
	return __ntapi_vms_ref_count_inc_dec(
		hvms,
		ref_cnt_info,
		NT_VMS_REF_COUNT_DEC);
}


int32_t __stdcall __ntapi_vms_point_detach(
	__in	void *			hvms,
	__in	nt_vms_ref_count_info *	ref_cnt_info)
{
	return __ntapi_vms_ref_count_inc_dec(
		hvms,
		ref_cnt_info,
		NT_VMS_POINT_DETACH);
}


int32_t __stdcall __ntapi_vms_point_get_handles(
	__in	void *			hvms,
	__in	nt_vms_ref_count_info *	ref_cnt_info)
{
	return __ntapi_vms_ref_count_inc_dec(
		hvms,
		ref_cnt_info,
		NT_VMS_POINT_GET_HANDLES);
}