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

/*  ptycon: a pty-console bridge                         */

/*  Copyright (C) 2016--2017  SysDeer Technologies, LLC  */

/*  Released under GPLv2 and GPLv3; see COPYING.PTYCON.  */

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

#include <psxtypes/psxtypes.h>

#include <pemagine/pemagine.h>

#include <ntapi/nt_status.h>

#include <ntapi/nt_object.h>

#include <ntapi/nt_thread.h>

#include <ntapi/nt_process.h>

#include <ntapi/nt_string.h>

#include <ntapi/ntapi.h>

#include <ptycon/ptycon.h>

#include "ptycon_driver_impl.h"

int ptyc_spawn(struct ptyc_driver_ctx * dctx)

{

	int32_t				status;

	nt_spawn_process_params		sparams;

	nt_runtime_data			rtctx;

	struct ptyc_driver_ctx_impl *	ictx;

	struct ptyc_client_ctx *	clctx;

	nt_rtdata *			self;

	/* init */

	if (!(ictx = ptyc_get_driver_ictx(dctx)))

		return NT_STATUS_INVALID_HANDLE;

	if (!dctx->cctx->eargv)

		return NT_STATUS_SUCCESS;

	self  = ictx->rtdata;

	clctx = &ictx->clctx;

	/* sparams */

	ntapi->tt_aligned_block_memset(

		&sparams,0,sizeof(sparams));

	sparams.rtctx    = &rtctx;

	sparams.patharg  = dctx->cctx->eargv[0];

	sparams.argv     = dctx->cctx->eargv;

	sparams.envp     = self->envp;

	sparams.hsession = self->hsession;

	sparams.hroot    = dctx->cctx->hroot

				? dctx->cctx->hroot

				: self->hroot;

	/* rtctx */

	ntapi->tt_aligned_block_memset(

		&rtctx,0,sizeof(rtctx));

	ntapi->tt_aligned_block_memcpy(

		(uintptr_t *)&rtctx.cid_parent,

		(uintptr_t *)&self->cid_self,

		sizeof(nt_cid));

	rtctx.hcwd  = self->hcwd;

	rtctx.hroot = sparams.hroot;

	rtctx.hntpipcdir  = self->hntpipcdir;

	rtctx.hpidmapdir  = self->hpidmapdir;

	rtctx.hpidanydir  = self->hpidanydir;

	rtctx.hntpgrpdir  = self->hntpgrpdir;

	rtctx.htmpfsldir  = self->htmpfsldir;

	rtctx.htmpfslroot = self->htmpfslroot;

	rtctx.hshmctldir  = self->hshmctldir;

	rtctx.hshmctlroot = self->hshmctlroot;

	rtctx.hshmkeydir  = self->hshmkeydir;

	rtctx.hshmkeyroot = self->hshmkeyroot;

	rtctx.tty_type    = self->tty_type;

	rtctx.tty_subtype = self->tty_subtype;

	rtctx.tty_keys[0] = self->tty_keys[0];

	rtctx.tty_keys[1] = self->tty_keys[1];

	rtctx.tty_keys[2] = self->tty_keys[2];

	rtctx.tty_keys[3] = self->tty_keys[3];

	rtctx.tty_keys[4] = self->tty_keys[4];

	rtctx.tty_keys[5] = self->tty_keys[5];

	ntapi->tt_guid_copy(

		&rtctx.tty_guid,

		&self->tty_guid);

	rtctx.hstdin      = NT_INVALID_HANDLE_VALUE;

	rtctx.hstdout     = NT_INVALID_HANDLE_VALUE;

	rtctx.hstderr     = NT_INVALID_HANDLE_VALUE;

	rtctx.stdin_type  = NT_FILE_TYPE_PTY;

	rtctx.stdout_type = NT_FILE_TYPE_PTY;

	rtctx.stderr_type = NT_FILE_TYPE_PTY;

	rtctx.ptyin [0]   = clctx->clinfo.any[0];

	rtctx.ptyin [1]   = clctx->clinfo.any[1];

	rtctx.ptyin [2]   = clctx->clinfo.any[2];

	rtctx.ptyin [3]   = clctx->clinfo.any[3];

	rtctx.ptyout[0]   = clctx->clinfo.any[0];

	rtctx.ptyout[1]   = clctx->clinfo.any[1];

	rtctx.ptyout[2]   = clctx->clinfo.any[2];

	rtctx.ptyout[3]   = clctx->clinfo.any[3];

	rtctx.ptyerr[0]   = clctx->clinfo.any[0];

	rtctx.ptyerr[1]   = clctx->clinfo.any[1];

	rtctx.ptyerr[2]   = clctx->clinfo.any[2];

	rtctx.ptyerr[3]   = clctx->clinfo.any[3];

	rtctx.ptyctl[0]   = clctx->clinfo.any[0];

	rtctx.ptyctl[1]   = clctx->clinfo.any[1];

	rtctx.ptyctl[2]   = clctx->clinfo.any[2];

	rtctx.ptyctl[3]   = clctx->clinfo.any[3];

	/* hoppla */

	if ((status = ntapi->tt_spawn_native_process(&sparams)))

		return status;

	/* clctx */

	clctx->hprocess		= sparams.hprocess;

	clctx->hthread		= sparams.hthread;

	clctx->cid.process_id	= sparams.cid.process_id;

	clctx->cid.thread_id	= sparams.cid.thread_id;

	/* child ready? */

	if (!(sparams.eready.signal_state))

		return NT_STATUS_GENERIC_COMMAND_FAILED;

	/* finalize */

	if (dctx->cctx->drvflags & (PTYC_DRIVER_DBG_RAW|PTYC_DRIVER_DBG_OVEN))

		return NT_STATUS_SUCCESS;

	if ((status = ntapi->pty_close(ictx->cctx.hpts)))

		return status;

	ictx->cctx.hpts = 0;

	/* all done */

	return NT_STATUS_SUCCESS;

}