summaryrefslogtreecommitdiffhomepage
path: root/src/ptycon.c
blob: dd483b83e2ada250dbd34afa0a988b2159df7949 (plain)
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
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
149
150
151
152
153
154
155
156
157
158
159
160
161
162
163
164
165
166
167
168
169
/*********************************************************/
/*  ptycon: a pty-console bridge                         */
/*  Copyright (C) 2016  Z. Gilboa                        */
/*  Released under GPLv2 and GPLv3; see COPYING.PTYCON.  */
/*********************************************************/

#include <pemagine/pemagine.h>
#include <ntapi/ntapi.h>
#include <ptycon/ptycon.h>
#include "ptycon_init_impl.h"
#include "ptycon_driver_impl.h"

static const nt_tty_affiliation tty_affiliation
	__attr_section__(".midipix")
	= NT_TTY_AFFILIATION_DEFAULT;

static void ptycon_exit(int code)
{
	/* posix exit code? */
	if (code <= 0xff)
		code <<= 8;

	ntapi->zw_terminate_process(
		NT_CURRENT_PROCESS_HANDLE,
		code);
}

static int32_t ptycon_start(int argc, char ** argv, char ** envp)
{
	int32_t				status;
	nt_runtime_data *		rtdata;
	nt_port_attr			port_attr;
        nt_pty_client_info		client_info;
        nt_pty *			hpty;
	nt_iosb				iosb;

	/* rtdata */
	if ((status = ntapi->tt_get_runtime_data(&rtdata,0)))
		return status;

	if (rtdata->argv) {
		argc = rtdata->argc;
		argv = rtdata->argv;
		envp = rtdata->envp;
	}

	/* no tty session? */
	if (!rtdata->srv_keys[0])
		return ptyc_main(argc,argv,envp);

	/* tty */
	ntapi->tt_aligned_block_memset(
		&port_attr,0,sizeof(port_attr));

	port_attr.type		= NT_PORT_TYPE_SUBSYSTEM;
	port_attr.subtype	= NT_PORT_SUBTYPE_DEFAULT;

	port_attr.keys.key[0]	= rtdata->srv_keys[0];
	port_attr.keys.key[1]	= rtdata->srv_keys[1];
	port_attr.keys.key[2]	= rtdata->srv_keys[2];
	port_attr.keys.key[3]	= rtdata->srv_keys[3];
	port_attr.keys.key[4]	= rtdata->srv_keys[4];
	port_attr.keys.key[5]	= rtdata->srv_keys[5];

	ntapi->tt_port_guid_from_type(
		&port_attr.guid,
		port_attr.type,
		port_attr.subtype);

	if ((status = ntapi->tty_join_session(
			&rtdata->hsession,0,
			&port_attr,
			NT_TTY_SESSION_PRIMARY)))
		return status;

	/* pty */
	client_info.any[0] = rtdata->ptyany[0];
	client_info.any[1] = rtdata->ptyany[1];
	client_info.any[2] = rtdata->ptyany[2];
	client_info.any[3] = rtdata->ptyany[3];

	status = ntapi->pty_inherit(
		rtdata->hsession,
		&hpty,
		&client_info);

	/* ntaio */
	if (status == NT_STATUS_SUCCESS) {
		if (rtdata->stdin_type == NT_FILE_TYPE_PTY)
			rtdata->hstdin = hpty;

		if (rtdata->stdout_type == NT_FILE_TYPE_PTY)
			rtdata->hstdout = hpty;

		if (rtdata->stderr_type == NT_FILE_TYPE_PTY)
			rtdata->hstderr = hpty;

		/* ctty identification */
		client_info.any[0] = 0;
		client_info.any[1] = 0;
		client_info.any[2] = 0;
		client_info.any[3] = 0;

		if ((status = ntapi->pty_set(
				hpty,&iosb,
				&client_info,sizeof(client_info),
				NT_PTY_CLIENT_INFORMATION)))
			return status;
	}

	/* ready */
	if (rtdata->hready)
		if ((status = ntapi->zw_set_event(rtdata->hready,0)))
			return status;

	/* main */
	return ptyc_main(argc,argv,envp);
}

static int ptycon_daemon_entry_point(void * arg)
{
	int32_t		status;
	int		argc;
	char **		argv;
	char **		envp;

	(void)arg;
	(void)tty_affiliation;

	if ((status = ntapi->tt_get_argv_envp_utf8(
			&argc,&argv,&envp,
			0,0,0)))
		ptycon_exit(status);

	ptycon_exit(ptycon_start(
		argc,argv,envp));

	return NT_STATUS_INTERNAL_ERROR;
}

int ptycon_entry_point(void)
{
	int32_t			status;
	nt_thread_params	params;

	if ((status = ptyc_init()))
		return status;

	ntapi->tt_aligned_block_memset(
		&params,0,sizeof(params));

	params.hprocess		  = NT_CURRENT_PROCESS_HANDLE;
	params.start		  = ptycon_daemon_entry_point;
	params.stack_size_commit  = 128 * 1024;
	params.stack_size_reserve = 128 * 1024;
	params.creation_flags	  = NT_CREATE_LOCAL_THREAD;

	if ((status = ntapi->tt_create_thread(&params)))
		ptycon_exit(status);

	ntapi->zw_close(
		params.hthread);

	ntapi->zw_terminate_thread(
		NT_CURRENT_THREAD_HANDLE,
		NT_STATUS_SUCCESS);

	return NT_STATUS_INTERNAL_ERROR;
}