1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
|
/********************************************************/
/* ntapi: Native API core library */
/* Copyright (C) 2013--2016 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;
}
|
