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 +#include +#include +#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; +}