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

/*  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 void toks_daemon_token_reset(struct toks_token * token)

{

	token->self        = 0;

	token->keys.key[0] = 0;

	token->keys.key[1] = 0;

	token->keys.key[2] = 0;

	token->keys.key[3] = 0;

	token->keys.key[4] = 0;

	token->keys.key[5] = 0;

}

static int32_t toks_daemon_unqueue(struct toks_daemon_ctx * dctx)

{

	int                         status;

	int                         nwaiters;

	struct toks_waiter *        waiter;

	nt_tty_port_msg *           msg;

	nt_tty_port_msg             cmsg;

	void *                      hcaller;

	msg      = &dctx->reply;

	nwaiters = toks_get_driver_nwaiters(dctx->driver_ctx);

	if ((nwaiters == 0) || (dctx->ftokens == 0))

		return NT_STATUS_SUCCESS;

	waiter = dctx->waiter_first;

	toks_set_driver_nwaiters(

		dctx->driver_ctx,

		--nwaiters);

	dctx->waiter_first++;

	if (dctx->waiter_first == dctx->waiter_cap)

		dctx->waiter_first = dctx->waiter_base;

	if (!waiter->msg.header.msg_id)

		return NT_STATUS_CANCELLED;

	if ((hcaller = waiter->client.hcaller)) {

		status = ntapi->zw_wait_for_single_object(

			hcaller,

			NT_SYNC_NON_ALERTABLE,

			&(nt_timeout){.quad=0});

		ntapi->zw_close(hcaller);

		switch (status) {

			case NT_STATUS_TIMEOUT:

				break;

			default:

				waiter->msg.header.msg_id = 0;

				return NT_STATUS_REQUEST_ABORTED;

		}

	}

	ntapi->tt_generic_memcpy(

		&cmsg,msg,

		sizeof(*msg));

	ntapi->tt_generic_memcpy(

		msg,&waiter->msg,

		sizeof(*msg));

	msg->ttyinfo.exarg  = waiter->client.hprocess;

	msg->ttyinfo.status = toks_daemon_acquire(dctx);

	status              = msg->ttyinfo.status;

	ntapi->zw_set_event(waiter->client.hevent,0);

	ntapi->zw_close(waiter->client.hevent);

	ntapi->tt_generic_memcpy(

		msg,&cmsg,

		sizeof(*msg));

	return status;

}

static int toks_daemon_unqueue_ioctl(struct toks_daemon_ctx * dctx)

{

	int ntokens = toks_get_driver_ntokens(dctx->driver_ctx);

	dctx->opcode = TOKS_DAEMON_RELEASE;

	while (dctx->utokens < ntokens) {

		if (toks_get_driver_nwaiters(dctx->driver_ctx) == 0) {

			dctx->opcode = TOKS_DAEMON_IOCTL;

			return NT_STATUS_SUCCESS;

		}

		toks_daemon_unqueue(dctx);

	}

	dctx->opcode = TOKS_DAEMON_IOCTL;

	return NT_STATUS_SUCCESS;

}

int32_t __stdcall toks_daemon_release(struct toks_daemon_ctx * dctx)

{

	nt_tty_port_msg *           msg;

	struct toks_token *         token;

	struct toks_token *         toptr;

	struct toks_token *         tocap;

	uint32_t *                  keys;

	if (dctx->opcode == TOKS_DAEMON_IOCTL)

		return toks_daemon_unqueue_ioctl(dctx);

	msg   = &dctx->reply;

	keys  = msg->syncinfo.ipckeys;

	toptr = toks_get_driver_tokens(dctx->driver_ctx);

	tocap = &toptr[toks_get_driver_atokens(dctx->driver_ctx)];

	for (token=0; !token && (toptr

		if ((toptr->self)

				&& (toptr->keys.key[0] == keys[0])

				&& (toptr->keys.key[1] == keys[1])

				&& (toptr->keys.key[2] == keys[2])

				&& (toptr->keys.key[3] == keys[3])

				&& (toptr->keys.key[4] == keys[4])

				&& (toptr->keys.key[5] == keys[5]))

			token = toptr;

	if (!token)

		return NT_STATUS_INVALID_PARAMETER;

	if (--dctx->utokens < toks_get_driver_ntokens(dctx->driver_ctx))

		dctx->ftokens++;

	toks_daemon_token_reset(token);

	ntapi->zw_close(token->client.hinstance);

	ntapi->zw_set_event(

		token->client.halert,

		0);

	do {

		(void)0;

	} while (toks_daemon_unqueue(dctx));

	return NT_STATUS_SUCCESS;

}