tty client-side interfaces(): added ntapi_tty_query_server_pts_slot_info().

3 files changed, 52 insertions, 0 deletions

diff --git a/src/internal/ntapi.c b/src/internal/ntapi.c
index 95abd45..20b49fb 100644
--- a/src/internal/ntapi.c
+++ b/src/internal/ntapi.c
@@ -392,6 +392,7 @@ static int32_t __fastcall __ntapi_init_once(ntapi_vtbl ** pvtbl)
 	__ntapi->tty_query_information_section			= __ntapi_tty_query_information_section;
 	__ntapi->tty_query_information_server			= __ntapi_tty_query_information_server;
 	__ntapi->tty_query_information_service			= __ntapi_tty_query_information_service;
+	__ntapi->tty_query_server_pts_slot_info			= __ntapi_tty_query_server_pts_slot_info;
 	__ntapi->tty_request_peer				= __ntapi_tty_request_peer;
 	__ntapi->tty_vms_query					= __ntapi_tty_vms_query;
 	__ntapi->tty_vms_request 				= __ntapi_tty_vms_request;
diff --git a/src/internal/ntapi_fnapi.h b/src/internal/ntapi_fnapi.h
index 3477127..42bbdbf 100644
--- a/src/internal/ntapi_fnapi.h
+++ b/src/internal/ntapi_fnapi.h
@@ -258,6 +258,7 @@ ntapi_tty_client_process_register		__ntapi_tty_client_process_register;
 ntapi_tty_query_information_section		__ntapi_tty_query_information_section;
 ntapi_tty_query_information_server		__ntapi_tty_query_information_server;
 ntapi_tty_query_information_service		__ntapi_tty_query_information_service;
+ntapi_tty_query_server_pts_slot_info		__ntapi_tty_query_server_pts_slot_info;
 ntapi_tty_request_peer				__ntapi_tty_request_peer;
 ntapi_tty_vms_query				__ntapi_tty_vms_query;
 ntapi_tty_vms_request				__ntapi_tty_vms_request;
diff --git a/src/tty/ntapi_tty_query_server_pts_slot_info.c b/src/tty/ntapi_tty_query_server_pts_slot_info.c
new file mode 100644
index 0000000..228cf1c
--- /dev/null
+++ b/src/tty/ntapi_tty_query_server_pts_slot_info.c
@@ -0,0 +1,50 @@
+/********************************************************/
+/*  ntapi: Native API core library                      */
+/*  Copyright (C) 2013--2018  Z. Gilboa                 */
+/*  Released under GPLv2 and GPLv3; see COPYING.NTAPI.  */
+/********************************************************/
+
+#include <psxtypes/psxtypes.h>
+#include <ntapi/nt_tty.h>
+#include <ntapi/ntapi.h>
+#include "ntapi_impl.h"
+
+#define __MSGBUF_ELEMENTS (NT_LPC_MAX_MSG_DATA_SIZE/sizeof(size_t)) + 1
+
+int32_t	__stdcall __ntapi_tty_query_server_pts_slot_info(
+	__in	void *			hport,
+	__out	 nt_tty_pts_slot_info *	pts_slot_info,
+	__in	intptr_t		blkidx)
+{
+	int32_t			status;
+	nt_tty_server_msg *	msg;
+	uintptr_t		msgbuf[__MSGBUF_ELEMENTS];
+
+	__ntapi->tt_aligned_block_memset(
+		msgbuf,0,
+		sizeof(msgbuf));
+
+	msg				= (nt_tty_server_msg *)msgbuf;
+	msg->header.msg_type		= NT_LPC_NEW_MESSAGE;
+	msg->header.data_size		= sizeof(msg->data);
+	msg->header.msg_size		= sizeof(*msg);
+	msg->data.ttyinfo.exarg         = (void *)blkidx;
+	msg->data.ttyinfo.opcode	= NT_TTY_QUERY_INFORMATION_SERVER;
+	msg->data.srvinfo.info_class	= NT_TTY_SERVER_PTS_SLOT_INFORMATION;
+
+	if ((status = __ntapi->zw_request_wait_reply_port(hport,msg,msg)))
+		return status;
+
+	else if (msg->data.ttyinfo.status)
+		return msg->data.ttyinfo.status;
+
+	if (sizeof(*pts_slot_info) < msg->data.srvinfo.info_length)
+		return NT_STATUS_INTERNAL_ERROR;
+
+	__ntapi->tt_generic_memcpy(
+		pts_slot_info,
+		msg->data.srvinfo.info_buffer,
+		msg->data.srvinfo.info_length);
+
+	return NT_STATUS_SUCCESS;
+}