internals: __ntapi_init_solib_fork_child_finalize() [rare use-case support].

1 files changed, 83 insertions, 1 deletions

diff --git a/src/internal/ntapi.c b/src/internal/ntapi.c
index b1d282d..7562b35 100644
--- a/src/internal/ntapi.c
+++ b/src/internal/ntapi.c
@@ -588,6 +588,88 @@ static int32_t __fastcall __ntapi_init_once(ntapi_vtbl ** pvtbl)
 	return NT_STATUS_SUCCESS;
 }
 
+static int32_t __ntapi_init_solib_fork_child_finalize(void)
+{
+	int32_t			status;
+	int			page;
+	nt_rtdata *		rtdata;
+	nt_oa			oa;
+	ntapi_internals *	__internals;
+
+	/* detect the rare scenario where libntapi.so is (indirectly)      */
+	/* used by a hosted process, in which case post-fork adjustments  */
+	/* to internal structures would take place in the copy of ntapi  */
+	/* that is statically linked into the system call layer         */
+	/* library, thereby requiring similar adjustmets to be made    */
+	/* once here as well.                                         */
+
+	__internals = __ntapi_internals();
+	rtdata      = __internals->rtdata;
+
+	if (rtdata->cid_self.process_id == pe_get_current_process_id())
+		return NT_STATUS_SUCCESS;
+
+	rtdata->cid_parent.process_id = rtdata->cid_self.process_id;
+	rtdata->cid_parent.thread_id  = rtdata->cid_self.thread_id;
+
+	rtdata->cid_self.process_id   = pe_get_current_process_id();
+	rtdata->cid_self.thread_id    = pe_get_current_thread_id();
+
+	if (rtdata->hparent)
+		__ntapi->zw_close(rtdata->hparent);
+
+	oa.len		= sizeof(oa);
+	oa.root_dir	= 0;
+	oa.obj_name	= 0;
+	oa.obj_attr	= 0;
+	oa.sec_desc	= &__internals->seq_desc;
+	oa.sec_qos	= &__internals->seq_qos;
+
+	status = __ntapi->zw_open_process(
+		&__internals->hprocess,
+		NT_PROCESS_ALL_ACCESS,
+		&oa,&rtdata->cid_self);
+
+	rtdata->hparent = rtdata->hself;
+	rtdata->hself   = __internals->hprocess;
+
+	if ((status = dalist_init_ex(
+			&__internals->ipc_conns,
+			sizeof(nt_ipc_conn),
+			NT_ALLOCATION_GRANULARITY,
+			__ntapi_ipc_page_alloc,
+			DALIST_MEMFN_CUSTOM)))
+		return status;
+
+	dalist_deposit_memory_block(
+		&__internals->ipc_conns,
+		__internals->ntapi_img_sec_bss->ipc_buffer,
+		__NT_BSS_IPC_BUFFER_SIZE);
+
+	for (page=0; page<__internals->ipc_page; page++)
+		dalist_deposit_memory_block(
+			&__internals->ipc_conns,
+			__internals->ipc_pages[page],
+			NT_ALLOCATION_GRANULARITY);
+
+	rtdata->hsemctl     = 0;
+	rtdata->hsempid     = 0;
+
+	rtdata->hmsqctl     = 0;
+	rtdata->hmsqpid     = 0;
+
+	rtdata->haflctl     = 0;
+	rtdata->haflpid     = 0;
+
+	rtdata->ipc_keys[0] = 0;
+	rtdata->ipc_keys[1] = 0;
+	rtdata->ipc_keys[2] = 0;
+	rtdata->ipc_keys[3] = 0;
+	rtdata->ipc_keys[4] = 0;
+	rtdata->ipc_keys[5] = 0;
+
+	return 0;
+}
 
 static int32_t __fastcall __ntapi_init_pending(ntapi_vtbl ** pvtbl)
 {
@@ -598,7 +680,7 @@ static int32_t __fastcall __ntapi_init_pending(ntapi_vtbl ** pvtbl)
 static int32_t __fastcall __ntapi_init_completed(ntapi_vtbl ** pvtbl)
 {
 	*pvtbl = &___ntapi_shadow;
-	return NT_STATUS_SUCCESS;
+	return __ntapi_init_solib_fork_child_finalize();
 };