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

/*  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"

int32_t toks_service_ioctl(

	struct toks_driver_ctx *  dctx,

	uint32_t                  ctlcode,

	uintptr_t *               data,

	void *                    any)

{

	int32_t			status;

	void *			hport;

	const char *		refstr;

	size_t			reflen;

	struct _nt_tty_sync_msg	msg;

	hport = toks_get_driver_hservice(dctx);

	refstr = toks_get_driver_refstr(dctx);

	reflen = 0;

	ntapi->tt_aligned_block_memset(

		&msg,0,sizeof(msg));

	msg.header.msg_type	= NT_LPC_NEW_MESSAGE;

	msg.header.data_size	= sizeof(msg.data);

	msg.header.msg_size	= sizeof(msg);

	msg.data.ttyinfo.opcode	= TOKS_DAEMON_IOCTL;

	msg.data.ttyinfo.opdata	= ctlcode;

	if (refstr && (reflen = toks_strlen(refstr))) {

		if (reflen > sizeof(msg.data.syncinfo.ipcsvc.keys.key))

			reflen = sizeof(msg.data.syncinfo.ipcsvc.keys.key);

		ntapi->tt_generic_memcpy(

			msg.data.syncinfo.ipcsvc.keys.key,

			refstr,reflen);

	}

	switch (ctlcode) {

		case TOKS_IOCTL_SET_TOKEN_COUNT:

		case TOKS_IOCTL_SET_LOG_LEVEL:

			msg.data.ttyinfo.exarg = (void *)(intptr_t)*data;

			break;

		default:

			(void)any;

			break;

	}

	if ((status = ntapi->zw_request_wait_reply_port(hport,&msg,&msg)))

		return status;

	else if (msg.data.ttyinfo.status)

		return msg.data.ttyinfo.status;

	switch (ctlcode) {

		case TOKS_IOCTL_GET_TOKEN_COUNT:

		case TOKS_IOCTL_GET_LOG_LEVEL:

			*data = (int)(intptr_t)msg.data.ttyinfo.exarg;

			break;

		default:

			(void)any;

			break;

	}

	return NT_STATUS_SUCCESS;

}