/********************************************************/

/*  ntapi: Native API core library                      */

/*  Copyright (C) 2013--2021  SysDeer Technologies, LLC */

/*  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),0,0,0,0,0};

	/* 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->hself,

			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->hself,

			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->hparent               = 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),0,0,0,0,0};

	/* 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;

}