From edb08567093daae164f6d3955884117dee74fe98 Mon Sep 17 00:00:00 2001 From: midipix Date: Fri, 4 Aug 2017 22:22:22 +0000 Subject: subsystem interfaces: added server-to-server synchronization interfaces. --- src/ipc/ntapi_ipc_connect.c | 212 ++++++++++++++++++++++++++++++++++++++++++-- 1 file changed, 205 insertions(+), 7 deletions(-) (limited to 'src/ipc') diff --git a/src/ipc/ntapi_ipc_connect.c b/src/ipc/ntapi_ipc_connect.c index fceb527..6584f49 100644 --- a/src/ipc/ntapi_ipc_connect.c +++ b/src/ipc/ntapi_ipc_connect.c @@ -20,14 +20,142 @@ static int32_t __ipc_connect_return( } +static void __ipc_init_ctrl_msg_semctl( + const nt_port_attr * attr, + nt_tty_ipc_msg * msg) +{ + __ntapi->tt_aligned_block_memset( + (uintptr_t *)msg, + 0,sizeof(*msg)); + + msg->data.ipcinfo.ctrlsvc.type = attr->type; + msg->data.ipcinfo.ctrlsvc.subtype = attr->subtype; + + msg->data.ipcinfo.ctrlsvc.keys.key[0] = attr->keys.key[0]; + msg->data.ipcinfo.ctrlsvc.keys.key[1] = attr->keys.key[1]; + msg->data.ipcinfo.ctrlsvc.keys.key[2] = attr->keys.key[2]; + msg->data.ipcinfo.ctrlsvc.keys.key[3] = attr->keys.key[3]; + msg->data.ipcinfo.ctrlsvc.keys.key[4] = attr->keys.key[4]; + msg->data.ipcinfo.ctrlsvc.keys.key[5] = attr->keys.key[5]; + + __ntapi->tt_guid_copy( + &msg->data.ipcinfo.ctrlsvc.guid, + &attr->guid); +} + + +static void __ipc_init_ctrl_msg_semsvc( + nt_rtdata * rtdata, + nt_tty_ipc_msg * msg) +{ + __ntapi->tt_aligned_block_memset( + (uintptr_t *)msg, + 0,sizeof(*msg)); + + msg->data.ipcinfo.ctrlsvc.type = rtdata->semctl_type; + msg->data.ipcinfo.ctrlsvc.subtype = rtdata->semctl_subtype; + + msg->data.ipcinfo.ctrlsvc.keys.key[0] = rtdata->semctl_keys[0]; + msg->data.ipcinfo.ctrlsvc.keys.key[1] = rtdata->semctl_keys[1]; + msg->data.ipcinfo.ctrlsvc.keys.key[2] = rtdata->semctl_keys[2]; + msg->data.ipcinfo.ctrlsvc.keys.key[3] = rtdata->semctl_keys[3]; + msg->data.ipcinfo.ctrlsvc.keys.key[4] = rtdata->semctl_keys[4]; + msg->data.ipcinfo.ctrlsvc.keys.key[5] = rtdata->semctl_keys[5]; + + __ntapi->tt_guid_copy( + &msg->data.ipcinfo.ctrlsvc.guid, + &rtdata->semctl_guid); +} + + +static void __ipc_init_ctrl_msg_ipcpeer( + nt_rtdata * rtdata, + nt_tty_ipc_msg * msg) +{ + __ntapi->tt_aligned_block_memset( + (uintptr_t *)msg, + 0,sizeof(*msg)); + + msg->data.ipcinfo.ctrlsvc.type = rtdata->srv_type; + msg->data.ipcinfo.ctrlsvc.subtype = rtdata->srv_subtype; + + msg->data.ipcinfo.ctrlsvc.keys.key[0] = rtdata->srv_keys[0]; + msg->data.ipcinfo.ctrlsvc.keys.key[1] = rtdata->srv_keys[1]; + msg->data.ipcinfo.ctrlsvc.keys.key[2] = rtdata->srv_keys[2]; + msg->data.ipcinfo.ctrlsvc.keys.key[3] = rtdata->srv_keys[3]; + msg->data.ipcinfo.ctrlsvc.keys.key[4] = rtdata->srv_keys[4]; + msg->data.ipcinfo.ctrlsvc.keys.key[5] = rtdata->srv_keys[5]; + + __ntapi->tt_guid_copy( + &msg->data.ipcinfo.ctrlsvc.guid, + &rtdata->srv_guid); +} + + +static int32_t __ipc_set_client_keys( + void * hport, + const nt_port_attr * attr, + nt_rtdata * rtdata) +{ + int32_t status; + nt_tty_ipc_msg msg; + uint32_t opcode; + + switch (attr->type) { + case NT_PORT_TYPE_SEMCTL: + if (rtdata->srv_type == NT_PORT_TYPE_SEMSVC) + __ipc_init_ctrl_msg_ipcpeer( + rtdata,&msg); + else + __ipc_init_ctrl_msg_semctl( + attr,&msg); + + opcode = NT_TTY_SEM_FCNTL; + break; + + case NT_PORT_TYPE_SEMSVC: + if (rtdata->srv_type == NT_PORT_TYPE_SEMCTL) + return NT_STATUS_SUCCESS; + + __ipc_init_ctrl_msg_semsvc( + rtdata,&msg); + + opcode = NT_TTY_SEM_FCNTL; + break; + + default: + return NT_STATUS_SUCCESS; + } + + 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 = opcode; + + msg.data.ipcinfo.ipckeys[0] = rtdata->ipc_keys[0]; + msg.data.ipcinfo.ipckeys[1] = rtdata->ipc_keys[1]; + msg.data.ipcinfo.ipckeys[2] = rtdata->ipc_keys[2]; + msg.data.ipcinfo.ipckeys[3] = rtdata->ipc_keys[3]; + msg.data.ipcinfo.ipckeys[4] = rtdata->ipc_keys[4]; + msg.data.ipcinfo.ipckeys[5] = rtdata->ipc_keys[5]; + + if ((status = __ntapi->zw_request_wait_reply_port(hport,&msg,&msg))) + return status; + else if (msg.data.ttyinfo.status) + return msg.data.ttyinfo.status; + + return NT_STATUS_SUCCESS; +} + static int32_t __ipc_connect_by_attr( void ** hport, const nt_port_attr * attr, nt_unicode_string * str, void * hconn, - __out void ** hsection, - __out void ** secaddr, - __out size_t * secsize) + void ** hsection, + void ** secaddr, + size_t * secsize, + int fexisting) { int32_t status; struct dalist_node_ex * node; @@ -35,10 +163,12 @@ static int32_t __ipc_connect_by_attr( nt_port_attr * nconn; nt_ipc_conn * ipc; intptr_t * hlock; + nt_rtdata * rtdata; ntapi_internals * __internals; /* init */ __internals = __ntapi_internals(); + rtdata = __internals->rtdata; /* lock */ hlock = &(__internals->hlock); @@ -76,6 +206,10 @@ static int32_t __ipc_connect_by_attr( } } + /* __ipc_get_port? */ + if (fexisting) + return __ipc_connect_return(hlock,NT_STATUS_NOT_FOUND); + /* allocate list node */ if ((status = dalist_get_free_node( &__internals->ipc_conns, @@ -96,6 +230,41 @@ static int32_t __ipc_connect_by_attr( } } + /* server-to-server synchronization */ + if (!rtdata->ipc_keys[0]) { + rtdata->ipc_keys[0] = __ntapi->tt_buffer_crc32( + (uint32_t)(uintptr_t)&hport, + (char *)__internals,sizeof(*__internals)); + + rtdata->ipc_keys[1] = __ntapi->tt_buffer_crc32( + (uint32_t)(uintptr_t)&hconn, + (char *)rtdata,sizeof(*rtdata)); + + rtdata->ipc_keys[2] = __ntapi->tt_buffer_crc32( + (uint32_t)(uintptr_t)&attr, + (char *)attr,sizeof(*attr)); + + rtdata->ipc_keys[3] = __ntapi->tt_buffer_crc32( + (uint32_t)(uintptr_t)&node, + (char *)node,sizeof(*node)); + + rtdata->ipc_keys[4] = __ntapi->tt_buffer_crc32( + (uint32_t)(uintptr_t)&str, + (char *)pe_get_peb_address(), + sizeof(nt_peb)); + + rtdata->ipc_keys[5] = __ntapi->tt_buffer_crc32( + (uint32_t)(uintptr_t)&conn, + (char *)pe_get_teb_address(), + sizeof(nt_tib)); + } + + if ((status = __ipc_set_client_keys(hconn,attr,rtdata))) { + __ntapi->zw_close(hconn); + dalist_deposit_free_node(&__internals->ipc_conns,node); + return __ipc_connect_return(hlock,status); + } + /* add connection */ node->key = (uintptr_t)hconn; ipc = (nt_ipc_conn *)&node->dblock; @@ -143,7 +312,7 @@ int32_t __stdcall __ntapi_ipc_connect_section_by_attr( return __ipc_connect_by_attr( hport,attr,&str,0, - hsection,secaddr,secsize); + hsection,secaddr,secsize,0); } @@ -167,7 +336,7 @@ int32_t __stdcall __ntapi_ipc_connect_section_by_name( return __ipc_connect_by_attr( hport,&attr,&str,0, - hsection,secaddr,secsize); + hsection,secaddr,secsize,0); } @@ -204,7 +373,7 @@ int32_t __stdcall __ntapi_ipc_connect_section_by_symlink( return __ipc_connect_by_attr( hport,&attr,str,0, - hsection,secaddr,secsize); + hsection,secaddr,secsize,0); } @@ -217,7 +386,7 @@ int32_t __stdcall __ntapi_ipc_connect_section_by_port( { return __ipc_connect_by_attr( &(void *){0},attr,0,hconn, - hsection,secaddr,secsize); + hsection,secaddr,secsize,0); } @@ -298,6 +467,35 @@ int __ntapi_ipc_page_alloc( } +int32_t __stdcall __ntapi_ipc_get_port_by_attr( + __out void ** hport, + __in nt_port_attr * attr) +{ + return __ipc_connect_by_attr( + hport,attr,0,0, + &(void *){0}, + &(void *){0}, + &(size_t){0}, + 1); +} + + +int32_t __stdcall __ntapi_ipc_get_port_section_by_attr( + __out void ** hport, + __in nt_port_attr * attr, + __out void ** hsection, + __out void ** section_addr, + __out size_t * section_size) +{ + return __ipc_connect_by_attr( + hport,attr,0,0, + hsection, + section_addr, + section_size, + 1); +} + + int32_t __stdcall __ntapi_ipc_init_section_by_port( __in void * hconn, __out void ** hsection, -- cgit v1.2.3