summaryrefslogtreecommitdiffhomepage
path: root/src/ipc
diff options
context:
space:
mode:
Diffstat (limited to 'src/ipc')
-rw-r--r--src/ipc/ntapi_ipc_connect.c212
1 files changed, 205 insertions, 7 deletions
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,