/*********************************************************/
/*  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_daemon_impl.h"
#include "toksvc_driver_impl.h"

static int32_t toks_daemon_client_wait(void * rapunzel)
{
	struct toks_client_ctx * client;

	client = (struct toks_client_ctx *)rapunzel;

	ntapi->zw_set_event(client->hswap,0);

	ntapi->zw_wait_for_single_object(
		client->hprocess,
		NT_SYNC_NON_ALERTABLE,
		0);

	ntapi->zw_close(client->hprocess);
	ntapi->zw_close(client->hport);

	return ntapi->zw_terminate_thread(
		NT_CURRENT_THREAD_HANDLE,
		NT_STATUS_SUCCESS);
}

static int32_t toks_daemon_client_instance(struct toks_client_ctx * client)
{
	int32_t			status;
	nt_thread_params	params;

	if ((status = ntapi->tt_create_private_event(
			&client->hswap,
			NT_NOTIFICATION_EVENT,
			NT_EVENT_NOT_SIGNALED)))
		return status;

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

	params.hprocess		  = NT_CURRENT_PROCESS_HANDLE;
	params.start		  = toks_daemon_client_wait;
	params.ext_ctx            = client;
	params.ext_ctx_size       = sizeof(*client);
	params.stack_size_commit  = 4 * 1024;
	params.stack_size_reserve = 4 * 1024;
	params.creation_flags	  = NT_CREATE_LOCAL_THREAD;

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

	if ((status = ntapi->zw_wait_for_single_object(
			client->hswap,
			NT_SYNC_NON_ALERTABLE,
			0)))
		return status;

	ntapi->zw_close(client->hswap);

	return ntapi->zw_close(params.hthread);
}

int32_t __stdcall toks_daemon_connect(struct toks_daemon_ctx * dctx)
{
	int32_t                     status;
	nt_tty_port_msg *           msg;
	struct toks_client_ctx      client;
	nt_oa                       oa;

	msg                   = &dctx->reply;
	msg->ttyinfo.opdata   = 0;

	client.hport          = 0;
	client.hprocess       = 0;
	client.hswap          = 0;
	client.cid.process_id = msg->header.client_id.process_id;
	client.cid.thread_id  = 0;

	oa.len      = sizeof(oa);
	oa.root_dir = 0;
	oa.obj_name = 0;
	oa.obj_attr = 0;
	oa.sec_desc = 0;
	oa.sec_qos  = 0;

	status = ntapi->zw_open_process(
		&client.hprocess,
		NT_PROCESS_SYNCHRONIZE | NT_PROCESS_QUERY_INFORMATION,
		&oa,&client.cid);

	status = ntapi->zw_accept_connect_port(
		&client.hport,
		client.cid.process_id,
		&msg->header,
		NT_LPC_ACCEPT_CONNECTION,0,0);

	if (status) {
		ntapi->zw_close(client.hprocess);
		return status;
	}

	ntapi->zw_complete_connect_port(client.hport);

	return toks_daemon_client_instance(&client);
}