/*********************************************************/
/*  toksvc: a framework-native token broker service      */
/*  Copyright (C) 2020  SysDeer Technologies, LLC        */
/*  Released under GPLv2 and GPLv3; see COPYING.TOKSVC.  */
/*********************************************************/

#include <psxtypes/psxtypes.h>
#include <ntapi/ntapi.h>

#include <toksvc/toksvc.h>
#include "toksvc_driver_impl.h"
#include "toksvc_daemon_impl.h"

#define TOKS_STR(x)  #x
#define TOKS_XSTR(x) TOKS_STR(x)

static char toks_service_image[] = TOKS_XSTR(TOKS_SERVICE_IMAGE);

static int32_t toks_spawn_child_exit_code(void * hprocess)
{
	nt_process_basic_information pbi;

	if (ntapi->zw_query_information_process(
			hprocess,
			NT_PROCESS_BASIC_INFORMATION,
			&pbi,sizeof(pbi),
			&(uint32_t){0}))
		return NT_STATUS_GENERIC_COMMAND_FAILED;

	return pbi.exit_status;
}

static int toks_spawn_impl(
	struct toks_driver_ctx *        dctx,
	struct toks_common_ctx *        cctx,
	int                             ntokens,
	int32_t                         ctrlpid,
	int32_t                         csyspid)
{
	int32_t				status;
	nt_spawn_process_params		sparams;
	nt_runtime_data			rtctx;
	nt_rtdata *			self;
	char *				img;
	void *				hroot;
	void *				hlog;
	const char *			logfile;
	char **				pearg;
	char *				eargv[10];
	char				tokbuf[32];
	char				pidbuf[32];
	char				logbuf[32];
	char				guidstr[40];
	void *				hduo[2];

	/* init */
	self = toks_get_driver_rtdata(dctx);
	img  = toks_service_image;

	if (!(hroot = cctx->hroot) && cctx->sysroot)
		if ((status = toks_open_dir(&hroot,0,cctx->sysroot,false)))
			return status;

	if (hroot && (img[0] == '/'))
		img++;

	logfile = cctx->logfile;

	if (hroot && logfile && (logfile[0] == '/'))
		logfile++;

	if (logfile && (status = toks_open_log_file(&hlog,hroot,logfile,false)))
		return status;

	/* eargv */
	pearg    = eargv;
	*pearg++ = img;

	ntapi->sprintf(tokbuf,"%d",ntokens);
	*pearg++ = "-t";
	*pearg++ = tokbuf;

	ntapi->sprintf(logbuf,"%d",cctx->loglevel);
	*pearg++ = "-O";
	*pearg++ = logbuf;

	if (cctx->uuid) {
		toks_uuid_to_string(cctx->uuid,&guidstr);
		*pearg++ = "-u";
		*pearg++ = guidstr;
	}

	if (ctrlpid) {
		ntapi->sprintf(pidbuf,"%d",ctrlpid);
		*pearg++ = "-l";
		*pearg++ = pidbuf;
	} else if (csyspid) {
		ntapi->sprintf(pidbuf,"%d",csyspid);
		*pearg++ = "-L";
		*pearg++ = pidbuf;
	}

	*pearg++ = 0;

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

	sparams.rtctx    = &rtctx;
	sparams.patharg  = img;
	sparams.argv     = eargv;
	sparams.envp     = self->envp;
	sparams.hroot    = hroot ? hroot : self->hroot;

	if ((status = toks_open_file(&sparams.himage,hroot,img,true)))
		return status;

	/* 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.hlog  = hlog;
	rtctx.hcwd  = self->hcwd;
	rtctx.hroot = sparams.hroot;
	rtctx.flags = self->flags | NT_RUNTIME_DATA_TTY_TOP_LEVEL;

	rtctx.dbg_level = 0;
	rtctx.log_level = self->log_level;

	rtctx.hstdin      = NT_INVALID_HANDLE_VALUE;
	rtctx.hstdout     = NT_INVALID_HANDLE_VALUE;
	rtctx.hstderr     = NT_INVALID_HANDLE_VALUE;

	rtctx.stdin_type  = NT_FILE_TYPE_UNKNOWN;
	rtctx.stdout_type = NT_FILE_TYPE_UNKNOWN;
	rtctx.stderr_type = NT_FILE_TYPE_UNKNOWN;

	/* service synchronization */
	if ((status = ntapi->tt_create_inheritable_event(
			&rtctx.hsync,
			NT_NOTIFICATION_EVENT,
			NT_EVENT_NOT_SIGNALED)))
		return status;

	/* hoppla */
	status = ntapi->tt_spawn_native_process(&sparams);

	if (sparams.himage)
		ntapi->zw_close(sparams.himage);

	if (status)
		return status;

	/* child ready? */
	if (!(sparams.eready.signal_state))
		status = toks_spawn_child_exit_code(sparams.hprocess);

	/* service synchronization */
	hduo[0] = sparams.hprocess;
	hduo[1] = rtctx.hsync;

	ntapi->zw_wait_for_multiple_objects(
		2,hduo,
		NT_WAIT_ANY,
		NT_SYNC_NON_ALERTABLE,
		0);

	/* finalize */
	ntapi->zw_close(sparams.hprocess);
	ntapi->zw_close(sparams.hthread);

	/* all done */
	return status;
}

int32_t toks_service_start(
	struct toks_common_ctx *    cctx,
	int                         ntokens,
	int32_t                     ctrlpid,
	int32_t                     csyspid)
{
	int32_t                     status;
	struct toks_driver_ctx *    dctx;
	char *                      eargv[2];

	eargv[0] = "toksvc.driver";
	eargv[1] = 0;

	status = toks_get_driver_ctx(eargv,0,TOKS_DRIVER_VERSION,&dctx);

	if (status == NT_STATUS_SUCCESS) {
		status = toks_spawn_impl(dctx,cctx,ntokens,ctrlpid,csyspid);
		toks_free_driver_ctx(dctx);
	}

	return status;
}