1 files changed, 82 insertions, 0 deletions
diff --git a/src/process/ntapi_tt_init_runtime_data.c b/src/process/ntapi_tt_init_runtime_data.c
new file mode 100644
index 0000000..c7c2603
--- /dev/null
+++ b/src/process/ntapi_tt_init_runtime_data.c
@@ -0,0 +1,82 @@
+/********************************************************/
+/*  ntapi: Native API core library                      */
+/*  Copyright (C) 2013,2014,2015  Z. Gilboa             */
+/*  Released under GPLv2 and GPLv3; see COPYING.NTAPI.  */
+/********************************************************/
+
+#include <psxtypes/psxtypes.h>
+#include <ntapi/nt_memory.h>
+#include <ntapi/nt_process.h>
+#include <ntapi/ntapi.h>
+#include "ntapi_impl.h"
+
+int32_t __stdcall __ntapi_tt_update_runtime_data(nt_runtime_data * rtdata)
+{
+	int32_t				status;
+	nt_process_basic_information	pbi;
+	uint32_t			ret;
+	nt_oa				oa = {sizeof(oa)};
+
+	/* process (self) */
+	rtdata->cid_self.process_id = pe_get_current_process_id();
+	rtdata->cid_self.thread_id  = 0;
+
+	if ((status = __ntapi->zw_open_process(
+			&rtdata->hprocess_self,
+			NT_PROCESS_ALL_ACCESS,
+			&oa,&rtdata->cid_self)))
+		return status;
+
+	if (rtdata->cid_parent.process_id)
+		return NT_STATUS_SUCCESS;
+
+	/* process (parent) */
+	if ((status = __ntapi->zw_query_information_process(
+			rtdata->hprocess_self,
+			NT_PROCESS_BASIC_INFORMATION,
+			&pbi,sizeof(pbi),&ret)))
+		return status;
+
+	rtdata->cid_parent.process_id = pbi.inherited_from_unique_process_id;
+	rtdata->cid_parent.thread_id  = 0;
+	rtdata->hprocess_parent = 0;
+
+	return NT_STATUS_SUCCESS;
+}
+
+int32_t __stdcall __ntapi_tt_init_runtime_data(nt_runtime_data * rtdata)
+{
+	int32_t				status;
+	nt_peb *			peb;
+	nt_oa				oa = {sizeof(oa)};
+
+	/* init */
+	__ntapi->tt_aligned_block_memset(rtdata,0,sizeof(*rtdata));
+	peb = (nt_peb *)(pe_get_peb_address());
+
+	/* pid (self,parent) */
+	if ((status = __ntapi_tt_update_runtime_data(rtdata)))
+		return status;
+
+	/* std handles */
+	rtdata->hstdin  = peb->process_params->hstdin;
+	rtdata->hstdout = peb->process_params->hstdout;
+	rtdata->hstderr = peb->process_params->hstderr;
+
+	if (__ntapi->tt_get_file_handle_type(rtdata->hstdin,&rtdata->stdin_type)) {
+		rtdata->hstdin = NT_INVALID_HANDLE_VALUE;
+		rtdata->stdin_type = 0;
+	}
+
+	if (__ntapi->tt_get_file_handle_type(rtdata->hstdout,&rtdata->stdout_type)) {
+		rtdata->hstdout = NT_INVALID_HANDLE_VALUE;
+		rtdata->stdout_type = 0;
+	}
+
+	if (__ntapi->tt_get_file_handle_type(rtdata->hstderr,&rtdata->stderr_type)) {
+		rtdata->hstderr = NT_INVALID_HANDLE_VALUE;
+		rtdata->stderr_type = 0;
+	}
+
+	return 0;
+}