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

6 files changed, 69 insertions, 0 deletions

diff --git a/include/ntapi/nt_tty.h b/include/ntapi/nt_tty.h
index 8d5a18e..f5081a0 100644
--- a/include/ntapi/nt_tty.h
+++ b/include/ntapi/nt_tty.h
@@ -193,6 +193,15 @@ typedef struct __attr_ptr_size_aligned__ _nt_tty_server_basic_info {
 } nt_tty_server_basic_info;
 
 
+typedef struct __attr_ptr_size_aligned__ _nt_tty_pts_slot_info {
+	int32_t		blkidx;
+	int32_t		blkpts;
+	int32_t		blkcnt;
+	int32_t		blkcap;
+	uint32_t	ptsmap[32];
+} nt_tty_pts_slot_info;
+
+
 typedef struct __attr_ptr_size_aligned__ _nt_tty_server_info {
 	nt_tty_server_info_class	info_class;
 	uint32_t			info_length;
@@ -611,6 +620,12 @@ typedef int32_t __stdcall ntapi_tty_query_information_service(
 	__in	uint32_t		id);
 
 
+typedef 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);
+
+
 /* pty api */
 typedef struct nt_pty_context nt_pty;
 
diff --git a/include/ntapi/ntapi.h b/include/ntapi/ntapi.h
index 7714bce..82491a7 100644
--- a/include/ntapi/ntapi.h
+++ b/include/ntapi/ntapi.h
@@ -597,6 +597,7 @@ typedef struct _ntapi_vtbl {
 	ntapi_tty_query_information_section *		tty_query_information_section;
 	ntapi_tty_query_information_server *		tty_query_information_server;
 	ntapi_tty_query_information_service *		tty_query_information_service;
+	ntapi_tty_query_server_pts_slot_info *		tty_query_server_pts_slot_info;
 	ntapi_tty_request_peer *			tty_request_peer;
 	ntapi_tty_vms_query *				tty_vms_query;
 	ntapi_tty_vms_request *				tty_vms_request;
diff --git a/project/common.mk b/project/common.mk
index a5eb5ab..ffd7376 100644
--- a/project/common.mk
+++ b/project/common.mk
@@ -131,6 +131,7 @@ COMMON_SRCS = \
 	src/tty/ntapi_tty_query_information_section.c \
 	src/tty/ntapi_tty_query_information_server.c \
 	src/tty/ntapi_tty_query_information_service.c \
+	src/tty/ntapi_tty_query_server_pts_slot_info.c \
 	src/tty/ntapi_tty_request_peer.c \
 	src/tty/ntapi_tty_vms_query.c \
 	src/tty/ntapi_tty_vms_request.c \
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;
+}