diff options
Diffstat (limited to 'src/pty')
-rw-r--r-- | src/pty/ntapi_pty_fd.c | 10 | ||||
-rw-r--r-- | src/pty/ntapi_pty_query.c | 33 | ||||
-rw-r--r-- | src/pty/ntapi_pty_xquery.c | 87 |
3 files changed, 93 insertions, 37 deletions
diff --git a/src/pty/ntapi_pty_fd.c b/src/pty/ntapi_pty_fd.c index 8d21fd0..45a58ff 100644 --- a/src/pty/ntapi_pty_fd.c +++ b/src/pty/ntapi_pty_fd.c @@ -280,15 +280,11 @@ int32_t __stdcall __ntapi_pty_inherit( nt_pty_inherit_info inherit; nt_pty * ctx; - inherit.any[0] = client_info->any[0]; - inherit.any[1] = client_info->any[1]; - inherit.any[2] = client_info->any[2]; - inherit.any[3] = client_info->any[3]; - - if ((status = __ntapi_pty_query( + if ((status = __ntapi_pty_xquery( hport,&iosb, &inherit,sizeof(inherit), - NT_PTY_INHERIT_INFORMATION))) + NT_PTY_INHERIT_INFORMATION, + client_info))) return status; /* control block */ diff --git a/src/pty/ntapi_pty_query.c b/src/pty/ntapi_pty_query.c index 72b3b31..329516a 100644 --- a/src/pty/ntapi_pty_query.c +++ b/src/pty/ntapi_pty_query.c @@ -22,11 +22,13 @@ int32_t __stdcall __ntapi_pty_query( void * hport; nt_pty_sigctl_msg msg; uintptr_t * info; - nt_pty_inherit_info * inherit; if (pty_info_class>=NT_PTY_INFORMATION_CAP) return NT_STATUS_INVALID_INFO_CLASS; + if (pty_info_class == NT_PTY_INHERIT_INFORMATION) + return NT_STATUS_CONTEXT_MISMATCH; + if (pty_info_class == NT_PTY_BASIC_INFORMATION) return NT_STATUS_NOT_IMPLEMENTED; @@ -34,10 +36,6 @@ int32_t __stdcall __ntapi_pty_query( if (pty_info_length < sizeof(nt_pty_client_info)) return NT_STATUS_INVALID_PARAMETER; - if (pty_info_class == NT_PTY_INHERIT_INFORMATION) - if (pty_info_length != sizeof(nt_pty_inherit_info)) - return NT_STATUS_INVALID_PARAMETER; - __ntapi->tt_aligned_block_memset( &msg,0,sizeof(msg)); @@ -57,17 +55,6 @@ int32_t __stdcall __ntapi_pty_query( &msg.data.ctlinfo.guid, &pty->guid); - } else if (pty_info_class == NT_PTY_INHERIT_INFORMATION) { - msg.data.ctlinfo.hpty = NT_INVALID_HANDLE_VALUE; - msg.data.ctlinfo.ctlcode = pty_info_class; - - inherit = (nt_pty_inherit_info *)pty_info; - msg.data.ctlinfo.ctxarg[0] = inherit->any[0]; - msg.data.ctlinfo.ctxarg[1] = inherit->any[1]; - msg.data.ctlinfo.ctxarg[2] = inherit->any[2]; - msg.data.ctlinfo.ctxarg[3] = inherit->any[3]; - - hport = pty ? pty : __ntapi_internals()->hport_tty_session; } else { return NT_STATUS_INVALID_INFO_CLASS; } @@ -88,20 +75,6 @@ int32_t __stdcall __ntapi_pty_query( info[2] = msg.data.ctlinfo.ctxarg[2]; info[3] = msg.data.ctlinfo.ctxarg[3]; - } else if (pty_info_class == NT_PTY_INHERIT_INFORMATION) { - inherit = (nt_pty_inherit_info *)pty_info; - inherit->hpty = msg.data.ctlinfo.hpty; - inherit->luid.low = msg.data.ctlinfo.luid.low; - inherit->luid.high = msg.data.ctlinfo.luid.high; - - inherit->access = (uint32_t)msg.data.ctlinfo.ctxarg[0]; - inherit->flags = (uint32_t)msg.data.ctlinfo.ctxarg[1]; - inherit->share = (uint32_t)msg.data.ctlinfo.ctxarg[2]; - inherit->options = (uint32_t)msg.data.ctlinfo.ctxarg[3]; - - __ntapi->tt_guid_copy( - &inherit->guid, - &msg.data.ctlinfo.guid); } return NT_STATUS_SUCCESS; diff --git a/src/pty/ntapi_pty_xquery.c b/src/pty/ntapi_pty_xquery.c new file mode 100644 index 0000000..075d9b1 --- /dev/null +++ b/src/pty/ntapi_pty_xquery.c @@ -0,0 +1,87 @@ +/********************************************************/ +/* 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_port.h> +#include <ntapi/nt_tty.h> +#include <ntapi/ntapi.h> +#include "ntapi_impl.h" +#include "ntapi_pty.h" + + +static int32_t __pty_xquery_inherit_info( + void * hport, + nt_io_status_block * iosb, + void * pty_info, + uint32_t pty_info_length, + nt_pty_client_info * pty_client_info) +{ + int32_t status; + nt_pty_sigctl_msg msg; + nt_pty_inherit_info * inherit; + + if (pty_info_length < sizeof(nt_pty_inherit_info)) + return NT_STATUS_INVALID_PARAMETER; + + __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.ttyinfo.opcode = NT_TTY_PTY_QUERY; + + msg.data.ctlinfo.hpty = NT_INVALID_HANDLE_VALUE; + msg.data.ctlinfo.ctlcode = NT_PTY_INHERIT_INFORMATION; + + msg.data.ctlinfo.ctxarg[0] = pty_client_info->any[0]; + msg.data.ctlinfo.ctxarg[1] = pty_client_info->any[1]; + msg.data.ctlinfo.ctxarg[2] = pty_client_info->any[2]; + msg.data.ctlinfo.ctxarg[3] = pty_client_info->any[3]; + + if ((status = __ntapi->zw_request_wait_reply_port(hport,&msg,&msg))) + return status; + + else if (msg.data.ttyinfo.status) + return msg.data.ttyinfo.status; + + iosb->info = msg.data.ctlinfo.iosb.info; + iosb->status = msg.data.ctlinfo.iosb.status; + + inherit = (nt_pty_inherit_info *)pty_info; + inherit->hpty = msg.data.ctlinfo.hpty; + inherit->luid.low = msg.data.ctlinfo.luid.low; + inherit->luid.high = msg.data.ctlinfo.luid.high; + + inherit->access = (uint32_t)msg.data.ctlinfo.ctxarg[0]; + inherit->flags = (uint32_t)msg.data.ctlinfo.ctxarg[1]; + inherit->share = (uint32_t)msg.data.ctlinfo.ctxarg[2]; + inherit->options = (uint32_t)msg.data.ctlinfo.ctxarg[3]; + + __ntapi->tt_guid_copy( + &inherit->guid, + &msg.data.ctlinfo.guid); + + return NT_STATUS_SUCCESS; +} + + +int32_t __stdcall __ntapi_pty_xquery( + void * hport, + nt_io_status_block * iosb, + void * pty_info, + uint32_t pty_info_length, + nt_pty_info_class pty_info_class, + nt_pty_client_info * pty_client_info) +{ + if (pty_info_class == NT_PTY_INHERIT_INFORMATION) + return __pty_xquery_inherit_info( + hport,iosb,pty_info, + pty_info_length, + pty_client_info); + + return NT_STATUS_WRONG_COMPARTMENT; +} |