entered advanced internal development stage.

1 files changed, 96 insertions, 0 deletions

diff --git a/src/vmount/ntapi_vms_ref_count.c b/src/vmount/ntapi_vms_ref_count.c
new file mode 100644
index 0000000..3be149f
--- /dev/null
+++ b/src/vmount/ntapi_vms_ref_count.c
@@ -0,0 +1,96 @@
+/********************************************************/
+/*  ntapi: Native API core library                      */
+/*  Copyright (C) 2013,2014,2015  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);
+}